1

/* -------------------- dont cross this line --------------------- */

     2

#include "stdafx.h"

     3

#include "ttd.h"

     4

#include "console.h"

     5

#include "engine.h"

     6

#include "functions.h"

     7

#include "variables.h"

     8

#include "network_data.h"

     9

#include "network_client.h"

    10

#include "network_server.h"

    11

#include "network_udp.h"

    12

#include "command.h"

    13

#include "settings.h"

    14

    15

    16

// ** scriptfile handling ** //

    17

static FILE * _script_file;

    18

static bool _script_running;

    19

    20

// ** console command / variable defines ** //

    21

    22

#define DEF_CONSOLE_CMD(yyyy) static _iconsole_var * yyyy(byte argc, char* argv[], byte argt[])

    23

#define DEF_CONSOLE_CMD_HOOK(yyyy) static bool yyyy(_iconsole_cmd * hookcmd)

    24

#define DEF_CONSOLE_VAR_HOOK(yyyy) static bool yyyy(_iconsole_var * hookvar)

    25

    26

    27

// ** supporting functions ** //

    28

    29

static uint32 GetArgumentInteger(const char* arg)

    30

{

    31

	uint32 result;

    32

	sscanf(arg, "%u", &result);

    33

    34

	if (result == 0 && arg[0] == '0' && arg[1] == 'x')

    35

		sscanf(arg, "%x", &result);

    36

    37

	return result;

    38

}

    39

    40

/* **************************** */

    41

/* variable and command hooks   */

    42

/* **************************** */

    43

    44

#ifdef ENABLE_NETWORK

    45

    46

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetwork)

    47

{

    48

	if (_networking) {

    49

		IConsoleError("This command is forbidden in multiplayer.");

    50

		return false;

    51

	}

    52

	return true;

    53

}

    54

    55

DEF_CONSOLE_VAR_HOOK(ConVarHookNoNetClient)

    56

{

    57

	if (!_network_available) {

    58

		IConsoleError("You can not use this command because there is no network available.");

    59

		return false;

    60

	}

    61

	if (!_network_server) {

    62

		IConsoleError("This variable only makes sense for a network server.");

    63

		return false;

    64

	}

    65

	return true;

    66

}

    67

    68

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetClient)

    69

{

    70

	if (!_network_available) {

    71

		IConsoleError("You can not use this command because there is no network available.");

    72

		return false;

    73

	}

    74

	if (!_network_server) {

    75

		IConsoleError("This command is only available for a network server.");

    76

		return false;

    77

	}

    78

	return true;

    79

}

    80

    81

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetServer)

    82

{

    83

	if (!_network_available) {

    84

		IConsoleError("You can not use this command because there is no network available.");

    85

		return false;

    86

	}

    87

	if (_network_server) {

    88

		IConsoleError("You can not use this command because you are a network-server.");

    89

		return false;

    90

	}

    91

	return true;

    92

}

    93

    94

DEF_CONSOLE_CMD_HOOK(ConCmdHookNeedNetwork)

    95

{

    96

	if (!_network_available) {

    97

		IConsoleError("You can not use this command because there is no network available.");

    98

		return false;

    99

	}

   100

	if (!_networking) {

   101

		IConsoleError("Not connected. Multiplayer only command.");

   102

		return false;

   103

	}

   104

	return true;

   105

}

   106

   107

#endif /* ENABLE_NETWORK */

   108

   109

/* **************************** */

   110

/* reset commands               */

   111

/* **************************** */

   112

   113

DEF_CONSOLE_CMD(ConResetEngines)

   114

{

   115

	StartupEngines();

   116

	return 0;

   117

}

   118

   119

#ifdef _DEBUG

   120

DEF_CONSOLE_CMD(ConResetTile)

   121

{

   122

	if (argc == 2) {

   123

		TileIndex tile = (TileIndex)GetArgumentInteger(argv[1]);

   124

		DoClearSquare(tile);

   125

	}

   126

   127

	return 0;

   128

}

   129

#endif

   130

   131

DEF_CONSOLE_CMD(ConScrollToTile)

   132

{

   133

	if (argc == 2) {

   134

		TileIndex tile = (TileIndex)GetArgumentInteger(argv[1]);

   135

		ScrollMainWindowToTile(tile);

   136

	}

   137

   138

	return 0;

   139

}

   140

   141

   142

