From 577f42c075ca59d429bf8d8bc93df8018ea64eb9 Mon Sep 17 00:00:00 2001
From: Jfreegman <Jfreegman@gmail.com>
Date: Tue, 3 Sep 2013 01:02:58 -0400
Subject: [PATCH] fix more potential segfaults

---
 src/prompt.c | 29 +++++++++++++++--------------
 1 file changed, 15 insertions(+), 14 deletions(-)

diff --git a/src/prompt.c b/src/prompt.c
index 6e598c3..0444494 100644
--- a/src/prompt.c
+++ b/src/prompt.c
@@ -109,6 +109,11 @@ void cmd_accept(ToxWindow *self, Tox *m, int argc, char **argv)
 
 void cmd_add(ToxWindow *self, Tox *m, int argc, char **argv)
 {
+    if (argc != 1 && argc != 2) {
+        wprintw(self->window, "Invalid syntax.\n");
+        return;
+    }
+
     uint8_t id_bin[TOX_FRIEND_ADDRESS_SIZE];
     char xx[3];
     uint32_t x;
@@ -116,19 +121,14 @@ void cmd_add(ToxWindow *self, Tox *m, int argc, char **argv)
     char *msg;
     int i, num;
 
-    /* check arguments */
-    if (argv[2] && argv[2][0] != '\"') {
-        wprintw(self->window, "Messages must be enclosed in quotes.\n");
-        return;
-    }
-    if (argc != 1 && argc != 2) {
-        wprintw(self->window, "Invalid syntax.\n");
-        return;
-    }
-
     id = argv[1];
 
     if (argc == 2) {
+        if (argv[2][0] != '\"') {
+            wprintw(self->window, "Messages must be enclosed in quotes.\n");
+            return;
+        }
+
         msg = argv[2];
         msg[strlen(++msg)-1] = L'\0';
     } else
@@ -379,6 +379,11 @@ void cmd_status(ToxWindow *self, Tox *m, int argc, char **argv)
 
 void cmd_note(ToxWindow *self, Tox *m, int argc, char **argv)
 {
+    if (argc != 1) {
+        wprintw(self->window, "Wrong number of arguments.\n");
+        return;
+    }
+
     char *msg;
 
     /* check arguments */
@@ -386,10 +391,6 @@ void cmd_note(ToxWindow *self, Tox *m, int argc, char **argv)
         wprintw(self->window, "Messages must be enclosed in quotes.\n");
         return;
     }
-    if (argc != 1) {
-        wprintw(self->window, "Wrong number of arguments.\n");
-        return;
-    }
 
     msg = argv[1];
     msg[strlen(++msg)-1] = L'\0';