tests/streams/socketstream.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Name:        tests/streams/socketstream.cpp
   3 // Purpose:     Test wxSocketInputStream/wxSocketOutputStream
   4 // Author:      Vadim Zeitlin
   5 // RCS-ID:      $Id$
   6 // Copyright:   (c) 2008 Vadim Zeitlin
   7 // Licence:     wxWidgets licence
   8 ///////////////////////////////////////////////////////////////////////////////
   9
  10 // For compilers that support precompilation, includes "wx/wx.h".
  11 // and "wx/cppunit.h"
  12 #include "testprec.h"
  13
  14 #ifdef __BORLANDC__
  15     #pragma hdrstop
  16 #endif
  17
  18 // for all others, include the necessary headers
  19 #ifndef WX_PRECOMP
  20     #include "wx/log.h"
  21 #endif
  22
  23 #include "wx/socket.h"
  24 #include "wx/sckstrm.h"
  25 #include "wx/thread.h"
  26
  27 #include "bstream.h"
  28
  29 namespace
  30 {
  31
  32 const int TEST_PORT_READ = 0x7778;  // arbitrary, chosen because == "wx"
  33 const int TEST_PORT_WRITE = 0x7779; // well, "wy"
  34
  35 // these cond and mutex are used to minimize the risk of the main thread
  36 // Connect()-ing before this thread starts Accept()-ing connections but
  37 // unfortunately we can't make this truly safe, see comment in
  38 // SocketServerThread::Entry()
  39 wxMutex gs_mutex;
  40 wxCondition gs_cond(gs_mutex);
  41 } // anonymous namespace
  42
  43 // return address for the given port on local host
  44 static inline wxIPV4address LocalAddress(int port)
  45 {
  46     wxIPV4address addr;
  47     addr.LocalHost();
  48     addr.Service(port);
  49
  50     return addr;
  51 }
  52
  53 // A thread which creates a listening socket on the specified port and executes
  54 // the given function with each socket which connects to it
  55 class SocketServerThread : public wxThread
  56 {
  57 public:
  58     // port is the port to listen on and function will be called on each
  59     // accepted socket
  60     SocketServerThread(int port, void (*accept)(wxSocketBase&))
  61         : wxThread(wxTHREAD_JOINABLE),
  62           m_port(port),
  63           m_accept(accept)
  64     {
  65         Create();
  66         Run();
  67     }
  68
  69 protected:
  70     virtual void *Entry()
  71     {
  72         wxSocketServer srv(LocalAddress(m_port), wxSOCKET_REUSEADDR);
  73
  74         // FIXME: this is still not atomic, of course and the main thread could
  75         //        call Connect() before we have time to Accept() but there is
  76         //        no way to fix it with current API
  77         {
  78             wxMutexLocker lock(gs_mutex);
  79             gs_cond.Signal();
  80         }
  81
  82         wxSocketBase *socket = srv.Accept();
  83         if ( socket )
  84         {
  85             (*m_accept)(*socket);
  86             delete socket;
  87         }
  88
  89         return NULL;
  90     }
  91
  92     int m_port;
  93     void (*m_accept)(wxSocketBase&);
  94
  95     DECLARE_NO_COPY_CLASS(SocketServerThread)
  96 };
  97
  98 // The test case for socket streams
  99 class socketStream :
 100         public BaseStreamTestCase<wxSocketInputStream, wxSocketOutputStream>
 101 {
 102 public:
 103     socketStream();
 104     virtual ~socketStream();
 105
 106     virtual void setUp();
 107     virtual void tearDown();
 108
 109     CPPUNIT_TEST_SUITE(socketStream);
 110         // Base class stream tests the socketStream supports.
 111         CPPUNIT_TEST(Input_GetC);
 112         CPPUNIT_TEST(Input_Eof);
 113         CPPUNIT_TEST(Input_Read);
 114         CPPUNIT_TEST(Input_LastRead);
 115         CPPUNIT_TEST(Input_CanRead);
 116         CPPUNIT_TEST(Input_Peek);
 117         CPPUNIT_TEST(Input_Ungetch);
 118
 119         CPPUNIT_TEST(Output_PutC);
 120         CPPUNIT_TEST(Output_Write);
 121         CPPUNIT_TEST(Output_LastWrite);
 122     CPPUNIT_TEST_SUITE_END();
 123
 124 private:
 125     // Implement base class functions.
 126     virtual wxSocketInputStream  *DoCreateInStream();
 127     virtual wxSocketOutputStream *DoCreateOutStream();
 128
 129     // socket thread functions
 130     static void WriteSocket(wxSocketBase& socket)
 131     {
 132         socket.Write("hello, world!", 13);
 133     }
 134
 135     static void ReadSocket(wxSocketBase& socket)
 136     {
 137         char ch;
 138         while ( socket.Read(&ch, 1).LastCount() == 1 )
 139             ;
 140     }
 141
 142     wxSocketClient *m_readSocket,
 143                    *m_writeSocket;
 144     wxThread *m_writeThread,
 145              *m_readThread;
 146 };
 147
 148 socketStream::socketStream()
 149 {
 150     m_readSocket =
 151     m_writeSocket = NULL;
 152
 153     m_writeThread =
 154     m_readThread = NULL;
 155
 156     wxSocketBase::Initialize();
 157 }
 158
 159 socketStream::~socketStream()
 160 {
 161     wxSocketBase::Shutdown();
 162 }
 163
 164 void socketStream::setUp()
 165 {
 166     // create the socket threads and wait until they are ready to accept
 167     // connections (if we called Connect() before this happens, it would fail)
 168     {
 169         wxMutexLocker lock(gs_mutex);
 170
 171         m_writeThread =
 172             new SocketServerThread(TEST_PORT_READ, &socketStream::WriteSocket);
 173         CPPUNIT_ASSERT_EQUAL( wxCOND_NO_ERROR, gs_cond.Wait() );
 174
 175         m_readThread =
 176             new SocketServerThread(TEST_PORT_WRITE, &socketStream::ReadSocket);
 177         CPPUNIT_ASSERT_EQUAL( wxCOND_NO_ERROR, gs_cond.Wait() );
 178     }
 179
 180     m_readSocket = new wxSocketClient;
 181     m_readSocket->SetTimeout(3);
 182     CPPUNIT_ASSERT( m_readSocket->Connect(LocalAddress(TEST_PORT_READ)) );
 183
 184     m_writeSocket = new wxSocketClient;
 185     m_writeSocket->SetTimeout(3);
 186     CPPUNIT_ASSERT( m_writeSocket->Connect(LocalAddress(TEST_PORT_WRITE)) );
 187 }
 188
 189 void socketStream::tearDown()
 190 {
 191     wxDELETE(m_readSocket);
 192     wxDELETE(m_writeSocket);
 193
 194     m_writeThread->Wait();
 195     wxDELETE(m_writeThread);
 196
 197     m_readThread->Wait();
 198     wxDELETE(m_readThread);
 199 }
 200
 201 wxSocketInputStream *socketStream::DoCreateInStream()
 202 {
 203     wxSocketInputStream *pStrInStream = new wxSocketInputStream(*m_readSocket);
 204     CPPUNIT_ASSERT(pStrInStream->IsOk());
 205     return pStrInStream;
 206 }
 207
 208 wxSocketOutputStream *socketStream::DoCreateOutStream()
 209 {
 210     wxSocketOutputStream *pStrOutStream = new wxSocketOutputStream(*m_writeSocket);
 211     CPPUNIT_ASSERT(pStrOutStream->IsOk());
 212     return pStrOutStream;
 213 }
 214
 215 // Register the stream sub suite, by using some stream helper macro.
 216 STREAM_TEST_SUBSUITE_NAMED_REGISTRATION(socketStream)