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 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 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 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 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(_iconsole_color_warning, "- %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(_iconsole_color_default, "Saving map...");

   158

   159

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

   160

			IConsolePrint(_iconsole_color_error, "SaveMap failed");

   161

		} else

   162

			IConsolePrintF(_iconsole_color_default, "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

	BuildFileList();

   174

   175

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

   176

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

   177

	}

   178

   179

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

   180

		char* endptr;

   181

   182

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

   183

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

   184

	}

   185

   186

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

   187

}

   188

   189

   190

DEF_CONSOLE_CMD(ConLoad)

   191

{

   192

	const FiosItem *item;

   193

	const char *file;

   194

   195

	if (argc == 0) {

   196

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

   197

		return true;

   198

	}

   199

   200

	if (argc != 2) return false;

   201

   202

	file = argv[1];

   203

	item = GetFiosItem(file);

   204

	if (item != NULL) {

   205

		switch (item->type) {

   206

			case FIOS_TYPE_FILE: case FIOS_TYPE_OLDFILE:

   207

				_switch_mode = SM_LOAD;

   208

				SetFiosType(item->type);

   209

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

   210

				break;

   211

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

   212

		}

   213

	} else

   214

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

   215

   216

	FiosFreeSavegameList();

   217

	return true;

   218

}

   219

   220

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

   221

DEF_CONSOLE_CMD(ConListFiles)

   222

{

   223

	int i;

   224

   225

	if (argc == 0) {

   227

		return true;

   228

	}

   229

   231

	BuildFileList();

   232

   233

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

   234

		const FiosItem *item = &_fios_list[i];

   235

		IConsolePrintF(_iconsole_color_default, "%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(_iconsole_color_error, "%s: Not a directory.", file);

   263

		}

   264

	} else

   265

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

   266

   267

	FiosFreeSavegameList();

   268

	return true;

   269

}

   270

   271

   272

DEF_CONSOLE_CMD(ConPrintWorkingDirectory)

   273

{

   274

	const char *path;

   275

   276

	if (argc == 0) {

   277

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

   278

		return true;

   279

	}

   280

   281

	// XXX - Workaround for broken file handling

   282

	FiosGetSavegameList(&_fios_num, SLD_LOAD_GAME);

   283

	FiosFreeSavegameList();

   284

   285

	FiosGetDescText(&path, NULL);

   286

	IConsolePrint(_iconsole_color_default, path);

   287

	return true;

   288

}

   289

   290

   291

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

   292

// * Network Core Console Commands * //

   293

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

   294

#ifdef ENABLE_NETWORK

   295

   296

DEF_CONSOLE_CMD(ConBan)

   297

{

   298

	NetworkClientInfo *ci;

   299

	uint32 index;

   300

   301

	if (argc == 0) {

   302

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

   303

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

   304

		return true;

   305

	}

   306

   307

	if (argc != 2) return false;

   308

   309

	index = atoi(argv[1]);

   310

   311

	if (index == NETWORK_SERVER_INDEX) {

   312

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

   313

		return true;

   314

	}

   315

	if (index == 0) {

   316

		IConsoleError("Invalid Client-ID");

   317

		return true;

   318

	}

   319

   320

	ci = NetworkFindClientInfoFromIndex(index);

   321

   322

	if (ci != NULL) {

   323

		uint i;

   324

		/* Add user to ban-list */

   325

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

   326

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

   327

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

   328

				break;

   329

			}

   330

		}

   331

   332

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

   333

	} else

   334

		IConsoleError("Client-ID not found");

   335

   336

	return true;

   337

}

   338

   339

DEF_CONSOLE_CMD(ConUnBan)

   340

{

   341

	uint i;

   342

   343

	if (argc == 0) {

   344

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

   345

		return true;

   346

	}

   347

   348

	if (argc != 2) return false;

   349

   350

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

   351

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

   352

			continue;

   353

   354

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

   355

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

   356

			IConsolePrint(_iconsole_color_default, "IP unbanned.");

   357

			return true;

   358

		}

   359

	}

   360

   361

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

   362

	return true;

   363

}

   364

   365

DEF_CONSOLE_CMD(ConBanList)

   366

{

   367

	uint i;

   368

   369

	if (argc == 0) {

   370

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

   371

		return true;

   372

	}

   373

   374

	IConsolePrint(_iconsole_color_default, "Banlist: ");

   375

   376

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

   377

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

   378

			continue;

   379

   380

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

   381

	}

   382

   383

	return true;

   384

}

   385

   386

DEF_CONSOLE_CMD(ConPauseGame)

   387

{

   388

	if (argc == 0) {

   389

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

   390

		return true;

   391

	}

   392

   393

	if (_pause == 0) {

   394

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

   395

		IConsolePrint(_iconsole_color_default, "Game paused.");

   396

	} else

   397

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

   398

   399

	return true;

   400

}

   401

   402

DEF_CONSOLE_CMD(ConUnPauseGame)

   403

{

   404

	if (argc == 0) {

   405

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

   406

		return true;

   407

	}

   408

   409

	if (_pause != 0) {

   410

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

   411

		IConsolePrint(_iconsole_color_default, "Game unpaused.");

   412

	} else

   413

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

   414

   415

	return true;

   416

}

   417

   418

DEF_CONSOLE_CMD(ConRcon)

   419

{

   420

	if (argc == 0) {

   421

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

   422

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

   423

		return true;

   424

	}

   425

   426

	if (argc < 3) return false;

   427

   428

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

   429

	return true;

   430

}

   431

   432

DEF_CONSOLE_CMD(ConStatus)

   433

{

   434

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

   435

	const char *status;

   436

	const NetworkClientState *cs;

   437

   438

	if (argc == 0) {

   439

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

   440

		return true;

   441

	}

   442

   443

	FOR_ALL_CLIENTS(cs) {

   444

		int lag = NetworkCalculateLag(cs);

   445

		const NetworkClientInfo *ci = DEREF_CLIENT_INFO(cs);

   446

   447

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

   448

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

   449

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

   450

	}

   451

   452

	return true;

   453

}

   454

   455

DEF_CONSOLE_CMD(ConKick)

   456

{

   457

	NetworkClientInfo *ci;

   458

	uint32 index;

   459

   460

	if (argc == 0) {

   461

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

   462

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

   463

		return true;

   464

	}

   465

   466

	if (argc != 2) return false;

   467

   468

	index = atoi(argv[1]);

   469

	if (index == NETWORK_SERVER_INDEX) {

   470

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

   471

		return true;

   472

	}

   473

	if (index == 0) {

   474

		IConsoleError("Invalid Client-ID");

   475

		return true;

   476

	}

   477

   478

	ci = NetworkFindClientInfoFromIndex(index);

   479

   480

	if (ci != NULL) {

   481

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

   482

	} else

   483

		IConsoleError("Client-ID not found");

   484

   485

	return true;