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

   148

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

   149

static void LoadMap(uint no)

   150

{

   151

	/* Build file list */

   152

	BuildFileList();

   153

   154

	/* Check if in range */

   155

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

   156

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

   157

   158

		if (item->type == FIOS_TYPE_FILE) {

   159

			/* Load the file */

   160

			_switch_mode = SM_LOAD;

   161

			SetFiosType(item->type);

   162

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

   163

   164

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

   165

		} else

   166

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

   167

   168

	} else /* Show usages */

   169

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

   170

   171

	/* Free the file-list */

   172

	FiosFreeSavegameList();

   173

}

   174

   175

/* Load a file from a map */

   176

DEF_CONSOLE_CMD(ConLoad)

   177

{

   178

	/* We need 1 argument */

   179

	if (argc == 2) {

   180

		/* Load the map */

   181

		LoadMap(atoi(argv[1]));

   182

		return NULL;

   183

	}

   184

   185

	/* Give usage */

   186

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

   187

	return NULL;

   188

}

   189

   190

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

   191

DEF_CONSOLE_CMD(ConListFiles)

   192

{

   193

	const FiosItem *item;

   194

	int pos = 0;

   195

   196

	/* Build the file-list */

   197

	BuildFileList();

   198

   199

	/* As long as we have files */

   200

	while (pos < _fios_num) {

   201

		item = _fios_list + pos;

   202

		pos++;

   203

		/* Show them */

   204

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

   205

	}

   206

   207

	/* Destroy the file list */

   208

	FiosFreeSavegameList();

   209

   210

	return NULL;

   211

}

   212

   213

/* Get an Specific file */

   214

DEF_CONSOLE_CMD(ConScanFiles)

   215

{

   216

	const FiosItem *item;

   217

	int pos = 0;

   218

	_iconsole_var* result;

   219

   220

   221

	result = IConsoleVarAlloc(ICONSOLE_VAR_STRING);

   222

   223

	if (argc <= 1) {

   224

		IConsoleVarSetString(result, "0");

   225

		return result; // return an zero

   226

	}

   227

   228

	/* Build the file-list */

   229

	BuildFileList();

   230

   231

	/* As long as we have files */

   232

	while (pos < _fios_num) {

   233

		item = _fios_list + pos;

   234

		pos++;

   235

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

   236

			if (item->type == FIOS_TYPE_PARENT) {

   237

				// huh we are searching for the parent directory

   238

				char buffer[10];

   239

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

   240

				IConsoleVarSetString(result, buffer);

   241

				return result;

   242

			}

   243

		} else

   244

			// file records ?

   245

			if (item->type == FIOS_TYPE_FILE) {

   246

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

   247

					char buffer[10];

   248

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

   249

					IConsoleVarSetString(result, buffer);

   250

					return result;

   251

				}

   252

			}

   253

	}

   254

   255

	/* Destroy the file list */

   256

	FiosFreeSavegameList();

   257

   258

	return NULL;

   259

}

   260

   261

/* Change the dir via console */

   262

DEF_CONSOLE_CMD(ConGotoDir)

   263

{

   264

	char *file;

   265

	int no;

   266

   267

	/* We need 1 argument */

   268

	if (argc != 2) {

   269

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

   270

		return NULL;

   271

	}

   272

   273

	no = atoi(argv[1]);

   274

   275

	/* Make the file list */

   276

	BuildFileList();

   277

   278

	/* Check if we are in range */

   279

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

   280

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

   281

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

   282

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

   283

			/* Goto the map */

   284

			file = FiosBrowseTo(item);

   285

			FiosFreeSavegameList();

   286

			return NULL;

   287

		}

   288

	}

   289

   290

	/* Report error */

   291

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

   292

	FiosFreeSavegameList();

   293

   294

	return NULL;

   295

}

   296

   297

   298

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

   299

// * Network Core Console Commands * //

   300

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

   301

#ifdef ENABLE_NETWORK

   302

   303

DEF_CONSOLE_CMD(ConBan)

   304

