audioc.c

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