[disorder] / server / disorderd.c

/*
 * This file is part of DisOrder.
 * Copyright (C) 2004, 2005, 2006, 2007 Richard Kettlewell
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 * USA
 */

#include <config.h>
#include "types.h"

#include <stdio.h>
#include <getopt.h>
#include <pwd.h>
#include <grp.h>
#include <sys/types.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <sys/socket.h>
#include <time.h>
#include <locale.h>
#include <syslog.h>
#include <sys/time.h>
#include <pcre.h>
#include <fcntl.h>
#include <gcrypt.h>

#include "daemonize.h"
#include "event.h"
#include "log.h"
#include "configuration.h"
#include "rights.h"
#include "trackdb.h"
#include "queue.h"
#include "mem.h"
#include "play.h"
#include "server.h"
#include "state.h"
#include "syscalls.h"
#include "defs.h"
#include "user.h"
#include "mixer.h"
#include "eventlog.h"
#include "printf.h"

static ev_source *ev;

static void rescan_after(long offset);
static void dbgc_after(long offset);
static void volumecheck_after(long offset);

static const struct option options[] = {
  { "help", no_argument, 0, 'h' },
  { "version", no_argument, 0, 'V' },
  { "config", required_argument, 0, 'c' },
  { "debug", no_argument, 0, 'd' },
  { "foreground", no_argument, 0, 'f' },
  { "log", required_argument, 0, 'l' },
  { "pidfile", required_argument, 0, 'P' },
  { "wide-open", no_argument, 0, 'w' },
  { "syslog", no_argument, 0, 's' },
  { 0, 0, 0, 0 }
};

/* display usage message and terminate */
static void help(void) {
  xprintf("Usage:\n"
	  "  disorderd [OPTIONS]\n"
	  "Options:\n"
	  "  --help, -h               Display usage message\n"
	  "  --version, -V            Display version number\n"
	  "  --config PATH, -c PATH   Set configuration file\n"
	  "  --debug, -d              Turn on debugging\n"
	  "  --foreground, -f         Do not become a daemon\n"
	  "  --syslog, -s             Log to syslog even with -f\n"
	  "  --pidfile PATH, -P PATH  Leave a pidfile\n");
  xfclose(stdout);
  exit(0);
}

/* display version number and terminate */
static void version(void) {
  xprintf("%s", disorder_version_string);
  xfclose(stdout);
  exit(0);
}

/* SIGHUP callback */
static int handle_sighup(ev_source attribute((unused)) *ev_,
			 int attribute((unused)) sig,
			 void attribute((unused)) *u) {
  info("received SIGHUP");
  reconfigure(ev, 1);
  return 0;
}

/* fatal signals */

static int handle_sigint(ev_source attribute((unused)) *ev_,
			 int attribute((unused)) sig,
			 void attribute((unused)) *u) {
  info("received SIGINT");
  quit(ev);
}

static int handle_sigterm(ev_source attribute((unused)) *ev_,
			  int attribute((unused)) sig,
			  void attribute((unused)) *u) {
  info("received SIGTERM");
  quit(ev);
}

static int rescan_again(ev_source *ev_,
			const struct timeval attribute((unused)) *now,
			void attribute((unused)) *u) {
  trackdb_rescan(ev_, 1/*check*/);
  rescan_after(86400);
  return 0;
}

static void rescan_after(long offset) {
  struct timeval w;

  gettimeofday(&w, 0);
  w.tv_sec += offset;
  ev_timeout(ev, 0, &w, rescan_again, 0);
}

static int dbgc_again(ev_source attribute((unused)) *ev_,
		      const struct timeval attribute((unused)) *now,
		      void attribute((unused)) *u) {
  trackdb_gc();
  dbgc_after(60);
  return 0;
}

static void dbgc_after(long offset) {
  struct timeval w;

  gettimeofday(&w, 0);
  w.tv_sec += offset;
  ev_timeout(ev, 0, &w, dbgc_again, 0);
}

static int volumecheck_again(ev_source attribute((unused)) *ev_,
			     const struct timeval attribute((unused)) *now,
			     void attribute((unused)) *u) {
  int l, r;
  char lb[32], rb[32];

  if(!mixer_control(&l, &r, 0)) {
    if(l != volume_left || r != volume_right) {
      volume_left = l;
      volume_right = r;
      snprintf(lb, sizeof lb, "%d", l);
      snprintf(rb, sizeof rb, "%d", r);
      eventlog("volume", lb, rb, (char *)0);
    }
  }
  volumecheck_after(60);
  return 0;
}

