2005-07-24 14:12:37 +00:00
|
|
|
/* $Id$ */
|
|
|
|
|
2007-03-03 04:04:22 +00:00
|
|
|
/** @file fileio.cpp Standard In/Out file operations */
|
2007-03-01 01:24:44 +00:00
|
|
|
|
2004-08-09 17:04:08 +00:00
|
|
|
#include "stdafx.h"
|
2005-06-02 19:30:21 +00:00
|
|
|
#include "openttd.h"
|
2005-01-22 20:23:18 +00:00
|
|
|
#include "fileio.h"
|
2005-07-22 07:02:20 +00:00
|
|
|
#include "functions.h"
|
2006-07-31 22:11:34 +00:00
|
|
|
#include "string.h"
|
2005-07-21 22:15:02 +00:00
|
|
|
#include "macros.h"
|
2005-07-21 18:44:27 +00:00
|
|
|
#include "variables.h"
|
2007-02-08 23:46:25 +00:00
|
|
|
#include "debug.h"
|
2007-03-12 15:25:33 +00:00
|
|
|
#include "fios.h"
|
2007-06-17 15:48:57 +00:00
|
|
|
#ifdef WIN32
|
|
|
|
#include <windows.h>
|
|
|
|
#else
|
2007-03-18 18:02:24 +00:00
|
|
|
#include <pwd.h>
|
2007-03-17 11:36:04 +00:00
|
|
|
#include <unistd.h>
|
2007-03-12 15:25:33 +00:00
|
|
|
#endif
|
2007-09-15 15:21:17 +00:00
|
|
|
#include <sys/stat.h>
|
2004-08-09 17:04:08 +00:00
|
|
|
|
|
|
|
/*************************************************/
|
|
|
|
/* FILE IO ROUTINES ******************************/
|
|
|
|
/*************************************************/
|
|
|
|
|
|
|
|
#define FIO_BUFFER_SIZE 512
|
|
|
|
|
2007-03-07 12:11:48 +00:00
|
|
|
struct Fio {
|
2007-10-29 23:02:31 +00:00
|
|
|
byte *buffer, *buffer_end; ///< position pointer in local buffer and last valid byte of buffer
|
|
|
|
uint32 pos; ///< current (system) position in file
|
|
|
|
FILE *cur_fh; ///< current file handle
|
|
|
|
const char *filename; ///< current filename
|
|
|
|
FILE *handles[MAX_FILE_SLOTS]; ///< array of file handles we can have open
|
|
|
|
byte buffer_start[FIO_BUFFER_SIZE]; ///< local buffer when read from file
|
|
|
|
const char *filenames[MAX_FILE_SLOTS]; ///< array of filenames we (should) have open
|
2007-02-08 23:46:25 +00:00
|
|
|
#if defined(LIMITED_FDS)
|
2007-10-29 23:02:31 +00:00
|
|
|
uint open_handles; ///< current amount of open handles
|
|
|
|
uint usage_count[MAX_FILE_SLOTS]; ///< count how many times this file has been opened
|
2007-02-08 23:46:25 +00:00
|
|
|
#endif /* LIMITED_FDS */
|
2007-03-07 12:11:48 +00:00
|
|
|
};
|
2004-08-09 17:04:08 +00:00
|
|
|
|
|
|
|
static Fio _fio;
|
|
|
|
|
2007-03-01 01:24:44 +00:00
|
|
|
/* Get current position in file */
|
2007-03-07 11:47:46 +00:00
|
|
|
uint32 FioGetPos()
|
2004-08-09 17:04:08 +00:00
|
|
|
{
|
|
|
|
return _fio.pos + (_fio.buffer - _fio.buffer_start) - FIO_BUFFER_SIZE;
|
|
|
|
}
|
|
|
|
|
2007-06-13 16:21:11 +00:00
|
|
|
const char *FioGetFilename()
|
|
|
|
{
|
|
|
|
return _fio.filename;
|
|
|
|
}
|
|
|
|
|
2004-08-09 17:04:08 +00:00
|
|
|
void FioSeekTo(uint32 pos, int mode)
|
|
|
|
{
|
|
|
|
if (mode == SEEK_CUR) pos += FioGetPos();
|
|
|
|
_fio.buffer = _fio.buffer_end = _fio.buffer_start + FIO_BUFFER_SIZE;
|
2006-07-31 22:26:11 +00:00
|
|
|
_fio.pos = pos;
|
|
|
|
fseek(_fio.cur_fh, _fio.pos, SEEK_SET);
|
2004-08-09 17:04:08 +00:00
|
|
|
}
|
|
|
|
|
2007-02-08 23:46:25 +00:00
|
|
|
#if defined(LIMITED_FDS)
|
|
|
|
static void FioRestoreFile(int slot)
|
|
|
|
{
|
|
|
|
/* Do we still have the file open, or should we reopen it? */
|
|
|
|
if (_fio.handles[slot] == NULL) {
|
2007-07-28 12:45:27 +00:00
|
|
|
DEBUG(misc, 6, "Restoring file '%s' in slot '%d' from disk", _fio.filenames[slot], slot);
|
|
|
|
FioOpenFile(slot, _fio.filenames[slot]);
|
2007-02-08 23:46:25 +00:00
|
|
|
}
|
|
|
|
_fio.usage_count[slot]++;
|
|
|
|
}
|
|
|
|
#endif /* LIMITED_FDS */
|
|
|
|
|
2007-03-01 01:24:44 +00:00
|
|
|
/* Seek to a file and a position */
|
2007-09-13 18:22:34 +00:00
|
|
|
void FioSeekToFile(uint8 slot, uint32 pos)
|
2004-08-09 17:04:08 +00:00
|
|
|
{
|
2007-02-08 23:46:25 +00:00
|
|
|
FILE *f;
|
|
|
|
#if defined(LIMITED_FDS)
|
|
|
|
/* Make sure we have this file open */
|
2007-09-13 18:22:34 +00:00
|
|
|
FioRestoreFile(slot);
|
2007-02-08 23:46:25 +00:00
|
|
|
#endif /* LIMITED_FDS */
|
2007-09-13 18:22:34 +00:00
|
|
|
f = _fio.handles[slot];
|
2004-08-09 17:04:08 +00:00
|
|
|
assert(f != NULL);
|
|
|
|
_fio.cur_fh = f;
|
2007-09-13 18:22:34 +00:00
|
|
|
_fio.filename = _fio.filenames[slot];
|
|
|
|
FioSeekTo(pos, SEEK_SET);
|
2004-08-09 17:04:08 +00:00
|
|
|
}
|
|
|
|
|
2007-03-07 11:47:46 +00:00
|
|
|
byte FioReadByte()
|
2004-08-09 17:04:08 +00:00
|
|
|
{
|
|
|
|
if (_fio.buffer == _fio.buffer_end) {
|
|
|
|
_fio.pos += FIO_BUFFER_SIZE;
|
|
|
|
fread(_fio.buffer = _fio.buffer_start, 1, FIO_BUFFER_SIZE, _fio.cur_fh);
|
|
|
|
}
|
|
|
|
return *_fio.buffer++;
|
|
|
|
}
|
|
|
|
|
|
|
|
void FioSkipBytes(int n)
|
|
|
|
{
|
2006-02-01 07:36:15 +00:00
|
|
|
for (;;) {
|
2004-08-09 17:04:08 +00:00
|
|
|
int m = min(_fio.buffer_end - _fio.buffer, n);
|
|
|
|
_fio.buffer += m;
|
|
|
|
n -= m;
|
|
|
|
if (n == 0) break;
|
|
|
|
FioReadByte();
|
|
|
|
n--;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-03-07 11:47:46 +00:00
|
|
|
uint16 FioReadWord()
|
2004-08-09 17:04:08 +00:00
|
|
|
{
|
|
|
|
byte b = FioReadByte();
|
|
|
|
return (FioReadByte() << 8) | b;
|
|
|
|
}
|
|
|
|
|
2007-03-07 11:47:46 +00:00
|
|
|
uint32 FioReadDword()
|
2004-08-09 17:04:08 +00:00
|
|
|
{
|
|
|
|
uint b = FioReadWord();
|
|
|
|
return (FioReadWord() << 16) | b;
|
|
|
|
}
|
|
|
|
|
|
|
|
void FioReadBlock(void *ptr, uint size)
|
|
|
|
{
|
|
|
|
FioSeekTo(FioGetPos(), SEEK_SET);
|
|
|
|
_fio.pos += size;
|
|
|
|
fread(ptr, 1, size, _fio.cur_fh);
|
|
|
|
}
|
|
|
|
|
2005-01-16 12:40:25 +00:00
|
|
|
static inline void FioCloseFile(int slot)
|
|
|
|
{
|
|
|
|
if (_fio.handles[slot] != NULL) {
|
2005-01-23 13:09:35 +00:00
|
|
|
fclose(_fio.handles[slot]);
|
2005-01-16 12:40:25 +00:00
|
|
|
_fio.handles[slot] = NULL;
|
2007-02-08 23:46:25 +00:00
|
|
|
#if defined(LIMITED_FDS)
|
|
|
|
_fio.open_handles--;
|
|
|
|
#endif /* LIMITED_FDS */
|
2005-01-16 12:40:25 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-03-07 11:47:46 +00:00
|
|
|
void FioCloseAll()
|
2004-08-09 17:04:08 +00:00
|
|
|
{
|
|
|
|
int i;
|
|
|
|
|
2005-01-16 12:40:25 +00:00
|
|
|
for (i = 0; i != lengthof(_fio.handles); i++)
|
|
|
|
FioCloseFile(i);
|
2004-08-09 17:04:08 +00:00
|
|
|
}
|
|
|
|
|
2007-02-08 23:46:25 +00:00
|
|
|
#if defined(LIMITED_FDS)
|
|
|
|
static void FioFreeHandle()
|
|
|
|
{
|
|
|
|
/* If we are about to open a file that will exceed the limit, close a file */
|
|
|
|
if (_fio.open_handles + 1 == LIMITED_FDS) {
|
|
|
|
uint i, count;
|
|
|
|
int slot;
|
|
|
|
|
|
|
|
count = UINT_MAX;
|
|
|
|
slot = -1;
|
|
|
|
/* Find the file that is used the least */
|
|
|
|
for (i = 0; i < lengthof(_fio.handles); i++) {
|
|
|
|
if (_fio.handles[i] != NULL && _fio.usage_count[i] < count) {
|
|
|
|
count = _fio.usage_count[i];
|
|
|
|
slot = i;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
assert(slot != -1);
|
2007-07-28 12:45:27 +00:00
|
|
|
DEBUG(misc, 6, "Closing filehandler '%s' in slot '%d' because of fd-limit", _fio.filenames[slot], slot);
|
2007-02-08 23:46:25 +00:00
|
|
|
FioCloseFile(slot);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif /* LIMITED_FDS */
|
|
|
|
|
2007-03-12 15:40:12 +00:00
|
|
|
void FioOpenFile(int slot, const char *filename)
|
|
|
|
{
|
|
|
|
FILE *f;
|
|
|
|
|
|
|
|
#if defined(LIMITED_FDS)
|
|
|
|
FioFreeHandle();
|
|
|
|
#endif /* LIMITED_FDS */
|
|
|
|
f = FioFOpenFile(filename);
|
2007-04-04 12:03:10 +00:00
|
|
|
if (f == NULL) error("Cannot open file '%s'", filename);
|
2007-09-13 18:22:34 +00:00
|
|
|
uint32 pos = ftell(f);
|
2007-03-12 15:40:12 +00:00
|
|
|
|
|
|
|
FioCloseFile(slot); // if file was opened before, close it
|
|
|
|
_fio.handles[slot] = f;
|
2007-06-13 16:21:11 +00:00
|
|
|
_fio.filenames[slot] = filename;
|
2007-03-12 15:40:12 +00:00
|
|
|
#if defined(LIMITED_FDS)
|
|
|
|
_fio.usage_count[slot] = 0;
|
|
|
|
_fio.open_handles++;
|
|
|
|
#endif /* LIMITED_FDS */
|
2007-09-13 18:22:34 +00:00
|
|
|
FioSeekToFile(slot, pos);
|
2007-03-12 15:40:12 +00:00
|
|
|
}
|
|
|
|
|
2007-06-17 15:48:57 +00:00
|
|
|
const char *_subdirs[NUM_SUBDIRS] = {
|
|
|
|
"",
|
|
|
|
"save" PATHSEP,
|
|
|
|
"save" PATHSEP "autosave" PATHSEP,
|
|
|
|
"scenario" PATHSEP,
|
|
|
|
"scenario" PATHSEP "heightmap" PATHSEP,
|
|
|
|
"gm" PATHSEP,
|
|
|
|
"data" PATHSEP,
|
|
|
|
"lang" PATHSEP
|
|
|
|
};
|
|
|
|
|
|
|
|
const char *_searchpaths[NUM_SEARCHPATHS];
|
2007-09-16 18:10:52 +00:00
|
|
|
TarList _tar_list;
|
|
|
|
TarFileList _tar_filelist;
|
2007-06-17 15:48:57 +00:00
|
|
|
|
2007-03-12 15:40:12 +00:00
|
|
|
/**
|
|
|
|
* Check whether the given file exists
|
|
|
|
* @param filename the file to try for existance
|
2007-06-17 15:48:57 +00:00
|
|
|
* @param subdir the subdirectory to look in
|
2007-03-12 15:40:12 +00:00
|
|
|
* @return true if and only if the file can be opened
|
|
|
|
*/
|
2007-06-17 15:48:57 +00:00
|
|
|
bool FioCheckFileExists(const char *filename, Subdirectory subdir)
|
2007-03-12 15:40:12 +00:00
|
|
|
{
|
2007-06-17 15:48:57 +00:00
|
|
|
FILE *f = FioFOpenFile(filename, "rb", subdir);
|
2007-03-12 15:40:12 +00:00
|
|
|
if (f == NULL) return false;
|
|
|
|
|
2007-09-16 18:10:52 +00:00
|
|
|
FioFCloseFile(f);
|
2007-03-12 15:40:12 +00:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2007-09-16 18:10:52 +00:00
|
|
|
/**
|
|
|
|
* Close a file in a safe way.
|
|
|
|
*/
|
|
|
|
void FioFCloseFile(FILE *f)
|
|
|
|
{
|
|
|
|
fclose(f);
|
|
|
|
}
|
|
|
|
|
2007-06-17 15:48:57 +00:00
|
|
|
char *FioGetFullPath(char *buf, size_t buflen, Searchpath sp, Subdirectory subdir, const char *filename)
|
2005-12-10 11:16:45 +00:00
|
|
|
{
|
2007-06-17 15:48:57 +00:00
|
|
|
assert(subdir < NUM_SUBDIRS);
|
|
|
|
assert(sp < NUM_SEARCHPATHS);
|
|
|
|
|
2007-06-17 21:26:57 +00:00
|
|
|
snprintf(buf, buflen, "%s%s%s", _searchpaths[sp], _subdirs[subdir], filename);
|
2007-06-17 15:48:57 +00:00
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
|
|
|
char *FioFindFullPath(char *buf, size_t buflen, Subdirectory subdir, const char *filename)
|
|
|
|
{
|
|
|
|
Searchpath sp;
|
|
|
|
assert(subdir < NUM_SUBDIRS);
|
|
|
|
|
|
|
|
FOR_ALL_SEARCHPATHS(sp) {
|
|
|
|
FioGetFullPath(buf, buflen, sp, subdir, filename);
|
|
|
|
if (FileExists(buf)) break;
|
|
|
|
}
|
|
|
|
|
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
|
|
|
char *FioAppendDirectory(char *buf, size_t buflen, Searchpath sp, Subdirectory subdir)
|
|
|
|
{
|
|
|
|
assert(subdir < NUM_SUBDIRS);
|
|
|
|
assert(sp < NUM_SEARCHPATHS);
|
|
|
|
|
|
|
|
snprintf(buf, buflen, "%s%s", _searchpaths[sp], _subdirs[subdir]);
|
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
|
|
|
char *FioGetDirectory(char *buf, size_t buflen, Subdirectory subdir)
|
|
|
|
{
|
|
|
|
Searchpath sp;
|
|
|
|
|
|
|
|
/* Find and return the first valid directory */
|
|
|
|
FOR_ALL_SEARCHPATHS(sp) {
|
|
|
|
char *ret = FioAppendDirectory(buf, buflen, sp, subdir);
|
|
|
|
if (FileExists(buf)) return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Could not find the directory, fall back to a base path */
|
|
|
|
ttd_strlcpy(buf, _personal_dir, buflen);
|
|
|
|
|
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
2007-09-13 18:46:29 +00:00
|
|
|
FILE *FioFOpenFileSp(const char *filename, const char *mode, Searchpath sp, Subdirectory subdir, size_t *filesize)
|
2007-06-17 15:48:57 +00:00
|
|
|
{
|
|
|
|
#if defined(WIN32) && defined(UNICODE)
|
|
|
|
/* fopen is implemented as a define with ellipses for
|
|
|
|
* Unicode support (prepend an L). As we are not sending
|
|
|
|
* a string, but a variable, it 'renames' the variable,
|
|
|
|
* so make that variable to makes it compile happily */
|
|
|
|
wchar_t Lmode[5];
|
|
|
|
MultiByteToWideChar(CP_ACP, 0, mode, -1, Lmode, lengthof(Lmode));
|
|
|
|
#endif
|
|
|
|
FILE *f = NULL;
|
2005-12-10 11:16:45 +00:00
|
|
|
char buf[MAX_PATH];
|
|
|
|
|
2007-06-17 20:09:05 +00:00
|
|
|
if (subdir == NO_DIRECTORY) {
|
2007-04-04 12:03:10 +00:00
|
|
|
ttd_strlcpy(buf, filename, lengthof(buf));
|
2007-06-08 21:16:17 +00:00
|
|
|
} else {
|
2007-06-17 15:48:57 +00:00
|
|
|
snprintf(buf, lengthof(buf), "%s%s%s", _searchpaths[sp], _subdirs[subdir], filename);
|
2007-04-04 12:03:10 +00:00
|
|
|
}
|
2005-12-10 11:16:45 +00:00
|
|
|
|
2007-09-24 19:15:42 +00:00
|
|
|
#if defined(WIN32)
|
2007-09-25 14:15:38 +00:00
|
|
|
if (mode[0] == 'r' && GetFileAttributes(OTTD2FS(buf)) == INVALID_FILE_ATTRIBUTES) return NULL;
|
2007-09-24 19:15:42 +00:00
|
|
|
#endif
|
|
|
|
|
2007-06-17 15:48:57 +00:00
|
|
|
f = fopen(buf, mode);
|
2005-12-10 11:16:45 +00:00
|
|
|
#if !defined(WIN32)
|
|
|
|
if (f == NULL) {
|
2007-06-17 15:48:57 +00:00
|
|
|
strtolower(buf + strlen(_searchpaths[sp]) - 1);
|
|
|
|
f = fopen(buf, mode);
|
2005-12-10 11:16:45 +00:00
|
|
|
}
|
|
|
|
#endif
|
2007-09-13 18:46:29 +00:00
|
|
|
if (f != NULL && filesize != NULL) {
|
|
|
|
/* Find the size of the file */
|
|
|
|
fseek(f, 0, SEEK_END);
|
|
|
|
*filesize = ftell(f);
|
|
|
|
fseek(f, 0, SEEK_SET);
|
|
|
|
}
|
2007-06-17 15:48:57 +00:00
|
|
|
return f;
|
|
|
|
}
|
|
|
|
|
2007-09-16 18:10:52 +00:00
|
|
|
FILE *FioFOpenFileTar(TarFileListEntry *entry, size_t *filesize)
|
2007-09-14 22:25:00 +00:00
|
|
|
{
|
2007-09-16 18:10:52 +00:00
|
|
|
FILE *f = fopen(entry->tar->filename, "rb");
|
2007-09-14 22:25:00 +00:00
|
|
|
assert(f != NULL);
|
|
|
|
|
2007-09-16 18:10:52 +00:00
|
|
|
fseek(f, entry->position, SEEK_SET);
|
|
|
|
if (filesize != NULL) *filesize = entry->size;
|
|
|
|
return f;
|
2007-09-14 22:25:00 +00:00
|
|
|
}
|
|
|
|
|
2007-06-17 15:48:57 +00:00
|
|
|
/** Opens OpenTTD files somewhere in a personal or global directory */
|
2007-09-13 18:50:42 +00:00
|
|
|
FILE *FioFOpenFile(const char *filename, const char *mode, Subdirectory subdir, size_t *filesize)
|
2007-06-17 15:48:57 +00:00
|
|
|
{
|
|
|
|
FILE *f = NULL;
|
|
|
|
Searchpath sp;
|
|
|
|
|
2007-06-17 20:09:05 +00:00
|
|
|
assert(subdir < NUM_SUBDIRS || subdir == NO_DIRECTORY);
|
2007-06-17 15:48:57 +00:00
|
|
|
|
|
|
|
FOR_ALL_SEARCHPATHS(sp) {
|
2007-09-13 18:50:42 +00:00
|
|
|
f = FioFOpenFileSp(filename, mode, sp, subdir, filesize);
|
2007-06-17 20:09:05 +00:00
|
|
|
if (f != NULL || subdir == NO_DIRECTORY) break;
|
2007-06-17 15:48:57 +00:00
|
|
|
}
|
2007-09-16 18:10:52 +00:00
|
|
|
|
2007-09-14 22:25:00 +00:00
|
|
|
/* We can only use .tar in case of data-dir, and read-mode */
|
|
|
|
if (f == NULL && subdir == DATA_DIR && mode[0] == 'r') {
|
2007-09-16 18:10:52 +00:00
|
|
|
/* Filenames in tars are always forced to be lowercase */
|
|
|
|
char *lcfilename = strdup(filename);
|
|
|
|
strtolower(lcfilename);
|
|
|
|
TarFileList::iterator it = _tar_filelist.find(lcfilename);
|
|
|
|
free(lcfilename);
|
|
|
|
if (it != _tar_filelist.end()) {
|
|
|
|
f = FioFOpenFileTar(&((*it).second), filesize);
|
2007-09-14 22:25:00 +00:00
|
|
|
}
|
|
|
|
}
|
2005-12-10 11:16:45 +00:00
|
|
|
|
|
|
|
return f;
|
|
|
|
}
|
|
|
|
|
2007-03-12 15:25:33 +00:00
|
|
|
/**
|
|
|
|
* Create a directory with the given name
|
|
|
|
* @param name the new name of the directory
|
|
|
|
*/
|
|
|
|
void FioCreateDirectory(const char *name)
|
|
|
|
{
|
|
|
|
#if defined(WIN32) || defined(WINCE)
|
|
|
|
CreateDirectory(OTTD2FS(name), NULL);
|
|
|
|
#elif defined(OS2) && !defined(__INNOTEK_LIBC__)
|
|
|
|
mkdir(OTTD2FS(name));
|
2007-11-18 18:28:32 +00:00
|
|
|
#elif defined(__MORPHOS__) || defined(__AMIGAOS__)
|
|
|
|
char buf[MAX_PATH];
|
|
|
|
ttd_strlcpy(buf, name, MAX_PATH);
|
|
|
|
|
|
|
|
size_t len = strlen(name) - 1;
|
|
|
|
if (buf[len] == '/') {
|
|
|
|
buf[len] = '\0'; // Kill pathsep, so mkdir() will not fail
|
|
|
|
}
|
|
|
|
|
|
|
|
mkdir(OTTD2FS(buf), 0755);
|
2007-03-12 15:25:33 +00:00
|
|
|
#else
|
|
|
|
mkdir(OTTD2FS(name), 0755);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Appends, if necessary, the path separator character to the end of the string.
|
|
|
|
* It does not add the path separator to zero-sized strings.
|
|
|
|
* @param buf string to append the separator to
|
|
|
|
* @param buflen the length of the buf
|
|
|
|
*/
|
|
|
|
void AppendPathSeparator(char *buf, size_t buflen)
|
|
|
|
{
|
|
|
|
size_t s = strlen(buf);
|
|
|
|
|
|
|
|
/* Length of string + path separator + '\0' */
|
|
|
|
if (s != 0 && buf[s - 1] != PATHSEPCHAR && s + 2 < buflen) {
|
2007-03-12 15:40:12 +00:00
|
|
|
buf[s] = PATHSEPCHAR;
|
|
|
|
buf[s + 1] = '\0';
|
2007-03-12 15:25:33 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2007-06-08 21:16:17 +00:00
|
|
|
/**
|
|
|
|
* Allocates and files a variable with the full path
|
|
|
|
* based on the given directory.
|
|
|
|
* @param dir the directory to base the path on
|
|
|
|
* @return the malloced full path
|
|
|
|
*/
|
|
|
|
char *BuildWithFullPath(const char *dir)
|
|
|
|
{
|
|
|
|
char *dest = MallocT<char>(MAX_PATH);
|
|
|
|
ttd_strlcpy(dest, dir, MAX_PATH);
|
|
|
|
|
|
|
|
/* Check if absolute or relative path */
|
|
|
|
const char *s = strchr(dest, PATHSEPCHAR);
|
|
|
|
|
|
|
|
/* Add absolute path */
|
|
|
|
if (s == NULL || dest != s) {
|
|
|
|
getcwd(dest, MAX_PATH);
|
|
|
|
AppendPathSeparator(dest, MAX_PATH);
|
|
|
|
ttd_strlcat(dest, dir, MAX_PATH);
|
|
|
|
}
|
|
|
|
AppendPathSeparator(dest, MAX_PATH);
|
|
|
|
|
|
|
|
return dest;
|
|
|
|
}
|
|
|
|
|
2007-09-14 22:25:00 +00:00
|
|
|
static bool TarListAddFile(const char *filename)
|
|
|
|
{
|
2007-09-16 18:10:52 +00:00
|
|
|
/* The TAR-header, repeated for every file */
|
|
|
|
typedef struct TarHeader {
|
|
|
|
char name[100]; ///< Name of the file
|
|
|
|
char mode[8];
|
|
|
|
char uid[8];
|
|
|
|
char gid[8];
|
|
|
|
char size[12]; ///< Size of the file, in ASCII
|
|
|
|
char mtime[12];
|
|
|
|
char chksum[8];
|
|
|
|
char typeflag;
|
|
|
|
char linkname[100];
|
|
|
|
char magic[6];
|
|
|
|
char version[2];
|
|
|
|
char uname[32];
|
|
|
|
char gname[32];
|
|
|
|
char devmajor[8];
|
|
|
|
char devminor[8];
|
|
|
|
char prefix[155]; ///< Path of the file
|
|
|
|
|
|
|
|
char unused[12];
|
|
|
|
} TarHeader;
|
|
|
|
|
|
|
|
/* Check if we already seen this file */
|
|
|
|
TarList::iterator it = _tar_list.find(filename);
|
|
|
|
if (it != _tar_list.end()) return false;
|
|
|
|
|
|
|
|
FILE *f = fopen(filename, "rb");
|
|
|
|
assert(f != NULL);
|
|
|
|
|
|
|
|
TarListEntry *tar_entry = MallocT<TarListEntry>(1);
|
|
|
|
tar_entry->filename = strdup(filename);
|
|
|
|
_tar_list.insert(TarList::value_type(filename, tar_entry));
|
|
|
|
|
|
|
|
TarHeader th;
|
|
|
|
char buf[sizeof(th.name) + 1], *end;
|
|
|
|
char name[sizeof(th.prefix) + 1 + sizeof(th.name) + 1];
|
|
|
|
int num = 0, pos = 0;
|
|
|
|
|
|
|
|
/* Make a char of 512 empty bytes */
|
|
|
|
char empty[512];
|
|
|
|
memset(&empty[0], 0, sizeof(empty));
|
|
|
|
|
|
|
|
while (!feof(f)) {
|
|
|
|
fread(&th, 1, 512, f);
|
|
|
|
pos += 512;
|
|
|
|
|
|
|
|
/* Check if we have the new tar-format (ustar) or the old one (a lot of zeros after 'link' field) */
|
|
|
|
if (strncmp(th.magic, "ustar", 5) != 0 && memcmp(&th.magic, &empty[0], 512 - offsetof(TarHeader, magic)) != 0) {
|
|
|
|
/* If we have only zeros in the block, it can be an end-of-file indicator */
|
|
|
|
if (memcmp(&th, &empty[0], 512) == 0) continue;
|
|
|
|
|
|
|
|
DEBUG(misc, 0, "The file '%s' isn't a valid tar-file", filename);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
name[0] = '\0';
|
|
|
|
int len = 0;
|
|
|
|
|
|
|
|
/* The prefix contains the directory-name */
|
|
|
|
if (th.prefix[0] != '\0') {
|
|
|
|
memcpy(name, th.prefix, sizeof(th.prefix));
|
|
|
|
name[sizeof(th.prefix)] = '\0';
|
|
|
|
len = strlen(name);
|
|
|
|
name[len] = PATHSEPCHAR;
|
|
|
|
len++;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Copy the name of the file in a safe way at the end of 'name' */
|
|
|
|
memcpy(&name[len], th.name, sizeof(th.name));
|
|
|
|
name[len + sizeof(th.name)] = '\0';
|
|
|
|
|
|
|
|
/* Calculate the size of the file.. for some strange reason this is stored as a string */
|
|
|
|
memcpy(buf, th.size, sizeof(th.size));
|
|
|
|
buf[sizeof(th.size)] = '\0';
|
|
|
|
int skip = strtol(buf, &end, 8);
|
|
|
|
|
|
|
|
/* 0 byte sized files can be skipped (dirs, symlinks, ..) */
|
|
|
|
if (skip == 0) continue;
|
|
|
|
|
|
|
|
/* Store this entry in the list */
|
|
|
|
TarFileListEntry entry;
|
|
|
|
entry.tar = tar_entry;
|
|
|
|
entry.size = skip;
|
|
|
|
entry.position = pos;
|
|
|
|
/* Force lowercase */
|
|
|
|
strtolower(name);
|
|
|
|
|
|
|
|
/* Tar-files always have '/' path-seperator, but we want our PATHSEPCHAR */
|
|
|
|
#if (PATHSEPCHAR != '/')
|
|
|
|
for (char *n = name; *n != '\0'; n++) if (*n == '/') *n = PATHSEPCHAR;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
DEBUG(misc, 6, "Found file in tar: %s (%d bytes, %d offset)", name, skip, pos);
|
|
|
|
if (_tar_filelist.insert(TarFileList::value_type(name, entry)).second) num++;
|
|
|
|
|
|
|
|
/* Skip to the next block.. */
|
2007-11-19 20:40:14 +00:00
|
|
|
skip = Align(skip, 512);
|
2007-09-16 18:10:52 +00:00
|
|
|
fseek(f, skip, SEEK_CUR);
|
|
|
|
pos += skip;
|
2007-09-14 22:25:00 +00:00
|
|
|
}
|
|
|
|
|
2007-09-16 18:10:52 +00:00
|
|
|
DEBUG(misc, 1, "Found tar '%s' with %d new files", filename, num);
|
|
|
|
fclose(f);
|
2007-09-14 22:25:00 +00:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int ScanPathForTarFiles(const char *path, int basepath_length)
|
|
|
|
{
|
|
|
|
extern bool FiosIsValidFile(const char *path, const struct dirent *ent, struct stat *sb);
|
|
|
|
|
|
|
|
uint num = 0;
|
|
|
|
struct stat sb;
|
|
|
|
struct dirent *dirent;
|
|
|
|
DIR *dir;
|
|
|
|
|
|
|
|
if (path == NULL || (dir = ttd_opendir(path)) == NULL) return 0;
|
|
|
|
|
|
|
|
while ((dirent = readdir(dir)) != NULL) {
|
|
|
|
const char *d_name = FS2OTTD(dirent->d_name);
|
|
|
|
char filename[MAX_PATH];
|
|
|
|
|
|
|
|
if (!FiosIsValidFile(path, dirent, &sb)) continue;
|
|
|
|
|
|
|
|
snprintf(filename, lengthof(filename), "%s%s", path, d_name);
|
|
|
|
|
|
|
|
if (sb.st_mode & S_IFDIR) {
|
|
|
|
/* Directory */
|
|
|
|
if (strcmp(d_name, ".") == 0 || strcmp(d_name, "..") == 0) continue;
|
|
|
|
AppendPathSeparator(filename, lengthof(filename));
|
|
|
|
num += ScanPathForTarFiles(filename, basepath_length);
|
|
|
|
} else if (sb.st_mode & S_IFREG) {
|
|
|
|
/* File */
|
|
|
|
char *ext = strrchr(filename, '.');
|
|
|
|
|
|
|
|
/* If no extension or extension isn't .tar, skip the file */
|
|
|
|
if (ext == NULL) continue;
|
|
|
|
if (strcasecmp(ext, ".tar") != 0) continue;
|
|
|
|
|
|
|
|
if (TarListAddFile(filename)) num++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
closedir(dir);
|
|
|
|
return num;
|
|
|
|
}
|
|
|
|
|
2007-09-15 15:21:17 +00:00
|
|
|
void ScanForTarFiles()
|
2007-09-14 22:25:00 +00:00
|
|
|
{
|
|
|
|
Searchpath sp;
|
|
|
|
char path[MAX_PATH];
|
|
|
|
uint num = 0;
|
|
|
|
|
|
|
|
DEBUG(misc, 1, "Scanning for tars");
|
|
|
|
FOR_ALL_SEARCHPATHS(sp) {
|
|
|
|
FioAppendDirectory(path, MAX_PATH, sp, DATA_DIR);
|
|
|
|
num += ScanPathForTarFiles(path, strlen(path));
|
|
|
|
}
|
|
|
|
DEBUG(misc, 1, "Scan complete, found %d files", num);
|
|
|
|
}
|
|
|
|
|
2007-09-15 15:21:17 +00:00
|
|
|
#if defined(WIN32) || defined(WINCE)
|
|
|
|
/**
|
|
|
|
* Determine the base (personal dir and game data dir) paths
|
|
|
|
* @param exe the path from the current path to the executable
|
|
|
|
* @note defined in the OS related files (os2.cpp, win32.cpp, unix.cpp etc)
|
|
|
|
*/
|
|
|
|
extern void DetermineBasePaths(const char *exe);
|
|
|
|
#else /* defined(WIN32) || defined(WINCE) */
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Changes the working directory to the path of the give executable.
|
|
|
|
* For OSX application bundles '.app' is the required extension of the bundle,
|
|
|
|
* so when we crop the path to there, when can remove the name of the bundle
|
|
|
|
* in the same way we remove the name from the executable name.
|
|
|
|
* @param exe the path to the executable
|
|
|
|
*/
|
|
|
|
void ChangeWorkingDirectory(const char *exe)
|
|
|
|
{
|
|
|
|
#ifdef WITH_COCOA
|
|
|
|
char *app_bundle = strchr(exe, '.');
|
|
|
|
while (app_bundle != NULL && strncasecmp(app_bundle, ".app", 4) != 0) app_bundle = strchr(&app_bundle[1], '.');
|
|
|
|
|
|
|
|
if (app_bundle != NULL) app_bundle[0] = '\0';
|
|
|
|
#endif /* WITH_COCOA */
|
|
|
|
char *s = strrchr(exe, PATHSEPCHAR);
|
|
|
|
if (s != NULL) {
|
|
|
|
*s = '\0';
|
|
|
|
chdir(exe);
|
|
|
|
*s = PATHSEPCHAR;
|
|
|
|
}
|
|
|
|
#ifdef WITH_COCOA
|
|
|
|
if (app_bundle != NULL) app_bundle[0] = '.';
|
|
|
|
#endif /* WITH_COCOA */
|
|
|
|
}
|
|
|
|
|
2007-03-12 15:25:33 +00:00
|
|
|
/**
|
|
|
|
* Determine the base (personal dir and game data dir) paths
|
2007-03-17 11:36:04 +00:00
|
|
|
* @param exe the path to the executable
|
2007-03-12 15:25:33 +00:00
|
|
|
*/
|
2007-03-17 11:36:04 +00:00
|
|
|
void DetermineBasePaths(const char *exe)
|
|
|
|
{
|
2007-06-17 15:48:57 +00:00
|
|
|
char tmp[MAX_PATH];
|
2007-06-21 14:32:27 +00:00
|
|
|
#if defined(__MORPHOS__) || defined(__AMIGA__) || !defined(WITH_PERSONAL_DIR)
|
|
|
|
_searchpaths[SP_PERSONAL_DIR] = NULL;
|
|
|
|
#else
|
2007-03-17 11:36:04 +00:00
|
|
|
const char *homedir = getenv("HOME");
|
|
|
|
|
|
|
|
if (homedir == NULL) {
|
|
|
|
const struct passwd *pw = getpwuid(getuid());
|
2007-06-17 15:48:57 +00:00
|
|
|
homedir = (pw == NULL) ? "" : pw->pw_dir;
|
2007-03-17 11:36:04 +00:00
|
|
|
}
|
|
|
|
|
2007-06-17 15:48:57 +00:00
|
|
|
snprintf(tmp, MAX_PATH, "%s" PATHSEP "%s", homedir, PERSONAL_DIR);
|
|
|
|
AppendPathSeparator(tmp, MAX_PATH);
|
|
|
|
|
|
|
|
_searchpaths[SP_PERSONAL_DIR] = strdup(tmp);
|
|
|
|
#endif
|
|
|
|
_searchpaths[SP_SHARED_DIR] = NULL;
|
|
|
|
|
2007-06-21 14:32:27 +00:00
|
|
|
#if defined(__MORPHOS__) || defined(__AMIGA__)
|
|
|
|
_searchpaths[SP_WORKING_DIR] = NULL;
|
|
|
|
#else
|
2007-06-17 15:48:57 +00:00
|
|
|
getcwd(tmp, MAX_PATH);
|
|
|
|
AppendPathSeparator(tmp, MAX_PATH);
|
|
|
|
_searchpaths[SP_WORKING_DIR] = strdup(tmp);
|
2007-06-21 14:32:27 +00:00
|
|
|
#endif
|
2007-06-17 15:48:57 +00:00
|
|
|
|
|
|
|
/* Change the working directory to that one of the executable */
|
|
|
|
ChangeWorkingDirectory((char*)exe);
|
|
|
|
getcwd(tmp, MAX_PATH);
|
|
|
|
AppendPathSeparator(tmp, MAX_PATH);
|
|
|
|
_searchpaths[SP_BINARY_DIR] = strdup(tmp);
|
|
|
|
|
2007-06-21 14:32:27 +00:00
|
|
|
#if defined(__MORPHOS__) || defined(__AMIGA__)
|
|
|
|
_searchpaths[SP_INSTALLATION_DIR] = NULL;
|
|
|
|
#else
|
2007-06-17 15:48:57 +00:00
|
|
|
snprintf(tmp, MAX_PATH, "%s", GLOBAL_DATA_DIR);
|
|
|
|
AppendPathSeparator(tmp, MAX_PATH);
|
|
|
|
_searchpaths[SP_INSTALLATION_DIR] = strdup(tmp);
|
2007-06-21 14:32:27 +00:00
|
|
|
#endif
|
2007-06-17 15:48:57 +00:00
|
|
|
#ifdef WITH_COCOA
|
|
|
|
extern void cocoaSetApplicationBundleDir();
|
|
|
|
cocoaSetApplicationBundleDir();
|
|
|
|
#else
|
|
|
|
_searchpaths[SP_APPLICATION_BUNDLE_DIR] = NULL;
|
|
|
|
#endif
|
2007-09-14 22:25:00 +00:00
|
|
|
|
|
|
|
ScanForTarFiles();
|
2007-03-17 11:36:04 +00:00
|
|
|
}
|
|
|
|
#endif /* defined(WIN32) || defined(WINCE) */
|
2007-03-12 15:25:33 +00:00
|
|
|
|
2007-06-17 15:48:57 +00:00
|
|
|
char *_personal_dir;
|
|
|
|
|
2007-03-12 15:25:33 +00:00
|
|
|
/**
|
|
|
|
* Acquire the base paths (personal dir and game data dir),
|
|
|
|
* fill all other paths (save dir, autosave dir etc) and
|
|
|
|
* make the save and scenario directories.
|
2007-06-17 15:48:57 +00:00
|
|
|
* @param exe the path from the current path to the executable
|
2007-03-12 15:25:33 +00:00
|
|
|
*/
|
2007-03-17 11:36:04 +00:00
|
|
|
void DeterminePaths(const char *exe)
|
2007-03-12 15:25:33 +00:00
|
|
|
{
|
2007-03-17 11:36:04 +00:00
|
|
|
DetermineBasePaths(exe);
|
2007-03-12 15:25:33 +00:00
|
|
|
|
2007-06-17 15:48:57 +00:00
|
|
|
Searchpath sp;
|
|
|
|
FOR_ALL_SEARCHPATHS(sp) DEBUG(misc, 4, "%s added as search path", _searchpaths[sp]);
|
|
|
|
|
2007-06-17 21:26:57 +00:00
|
|
|
if (_config_file != NULL) {
|
|
|
|
_personal_dir = strdup(_config_file);
|
|
|
|
char *end = strrchr(_personal_dir , PATHSEPCHAR);
|
|
|
|
if (end == NULL) {
|
|
|
|
_personal_dir[0] = '\0';
|
|
|
|
} else {
|
|
|
|
end[1] = '\0';
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
char personal_dir[MAX_PATH];
|
|
|
|
FioFindFullPath(personal_dir, lengthof(personal_dir), BASE_DIR, "openttd.cfg");
|
|
|
|
|
|
|
|
if (FileExists(personal_dir)) {
|
|
|
|
char *end = strrchr(personal_dir, PATHSEPCHAR);
|
|
|
|
if (end != NULL) end[1] = '\0';
|
|
|
|
_personal_dir = strdup(personal_dir);
|
|
|
|
_config_file = str_fmt("%sopenttd.cfg", _personal_dir);
|
|
|
|
} else {
|
|
|
|
static const Searchpath new_openttd_cfg_order[] = {
|
|
|
|
SP_PERSONAL_DIR, SP_BINARY_DIR, SP_WORKING_DIR, SP_SHARED_DIR, SP_INSTALLATION_DIR
|
|
|
|
};
|
|
|
|
|
|
|
|
for (uint i = 0; i < lengthof(new_openttd_cfg_order); i++) {
|
|
|
|
if (IsValidSearchPath(new_openttd_cfg_order[i])) {
|
|
|
|
_personal_dir = strdup(_searchpaths[new_openttd_cfg_order[i]]);
|
|
|
|
_config_file = str_fmt("%sopenttd.cfg", _personal_dir);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2007-06-17 15:48:57 +00:00
|
|
|
}
|
2007-03-12 15:25:33 +00:00
|
|
|
|
2007-06-17 21:26:57 +00:00
|
|
|
DEBUG(misc, 3, "%s found as personal directory", _personal_dir);
|
2007-03-12 15:25:33 +00:00
|
|
|
|
2007-06-17 15:48:57 +00:00
|
|
|
_highscore_file = str_fmt("%shs.dat", _personal_dir);
|
|
|
|
_log_file = str_fmt("%sopenttd.log", _personal_dir);
|
|
|
|
|
|
|
|
char *save_dir = str_fmt("%s%s", _personal_dir, FioGetSubdirectory(SAVE_DIR));
|
|
|
|
char *autosave_dir = str_fmt("%s%s", _personal_dir, FioGetSubdirectory(AUTOSAVE_DIR));
|
|
|
|
|
|
|
|
/* Make the necessary folders */
|
2007-11-18 18:28:32 +00:00
|
|
|
#if !defined(__MORPHOS__) && !defined(__AMIGA__) && defined(WITH_PERSONAL_DIR)
|
2007-06-17 15:48:57 +00:00
|
|
|
FioCreateDirectory(_personal_dir);
|
2007-11-18 18:28:32 +00:00
|
|
|
#endif
|
|
|
|
|
2007-06-17 15:48:57 +00:00
|
|
|
FioCreateDirectory(save_dir);
|
|
|
|
FioCreateDirectory(autosave_dir);
|
2007-03-12 15:25:33 +00:00
|
|
|
|
2007-06-17 15:48:57 +00:00
|
|
|
free(save_dir);
|
|
|
|
free(autosave_dir);
|
2007-03-12 15:25:33 +00:00
|
|
|
}
|
2007-06-12 15:46:34 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Sanitizes a filename, i.e. removes all illegal characters from it.
|
|
|
|
* @param filename the "\0" terminated filename
|
|
|
|
*/
|
|
|
|
void SanitizeFilename(char *filename)
|
|
|
|
{
|
|
|
|
for (; *filename != '\0'; filename++) {
|
|
|
|
switch (*filename) {
|
|
|
|
/* The following characters are not allowed in filenames
|
|
|
|
* on at least one of the supported operating systems: */
|
|
|
|
case ':': case '\\': case '*': case '?': case '/':
|
2007-06-22 19:30:59 +00:00
|
|
|
case '<': case '>': case '|': case '"':
|
2007-06-12 15:46:34 +00:00
|
|
|
*filename = '_';
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|