1

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

     2

#include "stdafx.h"

     3

#include "ttd.h"

     4

#include "console.h"

     5

#include "debug.h"

     6

#include "engine.h"

     7

#include "functions.h"

     8

#include "string.h"

     9

#include "variables.h"

    10

#include "network_data.h"

    11

#include "network_client.h"

    12

#include "network_server.h"

    13

#include "network_udp.h"

    14

#include "command.h"

    15

#include "settings.h"

    16

#include "hal.h" /* for file list */

    17

    18

    19

// ** scriptfile handling ** //

    20

static FILE * _script_file;

    21

static bool _script_running;

    22

    23

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

    24

    25

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

    26

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

    27

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

    28

    29

    30

// ** supporting functions ** //

    31

    32

static uint32 GetArgumentInteger(const char* arg)

    33

{

    34

	uint32 result;

    35

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

    36

    37

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

    38

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

    39

    40

	return result;

    41

}

    42

    43

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

    44

/* variable and command hooks   */

    45

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

    46

    47

#ifdef ENABLE_NETWORK

    48

    49

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetwork)

    50

{

    51

	if (_networking) {

    52

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

    53

		return false;

    54

	}

    55

	return true;

    56

}

    57

    58

DEF_CONSOLE_VAR_HOOK(ConVarHookNoNetClient)

    59

{

    60

	if (!_network_available) {

    61

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

    62

		return false;

    63

	}

    64

	if (!_network_server) {

    65

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

    66

		return false;

    67

	}

    68

	return true;

    69

}

    70

    71

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetClient)

    72

{

    73

	if (!_network_available) {

    74

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

    75

		return false;

    76

	}

    77

	if (!_network_server) {

    78

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

    79

		return false;

    80

	}

    81

	return true;

    82

}

    83

    84

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetServer)

    85

{

    86

	if (!_network_available) {

    87

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

    88

		return false;

    89

	}

    90

	if (_network_server) {

    91

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

    92

		return false;

    93

	}

    94

	return true;

    95

}

    96

    97

DEF_CONSOLE_CMD_HOOK(ConCmdHookNeedNetwork)

    98

{

    99

	if (!_network_available) {

   100

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

   101

		return false;

   102

	}

   103

	if (!_networking) {

   104

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

   105

		return false;

   106

	}

   107

	return true;

   108

}

   109

   110

#endif /* ENABLE_NETWORK */

   111

   112

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

   113

/* reset commands               */

   114

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

   115

   116

DEF_CONSOLE_CMD(ConResetEngines)

   117

{

   118

	StartupEngines();

   119

	return 0;

   120

}

   121

   122

#ifdef _DEBUG

   123

DEF_CONSOLE_CMD(ConResetTile)

   124

{

   125

	if (argc == 2) {

   126

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

   127

		DoClearSquare(tile);

   128

	}

   129

   130

	return 0;

   131

}

   132

#endif

   133

   134

DEF_CONSOLE_CMD(ConScrollToTile)

   135

{

   136

	if (argc == 2) {

   137

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

   138

		ScrollMainWindowToTile(tile);

   139

	}

   140

   141

	return 0;

   142

}

   143

   144

extern bool SafeSaveOrLoad(const char *filename, int mode, int newgm);

   145

extern void BuildFileList(void);

   146

extern void SetFiosType(const byte fiostype);

   147

   177

/* Load a file-number from current dir */

   178

static void LoadMap(uint no)

   179

{

   180

	/* Build file list */

   181

	BuildFileList();

   182

   183

	/* Check if in range */

   184

	if (no != 0 && no <= (uint)_fios_num) {

   185

		const FiosItem *item = &_fios_list[no - 1];

   186

   187

		if (item->type == FIOS_TYPE_FILE) {

   188

			/* Load the file */

   189

			_switch_mode = SM_LOAD;

   190

			SetFiosType(item->type);

   191

			strcpy(_file_to_saveload.name, FiosBrowseTo(item));

   192

   193

			IConsolePrint(_iconsole_color_default, "Loading map...");

   194

		} else

   195

			IConsolePrint(_iconsole_color_error, "That is not a map.");

   196

   197

	} else /* Show usages */

   198

		IConsolePrint(_iconsole_color_error, "Unknown map. Use 'list_files' and 'goto_dir' to find the numbers of the savegame.");

   199

   200

	/* Free the file-list */

   201

	FiosFreeSavegameList();

   202

}

   203

   204

