128 lines
4.3 KiB
C++
128 lines
4.3 KiB
C++
// Config Needs To Load First
|
|
#include <libreborn/libreborn.h>
|
|
|
|
#include <string>
|
|
#include <cstring>
|
|
#include <cstdio>
|
|
#include <vector>
|
|
|
|
#include <symbols/minecraft.h>
|
|
#ifndef MCPI_HEADLESS_MODE
|
|
#include <media-layer/core.h>
|
|
#endif
|
|
|
|
#include <mods/init/init.h>
|
|
#include <mods/feature/feature.h>
|
|
#ifndef MCPI_HEADLESS_MODE
|
|
#include <mods/input/input.h>
|
|
#endif
|
|
#include "chat-internal.h"
|
|
#include <mods/chat/chat.h>
|
|
|
|
// Message Limitations
|
|
#define MAX_CHAT_MESSAGE_LENGTH 512
|
|
|
|
// Send API Command
|
|
std::string chat_send_api_command(Minecraft *minecraft, std::string str) {
|
|
struct ConnectedClient client;
|
|
client.sock = -1;
|
|
client.str = "";
|
|
client.time = 0;
|
|
CommandServer *command_server = minecraft->command_server;
|
|
if (command_server != NULL) {
|
|
return CommandServer_parse(command_server, &client, &str);
|
|
} else {
|
|
return "";
|
|
}
|
|
}
|
|
|
|
#ifndef MCPI_HEADLESS_MODE
|
|
// Send API Chat Command
|
|
static void send_api_chat_command(Minecraft *minecraft, char *str) {
|
|
char *command = NULL;
|
|
safe_asprintf(&command, "chat.post(%s)\n", str);
|
|
chat_send_api_command(minecraft, command);
|
|
free(command);
|
|
}
|
|
#endif
|
|
|
|
// Send Message To Players
|
|
void chat_send_message(ServerSideNetworkHandler *server_side_network_handler, char *username, char *message) {
|
|
char *full_message = NULL;
|
|
safe_asprintf(&full_message, "<%s> %s", username, message);
|
|
sanitize_string(&full_message, MAX_CHAT_MESSAGE_LENGTH, 0);
|
|
std::string cpp_string = full_message;
|
|
free(full_message);
|
|
ServerSideNetworkHandler_displayGameMessage(server_side_network_handler, &cpp_string);
|
|
}
|
|
// Handle Chat packet Send
|
|
void chat_handle_packet_send(Minecraft *minecraft, ChatPacket *packet) {
|
|
RakNetInstance *rak_net_instance = minecraft->rak_net_instance;
|
|
if (rak_net_instance->vtable->isServer(rak_net_instance)) {
|
|
// Hosting Multiplayer
|
|
char *message = packet->message;
|
|
ServerSideNetworkHandler *server_side_network_handler = (ServerSideNetworkHandler *) minecraft->network_handler;
|
|
chat_send_message(server_side_network_handler, Strings_default_username, message);
|
|
} else {
|
|
// Client
|
|
rak_net_instance->vtable->send(rak_net_instance, (Packet *) packet);
|
|
}
|
|
}
|
|
|
|
// Manually Send (And Loopback) ChatPacket
|
|
static void CommandServer_parse_CommandServer_dispatchPacket_injection(CommandServer *command_server, Packet *packet) {
|
|
Minecraft *minecraft = command_server->minecraft;
|
|
if (minecraft != NULL) {
|
|
chat_handle_packet_send(minecraft, (ChatPacket *) packet);
|
|
}
|
|
}
|
|
|
|
// Handle ChatPacket Server-Side
|
|
static void ServerSideNetworkHandler_handle_ChatPacket_injection(ServerSideNetworkHandler *server_side_network_handler, RakNet_RakNetGUID *rak_net_guid, ChatPacket *chat_packet) {
|
|
Player *player = ServerSideNetworkHandler_getPlayer(server_side_network_handler, rak_net_guid);
|
|
if (player != NULL) {
|
|
const char *username = player->username.c_str();
|
|
char *message = chat_packet->message;
|
|
chat_send_message(server_side_network_handler, (char *) username, message);
|
|
}
|
|
}
|
|
|
|
#ifndef MCPI_HEADLESS_MODE
|
|
// Message Queue
|
|
static std::vector<std::string> queue;
|
|
// Add To Queue
|
|
void _chat_queue_message(const char *message) {
|
|
// Add
|
|
std::string str = message;
|
|
queue.push_back(str);
|
|
}
|
|
// Empty Queue
|
|
unsigned int old_chat_counter = 0;
|
|
static void send_queued_messages(Minecraft *minecraft) {
|
|
// Loop
|
|
for (unsigned int i = 0; i < queue.size(); i++) {
|
|
send_api_chat_command(minecraft, (char *) queue[i].c_str());
|
|
}
|
|
queue.clear();
|
|
}
|
|
#endif
|
|
|
|
// Init
|
|
void init_chat() {
|
|
if (feature_has("Implement Chat", server_enabled)) {
|
|
// Disable Original ChatPacket Loopback
|
|
unsigned char disable_chat_packet_loopback_patch[4] = {0x00, 0xf0, 0x20, 0xe3}; // "nop"
|
|
patch((void *) 0x6b490, disable_chat_packet_loopback_patch);
|
|
// Manually Send (And Loopback) ChatPacket
|
|
overwrite_call((void *) 0x6b518, (void *) CommandServer_parse_CommandServer_dispatchPacket_injection);
|
|
// Re-Broadcast ChatPacket
|
|
patch_address(ServerSideNetworkHandler_handle_ChatPacket_vtable_addr, (void *) ServerSideNetworkHandler_handle_ChatPacket_injection);
|
|
#ifndef MCPI_HEADLESS_MODE
|
|
// Send Messages On Input Tick
|
|
input_run_on_tick(send_queued_messages);
|
|
// Init UI
|
|
_init_chat_ui();
|
|
#endif
|
|
}
|
|
}
|