{

   305

	NetworkClientInfo *ci;

   306

   307

	if (argc == 2) {

   308

		uint32 index = atoi(argv[1]);

   309

		if (index == NETWORK_SERVER_INDEX) {

   310

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

   311

			return NULL;

   312

		}

   313

		if (index == 0) {

   314

			IConsoleError("Invalid Client-ID");

   315

			return NULL;

   316

		}

   317

   318

		ci = NetworkFindClientInfoFromIndex(index);

   319

   320

		if (ci != NULL) {

   321

			uint i;

   322

			/* Add user to ban-list */

   323

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

   324

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

   325

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

   326

					break;

   327

				}

   328

			}

   329

   330

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

   331

			return NULL;

   332

		} else {

   333

			IConsoleError("Client-ID not found");

   334

			return NULL;

   335

		}

   336

	}

   337

   338

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

   339

   340

	return NULL;

   341

}

   342

   343

DEF_CONSOLE_CMD(ConUnBan)

   344

{

   345

	if (argc == 2) {

   346

		uint i;

   347

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

   348

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

   349

				continue;

   350

   351

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

   352

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

   353

				IConsolePrint(_iconsole_color_default, "IP unbanned.");

   354

				return NULL;

   355

			}

   356

		}

   357

   358

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

   359

   360

		return NULL;

   361

	}

   362

   363

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

   364

   365

	return NULL;

   366

}

   367

   368

DEF_CONSOLE_CMD(ConBanList)

   369

{

   370

	uint i;

   371

   372

	IConsolePrint(_iconsole_color_default, "Banlist: ");

   373

   374

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

   375

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

   376

			continue;

   377

   378

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

   379

	}

   380

   381

	return NULL;

   382

}

   383

   384

DEF_CONSOLE_CMD(ConPauseGame)

   385

{

   386

	if (_pause == 0) {

   387

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

   388

		IConsolePrint(_iconsole_color_default, "Game paused.");

   389

	} else

   390

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

   391

   392

	return NULL;

   393

}

   394

   395

DEF_CONSOLE_CMD(ConUnPauseGame)

   396

{

   397

	if (_pause != 0) {

   398

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

   399

		IConsolePrint(_iconsole_color_default, "Game unpaused.");

   400

	} else

   401

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

   402

   403

	return NULL;

   404

}

   405

   406

DEF_CONSOLE_CMD(ConRcon)

   407

{

   408

	if (argc < 3) {

   409

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

   410

		return NULL;

   411

	}

   412

   413

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

   414

   415

	return NULL;

   416

}

   417

   418

DEF_CONSOLE_CMD(ConStatus)

   419

{

   420

	const char *status;

   421

	int lag;

   422

	const NetworkClientState *cs;

   423

	const NetworkClientInfo *ci;

   424

	FOR_ALL_CLIENTS(cs) {

   425

		lag = NetworkCalculateLag(cs);

   426

		ci = DEREF_CLIENT_INFO(cs);

   427

   428

		switch (cs->status) {

   429

			case STATUS_INACTIVE:

   430

				status = "inactive";

   431

				break;

   432

			case STATUS_AUTH:

   433

				status = "authorized";

   434

				break;

   435

			case STATUS_MAP:

   436

				status = "loading map";

   437

				break;

   438

			case STATUS_DONE_MAP:

   439

				status = "done map";

   440

				break;

   441

			case STATUS_PRE_ACTIVE:

   442

				status = "ready";

   443

				break;

   444

			case STATUS_ACTIVE:

   445

				status = "active";

   446

				break;

   447

			default:

   448

				status = "unknown";

   449

				break;

   450

		}

   451

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

   452

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

   453

	}

   454

   455

	return NULL;

   456

}

   457

   458

DEF_CONSOLE_CMD(ConKick)

   459

{

   460

	NetworkClientInfo *ci;

   461

   462

	if (argc == 2) {

   463

		uint32 index = atoi(argv[1]);

   464

		if (index == NETWORK_SERVER_INDEX) {

   465

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

   466

			return NULL;

   467

		}

   468

		if (index == 0) {

   469

			IConsoleError("Invalid Client-ID");

   470

			return NULL;

   471

		}

   472

   473

		ci = NetworkFindClientInfoFromIndex(index);

   474

   475

		if (ci != NULL) {

   476

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

   477

			return NULL;

   478

		} else {

   479

			IConsoleError("Client-ID not found");

   480

			return NULL;

   481

		}

   482

	}

   483

   484

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

   485

   486

	return NULL;

   487

}

   488

   489

DEF_CONSOLE_CMD(ConResetCompany)

   490

{

   491

	Player *p;