X-Git-Url: http://xvm.mit.edu/gitweb/invirt/third/libt4.git/blobdiff_plain/eeab3e6cade87c1fe0a5f3d93522e12ccb9ec2ab..16e7c282c6fcec8189425bd15ec9e8a4a0ee857d:/rsm_tester.cc

diff --git a/rsm_tester.cc b/rsm_tester.cc
index 3ff4733..1a8b833 100644
--- a/rsm_tester.cc
+++ b/rsm_tester.cc
@@ -2,34 +2,27 @@
 // RSM test client
 //
 
+#include "types.h"
 #include "rsm_protocol.h"
 #include "rsmtest_client.h"
-#include "rpc/rpc.h"
-#include <arpa/inet.h>
-#include <vector>
-#include <stdlib.h>
-#include <stdio.h>
-#include <string>
 
-char tprintf_thread_prefix = 't';
+char log_thread_prefix = 't';
 
-int
-main(int argc, char *argv[])
-{
+int main(int argc, char *argv[]) {
     if(argc != 4){
-        fprintf(stderr, "Usage: %s [host:]port [partition] arg\n", argv[0]);
-        exit(1);
+        cerr << "Usage: " << argv[0] << " [host:]port [partition] arg" << endl;
+        return 1;
     }
 
     rsmtest_client *lc = new rsmtest_client(argv[1]);
-    std::string command(argv[2]);
+    string command(argv[2]);
     if (command == "partition") {
-        printf("net_repair returned %d\n", lc->net_repair(atoi(argv[3])));
+        cout << "net_repair returned " << lc->net_repair(stoi(argv[3]));
     } else if (command == "breakpoint") {
-        int b = atoi(argv[3]);
-        printf("breakpoint %d returned %d\n", b, lc->breakpoint(b));
+        int b = stoi(argv[3]);
+        cout << "breakpoint " << b << " returned " << lc->breakpoint(b);
     } else {
-        fprintf(stderr, "Unknown command %s\n", argv[2]);
+        cerr << "Unknown command " << argv[2] << endl;
     }
     return 0;
 }