96 lines
1.9 KiB
C++
96 lines
1.9 KiB
C++
#include "Socket/Socket.hpp"
|
|
#include "Socket/Log.hpp"
|
|
#include "precomp.hpp"
|
|
|
|
namespace Chattr {
|
|
|
|
Socket::Socket(int domain, int type, int protocol) {
|
|
init(domain, type, protocol);
|
|
}
|
|
|
|
Socket::~Socket() {
|
|
destruct();
|
|
}
|
|
|
|
int Socket::init(int domain, int type, int protocol) {
|
|
this->domain = domain;
|
|
|
|
sock_ = ::socket(domain, type, protocol);
|
|
if (sock_ == INVALID_SOCKET)
|
|
log::critical("socket()");
|
|
|
|
valid_ = true;
|
|
|
|
return 0;
|
|
}
|
|
|
|
void Socket::destruct() {
|
|
if (!valid_)
|
|
return;
|
|
#ifdef _WIN32
|
|
::closesocket(sock_);
|
|
#elif __linux__
|
|
::close(sock_);
|
|
#endif
|
|
valid_ = false;
|
|
}
|
|
|
|
Socket::operator SOCKET() {
|
|
if (valid_) {
|
|
valid_ = false;
|
|
return sock_;
|
|
}
|
|
spdlog::critical("No valid socket created.");
|
|
return INVALID_SOCKET;
|
|
}
|
|
|
|
void Socket::set(const SOCKET __sock, int __domain) {
|
|
if (__sock == INVALID_SOCKET)
|
|
log::critical("socket()");
|
|
|
|
destruct();
|
|
|
|
sock_ = __sock;
|
|
valid_ = true;
|
|
};
|
|
|
|
int Socket::bind(Address __addr) {
|
|
bindAddr = __addr;
|
|
int retVal = ::bind(sock_, &__addr.addr, __addr.length);
|
|
if (retVal == INVALID_SOCKET)
|
|
log::critical("bind()");
|
|
return retVal;
|
|
}
|
|
|
|
int Socket::recvfrom(void *__restrict __buf, size_t __n, int __flags, struct Address& __addr) {
|
|
int retVal = ::recvfrom(sock_, (char*)__buf, __n, __flags, &__addr.addr, &__addr.length);
|
|
if (retVal == SOCKET_ERROR)
|
|
log::error("recvfrom()");
|
|
return retVal;
|
|
}
|
|
|
|
int Socket::sendto(const void *__buf, size_t __n, int __flags, struct Address __addr) {
|
|
int retVal = ::sendto(sock_, (char*)__buf, __n, __flags, &__addr.addr, __addr.length);
|
|
if (retVal == SOCKET_ERROR)
|
|
log::error("sendto()");
|
|
return retVal;
|
|
}
|
|
|
|
Socket::Socket(Socket &&other_) {
|
|
other_.valid_ = false;
|
|
sock_ = other_.sock_;
|
|
valid_ = true;
|
|
}
|
|
|
|
Socket& Socket::operator=(Socket && other_) {
|
|
other_.valid_ = false;
|
|
sock_ = other_.sock_;
|
|
valid_ = true;
|
|
|
|
return *this;
|
|
}
|
|
|
|
}
|
|
|
|
|