shutdown command -r option complete #125

Bone-Project · Feb 4, 2017 · f616da3 · f616da3
1 parent 456895e
commit f616da3
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/sbin/shutdown/shutdown.c b/sbin/shutdown/shutdown.c
@@ -28,6 +28,7 @@
 #include <stdio/stdio.h>
 #include <stdlib/stdlib.h>
 #include <string/string.h>
+#include <ctype/ctype.h>
 #include <io/io.h>
 #include <misc/asm_util.h>
 #include <sys/sys_poweroff.h>
@@ -75,6 +76,16 @@ int cmd_shutdown_handler(char* cmd)
  }
  else
  {
+ //First checking if the string contains an alphabet
+ for (int i = 0; i < (int)strlen (opts [2].str); i ++)
+ {
+ if (isalpha (opts [2].str [i]))
+ {
+ printk (cmd_shutdown.invalid_use_msg);
+ return STATUS_OK;
+ }
+ }
+
  time = atoi (opts [2].str);
  }
 
@@ -105,6 +116,10 @@ int cmd_shutdown_handler(char* cmd)
 
  cmds [CMD_REBOOT_INDEX]->handler ("reboot");
  }
+ else if (time == 0)
+ {
+ cmds [CMD_REBOOT_INDEX]->handler ("reboot");
+ }
  }
  else
  {