bspc.c

   1 /* Copyright (c) 2012-2014, Bastien Dejean
   2  * All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are met:
   6  *
   7  * 1. Redistributions of source code must retain the above copyright notice, this
   8  *    list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright notice,
  10  *    this list of conditions and the following disclaimer in the documentation
  11  *    and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  14  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  15  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  16  * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
  17  * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  18  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  19  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  20  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  21  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  22  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  23  *
  24  * The views and conclusions contained in the software and documentation are those
  25  * of the authors and should not be interpreted as representing official policies,
  26  * either expressed or implied, of the FreeBSD Project.
  27  */
  28
  29 #include <stdlib.h>
  30 #ifdef __OpenBSD__
  31 #include <sys/types.h>
  32 #endif
  33 #include <sys/socket.h>
  34 #include <sys/un.h>
  35 #include <unistd.h>
  36 #include <ctype.h>
  37 #include "helpers.h"
  38 #include "common.h"
  39
  40 int main(int argc, char *argv[])
  41 {
  42         int fd;
  43         struct sockaddr_un sock_address;
  44         char msg[BUFSIZ], rsp[BUFSIZ];
  45
  46         if (argc < 2)
  47                 err("No arguments given.\n");
  48
  49         sock_address.sun_family = AF_UNIX;
  50         char *sp = getenv(SOCKET_ENV_VAR);
  51         if (sp != NULL)
  52                 snprintf(sock_address.sun_path, sizeof(sock_address.sun_path), "%s", sp);
  53         else
  54                 snprintf(sock_address.sun_path, sizeof(sock_address.sun_path), SOCKET_PATH_TPL, getenv("DISPLAY"));
  55
  56         argc--, argv++;
  57         int msg_len = 0;
  58
  59         for (int offset = 0, rem = sizeof(msg), n = 0; argc > 0 && rem > 0; offset += n, rem -= n, argc--, argv++) {
  60                 n = snprintf(msg + offset, rem, "%s%c", *argv, 0);
  61                 msg_len += n;
  62         }
  63
  64         if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
  65                 err("Failed to create the socket.\n");
  66
  67         if (connect(fd, (struct sockaddr *) &sock_address, sizeof(sock_address)) == -1)
  68                 err("Failed to connect to the socket.\n");
  69
  70         if (send(fd, msg, msg_len, 0) == -1)
  71                 err("Failed to send the data.\n");
  72
  73         int ret = 0, nb;
  74         while ((nb = recv(fd, rsp, sizeof(rsp), 0)) > 0) {
  75                 if (nb == 1 && rsp[0] < MSG_LENGTH) {
  76                         ret = rsp[0];
  77                         if (ret == MSG_UNKNOWN) {
  78                                 warn("Unknown command.\n");
  79                         } else if (ret == MSG_SYNTAX) {
  80                                 warn("Invalid syntax.\n");
  81                         }
  82                 } else {
  83                         int end = MIN(nb, (int) sizeof(rsp) - 1);
  84                         rsp[end--] = '\0';
  85                         while (end >= 0 && isspace(rsp[end]))
  86                                 rsp[end--] = '\0';
  87                         printf("%s\n", rsp);
  88                         fflush(stdout);
  89                 }
  90         }
  91
  92         close(fd);
  93         return ret;
  94 }