← Back to branch summary

~zulcss/samba/server-dailies-3.4

« back to all changes in this revision

Viewing changes to source4/smbd/pidfile.c

  • Committer: Chuck Short
  • Date: 2010-09-28 20:38:39 UTC
  • Revision ID: zulcss@ubuntu.com-20100928203839-pgjulytsi9ue63x1
Initial version

Show diffs side-by-side

added added

removed removed

Lines of Context:
source4/smbd/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 3 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, see <http://www.gnu.org/licenses/>.
 
20
*/
 
21
 
 
22
#include "includes.h"
 
23
#include "system/filesys.h"
 
24
#include "smbd/pidfile.h"
 
25
 
 
26
/**
 
27
 * @file
 
28
 * @brief Pid file handling
 
29
 */
 
30
 
 
31
/**
 
32
 * return the pid in a pidfile. return 0 if the process (or pidfile)
 
33
 * does not exist 
 
34
 */
 
35
pid_t pidfile_pid(const char *piddir, const char *name)
 
36
{
 
37
        int fd;
 
38
        char pidstr[20];
 
39
        pid_t ret;
 
40
        char *pidFile;
 
41
 
 
42
        asprintf(&pidFile, "%s/%s.pid", piddir, name);
 
43
 
 
44
        fd = open(pidFile, O_NONBLOCK | O_RDONLY, 0644);
 
45
 
 
46
        if (fd == -1) {
 
47
                SAFE_FREE(pidFile);
 
48
                return 0;
 
49
        }
 
50
 
 
51
        ZERO_STRUCT(pidstr);
 
52
 
 
53
        if (read(fd, pidstr, sizeof(pidstr)-1) <= 0) {
 
54
                goto noproc;
 
55
        }
 
56
 
 
57
        ret = (pid_t)atoi(pidstr);
 
58
        
 
59
        if (!process_exists_by_pid(ret)) {
 
60
                goto noproc;
 
61
        }
 
62
 
 
63
        if (fcntl_lock(fd,F_SETLK,0,1,F_RDLCK)) {
 
64
                /* we could get the lock - it can't be a Samba process */
 
65
                goto noproc;
 
66
        }
 
67
 
 
68
        close(fd);
 
69
        SAFE_FREE(pidFile);
 
70
        return ret;
 
71
 
 
72
 noproc:
 
73
        close(fd);
 
74
        unlink(pidFile);
 
75
        SAFE_FREE(pidFile);
 
76
        return 0;
 
77
}
 
78
 
 
79
/**
 
80
 * create a pid file in the pid directory. open it and leave it locked 
 
81
 */
 
82
void pidfile_create(const char *piddir, const char *name)
 
83
{
 
84
        int     fd;
 
85
        char    buf[20];
 
86
        char *pidFile;
 
87
        pid_t pid;
 
88
 
 
89
        asprintf(&pidFile, "%s/%s.pid", piddir, name);
 
90
 
 
91
        pid = pidfile_pid(piddir, name);
 
92
        if (pid != 0) {
 
93
                DEBUG(0,("ERROR: %s is already running. File %s exists and process id %d is running.\n", 
 
94
                         name, pidFile, (int)pid));
 
95
                exit(1);
 
96
        }
 
97
 
 
98
        fd = open(pidFile, O_NONBLOCK | O_CREAT | O_WRONLY | O_EXCL, 0644);
 
99
        if (fd == -1) {
 
100
                DEBUG(0,("ERROR: can't open %s: Error was %s\n", pidFile, 
 
101
                         strerror(errno)));
 
102
                exit(1);
 
103
        }
 
104
 
 
105
        if (fcntl_lock(fd,F_SETLK,0,1,F_WRLCK)==false) {
 
106
                DEBUG(0,("ERROR: %s : fcntl lock of file %s failed. Error was %s\n",  
 
107
              name, pidFile, strerror(errno)));
 
108
                exit(1);
 
109
        }
 
110
 
 
111
        memset(buf, 0, sizeof(buf));
 
112
        slprintf(buf, sizeof(buf) - 1, "%u\n", (unsigned int) getpid());
 
113
        if (write(fd, buf, strlen(buf)) != (ssize_t)strlen(buf)) {
 
114
                DEBUG(0,("ERROR: can't write to file %s: %s\n", 
 
115
                         pidFile, strerror(errno)));
 
116
                exit(1);
 
117
        }
 
118
 
 
119
        /* Leave pid file open & locked for the duration... */
 
120
        SAFE_FREE(pidFile);
 
121
}