From 2826c6082ba0d0d3c006b48c0ae76b32f29093dc Mon Sep 17 00:00:00 2001
From: Carson Fleming <cflems@cflems.net>
Date: Fri, 18 Aug 2017 10:44:23 -0700
Subject: Make sure there is a command to be run before daemonizing

---
 daemonize.c | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'daemonize.c')

diff --git a/daemonize.c b/daemonize.c
index ab231c5..94662d6 100644
--- a/daemonize.c
+++ b/daemonize.c
@@ -9,6 +9,10 @@ void q (char* m) {
     exit(1);
 }
 
+void qtoofewargs () {
+    q("[daemonize] Too few arguments.\nUsage: daemonize [--chdir] [--noclose] <command>");
+}
+
 char* lowerdup (char* a) {
     char* b = strdup(a);
     int i;
@@ -19,7 +23,7 @@ char* lowerdup (char* a) {
 }
 
 int main (int argc, char** argv) {
-    if (argc < 2) q("[daemonize] Too few arguments.\nUsage: daemonize [--chdir] [--noclose] <command>");
+    if (argc < 2) qtoofewargs();
     int i, cmdlen = 0, buff_size = 256, nochdir = 1, noclose = 0;
     char* buffer = malloc(buff_size);
     
@@ -41,6 +45,8 @@ int main (int argc, char** argv) {
         free(cmp);
     }
     buffer[cmdlen-1] = 0;
+    
+    if (cmdlen < 1) qtoofewargs();
 
     daemon(nochdir, noclose);
     system(buffer);
-- 
cgit v1.2.3