// ********************************* //

   143

// * Network Core Console Commands * //

   144

// ********************************* //

   145

#ifdef ENABLE_NETWORK

   146

   147

DEF_CONSOLE_CMD(ConBan)

   148

{

   149

	NetworkClientInfo *ci;

   150

   151

	if (argc == 2) {

   152

		uint32 index = atoi(argv[1]);

   153

		if (index == NETWORK_SERVER_INDEX && !_network_dedicated) {

   154

			IConsolePrint(_iconsole_color_default, "Silly boy, you can not ban yourself!");

   155

			return NULL;

   156

		}

   157

		if (index == 0) {

   158

			IConsoleError("Invalid Client-ID");

   159

			return NULL;

   160

		}

   161

   162

		ci = NetworkFindClientInfoFromIndex(index);

   163

   164

		if (ci != NULL) {

   165

			uint i;

   166

			/* Add user to ban-list */

   167

			for (i = 0; i < lengthof(_network_ban_list); i++) {

   168

				if (_network_ban_list[i] == NULL || _network_ban_list[i][0] == '\0') {

   169

					_network_ban_list[i] = strdup(inet_ntoa(*(struct in_addr *)&ci->client_ip));

   170

					break;

   171

				}

   172

			}

   173

   174

			SEND_COMMAND(PACKET_SERVER_ERROR)(NetworkFindClientStateFromIndex(index), NETWORK_ERROR_KICKED);

   175

			return NULL;

   176

		} else {

   177

			IConsoleError("Client-ID not found");

   178

			return NULL;

   179

		}

   180

	}

   181

   182

	IConsolePrint(_iconsole_color_default, "Unknown usage. Usage: ban <client-id>. For client-ids, see 'clients'.");

   183

   184

	return NULL;

   185

}

   186

   187

DEF_CONSOLE_CMD(ConUnBan)

   188

{

   189

	if (argc == 2) {

   190

		uint i;

   191

		for (i = 0; i < lengthof(_network_ban_list); i++) {

   192

			if (_network_ban_list[i] == NULL || _network_ban_list[i][0] == '\0')

   193

				continue;

   194

   195

			if (strncmp(_network_ban_list[i], argv[1], strlen(_network_ban_list[i])) == 0) {

   196

				_network_ban_list[i][0] = '\0';

   197

				IConsolePrint(_iconsole_color_default, "IP unbanned.");

   198

				return NULL;

   199

			}

   200

		}

   201

   202

		IConsolePrint(_iconsole_color_default, "IP not in ban-list.");

   203

   204

		return NULL;

   205

	}

   206

   207

	IConsolePrint(_iconsole_color_default, "Unknown usage. Usage: unban <ip>.");

   208

   209

	return NULL;

   210

}

   211

   212

DEF_CONSOLE_CMD(ConBanList)

   213

{

   214

	uint i;

   215

   216

	IConsolePrint(_iconsole_color_default, "Banlist: ");

   217

   218

	for (i = 0; i < lengthof(_network_ban_list); i++) {

   219

		if (_network_ban_list[i] == NULL || _network_ban_list[i][0] == '\0')

   220

			continue;

   221

   222

		IConsolePrintF(_iconsole_color_default, "  %d) %s", i + 1, _network_ban_list[i]);

   223

	}

   224

   225

	return NULL;

   226

}

   227

   228

DEF_CONSOLE_CMD(ConStatus)

   229

{

   230

	const char *status;

   231

	int lag;

   232

	const NetworkClientState *cs;

   233

	const NetworkClientInfo *ci;

   234

	FOR_ALL_CLIENTS(cs) {

   235

		lag = NetworkCalculateLag(cs);

   236

		ci = DEREF_CLIENT_INFO(cs);

   237

   238

		switch (cs->status) {

   239

			case STATUS_INACTIVE:

   240

				status = "inactive";

   241

				break;

   242

			case STATUS_AUTH:

   243

				status = "authorized";

   244

				break;

   245

			case STATUS_MAP:

   246

				status = "loading map";

   247

				break;

   248

			case STATUS_DONE_MAP:

   249

				status = "done map";

   250

				break;

   251

			case STATUS_PRE_ACTIVE:

   252

				status = "ready";

   253

				break;

   254

			case STATUS_ACTIVE:

   255

				status = "active";

   256

				break;

   257

			default:

   258

				status = "unknown";

   259

				break;

   260

		}

   261

		IConsolePrintF(8, "Client #%d/%s  status: %s  frame-lag: %d  play-as: %d  unique-id: %s",

   262

			cs->index, ci->client_name, status, lag, ci->client_playas, ci->unique_id);

   263

	}

   264

   265

	return NULL;

   266

}

   267

   268

