netsocket.h

Go to the documentation of this file.

/********************************************************************************/
/*                                                                              */
/*  Netrinjo-Engine - a library for game-development                            */
/*  Copyright (C) 2005-2006 by Tobias Nadler                                    */
/*                                                                              */
/*  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; either                */
/*  version 2.1 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           */
/*  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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA   */
/*                                                                              */
/********************************************************************************/

#ifndef NETSOCKET_H
#define NETSOCKET_H

#include "SDL/SDL_net.h"
#include "basicobject.h"


#define FILTER_SERVER   1
#define MAX_DATALEN     0xFFFE

#define E_NET_NONE      0

#define E_NET_NO_HOST       -16

#define E_NET_DISCONNECTED  -17

#define E_NET_RAW_DATA      -32

#define E_NET_BUFTOOSMALL   -33

#define E_NET_OTHER         -64

class NetBasicManager;

class DECLSPEC NetSocket: public BasicObject
{
public:
    DECLSPEC NetSocket( const char *name, NetBasicManager *p, TCPsocket connection = 0 );
    DECLSPEC NetSocket( const char *name, NetBasicManager *p, IPaddress ip );
    DECLSPEC NetSocket( TiXmlElement *el, NetBasicManager *p );
    virtual DECLSPEC ~NetSocket();

    virtual bool isNetSocket() const    { return true; }

    bool isValid() const    { return bValid; }

    DECLSPEC bool connect( const char *host, int port );
    DECLSPEC bool connect( IPaddress &ip );
    DECLSPEC void disconnect();
    virtual bool connected() const  { return sock != 0; }
    virtual bool deleteOnDisconnected() const   { return bDelOnDisconn; }

    virtual DECLSPEC bool wait4data( Uint32 time );

    virtual bool sendData( char *data, unsigned int size );
    virtual bool sendUDP( char *data, unsigned int size );
    virtual int readData( char *buffer, int maxsize );
    virtual char *readData( int *size ) = 0;

    virtual bool isActive() const   { return bActive; }
    TCPsocket getSocket()   { return sock; }
    const IPaddress *getAddress() const { if(!sock) return NULL; return SDLNet_TCP_GetPeerAddress( sock ); }
    const char *getHostName() const { if(!sock) return NULL; return SDLNet_ResolveIP( SDLNet_TCP_GetPeerAddress( sock ) ); }

protected:
    virtual DECLSPEC void onInit();
    virtual DECLSPEC bool onReceive();
    virtual DECLSPEC void onReceiveUDP(UDPpacket &udp);
    virtual void onConnect()    {}
    virtual void onDisconnect() {}
    TCPsocket sock;
    bool bValid;
    bool bActive;
    bool bDelOnDisconn;
private:
    bool init();
    friend class NetBasicManager;
};


class NetSimpleManager;

class DECLSPEC NetStandardSocket : public NetSocket
{
public:
    DECLSPEC NetStandardSocket( const char *name, NetBasicManager *p, TCPsocket connection = 0 );
    DECLSPEC NetStandardSocket( const char *name, NetBasicManager *p, IPaddress ip, int filter = 0 );
    DECLSPEC NetStandardSocket( TiXmlElement *el, NetBasicManager *p );

    virtual DECLSPEC ~NetStandardSocket();

    virtual const char *getClassName()  { return "NetStandardSocket"; }
    
    virtual bool sendData( char *data, unsigned int size );
    virtual DECLSPEC bool sendData( char *data, unsigned int size, Uint32 condition, void *buffer = NULL );
    virtual DECLSPEC int readData( char *data, int maxsize );
    virtual DECLSPEC char *readData( int *size );

    void setReceiveCallback(BasicObject::Callback cb, BasicObject *o)   {cbRecieve.func = cb; cbRecieve.obj = o;}
    
    Uint32 filter;

protected:
    virtual DECLSPEC void onInit();
    virtual DECLSPEC bool onReceive();
    virtual DECLSPEC void onConnect();

private:
    BasicObject::FunctionData cbRecieve;
    // number of bytes to read before starting to read the next packet's size
    unsigned int nBytesUnread;
};


#endif

---