User selectable prediction mode (fixes #9 github issue)

2012-02-13 03:50:03 -05:00
parent c34986471c
commit 8c7c5340d4
5 changed files with 71 additions and 10 deletions
@@ -22,14 +22,24 @@
 #include "stmclient.h"
 #include "crypto.h"

+void usage( const char *argv0 ) {
+  fprintf( stderr, "Usage: %s IP PORT [always|never|adaptive]\n", argv0 );
+}
+
 int main( int argc, char *argv[] )
 {
  /* Get arguments */
  char *ip;
  int port;

-  if ( argc != 3 ) {
-    fprintf( stderr, "Usage: %s IP PORT\n", argv[ 0 ] );
+  char *predict_mode = NULL;
+
+  if ( argc == 4 ) {
+    predict_mode = argv[ 3 ];
+  } else if ( argc == 3 ) {
+    /* do nothing */
+  } else {
+    usage( argv[ 0 ]);
    exit( 1 );
  }

@@ -61,7 +71,7 @@ int main( int argc, char *argv[] )
  }

  try {
-    STMClient client( ip, port, key );
+    STMClient client( ip, port, key, predict_mode );
    client.init();

    try {