static void volumecheck_after(long offset) {
  struct timeval w;

  gettimeofday(&w, 0);
  w.tv_sec += offset;
  ev_timeout(ev, 0, &w, volumecheck_again, 0);
}

/* We fix the path to include the bindir and sbindir we were installed into */
static void fix_path(void) {
  char *path = getenv("PATH");
  static char *newpath;
  /* static or libgc collects it! */

  if(!path)
    error(0, "PATH is not set at all!");

  if(*finkbindir && strcmp(finkbindir, "/"))
    /* We appear to be a finkized mac; include fink on the path in case the
     * tools we need are there. */
    byte_xasprintf(&newpath, "PATH=%s:%s:%s:%s", 
		   path, bindir, sbindir, finkbindir);
  else
    byte_xasprintf(&newpath, "PATH=%s:%s:%s", path, bindir, sbindir);
  putenv(newpath);
  info("%s", newpath); 
}

int main(int argc, char **argv) {
  int n, background = 1, logsyslog = 0;
  const char *pidfile = 0;

  set_progname(argv);
  mem_init();
  if(!setlocale(LC_CTYPE, "")) fatal(errno, "error calling setlocale");
  /* garbage-collect PCRE's memory */
  pcre_malloc = xmalloc;
  pcre_free = xfree;
  while((n = getopt_long(argc, argv, "hVc:dfP:Ns", options, 0)) >= 0) {
    switch(n) {
    case 'h': help();
    case 'V': version();
    case 'c': configfile = optarg; break;
    case 'd': debugging = 1; break;
    case 'f': background = 0; break;
    case 'P': pidfile = optarg; break;
    case 's': logsyslog = 1; break;
    case 'w': wideopen = 1; break;
    default: fatal(0, "invalid option");
    }
  }
  /* go into background if necessary */
  if(background)
    daemonize(progname, LOG_DAEMON, pidfile);
  else if(logsyslog) {
    /* If we're running under some kind of daemon supervisor then we may want
     * to log to syslog but not to go into background */
    openlog(progname, LOG_PID, LOG_DAEMON);
    log_default = &log_syslog;
  }
  info("process ID %lu", (unsigned long)getpid());
  fix_path();
  srand(time(0));			/* don't start the same every time */
  /* gcrypt initialization */
  gcry_control(GCRYCTL_INIT_SECMEM, 1);
  /* create event loop */
  ev = ev_new();
  if(ev_child_setup(ev)) fatal(0, "ev_child_setup failed");
  /* read config */
  if(config_read(1))
    fatal(0, "cannot read configuration");
  /* make sure the home directory exists and has suitable permissions */
  make_home();
  /* Start the speaker process (as root! - so it can choose its nice value) */
  speaker_setup(ev);
  /* set server nice value _after_ starting the speaker, so that they
   * are independently niceable */
  xnice(config->nice_server);
  /* change user */
  become_mortal();
  /* make sure we're not root, whatever the config says */
  if(getuid() == 0 || geteuid() == 0) fatal(0, "do not run as root");
  /* open a lockfile - we only want one copy of the server to run at once. */
  if(config->lock) {
    const char *lockfile;
    int lockfd;
    struct flock lock;

    lockfile = config_get_file("lock");
    if((lockfd = open(lockfile, O_RDWR|O_CREAT, 0600)) < 0)
      fatal(errno, "error opening %s", lockfile);
    cloexec(lockfd);
    memset(&lock, 0, sizeof lock);
    lock.l_type = F_WRLCK;
    lock.l_whence = SEEK_SET;
    if(fcntl(lockfd, F_SETLK, &lock) < 0)
      fatal(errno, "error locking %s", lockfile);
  }
  /* initialize database environment */
  trackdb_init(TRACKDB_NORMAL_RECOVER|TRACKDB_MAY_CREATE);
  trackdb_master(ev);
  /* install new config (calls trackdb_open()) */
  reconfigure(ev, 0);
  /* pull in old users */
  trackdb_old_users();
  /* create a root login */
  trackdb_create_root();
  /* re-read config if we receive a SIGHUP */
  if(ev_signal(ev, SIGHUP, handle_sighup, 0)) fatal(0, "ev_signal failed");
  /* exit on SIGINT/SIGTERM */
  if(ev_signal(ev, SIGINT, handle_sigint, 0)) fatal(0, "ev_signal failed");
  if(ev_signal(ev, SIGTERM, handle_sigterm, 0)) fatal(0, "ev_signal failed");
  /* ignore SIGPIPE */
  signal(SIGPIPE, SIG_IGN);
  /* Start a rescan straight away */
  trackdb_rescan(ev, 1/*check*/);
  /* We'll rescan again after a day */
  rescan_after(86400);
  /* periodically tidy up the database */
  dbgc_after(60);
  /* periodically check the volume */
  volumecheck_again(0, 0, 0);
  /* set initial state */
  add_random_track();
  play(ev);
  /* enter the event loop */
  n = ev_run(ev);
  /* if we exit the event loop, something must have gone wrong */
  fatal(errno, "ev_run returned %d", n);
}

