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

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

    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

   142

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

   143

extern void BuildFileList();

   144

extern void SetFiosType(const byte fiostype);

   145

   146

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

   147

static void LoadMap(uint no)

   148

{

   149

	/* Build file list */

   150

	BuildFileList();

   151

   152

	/* Check if in range */

   153

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

   154

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

   155

   156

		if (item->type == FIOS_TYPE_FILE) {

   157

			/* Load the file */

   158

			_switch_mode = SM_LOAD;

   159

			SetFiosType(item->type);

   160

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

   161

   162

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

   163

		} else

   164

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

   165

   166

	} else /* Show usages */

   167

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

   168

   169

	/* Free the file-list */

   170

	FiosFreeSavegameList();

   171

}

   172

   173

/* Load a file from a map */

   174

DEF_CONSOLE_CMD(ConLoad)

   175

{

   176

	/* We need 1 argument */

   177

	if (argc == 2) {

   178

		/* Load the map */

   179

		LoadMap(atoi(argv[1]));

   180

		return NULL;

   181

	}

   182

   183

	/* Give usage */

   184

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

   185

	return NULL;

   186

}

   187

   188

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

   189

DEF_CONSOLE_CMD(ConListFiles)

   190

{

   191

	const FiosItem *item;

   192

	int pos = 0;

   193

   194

	/* Build the file-list */

   195

	BuildFileList();

   196

   197

	/* As long as we have files */

   198

	while (pos < _fios_num) {

   199

		item = _fios_list + pos;

   200

		pos++;

   201

		/* Show them */

   202

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

   203

	}

   204

   205

	/* Destroy the file list */

   206

	FiosFreeSavegameList();

   207

   208

	return NULL;

   209

}

   210

   211

/* Get an Specific file */

   212

DEF_CONSOLE_CMD(ConScanFiles)

   213

{

   214

	const FiosItem *item;

   215

	int pos = 0;

   216

	_iconsole_var* result;

   217

   218

   219

	result = IConsoleVarAlloc(ICONSOLE_VAR_STRING);

   220

   221

	if (argc <= 1) {

   222

		IConsoleVarSetString(result, "0");

   223

		return result; // return an zero

   224

	}

   225

   226

	/* Build the file-list */

   227

	BuildFileList();

   228

   229

	/* As long as we have files */

   230

	while (pos < _fios_num) {

   231

		item = _fios_list + pos;

   232

		pos++;		

   233

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

   234

			if (item->type == FIOS_TYPE_PARENT) {

   235

				// huh we are searching for the parent directory

   236

				char buffer[10];

   237

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

   238

				IConsoleVarSetString(result, buffer);

   239

				return result;

   240

			}

   241

		} else 

   242

			// file records ?

   243

			if (item->type == FIOS_TYPE_FILE) {

   244

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

   245

					char buffer[10];

   246

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

   247

					IConsoleVarSetString(result, buffer);

   248

					return result;

   249

				}			

   250

			}

   251

	}

   252

   253

	/* Destroy the file list */

   254

	FiosFreeSavegameList();

   255

   256

	return NULL;

   257

}

   258

   259

/* Change the dir via console */

   260

DEF_CONSOLE_CMD(ConGotoDir)

   261

{

   262

	char *file;

   263

	int no;

   264

   265

	/* We need 1 argument */

   266

	if (argc != 2) {

   267

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

   268

		return NULL;

   269

	}

   270

   271

	no = atoi(argv[1]);

   272

   273

	/* Make the file list */

   274

	BuildFileList();

   275

   276

	/* Check if we are in range */

   277

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

   278

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

   279

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

   280

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

   281

			/* Goto the map */

   282

			file = FiosBrowseTo(item);

   283

			FiosFreeSavegameList();

   284

			return NULL;

   285

		}

   286

	}

   287

   288

	/* Report error */

   289

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

   290

	FiosFreeSavegameList();

   291

   292

	return NULL;

   293

}

   294

   295

   296

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

   297

// * Network Core Console Commands * //

   298

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

   299

#ifdef ENABLE_NETWORK

   300

   301

DEF_CONSOLE_CMD(ConBan)

   302

