1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "console.h"

     4

#include "debug.h"

     5

#include "engine.h"

     6

#include "functions.h"

     7

#include "string.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

// ** scriptfile handling ** //

    18

static FILE *_script_file;

    19

static bool _script_running;

    20

    21

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

    22

#define DEF_CONSOLE_CMD(function) static bool function(byte argc, char *argv[])

    23

#define DEF_CONSOLE_HOOK(function) static bool function(void)

    24

    25

    26

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

    27

/* variable and command hooks   */

    28

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

    29

    30

#ifdef ENABLE_NETWORK

    31

    32

static inline bool NetworkAvailable(void)

    33

{

    34

	if (!_network_available) {

    35

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

    36

		return false;

    37

	}

    38

	return true;

    39

}

    40

    41

DEF_CONSOLE_HOOK(ConHookServerOnly)

    42

{

    43

	if (!NetworkAvailable()) return false;

    44

    45

	if (!_network_server) {

    46

		IConsoleError("This command/variable is only available to a network server.");

    47

		return false;

    48

	}

    49

	return true;

    50

}

    51

    52

DEF_CONSOLE_HOOK(ConHookClientOnly)

    53

{

    54

	if (!NetworkAvailable()) return false;

    55

    56

	if (_network_server) {

    57

		IConsoleError("This command/variable is not available to a network server.");

    58

		return false;

    59

	}

    60

	return true;

    61

}

    62

    63

DEF_CONSOLE_HOOK(ConHookNeedNetwork)

    64

{

    65

	if (!NetworkAvailable()) return false;

    66

    67

	if (!_networking) {

    68

		IConsoleError("Not connected. This command/variable is only available in multiplayer.");

    69

		return false;

    70

	}

    71

	return true;

    72

}

    73

    74

DEF_CONSOLE_HOOK(ConHookNoNetwork)

    75

{

    76

	if (_networking) {

    77

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

    78

		return false;

    79

	}

    80

	return true;

    81

}

    82

    83

#endif /* ENABLE_NETWORK */

    84

    85

static void IConsoleHelp(const char *str)

    86

{

    87

	IConsolePrintF(_icolour_warn, "- %s", str);

    88

}

    89

    90

DEF_CONSOLE_CMD(ConResetEngines)

    91

{

    92

	if (argc == 0) {

    93

		IConsoleHelp("Reset status data of all engines. This might solve some issues with 'lost' engines. Usage: 'resetengines'");

    94

		return true;

    95

	}

    96

    97

	StartupEngines();

    98

	return true;

    99

}

   100

   101

#ifdef _DEBUG

   102

DEF_CONSOLE_CMD(ConResetTile)

   103

{

   104

	if (argc == 0) {

   105

		IConsoleHelp("Reset a tile to bare land. Usage: 'resettile <tile>'");

   106

		IConsoleHelp("Tile can be either decimal (34161) or hexadecimal (0x4a5B)");

   107

		return true;

   108

	}

   109

   110

	if (argc == 2) {

   111

		uint32 result;

   112

		if (GetArgumentInteger(&result, argv[1])) {

   113

			DoClearSquare((TileIndex)result);

   114

			return true;

   115

		}

   116

	}

   117

   118

	return false;

   119

}

   120

#endif /* _DEBUG */

   121

   122

DEF_CONSOLE_CMD(ConScrollToTile)

   123

{

   124

	if (argc == 0) {

   125

		IConsoleHelp("Center the screen on a given tile. Usage: 'scrollto <tile>'");

   126

		IConsoleHelp("Tile can be either decimal (34161) or hexadecimal (0x4a5B)");

   127

		return true;

   128

	}

   129

   130

	if (argc == 2) {

   131

		uint32 result;

   132

		if (GetArgumentInteger(&result, argv[1])) {

   133

			ScrollMainWindowToTile((TileIndex)result);

   134

			return true;

   135

		}

   136

	}

   137

   138

	return false;

   139

}

   140

   141

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

   142

extern void BuildFileList(void);

   143

extern void SetFiosType(const byte fiostype);

   144

   145

/* Save the map to a file */

   146

DEF_CONSOLE_CMD(ConSave)

   147

{

   148

	if (argc == 0) {

   149

		IConsoleHelp("Save the current game. Usage: 'save <filename>'");

   150

		return true;

   151

	}

   152

   153

	if (argc == 2) {

   154

		char buf[200];

   155

   156

		snprintf(buf, lengthof(buf), "%s%s%s.sav", _path.save_dir, PATHSEP, argv[1]);

   157

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

   158

   159

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

   160

			IConsolePrint(_icolour_err, "SaveMap failed");

   161

		} else

   162

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

   163

		return true;

   164

	}

   165

   166

	return false;

   167

}

   168

   169

