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

/* Save the map to current dir */

   149

static void SaveMap(const char *filename)

   150

{

   151

	char buf[200];

   152

   153

	snprintf(buf, lengthof(buf), "%s%s%s.sav", _path.save_dir, PATHSEP, filename);

   154

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

   155

   156

	if (SaveOrLoad(buf, SL_SAVE) != SL_OK) {

   157

		IConsolePrint(_iconsole_color_error, "SaveMap failed");

   158

	} else

   159

		IConsolePrintF(_iconsole_color_default, "Map sucessfully saved to %s", buf);

   160

}

   161

   162

/* Save the map to a file */

   163

DEF_CONSOLE_CMD(ConSave)

   164

{

   165

	/* We need 1 argument */

   166

	if (argc == 2) {

   167

		/* Save the map */

   168

		SaveMap(argv[1]);

   169

		return NULL;

   170

	}

   171

   172

	/* Give usage */

   173

	IConsolePrint(_iconsole_color_default, "Unknown usage. Usage: save <filename>");

   174

	return NULL;

   175

}

   176

   177

   178

static const FiosItem* GetFiosItem(const char* file)

   179

{

   180

	int i;

   181

   182

	BuildFileList();

   183

   184

	for (i = 0; i < _fios_num; i++) {

   185

		if (strcmp(file, _fios_list[i].name) == 0) break;

   186

	}

   187

   188

	if (i == _fios_num) { /* If no name matches, try to parse it as number */

   189

		char* endptr;

   190

   191

		i = strtol(file, &endptr, 10);

   192

		if (file == endptr || *endptr != '\0') i = -1;

   193

	}

   194

   195

	return IS_INT_INSIDE(i, 0, _fios_num) ? &_fios_list[i] : NULL;

   196

}

   197

   198

   199

DEF_CONSOLE_CMD(ConLoad)

   200

{

   201

	const FiosItem* item;

   202

	const char* file;

   203

   204

	if (argc != 2) {

   205

		IConsolePrint(_iconsole_color_default, "Usage: load <file | number>");

   206

		return NULL;

   207

	}

   208

   209

	file = argv[1];

   210

	item = GetFiosItem(file);

   211

	if (item != NULL) {

   212

		switch (item->type) {

   213

			case FIOS_TYPE_FILE:

   214

			case FIOS_TYPE_OLDFILE:

   215

				_switch_mode = SM_LOAD;

   216

				SetFiosType(item->type);

   217

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

   218

				break;

   219

   220

			default:

   221

				IConsolePrintF(_iconsole_color_error, "%s: Not a map.", file);

   222

				break;

   223

		}

   224

	} else {

   225

		IConsolePrintF(_iconsole_color_error, "%s: No such file or directory.",

   226

			file);

   227

	}

   228

   229

	FiosFreeSavegameList();

   230

	return NULL;

   231

}

   232

   233

   234

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

   235

DEF_CONSOLE_CMD(ConListFiles)

   236

{

   237

	int i;

   238

   239

	BuildFileList();

   240

   241

	for (i = 0; i < _fios_num; i++) {

   242

		const FiosItem* item = &_fios_list[i];

   243

   244

		IConsolePrintF(_iconsole_color_default, "%d) %s",

   245

			i, item->title[0] != '\0' ? item->title : item->name);

   246

	}

   247

   248

	FiosFreeSavegameList();

   249

	return NULL;

   250

}

   251

   252

   253

/* Change the dir via console */

   254

DEF_CONSOLE_CMD(ConChangeDirectory)

   255

{

   256

	const FiosItem* item;

   257

	const char* file;

   258

   259

	if (argc != 2) {

   260

		IConsolePrint(_iconsole_color_default, "Usage: cd <directory | number>");

   261

		return NULL;

   262

	}

   263

   264

	file = argv[1];

   265

	item = GetFiosItem(file);

   266

	if (item != NULL) {

   267

		switch (item->type) {

   268

			case FIOS_TYPE_DIR:

   269

			case FIOS_TYPE_DRIVE:

   270

			case FIOS_TYPE_PARENT:

   271

				FiosBrowseTo(item);

   272

				break;

   273

   274

			default:

   275

				IConsolePrintF(_iconsole_color_error, "%s: Not a directory.", file);

   276

				break;

   277

		}

   278

	} else {

   279

		IConsolePrintF(_iconsole_color_error, "%s: No such file or directory.",

   280

			file);

   281

	}

   282

   283

	FiosFreeSavegameList();

   284

	return NULL;

   285

}

   286

   287

   288

DEF_CONSOLE_CMD(ConPrintWorkingDirectory)

   289

{

   290

	const char* path;

   291

   292

	// XXX Workaround for broken file handling

   293

	FiosGetSavegameList(&_fios_num, SLD_LOAD_GAME);

   294

	FiosFreeSavegameList();

   295

   296

	FiosGetDescText(&path, NULL);

   297

	IConsolePrint(_iconsole_color_default, path);

   298

	return NULL;

   299

}

   300

   301

   302

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

   303

