GeneralsZeroHour/GeneralsMD/Code/Tools/matchbot/main.cpp

/*
**	Command & Conquer Generals Zero Hour(tm)
**	Copyright 2025 Electronic Arts Inc.
**
**	This program is free software: you can redistribute it and/or modify
**	it under the terms of the GNU General Public License as published by
**	the Free Software Foundation, either version 3 of the License, or
**	(at your option) any later version.
**
**	This program is distributed in the hope that it will be useful,
**	but WITHOUT ANY WARRANTY; without even the implied warranty of
**	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
**	GNU General Public License for more details.
**
**	You should have received a copy of the GNU General Public License
**	along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifdef _WIN32
#include <process.h>
#endif
#include <cstdlib>
#include <csignal>
#include <iostream>

#ifdef _WIN32
#include <direct.h>
#else
#include <sys/types.h>
#include <sys/stat.h>
#endif

//#define THREADSAFE_HEADER

#include <wstring.h>
#include <wdebug.h>
#include <filed.h>
#include <stdoutd.h>
#include <wstypes.h>
#include <xtime.h>
#include "global.h"
#include "generals.h"
#include "timezone.h"
#include <threadfac.h>

#include <tcp.h>
#include "mydebug.h"

#ifdef _UNIX
using namespace std;
#else
#define sleep(x) Sleep(1000 * (x))
#endif

static char *Program_Usage = "A config filename can be given on the command line (default=matchbot.cfg)\n";
void logMonitor(void *);
void paranoidLogMonitor(void *);

OutputDevice * output_device = NULL;
OutputDevice * paranoid_output_device = NULL;


void Signal_Quit(int)
{
	INFMSG("Exiting due to signal.");
	exit(2);
}

void Setup_Signals(void)
{
#ifdef _UNIX
	struct sigaction act, oact;
	act.sa_handler = Signal_Quit;
	sigemptyset(&act.sa_mask);
	act.sa_flags = 0;
	sigaction(SIGTERM, &act, &oact);
	sigaction(SIGINT, &act, &oact);
#endif
}

int VerifyFileDescriptors(int requested)
{
#ifdef _UNIX
	struct rlimit limit;
	if (!getrlimit(_SC_OPEN_MAX, &limit))
	{
		INFMSG("Hard limit on file descriptors: " << limit.rlim_max);
		if (limit.rlim_max < (unsigned int)requested)
		{
			ERRMSG("Too many file descriptors requested");
			ERRMSG("Hard Limit: " << limit.rlim_max << ", requested: " << requested);
			requested = limit.rlim_max;
		}

		limit.rlim_cur = limit.rlim_max; /* make soft limit the max */
		if (setrlimit(_SC_OPEN_MAX, &limit) == -1)
		{
			ERRMSG("Error setting max file descriptors to " << limit.rlim_cur);
			exit(-1);
		}
		INFMSG("Soft limit on file descriptors: " << limit.rlim_cur);
	}
	else
	{
		ERRMSG("Couldn't get limit for _SC_OPEN_MAX");
		exit(-1);
	}
#endif
	return requested;
}



GeneralsMatcher *s_generalsMatcher = NULL;
GeneralsClientMatcher *s_generalsClientMatcher = NULL;

