From 67e44b0f2c76a8c4913048fe3b8fe90116fa595b Mon Sep 17 00:00:00 2001 From: Max Kellermann Date: Wed, 7 Aug 2013 09:53:33 +0200 Subject: daemon: convert to C++ --- src/Daemon.cxx | 236 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 236 insertions(+) create mode 100644 src/Daemon.cxx (limited to 'src/Daemon.cxx') diff --git a/src/Daemon.cxx b/src/Daemon.cxx new file mode 100644 index 000000000..e14b8aee7 --- /dev/null +++ b/src/Daemon.cxx @@ -0,0 +1,236 @@ +/* + * Copyright (C) 2003-2013 The Music Player Daemon Project + * http://www.musicpd.org + * + * 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., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#include "config.h" +#include "Daemon.hxx" + +#include + +#include +#include +#include +#include +#include +#include +#include +#include + +#ifndef WIN32 +#include +#include +#include +#endif + +#undef G_LOG_DOMAIN +#define G_LOG_DOMAIN "daemon" + +#ifndef WIN32 + +/** the Unix user name which MPD runs as */ +static char *user_name; + +/** the Unix user id which MPD runs as */ +static uid_t user_uid = (uid_t)-1; + +/** the Unix group id which MPD runs as */ +static gid_t user_gid = (pid_t)-1; + +/** the absolute path of the pidfile */ +static char *pidfile; + +/* whether "group" conf. option was given */ +static bool had_group = false; + + +void +daemonize_kill(void) +{ + FILE *fp; + int pid, ret; + + if (pidfile == nullptr) + MPD_ERROR("no pid_file specified in the config file"); + + fp = fopen(pidfile, "r"); + if (fp == nullptr) + MPD_ERROR("unable to open pid file \"%s\": %s", + pidfile, g_strerror(errno)); + + if (fscanf(fp, "%i", &pid) != 1) { + MPD_ERROR("unable to read the pid from file \"%s\"", + pidfile); + } + fclose(fp); + + ret = kill(pid, SIGTERM); + if (ret < 0) + MPD_ERROR("unable to kill process %i: %s", + pid, g_strerror(errno)); + + exit(EXIT_SUCCESS); +} + +void +daemonize_close_stdin(void) +{ + close(STDIN_FILENO); + open("/dev/null", O_RDONLY); +} + +void +daemonize_set_user(void) +{ + if (user_name == nullptr) + return; + + /* set gid */ + if (user_gid != (gid_t)-1 && user_gid != getgid()) { + if (setgid(user_gid) == -1) { + MPD_ERROR("cannot setgid to %d: %s", + (int)user_gid, g_strerror(errno)); + } + } + +#ifdef _BSD_SOURCE + /* init suplementary groups + * (must be done before we change our uid) + */ + if (!had_group && initgroups(user_name, user_gid) == -1) { + g_warning("cannot init supplementary groups " + "of user \"%s\": %s", + user_name, g_strerror(errno)); + } +#endif + + /* set uid */ + if (user_uid != (uid_t)-1 && user_uid != getuid() && + setuid(user_uid) == -1) { + MPD_ERROR("cannot change to uid of user \"%s\": %s", + user_name, g_strerror(errno)); + } +} + +static void +daemonize_detach(void) +{ + /* flush all file handles before duplicating the buffers */ + + fflush(nullptr); + +#ifdef HAVE_DAEMON + + if (daemon(0, 1)) + MPD_ERROR("daemon() failed: %s", g_strerror(errno)); + +#elif defined(HAVE_FORK) + + /* detach from parent process */ + + switch (fork()) { + case -1: + MPD_ERROR("fork() failed: %s", g_strerror(errno)); + case 0: + break; + default: + /* exit the parent process */ + _exit(EXIT_SUCCESS); + } + + /* release the current working directory */ + + if (chdir("/") < 0) + MPD_ERROR("problems changing to root directory"); + + /* detach from the current session */ + + setsid(); + +#else + MPD_ERROR("no support for daemonizing"); +#endif + + g_debug("daemonized!"); +} + +void +daemonize(bool detach) +{ + FILE *fp = nullptr; + + if (pidfile != nullptr) { + /* do this before daemon'izing so we can fail gracefully if we can't + * write to the pid file */ + g_debug("opening pid file"); + fp = fopen(pidfile, "w+"); + if (!fp) { + MPD_ERROR("could not create pid file \"%s\": %s", + pidfile, g_strerror(errno)); + } + } + + if (detach) + daemonize_detach(); + + if (pidfile != nullptr) { + g_debug("writing pid file"); + fprintf(fp, "%lu\n", (unsigned long)getpid()); + fclose(fp); + } +} + +void +daemonize_init(const char *user, const char *group, const char *_pidfile) +{ + if (user) { + struct passwd *pwd = getpwnam(user); + if (pwd == nullptr) + MPD_ERROR("no such user \"%s\"", user); + + user_uid = pwd->pw_uid; + user_gid = pwd->pw_gid; + + user_name = g_strdup(user); + + /* this is needed by libs such as arts */ + g_setenv("HOME", pwd->pw_dir, true); + } + + if (group) { + struct group *grp = getgrnam(group); + if (grp == nullptr) + MPD_ERROR("no such group \"%s\"", group); + user_gid = grp->gr_gid; + had_group = true; + } + + + pidfile = g_strdup(_pidfile); +} + +void +daemonize_finish(void) +{ + if (pidfile != nullptr) + unlink(pidfile); + + g_free(user_name); + g_free(pidfile); +} + +#endif -- cgit v1.2.3