// * Network Core Console Commands * //

   304

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

   305

#ifdef ENABLE_NETWORK

   306

   307

DEF_CONSOLE_CMD(ConBan)

   308

{

   309

	NetworkClientInfo *ci;

   310

   311

	if (argc == 2) {

   312

		uint32 index = atoi(argv[1]);

   313

		if (index == NETWORK_SERVER_INDEX) {

   314

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

   315

			return NULL;

   316

		}

   317

		if (index == 0) {

   318

			IConsoleError("Invalid Client-ID");

   319

			return NULL;

   320

		}

   321

   322

		ci = NetworkFindClientInfoFromIndex(index);

   323

   324

		if (ci != NULL) {

   325

			uint i;

   326

			/* Add user to ban-list */

   327

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

   328

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

   329

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

   330

					break;

   331

				}

   332

			}

   333

   334

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

   335

			return NULL;

   336

		} else {

   337

			IConsoleError("Client-ID not found");

   338

			return NULL;

   339

		}

   340

	}

   341

   342

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

   343

   344

	return NULL;

   345

}

   346

   347

DEF_CONSOLE_CMD(ConUnBan)

   348

{

   349

	if (argc == 2) {

   350

		uint i;

   351

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

   352

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

   353

				continue;

   354

   355

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

   356

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

   357

				IConsolePrint(_iconsole_color_default, "IP unbanned.");

   358

				return NULL;

   359

			}

   360

		}

   361

   362

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

   363

   364

		return NULL;

   365

	}

   366

   367

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

   368

   369

	return NULL;

   370

}

   371

   372

DEF_CONSOLE_CMD(ConBanList)

   373

{

   374

	uint i;

   375

   376

	IConsolePrint(_iconsole_color_default, "Banlist: ");

   377

   378

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

   379

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

   380

			continue;

   381

   382

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

   383

	}

   384

   385

	return NULL;

   386

}

   387

   388

DEF_CONSOLE_CMD(ConPauseGame)

   389

{

   390

	if (_pause == 0) {

   391

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

   392

		IConsolePrint(_iconsole_color_default, "Game paused.");

   393

	} else

   394

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

   395

   396

	return NULL;

   397

}

   398

   399

DEF_CONSOLE_CMD(ConUnPauseGame)

   400

{

   401

	if (_pause != 0) {

   402

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

   403

		IConsolePrint(_iconsole_color_default, "Game unpaused.");

   404

	} else

   405

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

   406

   407

	return NULL;

   408

}

   409

   410

DEF_CONSOLE_CMD(ConRcon)

   411

{

   412

	if (argc < 3) {

   413

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

   414

		return NULL;

   415

	}

   416

   417

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

   418

   419

	return NULL;

   420

}

   421

   422

DEF_CONSOLE_CMD(ConStatus)

   423

{

   424

	const char *status;

   425

	int lag;

   426

	const NetworkClientState *cs;

   427

	const NetworkClientInfo *ci;

   428

	FOR_ALL_CLIENTS(cs) {

   429

		lag = NetworkCalculateLag(cs);

   430

		ci = DEREF_CLIENT_INFO(cs);

   431

   432

		switch (cs->status) {

   433

			case STATUS_INACTIVE:

   434

				status = "inactive";

   435

				break;

   436

			case STATUS_AUTH:

   437

				status = "authorized";

   438

				break;

   442

			case STATUS_MAP:

   443

				status = "loading map";

   444

				break;

   445

			case STATUS_DONE_MAP:

   446

				status = "done map";

   447

				break;

   448

			case STATUS_PRE_ACTIVE:

   449

				status = "ready";

   450

				break;

   451

			case STATUS_ACTIVE:

   452

				status = "active";

   453

				break;

   454

			default:

   455

				status = "unknown";

   456

				break;

   457

		}

   458

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

   459

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

   460

	}

   461

   462

	return NULL;

   463

}

   464

   465

DEF_CONSOLE_CMD(ConKick)

   466

{

   467

	NetworkClientInfo *ci;

   468

   469

	if (argc == 2) {

   470

		uint32 index = atoi(argv[1]);

   471

		if (index == NETWORK_SERVER_INDEX) {

   472

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

   473

			return NULL;

   474

		}

   475

		if (index == 0) {

   476

			IConsoleError("Invalid Client-ID");

   477

			return NULL;

   478

		}

   479

   480

		ci = NetworkFindClientInfoFromIndex(index);

   481

   482

		if (ci != NULL) {

   483

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

   484

			return NULL;

   485

		} else {

   486

			IConsoleError("Client-ID not found");

   487

			return NULL;

   488

		}

   489

	}