1

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

     2

#include "stdafx.h"

     3

#include "ttd.h"

     4

#include "console.h"

     6

#include "engine.h"

     7

#include "functions.h"

     8

#include "variables.h"

     9

#include "network_data.h"

    10

#include "network_client.h"

    11

#include "network_server.h"

    12

#include "network_udp.h"

    13

#include "command.h"

    14

#include "settings.h"

    15

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

    16

    17

    18

// ** scriptfile handling ** //

    19

static FILE * _script_file;

    20

static bool _script_running;

    21

    22

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

    23

    24

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

    25

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

    26

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

    27

    28

    29

// ** supporting functions ** //

    30

    31

static uint32 GetArgumentInteger(const char* arg)

    32

{

    33

	uint32 result;

    34

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

    35

    36

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

    37

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

    38

    39

	return result;

    40

}

    41

    42

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

    43

/* variable and command hooks   */

    44

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

    45

    46

#ifdef ENABLE_NETWORK

    47

    48

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetwork)

    49

{

    50

	if (_networking) {

    51

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

    52

		return false;

    53

	}

    54

	return true;

    55

}

    56

    57

DEF_CONSOLE_VAR_HOOK(ConVarHookNoNetClient)

    58

{

    59

	if (!_network_available) {

    60

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

    61

		return false;

    62

	}

    63

	if (!_network_server) {

    64

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

    65

		return false;

    66

	}

    67

	return true;

    68

}

    69

    70

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetClient)

    71

{

    72

	if (!_network_available) {

    73

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

    74

		return false;

    75

	}

    76

	if (!_network_server) {

    77

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

    78

		return false;

    79

	}

    80

	return true;

    81

}

    82

    83

DEF_CONSOLE_CMD_HOOK(ConCmdHookNoNetServer)

    84

{

    85

	if (!_network_available) {

    86

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

    87

		return false;

    88

	}

    89

	if (_network_server) {

    90

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

    91

		return false;

    92

	}

    93

	return true;

    94

}

    95

    96

DEF_CONSOLE_CMD_HOOK(ConCmdHookNeedNetwork)

    97

{

    98

	if (!_network_available) {

    99

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

   100

		return false;

   101

	}

   102

	if (!_networking) {

   103

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

   104

		return false;

   105

	}

   106

	return true;

   107

}

   108

   109

#endif /* ENABLE_NETWORK */

   110

   111

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

   112

/* reset commands               */

   113

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

   114

   115

DEF_CONSOLE_CMD(ConResetEngines)

   116

{

   117

	StartupEngines();

   118

	return 0;

   119

}

   120

   121

#ifdef _DEBUG

   122

DEF_CONSOLE_CMD(ConResetTile)

   123

{

   124

	if (argc == 2) {

   125

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

   126

		DoClearSquare(tile);

   127

	}

   128

   129

	return 0;

   130

}

   131

#endif

   132

   133

DEF_CONSOLE_CMD(ConScrollToTile)

   134

{

   135

	if (argc == 2) {

   136

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

   137

		ScrollMainWindowToTile(tile);

   138

	}

   139

   140

	return 0;

   141

}

   142

   143

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

   144

extern void BuildFileList(void);

   145

extern void SetFiosType(const byte fiostype);

   146

   147

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

   148

static void LoadMap(uint no)

   149

{

   150

	/* Build file list */

   151

	BuildFileList();

   152

   153

	/* Check if in range */

   154

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

   155

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

   156

   157

		if (item->type == FIOS_TYPE_FILE) {

   158

			/* Load the file */

   159

			_switch_mode = SM_LOAD;

   160

			SetFiosType(item->type);

   161

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

   162

   163

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

   164

		} else

   165

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

   166

   167

	} else /* Show usages */

   168

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

   169

   170

	/* Free the file-list */

   171

	FiosFreeSavegameList();

   172

}

   173

   174

/* Load a file from a map */

   175

DEF_CONSOLE_CMD(ConLoad)

   176

