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) {

   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

   230

	BuildFileList();

   231

   232

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

   233

		const FiosItem *item = &_fios_list[i];

   234

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

   235

	}

   236

   237

	FiosFreeSavegameList();

   238

	return true;

   239

}

   240

   241

/* Change the dir via console */

   242

DEF_CONSOLE_CMD(ConChangeDirectory)

   243

{

   244

	const FiosItem *item;

   245

	const char *file;

   246

   247

	if (argc == 0) {

   249

		return true;

   250

	}

   251

   252

	if (argc != 2) return false;

   253

   254

	file = argv[1];

   255

	item = GetFiosItem(file);

   256

	if (item != NULL) {

   257

		switch (item->type) {

   258

			case FIOS_TYPE_DIR: case FIOS_TYPE_DRIVE: case FIOS_TYPE_PARENT:

   259

				FiosBrowseTo(item);

   260

				break;

   261

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

   262

		}

   263

	} else

   264

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

   265

   266

	FiosFreeSavegameList();

   267

	return true;

   268

}

   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(_iconsole_color_default, path);

   286

	return true;

   287

}

   288

   289

   290

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

   291

// * Network Core Console Commands * //

   292

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

   293

#ifdef ENABLE_NETWORK

   294

   295

DEF_CONSOLE_CMD(ConBan)

   296

{

   297

	NetworkClientInfo *ci;

   298

	uint32 index;

   299

   300

	if (argc == 0) {

   301

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

   302

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

   303

		return true;

   304

	}

   305

   306

	if (argc != 2) return false;

   307

   308

	index = atoi(argv[1]);

   309

   310

	if (index == NETWORK_SERVER_INDEX) {

   311

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

   312

		return true;

   313

	}

   314

	if (index == 0) {

   315

		IConsoleError("Invalid Client-ID");

   316

		return true;

   317

	}

   318

   319

	ci = NetworkFindClientInfoFromIndex(index);

   320

   321

	if (ci != NULL) {

   322

		uint i;

   323

		/* Add user to ban-list */

   324

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

   325

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

   326

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

   327

				break;

   328

			}

   329

		}

   330

   331

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

   332

	} else

   333

		IConsoleError("Client-ID not found");

   334

   335

	return true;

   336

}

   337

   338

DEF_CONSOLE_CMD(ConUnBan)

   339

{

   340

	uint i;

   341

   342

	if (argc == 0) {

   343

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

   344

		return true;

   345

	}

   346

   347

	if (argc != 2) return false;

   348

   349

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

   350

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

   351

			continue;

   352

   353

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

   354

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

   355

			IConsolePrint(_iconsole_color_default, "IP unbanned.");

   356

			return true;

   357

		}

   358

	}

   359

   360

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

   361

	return true;

   362

}

   363

   364

DEF_CONSOLE_CMD(ConBanList)

   365

{

   366

	uint i;

   367

   368

	if (argc == 0) {

   369

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

   370

		return true;

   371

	}

   372

   373

	IConsolePrint(_iconsole_color_default, "Banlist: ");

   374

   375

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

   376

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

   377

			continue;

   378

   379

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

   380

	}

   381

   382

	return true;

   383

}

   384

   385

DEF_CONSOLE_CMD(ConPauseGame)

   386

{

   387

	if (argc == 0) {

   388

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

   389

		return true;

   390

	}

   391

   392

	if (_pause == 0) {

   393

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

   394

		IConsolePrint(_iconsole_color_default, "Game paused.");

   395

	} else

   396

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

   397

   398

	return true;

   399

}

   400

   401

DEF_CONSOLE_CMD(ConUnPauseGame)

   402

{

   403

	if (argc == 0) {

   404

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

   405

		return true;

   406

	}

   407

   408

	if (_pause != 0) {

   409

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

   410

		IConsolePrint(_iconsole_color_default, "Game unpaused.");

   411

	} else

   412

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

   413

   414

	return true;

   415

}

   416

   417

DEF_CONSOLE_CMD(ConRcon)

   418

{

   419

	if (argc == 0) {

   420

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

   421

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

   422

		return true;

   423

	}

   424

   425

	if (argc < 3) return false;

   426

   427

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

   428

	return true;

   429

}

   430

   431

DEF_CONSOLE_CMD(ConStatus)

   432

{

   433

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

   434

	const char *status;

   435

	const NetworkClientState *cs;

   436

   437

	if (argc == 0) {

   438

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

   439

		return true;

   440

	}

   441

   442

	FOR_ALL_CLIENTS(cs) {

   443

		int lag = NetworkCalculateLag(cs);

   444

		const NetworkClientInfo *ci = DEREF_CLIENT_INFO(cs);

   445

   446

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

   447

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

   448

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

   449

	}

   450

   451

	return true;

   452

}

   453

   454

DEF_CONSOLE_CMD(ConKick)

   455

{

   456

	NetworkClientInfo *ci;

   457

	uint32 index;

   458

   459

	if (argc == 0) {

   460

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

   461

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

   462

		return true;

   463

	}

   464

   465

	if (argc != 2) return false;

   466

   467

	index = atoi(argv[1]);

   468

	if (index == NETWORK_SERVER_INDEX) {

   469

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

   470

		return true;

   471

	}

   472

	if (index == 0) {

   473

		IConsoleError("Invalid Client-ID");

   474

		return true;

   475

	}

   476

   477

	ci = NetworkFindClientInfoFromIndex(index);

   478

   479

	if (ci != NULL) {

   480

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

   481

	} else

   482

		IConsoleError("Client-ID not found");

   483

   484

	return true;

   485

}