GSock/gsock.h
kiritow 5883ac5ca5 Update GSock
Add Log. GSock will output log if compiled with GSOCK_DEBUG.
Add error handling and return value reference.
Fix compile error on getaddrinfo().
Fix logic error on moving sockets. Now one socket will only be opened
and closed once, even if class sock is moved.
2017-08-15 09:54:10 +08:00

90 lines
2.1 KiB
C++

/** General Socket Wrapper
* Created By Kiritow. (https://github.com/kiritow)
* Licensed under MIT
*/
/** Version: 2.2 Update: 20170815*/
#pragma once
#include <cstdint>
#include <string>
class sock
{
public:
sock();
sock(const sock&)=delete;
sock& operator = (const sock&)=delete;
sock(sock&&);
sock& operator = (sock&&);
~sock();
/// Return:
/// 0: Connection Established. No Error.
/// -1: connect() call error. See errno.
/// -2: This socket has been connected before.
/// -3: socket() call error. Failed to create socket. See errno.
int connect(const std::string& IPStr,int Port);
template<typename T>
int send(const T&);
template<typename T>
int recv(T&);
/// Return:
/// return what send() and recv() call returns.
int send(const char* Buffer,int Length);
int recv(char* Buffer,int MaxToRecv);
int getsendtime(int& _out_Second,int& _out_uSecond);
int getrecvtime(int& _out_Second,int& _out_uSecond);
int setsendtime(int Second);
int setrecvtime(int Second);
private:
sock(int);
friend class serversock;
struct _impl;
_impl* _pp;
};
class serversock
{
public:
serversock();
serversock(const serversock&)=delete;
serversock& operator = (const serversock&) =delete;
serversock(serversock&&)=delete;
serversock& operator = (serversock&&) =delete;
~serversock();
/// Return:
/// 0: Bind Succeed. No Error.
/// -1: bind() call error. See errno.
/// -2: This socket has been created before.
/// -3: socket() call error. Failed to create socket. See errno.
int bind(int Port);
int set_reuse();
/// Return:
/// return what listen() call returns
int listen(int MaxCount);
/// Return:
/// 0: Accept Succeed. No Error. _out_s holds the new socket.
/// -1: accept() call error. See errno.
/// -2: _out_s is a connected socket, which should not be passed in.
int accept(sock& _out_s);
private:
struct _impl;
_impl* _pp;
};
/// Net Tools
int DNSResolve(const std::string& HostName,std::string& _out_IPStr);