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"

    10

#include "saveload.h"

    11

#include "md5.h"

    12

#include "newgrf.h"

    13

#include "newgrf_config.h"

    14

    15

#include "fileio.h"

    16

#include "fios.h"

    17

#include <sys/types.h>

    18

#include <sys/stat.h>

    19

    20

#ifdef WIN32

    21

# include <io.h>

    22

#else

    23

# include <unistd.h>

    24

# include <dirent.h>

    25

#endif /* WIN32 */

    26

    27

    28

GRFConfig *_all_grfs;

    29

GRFConfig *_grfconfig;

    30

GRFConfig *_grfconfig_newgame;

    31

    32

    33

/* Calculate the MD5 Sum for a GRF */

    34

static bool CalcGRFMD5Sum(GRFConfig *config)

    35

{

    36

	FILE *f;

    37

	char filename[MAX_PATH];

    38

	md5_state_t md5state;

    39

	md5_byte_t buffer[1024];

    40

	size_t len;

    41

    42

	/* open the file */

    43

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

    44

	f = fopen(filename, "rb");

    45

	if (f == NULL) return false;

    46

    47

	/* calculate md5sum */

    48

	md5_init(&md5state);

    49

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

    50

		md5_append(&md5state, buffer, len);

    51

	}

    52

	md5_finish(&md5state, config->md5sum);

    53

    54

	fclose(f);

    55

    56

	return true;

    57

}

    58

    59

    60

/* Find the GRFID and calculate the md5sum */

    61

bool FillGRFDetails(GRFConfig *config)

    62

{

    63

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

    64

		SETBIT(config->flags, GCF_NOT_FOUND);

    65

		return false;

    66

	}

    67

    68

	/* Find and load the Action 8 information */

    69

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

    70

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

    71

	LoadNewGRFFile(config, 62, GLS_FILESCAN);

    72

    73

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

    74

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

    75

    76

	return CalcGRFMD5Sum(config);

    77

}

    78

    79

    80

/* Clear a GRF Config list */

    81

void ClearGRFConfigList(GRFConfig *config)

    82

{

    83

	GRFConfig *c, *next;

    84

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

    85

		next = c->next;

    86

		free(c->filename);

    87

		free(c->name);

    88

		free(c->info);

    89

		free(c);

    90

	}

    91

}

    92

    93

    94

/* Copy a GRF Config list */

    95

static void CopyGRFConfigList(GRFConfig **dst, GRFConfig *src)

    96

{

    97

	GRFConfig *c;

    98

    99

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

   100

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

   101

		*c = *src;

   102

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

   103

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

   104

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

   105

   106

		*dst = c;

   107

		dst = &c->next;

   108

	}

   109

}

   110

   111

   112

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

   113

void ResetGRFConfig(bool defaults)

   114

{

   115

	ClearGRFConfigList(_grfconfig);

   116

	_grfconfig = NULL;

   117

	if (defaults) CopyGRFConfigList(&_grfconfig, _grfconfig_newgame);

   118

}

   119

   120

   121

/* Check if all GRFs in the GRF Config can be loaded */

   122

bool IsGoodGRFConfigList(void)

   123

{

   124

	bool res = true;

   125

	GRFConfig *c;

   126

   127

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

   128

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

   129

		if (f == NULL) {

   130

			char buf[512], *p = buf;

   131

			uint i;

   132

   133

			p += snprintf(p, lastof(buf) - p, "Couldn't find NewGRF %08X (%s) checksum ", BSWAP32(c->grfid), c->filename);

   134

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

   135

				p += snprintf(p, lastof(buf) - p, "%02X", c->md5sum[i]);

   136

			}

   137

			ShowInfo(buf);

   138

   139

			res = false;

   140

		} else {

   141

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

   142

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

   143

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

   144

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

   145

		}

   146

	}

   147

   148

	return res;

   149

}

   150

   151

   152

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

   153

   154

/* Scan a path for NewGRFs */

   155

static uint ScanPath(const char *path)

   156

{

   157

	uint num = 0;

   158

	struct stat sb;

   159

	struct dirent *dirent;

   160

	DIR *dir;

   161

	GRFConfig *c;

   162

   163

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

   164

   165

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

   166

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

   167

		char filename[MAX_PATH];

   168

   169

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

   170

   171

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

   172

   173

		if (sb.st_mode & S_IFDIR) {

   174

			/* Directory */

   175

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

   176

			num += ScanPath(filename);

   177

		} else if (sb.st_mode & S_IFREG) {

   178

			/* File */

   179

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

   180

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

   181

   182

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

   183

			if (ext == NULL) continue;

   184

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

   185

   186

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

   187

			c->filename = strdup(file);

   188

   189

			if (FillGRFDetails(c)) {

   190

				if (_all_grfs == NULL) {

   191

					_all_grfs = c;

   192

				} else {

   193

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

   194

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

   195

					GRFConfig **pd, *d;

   196

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

   197

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

   198

					}

   199

					c->next = d;

   200

					*pd = c;

   201

				}

   202

   203

				num++;

   204

			} else {

   205

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

   206

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

   207

				free(c->filename);

   208

				free(c->name);

   209

				free(c->info);

   210

				free(c);

   211

			}

   212

		}

   213

	}

   214

   215

	closedir(dir);

   216

   217

	return num;

   218

}

   219

   220

   221

/* Scan for all NewGRFs */

   222

void ScanNewGRFFiles(void)

   223

{

   224

	uint num;

   225

   226

	ClearGRFConfigList(_all_grfs);

   227

	_all_grfs = NULL;

   228

   229

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

   230

	num = ScanPath(_paths.data_dir);

   231

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

   232

}

   233

   234

   235

/* Find a NewGRF in the scanned list */

   236

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

   237

{

   238

	GRFConfig *c;

   239

	static const uint8 blanksum[sizeof(c->md5sum)] = { 0 };

   240

   241

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

   242

		if (c->grfid == grfid) {

   243

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

   244

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

   245

		}

   246

	}

   247

   248

	return NULL;

   249

}

   250

   251

   252

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

   253

const GRFConfig *GetGRFConfig(uint32 grfid)

   254

{

   255

	GRFConfig *c;

   256

   257

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

   258

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

   259

	}

   260

   261

	return NULL;

   262

}

   263

   264

   281

static const SaveLoad _grfconfig_desc[] = {

   282

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

   283

	SLE_VAR(GRFConfig, grfid,      SLE_UINT32),

   284

	SLE_ARR(GRFConfig, md5sum,     SLE_UINT8, 16),

   285

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

   286

	SLE_VAR(GRFConfig, num_params, SLE_UINT8),

   287

	SLE_END()

   288

};

   289

   290

   291

static void Save_NGRF(void)

   292

{

   293

	GRFConfig *c;

   294

	int index = 0;

   295

   296

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

   297

		SlSetArrayIndex(index++);

   298

		SlObject(c, _grfconfig_desc);

   299

	}

   300

}

   301

   302

   303

static void Load_NGRF(void)

   304

{

   305

	GRFConfig *first = NULL;

   306

	GRFConfig **last = &first;

   307

   308

	while (SlIterateArray() != -1) {

   309

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

   310

		SlObject(c, _grfconfig_desc);

   311

   312

		/* Append our configuration to the list */

   313

		*last = c;

   314

		last = &c->next;

   315

	}

   316

   317

	ClearGRFConfigList(_grfconfig);

   318

	_grfconfig = first;

   319

}

   320

   321

const ChunkHandler _newgrf_chunk_handlers[] = {

   322

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

   323

};

   324