mirror of
https://github.com/open-goal/jak-project
synced 2026-06-01 01:40:07 -04:00
Tyler Wilding
136 lines
3.6 KiB
C++
136 lines
3.6 KiB
C++
/*!
|
|
* @file Deci2Server.cpp
|
|
* Basic implementation of a DECI2 server.
|
|
* Works with deci2.cpp (sceDeci2) to implement the networking on target
|
|
*/
|
|
|
|
#include "Deci2Server.h"
|
|
|
|
#include "common/cross_sockets/XSocket.h"
|
|
|
|
#include "common/versions.h"
|
|
#include <common/listener_common.h>
|
|
#include <common/util/Assert.h>
|
|
|
|
#include "third-party/fmt/core.h"
|
|
|
|
void Deci2Server::write_on_accept() {
|
|
u32 versions[2] = {versions::GOAL_VERSION_MAJOR, versions::GOAL_VERSION_MINOR};
|
|
lock();
|
|
write_to_socket(accepted_socket, (char*)&versions, 8);
|
|
unlock();
|
|
}
|
|
|
|
/*!
|
|
* Wait for protocols to become ready.
|
|
* This avoids the case where we receive messages before protocol handlers are set up.
|
|
*/
|
|
void Deci2Server::wait_for_protos_ready() {
|
|
if (protocols_ready)
|
|
return;
|
|
std::unique_lock<std::mutex> lk(server_mutex);
|
|
cv.wait(lk, [&] { return protocols_ready; });
|
|
}
|
|
|
|
/*!
|
|
* Inform server that protocol handlers are ready.
|
|
* Will unblock wait_for_protos_ready and incoming messages will be dispatched to these
|
|
* protocols. You can change the protocol handlers, but you should lock the mutex before
|
|
* doing so.
|
|
*/
|
|
void Deci2Server::send_proto_ready(Deci2Driver* drivers, int* driver_count) {
|
|
lock();
|
|
d2_drivers = drivers;
|
|
d2_driver_count = driver_count;
|
|
protocols_ready = true;
|
|
unlock();
|
|
cv.notify_all();
|
|
}
|
|
|
|
void Deci2Server::read_data() {
|
|
int desired_size = (int)sizeof(Deci2Header);
|
|
int got = 0;
|
|
|
|
while (got < desired_size) {
|
|
ASSERT(got + desired_size < buffer_size);
|
|
auto x = read_from_socket(accepted_socket, buffer + got, desired_size - got);
|
|
if (want_exit_callback()) {
|
|
return;
|
|
}
|
|
got += x > 0 ? x : 0;
|
|
}
|
|
|
|
auto* hdr = (Deci2Header*)(buffer);
|
|
fprintf(stderr, "[DECI2] Got message: %d %d 0x%x %c -> %c\n", hdr->len, hdr->rsvd, hdr->proto,
|
|
hdr->src, hdr->dst);
|
|
|
|
hdr->rsvd = got;
|
|
|
|
// see what protocol we got:
|
|
lock();
|
|
|
|
int handler = -1;
|
|
for (int i = 0; i < *d2_driver_count; i++) {
|
|
auto& prot = d2_drivers[i];
|
|
if (prot.active && prot.protocol) {
|
|
if (handler != -1) {
|
|
printf("[DECI2] Warning: more than on protocol handler for this message!\n");
|
|
}
|
|
handler = i;
|
|
}
|
|
}
|
|
|
|
if (handler == -1) {
|
|
printf("[DECI2] Warning: no handler for this message, ignoring...\n");
|
|
unlock();
|
|
return;
|
|
}
|
|
|
|
auto& driver = d2_drivers[handler];
|
|
|
|
u32 sent_to_program = 0;
|
|
while (!want_exit_callback() && (hdr->rsvd < hdr->len || sent_to_program < hdr->rsvd)) {
|
|
// send what we have to the program
|
|
if (sent_to_program < hdr->rsvd) {
|
|
// driver.next_recv_size = 0;
|
|
// driver.next_recv = nullptr;
|
|
driver.recv_buffer = buffer + sent_to_program;
|
|
driver.available_to_receive = hdr->rsvd - sent_to_program;
|
|
(driver.handler)(DECI2_READ, driver.available_to_receive, driver.opt);
|
|
// memcpy(driver.next_recv, buffer + sent_to_program, driver.next_recv_size);
|
|
sent_to_program += driver.recv_size;
|
|
}
|
|
|
|
// receive from network
|
|
if (hdr->rsvd < hdr->len) {
|
|
auto x = read_from_socket(accepted_socket, buffer + hdr->rsvd, hdr->len - hdr->rsvd);
|
|
if (want_exit_callback()) {
|
|
return;
|
|
}
|
|
got += x > 0 ? x : 0;
|
|
hdr->rsvd = got;
|
|
}
|
|
}
|
|
|
|
(driver.handler)(DECI2_READDONE, 0, driver.opt);
|
|
unlock();
|
|
}
|
|
|
|
void Deci2Server::send_data(void* buf, u16 len) {
|
|
lock();
|
|
if (!client_connected) {
|
|
printf("[DECI2] send while not connected, not sending!\n");
|
|
} else {
|
|
uint16_t prog = 0;
|
|
while (prog < len) {
|
|
int wrote = write_to_socket(accepted_socket, (char*)(buf) + prog, len - prog);
|
|
prog += wrote;
|
|
if (!client_connected || want_exit_callback()) {
|
|
unlock();
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
unlock();
|
|
}
|