diff --git a/src/frontend/mosh-client.cc b/src/frontend/mosh-client.cc index a9644e3..69038ec 100644 --- 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 @@ -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 index d8be58e..de6d6bf 100644 --- 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;