-/***************************************************************************
- * 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 <sys/types.h>
#include <unistd.h>
#include <errno.h>
#include <boost/bind.hpp>
-#include <cppunit/extensions/TestFactoryRegistry.h>
-#include <cppunit/ui/text/TestRunner.h>
-#include <cppunit/extensions/HelperMacros.h>
+#define BOOST_TEST_DYN_LINK
+#include <boost/test/unit_test.hpp>
#include <boost/archive/binary_oarchive.hpp>
#include <boost/archive/binary_iarchive.hpp>
#include <command_client.hxx>
#include <command_server.hxx>
-#ifdef HAVE_CONFIG_H
+#include "test_fixtures.hxx"
+
#include <config.h>
-#endif
using namespace std;
using namespace libt2n;
-using namespace CppUnit;
string testfunc2(const string& str)
{
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)
{
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");
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");
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");
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++)
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()
git://developer.intra2net.com / libt2n / blobdiff