From 0d24370c76002bb9db91f23e74f887805b070304 Mon Sep 17 00:00:00 2001 From: Jeremy Allison Date: Thu, 19 Jul 2012 16:08:49 -0700 Subject: [PATCH] Make the s3 pidfile use the common code inside lib/util/pidfile.c --- source3/include/proto.h | 2 +- source3/lib/pidfile.c | 272 +++++++++++++------------------------------- source3/nmbd/nmbd.c | 2 +- source3/smbd/server_exit.c | 2 +- source3/winbindd/winbindd.c | 2 +- 5 files changed, 85 insertions(+), 195 deletions(-) rewrite source3/lib/pidfile.c (64%) diff --git a/source3/include/proto.h b/source3/include/proto.h index b58f9a29456..ede06bd6cee 100644 --- a/source3/include/proto.h +++ b/source3/include/proto.h @@ -162,7 +162,7 @@ int ms_fnmatch(const char *pattern, const char *string, bool translate_pattern, pid_t pidfile_pid_s3(const char *name); void pidfile_create_s3(const char *program_name); -void pidfile_unlink_s3(void); +void pidfile_unlink_s3(const char *program_name); /* The following definitions come from lib/recvfile.c */ diff --git a/source3/lib/pidfile.c b/source3/lib/pidfile.c dissimilarity index 64% index 1170f876c38..79ea3a53300 100644 --- a/source3/lib/pidfile.c +++ b/source3/lib/pidfile.c @@ -1,191 +1,81 @@ -/* this code is broken - there is a race condition with the unlink (tridge) */ - -/* - Unix SMB/CIFS implementation. - pidfile handling - Copyright (C) Andrew Tridgell 1998 - - 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 3 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, see . -*/ - -#include "includes.h" -#include "system/filesys.h" - -#ifndef O_NONBLOCK -#define O_NONBLOCK -#endif - -static char *pidFile_name = NULL; - -/* return the pid in a pidfile. return 0 if the process (or pidfile) - does not exist */ -pid_t pidfile_pid_s3(const char *program_name) -{ - int fd; - char pidstr[20]; - pid_t pid = 0; - unsigned int ret; - char *name; - const char *short_configfile; - char * pidFile; - - /* Add a suffix to the program name if this is a process with a - * none default configuration file name. */ - if (strcmp( CONFIGFILE, get_dyn_CONFIGFILE()) == 0) { - name = SMB_STRDUP(program_name); - } else { - short_configfile = strrchr( get_dyn_CONFIGFILE(), '/'); - if (short_configfile == NULL) { - /* conf file in current directory */ - short_configfile = get_dyn_CONFIGFILE(); - } else { - /* full/relative path provided */ - short_configfile++; - } - if (asprintf(&name, "%s-%s", program_name, - short_configfile) == -1) { - smb_panic("asprintf failed"); - } - } - - if (asprintf(&pidFile, "%s/%s.pid", lp_piddir(), name) == -1) { - SAFE_FREE(name); - return 0; - } - - SAFE_FREE(name); - - fd = open(pidFile, O_NONBLOCK | O_RDONLY, 0644); - if (fd == -1) { - SAFE_FREE(pidFile); - return 0; - } - - ZERO_ARRAY(pidstr); - - if (read(fd, pidstr, sizeof(pidstr)-1) <= 0) { - goto noproc; - } - - ret = atoi(pidstr); - - if (ret == 0) { - /* Obviously we had some garbage in the pidfile... */ - DEBUG(1, ("Could not parse contents of pidfile %s\n", - pidFile)); - goto noproc; - } - - pid = (pid_t)ret; - if (!process_exists_by_pid(pid)) { - DEBUG(10, ("Process with PID=%d does not exist.\n", (int)pid)); - goto noproc; - } - - if (fcntl_lock(fd,F_SETLK,0,1,F_RDLCK)) { - /* we could get the lock - it can't be a Samba process */ - DEBUG(10, ("Process with PID=%d is no Samba process.\n", - (int)pid)); - goto noproc; - } - - DEBUG(10, ("Process with PID=%d is running.\n", (int)pid)); - SAFE_FREE(pidFile); - close(fd); - return (pid_t)ret; - - noproc: - close(fd); - DEBUG(10, ("Deleting %s, since %d is no Samba process.\n", pidFile, - (int)pid)); - unlink(pidFile); - SAFE_FREE(pidFile); - return 0; -} - -/* create a pid file in the pid directory. open it and leave it locked */ -void pidfile_create_s3(const char *program_name) -{ - int fd; - char buf[20]; - const char *short_configfile; - char *name; - pid_t pid; - - /* Add a suffix to the program name if this is a process with a - * none default configuration file name. */ - if (strcmp( CONFIGFILE, get_dyn_CONFIGFILE()) == 0) { - name = SMB_STRDUP(program_name); - } else { - short_configfile = strrchr( get_dyn_CONFIGFILE(), '/'); - if (short_configfile == NULL) { - /* conf file in current directory */ - short_configfile = get_dyn_CONFIGFILE(); - } else { - /* full/relative path provided */ - short_configfile++; - } - if (asprintf(&name, "%s-%s", program_name, - short_configfile) == -1) { - smb_panic("asprintf failed"); - } - } - - if (asprintf(&pidFile_name, "%s/%s.pid", lp_piddir(), name) == -1) { - smb_panic("asprintf failed"); - } - - pid = pidfile_pid_s3(program_name); - if (pid != 0) { - DEBUG(0,("ERROR: %s is already running. File %s exists and process id %d is running.\n", - name, pidFile_name, (int)pid)); - exit(1); - } - - fd = open(pidFile_name, O_NONBLOCK | O_CREAT | O_WRONLY | O_EXCL, - 0644); - if (fd == -1) { - DEBUG(0,("ERROR: can't open %s: Error was %s\n", pidFile_name, - strerror(errno))); - exit(1); - } - - if (fcntl_lock(fd,F_SETLK,0,1,F_WRLCK)==False) { - DEBUG(0,("ERROR: %s : fcntl lock of file %s failed. Error was %s\n", - name, pidFile_name, strerror(errno))); - exit(1); - } - - memset(buf, 0, sizeof(buf)); - slprintf(buf, sizeof(buf) - 1, "%u\n", (unsigned int)getpid()); - if (write(fd, buf, strlen(buf)) != (ssize_t)strlen(buf)) { - DEBUG(0,("ERROR: can't write to file %s: %s\n", - pidFile_name, strerror(errno))); - exit(1); - } - /* Leave pid file open & locked for the duration... */ - SAFE_FREE(name); - - /* set the close on exec so that we don't leak the fd */ - fcntl(fd, F_SETFD, FD_CLOEXEC); -} - -void pidfile_unlink_s3(void) -{ - if (pidFile_name == NULL) { - return; - } - unlink(pidFile_name); - SAFE_FREE(pidFile_name); -} +/* this code is broken - there is a race condition with the unlink (tridge) */ + +/* + Unix SMB/CIFS implementation. + pidfile handling + Copyright (C) Andrew Tridgell 1998 + Copyright (C) Jeremy Allison 2012 + + 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 3 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, see . +*/ + +#include "includes.h" +#include "system/filesys.h" +#include "../lib/util/pidfile.h" + +/* Malloc a pidfile name. */ +static char *get_pidfile_name(const char *program_name) +{ + char *name = NULL; + + /* Add a suffix to the program name if this is a process with a + * none default configuration file name. */ + if (strcmp( CONFIGFILE, get_dyn_CONFIGFILE()) == 0) { + name = SMB_STRDUP(program_name); + } else { + const char *short_configfile; + short_configfile = strrchr( get_dyn_CONFIGFILE(), '/'); + if (short_configfile == NULL) { + /* conf file in current directory */ + short_configfile = get_dyn_CONFIGFILE(); + } else { + /* full/relative path provided */ + short_configfile++; + } + if (asprintf(&name, "%s-%s", program_name, + short_configfile) == -1) { + smb_panic("asprintf failed"); + } + } + return name; +} + +/* return the pid in a pidfile. return 0 if the process (or pidfile) + does not exist */ +pid_t pidfile_pid_s3(const char *program_name) +{ + pid_t pid = 0; + char *name = get_pidfile_name(program_name); + + pid = pidfile_pid(lp_piddir(), name); + SAFE_FREE(name); + return pid; +} + +/* create a pid file in the pid directory. open it and leave it locked */ +void pidfile_create_s3(const char *program_name) +{ + char *name = get_pidfile_name(program_name); + + pidfile_create(lp_piddir(), name); + SAFE_FREE(name); +} + +/* Remove a pidfile. */ +void pidfile_unlink_s3(const char *program_name) +{ + char *name = get_pidfile_name(program_name); + pidfile_unlink(lp_piddir(), name); + SAFE_FREE(name); +} diff --git a/source3/nmbd/nmbd.c b/source3/nmbd/nmbd.c index 35ed3c0d92a..df12405adac 100644 --- a/source3/nmbd/nmbd.c +++ b/source3/nmbd/nmbd.c @@ -70,7 +70,7 @@ static void terminate(struct messaging_context *msg) gencache_stabilize(); serverid_deregister(messaging_server_id(msg)); - pidfile_unlink_s3(); + pidfile_unlink_s3("nmbd"); exit(0); } diff --git a/source3/smbd/server_exit.c b/source3/smbd/server_exit.c index 86a621f551a..ab64b9c8a9c 100644 --- a/source3/smbd/server_exit.c +++ b/source3/smbd/server_exit.c @@ -216,7 +216,7 @@ static void exit_server_common(enum server_exit_reason how, DEBUG(3,("Server exit (%s)\n", (reason ? reason : "normal exit"))); if (am_parent) { - pidfile_unlink_s3(); + pidfile_unlink_s3("smbd"); } gencache_stabilize(); } diff --git a/source3/winbindd/winbindd.c b/source3/winbindd/winbindd.c index dfbcf13eb4b..c2ac2e00324 100644 --- a/source3/winbindd/winbindd.c +++ b/source3/winbindd/winbindd.c @@ -184,7 +184,7 @@ static void terminate(bool is_parent) if (is_parent) { serverid_deregister(procid_self()); - pidfile_unlink_s3(); + pidfile_unlink_s3("winbindd"); } exit(0); -- 2.11.4.GIT