code.delx.au - pulseaudio/blob - src/daemon/main.c

   1 /* $Id$ */
   2
   3 /***
   4   This file is part of polypaudio.
   5
   6   polypaudio is free software; you can redistribute it and/or modify
   7   it under the terms of the GNU Lesser General Public License as published
   8   by the Free Software Foundation; either version 2 of the License,
   9   or (at your option) any later version.
  10
  11   polypaudio is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with polypaudio; if not, write to the Free Software
  18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  19   USA.
  20 ***/
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <unistd.h>
  27 #include <errno.h>
  28 #include <string.h>
  29 #include <stdlib.h>
  30 #include <stdio.h>
  31 #include <signal.h>
  32 #include <stddef.h>
  33 #include <assert.h>
  34 #include <ltdl.h>
  35 #include <limits.h>
  36 #include <fcntl.h>
  37 #include <unistd.h>
  38 #include <sys/types.h>
  39 #include <liboil/liboil.h>
  40
  41 #ifdef HAVE_SYS_IOCTL_H
  42 #include <sys/ioctl.h>
  43 #endif
  44
  45 #ifdef HAVE_LIBWRAP
  46 #include <syslog.h>
  47 #include <tcpd.h>
  48 #endif
  49
  50 #include "../polypcore/winsock.h"
  51
  52 #include <polyp/mainloop.h>
  53 #include <polyp/mainloop-signal.h>
  54
  55 #include <polypcore/core.h>
  56 #include <polypcore/memblock.h>
  57 #include <polypcore/module.h>
  58 #include <polypcore/cli-command.h>
  59 #include <polypcore/log.h>
  60 #include <polypcore/util.h>
  61 #include <polypcore/sioman.h>
  62 #include <polypcore/xmalloc.h>
  63 #include <polypcore/cli-text.h>
  64 #include <polypcore/pid.h>
  65 #include <polypcore/namereg.h>
  66
  67 #include "cmdline.h"
  68 #include "cpulimit.h"
  69 #include "daemon-conf.h"
  70 #include "dumpmodules.h"
  71 #include "caps.h"
  72
  73 #ifdef HAVE_LIBWRAP
  74 /* Only one instance of these variables */
  75 int allow_severity = LOG_INFO;
  76 int deny_severity = LOG_WARNING;
  77 #endif
  78
  79 #ifdef OS_IS_WIN32
  80
  81 static void message_cb(pa_mainloop_api*a, pa_defer_event *e, void *userdata) {
  82     MSG msg;
  83
  84     while(PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {
  85         if (msg.message == WM_QUIT)
  86             raise(SIGTERM);
  87         else {
  88             TranslateMessage(&msg);
  89             DispatchMessage(&msg);
  90         }
  91     }
  92 }
  93
  94 #endif
  95
  96 static void signal_callback(pa_mainloop_api*m, PA_GCC_UNUSED pa_signal_event *e, int sig, void *userdata) {
  97     pa_log_info(__FILE__": Got signal %s.", pa_strsignal(sig));
  98
  99     switch (sig) {
 100 #ifdef SIGUSR1
 101         case SIGUSR1:
 102             pa_module_load(userdata, "module-cli", NULL);
 103             break;
 104 #endif
 105
 106 #ifdef SIGUSR2
 107         case SIGUSR2:
 108             pa_module_load(userdata, "module-cli-protocol-unix", NULL);
 109             break;
 110 #endif
 111
 112 #ifdef SIGHUP
 113         case SIGHUP: {
 114             char *c = pa_full_status_string(userdata);
 115             pa_log_notice("%s", c);
 116             pa_xfree(c);
 117             return;
 118         }
 119 #endif
 120
 121         case SIGINT:
 122         case SIGTERM:
 123         default:
 124             pa_log_info(__FILE__": Exiting.");
 125             m->quit(m, 1);
 126             break;
 127     }
 128 }
 129
 130 static void close_pipe(int p[2]) {
 131     if (p[0] != -1)
 132         close(p[0]);
 133     if (p[1] != -1)
 134         close(p[1]);
 135     p[0] = p[1] = -1;
 136 }
 137
 138 static void set_random_seed(void) {
 139     unsigned int seed = 0;
 140
 141 #ifdef RANDOM_DEVICE
 142     int fd;
 143
 144     if ((fd = open(RANDOM_DEVICE, O_RDONLY)) >= 0) {
 145         ssize_t r;
 146
 147         if ((r = pa_loop_read(fd, &seed, sizeof(seed))) < 0 || (size_t) r != sizeof(seed)) {
 148             pa_log_error(__FILE__": failed to read entropy from '"RANDOM_DEVICE"'");
 149             seed += (unsigned int) time(NULL);
 150         }
 151
 152         close(fd);
 153     }
 154 #else
 155     seed = (unsigned int) time(NULL);
 156 #endif
 157
 158     srand(seed);
 159 }
 160
 161 int main(int argc, char *argv[]) {
 162     pa_core *c;
 163     pa_strbuf *buf = NULL;
 164     pa_daemon_conf *conf;
 165     pa_mainloop *mainloop;
 166
 167     char *s;
 168     int r, retval = 1, d = 0;
 169     int daemon_pipe[2] = { -1, -1 };
 170     int suid_root;
 171     int valid_pid_file = 0;
 172
 173 #ifdef HAVE_GETUID
 174     gid_t gid = (gid_t) -1;
 175 #endif
 176
 177 #ifdef OS_IS_WIN32
 178     pa_defer_event *defer;
 179 #endif
 180
 181     pa_limit_caps();
 182
 183 #ifdef HAVE_GETUID
 184     suid_root = getuid() != 0 && geteuid() == 0;
 185
 186     if (suid_root && (pa_own_uid_in_group("realtime", &gid) <= 0 || gid >= 1000)) {
 187         pa_log_warn(__FILE__": WARNING: called SUID root, but not in group 'realtime'.");
 188         pa_drop_root();
 189     }
 190 #else
 191     suid_root = 0;
 192 #endif
 193
 194     LTDL_SET_PRELOADED_SYMBOLS();
 195
 196     r = lt_dlinit();
 197     assert(r == 0);
 198
 199 #ifdef OS_IS_WIN32
 200     {
 201         WSADATA data;
 202         WSAStartup(MAKEWORD(2, 0), &data);
 203     }
 204 #endif
 205
 206     set_random_seed();
 207
 208     pa_log_set_ident("polypaudio");
 209
 210     conf = pa_daemon_conf_new();
 211
 212     if (pa_daemon_conf_load(conf, NULL) < 0)
 213         goto finish;
 214
 215     if (pa_daemon_conf_env(conf) < 0)
 216         goto finish;
 217
 218     if (pa_cmdline_parse(conf, argc, argv, &d) < 0) {
 219         pa_log(__FILE__": failed to parse command line.");
 220         goto finish;
 221     }
 222
 223     pa_log_set_maximal_level(conf->log_level);
 224     pa_log_set_target(conf->auto_log_target ? PA_LOG_STDERR : conf->log_target, NULL);
 225
 226     if (conf->high_priority && conf->cmd == PA_CMD_DAEMON)
 227         pa_raise_priority();
 228
 229     pa_drop_caps();
 230
 231     if (suid_root)
 232         pa_drop_root();
 233
 234     if (conf->dl_search_path)
 235         lt_dlsetsearchpath(conf->dl_search_path);
 236
 237     switch (conf->cmd) {
 238         case PA_CMD_DUMP_MODULES:
 239             pa_dump_modules(conf, argc-d, argv+d);
 240             retval = 0;
 241             goto finish;
 242
 243         case PA_CMD_DUMP_CONF: {
 244             s = pa_daemon_conf_dump(conf);
 245             fputs(s, stdout);
 246             pa_xfree(s);
 247             retval = 0;
 248             goto finish;
 249         }
 250
 251         case PA_CMD_HELP :
 252             pa_cmdline_help(argv[0]);
 253             retval = 0;
 254             goto finish;
 255
 256         case PA_CMD_VERSION :
 257             printf(PACKAGE_NAME" "PACKAGE_VERSION"\n");
 258             retval = 0;
 259             goto finish;
 260
 261         case PA_CMD_CHECK: {
 262             pid_t pid;
 263
 264             if (pa_pid_file_check_running(&pid) < 0) {
 265                 pa_log_info(__FILE__": daemon not running");
 266             } else {
 267                 pa_log_info(__FILE__": daemon running as PID %u", pid);
 268                 retval = 0;
 269             }
 270
 271             goto finish;
 272
 273         }
 274         case PA_CMD_KILL:
 275
 276             if (pa_pid_file_kill(SIGINT, NULL) < 0)
 277                 pa_log(__FILE__": failed to kill daemon.");
 278             else
 279                 retval = 0;
 280
 281             goto finish;
 282
 283         default:
 284             assert(conf->cmd == PA_CMD_DAEMON);
 285     }
 286
 287     if (conf->daemonize) {
 288         pid_t child;
 289         int tty_fd;
 290
 291         if (pa_stdio_acquire() < 0) {
 292             pa_log(__FILE__": failed to acquire stdio.");
 293             goto finish;
 294         }
 295
 296 #ifdef HAVE_FORK
 297         if (pipe(daemon_pipe) < 0) {
 298             pa_log(__FILE__": failed to create pipe.");
 299             goto finish;
 300         }
 301
 302         if ((child = fork()) < 0) {
 303             pa_log(__FILE__": fork() failed: %s", strerror(errno));
 304             goto finish;
 305         }
 306
 307         if (child != 0) {
 308             /* Father */
 309
 310             close(daemon_pipe[1]);
 311             daemon_pipe[1] = -1;
 312
 313             if (pa_loop_read(daemon_pipe[0], &retval, sizeof(retval)) != sizeof(retval)) {
 314                 pa_log(__FILE__": read() failed: %s", strerror(errno));
 315                 retval = 1;
 316             }
 317
 318             if (retval)
 319                 pa_log(__FILE__": daemon startup failed.");
 320             else
 321                 pa_log_info(__FILE__": daemon startup successful.");
 322
 323             goto finish;
 324         }
 325
 326         close(daemon_pipe[0]);
 327         daemon_pipe[0] = -1;
 328 #endif
 329
 330         if (conf->auto_log_target)
 331             pa_log_set_target(PA_LOG_SYSLOG, NULL);
 332
 333 #ifdef HAVE_SETSID
 334         setsid();
 335 #endif
 336 #ifdef HAVE_SETPGID
 337         setpgid(0,0);
 338 #endif
 339
 340 #ifndef OS_IS_WIN32
 341         close(0);
 342         close(1);
 343         close(2);
 344
 345         open("/dev/null", O_RDONLY);
 346         open("/dev/null", O_WRONLY);
 347         open("/dev/null", O_WRONLY);
 348 #else
 349         FreeConsole();
 350 #endif
 351
 352 #ifdef SIGTTOU
 353         signal(SIGTTOU, SIG_IGN);
 354 #endif
 355 #ifdef SIGTTIN
 356         signal(SIGTTIN, SIG_IGN);
 357 #endif
 358 #ifdef SIGTSTP
 359         signal(SIGTSTP, SIG_IGN);
 360 #endif
 361
 362 #ifdef TIOCNOTTY
 363         if ((tty_fd = open("/dev/tty", O_RDWR)) >= 0) {
 364             ioctl(tty_fd, TIOCNOTTY, (char*) 0);
 365             close(tty_fd);
 366         }
 367 #endif
 368     }
 369
 370     chdir("/");
 371
 372     if (conf->use_pid_file) {
 373         if (pa_pid_file_create() < 0) {
 374             pa_log(__FILE__": pa_pid_file_create() failed.");
 375 #ifdef HAVE_FORK
 376             if (conf->daemonize)
 377                 pa_loop_write(daemon_pipe[1], &retval, sizeof(retval));
 378 #endif
 379             goto finish;
 380         }
 381
 382         valid_pid_file = 1;
 383     }
 384
 385 #ifdef SIGPIPE
 386     signal(SIGPIPE, SIG_IGN);
 387 #endif
 388
 389     mainloop = pa_mainloop_new();
 390     assert(mainloop);
 391
 392     c = pa_core_new(pa_mainloop_get_api(mainloop));
 393     assert(c);
 394
 395     r = pa_signal_init(pa_mainloop_get_api(mainloop));
 396     assert(r == 0);
 397     pa_signal_new(SIGINT, signal_callback, c);
 398     pa_signal_new(SIGTERM, signal_callback, c);
 399
 400 #ifdef SIGUSR1
 401     pa_signal_new(SIGUSR1, signal_callback, c);
 402 #endif
 403 #ifdef SIGUSR2
 404     pa_signal_new(SIGUSR2, signal_callback, c);
 405 #endif
 406 #ifdef SIGHUP
 407     pa_signal_new(SIGHUP, signal_callback, c);
 408 #endif
 409
 410 #ifdef OS_IS_WIN32
 411     defer = pa_mainloop_get_api(mainloop)->defer_new(pa_mainloop_get_api(mainloop), message_cb, NULL);
 412     assert(defer);
 413 #endif
 414
 415     if (conf->daemonize)
 416         c->running_as_daemon = 1;
 417
 418     oil_init();
 419
 420     r = pa_cpu_limit_init(pa_mainloop_get_api(mainloop));
 421     assert(r == 0);
 422
 423     buf = pa_strbuf_new();
 424     if (conf->default_script_file)
 425         r = pa_cli_command_execute_file(c, conf->default_script_file, buf, &conf->fail);
 426
 427     if (r >= 0)
 428         r = pa_cli_command_execute(c, conf->script_commands, buf, &conf->fail);
 429     pa_log_error("%s", s = pa_strbuf_tostring_free(buf));
 430     pa_xfree(s);
 431
 432     if (r < 0 && conf->fail) {
 433         pa_log(__FILE__": failed to initialize daemon.");
 434 #ifdef HAVE_FORK
 435         if (conf->daemonize)
 436             pa_loop_write(daemon_pipe[1], &retval, sizeof(retval));
 437 #endif
 438     } else if (!c->modules || pa_idxset_size(c->modules) == 0) {
 439         pa_log(__FILE__": daemon startup without any loaded modules, refusing to work.");
 440 #ifdef HAVE_FORK
 441         if (conf->daemonize)
 442             pa_loop_write(daemon_pipe[1], &retval, sizeof(retval));
 443 #endif
 444     } else {
 445
 446         retval = 0;
 447 #ifdef HAVE_FORK
 448         if (conf->daemonize)
 449             pa_loop_write(daemon_pipe[1], &retval, sizeof(retval));
 450 #endif
 451
 452         c->disallow_module_loading = conf->disallow_module_loading;
 453         c->exit_idle_time = conf->exit_idle_time;
 454         c->module_idle_time = conf->module_idle_time;
 455         c->scache_idle_time = conf->scache_idle_time;
 456         c->resample_method = conf->resample_method;
 457
 458         if (c->default_sink_name &&
 459             pa_namereg_get(c, c->default_sink_name, PA_NAMEREG_SINK, 1) == NULL) {
 460             pa_log_error("%s : Fatal error. Default sink name (%s) does not exist in name register.", __FILE__, c->default_sink_name);
 461             retval = 1;
 462         } else {
 463             pa_log_info(__FILE__": Daemon startup complete.");
 464             if (pa_mainloop_run(mainloop, &retval) < 0)
 465                 retval = 1;
 466             pa_log_info(__FILE__": Daemon shutdown initiated.");
 467         }
 468     }
 469
 470 #ifdef OS_IS_WIN32
 471     pa_mainloop_get_api(mainloop)->defer_free(defer);
 472 #endif
 473
 474     pa_core_free(c);
 475
 476     pa_cpu_limit_done();
 477     pa_signal_done();
 478     pa_mainloop_free(mainloop);
 479
 480     pa_log_info(__FILE__": Daemon terminated.");
 481
 482 finish:
 483
 484     if (conf)
 485         pa_daemon_conf_free(conf);
 486
 487     if (valid_pid_file)
 488         pa_pid_file_remove();
 489
 490     close_pipe(daemon_pipe);
 491
 492 #ifdef OS_IS_WIN32
 493     WSACleanup();
 494 #endif
 495
 496     lt_dlexit();
 497
 498     return retval;
 499 }