import HEAD into svn+ssh://svn.samba.org/home/svn/samba/trunk
[metze/old/v3-2-winbind-ndr.git] / source / lib / pidfile.c
1 /* this code is broken - there is a race condition with the unlink (tridge) */
2
3 /* 
4    Unix SMB/CIFS implementation.
5    pidfile handling
6    Copyright (C) Andrew Tridgell 1998
7    
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 2 of the License, or
11    (at your option) any later version.
12    
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22
23 #include "includes.h"
24
25 #ifndef O_NONBLOCK
26 #define O_NONBLOCK
27 #endif
28
29 /* return the pid in a pidfile. return 0 if the process (or pidfile)
30    does not exist */
31 pid_t pidfile_pid(const char *name)
32 {
33         int fd;
34         char pidstr[20];
35         unsigned ret;
36         pstring pidFile;
37
38         slprintf(pidFile, sizeof(pidFile)-1, "%s/%s.pid", lp_piddir(), name);
39
40         fd = sys_open(pidFile, O_NONBLOCK | O_RDONLY, 0644);
41         if (fd == -1) {
42                 return 0;
43         }
44
45         ZERO_ARRAY(pidstr);
46
47         if (read(fd, pidstr, sizeof(pidstr)-1) <= 0) {
48                 goto noproc;
49         }
50
51         ret = atoi(pidstr);
52         
53         if (!process_exists((pid_t)ret)) {
54                 goto noproc;
55         }
56
57         if (fcntl_lock(fd,SMB_F_SETLK,0,1,F_RDLCK)) {
58                 /* we could get the lock - it can't be a Samba process */
59                 goto noproc;
60         }
61
62         close(fd);
63         return (pid_t)ret;
64
65  noproc:
66         close(fd);
67         unlink(pidFile);
68         return 0;
69 }
70
71 /* create a pid file in the pid directory. open it and leave it locked */
72 void pidfile_create(const char *name)
73 {
74         int     fd;
75         char    buf[20];
76         pstring pidFile;
77         pid_t pid;
78
79         slprintf(pidFile, sizeof(pidFile)-1, "%s/%s.pid", lp_piddir(), name);
80
81         pid = pidfile_pid(name);
82         if (pid != 0) {
83                 DEBUG(0,("ERROR: %s is already running. File %s exists and process id %d is running.\n", 
84                          name, pidFile, (int)pid));
85                 exit(1);
86         }
87
88         fd = sys_open(pidFile, O_NONBLOCK | O_CREAT | O_WRONLY | O_EXCL, 0644);
89         if (fd == -1) {
90                 DEBUG(0,("ERROR: can't open %s: Error was %s\n", pidFile, 
91                          strerror(errno)));
92                 exit(1);
93         }
94
95         if (fcntl_lock(fd,SMB_F_SETLK,0,1,F_WRLCK)==False) {
96                 DEBUG(0,("ERROR: %s : fcntl lock of file %s failed. Error was %s\n",  
97               name, pidFile, strerror(errno)));
98                 exit(1);
99         }
100
101         memset(buf, 0, sizeof(buf));
102         slprintf(buf, sizeof(buf) - 1, "%u\n", (unsigned int) sys_getpid());
103         if (write(fd, buf, strlen(buf)) != (ssize_t)strlen(buf)) {
104                 DEBUG(0,("ERROR: can't write to file %s: %s\n", 
105                          pidFile, strerror(errno)));
106                 exit(1);
107         }
108         /* Leave pid file open & locked for the duration... */
109 }