DEF_CONSOLE_CMD(ConKick)

   269

{

   270

	NetworkClientInfo *ci;

   271

   272

	if (argc == 2) {

   273

		uint32 index = atoi(argv[1]);

   274

		if (index == NETWORK_SERVER_INDEX && !_network_dedicated) {

   275

			IConsolePrint(_iconsole_color_default, "Silly boy, you can not kick yourself!");

   276

			return NULL;

   277

		}

   278

		if (index == 0) {

   279

			IConsoleError("Invalid Client-ID");

   280

			return NULL;

   281

		}

   282

   283

		ci = NetworkFindClientInfoFromIndex(index);

   284

   285

		if (ci != NULL) {

   286

			SEND_COMMAND(PACKET_SERVER_ERROR)(NetworkFindClientStateFromIndex(index), NETWORK_ERROR_KICKED);

   287

			return NULL;

   288

		} else {

   289

			IConsoleError("Client-ID not found");

   290

			return NULL;

   291

		}

   292

	}

   293

   294

	IConsolePrint(_iconsole_color_default, "Unknown usage. Usage: kick <client-id>. For client-ids, see 'clients'.");

   295

   296

	return NULL;

   297

}

   298

   299

DEF_CONSOLE_CMD(ConResetCompany)

   300

{

   301

	Player *p;

   302

	NetworkClientState *cs;

   303

	NetworkClientInfo *ci;

   304

   305

	if (argc == 2) {

   306

		uint32 index = atoi(argv[1]);

   307

   308

		/* Check valid range */

   309

		if (index < 1 || index > MAX_PLAYERS) {

   310

			IConsolePrintF(_iconsole_color_error, "Company does not exist. Company-ID must be between 1 and %d.", MAX_PLAYERS);

   311

			return NULL;

   312

		}

   313

   314

		/* Check if company does exist */

   315

		index--;

   316

		p = DEREF_PLAYER(index);

   317

		if (!p->is_active) {

   318

			IConsolePrintF(_iconsole_color_error, "Company does not exist.");

   319

			return NULL;

   320

		}

   321

   322

		if (p->is_ai) {

   323

			IConsolePrintF(_iconsole_color_error, "Company is owned by an AI.");

   324

			return NULL;

   325

		}

   326

   327

		/* Check if the company has active players */

   328

		FOR_ALL_CLIENTS(cs) {

   329

			ci = DEREF_CLIENT_INFO(cs);

   330

			if (ci->client_playas-1 == index) {

   331

				IConsolePrintF(_iconsole_color_error, "Cannot remove company: a client is connected to that company.");

   332

				return NULL;

   333

			}

   334

		}

   335

		ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   336

		if (ci->client_playas-1 == index) {

   337

			IConsolePrintF(_iconsole_color_error, "Cannot remove company: a client is connected to that company.");

   338

			return NULL;

   339

		}

   340

   341

		/* It is safe to remove this company */

   342

		DoCommandP(0, 2, index, NULL, CMD_PLAYER_CTRL);

   343

		IConsolePrint(_iconsole_color_default, "Company deleted.");

   344

		return NULL;

   345

	}

   346

   347

	IConsolePrint(_iconsole_color_default, "Unknown usage. Usage: reset_company <company-id>.");

   348

   349

	return NULL;

   350

}

   351

   352

DEF_CONSOLE_CMD(ConNetworkClients)

   353

{

   354

	NetworkClientInfo *ci;

   355

	for (ci = _network_client_info; ci != &_network_client_info[MAX_CLIENT_INFO]; ci++) {

   356

		if (ci->client_index != NETWORK_EMPTY_INDEX) {

   357

			IConsolePrintF(8,"Client #%d   name: %s  play-as: %d", ci->client_index, ci->client_name, ci->client_playas);

   358

		}

   359

	}

   360

   361

	return NULL;

   362

}

   363

   364