int main(int argc, char ** argv)
{
	Wstring config_fname = "matchbot.cfg";

	// You can specify the config file on the command line
	if (argc == 2)
		config_fname = argv[1];

	// Read the config file
	FILE *fp;
	if ((fp = fopen(config_fname.get(), "r")) == NULL)
	{
		cerr << "\nCan't open the config file '" << config_fname.get() << "'\n\n";
		cerr << Program_Usage << endl;
		exit( -1);
	}
	fclose(fp);

	Global.ReadFile(config_fname.get());

	// Setup debugging & logging output
	Wstring output_file;
	output_file.set("matchbot.log");
	Global.config.getString("OUTPUTFILE", output_file);
	if (output_file != "STDOUT")
	{
		int append = 1;
		Global.config.getInt("APPENDTOLOG", append);
		if (append)
			output_device = new FileD(output_file.get(), "a");
		else
			output_device = new FileD(output_file.get(), "w");
		if (!output_device)
		{
			cerr << "Could not open " << output_file.get() << " for writing!" << endl;
			exit( -1);
		}
	}
	else
	{
		output_device = new StdoutD;
	}
	MsgManager::setAllStreams(output_device);
	DBGMSG("Matching bot started!");
	INFMSG("Matching bot " << argv[0] << " started!");

	// Setup logging of suspicious activity
	Wstring paranoid_output_file;
	paranoid_output_file.set("hacks.log");
	Global.config.getString("PARANOIDFILE", paranoid_output_file);
	if (paranoid_output_file != "STDOUT")
	{
		paranoid_output_device = new FileD(paranoid_output_file.get(), "a");
		if (!paranoid_output_device)
		{
			cerr << "Could not open " << paranoid_output_file.get() << " for writing!" << endl;
			exit( -1);
		}
	}
	else
	{
		paranoid_output_device = new StdoutD;
	}
	MyMsgManager::setParanoidStream(paranoid_output_device);
	DBGMSG("Hack log started!");
	PARANOIDMSG("Hack log started!");

	Setup_Signals();

#ifdef _WINDOWS
	// ----- Initialize Winsock -----
	WORD verReq = MAKEWORD(2, 2);
	WSADATA wsadata;

	int err = WSAStartup(verReq, &wsadata);
	if (err != 0)
	{
		ERRMSG("Winsock Init failed.");
		return 1;
	}

	if ((LOBYTE(wsadata.wVersion) != 2) || (HIBYTE(wsadata.wVersion) !=2))
	{
		ERRMSG("Winsock DLL is not 2.2");
		WSACleanup();
		ERRMSG("Winsock Init failed.");
		return 1;
	}
	INFMSG("Winsock Init done.");
#endif

	// Check game type & start matcher
	Wstring gametype = "unknown";
	Global.config.getString("GAME", gametype);

	// Command-line override for gamemode.
	// This is for test suites, so they can use
	// the same config file as the corresponding
	// matchbot.
	const char *s = argv[0] + strlen(argv[0]);
	while (s > argv[0] && *s != '/')
		--s;
	if (*s == '/')
		++s;
	Wstring exe = s;
	exe.toLower();
	DBGMSG("Executable file is [" << exe.get() << "]");

	if (gametype == "Generals")
	{
		DBGMSG("Generals matching behavior");
		s_generalsMatcher = new GeneralsMatcher;
		s_generalsMatcher->connectAndLoop();
	}
	else if (gametype == "GeneralsClient")
	{
		DBGMSG("Generals TEST client matching behavior");
		s_generalsClientMatcher = new GeneralsClientMatcher;
		s_generalsClientMatcher->connectAndLoop();
	}
	else
	{
		cerr << "\nNo valid GAME entry found!" << endl;
		exit( -1);
	}

	if (s_generalsMatcher)
		delete s_generalsMatcher;
	if (s_generalsClientMatcher)
		delete s_generalsClientMatcher;

	return 0;
}

/*----------------------------------------------------------------------+
| THREAD: logMonitor                                                    |
| This thread is spawned once per execution.  It will activate after    |
| midnight and create a new log file.  The old one gets put into the    |
| logfiles directory.                                                   |
`----------------------------------------------------------------------*/
void logMonitor(void *)
{
#ifdef _UNIX
	Xtime xtime;
	time_t curtime;
	//char timebuf[40];
	char filenamebuf[128];
	int delay = -1;
	Global.config.getInt("ROTATEDELAY", delay);
	DBGMSG("ROTATEDELAY: " << delay);
	if (delay == -1)
		return ;
	while (1)
	{
		curtime = time(NULL);
		// get the number of seconds that have passed since midnight
		// of the current day.
		curtime -= TimezoneOffset();
		time_t timeofday = curtime % (delay);
		if ((timeofday > 0) && (timeofday <= 300))
		{
			// We're within 5 minutes of midnight, switch the files.

			DBGMSG("about to switch.");
			Wstring logfilename = "matchbot.log";
			Global.config.getString("OUTPUTFILE", logfilename);
			Wstring newfilename = "tmp.log";
			Global.config.getString("ROTATEFILE", newfilename);
			MsgManager::ReplaceAllStreams((FileD*)output_device, logfilename.get(), newfilename.get());
			Wstring logpath = "logs";
			Global.config.getString("LOGPATH", logpath);
			xtime.update();
			sprintf(filenamebuf, "%s/%02d%02d%04d_%02d%02d%02d_log", logpath.get(), xtime.getMonth(),
			        xtime.getMDay(), xtime.getYear(), xtime.getHour(), xtime.getMinute(), xtime.getSecond());
			rename(newfilename.get(), filenamebuf);
			DBGMSG("Normal: Just been switched.  " << logfilename.get() << ", " << newfilename.get());
			sleep(60*60*23); // sleep the next 23 hours
		}
		sleep(300);
	}
#endif
}

