Added custom broadcasts file

merge-requests/6/head
Arkhist 2019-04-21 23:25:37 +02:00
parent 0aff1032dd
commit b9e2b5fa40
5 changed files with 95 additions and 15 deletions

17
.gitignore vendored
View File

@ -1,5 +1,12 @@
.vscode/* .vscode/*
*.bin *.bin
*.o *.o
net.pb.* net.pb.*
*steam_api* *steam_api*
release/*
*.obj
base.exp
base.lib
rtlgenrandom*
steamclient.exp
steamclient.lib

View File

@ -25,8 +25,8 @@ release: CXX_FLAGS += -DNDEBUG -DEMU_RELEASE_BUILD -Ofast
release: LD_FLAGS += -lpthread release: LD_FLAGS += -lpthread
release32: CXX_FLAGS += -m32 release32: CXX_FLAGS += -m32
release32: LD_FLAGS += -m32 release32: LD_FLAGS += -m32
debug: CXX_FLAGS += -g3 -fsanitize=address debug: CXX_FLAGS += -g3
debug: LD_FLAGS += -lasan debug: LD_FLAGS += -lpthread
release: library release: library
release32: release release32: release
debug: library debug: library

View File

@ -280,7 +280,7 @@ static int receive_packet(sock_t sock, IP_PORT *ip_port, char *data, unsigned lo
return -1; return -1;
} }
static bool send_broadcasts(sock_t sock, uint16 port, char *data, unsigned long length) static bool send_broadcasts(sock_t sock, uint16 port, char *data, unsigned long length, uint32_t *custom_broadcasts)
{ {
static std::chrono::high_resolution_clock::time_point last_get_broadcast_info; static std::chrono::high_resolution_clock::time_point last_get_broadcast_info;
if (number_broadcasts < 0 || check_timedout(last_get_broadcast_info, 60.0)) { if (number_broadcasts < 0 || check_timedout(last_get_broadcast_info, 60.0)) {
@ -302,6 +302,23 @@ static bool send_broadcasts(sock_t sock, uint16 port, char *data, unsigned long
IP_PORT ip_port = broadcasts[i]; IP_PORT ip_port = broadcasts[i];
} }
/**
* Custom targeted clients server broadcaster
*
* Sends to custom IPs the broadcast packet
* This is useful in cases of undetected network interfaces
*/
PRINT_DEBUG("start custom broadcasts\n");
IP_PORT custom_targeted_broadcast;
custom_targeted_broadcast.port = port;
for(int i = 0; i < MAX_CUSTOM_BROADCASTS; i++) {
if(custom_broadcasts[i] == 0)
break;
custom_targeted_broadcast.ip = custom_broadcasts[i];
send_packet_to(sock, custom_targeted_broadcast, data, length);
}
PRINT_DEBUG("end custom broadcasts\n");
return true; return true;
} }
@ -666,7 +683,7 @@ bool Networking::handle_low_level_udp(Common_Message *msg, IP_PORT ip_port)
#define NUM_TCP_WAITING 128 #define NUM_TCP_WAITING 128
Networking::Networking(CSteamID id, uint32 appid, uint16 port) Networking::Networking(CSteamID id, uint32 appid, uint16 port, uint32_t *custom_broadcasts)
{ {
run_at_startup(); run_at_startup();
tcp_port = udp_port = port; tcp_port = udp_port = port;
@ -674,6 +691,8 @@ Networking::Networking(CSteamID id, uint32 appid, uint16 port)
alive = true; alive = true;
last_run = std::chrono::high_resolution_clock::now(); last_run = std::chrono::high_resolution_clock::now();
this->appid = appid; this->appid = appid;
for(int i = 0; i < MAX_CUSTOM_BROADCASTS; i++)
this->custom_broadcasts[i] = custom_broadcasts[i];
sock_t sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); sock_t sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
PRINT_DEBUG("UDP socket: %u\n", sock); PRINT_DEBUG("UDP socket: %u\n", sock);
@ -783,9 +802,9 @@ void Networking::send_announce_broadcasts()
size_t size = msg.ByteSizeLong(); size_t size = msg.ByteSizeLong();
char *buffer = new char[size]; char *buffer = new char[size];
msg.SerializeToArray(buffer, size); msg.SerializeToArray(buffer, size);
send_broadcasts(udp_socket, htons(DEFAULT_PORT), buffer, size); send_broadcasts(udp_socket, htons(DEFAULT_PORT), buffer, size, this->custom_broadcasts);
if (udp_port != DEFAULT_PORT) { if (udp_port != DEFAULT_PORT) {
send_broadcasts(udp_socket, htons(udp_port), buffer, size); send_broadcasts(udp_socket, htons(udp_port), buffer, size, this->custom_broadcasts);
} }
delete[] buffer; delete[] buffer;

View File

@ -20,6 +20,8 @@
#ifndef NETWORK_INCLUDE #ifndef NETWORK_INCLUDE
#define NETWORK_INCLUDE #define NETWORK_INCLUDE
#define MAX_CUSTOM_BROADCASTS 128
#include "net.pb.h" #include "net.pb.h"
#include <chrono> #include <chrono>
@ -106,6 +108,7 @@ class Networking {
std::vector<CSteamID> ids; std::vector<CSteamID> ids;
uint32 appid; uint32 appid;
std::chrono::high_resolution_clock::time_point last_broadcast; std::chrono::high_resolution_clock::time_point last_broadcast;
uint32_t custom_broadcasts[MAX_CUSTOM_BROADCASTS];
std::vector<struct TCP_Socket> accepted; std::vector<struct TCP_Socket> accepted;
std::recursive_mutex mutex; std::recursive_mutex mutex;
@ -120,7 +123,7 @@ class Networking {
Common_Message create_announce(bool request); Common_Message create_announce(bool request);
public: public:
Networking(CSteamID id, uint32 appid, uint16 port); Networking(CSteamID id, uint32 appid, uint16 port, uint32_t *custom_broadcasts);
void addListenId(CSteamID id); void addListenId(CSteamID id);
void setAppID(uint32 appid); void setAppID(uint32 appid);
void Run(); void Run();

View File

@ -19,6 +19,7 @@
#include <fstream> #include <fstream>
static void network_thread(Networking *network) static void network_thread(Networking *network)
{ {
PRINT_DEBUG("network thread starting\n"); PRINT_DEBUG("network thread starting\n");
@ -112,6 +113,7 @@ Steam_Client::Steam_Client()
local_storage = new Local_Storage(save_path); local_storage = new Local_Storage(save_path);
local_storage->setAppId(appid); local_storage->setAppId(appid);
// Listen port
char array_port[10] = {}; char array_port[10] = {};
array_port[0] = '0'; array_port[0] = '0';
local_storage->get_data_settings("listen_port.txt", array_port, sizeof(array_port) - 1); local_storage->get_data_settings("listen_port.txt", array_port, sizeof(array_port) - 1);
@ -122,19 +124,68 @@ Steam_Client::Steam_Client()
local_storage->store_data_settings("listen_port.txt", array_port, strlen(array_port)); local_storage->store_data_settings("listen_port.txt", array_port, strlen(array_port));
} }
char name[32] = {}; // Custom broadcasts
uint32_t custom_broadcasts[MAX_CUSTOM_BROADCASTS] = {0};
int readIP = 0;
std::string broadcasts_filepath = Local_Storage::get_game_settings_path() + "custom_broadcasts.txt";
std::ifstream broadcasts_file(broadcasts_filepath);
PRINT_DEBUG("Broadcasts file path: %s\n", broadcasts_filepath.c_str());
if (broadcasts_file.is_open()) {
std::string line;
while (std::getline(broadcasts_file, line) && readIP < MAX_CUSTOM_BROADCASTS) {
int offset = 0;
size_t pos = 0;
std::string tok;
uint32_t current_ip = 0;
while((pos = line.find(".")) != std::string::npos && offset < 32)
{
tok = line.substr(0, pos);
try
{
current_ip += (std::stoi(tok) << offset);
}
catch(std::invalid_argument ex)
{
offset = -1;
break;
}
line.erase(0, pos+1);
offset += 8;
}
if(pos == std::string::npos && offset != -1)
{
try
{
current_ip += (std::stoi(line) << offset);
custom_broadcasts[readIP++] = current_ip;
}
catch(std::invalid_argument ex)
{
}
}
}
}
// Acount name
char name[32] = {};
if (local_storage->get_data_settings("account_name.txt", name, sizeof(name) - 1) <= 0) { if (local_storage->get_data_settings("account_name.txt", name, sizeof(name) - 1) <= 0) {
strcpy(name, DEFAULT_NAME); strcpy(name, DEFAULT_NAME);
local_storage->store_data_settings("account_name.txt", name, strlen(name)); local_storage->store_data_settings("account_name.txt", name, strlen(name));
} }
// Language
char language[32] = {}; char language[32] = {};
if (local_storage->get_data_settings("language.txt", language, sizeof(language) - 1) <= 0) { if (local_storage->get_data_settings("language.txt", language, sizeof(language) - 1) <= 0) {
strcpy(language, DEFAULT_LANGUAGE); strcpy(language, DEFAULT_LANGUAGE);
local_storage->store_data_settings("language.txt", language, strlen(language)); local_storage->store_data_settings("language.txt", language, strlen(language));
} }
// Steam ID
char array_steam_id[32] = {}; char array_steam_id[32] = {};
CSteamID user_id; CSteamID user_id;
uint64 steam_id = 0; uint64 steam_id = 0;
@ -222,7 +273,7 @@ Steam_Client::Steam_Client()
} }
} }
network = new Networking(settings_server->get_local_steam_id(), appid, port); network = new Networking(settings_server->get_local_steam_id(), appid, port, custom_broadcasts);
callback_results_client = new SteamCallResults(); callback_results_client = new SteamCallResults();
callback_results_server = new SteamCallResults(); callback_results_server = new SteamCallResults();
@ -230,7 +281,7 @@ Steam_Client::Steam_Client()
callbacks_server = new SteamCallBacks(callback_results_server); callbacks_server = new SteamCallBacks(callback_results_server);
run_every_runcb = new RunEveryRunCB(); run_every_runcb = new RunEveryRunCB();
PRINT_DEBUG("steam client init: id: %llu server id: %llu appid: %u port: %u \n", user_id.ConvertToUint64(), settings_server->get_local_steam_id().ConvertToUint64(), appid, port); PRINT_DEBUG("steam client init: id: %llu server id: %llu appid: %u port: %u custom broadcasts: %u\n", user_id.ConvertToUint64(), settings_server->get_local_steam_id().ConvertToUint64(), appid, port, sizeof(custom_broadcasts));
steam_user = new Steam_User(settings_client, local_storage, network, callback_results_client, callbacks_client); steam_user = new Steam_User(settings_client, local_storage, network, callback_results_client, callbacks_client);
steam_friends = new Steam_Friends(settings_client, network, callback_results_client, callbacks_client, run_every_runcb); steam_friends = new Steam_Friends(settings_client, network, callback_results_client, callbacks_client, run_every_runcb);