SimpleUDP.h

Go to the documentation of this file.

 
#pragma once
 
#include <aws/core/Core_EXPORTS.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <cstdint>
struct sockaddr;
 
namespace Aws
{
    namespace Net
    {
        // 8K is aligned with default monitoring packet size.
        const static size_t UDP_BUFFER_SIZE = 8192;
        class AWS_CORE_API SimpleUDP
        {
        public:
            SimpleUDP(int addressFamily, size_t sendBufSize = UDP_BUFFER_SIZE, size_t receiveBufSize = UDP_BUFFER_SIZE, bool nonBlocking = true);
 
            SimpleUDP(bool IPV4 = true, size_t sendBufSize = UDP_BUFFER_SIZE, size_t receiveBufSize = UDP_BUFFER_SIZE, bool nonBlocking = true);
 
            SimpleUDP(const char* host, unsigned short port, size_t sendBufSize = UDP_BUFFER_SIZE, size_t receiveBufSize = UDP_BUFFER_SIZE, bool nonBlocking = true);
 
            ~SimpleUDP();
 
            int Connect(const sockaddr* address, size_t addressLength);
 
            int ConnectToHost(const char* hostIP, unsigned short port) const;
 
            int ConnectToLocalHost(unsigned short port) const;
 
            int Bind(const sockaddr* address, size_t addressLength) const;
 
            int BindToLocalHost(unsigned short port) const;
 
            int SendData(const uint8_t* data, size_t dataLen) const;
 
            int SendDataTo(const sockaddr* address, size_t addressLength, const uint8_t* data, size_t dataLen) const;
 
            int SendDataToLocalHost(const uint8_t* data, size_t dataLen, unsigned short port) const;
 
            int ReceiveData(uint8_t* buffer, size_t bufferLen) const;
 
            int ReceiveDataFrom(sockaddr* address, size_t* addressLength, uint8_t* buffer, size_t bufferLen) const;
 
            inline int GetAddressFamily() const { return m_addressFamily; }
 
            inline bool IsConnected() const { return m_connected; }
 
        private:
            void CreateSocket(int addressFamily, size_t sendBufSize, size_t receiveBufSize, bool nonBlocking);
            int GetUnderlyingSocket() const { return m_socket; }
            void SetUnderlyingSocket(int socket) { m_socket = socket; } 
            int m_addressFamily;
            // if not connected, you can't perform SendData, if connected,  SendDataTo will call SendData
            mutable bool m_connected;
            int m_socket;
            unsigned short m_port;
            Aws::String m_hostIP;
        };
    }
}