/* Load a file from a map */

   205

DEF_CONSOLE_CMD(ConLoad)

   206

{

   207

	/* We need 1 argument */

   208

	if (argc == 2) {

   209

		/* Load the map */

   210

		LoadMap(atoi(argv[1]));

   211

		return NULL;

   212

	}

   213

   214

	/* Give usage */

   215

	IConsolePrint(_iconsole_color_default, "Unknown usage. Usage: load <file-no>");

   216

	return NULL;

   217

}

   218

   219

/* List all the files in the current dir via console */

   220

DEF_CONSOLE_CMD(ConListFiles)

   221

{

   222

	const FiosItem *item;

   223

	int pos = 0;

   224

   225

	/* Build the file-list */

   226

	BuildFileList();

   227

   228

	/* As long as we have files */

   229

	while (pos < _fios_num) {

   230

		item = _fios_list + pos;

   231

		pos++;

   232

		/* Show them */

   233

		IConsolePrintF(_iconsole_color_default, "%d) %s", pos, item->title[0] ? item->title : item->name);

   234

	}

   235

   236

	/* Destroy the file list */

   237

	FiosFreeSavegameList();

   238

   239

	return NULL;

   240

}

   241

   242

/* Get an Specific file */

   243

DEF_CONSOLE_CMD(ConScanFiles)

   244

{

   245

	const FiosItem *item;

   246

	int pos = 0;

   247

	_iconsole_var* result;

   248

   249

   250

	result = IConsoleVarAlloc(ICONSOLE_VAR_STRING);

   251

   252

	if (argc <= 1) {

   253

		IConsoleVarSetString(result, "0");

   254

		return result; // return an zero

   255

	}

   256

   257

	/* Build the file-list */

   258

	BuildFileList();

   259

   260

	/* As long as we have files */

   261

	while (pos < _fios_num) {

   262

		item = _fios_list + pos;

   263

		pos++;

   264

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

   265

			if (item->type == FIOS_TYPE_PARENT) {

   266

				// huh we are searching for the parent directory

   267

				char buffer[10];

   268

				sprintf(buffer, "%d", pos);

   269

				IConsoleVarSetString(result, buffer);

   270

				return result;

   271

			}

   272

		} else

   273

			// file records ?

   274

			if (item->type == FIOS_TYPE_FILE) {

   275

				if (strcmp(argv[1], item->name) == 0) {

   276

					char buffer[10];

   277

					sprintf(buffer, "%d", pos);

   278

					IConsoleVarSetString(result, buffer);

   279

					return result;

   280

				}

   281

			}

   282

	}

   283

   284

	/* Destroy the file list */

   285

	FiosFreeSavegameList();

   286

   287

	return NULL;

   288

}

   289

   290

/* Change the dir via console */

   291

DEF_CONSOLE_CMD(ConGotoDir)

   292

{

   293

	char *file;

   294

	int no;

   295

   296

	/* We need 1 argument */

   297

	if (argc != 2) {

   298

		IConsolePrint(_iconsole_color_default, "Unknown usage. Usage: goto_dir <dir-no>");

   299

		return NULL;

   300

	}

   301

   302

	no = atoi(argv[1]);

   303

   304

	/* Make the file list */

   305

	BuildFileList();

   306

   307

	/* Check if we are in range */

   308

	if (no != 0 && no <= _fios_num) {

   309

		const FiosItem *item = &_fios_list[no - 1];

   310

		/* Only DIR and PARENT we do allow here */

   311

		if (item->type == FIOS_TYPE_DIR || item->type == FIOS_TYPE_PARENT) {

   312

			/* Goto the map */

   313

			file = FiosBrowseTo(item);

   314

			FiosFreeSavegameList();

   315

			return NULL;

   316

		}

   317

	}

   318

   319

	/* Report error */

   320

	IConsolePrint(_iconsole_color_default, "That number is no directory.");

   321

	FiosFreeSavegameList();

   322

   323

	return NULL;

   324

}

   325

   326

   327

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

   328

// * Network Core Console Commands * //

   329

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

   330

#ifdef ENABLE_NETWORK

   331

   332

DEF_CONSOLE_CMD(ConBan)

   333

