pulseaudio/src/utils/pacmd.c

/***
  This file is part of PulseAudio.

  Copyright 2004-2006 Lennart Poettering

  PulseAudio is free software; you can redistribute it and/or modify
  it under the terms of the GNU Lesser General Public License as published
  by the Free Software Foundation; either version 2.1 of the License,
  or (at your option) any later version.

  PulseAudio 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 Lesser General Public License
  along with PulseAudio; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  USA.
***/

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <assert.h>
#include <signal.h>
#include <sys/poll.h>
#include <sys/socket.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <sys/un.h>
#include <locale.h>

#include <pulse/error.h>
#include <pulse/util.h>
#include <pulse/xmalloc.h>
#include <pulse/i18n.h>

#include <pulsecore/macro.h>
#include <pulsecore/core-util.h>
#include <pulsecore/log.h>
#include <pulsecore/pid.h>

int main(int argc, char*argv[]) {

    enum {
        WATCH_STDIN,
        WATCH_STDOUT,
        WATCH_SOCKET,
        N_WATCH
    };

    pid_t pid ;
    int fd = -1;
    int ret = 1, i;
    struct sockaddr_un sa;
    char ibuf[PIPE_BUF], obuf[PIPE_BUF];
    size_t ibuf_index, ibuf_length, obuf_index, obuf_length;
    char *cli;
    pa_bool_t ibuf_eof, obuf_eof, ibuf_closed, obuf_closed;
    struct pollfd pollfd[N_WATCH];

    setlocale(LC_ALL, "");
    bindtextdomain(GETTEXT_PACKAGE, PULSE_LOCALEDIR);

    if (pa_pid_file_check_running(&pid, "pulseaudio") < 0) {
        pa_log(_("No PulseAudio daemon running, or not running as session daemon."));
        goto fail;
    }

    if ((fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0) {
        pa_log(_("socket(PF_UNIX, SOCK_STREAM, 0): %s"), strerror(errno));
        goto fail;
    }

    pa_zero(sa);
    sa.sun_family = AF_UNIX;

    if (!(cli = pa_runtime_path("cli")))
        goto fail;

    pa_strlcpy(sa.sun_path, cli, sizeof(sa.sun_path));
    pa_xfree(cli);

    for (i = 0; i < 5; i++) {
        int r;

        if ((r = connect(fd, (struct sockaddr*) &sa, sizeof(sa))) < 0 && (errno != ECONNREFUSED && errno != ENOENT)) {
            pa_log(_("connect(): %s"), strerror(errno));
            goto fail;
        }

        if (r >= 0)
            break;

        if (pa_pid_file_kill(SIGUSR2, NULL, "pulseaudio") < 0) {
            pa_log(_("Failed to kill PulseAudio daemon."));
            goto fail;
        }

        pa_msleep(300);
    }

    if (i >= 5) {
        pa_log(_("Daemon not responding."));
        goto fail;
    }

    ibuf_index = ibuf_length = obuf_index = obuf_length = 0;
    ibuf_eof = obuf_eof = ibuf_closed = obuf_closed = FALSE;

    if (argc > 1) {
        for (i = 1; i < argc; i++) {
            size_t k;

            k = PA_MIN(sizeof(ibuf) - ibuf_length, strlen(argv[i]));
            memcpy(ibuf + ibuf_length, argv[i], k);
            ibuf_length += k;

            if (ibuf_length < sizeof(ibuf)) {
                ibuf[ibuf_length] = i < argc-1 ? ' ' : '\n';
                ibuf_length++;
            }
        }

        ibuf_eof = TRUE;
    }

    pa_zero(pollfd);

    pollfd[WATCH_STDIN].fd = STDIN_FILENO;
    pollfd[WATCH_STDOUT].fd = STDOUT_FILENO;
    pollfd[WATCH_SOCKET].fd = fd;

    for (;;) {
        if (ibuf_eof &&
            obuf_eof &&
            ibuf_length <= 0 &&
            obuf_length <= 0)
            break;

        pollfd[WATCH_STDIN].events = pollfd[WATCH_STDOUT].events = pollfd[WATCH_SOCKET].events = 0;

        if (obuf_length > 0)
            pollfd[WATCH_STDOUT].events |= POLLOUT;
        else if (!obuf_eof)
            pollfd[WATCH_SOCKET].events |= POLLIN;

        if (ibuf_length > 0)
            pollfd[WATCH_SOCKET].events |= POLLOUT;
        else if (!ibuf_eof)
            pollfd[WATCH_STDIN].events |= POLLIN;

        if (poll(pollfd, N_WATCH, -1) < 0) {

            if (errno == EINTR)
                continue;

            pa_log(_("poll(): %s"), strerror(errno));
            goto fail;
        }

        if (pollfd[WATCH_STDIN].revents & POLLIN) {
            ssize_t r;
            pa_assert(!ibuf_length);

            if ((r = pa_read(STDIN_FILENO, ibuf, sizeof(ibuf), NULL)) <= 0) {
                if (r < 0) {
                    pa_log(_("read(): %s"), strerror(errno));
                    goto fail;
                }

                ibuf_eof = TRUE;
            } else {
                ibuf_length = (size_t) r;
                ibuf_index = 0;
            }
        }

        if (pollfd[WATCH_SOCKET].revents & POLLIN) {
            ssize_t r;
            pa_assert(!obuf_length);

            if ((r = pa_read(fd, obuf, sizeof(obuf), NULL)) <= 0) {
                if (r < 0) {
                    pa_log(_("read(): %s"), strerror(errno));
                    goto fail;
                }

                obuf_eof = TRUE;
            } else {
                obuf_length = (size_t) r;
                obuf_index = 0;
            }
        }

        if (pollfd[WATCH_STDOUT].revents & POLLHUP) {
            obuf_eof = TRUE;
            obuf_length = 0;
        } else if (pollfd[WATCH_STDOUT].revents & POLLOUT) {
            ssize_t r;
            pa_assert(obuf_length);

            if ((r = pa_write(STDOUT_FILENO, obuf + obuf_index, obuf_length, NULL)) < 0) {
                pa_log(_("write(): %s"), strerror(errno));
                goto fail;
            }

            obuf_length -= (size_t) r;
            obuf_index += obuf_index;
        }

        if (pollfd[WATCH_SOCKET].revents & POLLHUP) {
            ibuf_eof = TRUE;
            ibuf_length = 0;
        } if (pollfd[WATCH_SOCKET].revents & POLLOUT) {
            ssize_t r;
            pa_assert(ibuf_length);

            if ((r = pa_write(fd, ibuf + ibuf_index, ibuf_length, NULL)) < 0) {
                pa_log(_("write(): %s"), strerror(errno));
                goto fail;
            }

            ibuf_length -= (size_t) r;
            ibuf_index += obuf_index;
        }

        if (ibuf_length <= 0 && ibuf_eof && !ibuf_closed) {
            pa_close(STDIN_FILENO);
            shutdown(fd, SHUT_WR);
            ibuf_closed = TRUE;
        }

        if (obuf_length <= 0 && obuf_eof && !obuf_closed) {
            shutdown(fd, SHUT_RD);
            pa_close(STDOUT_FILENO);
            obuf_closed = TRUE;
        }
    }

    ret = 0;

fail:
    if (fd >= 0)
        pa_close(fd);

    return ret;
}