socket.h

Go to the documentation of this file.

/************************************************************************
 ** This file is part of the network simulator Shawn.                  **
 ** Copyright (C) 2004-2007 by the SwarmNet (www.swarmnet.de) project  **
 ** Shawn is free software; you can redistribute it and/or modify it   **
 ** under the terms of the BSD License. Refer to the shawn-licence.txt **
 ** file in the root of the Shawn source tree for further details.     **
 ************************************************************************/

#ifndef __SHAWN_APPS_TCPIP_SOCKET_H
#define __SHAWN_APPS_TCPIP_SOCKET_H

#ifdef SHAWN
     #include <shawn_config.h>
     #include "_apps_enable_cmake.h"
     #ifdef ENABLE_TCPIP
            #define BUILD_TCPIP
     #endif
#else
     #define BUILD_TCPIP
#endif


#ifdef BUILD_TCPIP

// Get Storage
#ifdef SHAWN
    #include <apps/tcpip/storage.h>
#else
    #include "storage.h"
#endif

#ifdef SHAWN
     namespace shawn
      { class SimulationController; }

     // Dummy function is called when Shawn Simulation starts. Does nothing up to now.
     extern "C" void init_tcpip( shawn::SimulationController& );
#endif

#include <string>
#include <map>
#include <vector>
#include <list>
#include <deque>
#include <iostream>
#include <cstddef>


struct sockaddr_in;

namespace tcpip
{

    class SocketException: public std::runtime_error
    {
    public:
        SocketException(std::string what) : std::runtime_error(what.c_str()) {}
    };

    class Socket
    {
        friend class Response;
    public:
        Socket(std::string host, int port);
        
        Socket(int port);

        ~Socket();

        void connect();

        Socket* accept(const bool create = false);

        void send( const std::vector<unsigned char> &buffer);
        void sendExact( const Storage & );
        std::vector<unsigned char> receive( int bufSize = 2048 );
        bool receiveExact( Storage &);
        void close();
        int port();
        void set_blocking(bool);
        bool is_blocking();
        bool has_client_connection() const;

        // If verbose, each send and received data is written to stderr
        bool verbose() { return verbose_; }
        void set_verbose(bool newVerbose) { verbose_ = newVerbose; }

    protected:
        static const int lengthLen;

        void receiveComplete(unsigned char * const buffer, std::size_t len) const;
        size_t recvAndCheck(unsigned char * const buffer, std::size_t len) const;
        void printBufferOnVerbose(const std::vector<unsigned char> buffer, const std::string &label) const;

    private:
        void init();
        void BailOnSocketError(std::string context) const;
#ifdef WIN32
        std::string GetWinsockErrorString(int err) const;
#endif
        bool atoaddr(std::string, struct sockaddr_in& addr);
        bool datawaiting(int sock) const;

        std::string host_;
        int port_;
        int socket_;
        int server_socket_;
        bool blocking_;

        bool verbose_;
#ifdef WIN32
        static bool init_windows_sockets_;
        static bool windows_sockets_initialized_;
        static int instance_count_;
#endif
    };

}   // namespace tcpip

#endif // BUILD_TCPIP

#endif

/*-----------------------------------------------------------------------
* Source  $Source: $
* Version $Revision: 612 $
* Date    $Date: 2011-06-14 15:16:52 +0200 (Tue, 14 Jun 2011) $
*-----------------------------------------------------------------------
* $Log:$
*-----------------------------------------------------------------------*/