{

   303

	NetworkClientInfo *ci;

   304

   305

	if (argc == 2) {

   306

		uint32 index = atoi(argv[1]);

   307

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

   308

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

   309

			return NULL;

   310

		}

   311

		if (index == 0) {

   312

			IConsoleError("Invalid Client-ID");

   313

			return NULL;

   314

		}

   315

   316

		ci = NetworkFindClientInfoFromIndex(index);

   317

   318

		if (ci != NULL) {

   319

			uint i;

   320

			/* Add user to ban-list */

   321

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

   322

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

   323

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

   324

					break;

   325

				}

   326

			}

   327

   328

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

   329

			return NULL;

   330

		} else {

   331

			IConsoleError("Client-ID not found");

   332

			return NULL;

   333

		}

   334

	}

   335

   336

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

   337

   338

	return NULL;

   339

}

   340

   341

DEF_CONSOLE_CMD(ConUnBan)

   342

{

   343

	if (argc == 2) {

   344

		uint i;

   345

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

   346

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

   347

				continue;

   348

   349

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

   350

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

   351

				IConsolePrint(_iconsole_color_default, "IP unbanned.");

   352

				return NULL;

   353

			}

   354

		}

   355

   356

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

   357

   358

		return NULL;

   359

	}

   360

   361

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

   362

   363

	return NULL;

   364

}

   365

   366

DEF_CONSOLE_CMD(ConBanList)

   367

{

   368

	uint i;

   369

   370

	IConsolePrint(_iconsole_color_default, "Banlist: ");

   371

   372

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

   373

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

   374

			continue;

   375

   376

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

   377

	}

   378

   379

	return NULL;

   380

}

   381

   382

DEF_CONSOLE_CMD(ConStatus)

   383

{

   384

	const char *status;

   385

	int lag;

   386

	const NetworkClientState *cs;

   387

	const NetworkClientInfo *ci;

   388

	FOR_ALL_CLIENTS(cs) {

   389

		lag = NetworkCalculateLag(cs);

   390

		ci = DEREF_CLIENT_INFO(cs);

   391

   392

		switch (cs->status) {

   393

			case STATUS_INACTIVE:

   394

				status = "inactive";

   395

				break;

   396

			case STATUS_AUTH:

   397

				status = "authorized";

   398

				break;

   399

			case STATUS_MAP:

   400

				status = "loading map";

   401

				break;

   402

			case STATUS_DONE_MAP:

   403

				status = "done map";

   404

				break;

   405

			case STATUS_PRE_ACTIVE:

   406

				status = "ready";

   407

				break;

   408

			case STATUS_ACTIVE:

   409

				status = "active";

   410

				break;

   411

			default:

   412

				status = "unknown";

   413

				break;

   414

		}

   415

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

   416

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

   417

	}

   418

   419

	return NULL;

   420

}

   421

   422

DEF_CONSOLE_CMD(ConKick)

   423

{

   424

	NetworkClientInfo *ci;

   425

   426

	if (argc == 2) {

   427

		uint32 index = atoi(argv[1]);

   428

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

   429

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

   430

			return NULL;

   431

		}

   432

		if (index == 0) {

   433

			IConsoleError("Invalid Client-ID");

   434

			return NULL;

   435

		}

   436

   437

		ci = NetworkFindClientInfoFromIndex(index);

   438

   439

		if (ci != NULL) {

   440

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

   441

			return NULL;

   442

		} else {

   443

			IConsoleError("Client-ID not found");

   444

			return NULL;

   445

		}

   446

	}

   447

   448

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

   449

   450

	return NULL;

   451

}

   452

   453

DEF_CONSOLE_CMD(ConResetCompany)

   454

{

   455

	Player *p;

   456

	NetworkClientState *cs;

   457

	NetworkClientInfo *ci;

   458

   459

	if (argc == 2) {

   460

		uint32 index = atoi(argv[1]);

   461

   462

		/* Check valid range */

   463

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

   464

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

   465

			return NULL;

   466

		}

   467

   468

		/* Check if company does exist */

   469

		index--;

   470

		p = DEREF_PLAYER(index);

   471

		if (!p->is_active) {

   472

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

   473

			return NULL;

   474

		}

   475

   476

		if (p->is_ai) {

   477

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

   478

			return NULL;

   479

		}

   480

   481

		/* Check if the company has active players */

   482

		FOR_ALL_CLIENTS(cs) {

   483

			ci = DEREF_CLIENT_INFO(cs);

   484

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

   485

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

   486

				return NULL;

   487

			}

   488

		}

   489

		ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   490

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

   491

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

   492

			return NULL;