+ char *buffer = (char *)buffer_;
+
+ int total;
+
+ // Try the pushback buffer first
+ total = GetPushback(buffer, nbytes, false);
+ nbytes -= total;
+ buffer = (char *)buffer + total;
+
+ // Return now in one of the following cases:
+ // - the socket is invalid,
+ // - we got all the data
+ if ( !m_socket ||
+ !nbytes )
+ return total;
+
+ // Possible combinations (they are checked in this order)
+ // wxSOCKET_NOWAIT
+ // wxSOCKET_WAITALL (with or without wxSOCKET_BLOCK)
+ // wxSOCKET_BLOCK
+ // wxSOCKET_NONE
+ //
+ int ret;
+ if (m_flags & wxSOCKET_NOWAIT)
+ {
+ m_socket->SetNonBlocking(1);
+ ret = m_socket->Read(buffer, nbytes);
+ m_socket->SetNonBlocking(0);
+
+ if ( ret < 0 )
+ return 0;
+
+ total += ret;
+ }
+ else // blocking socket
+ {
+ for ( ;; )
+ {
+ // dispatch events unless disabled
+ if ( !(m_flags & wxSOCKET_BLOCK) && !WaitForRead() )
+ break;
+
+ ret = m_socket->Read(buffer, nbytes);
+ if ( ret == 0 )
+ {
+ // for connection-oriented (e.g. TCP) sockets we can only read 0
+ // bytes if the other end has been closed, and for connectionless
+ // ones (UDP) this flag doesn't make sense anyhow so we can set it to
+ // true too without doing any harm
+ m_closed = true;
+ break;
+ }
+
+ if ( ret < 0 )
+ {
+ // this will be always interpreted as error by Read()
+ return 0;
+ }
+
+ total += ret;
+
+ // if wxSOCKET_WAITALL is not set, we can leave now as we did read
+ // something
+ if ( !(m_flags & wxSOCKET_WAITALL) )
+ break;
+
+ // otherwise check if read the maximal requested amount of data
+ nbytes -= ret;
+ if ( !nbytes )
+ break;
+
+ // we didn't, so continue reading
+ buffer = (char *)buffer + ret;
+ }
+ }
+
+ return total;
+}
+
+wxSocketBase& wxSocketBase::ReadMsg(void* buffer, wxUint32 nbytes)
+{
+ wxUint32 len, len2, sig, total;
+ bool error;
+ int old_flags;
+ struct
+ {
+ unsigned char sig[4];
+ unsigned char len[4];
projects / wxWidgets.git / blobdiff