/*
DarkPlaces file system
- Copyright (C) 2003 Mathieu Olivier
+ Copyright (C) 2003-2005 Mathieu Olivier
Copyright (C) 1999,2000 contributors of the QuakeForge project
This program is free software; you can redistribute it and/or
#include "quakedef.h"
-#include <stdlib.h>
-#include <string.h>
#include <limits.h>
#include <fcntl.h>
# include <unistd.h>
#endif
-#ifndef PATH_MAX
-# define PATH_MAX 512
-#endif
-
#include "fs.h"
+// Win32 requires us to add O_BINARY, but the other OSes don't have it
+#ifndef O_BINARY
+# define O_BINARY 0
+#endif
+
/*
all generated files (savegames, screenshots, demos, config files) will be
saved to. This can be overridden with the "-game" command line parameter.
The game directory can never be changed while quake is executing. This is a
-precacution against having a malicious server instruct clients to write files
+precaution against having a malicious server instruct clients to write files
over areas they shouldn't.
*/
} z_stream;
-// Our own file structure on top of FILE
typedef enum
{
- FS_FLAG_NONE = 0,
- FS_FLAG_PACKED = (1 << 0), // inside a package (PAK or PK3)
- FS_FLAG_DEFLATED = (1 << 1) // file is compressed using the deflate algorithm (PK3 only)
-} fs_flags_t;
+ QFILE_FLAG_NONE = 0,
+ QFILE_FLAG_PACKED = (1 << 0), // inside a package (PAK or PK3)
+ QFILE_FLAG_DEFLATED = (1 << 1) // file is compressed using the deflate algorithm (PK3 only)
+} qfile_flags_t;
-#define ZBUFF_SIZE 1024
+#define FILE_BUFF_SIZE 2048
typedef struct
{
z_stream zstream;
- size_t real_length; // length of the uncompressed file
- size_t in_ind, in_max; // input buffer index and counter
+ size_t comp_length; // length of the compressed file
+ size_t in_ind, in_len; // input buffer current index and length
size_t in_position; // position in the compressed file
- size_t out_ind, out_max; // output buffer index and counter
- size_t out_position; // how many bytes did we uncompress until now?
- qbyte input [ZBUFF_SIZE];
- qbyte output [ZBUFF_SIZE];
+ qbyte input [FILE_BUFF_SIZE];
} ztoolkit_t;
struct qfile_s
{
- fs_flags_t flags;
- FILE* stream;
- size_t length; // file size on disk (PACKED only)
- size_t offset; // offset into a package (PACKED only)
- size_t position; // current position in the file (PACKED only)
- ztoolkit_t* z; // used for inflating (DEFLATED only)
+ qfile_flags_t flags;
+ int handle; // file descriptor
+ size_t real_length; // uncompressed file size (for files opened in "read" mode)
+ size_t position; // current position in the file
+ size_t offset; // offset into the package (0 if external file)
+ int ungetc; // single stored character from ungetc, cleared to EOF when read
+
+ // Contents buffer
+ size_t buff_ind, buff_len; // buffer current index and length
+ qbyte buff [FILE_BUFF_SIZE];
+
+ // For zipped files
+ ztoolkit_t* ztk;
};
// Packages in memory
typedef enum
{
- FILE_FLAG_NONE = 0,
- FILE_FLAG_TRUEOFFS = (1 << 0), // the offset in packfile_t is the true contents offset
- FILE_FLAG_DEFLATED = (1 << 1) // file compressed using the deflate algorithm
-} file_flags_t;
+ PACKFILE_FLAG_NONE = 0,
+ PACKFILE_FLAG_TRUEOFFS = (1 << 0), // the offset in packfile_t is the true contents offset
+ PACKFILE_FLAG_DEFLATED = (1 << 1) // file compressed using the deflate algorithm
+} packfile_flags_t;
typedef struct
{
char name [MAX_QPATH];
- file_flags_t flags;
+ packfile_flags_t flags;
size_t offset;
size_t packsize; // size in the package
size_t realsize; // real file size (uncompressed)
typedef struct pack_s
{
char filename [MAX_OSPATH];
- FILE *handle;
- int ignorecase; // PK3 ignores case
+ int handle;
+ int ignorecase; // PK3 ignores case
int numfiles;
packfile_t *files;
mempool_t *mempool;
static packfile_t* FS_AddFileToPack (const char* name, pack_t* pack,
size_t offset, size_t packsize,
- size_t realsize, file_flags_t flags);
+ size_t realsize, packfile_flags_t flags);
/*
pack_t *packlist = NULL;
-searchpath_t *fs_searchpaths;
+searchpath_t *fs_searchpaths = NULL;
#define MAX_FILES_IN_PACK 65536
*/
void PK3_CloseLibrary (void)
{
- if (!zlib_dll)
- return;
-
- Sys_UnloadLibrary (zlib_dll);
- zlib_dll = NULL;
+ Sys_UnloadLibrary (&zlib_dll);
}
*/
qboolean PK3_OpenLibrary (void)
{
- const char* dllname;
- const dllfunction_t *func;
-
- // Already loaded?
- if (zlib_dll)
- return true;
-
+ const char* dllnames [] =
+ {
#ifdef WIN32
- dllname = "zlib.dll";
+ "zlib.dll",
+#elif defined(MACOSX)
+ "libz.dylib",
#else
- dllname = "libz.so";
+ "libz.so.1",
+ "libz.so",
#endif
+ NULL
+ };
- // Initializations
- for (func = zlibfuncs; func && func->name != NULL; func++)
- *func->funcvariable = NULL;
+ // Already loaded?
+ if (zlib_dll)
+ return true;
// Load the DLL
- if (! (zlib_dll = Sys_LoadLibrary (dllname)))
+ if (! Sys_LoadLibrary (dllnames, &zlib_dll, zlibfuncs))
{
- Con_Printf("Can't find %s. Compressed files support disabled\n", dllname);
+ Con_Printf ("Compressed files support disabled\n");
return false;
}
- // Get the function adresses
- for (func = zlibfuncs; func && func->name != NULL; func++)
- if (!(*func->funcvariable = (void *) Sys_GetProcAddress (zlib_dll, func->name)))
- {
- Con_Printf("missing function \"%s\" - broken Zlib library!\n", func->name);
- PK3_CloseLibrary ();
- return false;
- }
-
- Con_Printf("%s loaded. Compressed files support enabled\n", dllname);
+ Con_Printf ("Compressed files support enabled\n");
return true;
}
Extract the end of the central directory from a PK3 package
====================
*/
-qboolean PK3_GetEndOfCentralDir (const char *packfile, FILE *packhandle, pk3_endOfCentralDir_t *eocd)
+qboolean PK3_GetEndOfCentralDir (const char *packfile, int packhandle, pk3_endOfCentralDir_t *eocd)
{
long filesize, maxsize;
qbyte *buffer, *ptr;
int ind;
// Get the package size
- fseek (packhandle, 0, SEEK_END);
- filesize = ftell (packhandle);
+ filesize = lseek (packhandle, 0, SEEK_END);
if (filesize < ZIP_END_CDIR_SIZE)
return false;
else
maxsize = ZIP_MAX_COMMENTS_SIZE + ZIP_END_CDIR_SIZE;
buffer = Mem_Alloc (tempmempool, maxsize);
- fseek (packhandle, filesize - maxsize, SEEK_SET);
- if (fread (buffer, 1, maxsize, packhandle) != (unsigned long) maxsize)
+ lseek (packhandle, filesize - maxsize, SEEK_SET);
+ if (read (packhandle, buffer, maxsize) != (ssize_t) maxsize)
{
Mem_Free (buffer);
return false;
// Load the central directory in memory
central_dir = Mem_Alloc (tempmempool, eocd->cdir_size);
- fseek (pack->handle, eocd->cdir_offset, SEEK_SET);
- fread (central_dir, 1, eocd->cdir_size, pack->handle);
+ lseek (pack->handle, eocd->cdir_offset, SEEK_SET);
+ read (pack->handle, central_dir, eocd->cdir_size);
// Extract the files properties
// The parsing is done "by hand" because some fields have variable sizes and
{
char filename [sizeof (pack->files[0].name)];
size_t offset, packsize, realsize;
- file_flags_t flags;
+ packfile_flags_t flags;
// Extract the name (strip it if necessary)
if (namesize >= sizeof (filename))
filename[namesize] = '\0';
if (BuffLittleShort (&ptr[10]))
- flags = FILE_FLAG_DEFLATED;
+ flags = PACKFILE_FLAG_DEFLATED;
else
flags = 0;
offset = BuffLittleLong (&ptr[42]);
remaining -= count;
}
- Mem_Free (central_dir);
+ // If the package is empty, central_dir is NULL here
+ if (central_dir != NULL)
+ Mem_Free (central_dir);
return pack->numfiles;
}
*/
pack_t *FS_LoadPackPK3 (const char *packfile)
{
- FILE *packhandle;
+ int packhandle;
pk3_endOfCentralDir_t eocd;
pack_t *pack;
int real_nb_files;
- packhandle = fopen (packfile, "rb");
- if (!packhandle)
+ packhandle = open (packfile, O_RDONLY | O_BINARY);
+ if (packhandle < 0)
return NULL;
if (! PK3_GetEndOfCentralDir (packfile, packhandle, &eocd))
// We only need to do this test if MAX_FILES_IN_PACK is lesser than 65535
// since eocd.nbentries is an unsigned 16 bits integer
- #if MAX_FILES_IN_PACK < 65535
+#if MAX_FILES_IN_PACK < 65535
if (eocd.nbentries > MAX_FILES_IN_PACK)
Sys_Error ("%s contains too many files (%hu)", packfile, eocd.nbentries);
- #endif
+#endif
// Create a package structure in memory
pack = Mem_Alloc (pak_mempool, sizeof (pack_t));
strlcpy (pack->filename, packfile, sizeof (pack->filename));
pack->handle = packhandle;
pack->numfiles = eocd.nbentries;
- pack->mempool = Mem_AllocPool (packfile);
+ pack->mempool = Mem_AllocPool (packfile, 0, NULL);
pack->files = Mem_Alloc (pack->mempool, eocd.nbentries * sizeof(packfile_t));
pack->next = packlist;
packlist = pack;
real_nb_files = PK3_BuildFileList (pack, &eocd);
- if (real_nb_files <= 0)
+ if (real_nb_files < 0)
Sys_Error ("%s is not a valid PK3 file", packfile);
Con_Printf("Added packfile %s (%i files)\n", packfile, real_nb_files);
Find where the true file data offset is
====================
*/
-void PK3_GetTrueFileOffset (packfile_t *file, pack_t *pack)
+void PK3_GetTrueFileOffset (packfile_t *pfile, pack_t *pack)
{
qbyte buffer [ZIP_LOCAL_CHUNK_BASE_SIZE];
size_t count;
// Already found?
- if (file->flags & FILE_FLAG_TRUEOFFS)
+ if (pfile->flags & PACKFILE_FLAG_TRUEOFFS)
return;
// Load the local file description
- fseek (pack->handle, file->offset, SEEK_SET);
- count = fread (buffer, 1, ZIP_LOCAL_CHUNK_BASE_SIZE, pack->handle);
+ lseek (pack->handle, pfile->offset, SEEK_SET);
+ count = read (pack->handle, buffer, ZIP_LOCAL_CHUNK_BASE_SIZE);
if (count != ZIP_LOCAL_CHUNK_BASE_SIZE || BuffBigLong (buffer) != ZIP_DATA_HEADER)
- Sys_Error ("Can't retrieve file %s in package %s", file->name, pack->filename);
+ Sys_Error ("Can't retrieve file %s in package %s", pfile->name, pack->filename);
// Skip name and extra field
- file->offset += BuffLittleShort (&buffer[26]) + BuffLittleShort (&buffer[28]) + ZIP_LOCAL_CHUNK_BASE_SIZE;
+ pfile->offset += BuffLittleShort (&buffer[26]) + BuffLittleShort (&buffer[28]) + ZIP_LOCAL_CHUNK_BASE_SIZE;
- file->flags |= FILE_FLAG_TRUEOFFS;
+ pfile->flags |= PACKFILE_FLAG_TRUEOFFS;
}
*/
static packfile_t* FS_AddFileToPack (const char* name, pack_t* pack,
size_t offset, size_t packsize,
- size_t realsize, file_flags_t flags)
+ size_t realsize, packfile_flags_t flags)
{
int (*strcmp_funct) (const char* str1, const char* str2);
- size_t left, right, middle;
- int diff;
- packfile_t *file;
+ int left, right, middle;
+ packfile_t *pfile;
strcmp_funct = pack->ignorecase ? strcasecmp : strcmp;
// Look for the slot we should put that file into (binary search)
left = 0;
- right = pack->numfiles;
- while (left != right)
+ right = pack->numfiles - 1;
+ while (left <= right)
{
- middle = (left + right - 1) / 2;
+ int diff;
+
+ middle = (left + right) / 2;
diff = strcmp_funct (pack->files[middle].name, name);
// If we found the file, there's a problem
if (!diff)
- Sys_Error ("Package %s contains several time the file %s\n",
+ Sys_Error ("Package %s contains the file %s several times\n",
pack->filename, name);
// If we're too far in the list
if (diff > 0)
- right = middle;
+ right = middle - 1;
else
left = middle + 1;
}
// We have to move the right of the list by one slot to free the one we need
- file = &pack->files[left];
- memmove (file + 1, file, (pack->numfiles - left) * sizeof (*file));
+ pfile = &pack->files[left];
+ memmove (pfile + 1, pfile, (pack->numfiles - left) * sizeof (*pfile));
pack->numfiles++;
- strlcpy (file->name, name, sizeof (file->name));
- file->offset = offset;
- file->packsize = packsize;
- file->realsize = realsize;
- file->flags = flags;
+ strlcpy (pfile->name, name, sizeof (pfile->name));
+ pfile->offset = offset;
+ pfile->packsize = packsize;
+ pfile->realsize = realsize;
+ pfile->flags = flags;
- return file;
+ return pfile;
}
{
dpackheader_t header;
int i, numpackfiles;
- FILE *packhandle;
+ int packhandle;
pack_t *pack;
- dpackfile_t *info; // temporary alloc, allowing huge pack directories
+ dpackfile_t *info;
- packhandle = fopen (packfile, "rb");
- if (!packhandle)
+ packhandle = open (packfile, O_RDONLY | O_BINARY);
+ if (packhandle < 0)
return NULL;
-
- fread ((void *)&header, 1, sizeof(header), packhandle);
+ read (packhandle, (void *)&header, sizeof(header));
if (memcmp(header.id, "PACK", 4))
Sys_Error ("%s is not a packfile", packfile);
header.dirofs = LittleLong (header.dirofs);
strlcpy (pack->filename, packfile, sizeof (pack->filename));
pack->handle = packhandle;
pack->numfiles = 0;
- pack->mempool = Mem_AllocPool(packfile);
+ pack->mempool = Mem_AllocPool(packfile, 0, NULL);
pack->files = Mem_Alloc(pack->mempool, numpackfiles * sizeof(packfile_t));
pack->next = packlist;
packlist = pack;
info = Mem_Alloc(tempmempool, sizeof(*info) * numpackfiles);
- fseek (packhandle, header.dirofs, SEEK_SET);
- fread ((void *)info, 1, header.dirlen, packhandle);
+ lseek (packhandle, header.dirofs, SEEK_SET);
+ read (packhandle, (void *)info, header.dirlen);
// parse the directory
for (i = 0;i < numpackfiles;i++)
size_t offset = LittleLong (info[i].filepos);
size_t size = LittleLong (info[i].filelen);
- FS_AddFileToPack (info[i].name, pack, offset, size, size, FILE_FLAG_TRUEOFFS);
+ FS_AddFileToPack (info[i].name, pack, offset, size, size, PACKFILE_FLAG_TRUEOFFS);
}
Mem_Free(info);
then loads and adds pak1.pak pak2.pak ...
================
*/
-void FS_AddGameDirectory (char *dir)
+void FS_AddGameDirectory (const char *dir)
{
stringlist_t *list, *current;
searchpath_t *search;
strlcpy (fs_gamedir, dir, sizeof (fs_gamedir));
-#ifndef AKVERSION
- // add the directory to the search path
- search = Mem_Alloc(pak_mempool, sizeof(searchpath_t));
- strlcpy (search->filename, dir, sizeof (search->filename));
- search->next = fs_searchpaths;
- fs_searchpaths = search;
-#endif
-
list = listdirectory(dir);
// add any PAK package in the directory
{
if (matchpattern(current->text, "*.pak", true))
{
- snprintf (pakfile, sizeof (pakfile), "%s/%s", dir, current->text);
+ dpsnprintf (pakfile, sizeof (pakfile), "%s/%s", dir, current->text);
pak = FS_LoadPackPAK (pakfile);
if (pak)
{
{
if (matchpattern(current->text, "*.pk3", true))
{
- snprintf (pakfile, sizeof (pakfile), "%s/%s", dir, current->text);
+ dpsnprintf (pakfile, sizeof (pakfile), "%s/%s", dir, current->text);
pak = FS_LoadPackPK3 (pakfile);
if (pak)
{
}
freedirectory(list);
-// Unpacked files have the priority over packed files if AKVERSION is defined
-#ifdef AKVERSION
- // add the directory to the search path
+ // Add the directory to the search path
+ // (unpacked files have the priority over packed files)
search = Mem_Alloc(pak_mempool, sizeof(searchpath_t));
strlcpy (search->filename, dir, sizeof (search->filename));
search->next = fs_searchpaths;
fs_searchpaths = search;
-#endif
+}
+
+
+/*
+================
+FS_AddGameHierarchy
+================
+*/
+void FS_AddGameHierarchy (const char *dir)
+{
+ const char *homedir;
+
+ strlcpy (com_modname, dir, sizeof (com_modname));
+
+ // Add the common game directory
+ FS_AddGameDirectory (va("%s/%s", fs_basedir, dir));
+
+ // Add the personal game directory
+ homedir = getenv ("HOME");
+ if (homedir != NULL && homedir[0] != '\0')
+ FS_AddGameDirectory (va("%s/.%s/%s", homedir, gameuserdirname, dir));
}
FS_FileExtension
============
*/
-char *FS_FileExtension (const char *in)
+static const char *FS_FileExtension (const char *in)
{
- static char exten[8];
- const char *slash, *backslash, *colon, *dot, *separator;
- int i;
+ const char *separator, *backslash, *colon, *dot;
- slash = strrchr(in, '/');
+ separator = strrchr(in, '/');
backslash = strrchr(in, '\\');
- colon = strrchr(in, ':');
- dot = strrchr(in, '.');
- separator = slash;
if (separator < backslash)
separator = backslash;
+ colon = strrchr(in, ':');
if (separator < colon)
separator = colon;
- if (dot < separator)
+
+ dot = strrchr(in, '.');
+ if (dot == NULL || dot < separator)
return "";
- dot++;
- for (i = 0;i < 7 && dot[i];i++)
- exten[i] = dot[i];
- exten[i] = 0;
- return exten;
+
+ return dot + 1;
}
int i;
searchpath_t *search;
- fs_mempool = Mem_AllocPool("file management");
- pak_mempool = Mem_AllocPool("paks");
+ fs_mempool = Mem_AllocPool("file management", 0, NULL);
+ pak_mempool = Mem_AllocPool("paks", 0, NULL);
+
+ Cvar_RegisterVariable (&scr_screenshot_name);
Cmd_AddCommand ("path", FS_Path_f);
Cmd_AddCommand ("dir", FS_Dir_f);
Cmd_AddCommand ("ls", FS_Ls_f);
strcpy(fs_basedir, ".");
+ strcpy(fs_gamedir, ".");
PK3_OpenLibrary ();
// -basedir <path>
// Overrides the system supplied base directory (under GAMENAME)
+// COMMANDLINEOPTION: Filesystem: -basedir <path> chooses what base directory the game data is in, inside this there should be a data directory for the game (for example id1)
i = COM_CheckParm ("-basedir");
- if (i && i < com_argc-1)
- strlcpy (fs_basedir, com_argv[i+1], sizeof (fs_basedir));
-
- i = strlen (fs_basedir);
- if (i > 0 && (fs_basedir[i-1] == '\\' || fs_basedir[i-1] == '/'))
- fs_basedir[i-1] = 0;
-
- // start up with GAMENAME by default (id1)
- strlcpy (com_modname, GAMENAME, sizeof (com_modname));
- FS_AddGameDirectory (va("%s/"GAMENAME, fs_basedir));
- if (gamedirname[0])
- {
- fs_modified = true;
- strlcpy (com_modname, gamedirname, sizeof (com_modname));
- FS_AddGameDirectory (va("%s/%s", fs_basedir, gamedirname));
- }
-
- // -game <gamedir>
- // Adds basedir/gamedir as an override game
- i = COM_CheckParm ("-game");
if (i && i < com_argc-1)
{
- fs_modified = true;
- strlcpy (com_modname, com_argv[i+1], sizeof (com_modname));
- FS_AddGameDirectory (va("%s/%s", fs_basedir, com_argv[i+1]));
+ strlcpy (fs_basedir, com_argv[i+1], sizeof (fs_basedir));
+ i = strlen (fs_basedir);
+ if (i > 0 && (fs_basedir[i-1] == '\\' || fs_basedir[i-1] == '/'))
+ fs_basedir[i-1] = 0;
}
// -path <dir or packfile> [<dir or packfile>] ...
// Fully specifies the exact search path, overriding the generated one
+// COMMANDLINEOPTION: Filesystem: -path <path ..> specifies the full search path manually, overriding the generated one, example: -path c:\quake\id1 c:\quake\pak0.pak c:\quake\pak1.pak (not recommended)
i = COM_CheckParm ("-path");
if (i)
{
fs_modified = true;
- fs_searchpaths = NULL;
while (++i < com_argc)
{
if (!com_argv[i] || com_argv[i][0] == '+' || com_argv[i][0] == '-')
search->next = fs_searchpaths;
fs_searchpaths = search;
}
+ return;
+ }
+
+ // add the game-specific paths
+ // gamedirname1 (typically id1)
+ FS_AddGameHierarchy (gamedirname1);
+ Cvar_SetQuick (&scr_screenshot_name, gamescreenshotname);
+
+ // add the game-specific path, if any
+ if (gamedirname2)
+ {
+ fs_modified = true;
+ FS_AddGameHierarchy (gamedirname2);
+ }
+
+ // -game <gamedir>
+ // Adds basedir/gamedir as an override game
+ // LordHavoc: now supports multiple -game directories
+ for (i = 1;i < com_argc;i++)
+ {
+ if (!com_argv[i])
+ continue;
+ if (!strcmp (com_argv[i], "-game") && i < com_argc-1)
+ {
+ i++;
+ fs_modified = true;
+ FS_AddGameHierarchy (com_argv[i]);
+ Cvar_SetQuick (&scr_screenshot_name, com_modname);
+ }
}
+
+ // If "-condebug" is in the command line, remove the previous log file
+ if (COM_CheckParm ("-condebug") != 0)
+ unlink (va("%s/qconsole.log", fs_gamedir));
}
+/*
+================
+FS_Shutdown
+================
+*/
+void FS_Shutdown (void)
+{
+ Mem_FreePool (&pak_mempool);
+ Mem_FreePool (&fs_mempool);
+}
/*
====================
FS_SysOpen
-Internal function used to create a qfile_t and open the relevant file on disk
+Internal function used to create a qfile_t and open the relevant non-packed file on disk
====================
*/
-static qfile_t* FS_SysOpen (const char* filepath, const char* mode)
+static qfile_t* FS_SysOpen (const char* filepath, const char* mode, qboolean nonblocking)
{
qfile_t* file;
+ int mod, opt;
+ unsigned int ind;
+
+ // Parse the mode string
+ switch (mode[0])
+ {
+ case 'r':
+ mod = O_RDONLY;
+ opt = 0;
+ break;
+ case 'w':
+ mod = O_WRONLY;
+ opt = O_CREAT | O_TRUNC;
+ break;
+ case 'a':
+ mod = O_WRONLY;
+ opt = O_CREAT | O_APPEND;
+ break;
+ default:
+ Con_Printf ("FS_SysOpen(%s, %s): invalid mode\n", filepath, mode);
+ return NULL;
+ }
+ for (ind = 1; mode[ind] != '\0'; ind++)
+ {
+ switch (mode[ind])
+ {
+ case '+':
+ mod = O_RDWR;
+ break;
+ case 'b':
+ opt |= O_BINARY;
+ break;
+ default:
+ Con_Printf ("FS_SysOpen(%s, %s): unknown character in mode (%c)\n",
+ filepath, mode, mode[ind]);
+ }
+ }
+
+ if (nonblocking)
+ opt |= O_NONBLOCK;
file = Mem_Alloc (fs_mempool, sizeof (*file));
memset (file, 0, sizeof (*file));
+ file->ungetc = EOF;
- file->stream = fopen (filepath, mode);
- if (!file->stream)
+ file->handle = open (filepath, mod | opt, 0666);
+ if (file->handle < 0)
{
Mem_Free (file);
return NULL;
}
+ // For files opened in read mode, we now need to get the length
+ if (mod == O_RDONLY)
+ {
+ file->real_length = lseek (file->handle, 0, SEEK_END);
+ lseek (file->handle, 0, SEEK_SET);
+ }
+
return file;
}
/*
===========
-FS_OpenRead
+FS_OpenPackedFile
+
+Open a packed file using its package file descriptor
===========
*/
-qfile_t *FS_OpenRead (const char *path, int offs, int len)
+qfile_t *FS_OpenPackedFile (pack_t* pack, int pack_ind)
{
+ packfile_t *pfile;
+ int dup_handle;
qfile_t* file;
- file = FS_SysOpen (path, "rb");
- if (!file)
+ pfile = &pack->files[pack_ind];
+
+ // If we don't have the true offset, get it now
+ if (! (pfile->flags & PACKFILE_FLAG_TRUEOFFS))
+ PK3_GetTrueFileOffset (pfile, pack);
+
+ // No Zlib DLL = no compressed files
+ if (!zlib_dll && (pfile->flags & PACKFILE_FLAG_DEFLATED))
{
- Sys_Error ("Couldn't open %s", path);
+ Con_Printf("WARNING: can't open the compressed file %s\n"
+ "You need the Zlib DLL to use compressed files\n",
+ pfile->name);
+ fs_filesize = -1;
return NULL;
}
- // Normal file
- if (offs < 0 || len < 0)
- {
- // We set fs_filesize here for normal files
- fseek (file->stream, 0, SEEK_END);
- fs_filesize = ftell (file->stream);
- fseek (file->stream, 0, SEEK_SET);
- }
- // Packed file
- else
+ dup_handle = dup (pack->handle);
+ if (dup_handle < 0)
+ Sys_Error ("FS_OpenPackedFile: can't dup package's handle (pack: %s)", pack->filename);
+
+ file = Mem_Alloc (fs_mempool, sizeof (*file));
+ memset (file, 0, sizeof (*file));
+ file->handle = dup_handle;
+ file->flags = QFILE_FLAG_PACKED;
+ file->real_length = pfile->realsize;
+ file->offset = pfile->offset;
+ file->position = 0;
+ file->ungetc = EOF;
+
+ if (lseek (file->handle, file->offset, SEEK_SET) == -1)
+ Sys_Error ("FS_OpenPackedFile: can't lseek to %s in %s (offset: %d)",
+ pfile->name, pack->filename, file->offset);
+
+ if (pfile->flags & PACKFILE_FLAG_DEFLATED)
{
- fseek (file->stream, offs, SEEK_SET);
+ ztoolkit_t *ztk;
- file->flags |= FS_FLAG_PACKED;
- file->length = len;
- file->offset = offs;
- file->position = 0;
+ file->flags |= QFILE_FLAG_DEFLATED;
+
+ // We need some more variables
+ ztk = Mem_Alloc (fs_mempool, sizeof (*ztk));
+
+ ztk->comp_length = pfile->packsize;
+
+ // Initialize zlib stream
+ ztk->zstream.next_in = ztk->input;
+ ztk->zstream.avail_in = 0;
+
+ /* From Zlib's "unzip.c":
+ *
+ * windowBits is passed < 0 to tell that there is no zlib header.
+ * Note that in this case inflate *requires* an extra "dummy" byte
+ * after the compressed stream in order to complete decompression and
+ * return Z_STREAM_END.
+ * In unzip, i don't wait absolutely Z_STREAM_END because I known the
+ * size of both compressed and uncompressed data
+ */
+ if (qz_inflateInit2 (&ztk->zstream, -MAX_WBITS) != Z_OK)
+ Sys_Error ("FS_OpenPackedFile: inflate init error (file: %s)", pfile->name);
+
+ ztk->zstream.next_out = file->buff;
+ ztk->zstream.avail_out = sizeof (file->buff);
+
+ file->ztk = ztk;
}
+ fs_filesize = pfile->realsize;
+
return file;
}
+/*
+====================
+FS_CheckNastyPath
+
+Return true if the path should be rejected due to one of the following:
+1: path elements that are non-portable
+2: path elements that would allow access to files outside the game directory,
+ or are just not a good idea for a mod to be using.
+====================
+*/
+int FS_CheckNastyPath (const char *path)
+{
+ // Windows: don't allow \ in filenames (windows-only), period.
+ // (on Windows \ is a directory separator, but / is also supported)
+ if (strstr(path, "\\"))
+ return 1; // non-portable
+
+ // Mac: don't allow Mac-only filenames - : is a directory separator
+ // instead of /, but we rely on / working already, so there's no reason to
+ // support a Mac-only path
+ // Amiga and Windows: : tries to go to root of drive
+ if (strstr(path, ":"))
+ return 1; // non-portable attempt to go to root of drive
+
+ // Amiga: // is parent directory
+ if (strstr(path, "//"))
+ return 1; // non-portable attempt to go to parent directory
+
+ // all: don't allow going to current directory (./) or parent directory (../ or /../)
+ if (strstr(path, "./"))
+ return 2; // attempt to go outside the game directory
+
+ // Windows and UNIXes: don't allow absolute paths
+ if (path[0] == '/')
+ return 2; // attempt to go outside the game directory
+
+ // after all these checks we're pretty sure it's a / separated filename
+ // and won't do much if any harm
+ return false;
+}
+
/*
====================
{
searchpath_t *search;
pack_t *pak;
- int (*strcmp_funct) (const char* str1, const char* str2);
// search through the path, one element at a time
for (search = fs_searchpaths;search;search = search->next)
// is the element a pak file?
if (search->pack)
{
- size_t left, right, middle;
+ int (*strcmp_funct) (const char* str1, const char* str2);
+ int left, right, middle;
pak = search->pack;
strcmp_funct = pak->ignorecase ? strcasecmp : strcmp;
// Look for the file (binary search)
left = 0;
- right = pak->numfiles;
- while (left != right)
+ right = pak->numfiles - 1;
+ while (left <= right)
{
int diff;
- middle = (left + right - 1) / 2;
+ middle = (left + right) / 2;
diff = strcmp_funct (pak->files[middle].name, name);
// Found it
if (!diff)
{
if (!quiet)
- Sys_Printf("FS_FindFile: %s in %s\n",
+ Con_DPrintf("FS_FindFile: %s in %s\n",
pak->files[middle].name, pak->filename);
if (index != NULL)
// If we're too far in the list
if (diff > 0)
- right = middle;
+ right = middle - 1;
else
left = middle + 1;
}
else
{
char netpath[MAX_OSPATH];
- snprintf(netpath, sizeof(netpath), "%s/%s", search->filename, name);
+ dpsnprintf(netpath, sizeof(netpath), "%s/%s", search->filename, name);
if (FS_SysFileExists (netpath))
{
if (!quiet)
- Sys_Printf("FS_FindFile: %s\n", netpath);
+ Con_DPrintf("FS_FindFile: %s\n", netpath);
if (index != NULL)
*index = -1;
}
if (!quiet)
- Sys_Printf("FS_FindFile: can't find %s\n", name);
+ Con_DPrintf("FS_FindFile: can't find %s\n", name);
if (index != NULL)
*index = -1;
/*
===========
-FS_FOpenFile
+FS_OpenReadFile
-If the requested file is inside a packfile, a new qfile_t* will be opened
-into the file.
+Look for a file in the search paths and open it in read-only mode
Sets fs_filesize
===========
*/
-qfile_t *FS_FOpenFile (const char *filename, qboolean quiet)
+qfile_t *FS_OpenReadFile (const char *filename, qboolean quiet, qboolean nonblocking)
{
searchpath_t *search;
- packfile_t *packfile;
- int i;
- qfile_t *file;
+ int pack_ind;
- search = FS_FindFile (filename, &i, quiet);
+ search = FS_FindFile (filename, &pack_ind, quiet);
// Not found?
if (search == NULL)
}
// Found in the filesystem?
- if (i < 0)
+ if (pack_ind < 0)
{
- char netpath[MAX_OSPATH];
- snprintf(netpath, sizeof(netpath), "%s/%s", search->filename, filename);
- return FS_OpenRead(netpath, -1, -1);
+ char path [MAX_OSPATH];
+ dpsnprintf (path, sizeof (path), "%s/%s", search->filename, filename);
+ return FS_SysOpen (path, "rb", nonblocking);
}
// So, we found it in a package...
- packfile = &search->pack->files[i];
-
- // If we don't have the true offset, get it now
- if (! (packfile->flags & FILE_FLAG_TRUEOFFS))
- PK3_GetTrueFileOffset (packfile, search->pack);
-
- // No Zlib DLL = no compressed files
- if (!zlib_dll && (packfile->flags & FILE_FLAG_DEFLATED))
- {
- Con_Printf("WARNING: can't open the compressed file %s\n"
- "You need the Zlib DLL to use compressed files\n",
- filename);
- fs_filesize = -1;
- return NULL;
- }
-
- // open a new file in the pakfile
- file = FS_OpenRead (search->pack->filename, packfile->offset, packfile->packsize);
- fs_filesize = packfile->realsize;
-
- if (packfile->flags & FILE_FLAG_DEFLATED)
- {
- ztoolkit_t *ztk;
-
- file->flags |= FS_FLAG_DEFLATED;
-
- // We need some more variables
- ztk = Mem_Alloc (fs_mempool, sizeof (*file->z));
-
- ztk->real_length = packfile->realsize;
-
- // Initialize zlib stream
- ztk->zstream.next_in = ztk->input;
- ztk->zstream.avail_in = 0;
-
- /* From Zlib's "unzip.c":
- *
- * windowBits is passed < 0 to tell that there is no zlib header.
- * Note that in this case inflate *requires* an extra "dummy" byte
- * after the compressed stream in order to complete decompression and
- * return Z_STREAM_END.
- * In unzip, i don't wait absolutely Z_STREAM_END because I known the
- * size of both compressed and uncompressed data
- */
- if (qz_inflateInit2 (&ztk->zstream, -MAX_WBITS) != Z_OK)
- Sys_Error ("inflate init error (file: %s)", filename);
-
- ztk->zstream.next_out = ztk->output;
- ztk->zstream.avail_out = sizeof (ztk->output);
-
- file->z = ztk;
- }
-
- return file;
+ return FS_OpenPackedFile (search->pack, pack_ind);
}
Open a file. The syntax is the same as fopen
====================
*/
-qfile_t* FS_Open (const char* filepath, const char* mode, qboolean quiet)
+qfile_t* FS_Open (const char* filepath, const char* mode, qboolean quiet, qboolean nonblocking)
{
- // If the file is opened in "write" or "append" mode
- if (strchr (mode, 'w') || strchr (mode, 'a'))
+ qfile_t* file;
+
+ if (FS_CheckNastyPath(filepath))
+ {
+ Con_Printf("FS_Open(\"%s\", \"%s\", %s): nasty filename rejected\n", filepath, mode, quiet ? "true" : "false");
+ return NULL;
+ }
+
+ // If the file is opened in "write", "append", or "read/write" mode
+ if (mode[0] == 'w' || mode[0] == 'a' || strchr (mode, '+'))
{
char real_path [MAX_OSPATH];
// Open the file on disk directly
- snprintf (real_path, sizeof (real_path), "%s/%s", fs_gamedir, filepath);
+ dpsnprintf (real_path, sizeof (real_path), "%s/%s", fs_gamedir, filepath);
// Create directories up to the file
FS_CreatePath (real_path);
- return FS_SysOpen (real_path, mode);
+ return FS_SysOpen (real_path, mode, nonblocking);
}
- // Else, we look at the various search paths
- return FS_FOpenFile (filepath, quiet);
+ // Else, we look at the various search paths and open the file in read-only mode
+ file = FS_OpenReadFile (filepath, quiet, nonblocking);
+ if (file != NULL)
+ fs_filesize = file->real_length;
+
+ return file;
}
*/
int FS_Close (qfile_t* file)
{
- if (fclose (file->stream))
+ if (close (file->handle))
return EOF;
- if (file->z)
+ if (file->ztk)
{
- qz_inflateEnd (&file->z->zstream);
- Mem_Free (file->z);
+ qz_inflateEnd (&file->ztk->zstream);
+ Mem_Free (file->ztk);
}
Mem_Free (file);
*/
size_t FS_Write (qfile_t* file, const void* data, size_t datasize)
{
- return fwrite (data, 1, datasize, file->stream);
+ ssize_t result = write (file->handle, data, datasize);
+ if (result < 0)
+ return 0;
+
+ return result;
}
*/
size_t FS_Read (qfile_t* file, void* buffer, size_t buffersize)
{
- size_t count, nb;
- ztoolkit_t *ztk;
-
- // Quick path for unpacked files
- if (! (file->flags & FS_FLAG_PACKED))
- return fread (buffer, 1, buffersize, file->stream);
+ size_t count, done;
- // If the file isn't compressed
- if (! (file->flags & FS_FLAG_DEFLATED))
+ // First, we copy as many bytes as we can from "buff"
+ if (file->buff_ind < file->buff_len)
{
- // We must take care to not read after the end of the file
- count = file->length - file->position;
- if (buffersize > count)
- buffersize = count;
+ count = file->buff_len - file->buff_ind;
- nb = fread (buffer, 1, buffersize, file->stream);
+ done = (buffersize > count) ? count : buffersize;
+ memcpy (buffer, &file->buff[file->buff_ind], done);
+ file->buff_ind += done;
- file->position += nb;
- return nb;
+ buffersize -= done;
+ if (buffersize == 0)
+ return done;
}
+ else
+ done = 0;
- // If the file is compressed, it's more complicated...
- ztk = file->z;
+ // NOTE: at this point, the read buffer is always empty
- // First, we copy as many bytes as we can from "output"
- if (ztk->out_ind < ztk->out_max)
+ // If the file isn't compressed
+ if (! (file->flags & QFILE_FLAG_DEFLATED))
{
- count = ztk->out_max - ztk->out_ind;
+ int nb;
+
+ // We must take care to not read after the end of the file
+ count = file->real_length - file->position;
- nb = (buffersize > count) ? count : buffersize;
- memcpy (buffer, &ztk->output[ztk->out_ind], nb);
- ztk->out_ind += nb;
- file->position += nb;
+ // If we have a lot of data to get, put them directly into "buffer"
+ if (buffersize > sizeof (file->buff) / 2)
+ {
+ if (count > buffersize)
+ count = buffersize;
+ lseek (file->handle, file->offset + file->position, SEEK_SET);
+ nb = read (file->handle, &((qbyte*)buffer)[done], count);
+ if (nb > 0)
+ {
+ done += nb;
+ file->position += nb;
+
+ // Invalidate the output data (for FS_Seek)
+ file->buff_len = 0;
+ file->buff_ind = 0;
+ }
+ }
+ else
+ {
+ if (count > sizeof (file->buff))
+ count = sizeof (file->buff);
+ lseek (file->handle, file->offset + file->position, SEEK_SET);
+ nb = read (file->handle, file->buff, count);
+ if (nb > 0)
+ {
+ file->buff_len = nb;
+ file->position += nb;
+
+ // Copy the requested data in "buffer" (as much as we can)
+ count = (buffersize > file->buff_len) ? file->buff_len : buffersize;
+ memcpy (&((qbyte*)buffer)[done], file->buff, count);
+ file->buff_ind = count;
+ done += count;
+ }
+ }
+
+ return done;
}
- else
- nb = 0;
- // We cycle through a few operations until we have inflated enough data
- while (nb < buffersize)
+ // If the file is compressed, it's more complicated...
+ // We cycle through a few operations until we have read enough data
+ while (buffersize > 0)
{
- // NOTE: at this point, "output" should always be empty
+ ztoolkit_t *ztk = file->ztk;
+ int error;
- // If "input" is also empty, we need to fill it
- if (ztk->in_ind == ztk->in_max)
- {
- size_t remain;
+ // NOTE: at this point, the read buffer is always empty
+ // If "input" is also empty, we need to refill it
+ if (ztk->in_ind == ztk->in_len)
+ {
// If we are at the end of the file
- if (ztk->out_position == ztk->real_length)
- return nb;
+ if (file->position == file->real_length)
+ return done;
- remain = file->length - ztk->in_position;
- count = (remain > sizeof (ztk->input)) ? sizeof (ztk->input) : remain;
- fread (ztk->input, 1, count, file->stream);
+ count = ztk->comp_length - ztk->in_position;
+ if (count > sizeof (ztk->input))
+ count = sizeof (ztk->input);
+ lseek (file->handle, file->offset + ztk->in_position, SEEK_SET);
+ if (read (file->handle, ztk->input, count) != (ssize_t)count)
+ Sys_Error ("FS_Read: unexpected end of file");
- // Update indexes and counters
ztk->in_ind = 0;
- ztk->in_max = count;
+ ztk->in_len = count;
ztk->in_position += count;
}
- // Now that we are sure we have compressed data available, we need to determine
- // if it's better to inflate it in "output" or directly in "buffer" (we are in this
- // case if we still need more bytes than "output" can contain)
-
ztk->zstream.next_in = &ztk->input[ztk->in_ind];
- ztk->zstream.avail_in = ztk->in_max - ztk->in_ind;
+ ztk->zstream.avail_in = ztk->in_len - ztk->in_ind;
- // If output will be able to contain at least 1 more byte than the data we need
- if (buffersize - nb < sizeof (ztk->output))
- {
- int error;
+ // Now that we are sure we have compressed data available, we need to determine
+ // if it's better to inflate it in "file->buff" or directly in "buffer"
- // Inflate the data in "output"
- ztk->zstream.next_out = ztk->output;
- ztk->zstream.avail_out = sizeof (ztk->output);
+ // Inflate the data in "file->buff"
+ if (buffersize < sizeof (file->buff) / 2)
+ {
+ ztk->zstream.next_out = file->buff;
+ ztk->zstream.avail_out = sizeof (file->buff);
error = qz_inflate (&ztk->zstream, Z_SYNC_FLUSH);
if (error != Z_OK && error != Z_STREAM_END)
Sys_Error ("Can't inflate file");
- ztk->in_ind = ztk->in_max - ztk->zstream.avail_in;
- ztk->out_max = sizeof (ztk->output) - ztk->zstream.avail_out;
- ztk->out_position += ztk->out_max;
+ ztk->in_ind = ztk->in_len - ztk->zstream.avail_in;
+
+ file->buff_len = sizeof (file->buff) - ztk->zstream.avail_out;
+ file->position += file->buff_len;
// Copy the requested data in "buffer" (as much as we can)
- count = (buffersize - nb > ztk->out_max) ? ztk->out_max : buffersize - nb;
- memcpy (&((qbyte*)buffer)[nb], ztk->output, count);
- ztk->out_ind = count;
+ count = (buffersize > file->buff_len) ? file->buff_len : buffersize;
+ memcpy (&((qbyte*)buffer)[done], file->buff, count);
+ file->buff_ind = count;
}
// Else, we inflate directly in "buffer"
else
{
- int error;
-
- // Inflate the data in "buffer"
- ztk->zstream.next_out = &((qbyte*)buffer)[nb];
- ztk->zstream.avail_out = buffersize - nb;
+ ztk->zstream.next_out = &((qbyte*)buffer)[done];
+ ztk->zstream.avail_out = buffersize;
error = qz_inflate (&ztk->zstream, Z_SYNC_FLUSH);
if (error != Z_OK && error != Z_STREAM_END)
Sys_Error ("Can't inflate file");
- ztk->in_ind = ztk->in_max - ztk->zstream.avail_in;
-
- // Invalidate the output data (for FS_Seek)
- ztk->out_max = 0;
- ztk->out_ind = 0;
+ ztk->in_ind = ztk->in_len - ztk->zstream.avail_in;
// How much data did it inflate?
- count = buffersize - nb - ztk->zstream.avail_out;
- ztk->out_position += count;
+ count = buffersize - ztk->zstream.avail_out;
+ file->position += count;
+
+ // Invalidate the output data (for FS_Seek)
+ file->buff_len = 0;
+ file->buff_ind = 0;
}
- nb += count;
- file->position += count;
+ done += count;
+ buffersize -= count;
}
- return nb;
-}
-
-
-/*
-====================
-FS_Flush
-
-Flush the file output stream
-====================
-*/
-int FS_Flush (qfile_t* file)
-{
- return fflush (file->stream);
+ return done;
}
Print a string into a file
====================
*/
-int FS_Print(qfile_t* file, const char *msg)
+int FS_Print (qfile_t* file, const char *msg)
{
- return FS_Write(file, msg, strlen(msg));
+ return FS_Write (file, msg, strlen (msg));
}
/*
va_list args;
va_start (args, format);
- result = vfprintf (file->stream, format, args);
+ result = FS_VPrintf (file, format, args);
va_end (args);
return result;
/*
====================
-FS_Getc
+FS_VPrintf
-Get the next character of a file
+Print a string into a file
====================
*/
-int FS_Getc (qfile_t* file)
+int FS_VPrintf (qfile_t* file, const char* format, va_list ap)
{
- char c;
+ int len;
+ size_t buff_size;
+ char *tempbuff = NULL;
+
+ buff_size = 1024;
+ tempbuff = Mem_Alloc (tempmempool, buff_size);
+ len = dpvsnprintf (tempbuff, buff_size, format, ap);
+ while (len < 0)
+ {
+ Mem_Free (tempbuff);
+ buff_size *= 2;
+ tempbuff = Mem_Alloc (tempmempool, buff_size);
+ len = dpvsnprintf (tempbuff, buff_size, format, ap);
+ }
- if (FS_Read (file, &c, 1) != 1)
- return EOF;
+ len = write (file->handle, tempbuff, len);
+ Mem_Free (tempbuff);
- return c;
+ return len;
}
/*
====================
-FS_Seek
+FS_Getc
-Move the position index in a file
+Get stored ungetc character or the next character of a file
====================
*/
-int FS_Seek (qfile_t* file, long offset, int whence)
+int FS_Getc (qfile_t* file)
{
- // Quick path for unpacked files
- if (! (file->flags & FS_FLAG_PACKED))
- return fseek (file->stream, offset, whence);
-
- // Seeking in compressed files is more a hack than anything else,
- // but we need to support it, so here it is.
- if (file->flags & FS_FLAG_DEFLATED)
- {
- ztoolkit_t *ztk = file->z;
- qbyte buffer [sizeof (ztk->output)]; // it's big to force inflating into buffer directly
-
- switch (whence)
- {
- case SEEK_CUR:
- offset += file->position;
- break;
-
- case SEEK_SET:
- break;
-
- case SEEK_END:
- offset += ztk->real_length;
- break;
-
- default:
- return -1;
- }
- if (offset < 0 || offset > (long) ztk->real_length)
- return -1;
-
- // If we need to go back in the file
- if (offset <= (long) file->position)
- {
- // If we still have the data we need in the output buffer
- if (file->position - offset <= ztk->out_ind)
- {
- ztk->out_ind -= file->position - offset;
- file->position = offset;
- return 0;
- }
-
- // Else, we restart from the beginning of the file
- ztk->in_ind = 0;
- ztk->in_max = 0;
- ztk->in_position = 0;
- ztk->out_ind = 0;
- ztk->out_max = 0;
- ztk->out_position = 0;
- file->position = 0;
- fseek (file->stream, file->offset, SEEK_SET);
-
- // Reset the Zlib stream
- ztk->zstream.next_in = ztk->input;
- ztk->zstream.avail_in = 0;
- qz_inflateReset (&ztk->zstream);
- }
-
- // Skip all data until we reach the requested offset
- while ((long) file->position < offset)
- {
- size_t diff = offset - file->position;
- size_t count, len;
-
- count = (diff > sizeof (buffer)) ? sizeof (buffer) : diff;
- len = FS_Read (file, buffer, count);
- if (len != count)
- return -1;
- }
-
- return 0;
- }
+ char c;
- // Packed files receive a special treatment too, because
- // we need to make sure it doesn't go outside of the file
- switch (whence)
+ if (file->ungetc != EOF)
{
- case SEEK_CUR:
- offset += file->position;
- break;
-
- case SEEK_SET:
- break;
-
- case SEEK_END:
- offset += file->length;
- break;
-
- default:
- return -1;
+ c = file->ungetc;
+ file->ungetc = EOF;
}
- if (offset < 0 || offset > (long) file->length)
- return -1;
+ else if (FS_Read (file, &c, 1) != 1)
+ return EOF;
- if (fseek (file->stream, file->offset + offset, SEEK_SET) == -1)
- return -1;
- file->position = offset;
- return 0;
+ return c;
}
/*
====================
-FS_Tell
+FS_UnGetc
-Give the current position in a file
+Put a character back into the Getc buffer (only supports one character!)
====================
*/
-long FS_Tell (qfile_t* file)
+void FS_UnGetc (qfile_t* file, unsigned char c)
{
- if (file->flags & FS_FLAG_PACKED)
- return file->position;
-
- return ftell (file->stream);
+ file->ungetc = c;
}
/*
====================
-FS_Gets
+FS_Seek
-Extract a line from a file
+Move the position index in a file
====================
*/
-char* FS_Gets (qfile_t* file, char* buffer, int buffersize)
+int FS_Seek (qfile_t* file, long offset, int whence)
{
- size_t ind;
+ ztoolkit_t *ztk;
+ qbyte* buffer;
+ size_t buffersize;
- // Quick path for unpacked files
- if (! (file->flags & FS_FLAG_PACKED))
- return fgets (buffer, buffersize, file->stream);
+ // if this is an uncompressed real file we can just call the kernel seek
+ // (necessary when writing files)
+ if (file->offset == 0 && ! (file->flags & QFILE_FLAG_DEFLATED))
+ return lseek (file->handle, offset, whence);
- for (ind = 0; ind < (size_t) buffersize - 1; ind++)
+ // Compute the file offset
+ switch (whence)
{
- int c = FS_Getc (file);
- switch (c)
- {
- // End of file
- case -1:
- if (!ind)
- return NULL;
+ case SEEK_CUR:
+ offset += file->position - file->buff_len + file->buff_ind;
+ break;
- buffer[ind] = '\0';
- return buffer;
+ case SEEK_SET:
+ break;
- // End of line
- case '\r':
- case '\n':
- buffer[ind] = '\n';
- buffer[ind + 1] = '\0';
- return buffer;
+ case SEEK_END:
+ offset += file->real_length;
+ break;
- default:
- buffer[ind] = c;
- }
+ default:
+ return -1;
+ }
+ if (offset < 0 || offset > (long) file->real_length)
+ return -1;
+ // If we have the data in our read buffer, we don't need to actually seek
+ if (file->position - file->buff_len <= (size_t)offset &&
+ (size_t)offset <= file->position)
+ {
+ file->buff_ind = offset + file->buff_len - file->position;
+ return 0;
}
- buffer[buffersize - 1] = '\0';
- return buffer;
-}
+ // Invalidate the read buffer contents
+ file->buff_ind = 0;
+ file->buff_len = 0;
+ // Unpacked or uncompressed files can seek directly
+ if (! (file->flags & QFILE_FLAG_DEFLATED))
+ {
+ if (lseek (file->handle, file->offset + offset, SEEK_SET) == -1)
+ return -1;
+ file->position = offset;
+ return 0;
+ }
-/*
-==========
-FS_Getline
+ // Seeking in compressed files is more a hack than anything else,
+ // but we need to support it, so here we go.
+ ztk = file->ztk;
-Dynamic length version of fgets. DO NOT free the buffer.
-==========
-*/
-char *FS_Getline (qfile_t *file)
-{
- static int size = 256;
- static char *buf = 0;
- char *t;
- int len;
+ // If we have to go back in the file, we need to restart from the beginning
+ if ((size_t)offset <= file->position)
+ {
+ ztk->in_ind = 0;
+ ztk->in_len = 0;
+ ztk->in_position = 0;
+ file->position = 0;
+ lseek (file->handle, file->offset, SEEK_SET);
- if (!buf)
- buf = Mem_Alloc (fs_mempool, size);
+ // Reset the Zlib stream
+ ztk->zstream.next_in = ztk->input;
+ ztk->zstream.avail_in = 0;
+ qz_inflateReset (&ztk->zstream);
+ }
- if (!FS_Gets (file, buf, size))
- return 0;
+ // We need a big buffer to force inflating into it directly
+ buffersize = 2 * sizeof (file->buff);
+ buffer = Mem_Alloc (tempmempool, buffersize);
- len = strlen (buf);
- while (buf[len - 1] != '\n' && buf[len - 1] != '\r')
+ // Skip all data until we reach the requested offset
+ while ((size_t)offset > file->position)
{
- t = Mem_Alloc (fs_mempool, size + 256);
- memcpy(t, buf, size);
- Mem_Free(buf);
- size += 256;
- buf = t;
- if (!FS_Gets (file, buf + len, size - len))
- break;
- len = strlen (buf);
+ size_t diff = offset - file->position;
+ size_t count, len;
+
+ count = (diff > buffersize) ? buffersize : diff;
+ len = FS_Read (file, buffer, count);
+ if (len != count)
+ {
+ Mem_Free (buffer);
+ return -1;
+ }
}
- while ((len = strlen(buf)) && (buf[len - 1] == '\n' || buf[len - 1] == '\r'))
- buf[len - 1] = 0;
- return buf;
+
+ Mem_Free (buffer);
+ return 0;
}
/*
====================
-FS_Eof
+FS_Tell
-Extract a line from a file
+Give the current position in a file
====================
*/
-int FS_Eof (qfile_t* file)
+long FS_Tell (qfile_t* file)
{
- if (file->flags & FS_FLAG_PACKED)
- {
- if (file->flags & FS_FLAG_DEFLATED)
- return (file->position == file->z->real_length);
-
- return (file->position == file->length);
- }
-
- return feof (file->stream);
+ // if this is an uncompressed real file we can just call the kernel tell
+ // (necessary when writing files)
+ if (file->offset == 0 && ! (file->flags & QFILE_FLAG_DEFLATED))
+ return lseek (file->handle, 0, SEEK_CUR);
+ else
+ return file->position - file->buff_len + file->buff_ind;
}
*/
qbyte *FS_LoadFile (const char *path, mempool_t *pool, qboolean quiet)
{
- qfile_t *h;
+ qfile_t *file;
qbyte *buf;
- // look for it in the filesystem or pack files
- h = FS_Open (path, "rb", quiet);
- if (!h)
+ file = FS_Open (path, "rb", quiet, false);
+ if (!file)
return NULL;
- buf = Mem_Alloc(pool, fs_filesize+1);
- if (!buf)
- Sys_Error ("FS_LoadFile: not enough available memory for %s (size %i)", path, fs_filesize);
-
- ((qbyte *)buf)[fs_filesize] = 0;
+ buf = Mem_Alloc (pool, fs_filesize + 1);
+ buf[fs_filesize] = '\0';
- FS_Read (h, buf, fs_filesize);
- FS_Close (h);
+ FS_Read (file, buf, fs_filesize);
+ FS_Close (file);
return buf;
}
*/
qboolean FS_WriteFile (const char *filename, void *data, int len)
{
- qfile_t *handle;
+ qfile_t *file;
- handle = FS_Open (filename, "wb", false);
- if (!handle)
+ file = FS_Open (filename, "wb", false, false);
+ if (!file)
{
Con_Printf("FS_WriteFile: failed on %s\n", filename);
return false;
}
Con_DPrintf("FS_WriteFile: %s\n", filename);
- FS_Write (handle, data, len);
- FS_Close (handle);
+ FS_Write (file, data, len);
+ FS_Close (file);
return true;
}
qboolean FS_SysFileExists (const char *path)
{
#if WIN32
- FILE *f;
+ int desc;
- f = fopen (path, "rb");
- if (f)
- {
- fclose (f);
- return true;
- }
+ // TODO: use another function instead, to avoid opening the file
+ desc = open (path, O_RDONLY | O_BINARY);
+ if (desc < 0)
+ return false;
- return false;
+ close (desc);
+ return true;
#else
struct stat buf;
if (separator < colon)
separator = colon;
basepathlength = separator - pattern;
- basepath = Z_Malloc(basepathlength + 1);
+ basepath = Mem_Alloc (tempmempool, basepathlength + 1);
if (basepathlength)
memcpy(basepath, pattern, basepathlength);
basepath[basepathlength] = 0;
if (liststart == NULL)
liststart = listcurrent;
if (!quiet)
- Sys_Printf("SearchPackFile: %s : %s\n", pak->filename, temp);
+ Con_DPrintf("SearchPackFile: %s : %s\n", pak->filename, temp);
}
}
// strip off one path element at a time until empty
else
{
// get a directory listing and look at each name
- snprintf(netpath, sizeof (netpath), "%s/%s", searchpath->filename, basepath);
+ dpsnprintf(netpath, sizeof (netpath), "%s/%s", searchpath->filename, basepath);
if ((dir = listdirectory(netpath)))
{
for (dirfile = dir;dirfile;dirfile = dirfile->next)
{
- snprintf(temp, sizeof(temp), "%s/%s", basepath, dirfile->text);
+ dpsnprintf(temp, sizeof(temp), "%s/%s", basepath, dirfile->text);
if (matchpattern(temp, (char *)pattern, true))
{
for (listtemp = liststart;listtemp;listtemp = listtemp->next)
if (liststart == NULL)
liststart = listcurrent;
if (!quiet)
- Sys_Printf("SearchDirFile: %s\n", temp);
+ Con_DPrintf("SearchDirFile: %s\n", temp);
}
}
}
stringlistfree(liststart);
}
- Z_Free(basepath);
+ Mem_Free(basepath);
return search;
}
const char *name;
char linebuf[4096];
fssearch_t *search;
- search = FS_Search(pattern, true, false);
+ search = FS_Search(pattern, true, true);
if (!search)
return 0;
numfiles = search->numfilenames;
projects / xonotic / darkplaces.git / blobdiff