X-Git-Url: http://xvm.mit.edu/gitweb/invirt/third/libt4.git/blobdiff_plain/61809b48ade4c21b1b01931d520aa2abc7507032..a4175b2e216a20b86cc872dea8a08005c60617a5:/lock_demo.cc

diff --git a/lock_demo.cc b/lock_demo.cc
index 4d74acb..4285269 100644
--- a/lock_demo.cc
+++ b/lock_demo.cc
@@ -4,27 +4,20 @@
 
 #include "lock_protocol.h"
 #include "lock_client.h"
-#include "rpc.h"
+#include "rpc/rpc.h"
 #include <arpa/inet.h>
 #include <vector>
 #include <stdlib.h>
 #include <stdio.h>
 
-std::string dst;
-lock_client *lc;
-
 int
 main(int argc, char *argv[])
 {
-  int r;
-
-  if(argc != 2){
-    fprintf(stderr, "Usage: %s [host:]port\n", argv[0]);
-    exit(1);
-  }
+    if(argc != 2) {
+        fprintf(stderr, "Usage: %s [host:]port\n", argv[0]);
+        return 1;
+    }
 
-  dst = argv[1];
-  lc = new lock_client(dst);
-  r = lc->stat(1);
-  printf ("stat returned %d\n", r);
+    lock_client *lc = new lock_client(argv[1]);
+    printf ("stat returned %d\n", lc->stat("1"));
 }