]> git.samba.org - vlendec/samba-autobuild/.git/blobdiff - source3/lib/pidfile.c
r23784: use the GPLv3 boilerplate as recommended by the FSF and the license text
[vlendec/samba-autobuild/.git] / source3 / lib / pidfile.c
index 9cb3f5afef447a54694e7584fdad1dc231b12512..5506b46961ddbbe9305a4d0fcabed52bb905b7ee 100644 (file)
@@ -1,12 +1,13 @@
+/* this code is broken - there is a race condition with the unlink (tridge) */
+
 /* 
-   Unix SMB/Netbios implementation.
-   Version 1.9.
+   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 2 of the License, or
+   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,
    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., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
 
-
-extern int DEBUGLEVEL;
-
 #ifndef O_NONBLOCK
 #define O_NONBLOCK
 #endif
 
+/* return the pid in a pidfile. return 0 if the process (or pidfile)
+   does not exist */
+pid_t pidfile_pid(const char *name)
+{
+       int fd;
+       char pidstr[20];
+       pid_t pid;
+       unsigned int ret;
+       char * pidFile;
+
+       if (asprintf(&pidFile, "%s/%s.pid", lp_piddir(), name) == -1) {
+               return 0;
+       }
+
+       fd = sys_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)) {
+               goto noproc;
+       }
+
+       if (fcntl_lock(fd,SMB_F_SETLK,0,1,F_RDLCK)) {
+               /* we could get the lock - it can't be a Samba process */
+               goto noproc;
+       }
 
-/* create a pid file in the lock directory. open it and leave it locked */
-void pidfile_create(char *name)
+       SAFE_FREE(pidFile);
+       close(fd);
+       return (pid_t)ret;
+
+ noproc:
+       close(fd);
+       unlink(pidFile);
+       SAFE_FREE(pidFile);
+       return 0;
+}
+
+/* create a pid file in the pid directory. open it and leave it locked */
+void pidfile_create(const char *program_name)
 {
        int     fd;
        char    buf[20];
-       pstring pidFile;
-       int pid;
+       char    *short_configfile;
+       char *name;
+       char *pidFile;
+       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, dyn_CONFIGFILE) == 0) {
+               name = SMB_STRDUP(program_name);
+       } else {
+               short_configfile = strrchr( dyn_CONFIGFILE, '/');
+               if (short_configfile == NULL) {
+                       /* conf file in current directory */
+                       short_configfile = dyn_CONFIGFILE;
+               } else {
+                       /* full/relative path provided */
+                       short_configfile++;
+               }
+               if (asprintf(&name, "%s-%s", program_name,
+                            short_configfile+1) == -1) {
+                       smb_panic("asprintf failed");
+               }
+       }
 
-       slprintf(pidFile, sizeof(pidFile)-1, "%s/%s.pid", lp_lockdir(), name);
+       if (asprintf(&pidFile, "%s/%s.pid", lp_piddir(), name) == -1) {
+               smb_panic("asprintf failed");
+       }
 
        pid = pidfile_pid(name);
-       if (pid > 0 && process_exists(pid)) {
-               DEBUG(0,("ERROR: %s is already running\n", name));
+       if (pid != 0) {
+               DEBUG(0,("ERROR: %s is already running. File %s exists and process id %d is running.\n", 
+                        name, pidFile, (int)pid));
                exit(1);
        }
 
-       fd = open(pidFile, O_NONBLOCK | O_CREAT | O_WRONLY, 0644);
-       if (fd < 0) {
-               DEBUG(0,("ERROR: can't open %s: %s\n", pidFile, 
+       fd = sys_open(pidFile, O_NONBLOCK | O_CREAT | O_WRONLY | O_EXCL, 0644);
+       if (fd == -1) {
+               DEBUG(0,("ERROR: can't open %s: Error was %s\n", pidFile, 
                         strerror(errno)));
                exit(1);
        }
 
-       if (fcntl_lock(fd,F_SETLK,0,1,F_WRLCK)==False) {
-               DEBUG(0,("ERROR: %s is already running\n", name));
+       if (fcntl_lock(fd,SMB_F_SETLK,0,1,F_WRLCK)==False) {
+               DEBUG(0,("ERROR: %s : fcntl lock of file %s failed. Error was %s\n",  
+              name, pidFile, strerror(errno)));
                exit(1);
        }
 
        memset(buf, 0, sizeof(buf));
-       slprintf(buf, sizeof(buf) - 1, "%u\n", (unsigned int) getpid());
-       if (write(fd, buf, sizeof(buf)) != sizeof(buf)) {
-               DEBUG(0,("ERROR: can't write to %s: %s\n", 
+       slprintf(buf, sizeof(buf) - 1, "%u\n", (unsigned int) sys_getpid());
+       if (write(fd, buf, strlen(buf)) != (ssize_t)strlen(buf)) {
+               DEBUG(0,("ERROR: can't write to file %s: %s\n", 
                         pidFile, strerror(errno)));
                exit(1);
        }
        /* Leave pid file open & locked for the duration... */
+       SAFE_FREE(name);
+       SAFE_FREE(pidFile);
 }
-
-
-/* return the pid in a pidfile. return 0 if the process (or pidfile)
-   does not exist */
-int pidfile_pid(char *name)
-{
-       FILE *f;
-       pstring pidFile;
-       unsigned ret;
-
-       slprintf(pidFile, sizeof(pidFile)-1, "%s/%s.pid", lp_lockdir(), name);
-
-       f = fopen(pidFile, "r");
-       if (!f) {
-               return 0;
-       }
-
-       if (fscanf(f,"%u", &ret) != 1) {
-               fclose(f);
-               return 0;
-       }
-       fclose(f);
-       
-       if (!process_exists(ret)) return 0;
-
-       return ret;
-}
-