1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "console.h"

     6

#include "debug.h"

     7

#include "engine.h"

     8

#include "functions.h"

     9

#include "saveload.h"

    10

#include "string.h"

    11

#include "variables.h"

    12

#include "network_data.h"

    13

#include "network_client.h"

    14

#include "network_server.h"

    15

#include "network_udp.h"

    16

#include "command.h"

    17

#include "settings.h"

    18

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

    19

#include "vehicle.h"

    20

    21

// ** scriptfile handling ** //

    22

static FILE *_script_file;

    23

static bool _script_running;

    24

    25

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

    26

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

    27

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

    28

    29

    30

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

    31

/* variable and command hooks   */

    32

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

    33

    34

#ifdef ENABLE_NETWORK

    35

    36

static inline bool NetworkAvailable(void)

    37

{

    38

	if (!_network_available) {

    39

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

    40

		return false;

    41

	}

    42

	return true;

    43

}

    44

    45

DEF_CONSOLE_HOOK(ConHookServerOnly)

    46

{

    47

	if (!NetworkAvailable()) return false;

    48

    49

	if (!_network_server) {

    50

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

    51

		return false;

    52

	}

    53

	return true;

    54

}

    55

    56

DEF_CONSOLE_HOOK(ConHookClientOnly)

    57

{

    58

	if (!NetworkAvailable()) return false;

    59

    60

	if (_network_server) {

    61

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

    62

		return false;

    63

	}

    64

	return true;

    65

}

    66

    67

DEF_CONSOLE_HOOK(ConHookNeedNetwork)

    68

{

    69

	if (!NetworkAvailable()) return false;

    70

    71

	if (!_networking) {

    72

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

    73

		return false;

    74

	}

    75

	return true;

    76

}

    77

    78

DEF_CONSOLE_HOOK(ConHookNoNetwork)

    79

{

    80

	if (_networking) {

    81

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

    82

		return false;

    83

	}

    84

	return true;

    85

}

    86

    87

#endif /* ENABLE_NETWORK */

    88

    89

static void IConsoleHelp(const char *str)

    90

{

    91

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

    92

}

    93

    94

DEF_CONSOLE_CMD(ConStopAllVehicles)

    95

{

    96

	Vehicle* v;

    97

	if (argc == 0) {

    98

		IConsoleHelp("Stops all vehicles in the game. For debugging only! Use at your own risk... Usage: 'stopall'");

    99

		return true;

   100

	}

   101

   102

	FOR_ALL_VEHICLES(v) {

   103

		if (IsValidVehicle(v)) {

   104

			/* Code ripped from CmdStartStopTrain. Can't call it, because of

   105

			 * ownership problems, so we'll duplicate some code, for now */

   106

			if (v->type == VEH_Train)

   107

				v->u.rail.days_since_order_progr = 0;

   108

			v->vehstatus |= VS_STOPPED;

   109

			InvalidateWindowWidget(WC_VEHICLE_VIEW, v->index, STATUS_BAR);

   110

			InvalidateWindow(WC_VEHICLE_DEPOT, v->tile);

   111

		}

   112

	}

   113

	return true;

   114

}

   115

   116

DEF_CONSOLE_CMD(ConResetEngines)

   117

{

   118

	if (argc == 0) {

   119

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

   120

		return true;

   121

	}

   122

   123

	StartupEngines();

   124

	return true;

   125

}

   126

   127

#ifdef _DEBUG

   128

DEF_CONSOLE_CMD(ConResetTile)

   129

{

   130

	if (argc == 0) {

   131

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

   132

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

   133

		return true;

   134

	}

   135

   136

	if (argc == 2) {

   137

		uint32 result;

   138

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

   139

			DoClearSquare((TileIndex)result);

   140

			return true;

   141

		}

   142

	}

   143

   144

	return false;

   145

}

   146

#endif /* _DEBUG */

   147

   148

DEF_CONSOLE_CMD(ConScrollToTile)

   149

