773 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			773 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
 * This file is part of OpenTTD.
 | 
						|
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 | 
						|
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 | 
						|
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 | 
						|
 */
 | 
						|
 | 
						|
/**
 | 
						|
 * @file fios.cpp
 | 
						|
 * This file contains functions for building file lists for the save/load dialogs.
 | 
						|
 */
 | 
						|
 | 
						|
#include "stdafx.h"
 | 
						|
#include "3rdparty/md5/md5.h"
 | 
						|
#include "fileio_func.h"
 | 
						|
#include "fios.h"
 | 
						|
#include "network/network_content.h"
 | 
						|
#include "screenshot.h"
 | 
						|
#include "string_func.h"
 | 
						|
#include "tar_type.h"
 | 
						|
#include <sys/stat.h>
 | 
						|
 | 
						|
#ifndef _WIN32
 | 
						|
# include <unistd.h>
 | 
						|
#endif /* _WIN32 */
 | 
						|
 | 
						|
#include "table/strings.h"
 | 
						|
 | 
						|
#include "safeguards.h"
 | 
						|
 | 
						|
/* Variables to display file lists */
 | 
						|
static char *_fios_path;
 | 
						|
static const char *_fios_path_last;
 | 
						|
SortingBits _savegame_sort_order = SORT_BY_DATE | SORT_DESCENDING;
 | 
						|
 | 
						|
/* OS-specific functions are taken from their respective files (win32/unix/os2 .c) */
 | 
						|
extern bool FiosIsRoot(const char *path);
 | 
						|
extern bool FiosIsValidFile(const char *path, const struct dirent *ent, struct stat *sb);
 | 
						|
extern bool FiosIsHiddenFile(const struct dirent *ent);
 | 
						|
extern void FiosGetDrives(FileList &file_list);
 | 
						|
extern bool FiosGetDiskFreeSpace(const char *path, uint64 *tot);
 | 
						|
 | 
						|
/* get the name of an oldstyle savegame */
 | 
						|
extern void GetOldSaveGameName(const char *file, char *title, const char *last);
 | 
						|
 | 
						|
/**
 | 
						|
 * Compare two FiosItem's. Used with sort when sorting the file list.
 | 
						|
 * @param other The FiosItem to compare to.
 | 
						|
 * @return for ascending order: returns true if da < db. Vice versa for descending order.
 | 
						|
 */
 | 
						|
bool FiosItem::operator< (const FiosItem &other) const
 | 
						|
{
 | 
						|
	int r = false;
 | 
						|
 | 
						|
	if ((_savegame_sort_order & SORT_BY_NAME) == 0 && (*this).mtime != other.mtime) {
 | 
						|
		r = (*this).mtime - other.mtime;
 | 
						|
	} else {
 | 
						|
		r = strnatcmp((*this).title, other.title);
 | 
						|
	}
 | 
						|
	if (r == 0) return false;
 | 
						|
	return (_savegame_sort_order & SORT_DESCENDING) ? r > 0 : r < 0;
 | 
						|
}
 | 
						|
 | 
						|
