dccp_send: Fix a fd leak.
[paraslash.git] / audioc.c
1 /*
2  * Copyright (C) 2005-2008 Andre Noll <maan@systemlinux.org>
3  *
4  * Licensed under the GPL v2. For licencing details see COPYING.
5  */
6
7 /** \file audioc.c the client program used to connect to para_audiod */
8
9 #include <sys/types.h>
10 #include <dirent.h>
11
12 #include "audioc.cmdline.h"
13 #include "para.h"
14 #include "error.h"
15 #include "net.h"
16 #include "string.h"
17 #include "fd.h"
18
19 INIT_AUDIOC_ERRLISTS;
20
21 /** the gengetopt structure containing command line args */
22 struct audioc_args_info conf;
23
24 INIT_STDERR_LOGGING(conf.loglevel_arg);
25
26 static char *concat_args(unsigned argc, char * const *argv)
27 {
28         int i; char *buf = NULL;
29         for (i = 0; i < argc; i++) {
30                 buf = para_strcat(buf, argv[i]);
31                 if (i != argc - 1)
32                         buf = para_strcat(buf, "\n");
33         }
34         return buf;
35 }
36
37 static char *configfile_exists(void)
38 {
39         static char *config_file;
40         struct stat statbuf;
41
42
43         if (!config_file) {
44                 char *home = para_homedir();
45                 config_file = make_message("%s/.paraslash/audioc.conf", home);
46                 free(home);
47         }
48         if (!stat(config_file, &statbuf))
49                 return config_file;
50         return NULL;
51 }
52
53 /**
54  * the client program to connect to para_audiod
55  *
56  * \param argc usual argument count
57  * \param argv usual argument vector
58  *
59  * It creates a temporary local socket in order to communicate with para_audiod.
60  * Authentication consists in sending a ucred buffer that contains the user id.
61  *
62  * Any output received through the local socket is sent to stdout.
63  *
64  * \return EXIT_SUCCESS or EXIT_FAILURE
65  *
66  * \sa send_cred_buffer(), para_audioc(1), para_audiod(1).
67  */
68 int main(int argc, char *argv[])
69 {
70         int ret = -E_AUDIOC_SYNTAX, fd;
71         char *cf, *buf = NULL, *args;
72         size_t bufsize, loaded = 0;
73
74         if (audioc_cmdline_parser(argc, argv, &conf))
75                 goto out;
76         HANDLE_VERSION_FLAG("audioc", conf);
77         cf = configfile_exists();
78         if (cf) {
79                 struct audioc_cmdline_parser_params params = {
80                         .override = 0,
81                         .initialize = 0,
82                         .check_required = 0,
83                         .check_ambiguity = 0
84                 };
85                 if (audioc_cmdline_parser_config_file(cf, &conf, &params)) {
86                         fprintf(stderr, "parse error in config file\n");
87                         exit(EXIT_FAILURE);
88                 }
89         }
90         args = conf.inputs_num?
91                 concat_args(conf.inputs_num, conf.inputs) :
92                 para_strdup("stat");
93         bufsize = conf.bufsize_arg;
94         buf = para_malloc(bufsize);
95
96         if (conf.socket_given) {
97                 ret = create_remote_socket(conf.socket_arg);
98         } else {
99                 char *hn = para_hostname(),
100                      *socket_name = make_message("/var/paraslash/audiod_socket.%s", hn);
101
102                 ret = create_remote_socket(socket_name);
103                 free(hn);
104                 free(socket_name);
105         }
106         if (ret < 0)
107                 goto out;
108         fd = ret;
109
110         ret = send_cred_buffer(fd, args);
111         if (ret < 0)
112                 goto out;
113         for (;;) {
114                 int max_fileno = -1, check_write = 0;
115                 ssize_t len;
116                 fd_set rfd, wfd;
117                 FD_ZERO(&rfd);
118                 FD_ZERO(&wfd);
119                 if (loaded < bufsize)
120                         para_fd_set(fd, &rfd, &max_fileno);
121                 if (loaded > 0) {
122                         para_fd_set(STDOUT_FILENO, &wfd, &max_fileno);
123                         check_write = 1;
124                 }
125                 ret = -E_AUDIOC_OVERRUN;
126                 if (max_fileno < 0)
127                         goto out;
128                 ret = para_select(max_fileno + 1, &rfd, &wfd, NULL);
129                 if (ret < 0)
130                         goto out;
131                 if (loaded < bufsize && FD_ISSET(fd, &rfd)) {
132                         len = recv_bin_buffer(fd, buf + loaded,
133                                 bufsize - loaded);
134                         if (len <= 0) {
135                                 ret = len < 0? -E_AUDIOC_READ : 0;
136                                 goto out;
137                         }
138                         loaded += len;
139                 }
140                 if (check_write && FD_ISSET(STDOUT_FILENO, &wfd)) {
141                         ret = write(STDOUT_FILENO, buf, loaded);
142                         if (ret < 0) {
143                                 ret = -E_AUDIOC_WRITE;
144                                 goto out;
145                         }
146                         loaded -= ret;
147                 }
148         }
149 out:
150         if (!ret && loaded && buf)
151                 ret = write(STDOUT_FILENO, buf, loaded);
152         if (ret < 0)
153                 PARA_ERROR_LOG("%s\n", para_strerror(-ret));
154         return ret < 0? EXIT_FAILURE : EXIT_SUCCESS;
155 }