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 #define _GNU_SOURCE
  24
  25 #include <errno.h>
  26 #include <stdio.h>
  27 #include <unistd.h>
  28 #include <sys/socket.h>
  29 #include <sys/un.h>
  30 #include <fcntl.h>
  31
  32 #include <stdio.h>
  33 #include <readline/readline.h>
  34 #include <readline/history.h>
  35
  36 #include "common/socket.h"
  37 #include "common/version.h"
  38 #include "common/util.h"
  39
  40 /* GNU/Hurd workaround */
  41 #ifndef MSG_NOSIGNAL
  42 #define MSG_NOSIGNAL 0
  43 #endif
  44
  45 struct sockaddr_un *addr;
  46 int csfd;
  47 char *display;
  48
  49 /** Initialize the client and server socket connections.
  50  * If something goes wrong, preserves errno.
  51  * \return 0 if everything worked, 1 otherwise.
  52  */
  53 static bool
  54 sockets_init(void)
  55 {
  56     if((csfd = socket_getclient()) < 0)
  57         return false;
  58
  59     if(!(addr = socket_getaddr(display)))
  60         return false;
  61
  62     if(connect(csfd, addr, sizeof(struct sockaddr_un)) == -1)
  63         return false;
  64
  65     return true;
  66 }
  67
  68 /** Close the client and server socket connections.
  69  */
  70 static void
  71 sockets_close(void)
  72 {
  73     close(csfd);
  74     p_delete(&addr);
  75 }
  76
  77 /** Reconnect sockets.
  78  */
  79 static void
  80 sockets_reconnect(void)
  81 {
  82     warn("connection lost, reconnecting…");
  83     sockets_close();
  84     sockets_init();
  85 }
  86
  87 /** Send a message to awesome.
  88  * \param msg The message.
  89  * \param msg_len The message length.
  90  * \return The errno of sendto().
  91  */
  92 static int
  93 send_msg(const char *msg, ssize_t msg_len)
  94 {
  95     if(send(csfd, msg, msg_len, MSG_NOSIGNAL | MSG_EOR) == -1)
  96     {
  97         switch (errno)
  98         {
  99           case ENOENT:
 100             warn("can't write to %s", addr->sun_path);
 101             break;
 102           case EPIPE:
 103           case ENOTCONN:
 104           case ECONNRESET:
 105             sockets_reconnect();
 106             return send_msg(msg, msg_len);
 107           default:
 108             warn("error sending packet: %s", strerror(errno));
 109         }
 110         return errno;
 111     }
 112
 113     return EXIT_SUCCESS;
 114 }
 115
 116
 117 /** Recieve a message from awesome.
 118  */
 119 static void
 120 recv_msg(void)
 121 {
 122     ssize_t r;
 123     char buf[1024];
 124     int try = 10;
 125
 126     while(try)
 127     {
 128         r = recv(csfd, buf, sizeof(buf) - 1, MSG_TRUNC | MSG_DONTWAIT);
 129         if (r < 0)
 130         {
 131             if(errno != EAGAIN)
 132                 return warn("error recieving from UNIX domain socket: %s", strerror(errno));
 133             try--;
 134         }
 135         else
 136             break;
 137         usleep(100000);
 138     }
 139
 140     if(!try)
 141         sockets_reconnect();
 142     else if(r > 0)
 143     {
 144         buf[r] = '\0';
 145         puts(buf);
 146     }
 147 }
 148
 149
 150 /** Print help and exit(2) with given exit_code.
 151  * \param exit_code The exit code.
 152  * \return Never return.
 153  */
 154 static void __attribute__ ((noreturn))
 155 exit_help(int exit_code)
 156 {
 157     FILE *outfile = (exit_code == EXIT_SUCCESS) ? stdout : stderr;
 158     fprintf(outfile, "Usage: awesome-client [--version|--help]\n"
 159             "In normal operation, give no parameters and issue commands "
 160             "on standard input.\n");
 161     exit(exit_code);
 162 }
 163
 164 /** Main function of awesome-client.
 165  * \param argc Number of args.
 166  * \param argv Args array.
 167  * \return Value returned by send_msg().
 168  */
 169 int
 170 main(int argc, char **argv)
 171 {
 172     char buf[1024], *msg, *prompt;
 173     int ret_value = EXIT_SUCCESS;
 174     ssize_t len, msg_len = 1;
 175
 176     if(argc == 2)
 177     {
 178         if(!a_strcmp("-v", argv[1]) || !a_strcmp("--version", argv[1]))
 179             eprint_version("awesome-client");
 180         else if(!a_strcmp("-h", argv[1]) || !a_strcmp("--help", argv[1]))
 181             exit_help(EXIT_SUCCESS);
 182     }
 183     else if(argc > 2)
 184         exit_help(EXIT_SUCCESS);
 185
 186     display = getenv("DISPLAY");
 187
 188     if (!sockets_init())
 189     {
 190         warn("can't connect to UNIX domain socket: %s", strerror(errno));
 191         return EXIT_FAILURE;
 192     }
 193
 194     if(isatty(STDIN_FILENO))
 195     {
 196         asprintf(&prompt, "awesome@%s%% ", display ? display : "unknown");
 197         while((msg = readline(prompt)))
 198             if((msg_len = a_strlen(msg)))
 199             {
 200                 add_history (msg);
 201                 p_realloc(&msg, msg_len + 2);
 202                 msg[msg_len] = '\n';
 203                 msg[msg_len + 1] = '\0';
 204                 if(send_msg(msg, msg_len + 2) == EXIT_SUCCESS)
 205                     recv_msg();
 206                 p_delete(&msg);
 207             }
 208     }
 209     else
 210     {
 211         msg = p_new(char, 1);
 212         while(fgets(buf, sizeof(buf), stdin))
 213         {
 214             len = a_strlen(buf);
 215             if(len < 2 && msg_len > 1)
 216             {
 217                 ret_value = send_msg(msg, msg_len);
 218                 p_delete(&msg);
 219                 if (ret_value != EXIT_SUCCESS)
 220                     return ret_value;
 221                 msg = p_new(char, 1);
 222                 msg_len = 1;
 223             }
 224             else if (len > 1)
 225             {
 226                 msg_len += len;
 227                 p_realloc(&msg, msg_len);
 228                 a_strncat(msg, msg_len, buf, len);
 229             }
 230         }
 231         if(msg_len > 1)
 232         {
 233             if((ret_value = send_msg(msg, msg_len)) == EXIT_SUCCESS)
 234                 recv_msg();
 235         }
 236         p_delete(&msg);
 237     }
 238
 239     sockets_close();
 240     return ret_value;
 241 }
 242
 243 // vim: filetype=c:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=80