audioc.c: Always free IO buffer.
[paraslash.git] / audioc.c
1 /*
2 * Copyright (C) 2005-2012 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 <regex.h>
10 #include <sys/types.h>
11 #include <stdbool.h>
12 #include <signal.h>
13
14 #include "audioc.cmdline.h"
15 #include "para.h"
16 #include "error.h"
17 #include "net.h"
18 #include "string.h"
19 #include "fd.h"
20 #include "version.h"
21
22 INIT_AUDIOC_ERRLISTS;
23
24 /** The gengetopt structure containing command line args. */
25 static struct audioc_args_info conf;
26 static char *socket_name;
27
28
29 static int loglevel;
30 INIT_STDERR_LOGGING(loglevel);
31
32 static char *concat_args(unsigned argc, char * const *argv)
33 {
34 int i;
35 char *buf = NULL;
36
37 for (i = 0; i < argc; i++) {
38 buf = para_strcat(buf, argv[i]);
39 if (i != argc - 1)
40 buf = para_strcat(buf, "\n");
41 }
42 return buf;
43 }
44
45 #ifdef HAVE_READLINE
46 #include "list.h"
47 #include "sched.h"
48 #include "buffer_tree.h"
49 #include "interactive.h"
50 #include "audiod_completion.h"
51
52 static struct sched sched;
53
54 struct audioc_task {
55 int fd;
56 struct btr_node *btrn;
57 struct task task;
58 };
59
60 static struct i9e_completer audiod_completers[];
61
62 I9E_DUMMY_COMPLETER(cycle);
63 I9E_DUMMY_COMPLETER(off);
64 I9E_DUMMY_COMPLETER(on);
65 I9E_DUMMY_COMPLETER(sb);
66 I9E_DUMMY_COMPLETER(tasks);
67 I9E_DUMMY_COMPLETER(term);
68
69 static void help_completer(struct i9e_completion_info *ci,
70 struct i9e_completion_result *result)
71 {
72 result->matches = i9e_complete_commands(ci->word, audiod_completers);
73 }
74
75 static void stat_completer(struct i9e_completion_info *ci,
76 struct i9e_completion_result *cr)
77 {
78 char *sia[] = {STATUS_ITEM_ARRAY NULL};
79 char *opts[] = {"-p", NULL};
80
81 if (ci->word_num <= 2 && ci->word && ci->word[0] == '-')
82 i9e_complete_option(opts, ci, cr);
83 else
84 i9e_extract_completions(ci->word, sia, &cr->matches);
85 }
86
87 static void grab_completer(struct i9e_completion_info *ci,
88 struct i9e_completion_result *cr)
89 {
90 char *opts[] = {"-ms", "-ms", "-ma", "-p=", "-n=", "-o", NULL};
91 i9e_complete_option(opts, ci, cr);
92 }
93
94 static struct i9e_completer audiod_completers[] = {
95 AUDIOD_COMPLETERS
96 {.name = NULL}
97 };
98
99 static void audioc_pre_select(struct sched *s, struct task *t)
100 {
101 struct audioc_task *at = container_of(t, struct audioc_task, task);
102 int ret = btr_node_status(at->btrn, 0, BTR_NT_ROOT);
103
104 if (ret < 0)
105 sched_min_delay(s);
106 para_fd_set(at->fd, &s->rfds, &s->max_fileno);
107 }
108
109 static void audioc_post_select(struct sched *s, struct task *t)
110 {
111 char *buf = NULL;
112 struct audioc_task *at = container_of(t, struct audioc_task, task);
113 int ret = btr_node_status(at->btrn, 0, BTR_NT_ROOT);
114
115 if (ret < 0)
116 goto out;
117 if (!FD_ISSET(at->fd, &s->rfds))
118 return;
119 buf = para_malloc(conf.bufsize_arg);
120 ret = recv_bin_buffer(at->fd, buf, conf.bufsize_arg);
121 PARA_DEBUG_LOG("recv: %d\n", ret);
122 if (ret == 0)
123 ret = -E_AUDIOC_EOF;
124 if (ret < 0)
125 goto out;
126 btr_add_output(buf, ret, at->btrn);
127 return;
128 out:
129 if (ret < 0) {
130 free(buf);
131 btr_remove_node(at->btrn);
132 btr_free_node(at->btrn);
133 at->btrn = NULL;
134 close(at->fd);
135 }
136 t->error = ret;
137 }
138
139 static struct audioc_task audioc_task = {
140 .task = {
141 .pre_select = audioc_pre_select,
142 .post_select = audioc_post_select,
143 .status = "audioc task"
144 },
145 }, *at = &audioc_task;
146
147 static int audioc_i9e_line_handler(char *line)
148 {
149 char *args = NULL;
150 int ret;
151 if (!line || !*line)
152 return 0;
153
154 PARA_DEBUG_LOG("line: %s\n", line);
155 ret = create_argv(line, " ", &conf.inputs);
156 if (ret < 0)
157 return ret;
158 conf.inputs_num = ret;
159 args = concat_args(conf.inputs_num, conf.inputs);
160 free_argv(conf.inputs);
161 conf.inputs_num = 0; /* required for audioc_cmdline_parser_free() */
162 ret = connect_local_socket(socket_name);
163 if (ret < 0)
164 goto out;
165 at->fd = ret;
166 ret = mark_fd_nonblocking(at->fd);
167 if (ret < 0)
168 goto close;
169 ret = send_cred_buffer(at->fd, args);
170 if (ret < 0)
171 goto close;
172 free(args);
173 args = NULL;
174 at->btrn = btr_new_node(&(struct btr_node_description)
175 EMBRACE(.name = "audioc line handler"));
176 at->task.error = 0;
177 register_task(&sched, &at->task);
178 i9e_attach_to_stdout(at->btrn);
179 return 1;
180 close:
181 close(at->fd);
182 out:
183 free(args);
184 return ret;
185 }
186
187 __noreturn static void interactive_session(void)
188 {
189 int ret;
190 char *history_file;
191 struct sigaction act;
192 struct i9e_client_info ici = {
193 .fds = {0, 1, 2},
194 .prompt = "para_audioc> ",
195 .line_handler = audioc_i9e_line_handler,
196 .loglevel = loglevel,
197 .completers = audiod_completers,
198 };
199 PARA_NOTICE_LOG("\n%s\n", VERSION_TEXT("audioc"));
200 if (conf.history_file_given)
201 history_file = para_strdup(conf.history_file_arg);
202 else {
203 char *home = para_homedir();
204 history_file = make_message("%s/.paraslash/audioc.history",
205 home);
206 free(home);
207 }
208 ici.history_file = history_file;
209
210 act.sa_handler = i9e_signal_dispatch;
211 sigemptyset(&act.sa_mask);
212 act.sa_flags = 0;
213 sigaction(SIGINT, &act, NULL);
214 sched.select_function = i9e_select;
215
216 sched.default_timeout.tv_sec = 1;
217 ret = i9e_open(&ici, &sched);
218 if (ret < 0)
219 goto out;
220 para_log = i9e_log;
221 ret = schedule(&sched);
222 i9e_close();
223 para_log = stderr_log;
224 out:
225 free(history_file);
226 audioc_cmdline_parser_free(&conf);
227 if (ret < 0)
228 PARA_ERROR_LOG("%s\n", para_strerror(-ret));
229 exit(ret < 0? EXIT_FAILURE : EXIT_SUCCESS);
230 }
231
232 __noreturn static void print_completions(void)
233 {
234 int ret = i9e_print_completions(audiod_completers);
235 exit(ret <= 0? EXIT_FAILURE : EXIT_SUCCESS);
236 }
237
238 #else /* HAVE_READLINE */
239
240 __noreturn static void interactive_session(void)
241 {
242 PARA_EMERG_LOG("interactive sessions not available\n");
243 exit(EXIT_FAILURE);
244 }
245
246 __noreturn static void print_completions(void)
247 {
248 PARA_EMERG_LOG("command completion not available\n");
249 exit(EXIT_FAILURE);
250 }
251
252 #endif /* HAVE_READLINE */
253
254 static char *configfile_exists(void)
255 {
256 static char *config_file;
257 struct stat statbuf;
258
259 if (!config_file) {
260 char *home = para_homedir();
261 config_file = make_message("%s/.paraslash/audioc.conf", home);
262 free(home);
263 }
264 if (!stat(config_file, &statbuf))
265 return config_file;
266 return NULL;
267 }
268
269 /**
270 * The client program to connect to para_audiod.
271 *
272 * \param argc Usual argument count.
273 * \param argv Usual argument vector.
274 *
275 * It creates a temporary local socket in order to communicate with para_audiod.
276 * Authentication consists in sending a ucred buffer that contains the user id.
277 *
278 * Any output received through the local socket is sent to stdout.
279 *
280 * \return EXIT_SUCCESS or EXIT_FAILURE.
281 *
282 * \sa send_cred_buffer(), para_audioc(1), para_audiod(1).
283 */
284 int main(int argc, char *argv[])
285 {
286 int ret = -E_AUDIOC_SYNTAX, fd;
287 char *cf, *buf = NULL, *args;
288 size_t bufsize;
289
290 if (audioc_cmdline_parser(argc, argv, &conf))
291 goto out;
292 HANDLE_VERSION_FLAG("audioc", conf);
293 cf = configfile_exists();
294 if (cf) {
295 struct audioc_cmdline_parser_params params = {
296 .override = 0,
297 .initialize = 0,
298 .check_required = 0,
299 .check_ambiguity = 0
300 };
301 ret = audioc_cmdline_parser_config_file(cf, &conf, &params);
302 free(cf);
303 if (ret) {
304 fprintf(stderr, "parse error in config file\n");
305 exit(EXIT_FAILURE);
306 }
307 }
308 loglevel = get_loglevel_by_name(conf.loglevel_arg);
309 if (conf.socket_given)
310 socket_name = para_strdup(conf.socket_arg);
311 else {
312 char *hn = para_hostname();
313 socket_name = make_message("/var/paraslash/audiod_socket.%s",
314 hn);
315 free(hn);
316 }
317
318 if (conf.complete_given)
319 print_completions();
320
321 if (conf.inputs_num == 0)
322 interactive_session();
323 args = concat_args(conf.inputs_num, conf.inputs);
324
325 ret = connect_local_socket(socket_name);
326 free(socket_name);
327 if (ret < 0) {
328 PARA_EMERG_LOG("failed to connect to local socket\n");
329 goto out;
330 }
331 fd = ret;
332 ret = send_cred_buffer(fd, args);
333 if (ret < 0)
334 goto out;
335 bufsize = conf.bufsize_arg;
336 buf = para_malloc(bufsize);
337 ret = mark_fd_blocking(STDOUT_FILENO);
338 if (ret < 0)
339 goto out;
340 do {
341 size_t n = ret = recv_bin_buffer(fd, buf, bufsize);
342 if (ret <= 0)
343 break;
344 ret = write_all(STDOUT_FILENO, buf, &n);
345 } while (ret >= 0);
346 out:
347 free(buf);
348 if (ret < 0)
349 PARA_ERROR_LOG("%s\n", para_strerror(-ret));
350 return ret < 0? EXIT_FAILURE : EXIT_SUCCESS;
351 }