cgproxy/socket_client.hpp

#ifndef SOCKET_CLIENT_H
#define SOCKET_CLIENT_H

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/un.h>
#include <unistd.h>
#include "common.hpp"

using namespace std;

namespace CGPROXY::SOCKET{

#define return_if_error(flag, msg)                                             \
  if (flag == -1) {                                                            \
    perror(msg);                                                               \
    status = CONN_ERROR;                                                       \
    close(sfd); \
    return;                                                                    \
  }

void send(const char *msg, int &status) {
  debug("send msg: %s", msg);
  status = UNKNOWN_ERROR;

  int flag;
  int sfd = socket(AF_UNIX, SOCK_STREAM, 0);

  struct sockaddr_un unix_socket;
  memset(&unix_socket, '\0', sizeof(struct sockaddr_un));
  unix_socket.sun_family = AF_UNIX;
  strncpy(unix_socket.sun_path, SOCKET_PATH, sizeof(unix_socket.sun_path) - 1);

  flag =
      connect(sfd, (struct sockaddr *)&unix_socket, sizeof(struct sockaddr_un));
  return_if_error(flag, "connect");

  int msg_len = strlen(msg);
  flag = write(sfd, &msg_len, sizeof(int));
  return_if_error(flag, "write length");
  flag = write(sfd, msg, msg_len * sizeof(char));
  return_if_error(flag, "write msg");

  flag = read(sfd, &status, sizeof(int));
  return_if_error(flag, "read return value");

  close(sfd);
}

void send(const string msg, int &status) {
  int msg_len = msg.length();
  char buff[msg_len];
  msg.copy(buff, msg_len, 0);
  buff[msg_len] = '\0';
  send(buff, status);
  debug("return status: %d", status);
}

}
#endif