X-Git-Url: http://developer.intra2net.com/git/?p=libt2n;a=blobdiff_plain;f=test%2Ftimeout.cpp;h=936cc6ab2a8e3ffc19a9b7df50049c14ca99d5f0;hp=0493d627a38bdbe6a01ce44bac90ce879586ece3;hb=cf902cadb87cc6f75ae92c2dc64742555dd4f0f0;hpb=45a2ebc9695c4d7be6548b7e0f800d117ae56a0b diff --git a/test/timeout.cpp b/test/timeout.cpp index 0493d62..936cc6a 100644 --- a/test/timeout.cpp +++ b/test/timeout.cpp @@ -1,9 +1,24 @@ -/*************************************************************************** - * Copyright (C) 2004 by Intra2net AG * - * info@intra2net.com * - * * - ***************************************************************************/ +/* +Copyright (C) 2004 by Intra2net AG +The software in this package is distributed under the GNU General +Public License version 2 (with a special exception described below). + +A copy of GNU General Public License (GPL) is included in this distribution, +in the file COPYING.GPL. + +As a special exception, if other files instantiate templates or use macros +or inline functions from this file, or you compile this file and link it +with other works to produce a work based on this file, this file +does not by itself cause the resulting work to be covered +by the GNU General Public License. + +However the source code for this file must still be made available +in accordance with section (3) of the GNU General Public License. + +This exception does not invalidate any other reasons why a work based +on this file might be covered by the GNU General Public License. +*/ #include #include #include @@ -17,9 +32,8 @@ #include -#include -#include -#include +#define BOOST_TEST_DYN_LINK +#include #include #include @@ -33,13 +47,12 @@ #include #include -#ifdef HAVE_CONFIG_H +#include "test_fixtures.hxx" + #include -#endif using namespace std; using namespace libt2n; -using namespace CppUnit; string testfunc2(const string& str) { @@ -112,40 +125,18 @@ class testfunc2_cmd : public libt2n::command BOOST_CLASS_EXPORT(testfunc2_cmd) BOOST_CLASS_EXPORT(testfunc2_res) -// this is an evil hack to get access to real_write, don't ever do this in an app!!! -class real_write_connection: public socket_server_connection -{ - public: - void real_write(const std::string& data) - { socket_write(data); } -}; - -class test_timeout : public TestFixture +class test_timeoutFixture : public KillChildOnShutdownFixture { - CPPUNIT_TEST_SUITE(test_timeout); - - CPPUNIT_TEST(HelloTimeoutNothing); - CPPUNIT_TEST(HelloTimeoutSlowData); - CPPUNIT_TEST(CommandTimeout); - CPPUNIT_TEST(CommandSlowResponse); - - CPPUNIT_TEST_SUITE_END(); - - public: - +protected: typedef uint32_t packet_size_indicator; - void setUp() - { } - - void tearDown() - { } - - void send_hello(string hello_string, socket_server* ss, unsigned int conn_id) + // 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 { - server_connection *sc=ss->get_connection(conn_id); - sc->write(hello_string); - } + public: + void real_write(const std::string& data) + { socket_write(data); } + }; void send_slow_raw_socket(string data, socket_server* ss, unsigned int conn_id) { @@ -163,68 +154,115 @@ class test_timeout : public TestFixture usleep(200000); } } +}; - void HelloTimeoutNothing() - { - pid_t pid; +BOOST_FIXTURE_TEST_SUITE(test_timeout, test_timeoutFixture) - switch(pid=fork()) +BOOST_AUTO_TEST_CASE(ConnectTimeout) +{ + switch(child_pid=fork()) + { + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child { - case -1: + try + { + socket_server ss("./socket"); + } catch(...) { - CPPUNIT_FAIL("fork error"); - break; + std::cerr << "exception in child. ignoring\n"; } - case 0: - // child + + // don't call atexit and stuff + _exit(0); + } + + default: + // parent + { + string data; + + // wait till server is up + sleep(1); + + string errormsg; + + socket_client_connection sc("./socket"); + + BOOST_CHECK_MESSAGE(sc.connection::is_closed() == true, "connection not closed"); + + BOOST_CHECK_MESSAGE(sc.get_last_error_msg() == string("no more retries left after connect error"), "wrong errormessage"); + } + } +} + +BOOST_AUTO_TEST_CASE(HelloTimeoutNothing) +{ + switch(child_pid=fork()) + { + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { socket_server ss("./socket"); // max 10 sec for (int i=0; i < 10; i++) ss.fill_buffer(1000000); - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string data; + // don't call atexit and stuff + _exit(0); + } - // wait till server is up - sleep(1); - socket_client_connection sc("./socket"); + default: + // parent + { + string data; - string errormsg; + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); + command_client cc(&sc,1000000,1000000); - try - { - command_client cc(sc,1000000,1000000); - } - catch(t2n_transfer_error &e) - { errormsg=e.what(); } - catch(...) - { throw; } + t2n_exception* ep=cc.get_constuctor_exception(); - CPPUNIT_ASSERT_EQUAL(string("timeout exceeded"),errormsg); - } + string errormsg; + if (ep) + errormsg=ep->what(); + + BOOST_CHECK_EQUAL(string("timeout exceeded"),errormsg); } } +} - void HelloTimeoutSlowData() +BOOST_AUTO_TEST_CASE(HelloTimeoutSlowData) +{ + switch(child_pid=fork()) { - pid_t pid; - - switch(pid=fork()) + case -1: { - case -1: - { - CPPUNIT_FAIL("fork error"); - break; - } - case 0: - // child + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { socket_server ss("./socket"); @@ -239,53 +277,54 @@ class test_timeout : public TestFixture std::string send_data(hello.str()); send_data.insert(0,(char*)&psize,sizeof(packet_size_indicator)); - ss.add_callback(new_connection,bind(&test_timeout::send_slow_raw_socket, boost::ref(*this), send_data,&ss, _1)); + ss.add_callback(new_connection,bind(&test_timeout::HelloTimeoutSlowData::send_slow_raw_socket, boost::ref(*this), send_data,&ss, _1)); // max 10 sec for (int i=0; i < 10; i++) ss.fill_buffer(1000000); - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string data; + // don't call atexit and stuff + _exit(0); + } - // wait till server is up - sleep(1); - socket_client_connection sc("./socket"); + default: + // parent + { + string data; - string errormsg; + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); + command_client cc(&sc,1000000,1000000); - try - { - command_client cc(sc,1000000,1000000); - } - catch(t2n_transfer_error &e) - { errormsg=e.what(); } - catch(...) - { throw; } + t2n_exception* ep=cc.get_constuctor_exception(); - CPPUNIT_ASSERT_EQUAL(string("timeout exceeded"),errormsg); - } + string errormsg; + if (ep) + errormsg=ep->what(); + + BOOST_CHECK_EQUAL(string("timeout exceeded"),errormsg); } } +} - void CommandTimeout() +BOOST_AUTO_TEST_CASE(CommandTimeout) +{ + switch(child_pid=fork()) { - pid_t pid; - - switch(pid=fork()) + case -1: { - case -1: - { - CPPUNIT_FAIL("fork error"); - break; - } - case 0: - // child + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { socket_server ss("./socket"); @@ -295,56 +334,59 @@ class test_timeout : public TestFixture hello.write((char*)&byteordercheck,sizeof(byteordercheck)); hello << ';'; - ss.add_callback(new_connection,bind(&test_timeout::send_hello, boost::ref(*this), hello.str(),&ss, _1)); + ss.add_callback(new_connection,bind(&test_timeout::CommandTimeout::send_hello, boost::ref(*this), hello.str(),&ss, _1)); // max 10 sec for (int i=0; i < 10; i++) ss.fill_buffer(1000000); - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string data; + // don't call atexit and stuff + _exit(0); + } - // wait till server is up - sleep(1); - socket_client_connection sc("./socket"); + default: + // parent + { + string data; - command_client cc(sc,1000000,1000000); - result_container rc; + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); - string errormsg; + command_client cc(&sc,1000000,1000000); + result_container rc; - try - { - cc.send_command(new testfunc2_cmd("hello"),rc); - } - catch(t2n_transfer_error &e) - { errormsg=e.what(); } - catch(...) - { throw; } + string errormsg; - CPPUNIT_ASSERT_EQUAL(string("timeout exceeded"),errormsg); + try + { + cc.send_command(new testfunc2_cmd("hello"),rc); } + catch(t2n_transfer_error &e) + { errormsg=e.what(); } + + BOOST_CHECK_EQUAL(string("timeout exceeded"),errormsg); } } +} - void CommandSlowResponse() +BOOST_AUTO_TEST_CASE(CommandSlowResponse) +{ + switch(child_pid=fork()) { - pid_t pid; - - switch(pid=fork()) + case -1: { - case -1: - { - CPPUNIT_FAIL("fork error"); - break; - } - case 0: - // child + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { socket_server ss("./socket"); @@ -354,7 +396,7 @@ class test_timeout : public TestFixture hello.write((char*)&byteordercheck,sizeof(byteordercheck)); hello << ';'; - ss.add_callback(new_connection,bind(&test_timeout::send_hello, boost::ref(*this), hello.str(),&ss, _1)); + ss.add_callback(new_connection,bind(&test_timeout::CommandSlowResponse::send_hello, boost::ref(*this), hello.str(),&ss, _1)); // max 10 sec for (int i=0; i < 10; i++) @@ -374,38 +416,361 @@ class test_timeout : public TestFixture send_slow_raw_socket(send_data,&ss,cid); } } - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent + // don't call atexit and stuff + _exit(0); + } + + default: + // parent + { + string data; + + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); + + command_client cc(&sc,1000000,1000000); + result_container rc; + + string errormsg; + + try { - string data; + cc.send_command(new testfunc2_cmd("hello"),rc); + } + catch(t2n_transfer_error &e) + { errormsg=e.what(); } + + BOOST_CHECK_EQUAL(string("timeout exceeded"),errormsg); + } + } +} + +BOOST_AUTO_TEST_CASE(DisconnectOnWrite) +{ + switch(child_pid=fork()) + { + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try + { + socket_server ss("./socket"); + + // bail out as soon as we get something + ss.fill_buffer(-1); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } + + // 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(); } + + BOOST_CHECK_EQUAL(string("write() returned Broken pipe"),errormsg); + } + } +} + +BOOST_AUTO_TEST_CASE(WriteTwice) +{ + switch(child_pid=fork()) + { + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try + { + socket_server ss("./socket"); + + // bail out as soon as we get something + ss.fill_buffer(-1); + ss.fill_buffer(-1); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } + + // 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; + sc.write("somedata"); + + sleep(1); + + // server should disconnect now + try + { + sc.write("other data(2)"); + sleep(1); + sc.write("other data(3)"); + } + catch(t2n_transfer_error &e) + { errormsg=e.what(); } + + BOOST_CHECK_EQUAL(string("write() returned Broken pipe"),errormsg); + } + } +} + +BOOST_AUTO_TEST_CASE(DisconnectOnRead) +{ + pid_t pid2; + + switch(child_pid=fork()) + { + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try + { // wait till server is up sleep(1); + socket_client_connection sc("./socket"); - command_client cc(sc,1000000,1000000); - result_container rc; + // 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')); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } + + // 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); + } - string errormsg; + // are we still alive and able to process data? - try + switch(pid2=fork()) + { + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child { - cc.send_command(new testfunc2_cmd("hello"),rc); + try + { + socket_client_connection *sc=new socket_client_connection("./socket"); + sc->write(string(10000,'x')); + // socket is closed regularly + delete sc; + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } + + // don't run regular cleanup, otherwise cppunit stuff gets called + _exit(0); } - catch(t2n_transfer_error &e) - { errormsg=e.what(); } - catch(...) - { throw; } - CPPUNIT_ASSERT_EQUAL(string("timeout exceeded"),errormsg); + 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; + } + + BOOST_CHECK_EQUAL(string(10000,'x'),received); + } } } } + kill(pid2,SIGKILL); +} -}; +BOOST_AUTO_TEST_CASE(BreakAccept) +{ + pid_t pid2; + + switch(child_pid=fork()) + { + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try + { + // 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); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } + + // 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: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try + { + socket_client_connection *sc=new socket_client_connection("./socket"); + sc->write(string(10000,'x')); + delete sc; + // socket is closed regularly + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } + + // 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; + } + + BOOST_CHECK_EQUAL(string(10000,'x'),received); + } + } + } + } + kill(pid2,SIGKILL); +} -CPPUNIT_TEST_SUITE_REGISTRATION(test_timeout); +BOOST_AUTO_TEST_SUITE_END()