DEF_CONSOLE_CMD(ConNetworkConnect)

   365

{

   366

	char* ip;

   367

	const byte *port = NULL;

   368

	const byte *player = NULL;

   369

	uint16 rport;

   370

   371

	if (argc<2) return NULL;

   372

   373

	if (_networking) {

   374

		// We are in network-mode, first close it!

   375

		NetworkDisconnect();

   376

	}

   377

   378

	ip = argv[1];

   379

	rport = NETWORK_DEFAULT_PORT;

   380

   381

	ParseConnectionString(&player, &port, ip);

   382

   383

	IConsolePrintF(_iconsole_color_default, "Connecting to %s...", ip);

   384

	if (player != NULL) {

   385

		_network_playas = atoi(player);

   386

		IConsolePrintF(_iconsole_color_default, "    player-no: %s", player);

   387

	}

   388

	if (port != NULL) {

   389

		rport = atoi(port);

   390

		IConsolePrintF(_iconsole_color_default, "    port: %s", port);

   391

	}

   392

   393

	NetworkClientConnectGame(ip, rport);

   394

   395

	return NULL;

   396

}

   397

   398

#endif /* ENABLE_NETWORK */

   399

   400

/* ******************************** */

   401

/*   script file console commands   */

   402

/* ******************************** */

   403

   404

DEF_CONSOLE_CMD(ConExec)

   405

{

   406

	char cmd[1024];

   407

	bool doerror;

   408

   409

	if (argc<2) return NULL;

   410

   411

	doerror = true;

   412

	_script_file = fopen(argv[1], "r");

   413

   414

	if (_script_file == NULL) {

   415

		if (argc>2) if (atoi(argv[2])==0) doerror=false;

   416

		if (doerror) IConsoleError("script file not found");

   417

		return NULL;

   418

		}

   419

   420

	_script_running = true;

   421

   422

	fgets(cmd, sizeof(cmd), _script_file);

   423

	while (!feof(_script_file) && _script_running) {

   424

		strtok(cmd, "\r\n#");

   425

		if (strlen(cmd) > 0 && cmd[0] != '#')

   426

			IConsoleCmdExec(cmd);

   427

		fgets(cmd, sizeof(cmd), _script_file);

   428

	}

   429

   430

	_script_running = false;

   431

	fclose(_script_file);

   432

	return NULL;

   433

}

   434

   435

DEF_CONSOLE_CMD(ConReturn)

   436

{

   437

	_script_running = false;

   438

	return NULL;

   439

}

   440

   441

/* **************************** */

   442

/*   default console commands   */

   443

/* **************************** */

   444

   445

DEF_CONSOLE_CMD(ConScript)

   446

{

   447

	extern FILE* _iconsole_output_file;

   448

	if (_iconsole_output_file != NULL) {

   449

		IConsolePrintF(_iconsole_color_default, "file output complete");

   450

		fclose(_iconsole_output_file);

   451

	} else {

   452

		if (argc < 2) return NULL;

   453

		IConsolePrintF(_iconsole_color_default, "file output started to: %s",

   454

			argv[1]);

   455

		_iconsole_output_file = fopen(argv[1], "ab");

   456

		if (_iconsole_output_file == NULL) IConsoleError("could not open file");

   457

	}

   458

	return NULL;

   459

}

   460

   461

   462

DEF_CONSOLE_CMD(ConEcho)

   463

{

   464

	if (argc < 2) return NULL;

   465

	IConsolePrint(_iconsole_color_default, argv[1]);

   466

	return NULL;

   467

}

   468

   469

DEF_CONSOLE_CMD(ConEchoC)

   470

{

   471

	if (argc < 3) return NULL;

   472

	IConsolePrint(atoi(argv[1]), argv[2]);

   473

	return NULL;

   474

}

   475

   476

extern void SwitchMode(int new_mode);

   477

   478

DEF_CONSOLE_CMD(ConNewGame)

   479

{

   480

	_docommand_recursive = 0;

   481

   482

	_random_seeds[0][0] = Random();

   483

	_random_seeds[0][1] = InteractiveRandom();

   484

   485

	SwitchMode(SM_NEWGAME);

   486

	return NULL;

   487

}

   488

   489

DEF_CONSOLE_CMD(ConPrintF)

   490

{