#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.
-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];
-mutex count_mutex;
+static int ct[256];
+static std::mutex count_mutex;
-void check_grant(lock_protocol::lockid_t lid) {
+static 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;
}
-void check_release(lock_protocol::lockid_t lid) {
+static 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;
}
-void test1(void) {
- LOG_NONMEMBER("acquire a release a acquire a release a");
+static void test1(void) {
+ LOG_NONMEMBER << "acquire a release a acquire a release a";
lc[0]->acquire(a);
check_grant(a);
lc[0]->release(a);
lc[0]->release(a);
check_release(a);
- LOG_NONMEMBER("acquire a acquire b release b release a");
+ LOG_NONMEMBER << "acquire a acquire b release b release a";
lc[0]->acquire(a);
check_grant(a);
lc[0]->acquire(b);
check_release(a);
}
-void test2(int i) {
- LOG_NONMEMBER("test2: client " << i << " acquire a release a");
+static void test2(int i) {
+ LOG_NONMEMBER << "test2: client " << i << " acquire a release a";
lc[i]->acquire(a);
- LOG_NONMEMBER("test2: client " << i << " acquire done");
+ LOG_NONMEMBER << "test2: client " << i << " acquire done";
check_grant(a);
- sleep(1);
- LOG_NONMEMBER("test2: client " << i << " release");
+ usleep(100000);
+ LOG_NONMEMBER << "test2: client " << i << " release";
check_release(a);
lc[i]->release(a);
- LOG_NONMEMBER("test2: client " << i << " release done");
+ LOG_NONMEMBER << "test2: client " << i << " release done";
}
-void test3(int i) {
- LOG_NONMEMBER("test3: client " << i << " acquire a release a concurrent");
+static void test3(int i) {
+ LOG_NONMEMBER << "test3: client " << i << " acquire a release a concurrent";
for (int j = 0; j < 10; j++) {
lc[i]->acquire(a);
check_grant(a);
- LOG_NONMEMBER("test3: client " << i << " got lock");
+ LOG_NONMEMBER << "test3: client " << i << " got lock";
check_release(a);
lc[i]->release(a);
}
}
-void test4(int i) {
- LOG_NONMEMBER("test4: thread " << i << " acquire a release a concurrent; same clnt");
+static void test4(int i) {
+ LOG_NONMEMBER << "test4: thread " << i << " acquire a release a concurrent; same clnt";
for (int j = 0; j < 10; j++) {
lc[0]->acquire(a);
check_grant(a);
- LOG_NONMEMBER("test4: thread " << i << " on client 0 got lock");
+ LOG_NONMEMBER << "test4: thread " << i << " on client 0 got lock";
check_release(a);
lc[0]->release(a);
}
}
-void test5(int i) {
- LOG_NONMEMBER("test5: client " << i << " acquire a release a concurrent; same and diff clnt");
+static void test5(int i) {
+ LOG_NONMEMBER << "test5: client " << i << " acquire a release a concurrent; same and diff clnt";
for (int j = 0; j < 10; j++) {
if (i < 5) lc[0]->acquire(a);
else lc[1]->acquire(a);
check_grant(a);
- LOG_NONMEMBER("test5: client " << i << " got lock");
+ LOG_NONMEMBER << "test5: client " << i << " got lock";
check_release(a);
if (i < 5) lc[0]->release(a);
else lc[1]->release(a);
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);
}
if (argc > 2) {
test = atoi(argv[2]);
if (test < 1 || test > 5) {
- LOG_NONMEMBER("Test number must be between 1 and 5");
+ LOG_NONMEMBER << "Test number must be between 1 and 5";
exit(1);
}
}
- LOG_NONMEMBER("cache lock client");
+ LOG_NONMEMBER << "cache lock client";
for (int i = 0; i < nt; i++) lc[i] = new lock_client(dst);
if (!test || test == 1) {
}
if (!test || test == 3) {
- LOG_NONMEMBER("test 3");
+ LOG_NONMEMBER << "test 3";
for (int i = 0; i < nt; i++)
th[i] = thread(test3, i);
}
if (!test || test == 4) {
- LOG_NONMEMBER("test 4");
+ LOG_NONMEMBER << "test 4";
for (int i = 0; i < 2; i++)
th[i] = thread(test4, i);
}
if (!test || test == 5) {
- LOG_NONMEMBER("test 5");
+ LOG_NONMEMBER << "test 5";
for (int i = 0; i < nt; i++)
th[i] = thread(test5, i);
th[i].join();
}
- LOG_NONMEMBER(argv[0] << ": passed all tests successfully");
+ LOG_NONMEMBER << argv[0] << ": passed all tests successfully";
}