Specify the pid file by command argument -p/--pid-file, tgtd can write pid in pid file after running into daemon, and remove the file before exiting. Signed-off-by: zhenwei pi <pizhenwei@xxxxxxxxxxxxx> --- usr/tgtd.c | 39 ++++++++++++++++++++++++++++++++++++++- 1 file changed, 38 insertions(+), 1 deletion(-) diff --git a/usr/tgtd.c b/usr/tgtd.c index ae2f489..f92c6b0 100644 --- a/usr/tgtd.c +++ b/usr/tgtd.c @@ -55,13 +55,14 @@ static struct option const long_options[] = { {"foreground", no_argument, 0, 'f'}, {"control-port", required_argument, 0, 'C'}, {"nr_iothreads", required_argument, 0, 't'}, + {"pid-file", required_argument, 0, 'p'}, {"debug", required_argument, 0, 'd'}, {"version", no_argument, 0, 'V'}, {"help", no_argument, 0, 'h'}, {0, 0, 0, 0}, }; -static char *short_options = "fC:d:t:Vh"; +static char *short_options = "fC:d:t:p:Vh"; static char *spare_args; static void usage(int status) @@ -77,6 +78,7 @@ static void usage(int status) "-f, --foreground make the program run in the foreground\n" "-C, --control-port NNNN use port NNNN for the mgmt channel\n" "-t, --nr_iothreads NNNN specify the number of I/O threads\n" + "-p, --pid-file filename specify the pid file\n" "-d, --debug debuglevel print debugging information\n" "-V, --version print version and exit\n" "-h, --help display this help and exit\n", @@ -174,6 +176,27 @@ set_rlimit: return 0; } +void create_pid_file(const char *path) +{ + int fd, ret; + char buf[32] = {0}; + + fd = open(path, O_RDWR | O_CREAT); + if (fd < 0) { + eprintf("can't create %s, %m\n", path); + return; + } + + snprintf(buf, sizeof(buf), "%d", getpid()); + ret = write(fd, buf, strlen(buf)); + if (ret < 0) { + eprintf("can't write %s, %m\n", path); + return; + } + + close(fd); +} + int tgt_event_add(int fd, int events, event_handler_t handler, void *data) { struct epoll_event ev; @@ -525,6 +548,7 @@ int main(int argc, char **argv) int err, ch, longindex, nr_lld = 0; int is_daemon = 1, is_debug = 0; int ret; + char *pidfile = NULL; sa_new.sa_handler = signal_catch; sigemptyset(&sa_new.sa_mask); @@ -555,6 +579,13 @@ int main(int argc, char **argv) if (ret) bad_optarg(ret, ch, optarg); break; + case 'p': + pidfile = strdup(optarg); + if (pidfile == NULL) { + fprintf(stderr, "can't duplicate pid file\n"); + exit(1); + } + break; case 'd': ret = str_to_int_range(optarg, is_debug, 0, 1); if (ret) @@ -621,6 +652,9 @@ int main(int argc, char **argv) sd_notify(0, "READY=1\nSTATUS=Starting event loop..."); #endif + if (is_daemon && pidfile) + create_pid_file(pidfile); + event_loop(); lld_exit(); @@ -631,5 +665,8 @@ int main(int argc, char **argv) log_close(); + if (is_daemon && pidfile) + unlink(pidfile); + return 0; } -- 2.11.0