/*
Local Variables:
c-basic-offset:2
comment-column:40
fill-column:79
End:
*/
Commit	Line	Data
460b9539	1	/*
460b9539	2	* This file is part of DisOrder.
36be7e6a	3	* Copyright (C) 2004, 2005, 2006, 2007 Richard Kettlewell
460b9539	4	*
	5	* This program is free software; you can redistribute it and/or modify
	6	* it under the terms of the GNU General Public License as published by
	7	* the Free Software Foundation; either version 2 of the License, or
	8	* (at your option) any later version.
	9	*
	10	* This program is distributed in the hope that it will be useful, but
	11	* WITHOUT ANY WARRANTY; without even the implied warranty of
	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	13	* General Public License for more details.
	14	*
	15	* You should have received a copy of the GNU General Public License
	16	* along with this program; if not, write to the Free Software
	17	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
	18	* USA
	19	*/
	20
	21	#include <config.h>
6d2d327c	22	#include "types.h"
460b9539	23
	24	#include <stdio.h>
	25	#include <getopt.h>
	26	#include <pwd.h>
	27	#include <grp.h>
	28	#include <sys/types.h>
	29	#include <errno.h>
	30	#include <stdlib.h>
	31	#include <string.h>
	32	#include <unistd.h>
	33	#include <signal.h>
	34	#include <sys/socket.h>
	35	#include <time.h>
	36	#include <locale.h>
	37	#include <syslog.h>
	38	#include <sys/time.h>
	39	#include <pcre.h>
	40	#include <fcntl.h>
b12be54a	41	#include <gcrypt.h>
460b9539	42
	43	#include "daemonize.h"
	44	#include "event.h"
	45	#include "log.h"
	46	#include "configuration.h"
5df73aeb	47	#include "rights.h"
460b9539	48	#include "trackdb.h"
	49	#include "queue.h"
	50	#include "mem.h"
	51	#include "play.h"
	52	#include "server.h"
	53	#include "state.h"
	54	#include "syscalls.h"
	55	#include "defs.h"
	56	#include "user.h"
	57	#include "mixer.h"
	58	#include "eventlog.h"
