Denis Vlasenko | ebf48bb | 2007-03-27 22:02:06 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
| 2 | /* |
| 3 | * pid file routines |
| 4 | * |
| 5 | * Copyright (C) 2007 by Stephane Billiart <stephane.billiart@gmail.com> |
| 6 | * |
Denys Vlasenko | 0ef64bd | 2010-08-16 20:14:46 +0200 | [diff] [blame] | 7 | * Licensed under GPLv2 or later, see file LICENSE in this source tree. |
Denis Vlasenko | ebf48bb | 2007-03-27 22:02:06 +0000 | [diff] [blame] | 8 | */ |
Denis Vlasenko | 1caca34 | 2007-08-02 10:14:29 +0000 | [diff] [blame] | 9 | |
| 10 | /* Override ENABLE_FEATURE_PIDFILE */ |
| 11 | #define WANT_PIDFILE 1 |
Denis Vlasenko | ebf48bb | 2007-03-27 22:02:06 +0000 | [diff] [blame] | 12 | #include "libbb.h" |
| 13 | |
Denis Vlasenko | bb23c06 | 2007-08-15 20:05:37 +0000 | [diff] [blame] | 14 | smallint wrote_pidfile; |
| 15 | |
Denis Vlasenko | defc1ea | 2008-06-27 02:52:20 +0000 | [diff] [blame] | 16 | void FAST_FUNC write_pidfile(const char *path) |
Denis Vlasenko | ebf48bb | 2007-03-27 22:02:06 +0000 | [diff] [blame] | 17 | { |
| 18 | int pid_fd; |
| 19 | char *end; |
| 20 | char buf[sizeof(int)*3 + 2]; |
Denis Vlasenko | bb23c06 | 2007-08-15 20:05:37 +0000 | [diff] [blame] | 21 | struct stat sb; |
Denis Vlasenko | ebf48bb | 2007-03-27 22:02:06 +0000 | [diff] [blame] | 22 | |
Denis Vlasenko | 80edead | 2007-08-02 22:31:05 +0000 | [diff] [blame] | 23 | if (!path) |
Denis Vlasenko | bb23c06 | 2007-08-15 20:05:37 +0000 | [diff] [blame] | 24 | return; |
Denis Vlasenko | ebf48bb | 2007-03-27 22:02:06 +0000 | [diff] [blame] | 25 | /* we will overwrite stale pidfile */ |
Denis Vlasenko | 6a5598c | 2007-03-27 22:05:34 +0000 | [diff] [blame] | 26 | pid_fd = open(path, O_WRONLY|O_CREAT|O_TRUNC, 0666); |
Denis Vlasenko | ebf48bb | 2007-03-27 22:02:06 +0000 | [diff] [blame] | 27 | if (pid_fd < 0) |
Denis Vlasenko | bb23c06 | 2007-08-15 20:05:37 +0000 | [diff] [blame] | 28 | return; |
| 29 | |
| 30 | /* path can be "/dev/null"! Test for such cases */ |
| 31 | wrote_pidfile = (fstat(pid_fd, &sb) == 0) && S_ISREG(sb.st_mode); |
| 32 | |
| 33 | if (wrote_pidfile) { |
| 34 | /* few bytes larger, but doesn't use stdio */ |
| 35 | end = utoa_to_buf(getpid(), buf, sizeof(buf)); |
| 36 | *end = '\n'; |
| 37 | full_write(pid_fd, buf, end - buf + 1); |
| 38 | } |
Denis Vlasenko | ebf48bb | 2007-03-27 22:02:06 +0000 | [diff] [blame] | 39 | close(pid_fd); |
Denis Vlasenko | ebf48bb | 2007-03-27 22:02:06 +0000 | [diff] [blame] | 40 | } |
Denys Vlasenko | 5059653 | 2019-03-17 19:47:52 +0100 | [diff] [blame^] | 41 | |
| 42 | void FAST_FUNC write_pidfile_std_path_and_ext(const char *name) |
| 43 | { |
| 44 | char buf[sizeof(CONFIG_PID_FILE_PATH) + 64]; |
| 45 | |
| 46 | snprintf(buf, sizeof(buf), CONFIG_PID_FILE_PATH"/%s.pid", name); |
| 47 | write_pidfile(buf); |
| 48 | } |
| 49 | |
| 50 | void FAST_FUNC remove_pidfile_std_path_and_ext(const char *name) |
| 51 | { |
| 52 | char buf[sizeof(CONFIG_PID_FILE_PATH) + 64]; |
| 53 | |
| 54 | if (!wrote_pidfile) |
| 55 | return; |
| 56 | snprintf(buf, sizeof(buf), CONFIG_PID_FILE_PATH"/%s.pid", name); |
| 57 | unlink(buf); |
| 58 | } |