Permalink
Browse files

Detect edge case when argc == 0

  • Loading branch information...
1 parent 06661c9 commit da18a8c7eab4075ed1f4a6f799468e0923b678a6 @keithw keithw committed Oct 20, 2012
Showing with 8 additions and 0 deletions.
  1. +4 −0 src/frontend/mosh-client.cc
  2. +4 −0 src/frontend/mosh-server.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 ) {
@@ -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;

0 comments on commit da18a8c

Please sign in to comment.