Kayne Ruse
107 lines
2.4 KiB
C++
107 lines
2.4 KiB
C++
#include "udp_network_utility.hpp"
|
|
|
|
#include <stdexcept>
|
|
|
|
void UDPNetworkUtility::Open(int port, int packSize) {
|
|
if (!(socket = SDLNet_UDP_Open(port))) {
|
|
Close();
|
|
throw(std::runtime_error("Failed to open a UDP socket"));
|
|
}
|
|
|
|
if (!(packOut = SDLNet_AllocPacket(packSize))) {
|
|
Close();
|
|
throw(std::runtime_error("Failed to allocate the out packet"));
|
|
}
|
|
|
|
if (!(packIn = SDLNet_AllocPacket(packSize))) {
|
|
Close();
|
|
throw(std::runtime_error("Failed to allocate the in packet"));
|
|
}
|
|
}
|
|
|
|
void UDPNetworkUtility::Close() {
|
|
SDLNet_UDP_Close(socket);
|
|
SDLNet_FreePacket(packOut);
|
|
SDLNet_FreePacket(packIn);
|
|
socket = nullptr;
|
|
packOut = nullptr;
|
|
packIn = nullptr;
|
|
}
|
|
|
|
int UDPNetworkUtility::Bind(const char* ip, int port, int channel) {
|
|
IPaddress add;
|
|
if (SDLNet_ResolveHost(&add, ip, port) == -1) {
|
|
throw(std::runtime_error("Failed to resolve a host"));
|
|
}
|
|
|
|
return Bind(&add, channel);
|
|
}
|
|
|
|
int UDPNetworkUtility::Bind(IPaddress* add, int channel) {
|
|
int ret = SDLNet_UDP_Bind(socket, channel, add);
|
|
|
|
if (ret == -1) {
|
|
throw(std::runtime_error("Failed to bind to a channel"));
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
void UDPNetworkUtility::Unbind(int channel) {
|
|
SDLNet_UDP_Unbind(socket, channel);
|
|
}
|
|
|
|
int UDPNetworkUtility::Send(const char* ip, int port, void* data, int len) {
|
|
IPaddress add;
|
|
if (SDLNet_ResolveHost(&add, ip, port) == -1) {
|
|
throw(std::runtime_error("Failed to resolve a host"));
|
|
}
|
|
|
|
Send(&add, data, len);
|
|
}
|
|
|
|
int UDPNetworkUtility::Send(IPaddress* add, void* data, int len) {
|
|
if (len > packOut->maxlen) {
|
|
throw(std::runtime_error("Failed to copy the data into the packet"));
|
|
}
|
|
memset(packOut->data, 0, packOut->maxlen);
|
|
memcpy(packOut->data, data, len);
|
|
packOut->len = len;
|
|
packOut->address = *add;
|
|
|
|
int ret = SDLNet_UDP_Send(socket, -1, packOut);
|
|
|
|
if (ret <= 0) {
|
|
throw(std::runtime_error("Failed to send a packet"));
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int UDPNetworkUtility::Send(int channel, void* data, int len) {
|
|
if (len > packOut->maxlen) {
|
|
throw(std::runtime_error("Failed to copy the data into the packet"));
|
|
}
|
|
memset(packOut->data, 0, packOut->maxlen);
|
|
memcpy(packOut->data, data, len);
|
|
packOut->len = len;
|
|
|
|
int ret = SDLNet_UDP_Send(socket, channel, packOut);
|
|
|
|
if (ret <= 0) {
|
|
throw(std::runtime_error("Failed to send a packet"));
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int UDPNetworkUtility::Receive() {
|
|
int ret = SDLNet_UDP_Recv(socket, packIn);
|
|
|
|
if (ret < 0) {
|
|
throw(std::runtime_error("Unknown network error occured"));
|
|
}
|
|
|
|
return ret;
|
|
}
|