From: Jeremy Allison Date: Thu, 19 Jul 2012 23:08:49 +0000 (-0700) Subject: Make the s3 pidfile use the common code inside lib/util/pidfile.c X-Git-Url: http://git.samba.org/?a=commitdiff_plain;h=0d24370c76002bb9db91f23e74f887805b070304;p=metze%2Fsamba%2Fwip.git Make the s3 pidfile use the common code inside lib/util/pidfile.c --- diff --git a/source3/include/proto.h b/source3/include/proto.h index b58f9a294567..ede06bd6cee7 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 index 1170f876c386..79ea3a533006 100644 --- a/source3/lib/pidfile.c +++ b/source3/lib/pidfile.c @@ -1,50 +1,40 @@ /* 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" -#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) +/* Malloc a pidfile name. */ +static char *get_pidfile_name(const char *program_name) { - int fd; - char pidstr[20]; - pid_t pid = 0; - unsigned int ret; - char *name; - const char *short_configfile; - char * pidFile; + 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 */ @@ -58,134 +48,34 @@ pid_t pidfile_pid_s3(const char *program_name) smb_panic("asprintf failed"); } } + return name; +} - if (asprintf(&pidFile, "%s/%s.pid", lp_piddir(), name) == -1) { - SAFE_FREE(name); - return 0; - } +/* 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); - - 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; + return pid; } /* 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; + char *name = get_pidfile_name(program_name); - /* 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... */ + pidfile_create(lp_piddir(), name); 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) +/* Remove a pidfile. */ +void pidfile_unlink_s3(const char *program_name) { - if (pidFile_name == NULL) { - return; - } - unlink(pidFile_name); - SAFE_FREE(pidFile_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 35ed3c0d92aa..df12405adac8 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 86a621f551a3..ab64b9c8a9c0 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 dfbcf13eb4b3..c2ac2e003242 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);