TPSocket.hpp

/*
    Copyright (C) 2009-2011  EPFL (Ecole Polytechnique Fédérale de Lausanne)
    Michele Tavella <michele.tavella@epfl.ch>

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

    This program 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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.

        TPSocket.hpp/.cpp is adapted from libtransport
*/

#ifndef TPSOCKET_HPP 
#define TPSOCKET_HPP 

#ifdef _WIN32
#define UNICODE
#define _WIN32_WINNT 0x0501
#include <winsock2.h>
#include <ws2tcpip.h>
#else
#include <arpa/inet.h>
#include <netdb.h>
#endif
#include <string>

class TPHost {
        public:
                TPHost(void);

        public:
                const static size_t AddressSize = INET_ADDRSTRLEN + 1;
                char address[TPHost::AddressSize];
                short unsigned int port;
};

class TPSocket {
        public:
                TPSocket(int type = TPSocket::TCP, size_t bsize = 4096);
        
                virtual ~TPSocket(void);
                
                bool Open(bool asserver);
                
                bool Close(void);
                
                bool Async(bool block);
                
                bool Bind(const std::string &ip, const std::string& port);
                
                bool Listen(void);
                
                int Accept(TPSocket* endpoint);
                
                bool Connect(const std::string& ip, const std::string& port);
                
                ssize_t Send(const std::string& message);
                
                ssize_t Recv(std::string* message);
                
                bool IsConnected(void);

#ifdef _WIN32
                static bool InitializeWSA(void);
#endif

        protected:
                void Init(void);
                void Free(void);

                bool GetLocal(void);
                bool GetRemote(void);
                void GetMaxBSize(void);

        public:
                TPHost local;

                TPHost remote;

                enum {
                        TCP = 0,
                        UDP = 1
                };

        protected:
                int _fd;
                struct sockaddr_in _address;
                struct sockaddr_storage _endpoint;
                void* _buffer;
                size_t _bsize;
                unsigned int _mc;
                struct addrinfo _results;
                struct addrinfo* _info;
                size_t _bsizemax;
        private:
                int _type;
#ifdef _WIN32
                static bool _wsaInitialized;
#endif
};

#endif