awesome/awesome-client.c

/*
 * awesome-client.c - awesome client, communicate with socket
 *
 * Copyright © 2007-2008 Julien Danjou <julien@danjou.info>
 * Copyright © 2007 daniel@brinkers.de
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 */

#include <errno.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/un.h>

#include "common/socket.h"
#include "common/version.h"
#include "common/util.h"

/* GNU/Hurd workaround */
#ifndef MSG_NOSIGNAL
#define MSG_NOSIGNAL 0
#endif

static int
send_msg(char *msg, ssize_t msg_len)
{
    struct sockaddr_un *addr;
    int csfd, ret_value = EXIT_SUCCESS;
    csfd = get_client_socket();
    addr = get_client_addr(getenv("DISPLAY"));

    if(!addr || csfd < 0)
        return EXIT_FAILURE;

    if(sendto(csfd, msg, msg_len, MSG_NOSIGNAL,
              (const struct sockaddr *) addr, sizeof(struct sockaddr_un)) == -1)
    {
        switch (errno)
        {
          case ENOENT:
              warn("Can't write to %s\n", addr->sun_path);
              break;
          default:
              perror("error sending datagram");
         }
         ret_value = errno;
    }

    close(csfd);

    p_delete(&addr);
    return ret_value;
}


/** Print help and exit(2) with given exit_code.
 */
static void __attribute__ ((noreturn))
exit_help(int exit_code)
{
    FILE *outfile = (exit_code == EXIT_SUCCESS) ? stdout : stderr;
    fprintf(outfile, "Usage: awesome-client [--version|--help]\n"
	    "In normal operation, give no parameters and issue commands "
	    "on standard input.\n");
    exit(exit_code);
}


int
main(int argc, char *argv[])
{
    char buf[1024], *msg;
    int ret_value = EXIT_SUCCESS;
    ssize_t len, msg_len = 1;

    if (argc < 2)
    {
        /* no args to parse, nothing to do */
    }
    else if (argc == 2)
    {
        if(!a_strcmp("-v", argv[1]) || !a_strcmp("--version", argv[1]))
	    eprint_version("awesome-client");
        else if(!a_strcmp("-h", argv[1]) || !a_strcmp("--help", argv[1]))
            exit_help(EXIT_SUCCESS);
    }
    else
    {
        exit_help(EXIT_SUCCESS);
    }

    msg = p_new(char, 1);
    while(fgets(buf, sizeof(buf), stdin))
    {
        len = a_strlen(buf);
        if (len < 2 && msg_len > 1)
        {
            ret_value = send_msg(msg, msg_len);
            p_delete(&msg);
            if (ret_value != EXIT_SUCCESS)
                return ret_value;
            msg = p_new(char, 1);
            msg_len = 1;
        }
        else if (len > 1)
        {
            msg_len += len;
            p_realloc(&msg, msg_len);
            a_strncat(msg, msg_len, buf, len);
        }
    }

    if (msg_len > 1)
        ret_value = send_msg(msg, msg_len);

    p_delete(&msg);

    return ret_value;
}
// vim: filetype=c:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=80