X-Git-Url: http://developer.intra2net.com/git/?p=libt2n;a=blobdiff_plain;f=test%2Ftimeout.cpp;h=7a4e9ed7c9a4d58f5f087aacefb12029c43d0e80;hp=0493d627a38bdbe6a01ce44bac90ce879586ece3;hb=e72e07f3178ced4277066c5df4dfd5934fac49e8;hpb=45a2ebc9695c4d7be6548b7e0f800d117ae56a0b diff --git a/test/timeout.cpp b/test/timeout.cpp index 0493d62..7a4e9ed 100644 --- a/test/timeout.cpp +++ b/test/timeout.cpp @@ -120,17 +120,31 @@ class real_write_connection: public socket_server_connection { socket_write(data); } }; +// this is an evil hack to get access to real_write, don't ever do this in an app!!! +class real_write_client_connection: public socket_client_connection +{ + public: + void real_write(const std::string& data) + { socket_write(data); } +}; + class test_timeout : public TestFixture { CPPUNIT_TEST_SUITE(test_timeout); + CPPUNIT_TEST(ConnectTimeout); CPPUNIT_TEST(HelloTimeoutNothing); CPPUNIT_TEST(HelloTimeoutSlowData); CPPUNIT_TEST(CommandTimeout); CPPUNIT_TEST(CommandSlowResponse); + CPPUNIT_TEST(DisconnectOnWrite); + CPPUNIT_TEST(DisconnectOnRead); + CPPUNIT_TEST(BreakAccept); CPPUNIT_TEST_SUITE_END(); + pid_t child_pid; + public: typedef uint32_t packet_size_indicator; @@ -139,7 +153,11 @@ class test_timeout : public TestFixture { } void tearDown() - { } + { + // make sure the server-child is dead before the next test runs + kill(child_pid,SIGKILL); + sleep(1); + } void send_hello(string hello_string, socket_server* ss, unsigned int conn_id) { @@ -164,11 +182,46 @@ class test_timeout : public TestFixture } } - void HelloTimeoutNothing() + void ConnectTimeout() { - pid_t pid; + switch(child_pid=fork()) + { + case -1: + { + CPPUNIT_FAIL("fork error"); + break; + } + case 0: + // child + { + socket_server ss("./socket"); + + // don't call atexit and stuff + _exit(0); + } - switch(pid=fork()) + default: + // parent + { + string data; + + // wait till server is up + sleep(1); + + string errormsg; + + socket_client_connection sc("./socket"); + + CPPUNIT_ASSERT_EQUAL_MESSAGE("connection not closed",true,sc.connection::is_closed()); + + CPPUNIT_ASSERT_EQUAL_MESSAGE("wrong errormessage",string("no more retries left after connect error"),sc.get_last_error_msg()); + } + } + } + + void HelloTimeoutNothing() + { + switch(child_pid=fork()) { case -1: { @@ -195,17 +248,13 @@ class test_timeout : public TestFixture // wait till server is up sleep(1); socket_client_connection sc("./socket"); + command_client cc(&sc,1000000,1000000); - string errormsg; + t2n_exception* ep=cc.get_constuctor_exception(); - try - { - command_client cc(sc,1000000,1000000); - } - catch(t2n_transfer_error &e) - { errormsg=e.what(); } - catch(...) - { throw; } + string errormsg; + if (ep) + errormsg=ep->what(); CPPUNIT_ASSERT_EQUAL(string("timeout exceeded"),errormsg); } @@ -214,9 +263,7 @@ class test_timeout : public TestFixture void HelloTimeoutSlowData() { - pid_t pid; - - switch(pid=fork()) + switch(child_pid=fork()) { case -1: { @@ -256,17 +303,13 @@ class test_timeout : public TestFixture // wait till server is up sleep(1); socket_client_connection sc("./socket"); + command_client cc(&sc,1000000,1000000); - string errormsg; + t2n_exception* ep=cc.get_constuctor_exception(); - try - { - command_client cc(sc,1000000,1000000); - } - catch(t2n_transfer_error &e) - { errormsg=e.what(); } - catch(...) - { throw; } + string errormsg; + if (ep) + errormsg=ep->what(); CPPUNIT_ASSERT_EQUAL(string("timeout exceeded"),errormsg); } @@ -275,9 +318,7 @@ class test_timeout : public TestFixture void CommandTimeout() { - pid_t pid; - - switch(pid=fork()) + switch(child_pid=fork()) { case -1: { @@ -313,7 +354,7 @@ class test_timeout : public TestFixture sleep(1); socket_client_connection sc("./socket"); - command_client cc(sc,1000000,1000000); + command_client cc(&sc,1000000,1000000); result_container rc; string errormsg; @@ -334,9 +375,7 @@ class test_timeout : public TestFixture void CommandSlowResponse() { - pid_t pid; - - switch(pid=fork()) + switch(child_pid=fork()) { case -1: { @@ -387,7 +426,7 @@ class test_timeout : public TestFixture sleep(1); socket_client_connection sc("./socket"); - command_client cc(sc,1000000,1000000); + command_client cc(&sc,1000000,1000000); result_container rc; string errormsg; @@ -406,6 +445,237 @@ class test_timeout : public TestFixture } } + void DisconnectOnWrite() + { + switch(child_pid=fork()) + { + case -1: + { + CPPUNIT_FAIL("fork error"); + break; + } + case 0: + // child + { + socket_server ss("./socket"); + + // bail out as soon as we get something + ss.fill_buffer(-1); + // don't call atexit and stuff + _exit(0); + } + + default: + // parent + { + string data; + + // don't kill us on broken pipe + signal(SIGPIPE, SIG_IGN); + + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); + + string errormsg; + + string huge(5000000,'x'); + + try + { + sc.write(huge); + } + catch(t2n_transfer_error &e) + { errormsg=e.what(); } + catch(...) + { throw; } + + CPPUNIT_ASSERT_EQUAL(string("write() returned Broken pipe"),errormsg); + } + } + } + + void DisconnectOnRead() + { + pid_t pid2; + + switch(child_pid=fork()) + { + case -1: + { + CPPUNIT_FAIL("fork error"); + break; + } + case 0: + // child + { + // wait till server is up + sleep(1); + + socket_client_connection sc("./socket"); + + // this is an evil hack to get access to real_write, don't ever do this in an app!!! + real_write_client_connection *rwc=(real_write_client_connection*)≻ + rwc->real_write(string(10000,'x')); + + // don't call atexit and stuff + _exit(0); + } + + default: + // parent + { + // don't kill us on broken pipe + signal(SIGPIPE, SIG_IGN); + + socket_server ss("./socket"); + + time_t t0 = time(NULL); + + // max 5 sec + while (time(NULL) < t0 + 5 ) + { + ss.fill_buffer(1000000); + + string data; + ss.get_packet(data); + } + + // are we still alive and able to process data? + + switch(pid2=fork()) + { + case -1: + { + CPPUNIT_FAIL("fork error"); + break; + } + case 0: + // child + { + socket_client_connection *sc=new socket_client_connection("./socket"); + sc->write(string(10000,'x')); + delete sc; + // socket is closed regularly + + // don't run regular cleanup, otherwise cppunit stuff gets called + _exit(0); + } + + default: + // parent + { + string received; + + t0 = time(NULL); + + // max 10 sec + while (time(NULL) < t0 + 10 ) + { + ss.fill_buffer(1000000); + + if (ss.get_packet(received)) + break; + } + + CPPUNIT_ASSERT_EQUAL(string(10000,'x'),received); + } + } + } + } + kill(pid2,SIGKILL); + } + + void BreakAccept() + { + pid_t pid2; + + switch(child_pid=fork()) + { + case -1: + { + CPPUNIT_FAIL("fork error"); + break; + } + case 0: + // child + { + // wait till server is really up and waiting + sleep(2); + + // connect with very tight timeout and only 1 retry + socket_client_connection sc("./socket",50,1); + + // don't call atexit and stuff + _exit(0); + } + + default: + // parent + { + // don't kill us on broken pipe + signal(SIGPIPE, SIG_IGN); + + socket_server ss("./socket"); + + // server is "working" while client wants to connect + sleep(5); + + time_t t0 = time(NULL); + + // max 5 sec + while (time(NULL) < t0 + 5 ) + { + ss.fill_buffer(1000000); + + string data; + ss.get_packet(data); + } + + // are we still alive and able to process data? + + switch(pid2=fork()) + { + case -1: + { + CPPUNIT_FAIL("fork error"); + break; + } + case 0: + // child + { + socket_client_connection *sc=new socket_client_connection("./socket"); + sc->write(string(10000,'x')); + delete sc; + // socket is closed regularly + + // don't run regular cleanup, otherwise cppunit stuff gets called + _exit(0); + } + + default: + // parent + { + string received; + + t0 = time(NULL); + + // max 10 sec + while (time(NULL) < t0 + 10 ) + { + ss.fill_buffer(1000000); + + if (ss.get_packet(received)) + break; + } + + CPPUNIT_ASSERT_EQUAL(string(10000,'x'),received); + } + } + } + } + kill(pid2,SIGKILL); + } }; CPPUNIT_TEST_SUITE_REGISTRATION(test_timeout);