60 lines
1.8 KiB
C++
60 lines
1.8 KiB
C++
#include "socket/tcp_socket.h"
|
|
|
|
namespace Network {
|
|
|
|
int TCPSocket::init(int domain) { return init(domain, SOCK_STREAM, 0); }
|
|
|
|
int TCPSocket::listen(int __n) {
|
|
int retVal = ::listen(sock, __n);
|
|
if (retVal == INVALID_SOCKET) spdlog::error("listen()");
|
|
return retVal;
|
|
}
|
|
|
|
void TCPSocket::accept(TCPSocket &newSock, Address &__addr) {
|
|
SOCKET newsock = INVALID_SOCKET;
|
|
newsock = ::accept(sock, &__addr.addr, &__addr.length);
|
|
if (newsock == INVALID_SOCKET) {
|
|
spdlog::error("accept() errno:{} {}", errno, strerror(errno));
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
newSock.set(newsock, domain);
|
|
memcpy(&newSock.remoteAddr, &__addr, sizeof(Address));
|
|
}
|
|
|
|
int TCPSocket::connect(Address &serveraddr) {
|
|
std::string addr_string = serveraddr;
|
|
int retVal = -1;
|
|
if (serveraddr.family == AF_INET)
|
|
retVal = ::connect(sock, (const sockaddr *)&serveraddr.addr_in,
|
|
serveraddr.length);
|
|
else
|
|
retVal = ::connect(sock, (const sockaddr *)&serveraddr.addr_in6,
|
|
serveraddr.length);
|
|
memcpy(&remoteAddr, &serveraddr, sizeof(Address));
|
|
if (retVal == INVALID_SOCKET) {
|
|
#ifdef _WIN32
|
|
int err = WSAGetLastError();
|
|
spdlog::error("connect() failed: WSA error {} (0x{:X})", err, err);
|
|
#endif
|
|
spdlog::error("connect()");
|
|
}
|
|
return retVal;
|
|
}
|
|
|
|
int TCPSocket::recv(void *__restrict __buf, size_t __n, int __flags) {
|
|
int retVal = ::recv(sock, (char *)__buf, __n, __flags);
|
|
if (retVal == SOCKET_ERROR) {
|
|
if (errno == EAGAIN || errno == EWOULDBLOCK) return retVal;
|
|
spdlog::error("recv()");
|
|
}
|
|
return retVal;
|
|
}
|
|
|
|
int TCPSocket::send(const void *__buf, size_t __n, int __flags) {
|
|
int retVal = ::send(sock, (char *)__buf, __n, __flags);
|
|
if (retVal == SOCKET_ERROR) spdlog::error("send()");
|
|
return retVal;
|
|
}
|
|
|
|
} // namespace Network
|