main: moved code to daemon.c

Moved changeToUser(), cleanUpPidFile(), killFromPidFile() to
daemon.c.  These are daemonization functions.
This commit is contained in:
Max Kellermann 2009-01-18 17:15:34 +01:00
parent 1f0dfb4407
commit 90b34f8e6f
3 changed files with 119 additions and 98 deletions

View File

@ -28,6 +28,46 @@
#include <sys/stat.h> #include <sys/stat.h>
#include <fcntl.h> #include <fcntl.h>
#ifndef WIN32
#include <signal.h>
#include <pwd.h>
#include <grp.h>
#endif
void
daemonize_kill(void)
{
#ifndef WIN32
FILE *fp;
struct config_param *param =
parseConfigFilePath(CONF_PID_FILE, 0);
int pid, ret;
if (param == NULL)
g_error("no pid_file specified in the config file");
fp = fopen(param->value, "r");
if (fp == NULL)
g_error("unable to open %s \"%s\": %s",
CONF_PID_FILE, param->value, g_strerror(errno));
if (fscanf(fp, "%i", &pid) != 1) {
g_error("unable to read the pid from file \"%s\"",
param->value);
}
fclose(fp);
ret = kill(pid, SIGTERM);
if (ret < 0)
g_error("unable to kill proccess %i: %s",
pid, g_strerror(errno));
exit(EXIT_SUCCESS);
#else
g_error("--kill is not available on WIN32");
#endif
}
void void
daemonize_close_stdin(void) daemonize_close_stdin(void)
{ {
@ -41,6 +81,50 @@ daemonize_close_stdin(void)
} }
} }
void
daemonize_set_user(void)
{
#ifndef WIN32
struct config_param *param = config_get_param(CONF_USER);
if (param && strlen(param->value)) {
/* get uid */
struct passwd *userpwd;
if ((userpwd = getpwnam(param->value)) == NULL) {
g_error("no such user \"%s\" at line %i",
param->value, param->line);
}
if (setgid(userpwd->pw_gid) == -1) {
g_error("cannot setgid for user \"%s\" at line %i: %s",
param->value, param->line, strerror(errno));
}
#ifdef _BSD_SOURCE
/* init suplementary groups
* (must be done before we change our uid)
*/
if (initgroups(param->value, userpwd->pw_gid) == -1) {
g_warning("cannot init supplementary groups "
"of user \"%s\" at line %i: %s",
param->value, param->line, strerror(errno));
}
#endif
/* set uid */
if (setuid(userpwd->pw_uid) == -1) {
g_error("cannot change to uid of user "
"\"%s\" at line %i: %s",
param->value, param->line, strerror(errno));
}
/* this is needed by libs such as arts */
if (userpwd->pw_dir) {
g_setenv("HOME", userpwd->pw_dir, true);
}
}
#endif
}
void void
daemonize(Options *options) daemonize(Options *options)
{ {
@ -91,3 +175,16 @@ daemonize(Options *options)
(void)options; (void)options;
#endif #endif
} }
void
daemonize_delete_pidfile(void)
{
struct config_param *param = parseConfigFilePath(CONF_PID_FILE, 0);
if (param == NULL)
return;
g_debug("cleaning up pid file");
unlink(param->value);
}

View File

@ -21,13 +21,32 @@
#include "cmdline.h" #include "cmdline.h"
/**
* Kill the MPD which is currently running, pid determined from the
* pid file.
*/
void
daemonize_kill(void);
/** /**
* Close stdin (fd 0) and re-open it as /dev/null. * Close stdin (fd 0) and re-open it as /dev/null.
*/ */
void void
daemonize_close_stdin(void); daemonize_close_stdin(void);
/**
* Change to the configured Unix user.
*/
void
daemonize_set_user(void);
void void
daemonize(Options *options); daemonize(Options *options);
/**
* Deletes the pidfile which was created when MPD started.
*/
void
daemonize_delete_pidfile(void);
#endif #endif

View File

