Detect edge case when argc == 0

mobile-shell · Oct 20, 2012 · da18a8c · da18a8c
1 parent 06661c9
commit da18a8c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/frontend/mosh-client.cc b/src/frontend/mosh-client.cc
@@ -39,6 +39,7 @@
 #include "stmclient.h"
 #include "crypto.h"
 #include "locale_utils.h"
+#include "fatal_assert.h"
 
 /* these need to be included last because of conflicting defines */
 #include <curses.h>
@@ -72,6 +73,9 @@ int main( int argc, char *argv[] )
   /* For security, make sure we don't dump core */
   Crypto::disable_dumping_core();
 
+  /* Detect edge case */
+  fatal_assert( argc > 0 );
+
   /* Get arguments */
   int opt;
   while ( (opt = getopt( argc, argv, "c" )) != -1 ) {

diff --git a/src/frontend/mosh-server.cc b/src/frontend/mosh-server.cc
@@ -81,6 +81,7 @@
 #include "pty_compat.h"
 #include "select.h"
 #include "timestamp.h"
+#include "fatal_assert.h"
 
 #ifndef _PATH_BSHELL
 #define _PATH_BSHELL "/bin/sh"
@@ -159,6 +160,9 @@ int main( int argc, char *argv[] )
   /* For security, make sure we don't dump core */
   Crypto::disable_dumping_core();
 
+  /* Detect edge case */
+  fatal_assert( argc > 0 );
+
   char *desired_ip = NULL;
   char *desired_port = NULL;
   string command_path;