mirror of
https://github.com/electronicarts/CnC_Generals_Zero_Hour.git
synced 2025-12-16 23:51:41 -05:00
Initial commit of Command & Conquer Generals and Command & Conquer Generals Zero Hour source code.
This commit is contained in:
LFeenanEA
208
GeneralsMD/Code/Tools/mangler/mangler.cpp
Normal file
208
GeneralsMD/Code/Tools/mangler/mangler.cpp
Normal file
@@ -0,0 +1,208 @@
|
||||
/*
|
||||
** 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/>.
|
||||
*/
|
||||
|
||||
|
||||
#include <iostream.h>
|
||||
#include <signal.h>
|
||||
#ifdef _WINDOWS
|
||||
#include <process.h> // *MUST* be included before ANY Wnet/Wlib headers if _REENTRANT is defined
|
||||
#endif
|
||||
|
||||
#include "mangler.h"
|
||||
#include "crc.h"
|
||||
#include "endian.h"
|
||||
|
||||
#include <configfile.h>
|
||||
#include "threadfac.h"
|
||||
|
||||
#include "xtime.h"
|
||||
#include <filed.h>
|
||||
#include <wstring.h>
|
||||
#include <wdebug.h>
|
||||
#include <udp.h>
|
||||
|
||||
void DisplayHelp(const char *prog)
|
||||
{
|
||||
cout << "Usage: " << prog << " <config file>" << endl;
|
||||
exit(0);
|
||||
}
|
||||
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
ConfigFile config;
|
||||
FILE* conf;
|
||||
|
||||
if( argc <= 1 )
|
||||
{
|
||||
// No args - use a default config file
|
||||
if ((conf = fopen("mangler.cfg", "r")) == NULL) {
|
||||
cout << "Cannot open mangler.cfg for reading." << endl;
|
||||
DisplayHelp(argv[0]);
|
||||
}
|
||||
config.readFile(conf);
|
||||
fclose(conf);
|
||||
}
|
||||
else if( argc == 2 && (strcmp(argv[1], "help") == 0 || strcmp(argv[1], "?") == 0 ||
|
||||
strcmp(argv[1], "-h") == 0) )
|
||||
DisplayHelp(argv[0]);
|
||||
else if( argc == 2 )
|
||||
{
|
||||
// Use a user-supplied config file
|
||||
if ((conf = fopen(argv[1], "r")) == NULL) {
|
||||
cout << "Cannot open " << argv[1] << " for reading." << endl;
|
||||
DisplayHelp(argv[0]);
|
||||
}
|
||||
config.readFile(conf);
|
||||
fclose(conf);
|
||||
}
|
||||
|
||||
// ----- LOGGING -----
|
||||
// Setup debugging & logging output
|
||||
Wstring output_file("mangler.log");
|
||||
config.getString("LOGFILE", output_file);
|
||||
Wstring backup_file;
|
||||
backup_file = output_file;
|
||||
backup_file += ".bak";
|
||||
rename(output_file.get(),backup_file.get()); // save the old file
|
||||
FileD output_device(output_file.get());
|
||||
MsgManager::setAllStreams(&output_device);
|
||||
DBGMSG("DBG working...");
|
||||
INFMSG("INF working...");
|
||||
WRNMSG("WRN working...");
|
||||
|
||||
if (htonl(0x12345678) == 0x12345678)
|
||||
{
|
||||
INFMSG("Host is network-byte-order");
|
||||
}
|
||||
else
|
||||
{
|
||||
INFMSG("Host is Intel-byte-order");
|
||||
}
|
||||
|
||||
|
||||
// ----- Initialize Winsock -----
|
||||
#ifdef _WINDOWS
|
||||
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
|
||||
|
||||
|
||||
// Set up a UDP listener
|
||||
uint8 *buff=new uint8[1024];
|
||||
int retval;
|
||||
UDP udp;
|
||||
UDP udp2;
|
||||
UDP udp3;
|
||||
UDP udp4;
|
||||
int port = 4321;
|
||||
config.getInt("PORT", port);
|
||||
uint8 blitz = 0;
|
||||
|
||||
uint32 localIP = 0;
|
||||
Wstring hostIPStr = "";
|
||||
config.getString("IP", hostIPStr);
|
||||
if (hostIPStr.length())
|
||||
{
|
||||
INFMSG("Binding to "<<hostIPStr.get()<<":"<<port<<"-"<<(port+3));
|
||||
localIP = ntohl(inet_addr(hostIPStr.get()));
|
||||
}
|
||||
else
|
||||
{
|
||||
INFMSG("Binding to localhost:"<<port<<"-"<<(port+3));
|
||||
}
|
||||
|
||||
retval = udp.Bind(localIP,(uint16)port);
|
||||
retval |= udp2.Bind(localIP,(uint16)port+1);
|
||||
retval |= udp3.Bind(localIP,(uint16)port+2);
|
||||
retval |= udp4.Bind(localIP,(uint16)port+3);
|
||||
if (retval != 0)
|
||||
{
|
||||
ERRMSG("Couldn't bind - error " << retval);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
unsigned char buf[1024];
|
||||
struct sockaddr_in addr;
|
||||
int packet_size = sizeof(ManglerData);
|
||||
INFMSG("sizeof(packet) == " << packet_size);
|
||||
unsigned char *theAddr;
|
||||
fd_set fdset;
|
||||
while (1)
|
||||
{
|
||||
retval = udp.Wait(15, 0, fdset);
|
||||
if (!retval)
|
||||
continue;
|
||||
|
||||
//DBGMSG("Wait returned " << retval);
|
||||
retval = udp.Read(buf, packet_size, &addr); // Wait until there is something on the socket
|
||||
if (retval > 0)
|
||||
{
|
||||
ManglerData *packet = (ManglerData *)buf;
|
||||
theAddr = (unsigned char *)&(addr.sin_addr.s_addr);
|
||||
if (retval != packet_size)
|
||||
{
|
||||
WRNMSG("Recieved mis-sized packet (" << retval << " bytes) from " << theAddr[0] << "." << theAddr[1] << "." << theAddr[2] << "." << theAddr[3] << ":" << addr.sin_port);
|
||||
}
|
||||
else
|
||||
{
|
||||
if (!Passes_CRC_Check(buf, packet_size))
|
||||
{
|
||||
WRNMSG("Recieved a bad packet - good length!");
|
||||
continue;
|
||||
}
|
||||
packet->NetCommandType = 44;
|
||||
packet->MyMangledPortNumber = addr.sin_port; // not changing to host order, cause its in network byte order now, and the game will expect it to stay that way.
|
||||
packet->MyMangledAddress[0] = theAddr[0];
|
||||
packet->MyMangledAddress[1] = theAddr[1];
|
||||
packet->MyMangledAddress[2] = theAddr[2];
|
||||
packet->MyMangledAddress[3] = theAddr[3];
|
||||
blitz = packet->BlitzMe;
|
||||
INFMSG("Packet ID = " << packet->packetID);
|
||||
Build_Packet_CRC(buf, packet_size);
|
||||
udp.Write(buf,packet_size,ntohl(addr.sin_addr.s_addr), ntohs(addr.sin_port));
|
||||
INFMSG("Saw " << (int)theAddr[0] << "." << (int)theAddr[1] << "." << (int)theAddr[2] << "." << (int)theAddr[3] << ":" << ntohs(addr.sin_port) << ((blitz)?" Blitzed":"") );
|
||||
|
||||
if (blitz)
|
||||
{
|
||||
udp2.Write(buf,packet_size,ntohl(addr.sin_addr.s_addr), ntohs(addr.sin_port)+1);
|
||||
udp3.Write(buf,packet_size,ntohl(addr.sin_addr.s_addr), ntohs(addr.sin_port)+2);
|
||||
udp4.Write(buf,packet_size,ntohl(addr.sin_addr.s_addr), ntohs(addr.sin_port)+3);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user