Advertisement
ddto

Untitled

Aug 22nd, 2024
85
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.97 KB | None | 0 0
  1. // ------------ HEADER FILE ---------------------
  2.  
  3. #pragma once
  4.  
  5. #include <cstdint>  
  6. #include <unistd.h>
  7. #include <stdlib.h>
  8. #include <sys/socket.h>
  9. #include <sys/un.h>
  10. #include <cstring>
  11. #include <errno.h>
  12. #include <format>
  13. #include <string>
  14.  
  15. #include "Log.h"
  16.  
  17. #define CLIENT_SOCKET_PATH "test_program.client"
  18. #define SERVER_SOCKET_PATH "test_program.server"
  19. #define MSG_BUFFER_SIZE 256  
  20.  
  21. class Client
  22. {
  23. public:
  24.     void Init();
  25.     void Send(int sum);
  26.  
  27.     ~Client();
  28.  
  29. private:
  30.     struct sockaddr_un m_clientAddr;
  31.     struct sockaddr_un m_serverAddr;
  32.     int m_clientSock = -1,
  33.         m_len = -1,
  34.         m_rc = -1;
  35.     bool m_connected = false;
  36.     char m_buffer[MSG_BUFFER_SIZE] = {0};
  37. };
  38.  
  39. // -------------------- CPP FILE ----------------------
  40.  
  41. #include "Client.h"
  42.  
  43. void Client::Init()
  44. {
  45.     Log::GetInstance().Trace("Begin client initialization");
  46.  
  47.     memset(&m_clientAddr, 0, sizeof(m_clientAddr));
  48.     m_clientSock = socket(AF_UNIX, SOCK_STREAM, 0);
  49.     if(m_clientSock == -1)
  50.     {
  51.         Log::GetInstance().Trace(std::format("Error opening client socket: {}", strerror(errno)));
  52.         return;
  53.     }
  54.     else Log::GetInstance().Trace("Opened client socket");
  55.  
  56.     m_clientAddr.sun_family = AF_UNIX;
  57.     strcpy(m_clientAddr.sun_path, CLIENT_SOCKET_PATH);
  58.     m_len = sizeof(m_clientAddr);
  59.     //unlink (CLIENT_SOCKET_PATH);
  60.  
  61.     m_rc = bind(m_clientSock, (struct sockaddr *)&m_clientAddr, m_len);
  62.     if(m_rc == -1)
  63.     {
  64.         Log::GetInstance().Trace(std::format("Error binding the client: {}", strerror(errno)));
  65.         close(m_clientSock);
  66.         return;
  67.     }
  68.     else Log::GetInstance().Trace("Bound client");
  69.  
  70.     m_serverAddr.sun_family = AF_UNIX;
  71.     strcpy(m_serverAddr.sun_path, SERVER_SOCKET_PATH);
  72.    
  73.     int check = access(SERVER_SOCKET_PATH, F_OK);
  74.     Log::GetInstance().Trace(std::to_string(check));
  75.  
  76.     m_rc = connect(m_clientSock, (struct sockaddr*)&m_serverAddr, m_len);
  77.     if(m_rc == -1)
  78.     {
  79.         Log::GetInstance().Trace(std::format("Error connecting the client: {}", strerror(errno)));
  80.         close(m_clientSock);
  81.         return;
  82.     }
  83.     else
  84.     {
  85.         m_connected = true;
  86.         Log::GetInstance().Trace("Connected client to the server");
  87.     }
  88. }
  89.  
  90. void Client::Send(int sum)
  91. {
  92.     if(!m_connected)
  93.     {
  94.         Log::GetInstance().Trace("Failed to send data: client is not connected");
  95.         return;
  96.     }
  97.  
  98.     memset(m_buffer, 0, sizeof(m_buffer));
  99.  
  100.     std::string sumStr = std::to_string(sum);
  101.     strcpy(m_buffer, sumStr.c_str());
  102.     m_rc = send(m_clientSock, m_buffer, sizeof(m_buffer), 0);
  103.     if(m_rc == -1)
  104.     {
  105.         Log::GetInstance().Trace(std::format("Error sending data from client: {}", strerror(errno)));
  106.         close(m_clientSock);
  107.         return;
  108.     }
  109.     Log::GetInstance().Trace(std::format("Client sent data ({}) to the server", sumStr));
  110. }
  111.  
  112. Client::~Client()
  113. {
  114.     close(m_clientSock);
  115.     remove(CLIENT_SOCKET_PATH);
  116. }
  117.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement