1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "functions.h"

     6

#include "macros.h"

     7

#include "debug.h"

     8

#include "variables.h"

     9

#include "string.h"

    10

#include "saveload.h"

    11

#include "md5.h"

    12

#include "network_data.h"

    13

#include "newgrf.h"

    14

#include "newgrf_config.h"

    15

    16

#include "fileio.h"

    17

#include "fios.h"

    18

#include <sys/stat.h>

    19

    20

#ifdef WIN32

    21

# include <io.h>

    22

#endif /* WIN32 */

    23

    24

    25

GRFConfig *_all_grfs;

    26

GRFConfig *_grfconfig;

    27

GRFConfig *_grfconfig_newgame;

    28

GRFConfig *_grfconfig_static;

    29

    30

    31

/* Calculate the MD5 Sum for a GRF */

    32

static bool CalcGRFMD5Sum(GRFConfig *config)

    33

{

    34

	FILE *f;

    35

	char filename[MAX_PATH];

    36

	md5_state_t md5state;

    37

	md5_byte_t buffer[1024];

    38

	size_t len;

    39

    40

	/* open the file */

    41

	snprintf(filename, lengthof(filename), "%s%s", _paths.data_dir, config->filename);

    42

	f = fopen(filename, "rb");

    43

	if (f == NULL) return false;

    44

    45

	/* calculate md5sum */

    46

	md5_init(&md5state);

    47

	while ((len = fread(buffer, 1, sizeof(buffer), f)) != 0) {

    48

		md5_append(&md5state, buffer, len);

    49

	}

    50

	md5_finish(&md5state, config->md5sum);

    51

    52

	fclose(f);

    53

    54

	return true;

    55

}

    56

    57

    58

/* Find the GRFID and calculate the md5sum */

    59

bool FillGRFDetails(GRFConfig *config, bool is_static)

    60

{

    61

	if (!FioCheckFileExists(config->filename)) {

    62

		SETBIT(config->flags, GCF_NOT_FOUND);

    63

		return false;

    64

	}

    65

    66

	/* Find and load the Action 8 information */

    67

	/* 62 is the last file slot before sample.cat.

    68

	 * Should perhaps be some "don't care" value */

    69

	LoadNewGRFFile(config, 62, GLS_FILESCAN);

    70

    71

	/* Skip if the grfid is 0 (not read) or 0xFFFFFFFF (ttdp system grf) */

    72

	if (config->grfid == 0 || config->grfid == 0xFFFFFFFF) return false;

    73

    74

	if (is_static) {

    75

		/* Perform a 'safety scan' for static GRFs */

    76

		LoadNewGRFFile(config, 62, GLS_SAFETYSCAN);

    77

    78

		/* GCF_UNSAFE is set if GLS_SAFETYSCAN finds unsafe actions */

    79

		if (HASBIT(config->flags, GCF_UNSAFE)) return false;

    80

	}

    81

    82

	return CalcGRFMD5Sum(config);

    83

}

    84

    85

    86

void ClearGRFConfig(GRFConfig **config)

    87

{

    88

	/* GCF_COPY as in NOT strdupped/alloced the filename, name and info */

    89

	if (!HASBIT((*config)->flags, GCF_COPY)) {

    90

		free((*config)->filename);

    91

		free((*config)->name);

    92

		free((*config)->info);

    93

	}

    94

	free(*config);

    95

	*config = NULL;

    96

}

    97

    98

    99

/* Clear a GRF Config list */

   100

void ClearGRFConfigList(GRFConfig **config)

   101

{

   102

	GRFConfig *c, *next;

   103

	for (c = *config; c != NULL; c = next) {

   104

		next = c->next;

   105

		ClearGRFConfig(&c);

   106

	}

   107

	*config = NULL;

   108

}

   109

   110

   111

/** Copy a GRF Config list

   112

 * @param dst pointer to destination list

   113

 * @param srt pointer to source list values

   114

 * @return pointer to the last value added to the destination list */

   115

GRFConfig **CopyGRFConfigList(GRFConfig **dst, const GRFConfig *src)

   116

{

   117

	GRFConfig *c;

   118

   119

	/* Clear destination as it will be overwritten */

   120

	ClearGRFConfigList(dst);

   121

	for (; src != NULL; src = src->next) {

   122

		c = calloc(1, sizeof(*c));

   123

		*c = *src;

   124

		if (src->filename != NULL) c->filename = strdup(src->filename);

   125

		if (src->name     != NULL) c->name     = strdup(src->name);

   126

		if (src->info     != NULL) c->info     = strdup(src->info);

   127

   128

		*dst = c;

   129

		dst = &c->next;

   130

	}

   131

   132

	return dst;

   133

}

   134

   135

/**

   136

 * Removes duplicates from lists of GRFConfigs. These duplicates

   137

 * are introduced when the _grfconfig_static GRFs are appended

   138

 * to the _grfconfig on a newgame or savegame. As the parameters

   139

 * of the static GRFs could be different that the parameters of

   140

 * the ones used non-statically. This can result in desyncs in

   141

 * multiplayers, so the duplicate static GRFs have to be removed.

   142

 *

   143

 * This function _assumes_ that all static GRFs are placed after

   144

 * the non-static GRFs.

   145

 *

   146

 * @param list the list to remove the duplicates from

   147

 */

   148

static void RemoveDuplicatesFromGRFConfigList(GRFConfig *list)

   149

{

   150

	GRFConfig *prev;

   151

	GRFConfig *cur;

   152

   153

	if (list == NULL) return;

   154

   155

	for (prev = list, cur = list->next; cur != NULL; prev = cur, cur = cur->next) {

   156

		if (cur->grfid != list->grfid) continue;

   157

   158

		prev->next = cur->next;

   159

		ClearGRFConfig(&cur);

   160

		cur = prev; // Just go back one so it continues as normal later on

   161

	}

   162

   163

	RemoveDuplicatesFromGRFConfigList(list->next);

   164

}

   165

   166

/**

   167

 * Appends the static GRFs to a list of GRFs

   168

 * @param dst the head of the list to add to

   169

 */

   170

void AppendStaticGRFConfigs(GRFConfig **dst)

   171

{

   172

	GRFConfig **tail = dst;

   173

	while (*tail != NULL) tail = &(*tail)->next;

   174

   175

	CopyGRFConfigList(tail, _grfconfig_static);

   176

	RemoveDuplicatesFromGRFConfigList(*dst);

   177

}

   178

   179

   180

/* Reset the current GRF Config to either blank or newgame settings */

   181

void ResetGRFConfig(bool defaults)

   182

{

   183

	GRFConfig **c = &_grfconfig;

   184

   185

	if (defaults) {

   186

		c = CopyGRFConfigList(c, _grfconfig_newgame);

   187

	} else {

   188

		ClearGRFConfigList(c);

   189

	}

   190

   191

	AppendStaticGRFConfigs(&_grfconfig);

   192

}

   193

   194

   195

/** Check if all GRFs in the GRF config from a savegame can be loaded.

   196

 * @return will return any of the following 3 values:<br>

   197

 * <ul>

   198

 * <li> GCF_ACTIVATED: No problems occured, all GRF files were found and loaded

   199

 * <li> GCF_COMPATIBLE: For one or more GRF's no exact match was found, but a

   200

 *     compatible GRF with the same grfid was found and used instead

   201

 * <li> GCF_NOT_FOUND: For one or more GRF's no match was found at all

   202

 * </ul> */

   203

GCF_Flags IsGoodGRFConfigList(void)

   204

{

   205

	GCF_Flags res = GCF_ACTIVATED;

   206

	GRFConfig *c;

   207

   208

	for (c = _grfconfig; c != NULL; c = c->next) {

   209

		const GRFConfig *f = FindGRFConfig(c->grfid, c->md5sum);

   210

		if (f == NULL) {

   211

			char buf[256];

   212

   213

			/* If we have not found the exactly matching GRF try to find one with the

   214

			 * same grfid, as it most likely is compatible */

   215

			f = FindGRFConfig(c->grfid, NULL);

   216

			if (f != NULL) {

   217

				md5sumToString(buf, lastof(buf), c->md5sum);

   218

				DEBUG(grf, 1) ("[GRF] NewGRF %08X (%s) not found; checksum %s. Compatibility mode on", BSWAP32(c->grfid), c->filename, buf);

   219

				SETBIT(c->flags, GCF_COMPATIBLE);

   220

   221

				/* Non-found has precedence over compatibility load */

   222

				if (res != GCF_NOT_FOUND) res = GCF_COMPATIBLE;

   223

				goto compatible_grf;

   224

			}

   225

   226

			/* No compatible grf was found, mark it as disabled */

   227

			md5sumToString(buf, lastof(buf), c->md5sum);

   228

			DEBUG(grf, 0) ("[GRF] NewGRF %08X (%s) not found; checksum %s", BSWAP32(c->grfid), c->filename, buf);

   229

   230

			SETBIT(c->flags, GCF_NOT_FOUND);

   231

			res = GCF_NOT_FOUND;

   232

		} else {

   233

compatible_grf:

   234

			DEBUG(grf, 1) ("[GRF] Loading GRF %08X from %s", BSWAP32(f->grfid), f->filename);

   235

			/* The filename could be the filename as in the savegame. As we need

   236

			 * to load the GRF here, we need the correct filename, so overwrite that

   237

			 * in any case and set the name and info when it is not set already.

   238

			 * When the GCF_COPY flag is set, it is certain that the filename is

   239

			 * already a local one, so there is no need to replace it. */

   240

			if (!HASBIT(c->flags, GCF_COPY)) {

   241

				free(c->filename);

   242

				c->filename = strdup(f->filename);

   243

				memcpy(c->md5sum, f->md5sum, sizeof(c->md5sum));

   244

				if (c->name == NULL && f->name != NULL) c->name = strdup(f->name);

   245

				if (c->info == NULL && f->info != NULL) c->info = strdup(f->info);

   246

			}

   247

		}

   248

	}

   249

   250

	return res;

   251

}

   252

   253

   254

extern bool FiosIsValidFile(const char *path, const struct dirent *ent, struct stat *sb);

   255

   256

/* Scan a path for NewGRFs */

   257

static uint ScanPath(const char *path)

   258

{

   259

	uint num = 0;

   260

	struct stat sb;

   261

	struct dirent *dirent;

   262

	DIR *dir;

   263

	GRFConfig *c;

   264

   265

	if ((dir = ttd_opendir(path)) == NULL) return 0;

   266

   267

	while ((dirent = readdir(dir)) != NULL) {

   268

		const char *d_name = FS2OTTD(dirent->d_name);

   269

		char filename[MAX_PATH];

   270

   271

		if (!FiosIsValidFile(path, dirent, &sb)) continue;

   272

   273

		snprintf(filename, lengthof(filename), "%s" PATHSEP "%s", path, d_name);

   274

   275

		if (sb.st_mode & S_IFDIR) {

   276

			/* Directory */

   277

			if (strcmp(d_name, ".") == 0 || strcmp(d_name, "..") == 0) continue;

   278

			num += ScanPath(filename);

   279

		} else if (sb.st_mode & S_IFREG) {

   280

			/* File */

   281

			char *ext = strrchr(filename, '.');

   282

			char *file = filename + strlen(_paths.data_dir) + 1; // Crop base path

   283

   284

			/* If no extension or extension isn't .grf, skip the file */

   285

			if (ext == NULL) continue;

   286

			if (strcasecmp(ext, ".grf") != 0) continue;

   287

   288

			c = calloc(1, sizeof(*c));

   289

			c->filename = strdup(file);

   290

   291

			if (FillGRFDetails(c, false)) {

   292

				if (_all_grfs == NULL) {

   293

					_all_grfs = c;

   294

				} else {

   295

					/* Insert file into list at a position determined by its

   296

					 * name, so the list is sorted as we go along */

   297

					GRFConfig **pd, *d;

   298

					for (pd = &_all_grfs; (d = *pd) != NULL; pd = &d->next) {

   299

						if (strcasecmp(c->name, d->name) <= 0) break;

   300

					}

   301

					c->next = d;

   302

					*pd = c;

   303

				}

   304

   305

				num++;

   306

			} else {

   307

				/* File couldn't be opened, or is either not a NewGRF or is a

   308

				 * 'system' NewGRF, so forget about it. */

   309

				free(c->filename);

   310

				free(c->name);

   311

				free(c->info);

   312

				free(c);

   313

			}

   314

		}

   315

	}

   316

   317

	closedir(dir);

   318

   319

	return num;

   320

}

   321

   322

   323

/* Scan for all NewGRFs */

   324

void ScanNewGRFFiles(void)

   325

{

   326

	uint num;

   327

   328

	ClearGRFConfigList(&_all_grfs);

   329

   330

	DEBUG(grf, 1) ("[GRF] Scanning for NewGRFs");

   331

	num = ScanPath(_paths.data_dir);

   332

	DEBUG(grf, 1) ("[GRF] Scan complete, found %d files", num);

   333

}

   334

   335

   336

/* Find a NewGRF in the scanned list, if md5sum is NULL, we don't care about it*/

   337

const GRFConfig *FindGRFConfig(uint32 grfid, const uint8 *md5sum)

   338

{

   339

	const GRFConfig *c;

   340

	for (c = _all_grfs; c != NULL; c = c->next) {

   341

		if (c->grfid == grfid) {

   342

			if (md5sum == NULL) return c;

   343

   344

			if (memcmp(md5sum, c->md5sum, sizeof(c->md5sum)) == 0) return c;

   345

		}

   346

	}

   347

   348

	return NULL;

   349

}

   350

   351

#ifdef ENABLE_NETWORK

   352

   353

/** Structure for UnknownGRFs; this is a lightweight variant of GRFConfig */

   354

typedef struct UnknownGRF UnknownGRF;

   355

struct UnknownGRF {

   356

	UnknownGRF *next;

   357

	uint32 grfid;

   358

	uint8  md5sum[16];

   359

	char   name[NETWORK_GRF_NAME_LENGTH];

   360

};

   361

   362

/**

   363

 * Finds the name of a NewGRF in the list of names for unknown GRFs. An

   364

 * unknown GRF is a GRF where the .grf is not found during scanning.

   365

 *

   366

 * The names are resolved via UDP calls to servers that should know the name,

   367

 * though the replies may not come. This leaves "<Unknown>" as name, though

   368

 * that shouldn't matter _very_ much as they need GRF crawler or so to look

   369

 * up the GRF anyway and that works better with the GRF ID.

   370

 *

   371

 * @param grfid  the GRF ID part of the 'unique' GRF identifier

   372

 * @param md5sum the MD5 checksum part of the 'unique' GRF identifier

   373

 * @param create whether to create a new GRFConfig if the GRFConfig did not

   374

 *               exist in the fake list of GRFConfigs.

   375

 * @return the GRFConfig with the given GRF ID and MD5 checksum or NULL when

   376

 *         it does not exist and create is false. This value must NEVER be

   377

 *         freed by the caller.

   378

 */

   379

char *FindUnknownGRFName(uint32 grfid, uint8 *md5sum, bool create)

   380

{

   381

	UnknownGRF *grf;

   382

	static UnknownGRF *unknown_grfs = NULL;

   383

   384

	for (grf = unknown_grfs; grf != NULL; grf = grf->next) {

   385

		if (grf->grfid == grfid) {

   386

			if (memcmp(md5sum, grf->md5sum, sizeof(grf->md5sum)) == 0) return grf->name;

   387

		}

   388

	}

   389

   390

	if (!create) return NULL;

   391

   392

	grf = calloc(1, sizeof(*grf));

   393

	grf->grfid = grfid;

   394

	grf->next  = unknown_grfs;

   395

	ttd_strlcpy(grf->name, UNKNOWN_GRF_NAME_PLACEHOLDER, sizeof(grf->name));

   396

	memcpy(grf->md5sum, md5sum, sizeof(grf->md5sum));

   397

   398

	unknown_grfs = grf;

   399

	return grf->name;

   400

}

   401

   402

#endif /* ENABLE_NETWORK */

   403

   404

   405

/* Retrieve a NewGRF from the current config by its grfid */

   406

GRFConfig *GetGRFConfig(uint32 grfid)

   407

{

   408

	GRFConfig *c;

   409

   410

	for (c = _grfconfig; c != NULL; c = c->next) {

   411

		if (c->grfid == grfid) return c;

   412

	}

   413

   414

	return NULL;

   415

}

   416

   417

   418

/* Build a space separated list of parameters, and terminate */

   419

char *GRFBuildParamList(char *dst, const GRFConfig *c, const char *last)

   420

{

   421

	uint i;

   422

   423

	/* Return an empty string if there are no parameters */

   424

	if (c->num_params == 0) return strecpy(dst, "", last);

   425

   426

	for (i = 0; i < c->num_params; i++) {

   427

		if (i > 0) dst = strecpy(dst, " ", last);

   428

		dst += snprintf(dst, last - dst, "%d", c->param[i]);

   429

	}

   430

	return dst;

   431

}

   432

   433

   434

static const SaveLoad _grfconfig_desc[] = {

   435

	SLE_STR(GRFConfig, filename,   SLE_STR, 0x40),

   436

	SLE_VAR(GRFConfig, grfid,      SLE_UINT32),

   437

	SLE_ARR(GRFConfig, md5sum,     SLE_UINT8, 16),

   438

	SLE_ARR(GRFConfig, param,      SLE_UINT32, 0x80),

   439

	SLE_VAR(GRFConfig, num_params, SLE_UINT8),

   440

	SLE_END()

   441

};

   442

   443

   444

static void Save_NGRF(void)

   445

{

   446

	GRFConfig *c;

   447

	int index = 0;

   448

   449

	for (c = _grfconfig; c != NULL; c = c->next) {

   450

		if (HASBIT(c->flags, GCF_STATIC)) continue;

   451

		SlSetArrayIndex(index++);

   452

		SlObject(c, _grfconfig_desc);

   453

	}

   454

}

   455

   456

   457

static void Load_NGRF(void)

   458

{

   459

	GRFConfig *first = NULL;

   460

	GRFConfig **last = &first;

   461

   462

	while (SlIterateArray() != -1) {

   463

		GRFConfig *c = calloc(1, sizeof(*c));

   464

		SlObject(c, _grfconfig_desc);

   465

   466

		/* Append our configuration to the list */

   467

		*last = c;

   468

		last = &c->next;

   469

	}

   470

   471

	/* Append static NewGRF configuration */

   472

	CopyGRFConfigList(last, _grfconfig_static);

   473

   474

	ClearGRFConfigList(&_grfconfig);

   475

	_grfconfig = first;

   476

	AppendStaticGRFConfigs(&_grfconfig);

   477

}

   478

   479

const ChunkHandler _newgrf_chunk_handlers[] = {

   480

	{ 'NGRF', Save_NGRF, Load_NGRF, CH_ARRAY | CH_LAST }

   481

};

   482