{

   177

	/* We need 1 argument */

   178

	if (argc == 2) {

   179

		/* Load the map */

   180

		LoadMap(atoi(argv[1]));

   181

		return NULL;

   182

	}

   183

   184

	/* Give usage */

   185

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

   186

	return NULL;

   187

}

   188

   189

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

   190

DEF_CONSOLE_CMD(ConListFiles)

   191

{

   192

	const FiosItem *item;

   193

	int pos = 0;

   194

   195

	/* Build the file-list */

   196

	BuildFileList();

   197

   198

	/* As long as we have files */

   199

	while (pos < _fios_num) {

   200

		item = _fios_list + pos;

   201

		pos++;

   202

		/* Show them */

   203

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

   204

	}

   205

   206

	/* Destroy the file list */

   207

	FiosFreeSavegameList();

   208

   209

	return NULL;

   210

}

   211

   212

/* Get an Specific file */

   213

DEF_CONSOLE_CMD(ConScanFiles)

   214

{

   215

	const FiosItem *item;

   216

	int pos = 0;

   217

	_iconsole_var* result;

   218

   219

   220

	result = IConsoleVarAlloc(ICONSOLE_VAR_STRING);

   221

   222

	if (argc <= 1) {

   223

		IConsoleVarSetString(result, "0");

   224

		return result; // return an zero

   225

	}

   226

   227

	/* Build the file-list */

   228

	BuildFileList();

   229

   230

	/* As long as we have files */

   231

	while (pos < _fios_num) {

   232

		item = _fios_list + pos;

   233

		pos++;

   234

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

   235

			if (item->type == FIOS_TYPE_PARENT) {

   236

				// huh we are searching for the parent directory

   237

				char buffer[10];

   238

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

   239

				IConsoleVarSetString(result, buffer);

   240

				return result;

   241

			}

   242

		} else

   243

			// file records ?

   244

			if (item->type == FIOS_TYPE_FILE) {

   245

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

   246

					char buffer[10];

   247

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

   248

					IConsoleVarSetString(result, buffer);

   249

					return result;

   250

				}

   251

			}

   252

	}

   253

   254

	/* Destroy the file list */

   255

	FiosFreeSavegameList();

   256

   257

	return NULL;

   258

}

   259

   260

/* Change the dir via console */

   261

DEF_CONSOLE_CMD(ConGotoDir)

   262

{

   263

	char *file;

   264

	int no;

   265

   266

	/* We need 1 argument */

   267

	if (argc != 2) {

   268

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

   269

		return NULL;

   270

	}

   271

   272

	no = atoi(argv[1]);

   273

   274

	/* Make the file list */

   275

	BuildFileList();

   276

   277

	/* Check if we are in range */

   278

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

   279

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

   280

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

   281

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

   282

			/* Goto the map */

   283

			file = FiosBrowseTo(item);

   284

			FiosFreeSavegameList();

   285

			return NULL;

   286

		}

   287

	}

   288

   289

	/* Report error */

   290

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

   291

	FiosFreeSavegameList();

   292

   293

	return NULL;

   294

}

   295

   296

   297

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

   298

// * Network Core Console Commands * //

   299

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

   300

#ifdef ENABLE_NETWORK

   301

   302

DEF_CONSOLE_CMD(ConBan)

   303

{

   304

	NetworkClientInfo *ci;

   305

   306

	if (argc == 2) {

   307

		uint32 index = atoi(argv[1]);

   308

		if (index == NETWORK_SERVER_INDEX) {

   309

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

   310

			return NULL;

   311

		}

   312

		if (index == 0) {

   313

			IConsoleError("Invalid Client-ID");

   314

			return NULL;

   315

		}

   316

   317

		ci = NetworkFindClientInfoFromIndex(index);

   318

   319

		if (ci != NULL) {

   320

			uint i;

   321

			/* Add user to ban-list */

   322

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

   323

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

   324

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

   325

					break;

   326

				}

   327

			}

   328

   329

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

   330

			return NULL;

   331

		} else {

   332

			IConsoleError("Client-ID not found");

   333

			return NULL;

   334

		}

   335

	}

   336

   337

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

   338

   339

	return NULL;

   340

}

   341

   342