static const FiosItem* GetFiosItem(const char* file)

   170

{

   171

	int i;

   172

   173

	_saveload_mode = SLD_LOAD_GAME;

   174

	BuildFileList();

   175

   176

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

   177

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

   178

	}

   179

   180

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

   181

		char* endptr;

   182

   183

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

   184

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

   185

	}

   186

   187

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

   188

}

   189

   190

   191

DEF_CONSOLE_CMD(ConLoad)

   192

{

   193

	const FiosItem *item;

   194

	const char *file;

   195

   196

	if (argc == 0) {

   197

		IConsoleHelp("Load a game by name or index. Usage: 'load <file \\ number>'");

   198

		return true;

   199

	}

   200

   201

	if (argc != 2) return false;

   202

   203

	file = argv[1];

   204

	item = GetFiosItem(file);

   205

	if (item != NULL) {

   206

		switch (item->type) {

   207

			case FIOS_TYPE_FILE: case FIOS_TYPE_OLDFILE:

   208

				_switch_mode = SM_LOAD;

   209

				SetFiosType(item->type);

   210

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

   211

				break;

   212

			default: IConsolePrintF(_icolour_err, "%s: Not a savegame.", file);

   213

		}

   214

	} else

   215

		IConsolePrintF(_icolour_err, "%s: No such file or directory.", file);

   216

   217

	FiosFreeSavegameList();

   218

	return true;

   219

}

   220

   221

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

   222

DEF_CONSOLE_CMD(ConListFiles)

   223

{

   224

	int i;

   225

   226

	if (argc == 0) {

   227

		IConsoleHelp("List all loadable savegames and directories in the current dir via console. Usage: 'ls \\ dir'");

   228

		return true;

   229

	}

   230

   231

	BuildFileList();

   232

   233

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

   234

		const FiosItem *item = &_fios_list[i];

   235

		IConsolePrintF(_icolour_def, "%d) %s", i, (item->title[0] != '\0') ? item->title : item->name);

   236

	}

   237

   238

	FiosFreeSavegameList();

   239

	return true;

   240

}

   241

   242

/* Change the dir via console */

   243

DEF_CONSOLE_CMD(ConChangeDirectory)

   244

{

   245

	const FiosItem *item;

   246

	const char *file;

   247

   248

	if (argc == 0) {

   249

		IConsoleHelp("Change the dir via console. Usage: 'cd <directory \\ number>'");

   250

		return true;

   251

	}

   252

   253

	if (argc != 2) return false;

   254

   255

	file = argv[1];

   256

	item = GetFiosItem(file);

   257

	if (item != NULL) {

   258

		switch (item->type) {

   259

			case FIOS_TYPE_DIR: case FIOS_TYPE_DRIVE: case FIOS_TYPE_PARENT:

   260

				FiosBrowseTo(item);

   261

				break;

   262

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

   263

		}

   264

	} else

   265

		IConsolePrintF(_icolour_err, "%s: No such file or directory.", file);

   266

   267

	FiosFreeSavegameList();

   268

	return true;

   269

}

   270

   271

DEF_CONSOLE_CMD(ConPrintWorkingDirectory)

   272

{

   273

	const char *path;

   274

   275

	if (argc == 0) {

   276

		IConsoleHelp("Print out the current working directory. Usage: 'pwd'");

   277

		return true;

   278

	}

   279

   280

	// XXX - Workaround for broken file handling

   281

	FiosGetSavegameList(&_fios_num, SLD_LOAD_GAME);

   282

	FiosFreeSavegameList();

   283

   284

	FiosGetDescText(&path, NULL);

   285

	IConsolePrint(_icolour_def, path);

   286

	return true;

   287

}

   288

   289

DEF_CONSOLE_CMD(ConClearBuffer)

   290

