awesome-client.c

   1 /*
   2  * awesome-client.c - awesome client, communicate with socket
   3  *
   4  * Copyright © 2007-2008 Julien Danjou <julien@danjou.info>
   5  * Copyright © 2007 daniel@brinkers.de
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License along
  18  * with this program; if not, write to the Free Software Foundation, Inc.,
  19  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  */
  22
  23 #include <errno.h>
  24 #include <stdio.h>
  25 #include <unistd.h>
  26 #include <sys/socket.h>
  27 #include <sys/un.h>
  28
  29 #include "awesome-client.h"
  30 #include "common/awesome-version.h"
  31 #include "common/util.h"
  32
  33 /* GNU/Hurd workaround */
  34 #ifndef MSG_NOSIGNAL
  35 #define MSG_NOSIGNAL 0
  36 #endif
  37
  38 static int
  39 send_msg(char *msg, ssize_t msg_len)
  40 {
  41     struct sockaddr_un *addr;
  42     int csfd, ret_value = EXIT_SUCCESS;
  43     csfd = get_client_socket();
  44     addr = get_client_addr(getenv("DISPLAY"));
  45
  46     if(!addr || csfd < 0)
  47         return EXIT_FAILURE;
  48
  49     if(sendto(csfd, msg, msg_len, MSG_NOSIGNAL,
  50               (const struct sockaddr *) addr, sizeof(struct sockaddr_un)) == -1)
  51     {
  52         switch (errno)
  53         {
  54           case ENOENT:
  55               warn("Can't write to %s\n", addr->sun_path);
  56               break;
  57           default:
  58               perror("error sending datagram");
  59          }
  60          ret_value = errno;
  61     }
  62
  63     close(csfd);
  64
  65     p_delete(&addr);
  66     return ret_value;
  67 }
  68
  69
  70 /** Print help and exit(2) with given exit_code.
  71  */
  72 static void __attribute__ ((noreturn))
  73 exit_help(int exit_code)
  74 {
  75     FILE *outfile = (exit_code == EXIT_SUCCESS) ? stdout : stderr;
  76     fprintf(outfile, "Usage: awesome-client [--version|--help]\n"
  77             "In normal operation, give no parameters and issue commands "
  78             "on standard input.\n");
  79     exit(exit_code);
  80 }
  81
  82
  83 int
  84 main(int argc, char *argv[])
  85 {
  86     char buf[1024], *msg;
  87     int ret_value = EXIT_SUCCESS;
  88     ssize_t len, msg_len = 1;
  89
  90     if (argc < 2)
  91     {
  92         /* no args to parse, nothing to do */
  93     }
  94     else if (argc == 2)
  95     {
  96         if(!a_strcmp("-v", argv[1]) || !a_strcmp("--version", argv[1]))
  97             eprint_version("awesome-client");
  98         else if(!a_strcmp("-h", argv[1]) || !a_strcmp("--help", argv[1]))
  99             exit_help(EXIT_SUCCESS);
 100     }
 101     else
 102     {
 103         exit_help(EXIT_SUCCESS);
 104     }
 105
 106     msg = p_new(char, 1);
 107     while(fgets(buf, sizeof(buf), stdin))
 108     {
 109         len = a_strlen(buf);
 110         if (len < 2 && msg_len > 1)
 111         {
 112             ret_value = send_msg(msg, msg_len);
 113             p_delete(&msg);
 114             if (ret_value != EXIT_SUCCESS)
 115                 return ret_value;
 116             msg = p_new(char, 1);
 117             msg_len = 1;
 118         }
 119         else if (len > 1)
 120         {
 121             msg_len += len;
 122             p_realloc(&msg, msg_len);
 123             a_strncat(msg, msg_len, buf, len);
 124         }
 125     }
 126
 127     if (msg_len > 1)
 128         ret_value = send_msg(msg, msg_len);
 129
 130     p_delete(&msg);
 131
 132     return ret_value;
 133 }
 134 // vim: filetype=c:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=80