socket.hpp

来自「agentspark 机器人模拟代码 适用robocup 机器人步态模拟仿真（」· HPP 代码 · 共 175 行

// -*-c++-*-

/***************************************************************************
               socket.hpp  -  Base newtork socket class
                             -------------------
    begin                : 08-JAN-2003
    copyright            : (C) 2003 by The RoboCup Soccer Server
                           Maintenance Group.
    email                : [e-mail]
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU LGPL as published by the Free Software  *
 *   Foundation; either version 2 of the License, or (at your option) any  *
 *   later version.                                                        *
 *                                                                         *
 ***************************************************************************/

#ifndef RCSS_NET_SOCKET_HPP
#define RCSS_NET_SOCKET_HPP

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <boost/cstdint.hpp>
#include <boost/shared_ptr.hpp>
#include "addr.hpp"

#ifdef HAVE_WINSOCK2_H
#include "Winsock2.h"
#endif

namespace rcss
{
    namespace net
    {
                class Handler;

        class Socket
        {
        public:
#ifdef HAVE_SOCKETTYPE
                        typedef SOCKET SocketDesc;
#else
                        typedef int SocketDesc;
#endif

                        static const SocketDesc INVALIDSOCKET;

            enum CheckingType { CHECK, DONT_CHECK };

                public:
            static
            void
            closeFD( SocketDesc* s );

            Socket();

            Socket( SocketDesc s );

            virtual
            ~Socket();

            bool
            open();

            bool
            bind( const Addr& addr );

            Addr
            getName() const;

            bool
            connect( const Addr& addr );

            Addr
            getPeer() const;

            void
            close();

            int
            setCloseOnExec( bool on = true );

            int
            setNonBlocking( bool on = true );

            int
            setAsync( bool on = true );

            int
            setBroadcast( bool on = true );

            SocketDesc
            getFD() const;

            bool
            isOpen() const;

            bool
            isConnected() const;

            Addr
            getDest() const; // deprecated.  Use getPeer instead.

            int
            send( const char* msg,
                  size_t len,
                  const Addr& dest,
                  int flags = 0,
                  CheckingType check = CHECK );

            int
            send( const char* msg,
                  size_t len,
                  int flags = 0,
                  CheckingType check = CHECK );

            int
            recv( char* msg,
                  size_t len,
                  Addr& from,
                  int flags = 0,
                  CheckingType check = CHECK );

            int
            recv( char* msg,
                  size_t len,
                  int flags = 0,
                  CheckingType check = CHECK );

            bool
            accept( Socket& sock );
            
            Socket* accept(Addr& addr);

            bool
            listen( int backlog );

    // The following two methods allow a timeout to be specified.
    // Overall, it's slower than the untimed varients so if you do
    // need to specify a timeout and you just want it the recv to
    // block or not to block, then you are better off setting the
    // socket to blocking or non-blocking and using the version
    // without timeouts.
            int
            recv( int timeout,
                  char* msg,
                  size_t len,
                  Addr& from,
                  int flags = 0 );

            int
            recv( int timeout,
                  char* msg,
                  size_t len,
                  int flags = 0 );

        private:
            virtual
            bool
            doOpen( SocketDesc& fd ) = 0;

        private:
                        Handler* m_handler;
            boost::shared_ptr< SocketDesc > m_handle;
        };
    }
}

#endif