{

   291

	if (argc == 0) {

   292

		IConsoleHelp("Clear the console buffer. Usage: 'clear'");

   293

		return true;

   294

	}

   295

   296

	IConsoleClearBuffer();

   297

	InvalidateWindow(WC_CONSOLE, 0);

   298

	return true;

   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

	uint32 index;

   311

   312

	if (argc == 0) {

   313

		IConsoleHelp("Ban a player from a network game. Usage: 'ban <client-id>'");

   314

		IConsoleHelp("For client-id's, see the command 'clients'");

   315

		return true;

   316

	}

   317

   318

	if (argc != 2) return false;

   319

   320

	index = atoi(argv[1]);

   321

   322

	if (index == NETWORK_SERVER_INDEX) {

   323

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

   324

		return true;

   325

	}

   326

	if (index == 0) {

   327

		IConsoleError("Invalid Client-ID");

   328

		return true;

   329

	}

   330

   331

	ci = NetworkFindClientInfoFromIndex(index);

   332

   333

	if (ci != NULL) {

   334

		uint i;

   335

		/* Add user to ban-list */

   336

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

   337

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

   338

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

   339

				break;

   340

			}

   341

		}

   342

   343

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

   344

	} else

   345

		IConsoleError("Client-ID not found");

   346

   347

	return true;

   348

}

   349

   350

DEF_CONSOLE_CMD(ConUnBan)

   351

{

   352

	uint i;

   353

   354

	if (argc == 0) {

   355

		IConsoleHelp("Unban a player from a network game. Usage: 'unban <ip>'");

   356

		return true;

   357

	}

   358

   359

	if (argc != 2) return false;

   360

   361

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

   362

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

   363

			continue;

   364

   365

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

   366

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

   367

			IConsolePrint(_icolour_def, "IP unbanned.");

   368

			return true;

   369

		}

   370

	}

   371

   372

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

   373

	return true;

   374

}

   375

   376

DEF_CONSOLE_CMD(ConBanList)

   377

{

   378

	uint i;

   379

   380

	if (argc == 0) {

   381

		IConsoleHelp("List the IP's of banned clients: Usage 'banlist'");

   382

		return true;

   383

	}

   384

   385

	IConsolePrint(_icolour_def, "Banlist: ");

   386

   387

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

   388

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

   389

			continue;

   390

   391

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

   392

	}

   393

   394

	return true;

   395

}

   396

   397

DEF_CONSOLE_CMD(ConPauseGame)

   398

{

   399

	if (argc == 0) {

   400

		IConsoleHelp("Pause a network game. Usage: 'pause'");

   401

		return true;

   402

	}

   403

   404

	if (_pause == 0) {

   405

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

   406

		IConsolePrint(_icolour_def, "Game paused.");

   407

	} else

   408

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

   409

   410

	return true;

   411

}

   412

   413

DEF_CONSOLE_CMD(ConUnPauseGame)

   414

{

   415

	if (argc == 0) {

   416

		IConsoleHelp("Unpause a network game. Usage: 'unpause'");

   417

		return true;

   418

	}

   419

   420

	if (_pause != 0) {

   421

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

   422

		IConsolePrint(_icolour_def, "Game unpaused.");

   423

	} else

   424

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

   425

   426

	return true;

   427

}

   428

   429

DEF_CONSOLE_CMD(ConRcon)

   430

{

   431

	if (argc == 0) {

   432

		IConsoleHelp("Remote control the server from another client. Usage: 'rcon <password> <command>'");

   433

		IConsoleHelp("Remember to enclose the command in quotes, otherwise only the first parameter is sent");

   434

		return true;

   435

	}

   436

   437

	if (argc < 3) return false;

   438

   439

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

   440

	return true;

   441

}

   442

   443

DEF_CONSOLE_CMD(ConStatus)

   444

{

   445

	static const char *stat_str[] = {"inactive", "authorized", "waiting", "loading map", "map done", "ready", "active"};

   446

	const char *status;

   447

	const NetworkClientState *cs;

   448

   449

	if (argc == 0) {

   450

		IConsoleHelp("List the status of all clients connected to the server: Usage 'status'");

   451

		return true;

   452

	}

   453

   454

	FOR_ALL_CLIENTS(cs) {

   455

		int lag = NetworkCalculateLag(cs);

   456

		const NetworkClientInfo *ci = DEREF_CLIENT_INFO(cs);

   457

   458

		status = (cs->status <= STATUS_ACTIVE) ? stat_str[cs->status] : "unknown";

   459

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

   460

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

   461

	}

   462

   463

	return true;

   464

}

   465

   466

DEF_CONSOLE_CMD(ConKick)

   467

{

   468

	NetworkClientInfo *ci;

   469

	uint32 index;

   470

   471

	if (argc == 0) {

   472

		IConsoleHelp("Kick a player from a network game. Usage: 'kick <client-id>'");

   473

		IConsoleHelp("For client-id's, see the command 'clients'");

   474

		return true;

   475

	}

   476

   477

	if (argc != 2) return false;

   478

   479

	index = atoi(argv[1]);

   480

	if (index == NETWORK_SERVER_INDEX) {

   481

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

   482

		return true;

   483

	}