DEF_CONSOLE_CMD(ConUnBan)

   343

{

   344

	if (argc == 2) {

   345

		uint i;

   346

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

   347

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

   348

				continue;

   349

   350

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

   351

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

   352

				IConsolePrint(_iconsole_color_default, "IP unbanned.");

   353

				return NULL;

   354

			}

   355

		}

   356

   357

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

   358

   359

		return NULL;

   360

	}

   361

   362

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

   363

   364

	return NULL;

   365

}

   366

   367

DEF_CONSOLE_CMD(ConBanList)

   368

{

   369

	uint i;

   370

   371

	IConsolePrint(_iconsole_color_default, "Banlist: ");

   372

   373

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

   374

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

   375

			continue;

   376

   377

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

   378

	}

   379

   380

	return NULL;

   381

}

   382

   383

DEF_CONSOLE_CMD(ConPauseGame)

   384

{

   385

	if (_pause == 0) {

   386

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

   387

		IConsolePrint(_iconsole_color_default, "Game paused.");

   388

	} else

   389

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

   390

   391

	return NULL;

   392

}

   393

   394

DEF_CONSOLE_CMD(ConUnPauseGame)

   395

{

   396

	if (_pause != 0) {

   397

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

   398

		IConsolePrint(_iconsole_color_default, "Game unpaused.");

   399

	} else

   400

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

   401

   402

	return NULL;

   403

}

   404

   405

DEF_CONSOLE_CMD(ConRcon)

   406

{

   407

	if (argc < 3) {

   408

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

   409

		return NULL;

   410

	}

   411

   412

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

   413

   414

	return NULL;

   415

}

   416

   417

DEF_CONSOLE_CMD(ConStatus)

   418

{

   419

	const char *status;

   420

	int lag;

   421

	const NetworkClientState *cs;

   422

	const NetworkClientInfo *ci;

   423

	FOR_ALL_CLIENTS(cs) {

   424

		lag = NetworkCalculateLag(cs);

   425

		ci = DEREF_CLIENT_INFO(cs);

   426

   427

		switch (cs->status) {

   428

			case STATUS_INACTIVE:

   429

				status = "inactive";

   430

				break;

   431

			case STATUS_AUTH:

   432

				status = "authorized";

   433

				break;

   434

			case STATUS_MAP:

   435

				status = "loading map";

   436

				break;

   437

			case STATUS_DONE_MAP:

   438

				status = "done map";

   439

				break;

   440

			case STATUS_PRE_ACTIVE:

   441

				status = "ready";

   442

				break;

   443

			case STATUS_ACTIVE:

   444

				status = "active";

   445

				break;

   446

			default:

   447

				status = "unknown";

   448

				break;

   449

		}

   450

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

   451

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

   452

	}

   453

   454

	return NULL;

   455

}

   456

   457

DEF_CONSOLE_CMD(ConKick)

   458

{

   459

	NetworkClientInfo *ci;

   460

   461

	if (argc == 2) {

   462

		uint32 index = atoi(argv[1]);

   463

		if (index == NETWORK_SERVER_INDEX) {

   464

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

   465

			return NULL;

   466

		}

   467

		if (index == 0) {

   468

			IConsoleError("Invalid Client-ID");

   469

			return NULL;

   470

		}

   471

   472

		ci = NetworkFindClientInfoFromIndex(index);

   473

   474

		if (ci != NULL) {

   475

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

   476

			return NULL;

   477

		} else {

   478

			IConsoleError("Client-ID not found");

   479

			return NULL;

   480

		}

   481

	}

   482

   483

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

   484

   485

	return NULL;

   486

}

   487

   488

DEF_CONSOLE_CMD(ConResetCompany)

   489

{

   490

	Player *p;

   491

	NetworkClientState *cs;