e83d0967	59	#include "printf.h"
460b9539	60
	61	static ev_source *ev;
	62
	63	static void rescan_after(long offset);
	64	static void dbgc_after(long offset);
	65	static void volumecheck_after(long offset);
	66
	67	static const struct option options[] = {
	68	{ "help", no_argument, 0, 'h' },
	69	{ "version", no_argument, 0, 'V' },
	70	{ "config", required_argument, 0, 'c' },
	71	{ "debug", no_argument, 0, 'd' },
	72	{ "foreground", no_argument, 0, 'f' },
	73	{ "log", required_argument, 0, 'l' },
	74	{ "pidfile", required_argument, 0, 'P' },
18e6d6e6	75	{ "wide-open", no_argument, 0, 'w' },
e83d0967	76	{ "syslog", no_argument, 0, 's' },
460b9539	77	{ 0, 0, 0, 0 }
	78	};
	79
	80	/* display usage message and terminate */
	81	static void help(void) {
	82	xprintf("Usage:\n"
	83	" disorderd [OPTIONS]\n"
	84	"Options:\n"
	85	" --help, -h Display usage message\n"
	86	" --version, -V Display version number\n"
	87	" --config PATH, -c PATH Set configuration file\n"
	88	" --debug, -d Turn on debugging\n"
	89	" --foreground, -f Do not become a daemon\n"
e83d0967	90	" --syslog, -s Log to syslog even with -f\n"
460b9539	91	" --pidfile PATH, -P PATH Leave a pidfile\n");
	92	xfclose(stdout);
	93	exit(0);
	94	}
	95
	96	/* display version number and terminate */
	97	static void version(void) {
a05e4467	98	xprintf("%s", disorder_version_string);
460b9539	99	xfclose(stdout);
	100	exit(0);
	101	}
	102
	103	/* SIGHUP callback */
	104	static int handle_sighup(ev_source attribute((unused)) *ev_,
	105	int attribute((unused)) sig,
	106	void attribute((unused)) *u) {
	107	info("received SIGHUP");
	108	reconfigure(ev, 1);
	109	return 0;
	110	}
	111
	112	/* fatal signals */
	113
	114	static int handle_sigint(ev_source attribute((unused)) *ev_,
	115	int attribute((unused)) sig,
	116	void attribute((unused)) *u) {
	117	info("received SIGINT");
	118	quit(ev);
	119	}
	120
	121	static int handle_sigterm(ev_source attribute((unused)) *ev_,
	122	int attribute((unused)) sig,
	123	void attribute((unused)) *u) {
	124	info("received SIGTERM");
	125	quit(ev);
	126	}
	127
	128	static int rescan_again(ev_source *ev_,
	129	const struct timeval attribute((unused)) *now,
	130	void attribute((unused)) *u) {
ffac51d7	131	trackdb_rescan(ev_, 1/check/);
460b9539	132	rescan_after(86400);
	133	return 0;
	134	}
	135
	136	static void rescan_after(long offset) {
	137	struct timeval w;
	138
	139	gettimeofday(&w, 0);
	140	w.tv_sec += offset;
	141	ev_timeout(ev, 0, &w, rescan_again, 0);
	142	}
	143
	144	static int dbgc_again(ev_source attribute((unused)) *ev_,
	145	const struct timeval attribute((unused)) *now,
	146	void attribute((unused)) *u) {
	147	trackdb_gc();
	148	dbgc_after(60);
	149	return 0;
	150	}
	151
	152	static void dbgc_after(long offset) {
	153	struct timeval w;
	154
	155	gettimeofday(&w, 0);
	156	w.tv_sec += offset;
	157	ev_timeout(ev, 0, &w, dbgc_again, 0);
	158	}
	159
	160	static int volumecheck_again(ev_source attribute((unused)) *ev_,
	161	const struct timeval attribute((unused)) *now,
	162	void attribute((unused)) *u) {
	163	int l, r;
	164	char lb[32], rb[32];
	165
	166	if(!mixer_control(&l, &r, 0)) {
	167	if(l != volume_left \|\| r != volume_right) {
	168	volume_left = l;
	169	volume_right = r;
	170	snprintf(lb, sizeof lb, "%d", l);
	171	snprintf(rb, sizeof rb, "%d", r);
	172	eventlog("volume", lb, rb, (char *)0);
	173	}
	174	}
	175	volumecheck_after(60);
	176	return 0;
	177	}
	178
	179	static void volumecheck_after(long offset) {
	180	struct timeval w;
	181
	182	gettimeofday(&w, 0);
	183	w.tv_sec += offset;
	184	ev_timeout(ev, 0, &w, volumecheck_again, 0);
	185	}
	186