{

   150

	if (argc == 0) {

   151

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

   152

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

   153

		return true;

   154

	}

   155

   156

	if (argc == 2) {

   157

		uint32 result;

   158

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

   159

			ScrollMainWindowToTile((TileIndex)result);

   160

			return true;

   161

		}

   162

	}

   163

   164

	return false;

   165

}

   166

   167

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

   168

extern void BuildFileList(void);

   169

extern void SetFiosType(const byte fiostype);

   170

   171

/* Save the map to a file */

   172

DEF_CONSOLE_CMD(ConSave)

   173

{

   174

	if (argc == 0) {

   175

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

   176

		return true;

   177

	}

   178

   179

	if (argc == 2) {

   180

		char buf[200];

   181

   182

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

   183

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

   184

   185

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

   186

			IConsolePrint(_icolour_err, "SaveMap failed");

   187

		} else

   188

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

   189

		return true;

   190

	}

   191

   192

	return false;

   193

}

   194

   195

static const FiosItem* GetFiosItem(const char* file)

   196

{

   197

	int i;

   198

   199

	_saveload_mode = SLD_LOAD_GAME;

   200

	BuildFileList();

   201

   202

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

   203

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

   204

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

   205

	}

   206

   207

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

   208

		char* endptr;

   209

   210

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

   211

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

   212

	}

   213

   214

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

   215

}

   216

   217

   218

DEF_CONSOLE_CMD(ConLoad)

   219

{

   220

	const FiosItem *item;

   221

	const char *file;

   222

   223

	if (argc == 0) {

   224

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

   225

		return true;

   226

	}

   227

   228

	if (argc != 2) return false;

   229

   230

	file = argv[1];

   231

	item = GetFiosItem(file);

   232

	if (item != NULL) {

   233

		switch (item->type) {

   234

			case FIOS_TYPE_FILE: case FIOS_TYPE_OLDFILE: {

   235

				_switch_mode = SM_LOAD;

   236

				SetFiosType(item->type);

   237

   238

				ttd_strlcpy(_file_to_saveload.name, FiosBrowseTo(item), sizeof(_file_to_saveload.name));

   239

				ttd_strlcpy(_file_to_saveload.title, item->title, sizeof(_file_to_saveload.title));

   240

			} break;

   241

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

   242

		}

   243

	} else

   244

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

   245

   246

	FiosFreeSavegameList();

   247

	return true;

   248

}

   249

   250

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

   251

DEF_CONSOLE_CMD(ConListFiles)

   252

{

   253

	int i;

   254

   255

	if (argc == 0) {

   256

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

   257

		return true;

   258

	}

   259

   260

	BuildFileList();

   261

   262

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

   263

		const FiosItem *item = &_fios_list[i];

   264

		IConsolePrintF(_icolour_def, "%d) %s", i, item->title);

   265

	}

   266

   267

	FiosFreeSavegameList();

   268

	return true;

   269

}

   270

   271

/* Change the dir via console */

   272

DEF_CONSOLE_CMD(ConChangeDirectory)

   273

{

   274

	const FiosItem *item;

   275

	const char *file;

   276

   277

	if (argc == 0) {

   278

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

   279

		return true;

   280

	}

   281

   282

	if (argc != 2) return false;

   283

   284

	file = argv[1];

   285

	item = GetFiosItem(file);

   286

	if (item != NULL) {

   287

		switch (item->type) {

   288

			case FIOS_TYPE_DIR: case FIOS_TYPE_DRIVE: case FIOS_TYPE_PARENT:

   289

				FiosBrowseTo(item);

   290

				break;

   291

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

   292

		}

   293

	} else

   294

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

   295

   296

	FiosFreeSavegameList();

   297

	return true;

   298

}

   299

   300

DEF_CONSOLE_CMD(ConPrintWorkingDirectory)

   301