{

   334

	NetworkClientInfo *ci;

   335

   336

	if (argc == 2) {

   337

		uint32 index = atoi(argv[1]);

   338

		if (index == NETWORK_SERVER_INDEX) {

   339

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

   340

			return NULL;

   341

		}

   342

		if (index == 0) {

   343

			IConsoleError("Invalid Client-ID");

   344

			return NULL;

   345

		}

   346

   347

		ci = NetworkFindClientInfoFromIndex(index);

   348

   349

		if (ci != NULL) {

   350

			uint i;

   351

			/* Add user to ban-list */

   352

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

   353

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

   354

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

   355

					break;

   356

				}

   357

			}

   358

   359

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

   360

			return NULL;

   361

		} else {

   362

			IConsoleError("Client-ID not found");

   363

			return NULL;

   364

		}

   365

	}

   366

   367

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

   368

   369

	return NULL;

   370

}

   371

   372

DEF_CONSOLE_CMD(ConUnBan)

   373

{

   374

	if (argc == 2) {

   375

		uint i;

   376

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

   377

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

   378

				continue;

   379

   380

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

   381

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

   382

				IConsolePrint(_iconsole_color_default, "IP unbanned.");

   383

				return NULL;

   384

			}

   385

		}

   386

   387

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

   388

   389

		return NULL;

   390

	}

   391

   392

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

   393

   394

	return NULL;

   395

}

   396

   397

DEF_CONSOLE_CMD(ConBanList)

   398

{

   399

	uint i;

   400

   401

	IConsolePrint(_iconsole_color_default, "Banlist: ");

   402

   403

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

   404

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

   405

			continue;

   406

   407

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

   408

	}

   409

   410

	return NULL;

   411

}

   412

   413

DEF_CONSOLE_CMD(ConPauseGame)

   414

{

   415

	if (_pause == 0) {

   416

		DoCommandP(0, 1, 0, NULL, CMD_PAUSE);

   417

		IConsolePrint(_iconsole_color_default, "Game paused.");

   418

	} else

   419

		IConsolePrint(_iconsole_color_default, "Game is already paused.");

   420

   421

	return NULL;

   422

}

   423

   424

DEF_CONSOLE_CMD(ConUnPauseGame)

   425

{

   426

	if (_pause != 0) {

   427

		DoCommandP(0, 0, 0, NULL, CMD_PAUSE);

   428

		IConsolePrint(_iconsole_color_default, "Game unpaused.");

   429

	} else

   430

		IConsolePrint(_iconsole_color_default, "Game is already unpaused.");

   431

   432

	return NULL;

   433

}

   434

   435

DEF_CONSOLE_CMD(ConRcon)

   436

{

   437

	if (argc < 3) {

   438

		IConsolePrint(_iconsole_color_default, "Usage: rcon <password> <command>");

   439

		return NULL;

   440

	}

   441

   442

	SEND_COMMAND(PACKET_CLIENT_RCON)(argv[1], argv[2]);

   443

   444

	return NULL;

   445

}

   446

   447

DEF_CONSOLE_CMD(ConStatus)

   448

{

   449

	const char *status;

   450

	int lag;

   451

	const NetworkClientState *cs;

   452

	const NetworkClientInfo *ci;

   453

	FOR_ALL_CLIENTS(cs) {

   454

		lag = NetworkCalculateLag(cs);

   455

		ci = DEREF_CLIENT_INFO(cs);

   456

   457

		switch (cs->status) {

   458

			case STATUS_INACTIVE:

   459

				status = "inactive";

   460

				break;

   461

			case STATUS_AUTH:

   462

				status = "authorized";

   463

				break;

   464

			case STATUS_MAP:

   465

				status = "loading map";

   466

				break;

   467

			case STATUS_DONE_MAP:

   468

				status = "done map";

   469

				break;

   470

			case STATUS_PRE_ACTIVE:

   471

				status = "ready";

   472

				break;

   473

			case STATUS_ACTIVE:

   474

				status = "active";

   475

				break;

   476

			default:

   477

				status = "unknown";

   478

				break;

   479

		}

   480

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

   481

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

   482

	}

   483

   484

	return NULL;

   485

}

   486

   487

DEF_CONSOLE_CMD(ConKick)

   488

{

   489

	NetworkClientInfo *ci;

   490

   491

	if (argc == 2) {