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"

    15

    16

    17

// ** scriptfile handling ** //

    18

static FILE * _script_file;

    19

static bool _script_running;

    20

    21

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

    22

    23

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

    24

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

    25

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

    26

    27

    28

// ** supporting functions ** //

    29

    30

static uint32 GetArgumentInteger(const char* arg)

    31

{

    32

	uint32 result;

    33

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

    34

    35

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

    36

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

    37

    38

	return result;

    39

}

    40

    41

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

    42

/* variable and command hooks   */

    43

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

    44

    45

#ifdef ENABLE_NETWORK

    46

    47

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetwork)

    48

{

    49

	if (_networking) {

    50

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

    51

		return false;

    52

	}

    53

	return true;

    54

}

    55

    56

DEF_CONSOLE_VAR_HOOK(ConVarHookNoNetClient)

    57

{

    58

	if (!_network_available) {

    59

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

    60

		return false;

    61

	}

    62

	if (!_network_server) {

    63

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

    64

		return false;

    65

	}

    66

	return true;

    67

}

    68

    69

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetClient)

    70

{

    71

	if (!_network_available) {

    72

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

    73

		return false;

    74

	}

    75

	if (!_network_server) {

    76

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

    77

		return false;

    78

	}

    79

	return true;

    80

}

    81

    82

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetServer)

    83

{

    84

	if (!_network_available) {

    85

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

    86

		return false;

    87

	}

    88

	if (_network_server) {

    89

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

    90

		return false;

    91

	}

    92

	return true;

    93

}

    94

    95

DEF_CONSOLE_CMD_HOOK(ConCmdHookNeedNetwork)

    96

{

    97

	if (!_network_available) {

    98

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

    99

		return false;

   100

	}

   101

	if (!_networking) {

   102

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

   103

		return false;

   104

	}

   105

	return true;

   106

}

   107

   108

#endif /* ENABLE_NETWORK */

   109

   110

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

   111

/* reset commands               */

   112

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

   113

   114

DEF_CONSOLE_CMD(ConResetEngines)

   115

{

   116

	StartupEngines();

   117

	return 0;

   118

}

   119

   120

#ifdef _DEBUG

   121

DEF_CONSOLE_CMD(ConResetTile)

   122

{

   123

	if (argc == 2) {

   124

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

   125

		DoClearSquare(tile);

   126

	}

   127

   128

	return 0;

   129

}

   130

#endif

   131

   132

DEF_CONSOLE_CMD(ConScrollToTile)

   133

{

   134

	if (argc == 2) {

   135

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

   136

		ScrollMainWindowToTile(tile);

   137

	}

   138

   139

	return 0;

   140

}

   141

   245

   246

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

   247

// * Network Core Console Commands * //

   248

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

   249

#ifdef ENABLE_NETWORK

   250

   251

DEF_CONSOLE_CMD(ConBan)

   252

{

   253

	NetworkClientInfo *ci;

   254

   255

	if (argc == 2) {

   256

		uint32 index = atoi(argv[1]);

   257

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

   258

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

   259

			return NULL;

   260

		}

   261

		if (index == 0) {

   262

			IConsoleError("Invalid Client-ID");

   263

			return NULL;

   264

		}

   265

   266

		ci = NetworkFindClientInfoFromIndex(index);

   267

   268

		if (ci != NULL) {

   269

			uint i;

   270

			/* Add user to ban-list */

   271

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

   272

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

   273

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

   274

					break;

   275

				}

   276

			}

   277

   278

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

   279

			return NULL;

   280

		} else {

   281

			IConsoleError("Client-ID not found");

   282

			return NULL;

   283

		}

   284

	}

   285

   286

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

   287

   288

	return NULL;

   289

}

   290

   291

DEF_CONSOLE_CMD(ConUnBan)

   292

{

   293

	if (argc == 2) {

   294

		uint i;

   295

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

   296

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

   297

				continue;

   298

   299

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

   300

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

   301

				IConsolePrint(_iconsole_color_default, "IP unbanned.");

   302

				return NULL;

   303

			}

   304

		}

   305

   306

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

   307

   308

		return NULL;

   309

	}

   310

   311

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

   312

   313

	return NULL;

   314

}

   315

   316

DEF_CONSOLE_CMD(ConBanList)

   317

