bspc.c

   1 #include <stdlib.h>
   2 #ifdef __OpenBSD__
   3 #include <sys/types.h>
   4 #endif
   5 #include <sys/socket.h>
   6 #include <sys/un.h>
   7 #include <unistd.h>
   8 #include "helpers.h"
   9 #include "common.h"
  10
  11 int main(int argc, char *argv[])
  12 {
  13     int sock_fd;
  14     struct sockaddr_un sock_address;
  15     char msg[BUFSIZ];
  16     char rsp[BUFSIZ];
  17
  18     if (argc < 2)
  19         err("No arguments given.\n");
  20
  21     char *sp = getenv(SOCKET_ENV_VAR);
  22
  23     sock_address.sun_family = AF_UNIX;
  24     snprintf(sock_address.sun_path, sizeof(sock_address.sun_path), "%s", (sp == NULL ? DEFAULT_SOCKET_PATH : sp));
  25
  26     argc--, argv++;
  27     int msg_len = 0;
  28
  29     for (int offset = 0, rem = sizeof(msg), n = 0; argc > 0 && rem > 0; offset += n, rem -= n, argc--, argv++) {
  30         n = snprintf(msg + offset, rem, "%s%c", *argv, 0);
  31         msg_len += n;
  32     }
  33
  34     if ((sock_fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
  35         err("Failed to create the socket.\n");
  36
  37     if (connect(sock_fd, (struct sockaddr *) &sock_address, sizeof(sock_address)) == -1)
  38         err("Failed to connect to the socket.\n");
  39
  40     if (send(sock_fd, msg, msg_len, 0) == -1)
  41         err("Failed to send the data.\n");
  42
  43     int ret = EXIT_SUCCESS;
  44
  45     int n = recv(sock_fd, rsp, sizeof(rsp), 0);
  46     if (n == -1) {
  47         err("Failed to get the response.\n");
  48     } else if (n > 0) {
  49         if (n == 1 && rsp[0] == MESSAGE_FAILURE) {
  50             ret = EXIT_FAILURE;
  51         } else {
  52             rsp[n] = '\0';
  53             printf("%s\n", rsp);
  54         }
  55     }
  56
  57     if (sock_fd)
  58         close(sock_fd);
  59
  60     return ret;
  61 }