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
58 #include "sharedmemory.h"
61 char system_string[64], version_string[64];
67 void lockfile(const char *pidfile)
72 /* Don't use O_TRUNC here -- we want to leave the PID file
73 * unmodified if we cannot lock it.
75 lfp = open(pidfile, O_WRONLY|O_CREAT, 0640);
78 Log_fatal("Cannot open PID-file %s for writing", pidfile);
80 /* Try to lock the file. */
81 if (lockf(lfp, F_TLOCK, 0) < 0) {
84 if (errno == EACCES || errno == EAGAIN)
85 Log_fatal("PID file is locked -- uMurmur already running?");
87 Log_fatal("Cannot lock PID file: %s", strerror(errno));
90 /* Now that we locked the file, erase its contents. */
91 if (ftruncate(lfp, 0) < 0) {
93 Log_fatal("Cannot truncate PID file: %s", strerror(errno));
96 snprintf(str,16,"%d\n", getpid());
97 (void)write(lfp, str, strlen(str)); /* record pid to lockfile */
98 Log_info("PID-file: %s", pidfile);
100 /* If uMurmur ever starts to fork()+exec(), we don't want it to
101 * leak the fd to the forked process though. Set the close-on-exec
102 * flag to prevent leakage.
104 flags = fcntl(lfp, F_GETFD, 0);
106 fcntl(lfp, F_SETFD, (long) flags);
108 /* Don't close(lfp) here!
109 * We want the fd to remain opened so the lock is held until the
115 /* Drops privileges (if configured to do so). */
116 static void switch_user(void)
119 struct group *grp = NULL;
120 const char *username, *groupname;
123 username = getStrConf(USERNAME);
124 groupname = getStrConf(GROUPNAME);
127 /* It's an error to specify groupname
128 * but leave username empty.
131 Log_fatal("username missing");
137 pwd = getpwnam(username);
139 Log_fatal("Unknown user '%s'", username);
144 grp = getgrnam(groupname);
147 Log_fatal("Unknown group '%s'", groupname);
152 if (initgroups(pwd->pw_name, gid))
153 Log_fatal("initgroups() failed: %s", strerror(errno));
156 Log_fatal("setgid() failed: %s", strerror(errno));
158 if (setuid(pwd->pw_uid))
159 Log_fatal("setuid() failed: %s", strerror(errno));
164 Log_fatal("getgrgid() failed: %s", strerror(errno));
166 Log_info("Switch to user '%s' group '%s'", pwd->pw_name, grp->gr_name);
169 void signal_handler(int sig)
173 Log_info("HUP signal received.");
177 Log_info("TERM signal. Shutting down.");
188 return; /* already a daemon */
191 fprintf(stderr, "Fork error. Exiting\n");
192 exit(1); /* fork error */
195 exit(0); /* parent exits */
197 /* child (daemon) continues */
198 setsid(); /* obtain a new process group */
199 for (i = getdtablesize(); i >= 0; --i)
200 close(i); /* close all descriptors */
202 i = open("/dev/null",O_RDWR);
206 umask(027); /* set newly created file permissions */
211 #ifdef POSIX_PRIORITY_SCHEDULING
215 struct sched_param sp;
217 sp.sched_priority = sched_get_priority_min(SCHED_RR); /* Should suffice */
218 Log_info("Setting SCHED_RR prio %d", sp.sched_priority);
219 rc = sched_setscheduler(0, SCHED_RR, &sp);
221 Log_warn("Failed to set scheduler: %s", strerror(errno));
227 printf("uMurmur version %s ('%s'). Mumble protocol %d.%d.%d\n", UMURMUR_VERSION,
228 UMURMUR_CODENAME, PROTVER_MAJOR, PROTVER_MINOR, PROTVER_PATCH);
229 printf("Usage: umurmurd [-d] [-r] [-h] [-p <pidfile>] [-t] [-c <conf file>] [-a <addr>] [-b <port>]\n");
230 printf(" -d - Do not daemonize - run in foreground.\n");
231 #ifdef POSIX_PRIORITY_SCHEDULING
232 printf(" -r - Run with realtime priority\n");
234 printf(" -p <pidfile> - Write PID to this file\n");
235 printf(" -c <conf file> - Specify configuration file (default %s)\n", DEFAULT_CONFIG);
236 printf(" -t - Test config. Error message to stderr + non-zero exit code on error\n");
237 printf(" -a <address> - Bind to IP address\n");
238 printf(" -A <address> - Bind to IPv6 address\n");
239 printf(" -b <port> - Bind to port\n");
240 printf(" -B <port> - Bind to port (IPv6)\n");
241 printf(" -h - Print this help\n");
245 int main(int argc, char **argv)
247 bool_t nodaemon = false;
248 #ifdef POSIX_PRIORITY_SCHEDULING
249 bool_t realtime = false;
251 bool_t testconfig = false;
252 char *conffile = NULL, *pidfile = NULL;
254 struct utsname utsbuf;
257 #ifdef POSIX_PRIORITY_SCHEDULING
258 while ((c = getopt(argc, argv, "drp:c:a:A:b:B:ht")) != EOF) {
260 while ((c = getopt(argc, argv, "dp:c:a:A:b:B:ht")) != EOF) {
276 bindport = atoi(optarg);
279 bindport6 = atoi(optarg);
290 #ifdef POSIX_PRIORITY_SCHEDULING
296 fprintf(stderr, "Unrecognized option\n");
303 if (!Conf_ok(conffile))
309 /* Initialize the config subsystem early;
310 * switch_user() will need to read some config variables as well as logging.
314 /* Logging to terminal if not daemonizing, otherwise to syslog or log file.
324 /* Reopen log file. If user switch results in access denied, we catch
331 signal(SIGCHLD, SIG_IGN); /* ignore child */
332 signal(SIGTSTP, SIG_IGN); /* ignore tty signals */
333 signal(SIGTTOU, SIG_IGN);
334 signal(SIGTTIN, SIG_IGN);
335 signal(SIGPIPE, SIG_IGN);
336 signal(SIGHUP, signal_handler); /* catch hangup signal */
337 signal(SIGTERM, signal_handler); /* catch kill signal */
339 /* Build system string */
340 if (uname(&utsbuf) == 0) {
341 snprintf(system_string, 64, "%s %s", utsbuf.sysname, utsbuf.machine);
342 snprintf(version_string, 64, "%s", utsbuf.release);
345 snprintf(system_string, 64, "unknown unknown");
346 snprintf(version_string, 64, "unknown");
355 #ifdef USE_SHAREDMEMORY_API
356 Sharedmemory_init( bindport, bindport6 );
359 #ifdef POSIX_PRIORITY_SCHEDULING
366 #ifdef USE_SHAREDMEMORY_API
367 Sharedmemory_deinit();