StarMirror/GSock
1
0
Fork 0
mirror of https://github.com/Kiritow/GSock.git synced 2024-03-22 13:10:51 +08:00
Code Issues Projects Releases Wiki Activity
5883ac5ca5
GSock/gsock.cpp

399 lines
8.0 KiB
C++
Raw Normal View History

Initial Commit
2017-07-27 19:52:58 +08:00
/** General Socket Wrapper
* Created By Kiritow. (https://github.com/kiritow)
* Licensed under MIT
*/
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
/** Version: 2.2 Update: 20170815*/
Initial Commit
2017-07-27 19:52:58 +08:00
#include "gsock.h"
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
#ifdef GSOCK_DEBUG
#pragma message("GSock Debug mode compiled in")
#include <cstdio>
#define myliblog(fmt,args...) printf("GSock: " fmt,##args)
#else
#define myliblog(fmt,args...)
#endif
Initial Commit
2017-07-27 19:52:58 +08:00
#ifdef __WIN32__
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
/// Using Win8.1
#define _WIN32_WINNT 0x0603
Initial Commit
2017-07-27 19:52:58 +08:00
#include <winsock2.h>
#include <ws2tcpip.h>
#else
#include <arpa/inet.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <netdb.h>
#include <errno.h>
#include <sys/epoll.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <fcntl.h>
#define closesocket close
using BYTE = unsigned char;
#endif
#include <string>
#include <stdexcept>
class _init_winsock2_2_class
{
public:
_init_winsock2_2_class()
{
/// Windows Platform need WinSock2.DLL initialization.
#ifdef __WIN32__
WORD wd;
WSAData wdt;
wd=MAKEWORD(2,2);
int ret=WSAStartup(wd,&wdt);
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
myliblog("WSAStartup() Returns: %d\n",ret);
Initial Commit
2017-07-27 19:52:58 +08:00
if(ret<0)
{
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
myliblog("WSAGetLastError: %d\n",WSAGetLastError());
Initial Commit
2017-07-27 19:52:58 +08:00
throw std::runtime_error("Unable to load winsock2.dll. ");
}
#endif
}
~_init_winsock2_2_class()
{
/// Windows Platform need WinSock2.DLL clean up.
#ifdef __WIN32__
WSACleanup();
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
myliblog("WSACleanup() called.");
Initial Commit
2017-07-27 19:52:58 +08:00
#endif
}
} _init_winsock2_2_obj;
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
struct sock::_impl
Initial Commit
2017-07-27 19:52:58 +08:00
{
int sfd;
sockaddr_in saddr;
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
bool created;
Initial Commit
2017-07-27 19:52:58 +08:00
};
sock::sock() : _pp(new _impl)
{
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
myliblog("sock::sock() %p\n",this);
_pp->created=false;
Initial Commit
2017-07-27 19:52:58 +08:00
}
//private
sock::sock(int SocketValue) : _pp(new _impl)
{
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
myliblog("sock::sock(int) %p\n",this);
_pp->created=true;
Initial Commit
2017-07-27 19:52:58 +08:00
_pp->sfd=SocketValue;
}
sock::sock(sock&& tmp)
{
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
myliblog("sock::sock(sock&&) %p <- %p \n",this,&tmp);
_pp=tmp._pp;
tmp._pp=nullptr;
Initial Commit
2017-07-27 19:52:58 +08:00
}
sock& sock::operator = (sock&& tmp)
{
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
myliblog("sock::operator = (sock&&) %p <= %p\n",this,&tmp);
Initial Commit
2017-07-27 19:52:58 +08:00
if(_pp)
{
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
if(_pp->created)
{
myliblog("Socket closed: [%d] in %p\n",_pp->sfd,this);
closesocket(_pp->sfd);
}
delete _pp;
Initial Commit
2017-07-27 19:52:58 +08:00
}
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
_pp=tmp._pp;
tmp._pp=nullptr;
Initial Commit
2017-07-27 19:52:58 +08:00
return *this;
}
sock::~sock()
{
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
myliblog("sock::~sock() %p\n",this);
Initial Commit
2017-07-27 19:52:58 +08:00
if(_pp)
{
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
if(_pp->created)
{
myliblog("Socket closed: [%d] in %p\n",_pp->sfd,this);
closesocket(_pp->sfd);
}
delete _pp;
Initial Commit
2017-07-27 19:52:58 +08:00
}
}
int sock::connect(const std::string& IPStr,int Port)
{
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
myliblog("sock::connect() %p\n",this);
if(_pp->created)
{
return -2;
}
_pp->sfd=socket(AF_INET,SOCK_STREAM,0);
if(_pp->sfd<0)
{
myliblog("socket() returns %d. WSAGetLastError: %d\n",_pp->sfd,WSAGetLastError());
return -3;
}
myliblog("Socket created: [%d] in %p\n",_pp->sfd,this);
_pp->created=true;
Initial Commit
2017-07-27 19:52:58 +08:00
// refs
int& sfd=_pp->sfd;
sockaddr_in& saddr=_pp->saddr;
memset(&saddr,0,sizeof(saddr));
saddr.sin_addr.s_addr=inet_addr(IPStr.c_str());
saddr.sin_port=htons(Port);
saddr.sin_family=AF_INET;
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
Initial Commit
2017-07-27 19:52:58 +08:00
return ::connect(sfd,(sockaddr*)&saddr,sizeof(saddr));
}
int sock::send(const char* Buffer,int Length)
{
return ::send(_pp->sfd,Buffer,Length,0);
}
int sock::recv(char* Buffer,int MaxToRecv)
{
return ::recv(_pp->sfd,Buffer,MaxToRecv,0);
}
int sock::getsendtime(int& _out_Second, int& _out_uSecond)
{
// refs
int& sfd=_pp->sfd;
struct timeval outtime;
int _not_used_t;
int ret=getsockopt(sfd,SOL_SOCKET,SO_SNDTIMEO,(char*)&outtime,&_not_used_t);
if(ret<0) return ret;
/// We don't know why, but on Windows, 1 Second means 1000.
#ifdef __WIN32__
_out_Second=outtime.tv_sec/1000;
_out_uSecond=outtime.tv_usec;
#else
_out_Second=outtime.tv_sec;
_out_uSecond=outtime.tv_usec;
#endif
return ret;
}
int sock::getrecvtime(int& _out_Second, int& _out_uSecond)
{
// refs
int& sfd=_pp->sfd;
struct timeval outtime;
int _not_used_t;
int ret=getsockopt(sfd,SOL_SOCKET,SO_RCVTIMEO,(char*)&outtime,&_not_used_t);
if(ret<0) return ret;
/// We don't know why, but on Windows, 1 Second means 1000.
#ifdef __WIN32__
_out_Second=outtime.tv_sec/1000;
_out_uSecond=outtime.tv_usec;
#else
_out_Second=outtime.tv_sec;
_out_uSecond=outtime.tv_usec;
#endif
return ret;
}
int sock::setsendtime(int Second)
{
// refs
int& sfd=_pp->sfd;
struct timeval outtime;
/// We don't know why, but on Windows, 1 Second means 1000.
#ifdef __WIN32__
outtime.tv_sec=Second*1000;
outtime.tv_usec=0;
#else
outtime.tv_sec=Second;
outtime.tv_usec=0;
#endif
return setsockopt(sfd,SOL_SOCKET,SO_SNDTIMEO,(const char*)&outtime,sizeof(outtime));
}
int sock::setrecvtime(int Second)
{
// refs
int& sfd=_pp->sfd;
struct timeval outtime;
/// We don't know why, but on Windows, 1 Second means 1000.
#ifdef __WIN32__
outtime.tv_sec=Second*1000;
outtime.tv_usec=0;
#else
outtime.tv_sec=Second;
outtime.tv_usec=0;
#endif
return setsockopt(sfd,SOL_SOCKET,SO_RCVTIMEO,(const char*)&outtime,sizeof(outtime));
}
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
struct serversock::_impl
Initial Commit
2017-07-27 19:52:58 +08:00
{
int sfd;
sockaddr_in saddr;
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
bool created;
Initial Commit
2017-07-27 19:52:58 +08:00
};
serversock::serversock() : _pp(new _impl)
{
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
myliblog("serversock::serversock() %p\n",this);
_pp->created=false;
Initial Commit
2017-07-27 19:52:58 +08:00
}
serversock::~serversock()
{
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
myliblog("serversock::~serversock() %p\n",this);
if(_pp)
{
if(_pp->created)
{
myliblog("Server-Socket closed: [%d] in %p\n",_pp->sfd,this);
closesocket(_pp->sfd);
}
delete _pp;
}
Initial Commit
2017-07-27 19:52:58 +08:00
}
int serversock::bind(int Port)
{
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
myliblog("serversock::bind() %p\n",this);
if(_pp->created)
{
return -2;
}
_pp->sfd=socket(AF_INET,SOCK_STREAM,0);
if(_pp->sfd<0)
{
myliblog("socket() returns %d. WSAGetLastError: %d\n",_pp->sfd,WSAGetLastError());
return -3;
}
myliblog("Socket created: [%d] in %p\n",_pp->sfd,this);
_pp->created=true;
Initial Commit
2017-07-27 19:52:58 +08:00
// refs
int& sfd=_pp->sfd;
sockaddr_in& saddr=_pp->saddr;
memset(&saddr,0,sizeof(saddr));
saddr.sin_addr.s_addr=INADDR_ANY;
saddr.sin_port=htons(Port);
saddr.sin_family=AF_INET;
return ::bind(sfd,(sockaddr*)&saddr,sizeof(saddr));
}
int serversock::set_reuse()
{
// refs
int& sfd=_pp->sfd;
int opt=1;
return setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,(const char*)&opt,sizeof(opt));
}
int serversock::listen(int MaxCount)
{
// refs
int& sfd=_pp->sfd;
return ::listen(sfd,MaxCount);
}
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
int serversock::accept(sock& _out_s)
Initial Commit
2017-07-27 19:52:58 +08:00
{
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
if(_out_s._pp->created)
{
/// _out_s has been connected.
return -2;
}
Initial Commit
2017-07-27 19:52:58 +08:00
sock s;
int tmp=sizeof(s._pp->saddr);
int ret=::accept(_pp->sfd,(sockaddr*)&(s._pp->saddr),&tmp);
if(ret<0)
{
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
/// accept() call failed.
myliblog("accept() returns %d. WSAGetLastError: %d\n",ret,WSAGetLastError());
return -1;
Initial Commit
2017-07-27 19:52:58 +08:00
}
else
{
s._pp->sfd=ret;
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
s._pp->created=true;
Initial Commit
2017-07-27 19:52:58 +08:00
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
myliblog("Socket opened: [%d] in %p by serversock %p\n",s._pp->sfd,&s,this);
Initial Commit
2017-07-27 19:52:58 +08:00
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
/// Move resource.
_out_s=std::move(s);
return 0;
}
}
Initial Commit
2017-07-27 19:52:58 +08:00
int DNSResolve(const std::string& HostName, std::string& _out_IPStr)
{
/// Use getaddrinfo instead
struct addrinfo hints;
memset(&hints,0,sizeof(hints));
hints.ai_family=AF_UNSPEC;
hints.ai_socktype=SOCK_STREAM;
hints.ai_protocol=IPPROTO_TCP;
struct addrinfo* result=nullptr;
int ret=getaddrinfo(HostName.c_str(),NULL,&hints,&result);
if(ret!=0)
{
return -1;/// API Call Failed.
}
for(struct addrinfo* ptr=result; ptr!=nullptr; ptr=ptr->ai_next)
{
switch(ptr->ai_family)
{
case AF_INET:
sockaddr_in* addr=(struct sockaddr_in*) (ptr->ai_addr) ;
_out_IPStr=inet_ntoa(addr->sin_addr);
return 0;
break;
}
}
/// Unknown error.
return -2;
}
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
/// Undefine marcos
#undef myliblog
Reference in New Issue Copy Permalink