etc/SuperMarioOdysseyOnline
0
0
Fork 0
mirror of https://github.com/CraftyBoss/SuperMarioOdysseyOnline.git synced 2024-11-21 18:55:16 +00:00
Code Issues Projects Releases Wiki Activity

Attempt at full udp client

Browse source
This commit is contained in:
Jack Garrard 2022-08-30 15:18:44 -07:00
parent b18671f113
commit 1204e27adb
2 changed files with 74 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
include/server/SocketClient.hpp
View file

@ -34,6 +34,9 @@ class SocketClient : public SocketBase {
private: private:
int maxBufSize = 100; int maxBufSize = 100;
s32 udp_socket;
/** /**
* @param str a string containing an IPv4 address or a hostname that can be resolved via DNS * @param str a string containing an IPv4 address or a hostname that can be resolved via DNS
* @param out IPv4 address * @param out IPv4 address

77
source/server/SocketClient.cpp
View file

@ -16,6 +16,7 @@ nn::Result SocketClient::init(const char* ip, u16 port) {
in_addr hostAddress = { 0 }; in_addr hostAddress = { 0 };
sockaddr serverAddress = { 0 }; sockaddr serverAddress = { 0 };
sockaddr udpAddress = { 0 };
Logger::log("SocketClient::init: %s:%d sock %s\n", ip, port, getStateChar()); Logger::log("SocketClient::init: %s:%d sock %s\n", ip, port, getStateChar());
@ -65,6 +66,33 @@ nn::Result SocketClient::init(const char* ip, u16 port) {
return result; return result;
} }
if ((this->udp_socket = nn::socket::Socket(2, 2, 17)) <= 0) {
Logger::log("Udp Socket failed to create");
this->socket_errno = nn::socket::GetLastErrno();
this->socket_log_state = SOCKET_LOG_UNAVAILABLE;
return -1;
}
udpAddress.address = 0;
udpAddress.port = 0;
udpAddress.family = 2;
if ((nn::socket::Bind(this->udp_socket, &udpAddress, sizeof(serverAddress))).isFailure()){
Logger::log("Udp Socket failed to bind");
this->socket_errno = nn::socket::GetLastErrno();
this->socket_log_state = SOCKET_LOG_UNAVAILABLE;
return -1;
}
if((result = nn::socket::Connect(this->udp_socket, &serverAddress, sizeof(serverAddress))).isFailure()) {
Logger::log("Udp Socket Connection Failed!\n");
this->socket_errno = nn::socket::GetLastErrno();
this->socket_log_state = SOCKET_LOG_UNAVAILABLE;
return result;
}
this->socket_log_state = SOCKET_LOG_CONNECTED; this->socket_log_state = SOCKET_LOG_CONNECTED;
Logger::log("Socket fd: %d\n", socket_log_socket); Logger::log("Socket fd: %d\n", socket_log_socket);
@ -82,10 +110,22 @@ bool SocketClient::SEND(Packet *packet) {
int valread = 0; int valread = 0;
if (packet->mType != PLAYERINF && packet->mType != HACKCAPINF) if (packet->mType != PLAYERINF && packet->mType != HACKCAPINF) {
Logger::log("Sending packet: %s\n", packetNames[packet->mType]); Logger::log("Sending packet: %s\n", packetNames[packet->mType]);
} else {
if ((valread = nn::socket::Send(this->socket_log_socket, buffer, packet->mPacketSize + sizeof(Packet), 0) > 0)) { if ((valread = nn::socket::Send(this->udp_socket, buffer, packet->mPacketSize + sizeof(Packet), 0) > 0)) {
return true;
} else {
Logger::log("Failed to Fully Send Packet! Result: %d Type: %s Packet Size: %d\n", valread, packetNames[packet->mType], packet->mPacketSize);
this->socket_errno = nn::socket::GetLastErrno();
this->closeSocket();
return false;
}
}
if ((valread = nn::socket::Send(this->socket_log_socket, buffer, packet->mPacketSize + sizeof(Packet), 0) > 0)) {
return true; return true;
} else { } else {
Logger::log("Failed to Fully Send Packet! Result: %d Type: %s Packet Size: %d\n", valread, packetNames[packet->mType], packet->mPacketSize); Logger::log("Failed to Fully Send Packet! Result: %d Type: %s Packet Size: %d\n", valread, packetNames[packet->mType], packet->mPacketSize);
@ -108,9 +148,34 @@ bool SocketClient::RECV() {
char headerBuf[sizeof(Packet)] = {}; char headerBuf[sizeof(Packet)] = {};
int valread = 0; int valread = 0;
const int fd_count = 2;
struct pollfd pfds[fd_count] = {0};
pfds[0].fd = this->socket_log_socket;
pfds[0].events = 1;
pfds[0].revents = 0;
pfds[1].fd = this->udp_socket;
pfds[1].events = 1;
pfds[1].revents = 0;
if (poll(pfds, fd_count, -1) <= 0) {
return true;
}
s32 fd = -1;
for (int i = 0; i < fd_count; i++){
if (pfds[i].revents & 1) {
fd = pfds[i].fd;
}
}
if (fd == -1) {
return true;
}
// read only the size of a header // read only the size of a header
while(valread < headerSize) { while(valread < headerSize) {
int result = nn::socket::Recv(this->socket_log_socket, headerBuf + valread, int result = nn::socket::Recv(fd, headerBuf + valread,
headerSize - valread, this->sock_flags); headerSize - valread, this->sock_flags);
this->socket_errno = nn::socket::GetLastErrno(); this->socket_errno = nn::socket::GetLastErrno();
@ -155,10 +220,10 @@ bool SocketClient::RECV() {
while (valread < fullSize) { while (valread < fullSize) {
int result = nn::socket::Recv(this->socket_log_socket, packetBuf + valread, int result = nn::socket::Recv(fd, packetBuf + valread,
fullSize - valread, this->sock_flags); fullSize - valread, this->sock_flags);
this->socket_errno = nn::socket::GetLastErrno(); this->socket_errno = nn::socket::GetLastErrno();revents
if (result > 0) { if (result > 0) {
valread += result; valread += result;