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(ConStatus)

   148

{

   149

	const char *status;

   150

	int lag;

   151

	const NetworkClientState *cs;

   152

	const NetworkClientInfo *ci;

   153

	FOR_ALL_CLIENTS(cs) {

   154

		lag = NetworkCalculateLag(cs);

   155

		ci = DEREF_CLIENT_INFO(cs);

   156

   157

		switch (cs->status) {

   158

			case STATUS_INACTIVE:

   159

				status = "inactive";

   160

				break;

   161

			case STATUS_AUTH:

   162

				status = "authorized";

   163

				break;

   164

			case STATUS_MAP:

   165

				status = "loading map";

   166

				break;

   167

			case STATUS_DONE_MAP:

   168

				status = "done map";

   169

				break;

   170

			case STATUS_PRE_ACTIVE:

   171

				status = "ready";

   172

				break;

   173

			case STATUS_ACTIVE:

   174

				status = "active";

   175

				break;

   176

			default:

   177

				status = "unknown";

   178

				break;

   179

		}

   180

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

   181

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

   182

	}

   183

   184

	return NULL;

   185

}

   186

   187

DEF_CONSOLE_CMD(ConKick)

   188

{

   189

	NetworkClientInfo *ci;

   190

   191

	if (argc == 2) {

   192

		uint32 index = atoi(argv[1]);

   193

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

   194

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

   195

			return NULL;

   196

		}

   197

		if (index == 0) {

   198

			IConsoleError("Invalid Client-ID");

   199

			return NULL;

   200

		}

   201

   202

		ci = NetworkFindClientInfoFromIndex(index);

   203

   204

		if (ci != NULL) {

   205

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

   206

			return NULL;

   207

		} else {

   208

			IConsoleError("Client-ID not found");

   209

			return NULL;

   210

		}

   211

	}

   212

   213

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

   214

   215

	return NULL;

   216

}

   217

   218

DEF_CONSOLE_CMD(ConResetCompany)

   219

{

   220

	Player *p;

   221

	NetworkClientState *cs;

   222

	NetworkClientInfo *ci;

   223

   224

	if (argc == 2) {

   225

		uint32 index = atoi(argv[1]);

   226

   227

		/* Check valid range */

   228

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

   229

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

   230

			return NULL;

   231

		}

   232

   233

		/* Check if company does exist */

   234

		index--;

   235

		p = DEREF_PLAYER(index);

   236

		if (!p->is_active) {

   237

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

   238

			return NULL;

   239

		}

   240

   241

		if (p->is_ai) {

   242

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

   243

			return NULL;

   244

		}

   245

   246

		/* Check if the company has active players */

   247

		FOR_ALL_CLIENTS(cs) {

   248

			ci = DEREF_CLIENT_INFO(cs);

   249

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

   250

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

   251

				return NULL;

   252

			}

   253

		}

   254

		ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   255

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

   256

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

   257

			return NULL;

   258

		}

   259

   260

		/* It is safe to remove this company */

   261

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

   262

		IConsolePrint(_iconsole_color_default, "Company deleted.");

   263

		return NULL;

   264

	}

   265

   266

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

   267

   268

	return NULL;

   269

}

   270

   271

DEF_CONSOLE_CMD(ConNetworkClients)

   272

{

   273

	NetworkClientInfo *ci;

   274

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

   275

		if (ci->client_index != NETWORK_EMPTY_INDEX) {

   276

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

   277

		}

   278

	}

   279

   280

	return NULL;

   281

}

   282

   283

DEF_CONSOLE_CMD(ConNetworkConnect)

   284

{

   285

	char* ip;

   286

	const byte *port = NULL;

   287

	const byte *player = NULL;

   288

	uint16 rport;

   289

   290

	if (argc<2) return NULL;

   291

   292

	if (_networking) {

   293

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

   294

		NetworkDisconnect();

   295

	}

   296

   297

	ip = argv[1];

   298

	rport = NETWORK_DEFAULT_PORT;

   299

   300

	ParseConnectionString(&player, &port, ip);

   301

   302

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

   303

	if (player != NULL) {

   304

		_network_playas = atoi(player);

   305

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

   306

	}

   307

	if (port != NULL) {

   308

		rport = atoi(port);

   309

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

   310

	}

   311

   312

	NetworkClientConnectGame(ip, rport);

   313

   314

	return NULL;

   315

}

   316

   317

#endif /* ENABLE_NETWORK */

   318

   319

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

   320

/*   script file console commands   */

   321

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

   322

   323

DEF_CONSOLE_CMD(ConExec)

   324

{

   325

	char cmd[1024];

   326

	bool doerror;

   327

   328

	if (argc<2) return NULL;

   329

   330

	doerror = true;

   331

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

   332

   333

	if (_script_file == NULL) {

   334

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

   335

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

   336

		return NULL;

   337

		}

   338

   339

	_script_running = true;

   340

   341

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

   342

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

   343

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

   344

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

   345

			IConsoleCmdExec(cmd);

   346

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

   347

	}

   348

   349

	_script_running = false;

   350

	fclose(_script_file);

   351

	return NULL;

   352

}

   353

   354

