X-Git-Url: http://xvm.mit.edu/gitweb/invirt/third/libt4.git/blobdiff_plain/5a5c578e2e358a121cdb9234a6cb11c4ecfbf323..f0dcb6b97d6d40f67698d1f71ac26970f1776f82:/lock_tester.cc

diff --git a/lock_tester.cc b/lock_tester.cc
index c192128..f535d8f 100644
--- a/lock_tester.cc
+++ b/lock_tester.cc
@@ -4,31 +4,29 @@
 
 #include "lock_client.h"
 #include <arpa/inet.h>
-#include <sys/types.h>
 #include <unistd.h>
 
 char log_thread_prefix = 'c';
 
 // must be >= 2
 const int nt = 6; //XXX: lab1's rpc handlers are blocking. Since rpcs uses a thread pool of 10 threads, we cannot test more than 10 blocking rpc.
-std::string dst;
-lock_client **lc = new lock_client * [nt];
-lock_protocol::lockid_t a = "1";
-lock_protocol::lockid_t b = "2";
-lock_protocol::lockid_t c = "3";
+static string dst;
+static lock_client **lc = new lock_client * [nt];
+static lock_protocol::lockid_t a = "1";
+static lock_protocol::lockid_t b = "2";
+static lock_protocol::lockid_t c = "3";
 
 // check_grant() and check_release() check that the lock server
 // doesn't grant the same lock to both clients.
 // it assumes that lock names are distinct in the first byte.
-int ct[256];
-std::mutex count_mutex;
+static int ct[256];
+static mutex count_mutex;
 
 void check_grant(lock_protocol::lockid_t lid) {
     lock ml(count_mutex);
     int x = lid[0] & 0x0f;
     if (ct[x] != 0) {
-        cout << "error: server granted " << lid << " twice" << endl;
-        cerr << "error: server granted " << lid << " twice" << endl;
+        LOG_NONMEMBER("error: server granted " << lid << " twice");
         exit(1);
     }
     ct[x] += 1;
@@ -38,7 +36,7 @@ void check_release(lock_protocol::lockid_t lid) {
     lock ml(count_mutex);
     int x = lid[0] & 0x0f;
     if (ct[x] != 1) {
-        cerr << "error: client released un-held lock " << lid << endl;
+        LOG_NONMEMBER("error: client released un-held lock " << lid);
         exit(1);
     }
     ct[x] -= 1;
@@ -71,7 +69,7 @@ void test2(int i) {
     lc[i]->acquire(a);
     LOG_NONMEMBER("test2: client " << i << " acquire done");
     check_grant(a);
-    sleep(1);
+    usleep(100000);
     LOG_NONMEMBER("test2: client " << i << " release");
     check_release(a);
     lc[i]->release(a);
@@ -116,7 +114,7 @@ void test5(int i) {
 int
 main(int argc, char *argv[])
 {
-    std::thread th[nt];
+    thread th[nt];
     int test = 0;
 
     setvbuf(stdout, NULL, _IONBF, 0);
@@ -124,7 +122,7 @@ main(int argc, char *argv[])
     srandom((uint32_t)getpid());
 
     if (argc < 2) {
-        cerr << "Usage: " << argv[0] << " [host:]port [test]" << endl;
+        LOG_NONMEMBER("Usage: " << argv[0] << " [host:]port [test]");
         exit(1);
     }
 
@@ -148,7 +146,7 @@ main(int argc, char *argv[])
     if (!test || test == 2) {
         // test2
         for (int i = 0; i < nt; i++)
-            th[i] = std::thread(test2, i);
+            th[i] = thread(test2, i);
         for (int i = 0; i < nt; i++)
             th[i].join();
     }
@@ -157,7 +155,7 @@ main(int argc, char *argv[])
         LOG_NONMEMBER("test 3");
 
         for (int i = 0; i < nt; i++)
-            th[i] = std::thread(test3, i);
+            th[i] = thread(test3, i);
         for (int i = 0; i < nt; i++)
             th[i].join();
     }
@@ -166,7 +164,7 @@ main(int argc, char *argv[])
         LOG_NONMEMBER("test 4");
 
         for (int i = 0; i < 2; i++)
-            th[i] = std::thread(test4, i);
+            th[i] = thread(test4, i);
         for (int i = 0; i < 2; i++)
             th[i].join();
     }
@@ -175,7 +173,7 @@ main(int argc, char *argv[])
         LOG_NONMEMBER("test 5");
 
         for (int i = 0; i < nt; i++)
-            th[i] = std::thread(test5, i);
+            th[i] = thread(test5, i);
         for (int i = 0; i < nt; i++)
             th[i].join();
     }