@ -63,15 +63,9 @@
#include <unistd.h> #include <unistd.h>
#include <stdlib.h> #include <stdlib.h>
#include <signal.h>
#include <errno.h> #include <errno.h>
#include <string.h> #include <string.h>
#ifndef WIN32
#include <pwd.h>
#include <grp.h>
#endif
#ifdef HAVE_LOCALE_H #ifdef HAVE_LOCALE_H
#include <locale.h> #include <locale.h>
#endif #endif
@ -81,49 +75,6 @@ GMainLoop *main_loop;
struct notify main_notify; struct notify main_notify;
static void changeToUser(void)
{
#ifndef WIN32
struct config_param *param = config_get_param(CONF_USER);
if (param && strlen(param->value)) {
/* get uid */
struct passwd *userpwd;
if ((userpwd = getpwnam(param->value)) == NULL) {
g_error("no such user \"%s\" at line %i",
param->value, param->line);
}
if (setgid(userpwd->pw_gid) == -1) {
g_error("cannot setgid for user \"%s\" at line %i: %s",
param->value, param->line, strerror(errno));
}
#ifdef _BSD_SOURCE
/* init suplementary groups
* (must be done before we change our uid)
*/
if (initgroups(param->value, userpwd->pw_gid) == -1) {
g_warning("cannot init supplementary groups "
"of user \"%s\" at line %i: %s",
param->value, param->line, strerror(errno));
}
#endif
/* set uid */
if (setuid(userpwd->pw_uid) == -1) {
g_error("cannot change to uid of user "
"\"%s\" at line %i: %s",
param->value, param->line, strerror(errno));
}
/* this is needed by libs such as arts */
if (userpwd->pw_dir) {
g_setenv("HOME", userpwd->pw_dir, true);
}
}
#endif
}
static void openDB(Options * options, char *argv0) static void openDB(Options * options, char *argv0)
{ {
struct config_param *param; struct config_param *param;
@ -160,52 +111,6 @@ static void openDB(Options * options, char *argv0)
} }
} }
static void cleanUpPidFile(void)
{
struct config_param *pidFileParam =
parseConfigFilePath(CONF_PID_FILE, 0);
if (!pidFileParam)
return;
g_debug("cleaning up pid file");
unlink(pidFileParam->value);
}
static void killFromPidFile(void)
{
#ifndef WIN32
FILE *fp;
struct config_param *pidFileParam =
parseConfigFilePath(CONF_PID_FILE, 0);
int pid;
if (!pidFileParam) {
g_error("no pid_file specified in the config file");
}
fp = fopen(pidFileParam->value, "r");
if (!fp) {
g_error("unable to open %s \"%s\": %s",
CONF_PID_FILE, pidFileParam->value, strerror(errno));
}
if (fscanf(fp, "%i", &pid) != 1) {
g_error("unable to read the pid from file \"%s\"",
pidFileParam->value);
}
fclose(fp);
if (kill(pid, SIGTERM)) {
g_error("unable to kill proccess %i: %s",
pid, strerror(errno));
}
exit(EXIT_SUCCESS);
#else
g_error("--kill is not available on WIN32");
#endif
}
static gboolean static gboolean
timer_save_state_file(G_GNUC_UNUSED gpointer data) timer_save_state_file(G_GNUC_UNUSED gpointer data)
{ {
@ -253,7 +158,7 @@ int main(int argc, char *argv[])
parseOptions(argc, argv, &options); parseOptions(argc, argv, &options);
if (options.kill) if (options.kill)
killFromPidFile(); daemonize_kill();
stats_global_init(); stats_global_init();
tag_lib_init(); tag_lib_init();
@ -261,7 +166,7 @@ int main(int argc, char *argv[])
listenOnPort(); listenOnPort();
changeToUser(); daemonize_set_user();
main_task = g_thread_self(); main_task = g_thread_self();
main_loop = g_main_loop_new(NULL, FALSE); main_loop = g_main_loop_new(NULL, FALSE);
@ -356,7 +261,7 @@ int main(int argc, char *argv[])
archive_plugin_deinit_all(); archive_plugin_deinit_all();
#endif #endif
music_pipe_free(); music_pipe_free();
cleanUpPidFile(); daemonize_delete_pidfile();
config_global_finish(); config_global_finish();
tag_pool_deinit(); tag_pool_deinit();
songvec_deinit(); songvec_deinit();