Be explicit that per-host pages are directories.
[darkstat.git] / pidfile.c
1 /* darkstat 3
2  * copyright (c) 2007 Emil Mikulic.
3  *
4  * pidfile.h: pidfile manglement
5  *
6  * Permission to use, copy, modify, and distribute this file for any
7  * purpose with or without fee is hereby granted, provided that the above
8  * copyright notice and this permission notice appear in all copies.
9  *
10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17  */
18
19 #include "darkstat.h"
20 #include "err.h"
21 #include "str.h"
22 #include "pidfile.h"
23
24 #include <errno.h>
25 #include <fcntl.h>
26 #include <pwd.h>
27 #include <stdlib.h>
28 #include <unistd.h>
29
30 static int pidfd = -1;
31 static const char *pidname = NULL;
32
33 void
34 pidfile_create(const char *chroot_dir, const char *filename,
35    const char *privdrop_user)
36 {
37    struct passwd *pw;
38
39    if (pidfd != -1)
40       errx(1, "pidfile already created");
41
42    errno = 0;
43    pw = getpwnam(privdrop_user);
44
45    if (pw == NULL) {
46       if (errno == 0)
47          errx(1, "getpwnam(\"%s\") failed: no such user", privdrop_user);
48       else
49          err(1, "getpwnam(\"%s\") failed", privdrop_user);
50    }
51
52    if (chdir(chroot_dir) == -1)
53       err(1, "chdir(\"%s\") failed", chroot_dir);
54    pidname = filename;
55    pidfd = open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_EXCL, 0600);
56    if (pidfd == -1)
57       err(1, "couldn't not create pidfile");
58    if (chown(filename, pw->pw_uid, pw->pw_gid) == -1)
59       err(1, "couldn't chown pidfile");
60 }
61
62 void
63 pidfile_write_close(void)
64 {
65    struct str *s;
66    size_t len;
67    char *buf;
68
69    if (pidfd == -1)
70       errx(1, "cannot write pidfile: not created");
71
72    s = str_make();
73    str_appendf(s, "%u\n", (unsigned int)getpid());
74    str_extract(s, &len, &buf);
75
76    if (write(pidfd, buf, len) != (int)len)
77       err(1, "couldn't write to pidfile");
78    free(buf);
79    if (close(pidfd) == -1)
80       warn("problem closing pidfile");
81 }
82
83 void
84 pidfile_unlink(void)
85 {
86    if (pidname == NULL)
87       return; /* pidfile wasn't created */
88    if (unlink(pidname) == -1)
89       warn("problem unlinking pidfile");
90 }
91
92 /* vim:set ts=3 sw=3 tw=78 et: */