{

   302

	const char *path;

   303

   304

	if (argc == 0) {

   305

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

   306

		return true;

   307

	}

   308

   309

	// XXX - Workaround for broken file handling

   310

	FiosGetSavegameList(&_fios_num, SLD_LOAD_GAME);

   311

	FiosFreeSavegameList();

   312

   313

	FiosGetDescText(&path, NULL);

   314

	IConsolePrint(_icolour_def, path);

   315

	return true;

   316

}

   317

   318

DEF_CONSOLE_CMD(ConClearBuffer)

   319

{

   320

	if (argc == 0) {

   321

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

   322

		return true;

   323

	}

   324

   325

	IConsoleClearBuffer();

   326

	InvalidateWindow(WC_CONSOLE, 0);

   327

	return true;

   328

}

   329

   330

   331

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

   332

// * Network Core Console Commands * //

   333

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

   334

#ifdef ENABLE_NETWORK

   335

   336

DEF_CONSOLE_CMD(ConBan)

   337

{

   338

	NetworkClientInfo *ci;

   339

	uint32 index;

   340

   341

	if (argc == 0) {

   342

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

   343

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

   344

		return true;

   345

	}

   346

   347

	if (argc != 2) return false;

   348

   349

	index = atoi(argv[1]);

   350

   351

	if (index == NETWORK_SERVER_INDEX) {

   352

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

   353

		return true;

   354

	}

   355

	if (index == 0) {

   356

		IConsoleError("Invalid Client-ID");

   357

		return true;

   358

	}

   359

   360

	ci = NetworkFindClientInfoFromIndex(index);

   361

   362

	if (ci != NULL) {

   363

		uint i;

   364

		/* Add user to ban-list */

   365

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

   366

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

   367

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

   368

				break;

   369

			}

   370

		}

   371

   372

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

   373

	} else

   374

		IConsoleError("Client-ID not found");

   375

   376

	return true;

   377

}

   378

   379

DEF_CONSOLE_CMD(ConUnBan)

   380

{

   381

	uint i;

   382

   383

	if (argc == 0) {

   384

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

   385

		return true;

   386

	}

   387

   388

	if (argc != 2) return false;

   389

   390

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

   391

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

   392

			continue;

   393

   394

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

   395

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

   396

			IConsolePrint(_icolour_def, "IP unbanned.");

   397

			return true;

   398

		}

   399

	}

   400

   401

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

   402

	return true;

   403

}

   404

   405

DEF_CONSOLE_CMD(ConBanList)

   406

{

   407

	uint i;

   408

   409

	if (argc == 0) {

   410

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

   411

		return true;

   412

	}

   413

   414

	IConsolePrint(_icolour_def, "Banlist: ");

   415

   416

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

   417

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

   418

			continue;

   419

   420

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

   421

	}

   422

   423

	return true;

   424

}

   425

   426

DEF_CONSOLE_CMD(ConPauseGame)

   427

{

   428

	if (argc == 0) {

   429

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

   430

		return true;

   431

	}

   432

   433

	if (_pause == 0) {

   434

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

   435

		IConsolePrint(_icolour_def, "Game paused.");

   436

	} else

   437

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

   438

   439

	return true;

   440

}

   441

   442

DEF_CONSOLE_CMD(ConUnPauseGame)

   443

{

   444

	if (argc == 0) {

   445

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

   446

		return true;

   447

	}

   448

   449

	if (_pause != 0) {

   450

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

   451

		IConsolePrint(_icolour_def, "Game unpaused.");

   452

	} else

   453

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

   454

   455

	return true;

   456

}

   457

   458

DEF_CONSOLE_CMD(ConRcon)

   459

{

   460

	if (argc == 0) {

   461

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

   462

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

   463

		return true;

   464

	}

   465

   466

	if (argc < 3) return false;

   467

   468

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

   469

	return true;

   470

}

   471

   472

DEF_CONSOLE_CMD(ConStatus)

   473

{

   474

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

   475

	const char *status;

   476

	const NetworkClientState *cs;

   477

   478

	if (argc == 0) {

   479

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

   480

		return true;

   481

	}

   482

   483

	FOR_ALL_CLIENTS(cs) {