43 lines
1.3 KiB
C++
43 lines
1.3 KiB
C++
#include "socket/tcp_socket.h"
|
|
|
|
namespace Socket {
|
|
|
|
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) {
|
|
newSock.set(::accept(sock, &__addr.addr, &__addr.length), domain);
|
|
memcpy(&newSock.remoteAddr, &__addr, sizeof(Address));
|
|
if (newSock.sock == INVALID_SOCKET) spdlog::error("accept()");
|
|
}
|
|
|
|
int TCPSocket::connect(Address &serveraddr) {
|
|
int retVal =
|
|
::connect(sock, (struct sockaddr *)&serveraddr.addr, serveraddr.length);
|
|
memcpy(&remoteAddr, &serveraddr, sizeof(Address));
|
|
if (retVal == INVALID_SOCKET) 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 Socket
|