psocket.h

Go to the documentation of this file.

/*
 *   P::Classes - Portable C++ Application Framework
 *   Copyright (C) 2000-2004  Christian Prochnow <cproch@seculogix.de>
 *
 *   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 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 _psocket_h_
#define _psocket_h_

#ifdef WIN32
  #include <winsock2.h>
#else
  #include <sys/types.h>
  #include <sys/socket.h>
#endif

#include <pclasses/pexport.h>
#include <pclasses/ptypes.h>
#include <pclasses/piodevice.h>
#include <pclasses/pnetaddr.h>
#include <string>

namespace P {

typedef uint16_t port_t;


class PNET_EXPORT Socket: public IODevice {
  public:

    Socket(int domain, int type, int proto) throw(IOError);

    Socket(io_handle_t handle, int domain, int type, int proto) throw(IOError);

    ~Socket() throw();

    void close() throw(IOError);

    size_t write(const char* buffer, size_t count) throw(IOError);

    size_t peek(char* buffer, size_t count) throw(IOError);

    size_t read(char* buffer, size_t count) throw(IOError);

    inline bool isSeekable() const throw()
    { return false; }

    inline int domain() const throw()
    { return m_domain; }

    inline int type() const throw()
    { return m_type; }

    inline int protocol() const throw()
    { return m_protocol; }

    enum wait_t {
      pendingInput   = 0x01,  
      pendingOutput  = 0x02   
    };


    int wait(int wait, unsigned int timeout) throw(IOError);

    void setBroadcast(bool enable) throw(IOError);

    void setRouting(bool enable) throw(IOError);

    void bind(const NetworkAddress& addr, port_t port) throw(IOError);

    void getName(NetworkAddress& addr, port_t& port) const throw(LogicError,IOError);

    void getPeer(NetworkAddress& addr, port_t& port) const throw(LogicError,IOError);

  private:
    int m_domain;
    int m_type;
    int m_protocol;
};


class PNET_EXPORT StreamSocketServer: protected Socket {
  public:
    friend class StreamSocket;


    StreamSocketServer(int domain) throw(IOError);


    StreamSocketServer(const NetworkAddress& addr, port_t port) throw(IOError);

    ~StreamSocketServer() throw();


    void listen() throw(IOError);


    bool waitClient(unsigned int timeout) throw(IOError);

    inline void getName(NetworkAddress& addr, port_t& port) const throw(LogicError,IOError)
    { Socket::getName(addr, port); }

  protected:

    io_handle_t accept() throw(IOError);

  private:
    StreamSocketServer(const StreamSocketServer&);
    StreamSocketServer& operator=(const StreamSocketServer&);

    io_handle_t m_lastAcceptHandle;
};


class PNET_EXPORT StreamSocket: public Socket {
  public:
    enum sdmode_t {
      disableRecv = 0,  
      disableSend,      
      disableBoth       
    };

    StreamSocket(const NetworkAddress& addr, port_t port) throw(IOError);

    StreamSocket(int domain) throw(IOError);

    StreamSocket(StreamSocketServer& srv) throw(IOError);

    ~StreamSocket() throw();

    void close() throw(IOError);

    void shutdown(sdmode_t mode);

    void connect(const NetworkAddress& addr, port_t port) throw(IOError);

    std::string readLine(unsigned int timeout) throw(IOError);

};


class PNET_EXPORT DatagramSocket: public Socket {
  public:
    DatagramSocket(const NetworkAddress& addr, port_t port) throw(IOError);

    DatagramSocket(int domain) throw(IOError);

    ~DatagramSocket() throw();

    size_t write(const char* buffer, size_t count) throw(IOError);

    size_t read(char* buffer, size_t count) throw(IOError);

    size_t peek(char* buffer, size_t count) throw(IOError);

    void setPeer(const NetworkAddress& addr, port_t port) throw();

    const NetworkAddress& getSender(port_t& port) const throw();

  private:
    void retrSenderAddr(sockaddr* sa, size_t salen);

    NetworkAddress* m_peer;
    port_t          m_peerPort;
    NetworkAddress* m_sender;
    port_t          m_senderPort;
};

}

#endif

---