{

   318

	uint i;

   319

   320

	IConsolePrint(_iconsole_color_default, "Banlist: ");

   321

   322

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

   323

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

   324

			continue;

   325

   326

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

   327

	}

   328

   329

	return NULL;

   330

}

   331

   332

DEF_CONSOLE_CMD(ConStatus)

   333

{

   334

	const char *status;

   335

	int lag;

   336

	const NetworkClientState *cs;

   337

	const NetworkClientInfo *ci;

   338

	FOR_ALL_CLIENTS(cs) {

   339

		lag = NetworkCalculateLag(cs);

   340

		ci = DEREF_CLIENT_INFO(cs);

   341

   342

		switch (cs->status) {

   343

			case STATUS_INACTIVE:

   344

				status = "inactive";

   345

				break;

   346

			case STATUS_AUTH:

   347

				status = "authorized";

   348

				break;

   349

			case STATUS_MAP:

   350

				status = "loading map";

   351

				break;

   352

			case STATUS_DONE_MAP:

   353

				status = "done map";

   354

				break;

   355

			case STATUS_PRE_ACTIVE:

   356

				status = "ready";

   357

				break;

   358

			case STATUS_ACTIVE:

   359

				status = "active";

   360

				break;

   361

			default:

   362

				status = "unknown";

   363

				break;

   364

		}

   365

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

   366

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

   367

	}

   368

   369

	return NULL;

   370

}

   371

   372

DEF_CONSOLE_CMD(ConKick)

   373

{

   374

	NetworkClientInfo *ci;

   375

   376

	if (argc == 2) {

   377

		uint32 index = atoi(argv[1]);

   378

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

   379

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

   380

			return NULL;

   381

		}

   382

		if (index == 0) {

   383

			IConsoleError("Invalid Client-ID");

   384

			return NULL;

   385

		}

   386

   387

		ci = NetworkFindClientInfoFromIndex(index);

   388

   389

		if (ci != NULL) {

   390

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

   391

			return NULL;

   392

		} else {

   393

			IConsoleError("Client-ID not found");

   394

			return NULL;

   395

		}

   396

	}

   397

   398

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

   399

   400

	return NULL;

   401

}

   402

   403

DEF_CONSOLE_CMD(ConResetCompany)

   404

{

   405

	Player *p;

   406

	NetworkClientState *cs;

   407

	NetworkClientInfo *ci;

   408

   409

	if (argc == 2) {

   410

		uint32 index = atoi(argv[1]);

   411

   412

		/* Check valid range */

   413

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

   414

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

   415

			return NULL;

   416

		}

   417

   418

		/* Check if company does exist */

   419

		index--;

   420

		p = DEREF_PLAYER(index);

   421

		if (!p->is_active) {

   422

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

   423

			return NULL;

   424

		}

   425

   426

		if (p->is_ai) {

   427

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

   428

			return NULL;

   429

		}

   430

   431

		/* Check if the company has active players */

   432

		FOR_ALL_CLIENTS(cs) {

   433

			ci = DEREF_CLIENT_INFO(cs);

   434

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

   435

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

   436

				return NULL;

   437

			}

   438

		}

   439

		ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   440

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

   441

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

   442

			return NULL;

   443

		}

   444

   445

		/* It is safe to remove this company */

   446

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

   447

		IConsolePrint(_iconsole_color_default, "Company deleted.");

   448

		return NULL;

   449

	}

   450

   451

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

   452

   453

	return NULL;

   454

}

   455

   456

DEF_CONSOLE_CMD(ConNetworkClients)

   457

{

   458

	NetworkClientInfo *ci;

   459

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

   460

		if (ci->client_index != NETWORK_EMPTY_INDEX) {

   461

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

   462

		}

   463

	}

   464

   465

	return NULL;

   466

}

   467

   468

DEF_CONSOLE_CMD(ConNetworkConnect)

   469

{

   470

	char* ip;

   471

	const byte *port = NULL;

   472

	const byte *player = NULL;

   473

	uint16 rport;

   474

   475

	if (argc<2) return NULL;

   476

   477

	if (_networking) {

   478

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

   479

		NetworkDisconnect();

   480

	}

   481

   482

	ip = argv[1];

   483

	rport = NETWORK_DEFAULT_PORT;

   484

   485

	ParseConnectionString(&player, &port, ip);

   486

   487

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

   488

	if (player != NULL) {

   489

		_network_playas = atoi(player);

   490

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

   491

	}

   492

	if (port != NULL) {