FileList::~FileList()
 | 
						|
{
 | 
						|
	this->Clear();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Construct a file list with the given kind of files, for the stated purpose.
 | 
						|
 * @param abstract_filetype Kind of files to collect.
 | 
						|
 * @param fop Purpose of the collection, either #SLO_LOAD or #SLO_SAVE.
 | 
						|
 */
 | 
						|
void FileList::BuildFileList(AbstractFileType abstract_filetype, SaveLoadOperation fop)
 | 
						|
{
 | 
						|
	this->Clear();
 | 
						|
 | 
						|
	assert(fop == SLO_LOAD || fop == SLO_SAVE);
 | 
						|
	switch (abstract_filetype) {
 | 
						|
		case FT_NONE:
 | 
						|
			break;
 | 
						|
 | 
						|
		case FT_SAVEGAME:
 | 
						|
			FiosGetSavegameList(fop, *this);
 | 
						|
			break;
 | 
						|
 | 
						|
		case FT_SCENARIO:
 | 
						|
			FiosGetScenarioList(fop, *this);
 | 
						|
			break;
 | 
						|
 | 
						|
		case FT_HEIGHTMAP:
 | 
						|
			FiosGetHeightmapList(fop, *this);
 | 
						|
			break;
 | 
						|
 | 
						|
		default:
 | 
						|
			NOT_REACHED();
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Find file information of a file by its name from the file list.
 | 
						|
 * @param file The filename to return information about. Can be the actual name
 | 
						|
 *             or a numbered entry into the filename list.
 | 
						|
 * @return The information on the file, or \c nullptr if the file is not available.
 | 
						|
 */
 | 
						|
const FiosItem *FileList::FindItem(const char *file)
 | 
						|
{
 | 
						|
	for (const FiosItem *item = this->Begin(); item != this->End(); item++) {
 | 
						|
		if (strcmp(file, item->name) == 0) return item;
 | 
						|
		if (strcmp(file, item->title) == 0) return item;
 | 
						|
	}
 | 
						|
 | 
						|
	/* If no name matches, try to parse it as number */
 | 
						|
	char *endptr;
 | 
						|
	int i = strtol(file, &endptr, 10);
 | 
						|
	if (file == endptr || *endptr != '\0') i = -1;
 | 
						|
 | 
						|
	if (IsInsideMM(i, 0, this->Length())) return this->Get(i);
 | 
						|
 | 
						|
	/* As a last effort assume it is an OpenTTD savegame and
 | 
						|
	 * that the ".sav" part was not given. */
 | 
						|
	char long_file[MAX_PATH];
 | 
						|
	seprintf(long_file, lastof(long_file), "%s.sav", file);
 | 
						|
	for (const FiosItem *item = this->Begin(); item != this->End(); item++) {
 | 
						|
		if (strcmp(long_file, item->name) == 0) return item;
 | 
						|
		if (strcmp(long_file, item->title) == 0) return item;
 | 
						|
	}
 | 
						|
 | 
						|
	return nullptr;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get descriptive texts. Returns the path and free space
 | 
						|
 * left on the device
 | 
						|
 * @param path string describing the path
 | 
						|
 * @param total_free total free space in megabytes, optional (can be nullptr)
 | 
						|
 * @return StringID describing the path (free space or failure)
 | 
						|
 */
 | 
						|
StringID FiosGetDescText(const char **path, uint64 *total_free)
 | 
						|
{
 | 
						|
	*path = _fios_path;
 | 
						|
	return FiosGetDiskFreeSpace(*path, total_free) ? STR_SAVELOAD_BYTES_FREE : STR_ERROR_UNABLE_TO_READ_DRIVE;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Browse to a new path based on the passed \a item, starting at #_fios_path.
 | 
						|
 * @param *item Item telling us what to do.
 | 
						|
 * @return A filename w/path if we reached a file, otherwise \c nullptr.
 | 
						|
 */
 | 
						|
const char *FiosBrowseTo(const FiosItem *item)
 | 
						|
{
 | 
						|
	switch (item->type) {
 | 
						|
		case FIOS_TYPE_DRIVE:
 | 
						|
#if defined(_WIN32) || defined(__OS2__)
 | 
						|
			seprintf(_fios_path, _fios_path_last, "%c:" PATHSEP, item->title[0]);
 | 
						|
#endif
 | 
						|
			break;
 | 
						|
 | 
						|
		case FIOS_TYPE_INVALID:
 | 
						|
			break;
 | 
						|
 | 
						|
		case FIOS_TYPE_PARENT: {
 | 
						|
			/* Check for possible nullptr ptr */
 | 
						|
			char *s = strrchr(_fios_path, PATHSEPCHAR);
 | 
						|
			if (s != nullptr && s != _fios_path) {
 | 
						|
				s[0] = '\0'; // Remove last path separator character, so we can go up one level.
 | 
						|
			}
 | 
						|
			s = strrchr(_fios_path, PATHSEPCHAR);
 | 
						|
			if (s != nullptr) {
 | 
						|
				s[1] = '\0'; // go up a directory
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		}
 | 
						|
 | 
						|
		case FIOS_TYPE_DIR:
 | 
						|
			strecat(_fios_path, item->name, _fios_path_last);
 | 
						|
			strecat(_fios_path, PATHSEP, _fios_path_last);
 | 
						|
			break;
 | 
						|
 | 
						|
		case FIOS_TYPE_DIRECT:
 | 
						|
			seprintf(_fios_path, _fios_path_last, "%s", item->name);
 | 
						|
			break;
 | 
						|
 | 
						|
		case FIOS_TYPE_FILE:
 | 
						|
		case FIOS_TYPE_OLDFILE:
 | 
						|
		case FIOS_TYPE_SCENARIO:
 | 
						|
		case FIOS_TYPE_OLD_SCENARIO:
 | 
						|
		case FIOS_TYPE_PNG:
 | 
						|
		case FIOS_TYPE_BMP:
 | 
						|
			return item->name;
 | 
						|
	}
 | 
						|
 | 
						|
	return nullptr;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Construct a filename from its components in destination buffer \a buf.
 | 
						|
 * @param buf Destination buffer.
 | 
						|
 * @param path Directory path, may be \c nullptr.
 | 
						|
 * @param name Filename.
 | 
						|
 * @param ext Filename extension (use \c "" for no extension).
 | 
						|
 * @param last Last element of buffer \a buf.
 | 
						|
 */
 | 
						|
static void FiosMakeFilename(char *buf, const char *path, const char *name, const char *ext, const char *last)
 | 
						|
{
 | 
						|
	if (path != nullptr) {
 | 
						|
		const char *buf_start = buf;
 | 
						|
		buf = strecpy(buf, path, last);
 | 
						|
		/* Remove trailing path separator, if present */
 | 
						|
		if (buf > buf_start && buf[-1] == PATHSEPCHAR) buf--;
 | 
						|
	}
 | 
						|
 | 
						|
	/* Don't append the extension if it is already there */
 | 
						|
	const char *period = strrchr(name, '.');
 | 
						|
	if (period != nullptr && strcasecmp(period, ext) == 0) ext = "";
 | 
						|
 | 
						|
	seprintf(buf, last, PATHSEP "%s%s", name, ext);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Make a save game or scenario filename from a name.
 | 
						|
 * @param buf Destination buffer for saving the filename.
 | 
						|
 * @param name Name of the file.
 | 
						|
 * @param last Last element of buffer \a buf.
 | 
						|
 */
 | 
						|
void FiosMakeSavegameName(char *buf, const char *name, const char *last)
 | 
						|
{
 | 
						|
	const char *extension = (_game_mode == GM_EDITOR) ? ".scn" : ".sav";
 | 
						|
 | 
						|
	FiosMakeFilename(buf, _fios_path, name, extension, last);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Construct a filename for a height map.
 | 
						|
 * @param buf Destination buffer.
 | 
						|
 * @param name Filename.
 | 
						|
 * @param last Last element of buffer \a buf.
 | 
						|
 */
 | 
						|
void FiosMakeHeightmapName(char *buf, const char *name, const char *last)
 | 
						|
{
 | 
						|
	char ext[5];
 | 
						|
	ext[0] = '.';
 | 
						|
	strecpy(ext + 1, GetCurrentScreenshotExtension(), lastof(ext));
 | 
						|
 | 
						|
	FiosMakeFilename(buf, _fios_path, name, ext, last);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Delete a file.
 | 
						|
 * @param name Filename to delete.
 | 
						|
 * @return Whether the file deletion was successful.
 | 
						|
 */
 | 
						|
bool FiosDelete(const char *name)
 | 
						|
{
 | 
						|
	char filename[512];
 | 
						|
 | 
						|
	FiosMakeSavegameName(filename, name, lastof(filename));
 | 
						|
	return unlink(filename) == 0;
 | 
						|
}
 | 
						|
 | 
						|
typedef FiosType fios_getlist_callback_proc(SaveLoadOperation fop, const char *filename, const char *ext, char *title, const char *last);
 | 
						|
 | 
						|
/**
 | 
						|
 * Scanner to scan for a particular type of FIOS file.
 | 
						|
 */
 | 
						|
class FiosFileScanner : public FileScanner {
 | 
						|
	SaveLoadOperation fop;   ///< The kind of file we are looking for.
 | 
						|
	fios_getlist_callback_proc *callback_proc; ///< Callback to check whether the file may be added
 | 
						|
	FileList &file_list;     ///< Destination of the found files.
 | 
						|
public:
 | 
						|
	/**
 | 
						|
	 * Create the scanner
 | 
						|
	 * @param fop Purpose of collecting the list.
 | 
						|
	 * @param callback_proc The function that is called where you need to do the filtering.
 | 
						|
	 * @param file_list Destination of the found files.
 | 
						|
	 */
 | 
						|
	FiosFileScanner(SaveLoadOperation fop, fios_getlist_callback_proc *callback_proc, FileList &file_list) :
 | 
						|
			fop(fop), callback_proc(callback_proc), file_list(file_list)
 | 
						|
	{}
 | 
						|
 | 
						|
	bool AddFile(const char *filename, size_t basepath_length, const char *tar_filename) override;
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Try to add a fios item set with the given filename.
 | 
						|
 * @param filename        the full path to the file to read
 | 
						|
 * @param basepath_length amount of characters to chop of before to get a relative filename
 | 
						|
 * @return true if the file is added.
 | 
						|
 */
 | 
						|
bool FiosFileScanner::AddFile(const char *filename, size_t basepath_length, const char *tar_filename)
 | 
						|
{
 | 
						|
	const char *ext = strrchr(filename, '.');
 | 
						|
	if (ext == nullptr) return false;
 | 
						|
 | 
						|
	char fios_title[64];
 | 
						|
	fios_title[0] = '\0'; // reset the title;
 | 
						|
 | 
						|
	FiosType type = this->callback_proc(this->fop, filename, ext, fios_title, lastof(fios_title));
 | 
						|
	if (type == FIOS_TYPE_INVALID) return false;
 | 
						|
 | 
						|
	for (const FiosItem *fios = file_list.Begin(); fios != file_list.End(); fios++) {
 | 
						|
		if (strcmp(fios->name, filename) == 0) return false;
 | 
						|
	}
 | 
						|
 | 
						|
	FiosItem *fios = file_list.Append();
 | 
						|
#ifdef _WIN32
 | 
						|
	// Retrieve the file modified date using GetFileTime rather than stat to work around an obscure MSVC bug that affects Windows XP
 | 
						|
	HANDLE fh = CreateFile(OTTD2FS(filename), GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, nullptr, OPEN_EXISTING, 0, nullptr);
 | 
						|
 | 
						|
	if (fh != INVALID_HANDLE_VALUE) {
 | 
						|
		FILETIME ft;
 | 
						|
		ULARGE_INTEGER ft_int64;
 | 
						|
 | 
						|
		if (GetFileTime(fh, nullptr, nullptr, &ft) != 0) {
 | 
						|
			ft_int64.HighPart = ft.dwHighDateTime;
 | 
						|
			ft_int64.LowPart = ft.dwLowDateTime;
 | 
						|
 | 
						|
			// Convert from hectonanoseconds since 01/01/1601 to seconds since 01/01/1970
 | 
						|
			fios->mtime = ft_int64.QuadPart / 10000000ULL - 11644473600ULL;
 | 
						|
		} else {
 | 
						|
			fios->mtime = 0;
 | 
						|
		}
 | 
						|
 | 
						|
		CloseHandle(fh);
 | 
						|
#else
 | 
						|
	struct stat sb;
 | 
						|
	if (stat(filename, &sb) == 0) {
 | 
						|
		fios->mtime = sb.st_mtime;
 | 
						|
#endif
 | 
						|
	} else {
 | 
						|
		fios->mtime = 0;
 | 
						|
	}
 | 
						|
 | 
						|
	fios->type = type;
 | 
						|
	strecpy(fios->name, filename, lastof(fios->name));
 | 
						|
 | 
						|
	/* If the file doesn't have a title, use its filename */
 | 
						|
	const char *t = fios_title;
 | 
						|
	if (StrEmpty(fios_title)) {
 | 
						|
		t = strrchr(filename, PATHSEPCHAR);
 | 
						|
		t = (t == nullptr) ? filename : (t + 1);
 | 
						|
	}
 | 
						|
	strecpy(fios->title, t, lastof(fios->title));
 | 
						|
	str_validate(fios->title, lastof(fios->title));
 | 
						|
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
 * Fill the list of the files in a directory, according to some arbitrary rule.
 | 
						|
 * @param fop Purpose of collecting the list.
 | 
						|
 * @param callback_proc The function that is called where you need to do the filtering.
 | 
						|
 * @param subdir The directory from where to start (global) searching.
 | 
						|
 * @param file_list Destination of the found files.
 | 
						|
 */
 | 
						|
static void FiosGetFileList(SaveLoadOperation fop, fios_getlist_callback_proc *callback_proc, Subdirectory subdir, FileList &file_list)
 | 
						|
{
 | 
						|
	struct stat sb;
 | 
						|
	struct dirent *dirent;
 | 
						|
	DIR *dir;
 | 
						|
	FiosItem *fios;
 | 
						|
	size_t sort_start;
 | 
						|
	char d_name[sizeof(fios->name)];
 | 
						|
 | 
						|
	file_list.Clear();
 | 
						|
 | 
						|
	/* A parent directory link exists if we are not in the root directory */
 | 
						|
	if (!FiosIsRoot(_fios_path)) {
 | 
						|
		fios = file_list.Append();
 | 
						|
		fios->type = FIOS_TYPE_PARENT;
 | 
						|
		fios->mtime = 0;
 | 
						|
		strecpy(fios->name, "..", lastof(fios->name));
 | 
						|
		strecpy(fios->title, ".. (Parent directory)", lastof(fios->title));
 | 
						|
	}
 | 
						|
 | 
						|
	/* Show subdirectories */
 | 
						|
	if ((dir = ttd_opendir(_fios_path)) != nullptr) {
 | 
						|
		while ((dirent = readdir(dir)) != nullptr) {
 | 
						|
			strecpy(d_name, FS2OTTD(dirent->d_name), lastof(d_name));
 | 
						|
 | 
						|
			/* found file must be directory, but not '.' or '..' */
 | 
						|
			if (FiosIsValidFile(_fios_path, dirent, &sb) && S_ISDIR(sb.st_mode) &&
 | 
						|
					(!FiosIsHiddenFile(dirent) || strncasecmp(d_name, PERSONAL_DIR, strlen(d_name)) == 0) &&
 | 
						|
					strcmp(d_name, ".") != 0 && strcmp(d_name, "..") != 0) {
 | 
						|
				fios = file_list.Append();
 | 
						|
				fios->type = FIOS_TYPE_DIR;
 | 
						|
				fios->mtime = 0;
 | 
						|
				strecpy(fios->name, d_name, lastof(fios->name));
 | 
						|
				seprintf(fios->title, lastof(fios->title), "%s" PATHSEP " (Directory)", d_name);
 | 
						|
				str_validate(fios->title, lastof(fios->title));
 | 
						|
			}
 | 
						|
		}
 | 
						|
		closedir(dir);
 | 
						|
	}
 | 
						|
 | 
						|
	/* Sort the subdirs always by name, ascending, remember user-sorting order */
 | 
						|
	{
 | 
						|
		SortingBits order = _savegame_sort_order;
 | 
						|
		_savegame_sort_order = SORT_BY_NAME | SORT_ASCENDING;
 | 
						|
		std::sort(file_list.files.begin(), file_list.files.end());
 | 
						|
		_savegame_sort_order = order;
 | 
						|
	}
 | 
						|
 | 
						|
	/* This is where to start sorting for the filenames */
 | 
						|
	sort_start = file_list.Length();
 | 
						|
 | 
						|
	/* Show files */
 | 
						|
	FiosFileScanner scanner(fop, callback_proc, file_list);
 | 
						|
	if (subdir == NO_DIRECTORY) {
 | 
						|
		scanner.Scan(nullptr, _fios_path, false);
 | 
						|
	} else {
 | 
						|
		scanner.Scan(nullptr, subdir, true, true);
 | 
						|
	}
 | 
						|
 | 
						|
	std::sort(file_list.files.begin() + sort_start, file_list.files.end());
 | 
						|
 | 
						|
	/* Show drives */
 | 
						|
	FiosGetDrives(file_list);
 | 
						|
 | 
						|
	file_list.Compact();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get the title of a file, which (if exists) is stored in a file named
 | 
						|
 * the same as the data file but with '.title' added to it.
 | 
						|
 * @param file filename to get the title for
 | 
						|
 * @param title the title buffer to fill
 | 
						|
 * @param last the last element in the title buffer
 | 
						|
 * @param subdir the sub directory to search in
 | 
						|
 */
 | 
						|
static void GetFileTitle(const char *file, char *title, const char *last, Subdirectory subdir)
 | 
						|
{
 | 
						|
	char buf[MAX_PATH];
 | 
						|
	strecpy(buf, file, lastof(buf));
 | 
						|
	strecat(buf, ".title", lastof(buf));
 | 
						|
 | 
						|
	FILE *f = FioFOpenFile(buf, "r", subdir);
 | 
						|
	if (f == nullptr) return;
 | 
						|
 | 
						|
	size_t read = fread(title, 1, last - title, f);
 | 
						|
	assert(title + read <= last);
 | 
						|
	title[read] = '\0';
 | 
						|
	str_validate(title, last);
 | 
						|
	FioFCloseFile(f);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Callback for FiosGetFileList. It tells if a file is a savegame or not.
 | 
						|
 * @param fop Purpose of collecting the list.
 | 
						|
 * @param file Name of the file to check.
 | 
						|
 * @param ext A pointer to the extension identifier inside file
 | 
						|
 * @param title Buffer if a callback wants to lookup the title of the file; nullptr to skip the lookup
 | 
						|
 * @param last Last available byte in buffer (to prevent buffer overflows); not used when title == nullptr
 | 
						|
 * @return a FIOS_TYPE_* type of the found file, FIOS_TYPE_INVALID if not a savegame
 | 
						|
 * @see FiosGetFileList
 | 
						|
 * @see FiosGetSavegameList
 | 
						|
 */
 | 
						|
FiosType FiosGetSavegameListCallback(SaveLoadOperation fop, const char *file, const char *ext, char *title, const char *last)
 | 
						|
{
 | 
						|
	/* Show savegame files
 | 
						|
	 * .SAV OpenTTD saved game
 | 
						|
	 * .SS1 Transport Tycoon Deluxe preset game
 | 
						|
	 * .SV1 Transport Tycoon Deluxe (Patch) saved game
 | 
						|
	 * .SV2 Transport Tycoon Deluxe (Patch) saved 2-player game */
 | 
						|
 | 
						|
	/* Don't crash if we supply no extension */
 | 
						|
	if (ext == nullptr) return FIOS_TYPE_INVALID;
 | 
						|
 | 
						|
	if (strcasecmp(ext, ".sav") == 0) {
 | 
						|
		GetFileTitle(file, title, last, SAVE_DIR);
 | 
						|
		return FIOS_TYPE_FILE;
 | 
						|
	}
 | 
						|
 | 
						|
	if (fop == SLO_LOAD) {
 | 
						|
		if (strcasecmp(ext, ".ss1") == 0 || strcasecmp(ext, ".sv1") == 0 ||
 | 
						|
				strcasecmp(ext, ".sv2") == 0) {
 | 
						|
			if (title != nullptr) GetOldSaveGameName(file, title, last);
 | 
						|
			return FIOS_TYPE_OLDFILE;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return FIOS_TYPE_INVALID;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get a list of savegames.
 | 
						|
 * @param fop Purpose of collecting the list.
 | 
						|
 * @param file_list Destination of the found files.
 | 
						|
 * @see FiosGetFileList
 | 
						|
 */
 | 
						|
void FiosGetSavegameList(SaveLoadOperation fop, FileList &file_list)
 | 
						|
{
 | 
						|
	static char *fios_save_path = nullptr;
 | 
						|
	static char *fios_save_path_last = nullptr;
 | 
						|
 | 
						|
	if (fios_save_path == nullptr) {
 | 
						|
		fios_save_path = MallocT<char>(MAX_PATH);
 | 
						|
		fios_save_path_last = fios_save_path + MAX_PATH - 1;
 | 
						|
		FioGetDirectory(fios_save_path, fios_save_path_last, SAVE_DIR);
 | 
						|
	}
 | 
						|
 | 
						|
	_fios_path = fios_save_path;
 | 
						|
	_fios_path_last = fios_save_path_last;
 | 
						|
 | 
						|
	FiosGetFileList(fop, &FiosGetSavegameListCallback, NO_DIRECTORY, file_list);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Callback for FiosGetFileList. It tells if a file is a scenario or not.
 | 
						|
 * @param fop Purpose of collecting the list.
 | 
						|
 * @param file Name of the file to check.
 | 
						|
 * @param ext A pointer to the extension identifier inside file
 | 
						|
 * @param title Buffer if a callback wants to lookup the title of the file
 | 
						|
 * @param last Last available byte in buffer (to prevent buffer overflows)
 | 
						|
 * @return a FIOS_TYPE_* type of the found file, FIOS_TYPE_INVALID if not a scenario
 | 
						|
 * @see FiosGetFileList
 | 
						|
 * @see FiosGetScenarioList
 | 
						|
 */
 | 
						|
static FiosType FiosGetScenarioListCallback(SaveLoadOperation fop, const char *file, const char *ext, char *title, const char *last)
 | 
						|
{
 | 
						|
	/* Show scenario files
 | 
						|
	 * .SCN OpenTTD style scenario file
 | 
						|
	 * .SV0 Transport Tycoon Deluxe (Patch) scenario
 | 
						|
	 * .SS0 Transport Tycoon Deluxe preset scenario */
 | 
						|
	if (strcasecmp(ext, ".scn") == 0) {
 | 
						|
		GetFileTitle(file, title, last, SCENARIO_DIR);
 | 
						|
		return FIOS_TYPE_SCENARIO;
 | 
						|
	}
 | 
						|
 | 
						|
	if (fop == SLO_LOAD) {
 | 
						|
		if (strcasecmp(ext, ".sv0") == 0 || strcasecmp(ext, ".ss0") == 0 ) {
 | 
						|
			GetOldSaveGameName(file, title, last);
 | 
						|
			return FIOS_TYPE_OLD_SCENARIO;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return FIOS_TYPE_INVALID;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get a list of scenarios.
 | 
						|
 * @param fop Purpose of collecting the list.
 | 
						|
 * @param file_list Destination of the found files.
 | 
						|
 * @see FiosGetFileList
 | 
						|
 */
 | 
						|
void FiosGetScenarioList(SaveLoadOperation fop, FileList &file_list)
 | 
						|
{
 | 
						|
	static char *fios_scn_path = nullptr;
 | 
						|
	static char *fios_scn_path_last = nullptr;
 | 
						|
 | 
						|
	/* Copy the default path on first run or on 'New Game' */
 | 
						|
	if (fios_scn_path == nullptr) {
 | 
						|
		fios_scn_path = MallocT<char>(MAX_PATH);
 | 
						|
		fios_scn_path_last = fios_scn_path + MAX_PATH - 1;
 | 
						|
		FioGetDirectory(fios_scn_path, fios_scn_path_last, SCENARIO_DIR);
 | 
						|
	}
 | 
						|
 | 
						|
	_fios_path = fios_scn_path;
 | 
						|
	_fios_path_last = fios_scn_path_last;
 | 
						|
 | 
						|
	char base_path[MAX_PATH];
 | 
						|
	FioGetDirectory(base_path, lastof(base_path), SCENARIO_DIR);
 | 
						|
 | 
						|
	Subdirectory subdir = (fop == SLO_LOAD && strcmp(base_path, _fios_path) == 0) ? SCENARIO_DIR : NO_DIRECTORY;
 | 
						|
	FiosGetFileList(fop, &FiosGetScenarioListCallback, subdir, file_list);
 | 
						|
}
 | 
						|
 | 
						|
static FiosType FiosGetHeightmapListCallback(SaveLoadOperation fop, const char *file, const char *ext, char *title, const char *last)
 | 
						|
{
 | 
						|
	/* Show heightmap files
 | 
						|
	 * .PNG PNG Based heightmap files
 | 
						|
	 * .BMP BMP Based heightmap files
 | 
						|
	 */
 | 
						|
 | 
						|
	FiosType type = FIOS_TYPE_INVALID;
 | 
						|
 | 
						|
#ifdef WITH_PNG
 | 
						|
	if (strcasecmp(ext, ".png") == 0) type = FIOS_TYPE_PNG;
 | 
						|
#endif /* WITH_PNG */
 | 
						|
 | 
						|
	if (strcasecmp(ext, ".bmp") == 0) type = FIOS_TYPE_BMP;
 | 
						|
 | 
						|
	if (type == FIOS_TYPE_INVALID) return FIOS_TYPE_INVALID;
 | 
						|
 | 
						|
	TarFileList::iterator it = _tar_filelist[SCENARIO_DIR].find(file);
 | 
						|
	if (it != _tar_filelist[SCENARIO_DIR].end()) {
 | 
						|
		/* If the file is in a tar and that tar is not in a heightmap
 | 
						|
		 * directory we are for sure not supposed to see it.
 | 
						|
		 * Examples of this are pngs part of documentation within
 | 
						|
		 * collections of NewGRFs or 32 bpp graphics replacement PNGs.
 | 
						|
		 */
 | 
						|
		bool match = false;
 | 
						|
		Searchpath sp;
 | 
						|
		FOR_ALL_SEARCHPATHS(sp) {
 | 
						|
			char buf[MAX_PATH];
 | 
						|
			FioAppendDirectory(buf, lastof(buf), sp, HEIGHTMAP_DIR);
 | 
						|
 | 
						|
			if (strncmp(buf, it->second.tar_filename, strlen(buf)) == 0) {
 | 
						|
				match = true;
 | 
						|
				break;
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		if (!match) return FIOS_TYPE_INVALID;
 | 
						|
	}
 | 
						|
 | 
						|
	GetFileTitle(file, title, last, HEIGHTMAP_DIR);
 | 
						|
 | 
						|
	return type;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get a list of heightmaps.
 | 
						|
 * @param fop Purpose of collecting the list.
 | 
						|
 * @param file_list Destination of the found files.
 | 
						|
 */
 | 
						|
void FiosGetHeightmapList(SaveLoadOperation fop, FileList &file_list)
 | 
						|
{
 | 
						|
	static char *fios_hmap_path = nullptr;
 | 
						|
	static char *fios_hmap_path_last = nullptr;
 | 
						|
 | 
						|
	if (fios_hmap_path == nullptr) {
 | 
						|
		fios_hmap_path = MallocT<char>(MAX_PATH);
 | 
						|
		fios_hmap_path_last = fios_hmap_path + MAX_PATH - 1;
 | 
						|
		FioGetDirectory(fios_hmap_path, fios_hmap_path_last, HEIGHTMAP_DIR);
 | 
						|
	}
 | 
						|
 | 
						|
	_fios_path = fios_hmap_path;
 | 
						|
	_fios_path_last = fios_hmap_path_last;
 | 
						|
 | 
						|
	char base_path[MAX_PATH];
 | 
						|
	FioGetDirectory(base_path, lastof(base_path), HEIGHTMAP_DIR);
 | 
						|
 | 
						|
	Subdirectory subdir = strcmp(base_path, _fios_path) == 0 ? HEIGHTMAP_DIR : NO_DIRECTORY;
 | 
						|
	FiosGetFileList(fop, &FiosGetHeightmapListCallback, subdir, file_list);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get the directory for screenshots.
 | 
						|
 * @return path to screenshots
 | 
						|
 */
 | 
						|
const char *FiosGetScreenshotDir()
 | 
						|
{
 | 
						|
	static char *fios_screenshot_path = nullptr;
 | 
						|
 | 
						|
	if (fios_screenshot_path == nullptr) {
 | 
						|
		fios_screenshot_path = MallocT<char>(MAX_PATH);
 | 
						|
		FioGetDirectory(fios_screenshot_path, fios_screenshot_path + MAX_PATH - 1, SCREENSHOT_DIR);
 | 
						|
	}
 | 
						|
 | 
						|
	return fios_screenshot_path;
 | 
						|
}
 | 
						|
 | 
						|
/** Basic data to distinguish a scenario. Used in the server list window */
 | 
						|
struct ScenarioIdentifier {
 | 
						|
	uint32 scenid;           ///< ID for the scenario (generated by content).
 | 
						|
	uint8 md5sum[16];        ///< MD5 checksum of file.
 | 
						|
	char filename[MAX_PATH]; ///< filename of the file.
 | 
						|
 | 
						|
	bool operator == (const ScenarioIdentifier &other) const
 | 
						|
	{
 | 
						|
		return this->scenid == other.scenid &&
 | 
						|
				memcmp(this->md5sum, other.md5sum, sizeof(this->md5sum)) == 0;
 | 
						|
	}
 | 
						|
 | 
						|
	bool operator != (const ScenarioIdentifier &other) const
 | 
						|
	{
 | 
						|
		return !(*this == other);
 | 
						|
	}
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Scanner to find the unique IDs of scenarios
 | 
						|
 */
 | 
						|
class ScenarioScanner : protected FileScanner, public std::vector<ScenarioIdentifier> {
 | 
						|
	bool scanned; ///< Whether we've already scanned
 | 
						|
public:
 | 
						|
	/** Initialise */
 | 
						|
	ScenarioScanner() : scanned(false) {}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Scan, but only if it's needed.
 | 
						|
	 * @param rescan whether to force scanning even when it's not necessary
 | 
						|
	 */
 | 
						|
	void Scan(bool rescan)
 | 
						|
	{
 | 
						|
		if (this->scanned && !rescan) return;
 | 
						|
 | 
						|
		this->FileScanner::Scan(".id", SCENARIO_DIR, true, true);
 | 
						|
		this->scanned = true;
 | 
						|
	}
 | 
						|
 | 
						|
	bool AddFile(const char *filename, size_t basepath_length, const char *tar_filename) override
 | 
						|
	{
 | 
						|
		FILE *f = FioFOpenFile(filename, "r", SCENARIO_DIR);
 | 
						|
		if (f == nullptr) return false;
 | 
						|
 | 
						|
		ScenarioIdentifier id;
 | 
						|
		int fret = fscanf(f, "%u", &id.scenid);
 | 
						|
		FioFCloseFile(f);
 | 
						|
		if (fret != 1) return false;
 | 
						|
		strecpy(id.filename, filename, lastof(id.filename));
 | 
						|
 | 
						|
		Md5 checksum;
 | 
						|
		uint8 buffer[1024];
 | 
						|
		char basename[MAX_PATH]; ///< \a filename without the extension.
 | 
						|
		size_t len, size;
 | 
						|
 | 
						|
		/* open the scenario file, but first get the name.
 | 
						|
		 * This is safe as we check on extension which
 | 
						|
		 * must always exist. */
 | 
						|
		strecpy(basename, filename, lastof(basename));
 | 
						|
		*strrchr(basename, '.') = '\0';
 | 
						|
		f = FioFOpenFile(basename, "rb", SCENARIO_DIR, &size);
 | 
						|
		if (f == nullptr) return false;
 | 
						|
 | 
						|
		/* calculate md5sum */
 | 
						|
		while ((len = fread(buffer, 1, (size > sizeof(buffer)) ? sizeof(buffer) : size, f)) != 0 && size != 0) {
 | 
						|
			size -= len;
 | 
						|
			checksum.Append(buffer, len);
 | 
						|
		}
 | 
						|
		checksum.Finish(id.md5sum);
 | 
						|
 | 
						|
		FioFCloseFile(f);
 | 
						|
 | 
						|
		include(*this, id);
 | 
						|
		return true;
 | 
						|
	}
 | 
						|
};
 | 
						|
 | 
						|
/** Scanner for scenarios */
 | 
						|
static ScenarioScanner _scanner;
 | 
						|
 | 
						|
/**
 | 
						|
 * Find a given scenario based on its unique ID.
 | 
						|
 * @param ci The content info to compare it to.
 | 
						|
 * @param md5sum Whether to look at the md5sum or the id.
 | 
						|
 * @return The filename of the file, else \c nullptr.
 | 
						|
 */
 | 
						|
const char *FindScenario(const ContentInfo *ci, bool md5sum)
 | 
						|
{
 | 
						|
	_scanner.Scan(false);
 | 
						|
 | 
						|
	for (ScenarioIdentifier &id : _scanner) {
 | 
						|
		if (md5sum ? (memcmp(id.md5sum, ci->md5sum, sizeof(id.md5sum)) == 0)
 | 
						|
		           : (id.scenid == ci->unique_id)) {
 | 
						|
			return id.filename;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return nullptr;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Check whether we've got a given scenario based on its unique ID.
 | 
						|
 * @param ci The content info to compare it to.
 | 
						|
 * @param md5sum Whether to look at the md5sum or the id.
 | 
						|
 * @return True iff we've got the scenario.
 | 
						|
 */
 | 
						|
bool HasScenario(const ContentInfo *ci, bool md5sum)
 | 
						|
{
 | 
						|
	return (FindScenario(ci, md5sum) != nullptr);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Force a (re)scan of the scenarios.
 | 
						|
 */
 | 
						|
void ScanScenarios()
 | 
						|
{
 | 
						|
	_scanner.Scan(true);
 | 
						|
}
 |