Use flag macros and document -rn options.

1 files changed, 6 insertions, 5 deletions

diff --git a/toys/other/oneit.c b/toys/other/oneit.c
index fe657fb8..4c8bb1f6 100644
--- a/toys/other/oneit.c
+++ b/toys/other/oneit.c
@@ -8,7 +8,7 @@ config ONEIT
   bool "oneit"
   default y
   help
-    usage: oneit [-p] [-c /dev/tty0] command [...]
+    usage: oneit [-prn3] [-c CONSOLE] [COMMAND...]
 
     Simple init program that runs a single supplied command line with a
     controlling tty (so CTRL-C can kill it).
@@ -16,6 +16,7 @@ config ONEIT
     -c	Which console device to use (/dev/console doesn't do CTRL-C, etc)
     -p	Power off instead of rebooting when command exits
     -r	Restart child when it exits
+    -n	No reboot, just relaunch command line
     -3	Write 32 bit PID of each exiting reparented process to fd 3 of child
     	(Blocking writes, child must read to avoid eventual deadlock.)
 
@@ -68,7 +69,7 @@ void oneit_main(void)
   // Setup signal handlers for signals of interest
   for (i = 0; i<ARRAY_LEN(pipes); i++) xsignal(pipes[i], oneit_signaled);
 
-  if (toys.optflags & FLAG_3) {
+  if (FLAG(3)) {
     // Ensure next available filehandles are #3 and #4
     while (xopen_stdio("/", 0) < 3);
     close(3);
@@ -87,10 +88,10 @@ void oneit_main(void)
       // We ignore the return value of write (what would we do with it?)
       // but save it in a variable we never read to make fortify shut up.
       // (Real problem is if pid2 never reads, write() fills pipe and blocks.)
-      while (pid != wait(&i)) if (toys.optflags & FLAG_3) i = write(4, &pid, 4);
-      if (toys.optflags & FLAG_n) continue;
+      while (pid != wait(&i)) if (FLAG(3)) i = write(4, &pid, 4);
+      if (FLAG(n)) continue;
 
-      oneit_signaled((toys.optflags & FLAG_p) ? SIGUSR2 : SIGTERM);
+      oneit_signaled(FLAG(p) ? SIGUSR2 : SIGTERM);
     } else {
       // Redirect stdio to TT.c, with new session ID, so ctrl-c works.
       setsid();