Socket.h

Go to the documentation of this file.

/***********************************************************************************************************************
*                                                                                                                      *
* ANTIKERNEL                                                                                                           *
*                                                                                                                      *
* Copyright (c) 2012-2025 Andrew D. Zonenberg                                                                          *
* All rights reserved.                                                                                                 *
*                                                                                                                      *
* Redistribution and use in source and binary forms, with or without modification, are permitted provided that the     *
* following conditions are met:                                                                                        *
*                                                                                                                      *
*    * Redistributions of source code must retain the above copyright notice, this list of conditions, and the         *
*      following disclaimer.                                                                                           *
*                                                                                                                      *
*    * Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the       *
*      following disclaimer in the documentation and/or other materials provided with the distribution.                *
*                                                                                                                      *
*    * Neither the name of the author nor the names of any contributors may be used to endorse or promote products     *
*      derived from this software without specific prior written permission.                                           *
*                                                                                                                      *
* THIS SOFTWARE IS PROVIDED BY THE AUTHORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED   *
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL *
* THE AUTHORS BE HELD LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES        *
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR       *
* BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT *
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE       *
* POSSIBILITY OF SUCH DAMAGE.                                                                                          *
*                                                                                                                      *
***********************************************************************************************************************/
 
#ifndef Socket_h
#define Socket_h
 
#include <string>
#include <stdint.h>
 
//Pull in some OS specific stuff
#ifdef _WIN32
 
#include <ws2tcpip.h>
#define ZSOCKLEN int
#define ZSOCKET SOCKET
 
#else
 
#include <arpa/inet.h>
#include <netdb.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <unistd.h>
#define ZSOCKLEN socklen_t
#define ZSOCKET int
 
#endif
 
class Socket
{
public:
    Socket(int af, int type, int protocol);
 
    //Create a Socket object from an existing socket
    Socket(ZSOCKET sock, int af = PF_INET);
 
    //Destructor
    virtual ~Socket(void);
 
    //Connect to a host (automatic DNS resolution)
    bool Connect(const std::string& host, uint16_t port);
 
    //Bind to a port (any available interface)
    bool Bind(unsigned short port);
 
    //Put us in listening mode
    bool Listen();
 
    //Accept a new connection
    Socket Accept(sockaddr_in* addr, ZSOCKLEN len);
    Socket Accept(sockaddr_in6* addr, ZSOCKLEN len);
    Socket Accept();
 
    //Disconnect us from the socket object
    ZSOCKET Detach();
 
    // Flush the incoming socket
    void FlushRxBuffer(void);
 
    //Send / receive rawdata
    bool SendLooped(const unsigned char* buf, int count);
    bool RecvLooped(unsigned char* buf, int len);
    //size_t RecvFrom(void* buf, size_t len, sockaddr_in& addr, int flags = 0);
    //size_t SendTo(void* buf, size_t len, sockaddr_in& addr, int flags = 0);
 
    //Send/receive a string
    bool RecvPascalString(std::string& str);
    bool SendPascalString(const std::string& str);
 
    //Set TCP_NODELAY on our socket
    bool DisableNagle();
 
    //Set TCP_QUICKACK on our socket
    bool DisableDelayedACK();
 
    //Set SO_REUSEADDR on our socket
    bool SetReuseaddr(bool on = true);
 
    //Set RX/TX timeouts
    bool SetRxTimeout(unsigned int microSeconds);
    bool SetTxTimeout(unsigned int microSeconds);
 
    //Set buffer sizes
    bool SetTxBuffer(int bufsize);
    bool SetRxBuffer(int bufsize);
 
    //Peek if data is ready
    size_t GetRxBytesAvailable() const;
 
    operator const ZSOCKET&() const { return m_socket; }
 
    bool IsValid() const
    {
#ifdef _WIN32
        return (m_socket != INVALID_SOCKET);
#else
        return (m_socket >= 0);
#endif
    }
 
    Socket& operator=(ZSOCKET rhs);
 
    void Close();
 
protected:
    void Open();
 
    int m_af;
 
    int m_type;
 
    int m_protocol;
 
    double m_rxtimeout;
 
    double m_txtimeout;
 
    ZSOCKET m_socket;
};
 
#endif