DEF_CONSOLE_CMD(ConReturn)

   355

{

   356

	_script_running = false;

   357

	return NULL;

   358

}

   359

   360

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

   361

/*   default console commands   */

   362

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

   363

   364

DEF_CONSOLE_CMD(ConScript)

   365

{

   366

	extern FILE* _iconsole_output_file;

   367

	if (_iconsole_output_file != NULL) {

   368

		IConsolePrintF(_iconsole_color_default, "file output complete");

   369

		fclose(_iconsole_output_file);

   370

	} else {

   371

		if (argc < 2) return NULL;

   372

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

   373

			argv[1]);

   374

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

   375

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

   376

	}

   377

	return NULL;

   378

}

   379

   380

   381

DEF_CONSOLE_CMD(ConEcho)

   382

{

   383

	if (argc < 2) return NULL;

   384

	IConsolePrint(_iconsole_color_default, argv[1]);

   385

	return NULL;

   386

}

   387

   388

DEF_CONSOLE_CMD(ConEchoC)

   389

{

   390

	if (argc < 3) return NULL;

   391

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

   392

	return NULL;

   393

}

   394

   395

extern void SwitchMode(int new_mode);

   396

   397

DEF_CONSOLE_CMD(ConNewGame)

   398

{

   399

	_docommand_recursive = 0;

   400

   401

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

   402

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

   403

   404

	SwitchMode(SM_NEWGAME);

   405

	return NULL;

   406

}

   407

   408

DEF_CONSOLE_CMD(ConPrintF)

   409

{

   410

	if (argc < 3) return NULL;

   411

	IConsolePrintF(_iconsole_color_default, argv[1] , argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17], argv[18], argv[19]); /* XXX ugh... */

   412

	return NULL;

   413

}

   414

   415

DEF_CONSOLE_CMD(ConPrintFC)

   416

{

   417

	if (argc < 3) return NULL;

   418

	IConsolePrintF(atoi(argv[1]), argv[2] , argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17], argv[18], argv[19]); /* XXX ugh... */

   419

	return NULL;

   420

}

   421

   422

DEF_CONSOLE_CMD(ConAlias)

   423

{

   424

	_iconsole_alias* alias;

   425

   426

	if (argc < 3) return NULL;

   427

   428

	alias = IConsoleAliasGet(argv[1]);

   429

	if (alias == NULL) {

   430

		IConsoleAliasRegister(argv[1],argv[2]);

   431

	} else {

   432

		free(alias->cmdline);

   433

		alias->cmdline = strdup(argv[2]);

   434

	}

   435

	return NULL;

   436

}

   437

   438

DEF_CONSOLE_CMD(ConScreenShot)

   439

{

   440

	if (argc < 2) {

   441

		_make_screenshot = 1;

   442

	} else {

   443

		if (strcmp(argv[1], "big") == 0)

   444

			_make_screenshot=2;

   445

		if (strcmp(argv[1], "no_con") == 0) {

   446

			IConsoleClose();

   447

			_make_screenshot = 1;

   448

		}

   449

	}

   450

	return NULL;

   451

}

   452

   453

DEF_CONSOLE_CMD(ConInfoVar)

   454

{

   455

	if (argc < 2) return NULL;

   456

	if (argt[1] != ICONSOLE_VAR_REFERENCE) {

   457

		IConsoleError("first argument has to be a variable reference");

   458

	} else {

   459

		_iconsole_var* item;

   460

		item = (_iconsole_var*)argv[1];

   461

		IConsolePrintF(_iconsole_color_default, "var_name: %s", item->name);

   462

		IConsolePrintF(_iconsole_color_default, "var_type: %i", item->type);

   463

		IConsolePrintF(_iconsole_color_default, "var_addr: %i", item->data.addr);

   464

		if (item->_malloc)

   465

			IConsolePrintF(_iconsole_color_default, "var_malloc: internal");

   466

		else

   467

			IConsolePrintF(_iconsole_color_default, "var_malloc: external");

   468

		if (item->hook_access) IConsoleWarning("var_access hooked");

   469

		if (item->hook_before_change) IConsoleWarning("var_before_change hooked");

   470

		if (item->hook_after_change) IConsoleWarning("var_after_change hooked");

   471

	}

   472

	return NULL;

   473

}

   474

   475

   476

DEF_CONSOLE_CMD(ConInfoCmd)

   477

{

   478

	if (argc < 2) return NULL;

   479

	if (argt[1] != ICONSOLE_VAR_UNKNOWN) {

   480

		IConsoleError("first argument has to be a command name");

   481

	} else {

   482

		_iconsole_cmd* item;

   483

		item = IConsoleCmdGet(argv[1]);

   484

		if (item == NULL) {

   485

			IConsoleError("the given command was not found");

   486

			return NULL;

   487

		}

   488

		IConsolePrintF(_iconsole_color_default, "cmd_name: %s", item->name);