X-Git-Url: http://xvm.mit.edu/gitweb/invirt/third/libt4.git/blobdiff_plain/f2170465073de34adf89161d4287182b518352c4..2546a41ad36fdc9ef6471cb35a1d56930ae1b527:/rpc/rpctest.cc diff --git a/rpc/rpctest.cc b/rpc/rpctest.cc index dbb10c6..bf8a56c 100644 --- a/rpc/rpctest.cc +++ b/rpc/rpctest.cc @@ -3,9 +3,10 @@ #include "rpc.h" #include -#include +#include +#include +#include #include -#include #include #include #include @@ -14,11 +15,17 @@ #define NUM_CL 2 -char tprintf_thread_prefix = 'r'; +char log_thread_prefix = 'r'; + +using std::string; +using std::cout; +using std::endl; +using std::vector; +using std::thread; rpcs *server; // server rpc object rpcc *clients[NUM_CL]; // client rpc object -struct sockaddr_in dst; //server's ip address +string dst; //server's ip address int port; // server-side handlers. they must be methods of some class @@ -26,10 +33,10 @@ int port; // from multiple classes. class srv { public: - int handle_22(std::string & r, const std::string a, const std::string b); + int handle_22(string & r, const string a, const string b); int handle_fast(int &r, const int a); int handle_slow(int &r, const int a); - int handle_bigrep(std::string &r, const size_t a); + int handle_bigrep(string &r, const size_t a); }; // a handler. a and b are arguments, r is the result. @@ -40,7 +47,7 @@ class srv { // at these argument types, so this function definition // does what a .x file does in SunRPC. int -srv::handle_22(std::string &r, const std::string a, std::string b) +srv::handle_22(string &r, const string a, string b) { r = a + b; return 0; @@ -62,9 +69,9 @@ srv::handle_slow(int &r, const int a) } int -srv::handle_bigrep(std::string &r, const size_t len) +srv::handle_bigrep(string &r, const size_t len) { - r = std::string((size_t)len, 'x'); + r = string((size_t)len, 'x'); return 0; } @@ -88,7 +95,7 @@ testmarshall() VERIFY(m.size()==RPC_HEADER_SZ); int i = 12345; unsigned long long l = 1223344455L; - std::string s = std::string("hallo...."); + string s = "hallo...."; m << i; m << l; m << s; @@ -104,7 +111,7 @@ testmarshall() VERIFY(memcmp(&rh,&rh1,sizeof(rh))==0); int i1; unsigned long long l1; - std::string s1; + string s1; un >> i1; un >> l1; un >> s1; @@ -120,12 +127,11 @@ client1(size_t cl) for(int i = 0; i < 100; i++){ int arg = (random() % 2000); - std::string rep; + string rep; int ret = clients[which_cl]->call(25, rep, arg); VERIFY(ret == 0); - if ((int)rep.size()!=arg) { - printf("repsize wrong %d!=%d\n", (int)rep.size(), arg); - } + if ((int)rep.size()!=arg) + cout << "repsize wrong " << rep.size() << "!=" << arg << endl; VERIFY((int)rep.size() == arg); } @@ -142,7 +148,7 @@ client1(size_t cl) auto end = std::chrono::steady_clock::now(); auto diff = std::chrono::duration_cast(end - start).count(); if (ret != 0) - printf("%d ms have elapsed!!!\n", (int)diff); + cout << diff << " ms have elapsed!!!" << endl; VERIFY(ret == 0); VERIFY(rep == (which ? arg+1 : arg+2)); } @@ -158,12 +164,10 @@ client2(size_t cl) while(time(0) - t1 < 10){ int arg = (random() % 2000); - std::string rep; + string rep; int ret = clients[which_cl]->call(25, rep, arg); - if ((int)rep.size()!=arg) { - printf("ask for %d reply got %d ret %d\n", - arg, (int)rep.size(), ret); - } + if ((int)rep.size()!=arg) + cout << "ask for " << arg << " reply got " << rep.size() << " ret " << ret << endl; VERIFY((int)rep.size() == arg); } } @@ -184,57 +188,53 @@ client3(void *xx) void simple_tests(rpcc *c) { - printf("simple_tests\n"); + cout << "simple_tests" << endl; // an RPC call to procedure #22. // rpcc::call() looks at the argument types to decide how // to marshall the RPC call packet, and how to unmarshall // the reply packet. - std::string rep; - int intret = c->call(22, rep, (std::string)"hello", (std::string)" goodbye"); + string rep; + int intret = c->call(22, rep, (string)"hello", (string)" goodbye"); VERIFY(intret == 0); // this is what handle_22 returns VERIFY(rep == "hello goodbye"); - printf(" -- string concat RPC .. ok\n"); + cout << " -- string concat RPC .. ok" << endl; // small request, big reply (perhaps req via UDP, reply via TCP) intret = c->call_timeout(25, rpcc::to(200000), rep, 70000); VERIFY(intret == 0); VERIFY(rep.size() == 70000); - printf(" -- small request, big reply .. ok\n"); + cout << " -- small request, big reply .. ok" << endl; // specify a timeout value to an RPC that should succeed (udp) int xx = 0; intret = c->call_timeout(23, rpcc::to(3000), xx, 77); VERIFY(intret == 0 && xx == 78); - printf(" -- no spurious timeout .. ok\n"); + cout << " -- no spurious timeout .. ok" << endl; // specify a timeout value to an RPC that should succeed (tcp) { - std::string arg(1000, 'x'); - std::string rep2; - c->call_timeout(22, rpcc::to(3000), rep2, arg, (std::string)"x"); + string arg(1000, 'x'); + string rep2; + c->call_timeout(22, rpcc::to(3000), rep2, arg, (string)"x"); VERIFY(rep2.size() == 1001); - printf(" -- no spurious timeout .. ok\n"); + cout << " -- no spurious timeout .. ok" << endl; } // huge RPC - std::string big(1000000, 'x'); - intret = c->call(22, rep, big, (std::string)"z"); + string big(1000000, 'x'); + intret = c->call(22, rep, big, (string)"z"); VERIFY(rep.size() == 1000001); - printf(" -- huge 1M rpc request .. ok\n"); + cout << " -- huge 1M rpc request .. ok" << endl; // specify a timeout value to an RPC that should timeout (udp) - struct sockaddr_in non_existent; - memset(&non_existent, 0, sizeof(non_existent)); - non_existent.sin_family = AF_INET; - non_existent.sin_addr.s_addr = inet_addr("127.0.0.1"); - non_existent.sin_port = htons(7661); + string non_existent = "127.0.0.1:7661"; rpcc *c1 = new rpcc(non_existent); time_t t0 = time(0); intret = c1->bind(rpcc::to(3000)); time_t t1 = time(0); VERIFY(intret < 0 && (t1 - t0) <= 4); - printf(" -- rpc timeout .. ok\n"); - printf("simple_tests OK\n"); + cout << " -- rpc timeout .. ok" << endl; + cout << "simple_tests OK" << endl; } void @@ -243,23 +243,23 @@ concurrent_test(size_t nt) // create threads that make lots of calls in parallel, // to test thread synchronization for concurrent calls // and dispatches. - printf("start concurrent_test (%lu threads) ...", nt); + cout << "start concurrent_test (" << nt << " threads) ..."; - std::vector th(nt); + vector th(nt); for(size_t i = 0; i < nt; i++) - th[i] = std::thread(client1, i); + th[i] = thread(client1, i); for(size_t i = 0; i < nt; i++) th[i].join(); - printf(" OK\n"); + cout << " OK" << endl; } void lossy_test() { - printf("start lossy_test ..."); + cout << "start lossy_test ..."; VERIFY(setenv("RPC_LOSSY", "5", 1) == 0); if (server) { @@ -275,15 +275,15 @@ lossy_test() size_t nt = 1; - std::vector th(nt); + vector th(nt); for(size_t i = 0; i < nt; i++) - th[i] = std::thread(client2, i); + th[i] = thread(client2, i); for(size_t i = 0; i < nt; i++) th[i].join(); - printf(".. OK\n"); + cout << ".. OK" << endl; VERIFY(setenv("RPC_LOSSY", "0", 1) == 0); } @@ -293,22 +293,22 @@ failure_test() rpcc *client1; rpcc *client = clients[0]; - printf("failure_test\n"); + cout << "failure_test" << endl; delete server; client1 = new rpcc(dst); VERIFY (client1->bind(rpcc::to(3000)) < 0); - printf(" -- create new client and try to bind to failed server .. failed ok\n"); + cout << " -- create new client and try to bind to failed server .. failed ok" << endl; delete client1; startserver(); - std::string rep; - int intret = client->call(22, rep, (std::string)"hello", (std::string)" goodbye"); + string rep; + int intret = client->call(22, rep, (string)"hello", (string)" goodbye"); VERIFY(intret == rpc_const::oldsrv_failure); - printf(" -- call recovered server with old client .. failed ok\n"); + cout << " -- call recovered server with old client .. failed ok" << endl; delete client; @@ -316,25 +316,25 @@ failure_test() VERIFY (client->bind() >= 0); VERIFY (client->bind() < 0); - intret = client->call(22, rep, (std::string)"hello", (std::string)" goodbye"); + intret = client->call(22, rep, (string)"hello", (string)" goodbye"); VERIFY(intret == 0); VERIFY(rep == "hello goodbye"); - printf(" -- delete existing rpc client, create replacement rpc client .. ok\n"); + cout << " -- delete existing rpc client, create replacement rpc client .. ok" << endl; size_t nt = 10; - printf(" -- concurrent test on new rpc client w/ %lu threads ..", nt); + cout << " -- concurrent test on new rpc client w/ " << nt << " threads .."; - std::vector th(nt); + vector th(nt); for(size_t i = 0; i < nt; i++) - th[i] = std::thread(client3, client); + th[i] = thread(client3, client); for(size_t i = 0; i < nt; i++) th[i].join(); - printf("ok\n"); + cout << "ok" << endl; delete server; delete client; @@ -342,19 +342,19 @@ failure_test() startserver(); clients[0] = client = new rpcc(dst); VERIFY (client->bind() >= 0); - printf(" -- delete existing rpc client and server, create replacements.. ok\n"); + cout << " -- delete existing rpc client and server, create replacements.. ok" << endl; - printf(" -- concurrent test on new client and server w/ %lu threads ..", nt); + cout << " -- concurrent test on new client and server w/ " << nt << " threads .."; for(size_t i = 0; i < nt; i++) - th[i] = std::thread(client3, client); + th[i] = thread(client3, client); for(size_t i = 0; i < nt; i++) th[i].join(); - printf("ok\n"); + cout << "ok" << endl; - printf("failure_test OK\n"); + cout << "failure_test OK" << endl; } int @@ -406,16 +406,13 @@ main(int argc, char *argv[]) testmarshall(); if (isserver) { - printf("starting server on port %d RPC_HEADER_SZ %d\n", port, (int)RPC_HEADER_SZ); + cout << "starting server on port " << port << " RPC_HEADER_SZ " << (int)RPC_HEADER_SZ << endl; startserver(); } if (isclient) { // server's address. - memset(&dst, 0, sizeof(dst)); - dst.sin_family = AF_INET; - dst.sin_addr.s_addr = inet_addr("127.0.0.1"); - dst.sin_port = htons(port); + dst = "127.0.0.1:" + std::to_string(port); // start the client. bind it to the server. @@ -435,7 +432,7 @@ main(int argc, char *argv[]) failure_test(); } - printf("rpctest OK\n"); + cout << "rpctest OK" << endl; exit(0); }