Prepare libt2n 0.7 release
[libt2n] / test / timeout.cpp
index 0493d62..936cc6a 100644 (file)
@@ -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 <sys/types.h>
 #include <unistd.h>
 #include <errno.h>
@@ -17,9 +32,8 @@
 
 #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)
 {
@@ -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*)&sc;
+                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()