StreamedSocket.h

Go to the documentation of this file.

/*  StreamedSocket.h - Reliable send/recv socket class
    Copyright (C) 2001-2004 Mark Weaver
    Written by Mark Weaver <mark@npsl.co.uk>

    Part of the Open-Win32 library.
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public
    License along with this library; if not, write to the
    Free Software Foundation, Inc., 59 Temple Place - Suite 330,
    Boston, MA  02111-1307, USA.
*/

#ifndef OW32_StreamedSocket_h
#define OW32_StreamedSocket_h

#include <OW32/Socket.h>

// Open Win32 namespace
namespace OW32
{

template <class TSocket = CSocket> 
class CStreamedSocket : 
    public TSocket
{
private:
    CStreamedSocket(const CStreamedSocket& );
    CStreamedSocket& operator=(const CStreamedSocket& );

public:
    CStreamedSocket() {}

    CStreamedSocket(SOCKET s) :
        TSocket(s)
    {
    }

    // recv with a loop to read all requested data in stream
    virtual int recv(char* buf, int len)
    {
        int count=0;
        
        do
        {
            int result = TSocket::recv(buf+count, len-count);
            if (result <= 0)
                return result;
            count += result;
        } while (count < len);

        return count;
    }

    // send with the same
    virtual int send(const char *buf, int len)
    {
        int count=0;

        do
        {
            int result = TSocket::send(buf+count, len-count);
            if (result <= 0)
                return result;
            count += result;
        } while (count < len);

        return count;
    }
};

} // namespace OW32

#endif // OW32_StreamedSocket_h

---