Socket.hpp

//
// Created by xiong on 2022/6/9.
//
 
#ifndef EMX_Socket_HPP
#define EMX_Socket_HPP
 
#include "core/EmxTypeDef.hpp"
#include <sys/socket.h>
#include <sys/un.h>
 
namespace Emx {
    class Socket {
    public:
        Socket() {
            m_sock = -1;
            m_sendTimeout = -1;
            m_recvTimeout = -1;
        }
 
        virtual ~Socket() {
            Close();
        }
 
        virtual ErrCodeE Open() = 0;
 
        void Close() {
            if (m_sock > 0) {
                close(m_sock);
                m_sock = -1;
            }
        }
 
        ErrCodeE Bind(const struct sockaddr *addr, socklen_t addrSize);
 
        ErrCodeE SetSendTimeOut(int32_t timeoutMs);
 
        ErrCodeE SetRecvTimeOut(int32_t timeoutMs);
 
        int m_sock;
        int m_sendTimeout;
        int m_recvTimeout;
    };
 
    class SocketStream : public Socket {
    public:
        SocketStream() : Socket() {}
 
        virtual ~SocketStream() {};
 
        ErrCodeE Connect(const sockaddr *addr, int32_t addrSize, int32_t timeoutMs);
 
        ErrCodeE Send(const char *buffer, int32_t size, int32_t timeoutMs);
 
        ErrCodeE Recv(char *buffer, int32_t size, int32_t timeoutMs);
    };
 
    class SocketDgram : public Socket {
    public:
        SocketDgram() : Socket() {}
 
        virtual ~SocketDgram() {};
 
        ErrCodeE Send(const char *buffer, int32_t size, const sockaddr *addr, socklen_t addrLen, int32_t timeoutMs);
 
        ErrCodeE Recv(char *buffer, int32_t size, sockaddr *addr, socklen_t *addrLen, int32_t timeoutMs);
    };
 
    class SocketUnStream : public SocketStream {
    public:
        SocketUnStream() : SocketStream() {}
 
        virtual ~SocketUnStream();
 
        ErrCodeE Open() override;
 
        static void MakeAddr(sockaddr_un &un, const char *path);
    };
 
    class SocketUnDgram : public SocketDgram {
    public:
        SocketUnDgram() : SocketDgram() {}
 
        virtual ~SocketUnDgram();
 
        ErrCodeE Open() override;
 
        static void MakeAddr(sockaddr_un &un, const char *path);
    };
}
 
#endif //EMX_Socket_HPP