e83d0967 RK	187	/* We fix the path to include the bindir and sbindir we were installed into */
	188	static void fix_path(void) {
	189	char *path = getenv("PATH");
e99d42b1	190	static char *newpath;
e99d42b1	191	/* static or libgc collects it! */
e83d0967 RK	192
	193	if(!path)
	194	error(0, "PATH is not set at all!");
	195
31780899	196	if(*finkbindir && strcmp(finkbindir, "/"))
e83d0967 RK	197	/* We appear to be a finkized mac; include fink on the path in case the
	198	* tools we need are there. */
	199	byte_xasprintf(&newpath, "PATH=%s:%s:%s:%s",
	200	path, bindir, sbindir, finkbindir);
	201	else
	202	byte_xasprintf(&newpath, "PATH=%s:%s:%s", path, bindir, sbindir);
	203	putenv(newpath);
	204	info("%s", newpath);
	205	}
	206
	207	int main(int argc, char **argv) {
	208	int n, background = 1, logsyslog = 0;
460b9539	209	const char *pidfile = 0;
460b9539	210
460b9539	211	set_progname(argv);
320598d4	212	mem_init();
460b9539	213	if(!setlocale(LC_CTYPE, "")) fatal(errno, "error calling setlocale");
	214	/* garbage-collect PCRE's memory */
	215	pcre_malloc = xmalloc;
	216	pcre_free = xfree;
e83d0967	217	while((n = getopt_long(argc, argv, "hVc:dfP:Ns", options, 0)) >= 0) {
460b9539	218	switch(n) {
	219	case 'h': help();
	220	case 'V': version();
	221	case 'c': configfile = optarg; break;
	222	case 'd': debugging = 1; break;
	223	case 'f': background = 0; break;
	224	case 'P': pidfile = optarg; break;
e83d0967	225	case 's': logsyslog = 1; break;
18e6d6e6	226	case 'w': wideopen = 1; break;
460b9539	227	default: fatal(0, "invalid option");
	228	}
	229	}
	230	/* go into background if necessary */
	231	if(background)
	232	daemonize(progname, LOG_DAEMON, pidfile);
e83d0967 RK	233	else if(logsyslog) {
	234	/* If we're running under some kind of daemon supervisor then we may want
	235	* to log to syslog but not to go into background */
	236	openlog(progname, LOG_PID, LOG_DAEMON);
	237	log_default = &log_syslog;
	238	}
460b9539	239	info("process ID %lu", (unsigned long)getpid());
e83d0967	240	fix_path();
460b9539	241	srand(time(0)); /* don't start the same every time */
b12be54a RK	242	/* gcrypt initialization */
b12be54a RK	243	gcry_control(GCRYCTL_INIT_SECMEM, 1);
460b9539	244	/* create event loop */
	245	ev = ev_new();
	246	if(ev_child_setup(ev)) fatal(0, "ev_child_setup failed");
	247	/* read config */
c00fce3a	248	if(config_read(1))
460b9539	249	fatal(0, "cannot read configuration");
659d87e8 RK	250	/* make sure the home directory exists and has suitable permissions */
659d87e8 RK	251	make_home();
460b9539	252	/* Start the speaker process (as root! - so it can choose its nice value) */
	253	speaker_setup(ev);
	254	/* set server nice value _after_ starting the speaker, so that they
	255	* are independently niceable */
	256	xnice(config->nice_server);
	257	/* change user */
	258	become_mortal();
	259	/* make sure we're not root, whatever the config says */
	260	if(getuid() == 0 \|\| geteuid() == 0) fatal(0, "do not run as root");
	261	/* open a lockfile - we only want one copy of the server to run at once. */
	262	if(config->lock) {
	263	const char *lockfile;
	264	int lockfd;
9ba0617a RK	265	struct flock lock;
9ba0617a RK	266
460b9539	267	lockfile = config_get_file("lock");
	268	if((lockfd = open(lockfile, O_RDWR\|O_CREAT, 0600)) < 0)
	269	fatal(errno, "error opening %s", lockfile);
	270	cloexec(lockfd);
	271	memset(&lock, 0, sizeof lock);
	272	lock.l_type = F_WRLCK;
	273	lock.l_whence = SEEK_SET;
	274	if(fcntl(lockfd, F_SETLK, &lock) < 0)
	275	fatal(errno, "error locking %s", lockfile);
	276	}
	277	/* initialize database environment */
a745dd43	278	trackdb_init(TRACKDB_NORMAL_RECOVER\|TRACKDB_MAY_CREATE);
460b9539	279	trackdb_master(ev);
f0feb22e	280	/* install new config (calls trackdb_open()) */
460b9539	281	reconfigure(ev, 0);
f0feb22e RK	282	/* pull in old users */
	283	trackdb_old_users();
	284	/* create a root login */
	285	trackdb_create_root();
460b9539	286	/* re-read config if we receive a SIGHUP */
	287	if(ev_signal(ev, SIGHUP, handle_sighup, 0)) fatal(0, "ev_signal failed");
	288	/* exit on SIGINT/SIGTERM */
	289	if(ev_signal(ev, SIGINT, handle_sigint, 0)) fatal(0, "ev_signal failed");
	290	if(ev_signal(ev, SIGTERM, handle_sigterm, 0)) fatal(0, "ev_signal failed");
	291	/* ignore SIGPIPE */
	292	signal(SIGPIPE, SIG_IGN);
dbeb3844 RK	293	/* Start a rescan straight away */
	294	trackdb_rescan(ev, 1/check/);
	295	/* We'll rescan again after a day */
	296	rescan_after(86400);
460b9539	297	/* periodically tidy up the database */
	298	dbgc_after(60);
	299	/* periodically check the volume */
104ae93a	300	volumecheck_again(0, 0, 0);
460b9539	301	/* set initial state */
	302	add_random_track();
	303	play(ev);
	304	/* enter the event loop */
	305	n = ev_run(ev);
	306	/* if we exit the event loop, something must have gone wrong */
	307	fatal(errno, "ev_run returned %d", n);
	308	}
	309
	310	/*
	311	Local Variables:
	312	c-basic-offset:2
	313	comment-column:40
e83d0967	314	fill-column:79
460b9539	315	End:
460b9539	316	*/