Add '-P' option which allows to specify pidfile.

1 files changed, 28 insertions, 7 deletions

diff --git a/usr.sbin/powerd/powerd.c b/usr.sbin/powerd/powerd.c
index 80547d544ab5..90b89da1c9e4 100644
--- a/usr.sbin/powerd/powerd.c
+++ b/usr.sbin/powerd/powerd.c
@@ -28,9 +28,15 @@
 #include <sys/cdefs.h>
 __FBSDID("$FreeBSD$");
 
+#include <sys/param.h>
+#include <sys/ioctl.h>
+#include <sys/sysctl.h>
+#include <sys/resource.h>
+
 #include <err.h>
 #include <errno.h>
 #include <fcntl.h>
+#include <libutil.h>
 #include <signal.h>
 #include <stdio.h>
 #include <stdlib.h>
@@ -41,10 +47,6 @@ __FBSDID("$FreeBSD$");
 #include <machine/apm_bios.h>
 #endif
 
-#include <sys/ioctl.h>
-#include <sys/sysctl.h>
-#include <sys/resource.h>
-
 #define DEFAULT_ACTIVE_PERCENT	65
 #define DEFAULT_IDLE_PERCENT	90
 #define DEFAULT_POLL_INTERVAL	500	/* Poll interval in milliseconds */
@@ -247,13 +249,15 @@ usage(void)
 {
 
 	fprintf(stderr,
-"usage: powerd [-v] [-a mode] [-b mode] [-i %%] [-n mode] [-p ival] [-r %%]\n");
+"usage: powerd [-v] [-a mode] [-b mode] [-i %%] [-n mode] [-p ival] [-r %%] [-P pidfile]\n");
 	exit(1);
 }
 
 int
 main(int argc, char * argv[])
 {
+	struct pidfh *pfh;
+	const char *pidfile = NULL;
 	long idle, total;
 	int curfreq, *freqs, i, *mwatts, numfreqs;
 	int ch, mode_ac, mode_battery, mode_none, acline, mode, vflag;
@@ -273,7 +277,7 @@ main(int argc, char * argv[])
 	if (geteuid() != 0)
 		errx(1, "must be root to run");
 
-	while ((ch = getopt(argc, argv, "a:b:i:n:p:r:v")) != EOF)
+	while ((ch = getopt(argc, argv, "a:b:i:n:p:P:r:v")) != EOF)
 		switch (ch) {
 		case 'a':
 			parse_mode(optarg, &mode_ac, ch);
@@ -299,6 +303,9 @@ main(int argc, char * argv[])
 				usage();
 			}
 			break;
+		case 'P':
+			pidfile = optarg;
+			break;
 		case 'r':
 			cpu_running_mark = atoi(optarg);
 			if (cpu_running_mark < 0 || cpu_running_mark > 100) {
@@ -338,8 +345,20 @@ main(int argc, char * argv[])
 	acline_init();
 
 	/* Run in the background unless in verbose mode. */
-	if (!vflag)
+	if (!vflag) {
+		pid_t otherpid;
+
+		pfh = pidfile_open(pidfile, 0600, &otherpid);
+		if (pfh == NULL) {
+			if (errno == EEXIST) {
+				errx(1, "powerd already running, pid: %d",
+				    otherpid);
+			}
+			warn("cannot open pid file");
+		}
 		daemon(0, 0);
+		pidfile_write(pfh);
+	}
 	signal(SIGINT, handle_sigs);
 	signal(SIGTERM, handle_sigs);
 
@@ -460,6 +479,8 @@ main(int argc, char * argv[])
 	}
 	free(freqs);
 	free(mwatts);
+	if (!vflag)
+		pidfile_remove(pfh);
 
 	exit(0);
 }