1 /* Copyright (C) 2009-2014, Martin Johansson <martin@fatbob.nu>
2 Copyright (C) 2005-2014, Thorvald Natvig <thorvald@natvig.com>
6 Redistribution and use in source and binary forms, with or without
7 modification, are permitted provided that the following conditions
10 - Redistributions of source code must retain the above copyright notice,
11 this list of conditions and the following disclaimer.
12 - Redistributions in binary form must reproduce the above copyright notice,
13 this list of conditions and the following disclaimer in the documentation
14 and/or other materials provided with the distribution.
15 - Neither the name of the Developers nor the names of its contributors may
16 be used to endorse or promote products derived from this software without
17 specific prior written permission.
19 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
23 CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
24 EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
25 PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
26 PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
27 LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
28 NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
29 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34 #include <sys/types.h>
36 #include <sys/utsname.h>
44 #ifdef _POSIX_PRIORITY_SCHEDULING
45 #if (_POSIX_PRIORITY_SCHEDULING > 0)
46 #define POSIX_PRIORITY_SCHEDULING
59 char system_string[64], version_string[64];
63 void lockfile(const char *pidfile)
68 /* Don't use O_TRUNC here -- we want to leave the PID file
69 * unmodified if we cannot lock it.
71 lfp = open(pidfile, O_WRONLY|O_CREAT, 0640);
74 Log_fatal("Cannot open PID-file %s for writing", pidfile);
76 /* Try to lock the file. */
77 if (lockf(lfp, F_TLOCK, 0) < 0) {
80 if (errno == EACCES || errno == EAGAIN)
81 Log_fatal("PID file is locked -- uMurmur already running?");
83 Log_fatal("Cannot lock PID file: %s", strerror(errno));
86 /* Now that we locked the file, erase its contents. */
87 if (ftruncate(lfp, 0) < 0) {
89 Log_fatal("Cannot truncate PID file: %s", strerror(errno));
92 snprintf(str,16,"%d\n", getpid());
93 (void)write(lfp, str, strlen(str)); /* record pid to lockfile */
94 Log_info("PID-file: %s", pidfile);
96 /* If uMurmur ever starts to fork()+exec(), we don't want it to
97 * leak the fd to the forked process though. Set the close-on-exec
98 * flag to prevent leakage.
100 flags = fcntl(lfp, F_GETFD, 0);
102 fcntl(lfp, F_SETFD, (long) flags);
104 /* Don't close(lfp) here!
105 * We want the fd to remain opened so the lock is held until the
111 /* Drops privileges (if configured to do so). */
112 static void switch_user(void)
115 struct group *grp = NULL;
116 const char *username, *groupname;
119 username = getStrConf(USERNAME);
120 groupname = getStrConf(GROUPNAME);
123 /* It's an error to specify groupname
124 * but leave username empty.
127 Log_fatal("username missing");
133 pwd = getpwnam(username);
135 Log_fatal("Unknown user '%s'", username);
140 grp = getgrnam(groupname);
143 Log_fatal("Unknown group '%s'", groupname);
148 if (initgroups(pwd->pw_name, gid))
149 Log_fatal("initgroups() failed: %s", strerror(errno));
152 Log_fatal("setgid() failed: %s", strerror(errno));
154 if (setuid(pwd->pw_uid))
155 Log_fatal("setuid() failed: %s", strerror(errno));
160 Log_fatal("getgrgid() failed: %s", strerror(errno));
162 Log_info("Switch to user '%s' group '%s'", pwd->pw_name, grp->gr_name);
165 void signal_handler(int sig)
169 Log_info("HUP signal received.");
173 Log_info("TERM signal. Shutting down.");
184 return; /* already a daemon */
187 fprintf(stderr, "Fork error. Exiting\n");
188 exit(1); /* fork error */
191 exit(0); /* parent exits */
193 /* child (daemon) continues */
194 setsid(); /* obtain a new process group */
195 for (i = getdtablesize(); i >= 0; --i)
196 close(i); /* close all descriptors */
198 i = open("/dev/null",O_RDWR);
202 umask(027); /* set newly created file permissions */
207 #ifdef POSIX_PRIORITY_SCHEDULING
211 struct sched_param sp;
213 sp.sched_priority = sched_get_priority_min(SCHED_RR); /* Should suffice */
214 Log_info("Setting SCHED_RR prio %d", sp.sched_priority);
215 rc = sched_setscheduler(0, SCHED_RR, &sp);
217 Log_warn("Failed to set scheduler: %s", strerror(errno));
223 printf("uMurmur version %s ('%s'). Mumble protocol %d.%d.%d\n", UMURMUR_VERSION,
224 UMURMUR_CODENAME, PROTVER_MAJOR, PROTVER_MINOR, PROTVER_PATCH);
225 printf("Usage: umurmurd [-d] [-r] [-h] [-p <pidfile>] [-t] [-c <conf file>] [-a <addr>] [-b <port>]\n");
226 printf(" -d - Do not daemonize - run in foreground.\n");
227 #ifdef POSIX_PRIORITY_SCHEDULING
228 printf(" -r - Run with realtime priority\n");
230 printf(" -p <pidfile> - Write PID to this file\n");
231 printf(" -c <conf file> - Specify configuration file (default %s)\n", DEFAULT_CONFIG);
232 printf(" -t - Test config. Error message to stderr + non-zero exit code on error\n");
233 printf(" -a <address> - Bind to IP address\n");
234 printf(" -b <port> - Bind to port\n");
235 printf(" -h - Print this help\n");
239 int main(int argc, char **argv)
241 bool_t nodaemon = false;
242 #ifdef POSIX_PRIORITY_SCHEDULING
243 bool_t realtime = false;
245 bool_t testconfig = false;
246 char *conffile = NULL, *pidfile = NULL;
248 struct utsname utsbuf;
251 #ifdef POSIX_PRIORITY_SCHEDULING
252 while ((c = getopt(argc, argv, "drp:c:a:b:ht")) != EOF) {
254 while ((c = getopt(argc, argv, "dp:c:a:b:ht")) != EOF) {
267 bindport = atoi(optarg);
278 #ifdef POSIX_PRIORITY_SCHEDULING
284 fprintf(stderr, "Unrecognized option\n");
291 if (!Conf_ok(conffile))
297 /* Initialize the config subsystem early;
298 * switch_user() will need to read some config variables as well as logging.
302 /* Logging to terminal if not daemonizing, otherwise to syslog or log file.
312 /* Reopen log file. If user switch results in access denied, we catch
319 signal(SIGCHLD, SIG_IGN); /* ignore child */
320 signal(SIGTSTP, SIG_IGN); /* ignore tty signals */
321 signal(SIGTTOU, SIG_IGN);
322 signal(SIGTTIN, SIG_IGN);
323 signal(SIGPIPE, SIG_IGN);
324 signal(SIGHUP, signal_handler); /* catch hangup signal */
325 signal(SIGTERM, signal_handler); /* catch kill signal */
327 /* Build system string */
328 if (uname(&utsbuf) == 0) {
329 snprintf(system_string, 64, "%s %s", utsbuf.sysname, utsbuf.machine);
330 snprintf(version_string, 64, "%s", utsbuf.release);
333 snprintf(system_string, 64, "unknown unknown");
334 snprintf(version_string, 64, "unknown");
343 #ifdef POSIX_PRIORITY_SCHEDULING