win32/ntcpsocket.h

Go to the documentation of this file.

// LICENSE: (Please see the file COPYING for details)
//
// NUS - Nemesis Utilities System: A C++ application development framework 
// Copyright (C) 2006, 2008 Otavio Rodolfo Piske
//
//  This file is part of NUS
//
//  This library is free software; you can redistribute it and/or
//  modify it under the terms of the GNU Lesser General Public
//  License as published by the Free Software Foundation version 2.1
//  of the License.
//
//  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
//  Lesser General Public License for more details.
//
//  You should have received a copy of the GNU Lesser General Public
//  License along with this library; if not, write to the Free Software
//  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
//
#ifndef NTCPSOCKET_H
#define NTCPSOCKET_H

#include "ndefs.h"
#include "ndatastream.h"
#include "nstring.h"
#include "nabstractsocket.h"
#include "nexception.h"
#include "ndebug.h"

#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN
#endif
#include <winsock2.h>
#include <ws2tcpip.h>


class DllDeclSpec NTcpSocket: public NAbstractSocket {
      public:
            NTcpSocket(void);
            
            NTcpSocket(NHostAddress &host);
            
            NTcpSocket(const NString &host, nuint16 port);
            
            ~NTcpSocket(void);

            void setPort(nint16 port);
            void setSocketOptions(nint32 opt);
            
            NHostAddress peerAddress(void) const;
            
            NString peerName(void) const;
            
            nuint16 peerPort(void) const;
            
            void open(void);
            void closeDevice(void);
            
            void disconnectFromHost(void);
            void connectToHost();
            void execListen(void);
            
            NTcpSocket *waitForConnected(void);
            
            nint32 read(NDataStream *buffer);
            nint32 read(NDataStream *buffer, nuint32 bytes);
            nint32 readLine(NDataStream *buffer);
            nint32 write(const NDataStream &data, nuint32 bytes = 0);
            nint32 writeLine(const NDataStream &data, nuint32 bytes = 0);

      private:
            struct addrinfo *m_addr;
            SOCKET m_sock;
            
            NTcpSocket(nuint16 timeout, nint32 sock);
            
            void init(void);
            void execBind(void);
            
            void setPeer(struct sockaddr_in addr);
            void setSock(int sock);
            void setupNewSocket(NTcpSocket *sock, struct sockaddr_in addr);

            void throwSocketError(int err) const;
};

#endif // NTCPSOCKET_H

---