diff --git a/sysdep/unix/main.c b/sysdep/unix/main.c index 610d207..5d5f82e 100644 --- a/sysdep/unix/main.c +++ b/sysdep/unix/main.c @@ -478,16 +478,17 @@ signal_init(void) * Parsing of command-line arguments */ -static char *opt_list = "c:dD:ps:u:g:"; +static char *opt_list = "c:dD:ps:u:g:P:"; static int parse_and_exit; char *bird_name; static char *use_user; static char *use_group; +static char *pid_file; static void usage(void) { - fprintf(stderr, "Usage: %s [-c ] [-d] [-D ] [-p] [-s ] [-u ] [-g ]\n", bird_name); + fprintf(stderr, "Usage: %s [-c ] [-d] [-D ] [-p] [-s ] [-u ] [-g ] [-P ]\n", bird_name); exit(1); } @@ -551,6 +552,28 @@ get_gid(const char *s) return gr->gr_gid; } +static inline void +set_pid(const char *s) +{ + int fd; + char pid[6]; + + if (!s) + return; + + memset(pid, 0, sizeof(pid)); + + if((fd=open(s, O_WRONLY|O_CREAT, 0664)) == -1) + die("Cannot open pid file '%s'", s); + + snprintf(pid, sizeof(pid), "%u", getpid()); + + if(write(fd, pid, sizeof(pid)) == -1) + die("Cannot write pid file '%s'", s); + + return; +} + static void parse_args(int argc, char **argv) { @@ -592,6 +615,9 @@ parse_args(int argc, char **argv) case 'g': use_group = optarg; break; + case 'P': + pid_file = optarg; + break; default: usage(); } @@ -661,6 +687,7 @@ main(int argc, char **argv) dup2(0, 2); } + set_pid(pid_file); signal_init(); #ifdef LOCAL_DEBUG