/////////////////////////////////////////////////////////////////////////////
-// Name: sckstrm.h
+// Name: src/common/sckstrm.cpp
// Purpose: wxSocket*Stream
// Author: Guilhem Lavaux
// Modified by:
// Copyright: (c)
// Licence: wxWindows licence
/////////////////////////////////////////////////////////////////////////////
-#ifdef __GNUG__
-#pragma implementation "sckstrm.h"
-#endif
// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"
#ifdef __BORLANDC__
- #pragma hdrstop
+ #pragma hdrstop
#endif
+#if wxUSE_SOCKETS && wxUSE_STREAMS
+
+#include "wx/sckstrm.h"
+
#ifndef WX_PRECOMP
- #include "wx/defs.h"
+ #include "wx/stream.h"
#endif
-#if wxUSE_SOCKETS && wxUSE_STREAMS
-
-#include "wx/stream.h"
#include "wx/socket.h"
-#include "wx/sckstrm.h"
// ---------------------------------------------------------------------------
// wxSocketOutputStream
size_t wxSocketOutputStream::OnSysWrite(const void *buffer, size_t size)
{
- size_t ret;
-
- ret = m_o_socket->Write((const char *)buffer, size).LastCount();
-
- if (m_o_socket->Error())
- m_lasterror = wxStream_WRITE_ERR;
- else
- m_lasterror = wxStream_NOERROR;
-
- return ret;
-
+ const char *buf = (const char *)buffer;
+ size_t count = 0;
+
+ while ( count < size && m_o_socket->WaitForWrite() )
+ {
+ const size_t ret = m_o_socket->Write(buf, size - count).LastCount();
+
+ buf += ret;
+ count += ret;
+
+ if ( m_o_socket->Error() )
+ {
+ if (m_o_socket->LastError() != wxSOCKET_WOULDBLOCK)
+ {
+ m_lasterror = wxSTREAM_WRITE_ERROR;
+ return count;
+ }
+ }
+ }
+
+ m_lasterror = wxSTREAM_NO_ERROR;
+ return count;
}
// ---------------------------------------------------------------------------
size_t wxSocketInputStream::OnSysRead(void *buffer, size_t size)
{
- size_t ret;
-
- ret = m_i_socket->Read((char *)buffer, size).LastCount();
-
- if (m_i_socket->Error())
- m_lasterror = wxStream_READ_ERR;
- else
- m_lasterror = wxStream_NOERROR;
-
- return ret;
+ char *buf = (char *)buffer;
+ size_t count = 0;
+
+ while ( count < size && m_i_socket->WaitForRead() )
+ {
+ const size_t ret = m_i_socket->Read(buf, size - count).LastCount();
+
+ buf += ret;
+ count += ret;
+
+ if ( m_i_socket->Error() )
+ {
+ if (m_i_socket->LastError() != wxSOCKET_WOULDBLOCK)
+ {
+ m_lasterror = wxSTREAM_READ_ERROR;
+ return count;
+ }
+ }
+ }
+
+ m_lasterror = wxSTREAM_NO_ERROR;
+ return count;
}
// ---------------------------------------------------------------------------