Socket.h

Go to the documentation of this file.

/*
 * Copyright (c) 2018, NVIDIA CORPORATION. All rights reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */
 
#ifndef __NETWORK_SOCKET_H_
#define __NETWORK_SOCKET_H_
 
#include <stdint.h>
#include <cstddef>
 
 
enum SocketType
{
        SOCKET_UDP  = 0,
 
        SOCKET_TCP  = 1
};
 
 
#define IP_ANY                  0x00000000
#define IP_BROADCAST    0xFFFFFFFF
#define IP_LOOPBACK     0x7F000001
 
 
class Socket
{
public:
        static Socket* Create( SocketType type );
 
        ~Socket();
 
        bool Accept( uint64_t timeout=0 );      
 
        bool Bind( const char* localIP, uint16_t port );
 
        bool Bind( uint32_t localIP, uint16_t port );
 
        bool Bind( uint16_t port=0 );
 
        bool Connect( const char* remoteIP, uint16_t port );
 
        bool Connect( uint32_t remoteIP, uint16_t port );
 
        size_t Recieve( uint8_t* buffer, size_t size, uint32_t* remoteIP=NULL, uint16_t* remotePort=NULL );
 
        size_t Recieve( uint8_t* buffer, size_t size, uint32_t* remoteIP, uint16_t* remotePort, uint32_t* localIP );
        
        bool Send( void* buffer, size_t size, uint32_t remoteIP, uint16_t remotePort );
        
        bool SetRecieveTimeout( uint64_t timeout );
        
        bool SetBufferSize( size_t size );
 
        bool EnableJumboBuffer();
 
        inline int GetFD() const                                                                                        { return mSock; }
 
        inline SocketType GetType() const                                                                       { return mType; }
 
        void PrintIP() const;
 
        size_t GetMTU();
        
private:
        
        Socket( SocketType type );
        bool Init();
 
        int        mSock;
        SocketType mType;
        bool       mListening;
        bool       mPktInfoEnabled;
        bool       mBroadcastEnabled;
        
        uint32_t   mLocalIP;
        uint16_t   mLocalPort;
 
        uint32_t   mRemoteIP;
        uint16_t   mRemotePort; 
};
 
#endif