void rotateOutput(void)
{
	Xtime xtime;
	char filenamebuf[128];
	Wstring logfilename = "matchbot.log";
	Wstring newfilename = "tmp.log";
	Wstring logpath = "logs";

	DBGMSG("About to switch.");

	Global.config.getString("OUTPUTFILE", logfilename);
	Global.config.getString("ROTATEFILE", newfilename);
	Global.config.getString("LOGPATH", logpath);

	// This grabs the semaphore, renames the file, and switches the output device
	MsgManager::ReplaceAllStreams((FileD*)output_device, logfilename.get(),
	                              newfilename.get());

	// clean up the tmp filename and move it to the log dir.
	sprintf(filenamebuf, "%s/%02d%02d%04d_%02d%02d%02d_log", logpath.get(),
	        xtime.getMonth(), xtime.getMDay(), xtime.getYear(), xtime.getHour(),
	        xtime.getMinute(), xtime.getSecond());
#ifdef _WINDOWS
	mkdir(logpath.get());
#else
	mkdir(logpath.get(), 00666);
#endif
	rename(newfilename.get(), filenamebuf);

	DBGMSG("Normal: Just been switched.  " << logfilename.get() << ", " <<
	       newfilename.get());
}

void paranoidLogMonitor(void *)
{
#ifdef _UNIX
	Xtime xtime;
	time_t curtime;
	//char timebuf[40];
	char filenamebuf[128];
	int delay = -1;
	Global.config.getInt("ROTATEDELAY", delay);
	PARANOIDMSG("ROTATEDELAY: " << delay);
	if (delay == -1)
		return ;
	while (1)
	{
		curtime = time(NULL);
		// get the number of seconds that have passed since midnight
		// of the current day.
		curtime -= TimezoneOffset();
		time_t timeofday = curtime % (delay);
		if ((timeofday > 0) && (timeofday <= 300))
		{
			// We're within 5 minutes of midnight, switch the files.

			PARANOIDMSG("about to switch.");
			Wstring logfilename = "matchbot.log";
			Global.config.getString("PARANOIDFILE", logfilename);
			Wstring newfilename = "tmp.log";
			Global.config.getString("ROTATEPARANOIDFILE", newfilename);
			MyMsgManager::ReplaceAllStreams((FileD*)paranoid_output_device, logfilename.get(), newfilename.get());
			Wstring logpath = "logs";
			Global.config.getString("PARANOIDLOGPATH", logpath);
			xtime.update();
			sprintf(filenamebuf, "%s/%02d%02d%04d_%02d%02d%02d_log", logpath.get(), xtime.getMonth(),
			        xtime.getMDay(), xtime.getYear(), xtime.getHour(), xtime.getMinute(), xtime.getSecond());
			rename(newfilename.get(), filenamebuf);
			PARANOIDMSG("Paranoid: Just been switched.  " << logfilename.get() << ", " << newfilename.get());
			sleep(60*60*23); // sleep the next 23 hours
		}
		sleep(300);
	}
#endif
}

void rotateParanoid(void)
{
	Xtime xtime;
	char filenamebuf[128];
	Wstring logfilename = "matchbot.log";
	Wstring newfilename = "tmp.log";
	Wstring logpath = "logs";

	PARANOIDMSG("About to switch.");

	Global.config.getString("PARANOIDFILE", logfilename);
	Global.config.getString("ROTATEPARANOIDFILE", newfilename);
	Global.config.getString("PARANOIDLOGPATH", logpath);

	// This grabs the semaphore, renames the file, and switches the output device
	MyMsgManager::ReplaceAllStreams((FileD*)output_device, logfilename.get(),
	                                newfilename.get());

	// clean up the tmp filename and move it to the log dir.
	sprintf(filenamebuf, "%s/%02d%02d%04d_%02d%02d%02d_log", logpath.get(),
	        xtime.getMonth(), xtime.getMDay(), xtime.getYear(), xtime.getHour(),
	        xtime.getMinute(), xtime.getSecond());
#ifdef _WINDOWS
	mkdir(logpath.get());
#else
	mkdir(logpath.get(), 00666);
#endif
	rename(newfilename.get(), filenamebuf);

	PARANOIDMSG("Paranoid: Just been switched.  " << logfilename.get() << ", " <<
	            newfilename.get());
}