]> git.xonotic.org Git - xonotic/darkplaces.git/blob - fs.c
Add QC/cfg facilities to control deletion of automatically recorded demos
[xonotic/darkplaces.git] / fs.c
1 /*
2         DarkPlaces file system
3
4         Copyright (C) 2003-2006 Mathieu Olivier
5
6         This program is free software; you can redistribute it and/or
7         modify it under the terms of the GNU General Public License
8         as published by the Free Software Foundation; either version 2
9         of the License, or (at your option) any later version.
10
11         This program is distributed in the hope that it will be useful,
12         but WITHOUT ANY WARRANTY; without even the implied warranty of
13         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
14
15         See the GNU General Public License for more details.
16
17         You should have received a copy of the GNU General Public License
18         along with this program; if not, write to:
19
20                 Free Software Foundation, Inc.
21                 59 Temple Place - Suite 330
22                 Boston, MA  02111-1307, USA
23 */
24
25 #include "quakedef.h"
26
27 #include <limits.h>
28 #include <fcntl.h>
29
30 #ifdef WIN32
31 # include <direct.h>
32 # include <io.h>
33 # include <shlobj.h>
34 #else
35 # include <pwd.h>
36 # include <sys/stat.h>
37 # include <unistd.h>
38 #endif
39
40 #include "fs.h"
41 #include "wad.h"
42
43 // Win32 requires us to add O_BINARY, but the other OSes don't have it
44 #ifndef O_BINARY
45 # define O_BINARY 0
46 #endif
47
48 // In case the system doesn't support the O_NONBLOCK flag
49 #ifndef O_NONBLOCK
50 # define O_NONBLOCK 0
51 #endif
52
53 // largefile support for Win32
54 #ifdef WIN32
55 # define lseek _lseeki64
56 #endif
57
58 #if _MSC_VER >= 1400
59 // suppress deprecated warnings
60 # include <sys/stat.h>
61 # include <share.h>
62 # define read _read
63 # define write _write
64 # define close _close
65 # define unlink _unlink
66 # define dup _dup
67 #endif
68
69 /** \page fs File System
70
71 All of Quake's data access is through a hierchal file system, but the contents
72 of the file system can be transparently merged from several sources.
73
74 The "base directory" is the path to the directory holding the quake.exe and
75 all game directories.  The sys_* files pass this to host_init in
76 quakeparms_t->basedir.  This can be overridden with the "-basedir" command
77 line parm to allow code debugging in a different directory.  The base
78 directory is only used during filesystem initialization.
79
80 The "game directory" is the first tree on the search path and directory that
81 all generated files (savegames, screenshots, demos, config files) will be
82 saved to.  This can be overridden with the "-game" command line parameter.
83 The game directory can never be changed while quake is executing.  This is a
84 precaution against having a malicious server instruct clients to write files
85 over areas they shouldn't.
86
87 */
88
89
90 /*
91 =============================================================================
92
93 CONSTANTS
94
95 =============================================================================
96 */
97
98 // Magic numbers of a ZIP file (big-endian format)
99 #define ZIP_DATA_HEADER 0x504B0304  // "PK\3\4"
100 #define ZIP_CDIR_HEADER 0x504B0102  // "PK\1\2"
101 #define ZIP_END_HEADER  0x504B0506  // "PK\5\6"
102
103 // Other constants for ZIP files
104 #define ZIP_MAX_COMMENTS_SIZE           ((unsigned short)0xFFFF)
105 #define ZIP_END_CDIR_SIZE                       22
106 #define ZIP_CDIR_CHUNK_BASE_SIZE        46
107 #define ZIP_LOCAL_CHUNK_BASE_SIZE       30
108
109 #ifdef LINK_TO_ZLIB
110 #include <zlib.h>
111
112 #define qz_inflate inflate
113 #define qz_inflateEnd inflateEnd
114 #define qz_inflateInit2_ inflateInit2_
115 #define qz_inflateReset inflateReset
116 #define qz_deflateInit2_ deflateInit2_
117 #define qz_deflateEnd deflateEnd
118 #define qz_deflate deflate
119 #define Z_MEMLEVEL_DEFAULT 8
120 #else
121
122 // Zlib constants (from zlib.h)
123 #define Z_SYNC_FLUSH    2
124 #define MAX_WBITS               15
125 #define Z_OK                    0
126 #define Z_STREAM_END    1
127 #define Z_STREAM_ERROR  (-2)
128 #define Z_DATA_ERROR    (-3)
129 #define Z_MEM_ERROR     (-4)
130 #define Z_BUF_ERROR     (-5)
131 #define ZLIB_VERSION    "1.2.3"
132
133 #define Z_BINARY 0
134 #define Z_DEFLATED 8
135 #define Z_MEMLEVEL_DEFAULT 8
136
137 #define Z_NULL 0
138 #define Z_DEFAULT_COMPRESSION (-1)
139 #define Z_NO_FLUSH 0
140 #define Z_SYNC_FLUSH 2
141 #define Z_FULL_FLUSH 3
142 #define Z_FINISH 4
143
144 // Uncomment the following line if the zlib DLL you have still uses
145 // the 1.1.x series calling convention on Win32 (WINAPI)
146 //#define ZLIB_USES_WINAPI
147
148
149 /*
150 =============================================================================
151
152 TYPES
153
154 =============================================================================
155 */
156
157 /*! Zlib stream (from zlib.h)
158  * \warning: some pointers we don't use directly have
159  * been cast to "void*" for a matter of simplicity
160  */
161 typedef struct
162 {
163         unsigned char                   *next_in;       ///< next input byte
164         unsigned int    avail_in;       ///< number of bytes available at next_in
165         unsigned long   total_in;       ///< total nb of input bytes read so far
166
167         unsigned char                   *next_out;      ///< next output byte should be put there
168         unsigned int    avail_out;      ///< remaining free space at next_out
169         unsigned long   total_out;      ///< total nb of bytes output so far
170
171         char                    *msg;           ///< last error message, NULL if no error
172         void                    *state;         ///< not visible by applications
173
174         void                    *zalloc;        ///< used to allocate the internal state
175         void                    *zfree;         ///< used to free the internal state
176         void                    *opaque;        ///< private data object passed to zalloc and zfree
177
178         int                             data_type;      ///< best guess about the data type: ascii or binary
179         unsigned long   adler;          ///< adler32 value of the uncompressed data
180         unsigned long   reserved;       ///< reserved for future use
181 } z_stream;
182 #endif
183
184
185 /// inside a package (PAK or PK3)
186 #define QFILE_FLAG_PACKED (1 << 0)
187 /// file is compressed using the deflate algorithm (PK3 only)
188 #define QFILE_FLAG_DEFLATED (1 << 1)
189 /// file is actually already loaded data
190 #define QFILE_FLAG_DATA (1 << 2)
191 /// real file will be removed on close
192 #define QFILE_FLAG_REMOVE (1 << 3)
193
194 #define FILE_BUFF_SIZE 2048
195 typedef struct
196 {
197         z_stream        zstream;
198         size_t          comp_length;                    ///< length of the compressed file
199         size_t          in_ind, in_len;                 ///< input buffer current index and length
200         size_t          in_position;                    ///< position in the compressed file
201         unsigned char           input [FILE_BUFF_SIZE];
202 } ztoolkit_t;
203
204 struct qfile_s
205 {
206         int                             flags;
207         int                             handle;                                 ///< file descriptor
208         fs_offset_t             real_length;                    ///< uncompressed file size (for files opened in "read" mode)
209         fs_offset_t             position;                               ///< current position in the file
210         fs_offset_t             offset;                                 ///< offset into the package (0 if external file)
211         int                             ungetc;                                 ///< single stored character from ungetc, cleared to EOF when read
212
213         // Contents buffer
214         fs_offset_t             buff_ind, buff_len;             ///< buffer current index and length
215         unsigned char                   buff [FILE_BUFF_SIZE];
216
217         ztoolkit_t*             ztk;    ///< For zipped files.
218
219         const unsigned char *data;      ///< For data files.
220
221         const char *filename; ///< Kept around for QFILE_FLAG_REMOVE, unused otherwise
222 };
223
224
225 // ------ PK3 files on disk ------ //
226
227 // You can get the complete ZIP format description from PKWARE website
228
229 typedef struct pk3_endOfCentralDir_s
230 {
231         unsigned int signature;
232         unsigned short disknum;
233         unsigned short cdir_disknum;    ///< number of the disk with the start of the central directory
234         unsigned short localentries;    ///< number of entries in the central directory on this disk
235         unsigned short nbentries;               ///< total number of entries in the central directory on this disk
236         unsigned int cdir_size;                 ///< size of the central directory
237         unsigned int cdir_offset;               ///< with respect to the starting disk number
238         unsigned short comment_size;
239         fs_offset_t prepended_garbage;
240 } pk3_endOfCentralDir_t;
241
242
243 // ------ PAK files on disk ------ //
244 typedef struct dpackfile_s
245 {
246         char name[56];
247         int filepos, filelen;
248 } dpackfile_t;
249
250 typedef struct dpackheader_s
251 {
252         char id[4];
253         int dirofs;
254         int dirlen;
255 } dpackheader_t;
256
257
258 /*! \name Packages in memory
259  * @{
260  */
261 /// the offset in packfile_t is the true contents offset
262 #define PACKFILE_FLAG_TRUEOFFS (1 << 0)
263 /// file compressed using the deflate algorithm
264 #define PACKFILE_FLAG_DEFLATED (1 << 1)
265 /// file is a symbolic link
266 #define PACKFILE_FLAG_SYMLINK (1 << 2)
267
268 typedef struct packfile_s
269 {
270         char name [MAX_QPATH];
271         int flags;
272         fs_offset_t offset;
273         fs_offset_t packsize;   ///< size in the package
274         fs_offset_t realsize;   ///< real file size (uncompressed)
275 } packfile_t;
276
277 typedef struct pack_s
278 {
279         char filename [MAX_OSPATH];
280         char shortname [MAX_QPATH];
281         int handle;
282         int ignorecase;  ///< PK3 ignores case
283         int numfiles;
284         qboolean vpack;
285         packfile_t *files;
286 } pack_t;
287 //@}
288
289 /// Search paths for files (including packages)
290 typedef struct searchpath_s
291 {
292         // only one of filename / pack will be used
293         char filename[MAX_OSPATH];
294         pack_t *pack;
295         struct searchpath_s *next;
296 } searchpath_t;
297
298
299 /*
300 =============================================================================
301
302 FUNCTION PROTOTYPES
303
304 =============================================================================
305 */
306
307 void FS_Dir_f(void);
308 void FS_Ls_f(void);
309 void FS_Which_f(void);
310
311 static searchpath_t *FS_FindFile (const char *name, int* index, qboolean quiet);
312 static packfile_t* FS_AddFileToPack (const char* name, pack_t* pack,
313                                                                         fs_offset_t offset, fs_offset_t packsize,
314                                                                         fs_offset_t realsize, int flags);
315
316
317 /*
318 =============================================================================
319
320 VARIABLES
321
322 =============================================================================
323 */
324
325 mempool_t *fs_mempool;
326
327 searchpath_t *fs_searchpaths = NULL;
328 const char *const fs_checkgamedir_missing = "missing";
329
330 #define MAX_FILES_IN_PACK       65536
331
332 char fs_userdir[MAX_OSPATH];
333 char fs_gamedir[MAX_OSPATH];
334 char fs_basedir[MAX_OSPATH];
335 static pack_t *fs_selfpack = NULL;
336
337 // list of active game directories (empty if not running a mod)
338 int fs_numgamedirs = 0;
339 char fs_gamedirs[MAX_GAMEDIRS][MAX_QPATH];
340
341 // list of all gamedirs with modinfo.txt
342 gamedir_t *fs_all_gamedirs = NULL;
343 int fs_all_gamedirs_count = 0;
344
345 cvar_t scr_screenshot_name = {CVAR_NORESETTODEFAULTS, "scr_screenshot_name","dp", "prefix name for saved screenshots (changes based on -game commandline, as well as which game mode is running; the date is encoded using strftime escapes)"};
346 cvar_t fs_empty_files_in_pack_mark_deletions = {0, "fs_empty_files_in_pack_mark_deletions", "0", "if enabled, empty files in a pak/pk3 count as not existing but cancel the search in further packs, effectively allowing patch pak/pk3 files to 'delete' files"};
347 cvar_t cvar_fs_gamedir = {CVAR_READONLY | CVAR_NORESETTODEFAULTS, "fs_gamedir", "", "the list of currently selected gamedirs (use the 'gamedir' command to change this)"};
348
349
350 /*
351 =============================================================================
352
353 PRIVATE FUNCTIONS - PK3 HANDLING
354
355 =============================================================================
356 */
357
358 #ifndef LINK_TO_ZLIB
359 // Functions exported from zlib
360 #if defined(WIN32) && defined(ZLIB_USES_WINAPI)
361 # define ZEXPORT WINAPI
362 #else
363 # define ZEXPORT
364 #endif
365
366 static int (ZEXPORT *qz_inflate) (z_stream* strm, int flush);
367 static int (ZEXPORT *qz_inflateEnd) (z_stream* strm);
368 static int (ZEXPORT *qz_inflateInit2_) (z_stream* strm, int windowBits, const char *version, int stream_size);
369 static int (ZEXPORT *qz_inflateReset) (z_stream* strm);
370 static int (ZEXPORT *qz_deflateInit2_) (z_stream* strm, int level, int method, int windowBits, int memLevel, int strategy, const char *version, int stream_size);
371 static int (ZEXPORT *qz_deflateEnd) (z_stream* strm);
372 static int (ZEXPORT *qz_deflate) (z_stream* strm, int flush);
373 #endif
374
375 #define qz_inflateInit2(strm, windowBits) \
376         qz_inflateInit2_((strm), (windowBits), ZLIB_VERSION, sizeof(z_stream))
377 #define qz_deflateInit2(strm, level, method, windowBits, memLevel, strategy) \
378         qz_deflateInit2_((strm), (level), (method), (windowBits), (memLevel), (strategy), ZLIB_VERSION, sizeof(z_stream))
379
380 #ifndef LINK_TO_ZLIB
381 //        qz_deflateInit_((strm), (level), ZLIB_VERSION, sizeof(z_stream))
382
383 static dllfunction_t zlibfuncs[] =
384 {
385         {"inflate",                     (void **) &qz_inflate},
386         {"inflateEnd",          (void **) &qz_inflateEnd},
387         {"inflateInit2_",       (void **) &qz_inflateInit2_},
388         {"inflateReset",        (void **) &qz_inflateReset},
389         {"deflateInit2_",   (void **) &qz_deflateInit2_},
390         {"deflateEnd",      (void **) &qz_deflateEnd},
391         {"deflate",         (void **) &qz_deflate},
392         {NULL, NULL}
393 };
394
395 /// Handle for Zlib DLL
396 static dllhandle_t zlib_dll = NULL;
397 #endif
398
399 #ifdef WIN32
400 static HRESULT (WINAPI *qSHGetFolderPath) (HWND hwndOwner, int nFolder, HANDLE hToken, DWORD dwFlags, LPTSTR pszPath);
401 static dllfunction_t shfolderfuncs[] =
402 {
403         {"SHGetFolderPathA", (void **) &qSHGetFolderPath},
404         {NULL, NULL}
405 };
406 static dllhandle_t shfolder_dll = NULL;
407 #endif
408
409 /*
410 ====================
411 PK3_CloseLibrary
412
413 Unload the Zlib DLL
414 ====================
415 */
416 void PK3_CloseLibrary (void)
417 {
418 #ifndef LINK_TO_ZLIB
419         Sys_UnloadLibrary (&zlib_dll);
420 #endif
421 }
422
423
424 /*
425 ====================
426 PK3_OpenLibrary
427
428 Try to load the Zlib DLL
429 ====================
430 */
431 qboolean PK3_OpenLibrary (void)
432 {
433 #ifdef LINK_TO_ZLIB
434         return true;
435 #else
436         const char* dllnames [] =
437         {
438 #if defined(WIN32)
439 # ifdef ZLIB_USES_WINAPI
440                 "zlibwapi.dll",
441                 "zlib.dll",
442 # else
443                 "zlib1.dll",
444 # endif
445 #elif defined(MACOSX)
446                 "libz.dylib",
447 #else
448                 "libz.so.1",
449                 "libz.so",
450 #endif
451                 NULL
452         };
453
454         // Already loaded?
455         if (zlib_dll)
456                 return true;
457
458         // Load the DLL
459         return Sys_LoadLibrary (dllnames, &zlib_dll, zlibfuncs);
460 #endif
461 }
462
463 /*
464 ====================
465 FS_HasZlib
466
467 See if zlib is available
468 ====================
469 */
470 qboolean FS_HasZlib(void)
471 {
472 #ifdef LINK_TO_ZLIB
473         return true;
474 #else
475         PK3_OpenLibrary(); // to be safe
476         return (zlib_dll != 0);
477 #endif
478 }
479
480 /*
481 ====================
482 PK3_GetEndOfCentralDir
483
484 Extract the end of the central directory from a PK3 package
485 ====================
486 */
487 qboolean PK3_GetEndOfCentralDir (const char *packfile, int packhandle, pk3_endOfCentralDir_t *eocd)
488 {
489         fs_offset_t filesize, maxsize;
490         unsigned char *buffer, *ptr;
491         int ind;
492
493         // Get the package size
494         filesize = lseek (packhandle, 0, SEEK_END);
495         if (filesize < ZIP_END_CDIR_SIZE)
496                 return false;
497
498         // Load the end of the file in memory
499         if (filesize < ZIP_MAX_COMMENTS_SIZE + ZIP_END_CDIR_SIZE)
500                 maxsize = filesize;
501         else
502                 maxsize = ZIP_MAX_COMMENTS_SIZE + ZIP_END_CDIR_SIZE;
503         buffer = (unsigned char *)Mem_Alloc (tempmempool, maxsize);
504         lseek (packhandle, filesize - maxsize, SEEK_SET);
505         if (read (packhandle, buffer, maxsize) != (fs_offset_t) maxsize)
506         {
507                 Mem_Free (buffer);
508                 return false;
509         }
510
511         // Look for the end of central dir signature around the end of the file
512         maxsize -= ZIP_END_CDIR_SIZE;
513         ptr = &buffer[maxsize];
514         ind = 0;
515         while (BuffBigLong (ptr) != ZIP_END_HEADER)
516         {
517                 if (ind == maxsize)
518                 {
519                         Mem_Free (buffer);
520                         return false;
521                 }
522
523                 ind++;
524                 ptr--;
525         }
526
527         memcpy (eocd, ptr, ZIP_END_CDIR_SIZE);
528         eocd->signature = LittleLong (eocd->signature);
529         eocd->disknum = LittleShort (eocd->disknum);
530         eocd->cdir_disknum = LittleShort (eocd->cdir_disknum);
531         eocd->localentries = LittleShort (eocd->localentries);
532         eocd->nbentries = LittleShort (eocd->nbentries);
533         eocd->cdir_size = LittleLong (eocd->cdir_size);
534         eocd->cdir_offset = LittleLong (eocd->cdir_offset);
535         eocd->comment_size = LittleShort (eocd->comment_size);
536         eocd->prepended_garbage = filesize - (ind + ZIP_END_CDIR_SIZE) - eocd->cdir_offset - eocd->cdir_size; // this detects "SFX" zip files
537         eocd->cdir_offset += eocd->prepended_garbage;
538
539         Mem_Free (buffer);
540
541         return true;
542 }
543
544
545 /*
546 ====================
547 PK3_BuildFileList
548
549 Extract the file list from a PK3 file
550 ====================
551 */
552 int PK3_BuildFileList (pack_t *pack, const pk3_endOfCentralDir_t *eocd)
553 {
554         unsigned char *central_dir, *ptr;
555         unsigned int ind;
556         fs_offset_t remaining;
557
558         // Load the central directory in memory
559         central_dir = (unsigned char *)Mem_Alloc (tempmempool, eocd->cdir_size);
560         lseek (pack->handle, eocd->cdir_offset, SEEK_SET);
561         if(read (pack->handle, central_dir, eocd->cdir_size) != (fs_offset_t) eocd->cdir_size)
562         {
563                 Mem_Free (central_dir);
564                 return -1;
565         }
566
567         // Extract the files properties
568         // The parsing is done "by hand" because some fields have variable sizes and
569         // the constant part isn't 4-bytes aligned, which makes the use of structs difficult
570         remaining = eocd->cdir_size;
571         pack->numfiles = 0;
572         ptr = central_dir;
573         for (ind = 0; ind < eocd->nbentries; ind++)
574         {
575                 fs_offset_t namesize, count;
576
577                 // Checking the remaining size
578                 if (remaining < ZIP_CDIR_CHUNK_BASE_SIZE)
579                 {
580                         Mem_Free (central_dir);
581                         return -1;
582                 }
583                 remaining -= ZIP_CDIR_CHUNK_BASE_SIZE;
584
585                 // Check header
586                 if (BuffBigLong (ptr) != ZIP_CDIR_HEADER)
587                 {
588                         Mem_Free (central_dir);
589                         return -1;
590                 }
591
592                 namesize = BuffLittleShort (&ptr[28]);  // filename length
593
594                 // Check encryption, compression, and attributes
595                 // 1st uint8  : general purpose bit flag
596                 //    Check bits 0 (encryption), 3 (data descriptor after the file), and 5 (compressed patched data (?))
597                 //
598                 // LordHavoc: bit 3 would be a problem if we were scanning the archive
599                 // but is not a problem in the central directory where the values are
600                 // always real.
601                 //
602                 // bit 3 seems to always be set by the standard Mac OSX zip maker
603                 //
604                 // 2nd uint8 : external file attributes
605                 //    Check bits 3 (file is a directory) and 5 (file is a volume (?))
606                 if ((ptr[8] & 0x21) == 0 && (ptr[38] & 0x18) == 0)
607                 {
608                         // Still enough bytes for the name?
609                         if (remaining < namesize || namesize >= (int)sizeof (*pack->files))
610                         {
611                                 Mem_Free (central_dir);
612                                 return -1;
613                         }
614
615                         // WinZip doesn't use the "directory" attribute, so we need to check the name directly
616                         if (ptr[ZIP_CDIR_CHUNK_BASE_SIZE + namesize - 1] != '/')
617                         {
618                                 char filename [sizeof (pack->files[0].name)];
619                                 fs_offset_t offset, packsize, realsize;
620                                 int flags;
621
622                                 // Extract the name (strip it if necessary)
623                                 namesize = min(namesize, (int)sizeof (filename) - 1);
624                                 memcpy (filename, &ptr[ZIP_CDIR_CHUNK_BASE_SIZE], namesize);
625                                 filename[namesize] = '\0';
626
627                                 if (BuffLittleShort (&ptr[10]))
628                                         flags = PACKFILE_FLAG_DEFLATED;
629                                 else
630                                         flags = 0;
631                                 offset = BuffLittleLong (&ptr[42]) + eocd->prepended_garbage;
632                                 packsize = BuffLittleLong (&ptr[20]);
633                                 realsize = BuffLittleLong (&ptr[24]);
634
635                                 switch(ptr[5]) // C_VERSION_MADE_BY_1
636                                 {
637                                         case 3: // UNIX_
638                                         case 2: // VMS_
639                                         case 16: // BEOS_
640                                                 if((BuffLittleShort(&ptr[40]) & 0120000) == 0120000)
641                                                         // can't use S_ISLNK here, as this has to compile on non-UNIX too
642                                                         flags |= PACKFILE_FLAG_SYMLINK;
643                                                 break;
644                                 }
645
646                                 FS_AddFileToPack (filename, pack, offset, packsize, realsize, flags);
647                         }
648                 }
649
650                 // Skip the name, additionnal field, and comment
651                 // 1er uint16 : extra field length
652                 // 2eme uint16 : file comment length
653                 count = namesize + BuffLittleShort (&ptr[30]) + BuffLittleShort (&ptr[32]);
654                 ptr += ZIP_CDIR_CHUNK_BASE_SIZE + count;
655                 remaining -= count;
656         }
657
658         // If the package is empty, central_dir is NULL here
659         if (central_dir != NULL)
660                 Mem_Free (central_dir);
661         return pack->numfiles;
662 }
663
664
665 /*
666 ====================
667 FS_LoadPackPK3
668
669 Create a package entry associated with a PK3 file
670 ====================
671 */
672 pack_t *FS_LoadPackPK3FromFD (const char *packfile, int packhandle)
673 {
674         pk3_endOfCentralDir_t eocd;
675         pack_t *pack;
676         int real_nb_files;
677
678         if (! PK3_GetEndOfCentralDir (packfile, packhandle, &eocd))
679         {
680                 Con_Printf ("%s is not a PK3 file\n", packfile);
681                 close(packhandle);
682                 return NULL;
683         }
684
685         // Multi-volume ZIP archives are NOT allowed
686         if (eocd.disknum != 0 || eocd.cdir_disknum != 0)
687         {
688                 Con_Printf ("%s is a multi-volume ZIP archive\n", packfile);
689                 close(packhandle);
690                 return NULL;
691         }
692
693         // We only need to do this test if MAX_FILES_IN_PACK is lesser than 65535
694         // since eocd.nbentries is an unsigned 16 bits integer
695 #if MAX_FILES_IN_PACK < 65535
696         if (eocd.nbentries > MAX_FILES_IN_PACK)
697         {
698                 Con_Printf ("%s contains too many files (%hu)\n", packfile, eocd.nbentries);
699                 close(packhandle);
700                 return NULL;
701         }
702 #endif
703
704         // Create a package structure in memory
705         pack = (pack_t *)Mem_Alloc(fs_mempool, sizeof (pack_t));
706         pack->ignorecase = true; // PK3 ignores case
707         strlcpy (pack->filename, packfile, sizeof (pack->filename));
708         pack->handle = packhandle;
709         pack->numfiles = eocd.nbentries;
710         pack->files = (packfile_t *)Mem_Alloc(fs_mempool, eocd.nbentries * sizeof(packfile_t));
711
712         real_nb_files = PK3_BuildFileList (pack, &eocd);
713         if (real_nb_files < 0)
714         {
715                 Con_Printf ("%s is not a valid PK3 file\n", packfile);
716                 close(pack->handle);
717                 Mem_Free(pack);
718                 return NULL;
719         }
720
721         Con_DPrintf("Added packfile %s (%i files)\n", packfile, real_nb_files);
722         return pack;
723 }
724 pack_t *FS_LoadPackPK3 (const char *packfile)
725 {
726         int packhandle;
727 #if _MSC_VER >= 1400
728         _sopen_s(&packhandle, packfile, O_RDONLY | O_BINARY, _SH_DENYNO, _S_IREAD | _S_IWRITE);
729 #else
730         packhandle = open (packfile, O_RDONLY | O_BINARY);
731 #endif
732         if (packhandle < 0)
733                 return NULL;
734         return FS_LoadPackPK3FromFD(packfile, packhandle);
735 }
736
737
738 /*
739 ====================
740 PK3_GetTrueFileOffset
741
742 Find where the true file data offset is
743 ====================
744 */
745 qboolean PK3_GetTrueFileOffset (packfile_t *pfile, pack_t *pack)
746 {
747         unsigned char buffer [ZIP_LOCAL_CHUNK_BASE_SIZE];
748         fs_offset_t count;
749
750         // Already found?
751         if (pfile->flags & PACKFILE_FLAG_TRUEOFFS)
752                 return true;
753
754         // Load the local file description
755         lseek (pack->handle, pfile->offset, SEEK_SET);
756         count = read (pack->handle, buffer, ZIP_LOCAL_CHUNK_BASE_SIZE);
757         if (count != ZIP_LOCAL_CHUNK_BASE_SIZE || BuffBigLong (buffer) != ZIP_DATA_HEADER)
758         {
759                 Con_Printf ("Can't retrieve file %s in package %s\n", pfile->name, pack->filename);
760                 return false;
761         }
762
763         // Skip name and extra field
764         pfile->offset += BuffLittleShort (&buffer[26]) + BuffLittleShort (&buffer[28]) + ZIP_LOCAL_CHUNK_BASE_SIZE;
765
766         pfile->flags |= PACKFILE_FLAG_TRUEOFFS;
767         return true;
768 }
769
770
771 /*
772 =============================================================================
773
774 OTHER PRIVATE FUNCTIONS
775
776 =============================================================================
777 */
778
779
780 /*
781 ====================
782 FS_AddFileToPack
783
784 Add a file to the list of files contained into a package
785 ====================
786 */
787 static packfile_t* FS_AddFileToPack (const char* name, pack_t* pack,
788                                                                          fs_offset_t offset, fs_offset_t packsize,
789                                                                          fs_offset_t realsize, int flags)
790 {
791         int (*strcmp_funct) (const char* str1, const char* str2);
792         int left, right, middle;
793         packfile_t *pfile;
794
795         strcmp_funct = pack->ignorecase ? strcasecmp : strcmp;
796
797         // Look for the slot we should put that file into (binary search)
798         left = 0;
799         right = pack->numfiles - 1;
800         while (left <= right)
801         {
802                 int diff;
803
804                 middle = (left + right) / 2;
805                 diff = strcmp_funct (pack->files[middle].name, name);
806
807                 // If we found the file, there's a problem
808                 if (!diff)
809                         Con_Printf ("Package %s contains the file %s several times\n", pack->filename, name);
810
811                 // If we're too far in the list
812                 if (diff > 0)
813                         right = middle - 1;
814                 else
815                         left = middle + 1;
816         }
817
818         // We have to move the right of the list by one slot to free the one we need
819         pfile = &pack->files[left];
820         memmove (pfile + 1, pfile, (pack->numfiles - left) * sizeof (*pfile));
821         pack->numfiles++;
822
823         strlcpy (pfile->name, name, sizeof (pfile->name));
824         pfile->offset = offset;
825         pfile->packsize = packsize;
826         pfile->realsize = realsize;
827         pfile->flags = flags;
828
829         return pfile;
830 }
831
832
833 /*
834 ============
835 FS_CreatePath
836
837 Only used for FS_OpenRealFile.
838 ============
839 */
840 void FS_CreatePath (char *path)
841 {
842         char *ofs, save;
843
844         for (ofs = path+1 ; *ofs ; ofs++)
845         {
846                 if (*ofs == '/' || *ofs == '\\')
847                 {
848                         // create the directory
849                         save = *ofs;
850                         *ofs = 0;
851                         FS_mkdir (path);
852                         *ofs = save;
853                 }
854         }
855 }
856
857
858 /*
859 ============
860 FS_Path_f
861
862 ============
863 */
864 void FS_Path_f (void)
865 {
866         searchpath_t *s;
867
868         Con_Print("Current search path:\n");
869         for (s=fs_searchpaths ; s ; s=s->next)
870         {
871                 if (s->pack)
872                 {
873                         if(s->pack->vpack)
874                                 Con_Printf("%sdir (virtual pack)\n", s->pack->filename);
875                         else
876                                 Con_Printf("%s (%i files)\n", s->pack->filename, s->pack->numfiles);
877                 }
878                 else
879                         Con_Printf("%s\n", s->filename);
880         }
881 }
882
883
884 /*
885 =================
886 FS_LoadPackPAK
887 =================
888 */
889 /*! Takes an explicit (not game tree related) path to a pak file.
890  *Loads the header and directory, adding the files at the beginning
891  *of the list so they override previous pack files.
892  */
893 pack_t *FS_LoadPackPAK (const char *packfile)
894 {
895         dpackheader_t header;
896         int i, numpackfiles;
897         int packhandle;
898         pack_t *pack;
899         dpackfile_t *info;
900
901 #if _MSC_VER >= 1400
902         _sopen_s(&packhandle, packfile, O_RDONLY | O_BINARY, _SH_DENYNO, _S_IREAD | _S_IWRITE);
903 #else
904         packhandle = open (packfile, O_RDONLY | O_BINARY);
905 #endif
906         if (packhandle < 0)
907                 return NULL;
908         if(read (packhandle, (void *)&header, sizeof(header)) != sizeof(header))
909         {
910                 Con_Printf ("%s is not a packfile\n", packfile);
911                 close(packhandle);
912                 return NULL;
913         }
914         if (memcmp(header.id, "PACK", 4))
915         {
916                 Con_Printf ("%s is not a packfile\n", packfile);
917                 close(packhandle);
918                 return NULL;
919         }
920         header.dirofs = LittleLong (header.dirofs);
921         header.dirlen = LittleLong (header.dirlen);
922
923         if (header.dirlen % sizeof(dpackfile_t))
924         {
925                 Con_Printf ("%s has an invalid directory size\n", packfile);
926                 close(packhandle);
927                 return NULL;
928         }
929
930         numpackfiles = header.dirlen / sizeof(dpackfile_t);
931
932         if (numpackfiles > MAX_FILES_IN_PACK)
933         {
934                 Con_Printf ("%s has %i files\n", packfile, numpackfiles);
935                 close(packhandle);
936                 return NULL;
937         }
938
939         info = (dpackfile_t *)Mem_Alloc(tempmempool, sizeof(*info) * numpackfiles);
940         lseek (packhandle, header.dirofs, SEEK_SET);
941         if(header.dirlen != read (packhandle, (void *)info, header.dirlen))
942         {
943                 Con_Printf("%s is an incomplete PAK, not loading\n", packfile);
944                 Mem_Free(info);
945                 close(packhandle);
946                 return NULL;
947         }
948
949         pack = (pack_t *)Mem_Alloc(fs_mempool, sizeof (pack_t));
950         pack->ignorecase = false; // PAK is case sensitive
951         strlcpy (pack->filename, packfile, sizeof (pack->filename));
952         pack->handle = packhandle;
953         pack->numfiles = 0;
954         pack->files = (packfile_t *)Mem_Alloc(fs_mempool, numpackfiles * sizeof(packfile_t));
955
956         // parse the directory
957         for (i = 0;i < numpackfiles;i++)
958         {
959                 fs_offset_t offset = LittleLong (info[i].filepos);
960                 fs_offset_t size = LittleLong (info[i].filelen);
961
962                 FS_AddFileToPack (info[i].name, pack, offset, size, size, PACKFILE_FLAG_TRUEOFFS);
963         }
964
965         Mem_Free(info);
966
967         Con_DPrintf("Added packfile %s (%i files)\n", packfile, numpackfiles);
968         return pack;
969 }
970
971 /*
972 ====================
973 FS_LoadPackVirtual
974
975 Create a package entry associated with a directory file
976 ====================
977 */
978 pack_t *FS_LoadPackVirtual (const char *dirname)
979 {
980         pack_t *pack;
981         pack = (pack_t *)Mem_Alloc(fs_mempool, sizeof (pack_t));
982         pack->vpack = true;
983         pack->ignorecase = false;
984         strlcpy (pack->filename, dirname, sizeof(pack->filename));
985         pack->handle = -1;
986         pack->numfiles = -1;
987         pack->files = NULL;
988         Con_DPrintf("Added packfile %s (virtual pack)\n", dirname);
989         return pack;
990 }
991
992 /*
993 ================
994 FS_AddPack_Fullpath
995 ================
996 */
997 /*! Adds the given pack to the search path.
998  * The pack type is autodetected by the file extension.
999  *
1000  * Returns true if the file was successfully added to the
1001  * search path or if it was already included.
1002  *
1003  * If keep_plain_dirs is set, the pack will be added AFTER the first sequence of
1004  * plain directories.
1005  *
1006  */
1007 static qboolean FS_AddPack_Fullpath(const char *pakfile, const char *shortname, qboolean *already_loaded, qboolean keep_plain_dirs)
1008 {
1009         searchpath_t *search;
1010         pack_t *pak = NULL;
1011         const char *ext = FS_FileExtension(pakfile);
1012         size_t l;
1013
1014         for(search = fs_searchpaths; search; search = search->next)
1015         {
1016                 if(search->pack && !strcasecmp(search->pack->filename, pakfile))
1017                 {
1018                         if(already_loaded)
1019                                 *already_loaded = true;
1020                         return true; // already loaded
1021                 }
1022         }
1023
1024         if(already_loaded)
1025                 *already_loaded = false;
1026
1027         if(!strcasecmp(ext, "pk3dir"))
1028                 pak = FS_LoadPackVirtual (pakfile);
1029         else if(!strcasecmp(ext, "pak"))
1030                 pak = FS_LoadPackPAK (pakfile);
1031         else if(!strcasecmp(ext, "pk3"))
1032                 pak = FS_LoadPackPK3 (pakfile);
1033         else
1034                 Con_Printf("\"%s\" does not have a pack extension\n", pakfile);
1035
1036         if(pak)
1037         {
1038                 strlcpy(pak->shortname, shortname, sizeof(pak->shortname));
1039
1040                 //Con_DPrintf("  Registered pack with short name %s\n", shortname);
1041                 if(keep_plain_dirs)
1042                 {
1043                         // find the first item whose next one is a pack or NULL
1044                         searchpath_t *insertion_point = 0;
1045                         if(fs_searchpaths && !fs_searchpaths->pack)
1046                         {
1047                                 insertion_point = fs_searchpaths;
1048                                 for(;;)
1049                                 {
1050                                         if(!insertion_point->next)
1051                                                 break;
1052                                         if(insertion_point->next->pack)
1053                                                 break;
1054                                         insertion_point = insertion_point->next;
1055                                 }
1056                         }
1057                         // If insertion_point is NULL, this means that either there is no
1058                         // item in the list yet, or that the very first item is a pack. In
1059                         // that case, we want to insert at the beginning...
1060                         if(!insertion_point)
1061                         {
1062                                 search = (searchpath_t *)Mem_Alloc(fs_mempool, sizeof(searchpath_t));
1063                                 search->next = fs_searchpaths;
1064                                 fs_searchpaths = search;
1065                         }
1066                         else
1067                         // otherwise we want to append directly after insertion_point.
1068                         {
1069                                 search = (searchpath_t *)Mem_Alloc(fs_mempool, sizeof(searchpath_t));
1070                                 search->next = insertion_point->next;
1071                                 insertion_point->next = search;
1072                         }
1073                 }
1074                 else
1075                 {
1076                         search = (searchpath_t *)Mem_Alloc(fs_mempool, sizeof(searchpath_t));
1077                         search->next = fs_searchpaths;
1078                         fs_searchpaths = search;
1079                 }
1080                 search->pack = pak;
1081                 if(pak->vpack)
1082                 {
1083                         dpsnprintf(search->filename, sizeof(search->filename), "%s/", pakfile);
1084                         // if shortname ends with "pk3dir", strip that suffix to make it just "pk3"
1085                         // same goes for the name inside the pack structure
1086                         l = strlen(pak->shortname);
1087                         if(l >= 7)
1088                                 if(!strcasecmp(pak->shortname + l - 7, ".pk3dir"))
1089                                         pak->shortname[l - 3] = 0;
1090                         l = strlen(pak->filename);
1091                         if(l >= 7)
1092                                 if(!strcasecmp(pak->filename + l - 7, ".pk3dir"))
1093                                         pak->filename[l - 3] = 0;
1094                 }
1095                 return true;
1096         }
1097         else
1098         {
1099                 Con_Printf("unable to load pak \"%s\"\n", pakfile);
1100                 return false;
1101         }
1102 }
1103
1104
1105 /*
1106 ================
1107 FS_AddPack
1108 ================
1109 */
1110 /*! Adds the given pack to the search path and searches for it in the game path.
1111  * The pack type is autodetected by the file extension.
1112  *
1113  * Returns true if the file was successfully added to the
1114  * search path or if it was already included.
1115  *
1116  * If keep_plain_dirs is set, the pack will be added AFTER the first sequence of
1117  * plain directories.
1118  */
1119 qboolean FS_AddPack(const char *pakfile, qboolean *already_loaded, qboolean keep_plain_dirs)
1120 {
1121         char fullpath[MAX_OSPATH];
1122         int index;
1123         searchpath_t *search;
1124
1125         if(already_loaded)
1126                 *already_loaded = false;
1127
1128         // then find the real name...
1129         search = FS_FindFile(pakfile, &index, true);
1130         if(!search || search->pack)
1131         {
1132                 Con_Printf("could not find pak \"%s\"\n", pakfile);
1133                 return false;
1134         }
1135
1136         dpsnprintf(fullpath, sizeof(fullpath), "%s%s", search->filename, pakfile);
1137
1138         return FS_AddPack_Fullpath(fullpath, pakfile, already_loaded, keep_plain_dirs);
1139 }
1140
1141
1142 /*
1143 ================
1144 FS_AddGameDirectory
1145
1146 Sets fs_gamedir, adds the directory to the head of the path,
1147 then loads and adds pak1.pak pak2.pak ...
1148 ================
1149 */
1150 void FS_AddGameDirectory (const char *dir)
1151 {
1152         int i;
1153         stringlist_t list;
1154         searchpath_t *search;
1155
1156         strlcpy (fs_gamedir, dir, sizeof (fs_gamedir));
1157
1158         stringlistinit(&list);
1159         listdirectory(&list, "", dir);
1160         stringlistsort(&list);
1161
1162         // add any PAK package in the directory
1163         for (i = 0;i < list.numstrings;i++)
1164         {
1165                 if (!strcasecmp(FS_FileExtension(list.strings[i]), "pak"))
1166                 {
1167                         FS_AddPack_Fullpath(list.strings[i], list.strings[i] + strlen(dir), NULL, false);
1168                 }
1169         }
1170
1171         // add any PK3 package in the directory
1172         for (i = 0;i < list.numstrings;i++)
1173         {
1174                 if (!strcasecmp(FS_FileExtension(list.strings[i]), "pk3") || !strcasecmp(FS_FileExtension(list.strings[i]), "pk3dir"))
1175                 {
1176                         FS_AddPack_Fullpath(list.strings[i], list.strings[i] + strlen(dir), NULL, false);
1177                 }
1178         }
1179
1180         stringlistfreecontents(&list);
1181
1182         // Add the directory to the search path
1183         // (unpacked files have the priority over packed files)
1184         search = (searchpath_t *)Mem_Alloc(fs_mempool, sizeof(searchpath_t));
1185         strlcpy (search->filename, dir, sizeof (search->filename));
1186         search->next = fs_searchpaths;
1187         fs_searchpaths = search;
1188 }
1189
1190
1191 /*
1192 ================
1193 FS_AddGameHierarchy
1194 ================
1195 */
1196 void FS_AddGameHierarchy (const char *dir)
1197 {
1198         // Add the common game directory
1199         FS_AddGameDirectory (va("%s%s/", fs_basedir, dir));
1200
1201         if (*fs_userdir)
1202                 FS_AddGameDirectory(va("%s%s/", fs_userdir, dir));
1203 }
1204
1205
1206 /*
1207 ============
1208 FS_FileExtension
1209 ============
1210 */
1211 const char *FS_FileExtension (const char *in)
1212 {
1213         const char *separator, *backslash, *colon, *dot;
1214
1215         separator = strrchr(in, '/');
1216         backslash = strrchr(in, '\\');
1217         if (!separator || separator < backslash)
1218                 separator = backslash;
1219         colon = strrchr(in, ':');
1220         if (!separator || separator < colon)
1221                 separator = colon;
1222
1223         dot = strrchr(in, '.');
1224         if (dot == NULL || (separator && (dot < separator)))
1225                 return "";
1226
1227         return dot + 1;
1228 }
1229
1230
1231 /*
1232 ============
1233 FS_FileWithoutPath
1234 ============
1235 */
1236 const char *FS_FileWithoutPath (const char *in)
1237 {
1238         const char *separator, *backslash, *colon;
1239
1240         separator = strrchr(in, '/');
1241         backslash = strrchr(in, '\\');
1242         if (!separator || separator < backslash)
1243                 separator = backslash;
1244         colon = strrchr(in, ':');
1245         if (!separator || separator < colon)
1246                 separator = colon;
1247         return separator ? separator + 1 : in;
1248 }
1249
1250
1251 /*
1252 ================
1253 FS_ClearSearchPath
1254 ================
1255 */
1256 void FS_ClearSearchPath (void)
1257 {
1258         // unload all packs and directory information, close all pack files
1259         // (if a qfile is still reading a pack it won't be harmed because it used
1260         //  dup() to get its own handle already)
1261         while (fs_searchpaths)
1262         {
1263                 searchpath_t *search = fs_searchpaths;
1264                 fs_searchpaths = search->next;
1265                 if (search->pack && search->pack != fs_selfpack)
1266                 {
1267                         if(!search->pack->vpack)
1268                         {
1269                                 // close the file
1270                                 close(search->pack->handle);
1271                                 // free any memory associated with it
1272                                 if (search->pack->files)
1273                                         Mem_Free(search->pack->files);
1274                         }
1275                         Mem_Free(search->pack);
1276                 }
1277                 Mem_Free(search);
1278         }
1279 }
1280
1281 static void FS_AddSelfPack(void)
1282 {
1283         if(fs_selfpack)
1284         {
1285                 searchpath_t *search;
1286                 search = Mem_Alloc(fs_mempool, sizeof(searchpath_t));
1287                 search->next = fs_searchpaths;
1288                 search->pack = fs_selfpack;
1289                 fs_searchpaths = search;
1290         }
1291 }
1292
1293
1294 /*
1295 ================
1296 FS_Rescan
1297 ================
1298 */
1299 void FS_Rescan (void)
1300 {
1301         int i;
1302         qboolean fs_modified = false;
1303         char gamedirbuf[MAX_INPUTLINE];
1304
1305         FS_ClearSearchPath();
1306
1307         // add the game-specific paths
1308         // gamedirname1 (typically id1)
1309         FS_AddGameHierarchy (gamedirname1);
1310         // update the com_modname (used for server info)
1311         strlcpy(com_modname, gamedirname1, sizeof(com_modname));
1312
1313         // add the game-specific path, if any
1314         // (only used for mission packs and the like, which should set fs_modified)
1315         if (gamedirname2)
1316         {
1317                 fs_modified = true;
1318                 FS_AddGameHierarchy (gamedirname2);
1319         }
1320
1321         // -game <gamedir>
1322         // Adds basedir/gamedir as an override game
1323         // LordHavoc: now supports multiple -game directories
1324         // set the com_modname (reported in server info)
1325         *gamedirbuf = 0;
1326         for (i = 0;i < fs_numgamedirs;i++)
1327         {
1328                 fs_modified = true;
1329                 FS_AddGameHierarchy (fs_gamedirs[i]);
1330                 // update the com_modname (used server info)
1331                 strlcpy (com_modname, fs_gamedirs[i], sizeof (com_modname));
1332                 if(i)
1333                         strlcat(gamedirbuf, va(" %s", fs_gamedirs[i]), sizeof(gamedirbuf));
1334                 else
1335                         strlcpy(gamedirbuf, fs_gamedirs[i], sizeof(gamedirbuf));
1336         }
1337         Cvar_SetQuick(&cvar_fs_gamedir, gamedirbuf); // so QC or console code can query it
1338
1339         // add back the selfpack as new first item
1340         FS_AddSelfPack();
1341
1342         // set the default screenshot name to either the mod name or the
1343         // gamemode screenshot name
1344         if (strcmp(com_modname, gamedirname1))
1345                 Cvar_SetQuick (&scr_screenshot_name, com_modname);
1346         else
1347                 Cvar_SetQuick (&scr_screenshot_name, gamescreenshotname);
1348         
1349         if((i = COM_CheckParm("-modname")) && i < com_argc - 1)
1350                 strlcpy(com_modname, com_argv[i+1], sizeof(com_modname));
1351
1352         // If "-condebug" is in the command line, remove the previous log file
1353         if (COM_CheckParm ("-condebug") != 0)
1354                 unlink (va("%s/qconsole.log", fs_gamedir));
1355
1356         // look for the pop.lmp file and set registered to true if it is found
1357         if ((gamemode == GAME_NORMAL || gamemode == GAME_HIPNOTIC || gamemode == GAME_ROGUE) && !FS_FileExists("gfx/pop.lmp"))
1358         {
1359                 if (fs_modified)
1360                         Con_Print("Playing shareware version, with modification.\nwarning: most mods require full quake data.\n");
1361                 else
1362                         Con_Print("Playing shareware version.\n");
1363         }
1364         else
1365         {
1366                 Cvar_Set ("registered", "1");
1367                 if (gamemode == GAME_NORMAL || gamemode == GAME_HIPNOTIC || gamemode == GAME_ROGUE)
1368                         Con_Print("Playing registered version.\n");
1369         }
1370
1371         // unload all wads so that future queries will return the new data
1372         W_UnloadAll();
1373 }
1374
1375 void FS_Rescan_f(void)
1376 {
1377         FS_Rescan();
1378 }
1379
1380 /*
1381 ================
1382 FS_ChangeGameDirs
1383 ================
1384 */
1385 extern void Host_SaveConfig (void);
1386 extern void Host_LoadConfig_f (void);
1387 qboolean FS_ChangeGameDirs(int numgamedirs, char gamedirs[][MAX_QPATH], qboolean complain, qboolean failmissing)
1388 {
1389         int i;
1390         const char *p;
1391
1392         if (fs_numgamedirs == numgamedirs)
1393         {
1394                 for (i = 0;i < numgamedirs;i++)
1395                         if (strcasecmp(fs_gamedirs[i], gamedirs[i]))
1396                                 break;
1397                 if (i == numgamedirs)
1398                         return true; // already using this set of gamedirs, do nothing
1399         }
1400
1401         if (numgamedirs > MAX_GAMEDIRS)
1402         {
1403                 if (complain)
1404                         Con_Printf("That is too many gamedirs (%i > %i)\n", numgamedirs, MAX_GAMEDIRS);
1405                 return false; // too many gamedirs
1406         }
1407
1408         for (i = 0;i < numgamedirs;i++)
1409         {
1410                 // if string is nasty, reject it
1411                 p = FS_CheckGameDir(gamedirs[i]);
1412                 if(!p)
1413                 {
1414                         if (complain)
1415                                 Con_Printf("Nasty gamedir name rejected: %s\n", gamedirs[i]);
1416                         return false; // nasty gamedirs
1417                 }
1418                 if(p == fs_checkgamedir_missing && failmissing)
1419                 {
1420                         if (complain)
1421                                 Con_Printf("Gamedir missing: %s%s/\n", fs_basedir, gamedirs[i]);
1422                         return false; // missing gamedirs
1423                 }
1424         }
1425
1426         Host_SaveConfig();
1427
1428         fs_numgamedirs = numgamedirs;
1429         for (i = 0;i < fs_numgamedirs;i++)
1430                 strlcpy(fs_gamedirs[i], gamedirs[i], sizeof(fs_gamedirs[i]));
1431
1432         // reinitialize filesystem to detect the new paks
1433         FS_Rescan();
1434
1435         // exec the new config
1436         Host_LoadConfig_f();
1437
1438         // unload all sounds so they will be reloaded from the new files as needed
1439         S_UnloadAllSounds_f();
1440
1441         // reinitialize renderer (this reloads hud/console background/etc)
1442         R_Modules_Restart();
1443
1444         return true;
1445 }
1446
1447 /*
1448 ================
1449 FS_GameDir_f
1450 ================
1451 */
1452 void FS_GameDir_f (void)
1453 {
1454         int i;
1455         int numgamedirs;
1456         char gamedirs[MAX_GAMEDIRS][MAX_QPATH];
1457
1458         if (Cmd_Argc() < 2)
1459         {
1460                 Con_Printf("gamedirs active:");
1461                 for (i = 0;i < fs_numgamedirs;i++)
1462                         Con_Printf(" %s", fs_gamedirs[i]);
1463                 Con_Printf("\n");
1464                 return;
1465         }
1466
1467         numgamedirs = Cmd_Argc() - 1;
1468         if (numgamedirs > MAX_GAMEDIRS)
1469         {
1470                 Con_Printf("Too many gamedirs (%i > %i)\n", numgamedirs, MAX_GAMEDIRS);
1471                 return;
1472         }
1473
1474         for (i = 0;i < numgamedirs;i++)
1475                 strlcpy(gamedirs[i], Cmd_Argv(i+1), sizeof(gamedirs[i]));
1476
1477         if ((cls.state == ca_connected && !cls.demoplayback) || sv.active)
1478         {
1479                 // actually, changing during game would work fine, but would be stupid
1480                 Con_Printf("Can not change gamedir while client is connected or server is running!\n");
1481                 return;
1482         }
1483
1484         // halt demo playback to close the file
1485         CL_Disconnect();
1486
1487         FS_ChangeGameDirs(numgamedirs, gamedirs, true, true);
1488 }
1489
1490 static qfile_t* FS_SysOpen (const char* filepath, const char* mode, qboolean nonblocking);
1491 static const char *FS_SysCheckGameDir(const char *gamedir)
1492 {
1493         static char buf[8192];
1494         qboolean success;
1495         qfile_t *f;
1496         stringlist_t list;
1497         fs_offset_t n;
1498
1499         stringlistinit(&list);
1500         listdirectory(&list, gamedir, "");
1501         success = list.numstrings > 0;
1502         stringlistfreecontents(&list);
1503
1504         if(success)
1505         {
1506                 f = FS_SysOpen(va("%smodinfo.txt", gamedir), "r", false);
1507                 if(f)
1508                 {
1509                         n = FS_Read (f, buf, sizeof(buf) - 1);
1510                         if(n >= 0)
1511                                 buf[n] = 0;
1512                         else
1513                                 *buf = 0;
1514                         FS_Close(f);
1515                 }
1516                 else
1517                         *buf = 0;
1518                 return buf;
1519         }
1520
1521         return NULL;
1522 }
1523
1524 /*
1525 ================
1526 FS_CheckGameDir
1527 ================
1528 */
1529 const char *FS_CheckGameDir(const char *gamedir)
1530 {
1531         const char *ret;
1532
1533         if (FS_CheckNastyPath(gamedir, true))
1534                 return NULL;
1535
1536         ret = FS_SysCheckGameDir(va("%s%s/", fs_userdir, gamedir));
1537         if(ret)
1538         {
1539                 if(!*ret)
1540                 {
1541                         // get description from basedir
1542                         ret = FS_SysCheckGameDir(va("%s%s/", fs_basedir, gamedir));
1543                         if(ret)
1544                                 return ret;
1545                         return "";
1546                 }
1547                 return ret;
1548         }
1549
1550         ret = FS_SysCheckGameDir(va("%s%s/", fs_basedir, gamedir));
1551         if(ret)
1552                 return ret;
1553         
1554         return fs_checkgamedir_missing;
1555 }
1556
1557 static void FS_ListGameDirs(void)
1558 {
1559         stringlist_t list, list2;
1560         int i, j;
1561         const char *info;
1562
1563         fs_all_gamedirs_count = 0;
1564         if(fs_all_gamedirs)
1565                 Mem_Free(fs_all_gamedirs);
1566
1567         stringlistinit(&list);
1568         listdirectory(&list, va("%s/", fs_basedir), "");
1569         listdirectory(&list, va("%s/", fs_userdir), "");
1570         stringlistsort(&list);
1571
1572         stringlistinit(&list2);
1573         for(i = 0; i < list.numstrings; ++i)
1574         {
1575                 if(i)
1576                         if(!strcmp(list.strings[i-1], list.strings[i]))
1577                                 continue;
1578                 info = FS_CheckGameDir(list.strings[i]);
1579                 if(!info)
1580                         continue;
1581                 if(info == fs_checkgamedir_missing)
1582                         continue;
1583                 if(!*info)
1584                         continue;
1585                 stringlistappend(&list2, list.strings[i]); 
1586         }
1587         stringlistfreecontents(&list);
1588
1589         fs_all_gamedirs = (gamedir_t *)Mem_Alloc(fs_mempool, list2.numstrings * sizeof(*fs_all_gamedirs));
1590         for(i = 0; i < list2.numstrings; ++i)
1591         {
1592                 info = FS_CheckGameDir(list2.strings[i]);
1593                 // all this cannot happen any more, but better be safe than sorry
1594                 if(!info)
1595                         continue;
1596                 if(info == fs_checkgamedir_missing)
1597                         continue;
1598                 if(!*info)
1599                         continue;
1600                 strlcpy(fs_all_gamedirs[fs_all_gamedirs_count].name, list2.strings[i], sizeof(fs_all_gamedirs[j].name));
1601                 strlcpy(fs_all_gamedirs[fs_all_gamedirs_count].description, info, sizeof(fs_all_gamedirs[j].description));
1602                 ++fs_all_gamedirs_count;
1603         }
1604 }
1605
1606 /*
1607 ================
1608 FS_Init_SelfPack
1609 ================
1610 */
1611 void FS_Init_SelfPack (void)
1612 {
1613         PK3_OpenLibrary ();
1614         fs_mempool = Mem_AllocPool("file management", 0, NULL);
1615         if(com_selffd >= 0)
1616         {
1617                 fs_selfpack = FS_LoadPackPK3FromFD(com_argv[0], com_selffd);
1618                 if(fs_selfpack)
1619                 {
1620                         char *buf, *q;
1621                         const char *p;
1622                         FS_AddSelfPack();
1623                         buf = (char *) FS_LoadFile("darkplaces.opt", tempmempool, true, NULL);
1624                         if(buf)
1625                         {
1626                                 const char **new_argv;
1627                                 int i = 0;
1628                                 int args_left = 256;
1629                                 new_argv = (const char **)Mem_Alloc(fs_mempool, sizeof(*com_argv) * (com_argc + args_left + 2));
1630                                 if(com_argc == 0)
1631                                 {
1632                                         new_argv[0] = "dummy";
1633                                         com_argc = 1;
1634                                 }
1635                                 else
1636                                 {
1637                                         memcpy((char *)(&new_argv[0]), &com_argv[0], sizeof(*com_argv) * com_argc);
1638                                 }
1639                                 p = buf;
1640                                 while(COM_ParseToken_Console(&p))
1641                                 {
1642                                         if(i >= args_left)
1643                                                 break;
1644                                         q = (char *)Mem_Alloc(fs_mempool, strlen(com_token) + 1);
1645                                         strlcpy(q, com_token, strlen(com_token) + 1);
1646                                         new_argv[com_argc + i] = q;
1647                                         ++i;
1648                                 }
1649                                 new_argv[i+com_argc] = NULL;
1650                                 com_argv = new_argv;
1651                                 com_argc = com_argc + i;
1652                         }
1653                         Mem_Free(buf);
1654                 }
1655         }
1656 }
1657
1658 /*
1659 ================
1660 FS_Init
1661 ================
1662 */
1663 void FS_Init (void)
1664 {
1665         const char *p;
1666         int i;
1667 #ifdef WIN32
1668         TCHAR mydocsdir[MAX_PATH + 1];
1669 #if _MSC_VER >= 1400
1670         size_t homedirlen;
1671 #endif
1672 #endif
1673         char *homedir;
1674
1675 #ifdef WIN32
1676         const char* dllnames [] =
1677         {
1678                 "shfolder.dll",  // IE 4, or Win NT and higher
1679                 NULL
1680         };
1681         Sys_LoadLibrary(dllnames, &shfolder_dll, shfolderfuncs);
1682         // don't care for the result; if it fails, %USERPROFILE% will be used instead
1683 #endif
1684
1685         // Add the personal game directory
1686         if((i = COM_CheckParm("-userdir")) && i < com_argc - 1)
1687         {
1688                 dpsnprintf(fs_userdir, sizeof(fs_userdir), "%s/", com_argv[i+1]);
1689         }
1690         else if(COM_CheckParm("-nohome"))
1691         {
1692                 *fs_userdir = 0;
1693         }
1694         else
1695         {
1696 #ifdef WIN32
1697                 if(qSHGetFolderPath && (qSHGetFolderPath(NULL, CSIDL_PERSONAL, NULL, 0, mydocsdir) == S_OK))
1698                 {
1699                         dpsnprintf(fs_userdir, sizeof(fs_userdir), "%s/My Games/%s/", mydocsdir, gameuserdirname);
1700                         Con_DPrintf("Obtained personal directory %s from SHGetFolderPath\n", fs_userdir);
1701                 }
1702                 else
1703                 {
1704                         // use the environment
1705 #if _MSC_VER >= 1400
1706                         _dupenv_s (&homedir, &homedirlen, "USERPROFILE");
1707 #else
1708                         homedir = getenv("USERPROFILE");
1709 #endif
1710
1711                         if(homedir)
1712                         {
1713                                 dpsnprintf(fs_userdir, sizeof(fs_userdir), "%s/My Documents/My Games/%s/", homedir, gameuserdirname);
1714 #if _MSC_VER >= 1400
1715                                 free(homedir);
1716 #endif
1717                                 Con_DPrintf("Obtained personal directory %s from environment\n", fs_userdir);
1718                         }
1719                 }
1720
1721                 if(!*fs_userdir)
1722                         Con_DPrintf("Could not obtain home directory; not supporting -mygames\n");
1723 #else
1724                 homedir = getenv ("HOME");
1725                 if(homedir)
1726                         dpsnprintf(fs_userdir, sizeof(fs_userdir), "%s/.%s/", homedir, gameuserdirname);
1727
1728                 if(!*fs_userdir)
1729                         Con_DPrintf("Could not obtain home directory; assuming -nohome\n");
1730 #endif
1731
1732 #ifdef WIN32
1733                 if(!COM_CheckParm("-mygames"))
1734                 {
1735 #if _MSC_VER >= 1400
1736                         int fd;
1737                         _sopen_s(&fd, va("%s%s/config.cfg", fs_basedir, gamedirname1), O_WRONLY | O_CREAT, _SH_DENYNO, _S_IREAD | _S_IWRITE); // note: no O_TRUNC here!
1738 #else
1739                         int fd = open (va("%s%s/config.cfg", fs_basedir, gamedirname1), O_WRONLY | O_CREAT, 0666); // note: no O_TRUNC here!
1740 #endif
1741                         if(fd >= 0)
1742                         {
1743                                 close(fd);
1744                                 *fs_userdir = 0; // we have write access to the game dir, so let's use it
1745                         }
1746                 }
1747 #endif
1748         }
1749
1750         strlcpy(fs_gamedir, "", sizeof(fs_gamedir));
1751
1752 // If the base directory is explicitly defined by the compilation process
1753 #ifdef DP_FS_BASEDIR
1754         strlcpy(fs_basedir, DP_FS_BASEDIR, sizeof(fs_basedir));
1755 #else
1756         *fs_basedir = 0;
1757
1758 #ifdef MACOSX
1759         // FIXME: is there a better way to find the directory outside the .app?
1760         if (strstr(com_argv[0], ".app/"))
1761         {
1762                 char *split;
1763
1764                 split = strstr(com_argv[0], ".app/");
1765                 while (split > com_argv[0] && *split != '/')
1766                         split--;
1767                 strlcpy(fs_basedir, com_argv[0], sizeof(fs_basedir));
1768                 fs_basedir[split - com_argv[0]] = 0;
1769         }
1770 #endif
1771 #endif
1772
1773         // -basedir <path>
1774         // Overrides the system supplied base directory (under GAMENAME)
1775 // 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)
1776         i = COM_CheckParm ("-basedir");
1777         if (i && i < com_argc-1)
1778         {
1779                 strlcpy (fs_basedir, com_argv[i+1], sizeof (fs_basedir));
1780                 i = (int)strlen (fs_basedir);
1781                 if (i > 0 && (fs_basedir[i-1] == '\\' || fs_basedir[i-1] == '/'))
1782                         fs_basedir[i-1] = 0;
1783         }
1784
1785         // add a path separator to the end of the basedir if it lacks one
1786         if (fs_basedir[0] && fs_basedir[strlen(fs_basedir) - 1] != '/' && fs_basedir[strlen(fs_basedir) - 1] != '\\')
1787                 strlcat(fs_basedir, "/", sizeof(fs_basedir));
1788
1789         FS_ListGameDirs();
1790
1791         p = FS_CheckGameDir(gamedirname1);
1792         if(!p || p == fs_checkgamedir_missing)
1793                 Con_Printf("WARNING: base gamedir %s%s/ not found!\n", fs_basedir, gamedirname1);
1794
1795         if(gamedirname2)
1796         {
1797                 p = FS_CheckGameDir(gamedirname2);
1798                 if(!p || p == fs_checkgamedir_missing)
1799                         Con_Printf("WARNING: base gamedir %s%s/ not found!\n", fs_basedir, gamedirname2);
1800         }
1801
1802         // -game <gamedir>
1803         // Adds basedir/gamedir as an override game
1804         // LordHavoc: now supports multiple -game directories
1805         for (i = 1;i < com_argc && fs_numgamedirs < MAX_GAMEDIRS;i++)
1806         {
1807                 if (!com_argv[i])
1808                         continue;
1809                 if (!strcmp (com_argv[i], "-game") && i < com_argc-1)
1810                 {
1811                         i++;
1812                         p = FS_CheckGameDir(com_argv[i]);
1813                         if(!p)
1814                                 Sys_Error("Nasty -game name rejected: %s", com_argv[i]);
1815                         if(p == fs_checkgamedir_missing)
1816                                 Con_Printf("WARNING: -game %s%s/ not found!\n", fs_basedir, com_argv[i]);
1817                         // add the gamedir to the list of active gamedirs
1818                         strlcpy (fs_gamedirs[fs_numgamedirs], com_argv[i], sizeof(fs_gamedirs[fs_numgamedirs]));
1819                         fs_numgamedirs++;
1820                 }
1821         }
1822
1823         // generate the searchpath
1824         FS_Rescan();
1825 }
1826
1827 void FS_Init_Commands(void)
1828 {
1829         Cvar_RegisterVariable (&scr_screenshot_name);
1830         Cvar_RegisterVariable (&fs_empty_files_in_pack_mark_deletions);
1831         Cvar_RegisterVariable (&cvar_fs_gamedir);
1832
1833         Cmd_AddCommand ("gamedir", FS_GameDir_f, "changes active gamedir list (can take multiple arguments), not including base directory (example usage: gamedir ctf)");
1834         Cmd_AddCommand ("fs_rescan", FS_Rescan_f, "rescans filesystem for new pack archives and any other changes");
1835         Cmd_AddCommand ("path", FS_Path_f, "print searchpath (game directories and archives)");
1836         Cmd_AddCommand ("dir", FS_Dir_f, "list files in searchpath matching an * filename pattern, one per line");
1837         Cmd_AddCommand ("ls", FS_Ls_f, "list files in searchpath matching an * filename pattern, multiple per line");
1838         Cmd_AddCommand ("which", FS_Which_f, "accepts a file name as argument and reports where the file is taken from");
1839 }
1840
1841 /*
1842 ================
1843 FS_Shutdown
1844 ================
1845 */
1846 void FS_Shutdown (void)
1847 {
1848         // close all pack files and such
1849         // (hopefully there aren't any other open files, but they'll be cleaned up
1850         //  by the OS anyway)
1851         FS_ClearSearchPath();
1852         Mem_FreePool (&fs_mempool);
1853
1854 #ifdef WIN32
1855         Sys_UnloadLibrary (&shfolder_dll);
1856 #endif
1857 }
1858
1859 int FS_SysOpenFD(const char *filepath, const char *mode, qboolean nonblocking)
1860 {
1861         int handle;
1862         int mod, opt;
1863         unsigned int ind;
1864
1865         // Parse the mode string
1866         switch (mode[0])
1867         {
1868                 case 'r':
1869                         mod = O_RDONLY;
1870                         opt = 0;
1871                         break;
1872                 case 'w':
1873                         mod = O_WRONLY;
1874                         opt = O_CREAT | O_TRUNC;
1875                         break;
1876                 case 'a':
1877                         mod = O_WRONLY;
1878                         opt = O_CREAT | O_APPEND;
1879                         break;
1880                 default:
1881                         Con_Printf ("FS_SysOpen(%s, %s): invalid mode\n", filepath, mode);
1882                         return -1;
1883         }
1884         for (ind = 1; mode[ind] != '\0'; ind++)
1885         {
1886                 switch (mode[ind])
1887                 {
1888                         case '+':
1889                                 mod = O_RDWR;
1890                                 break;
1891                         case 'b':
1892                                 opt |= O_BINARY;
1893                                 break;
1894                         default:
1895                                 Con_Printf ("FS_SysOpen(%s, %s): unknown character in mode (%c)\n",
1896                                                         filepath, mode, mode[ind]);
1897                 }
1898         }
1899
1900         if (nonblocking)
1901                 opt |= O_NONBLOCK;
1902
1903 #if _MSC_VER >= 1400
1904         _sopen_s(&handle, filepath, mod | opt, _SH_DENYNO, _S_IREAD | _S_IWRITE);
1905 #else
1906         handle = open (filepath, mod | opt, 0666);
1907 #endif
1908         return handle;
1909 }
1910
1911 /*
1912 ====================
1913 FS_SysOpen
1914
1915 Internal function used to create a qfile_t and open the relevant non-packed file on disk
1916 ====================
1917 */
1918 static qfile_t* FS_SysOpen (const char* filepath, const char* mode, qboolean nonblocking)
1919 {
1920         qfile_t* file;
1921
1922         file = (qfile_t *)Mem_Alloc (fs_mempool, sizeof (*file));
1923         file->ungetc = EOF;
1924         file->handle = FS_SysOpenFD(filepath, mode, nonblocking);
1925         if (file->handle < 0)
1926         {
1927                 Mem_Free (file);
1928                 return NULL;
1929         }
1930
1931         file->filename = Mem_strdup(fs_mempool, filepath);
1932
1933         file->real_length = lseek (file->handle, 0, SEEK_END);
1934
1935         // For files opened in append mode, we start at the end of the file
1936         if (mode[0] == 'a')
1937                 file->position = file->real_length;
1938         else
1939                 lseek (file->handle, 0, SEEK_SET);
1940
1941         return file;
1942 }
1943
1944
1945 /*
1946 ===========
1947 FS_OpenPackedFile
1948
1949 Open a packed file using its package file descriptor
1950 ===========
1951 */
1952 qfile_t *FS_OpenPackedFile (pack_t* pack, int pack_ind)
1953 {
1954         packfile_t *pfile;
1955         int dup_handle;
1956         qfile_t* file;
1957
1958         pfile = &pack->files[pack_ind];
1959
1960         // If we don't have the true offset, get it now
1961         if (! (pfile->flags & PACKFILE_FLAG_TRUEOFFS))
1962                 if (!PK3_GetTrueFileOffset (pfile, pack))
1963                         return NULL;
1964
1965 #ifndef LINK_TO_ZLIB
1966         // No Zlib DLL = no compressed files
1967         if (!zlib_dll && (pfile->flags & PACKFILE_FLAG_DEFLATED))
1968         {
1969                 Con_Printf("WARNING: can't open the compressed file %s\n"
1970                                         "You need the Zlib DLL to use compressed files\n",
1971                                         pfile->name);
1972                 return NULL;
1973         }
1974 #endif
1975
1976         // LordHavoc: lseek affects all duplicates of a handle so we do it before
1977         // the dup() call to avoid having to close the dup_handle on error here
1978         if (lseek (pack->handle, pfile->offset, SEEK_SET) == -1)
1979         {
1980                 Con_Printf ("FS_OpenPackedFile: can't lseek to %s in %s (offset: %d)\n",
1981                                         pfile->name, pack->filename, (int) pfile->offset);
1982                 return NULL;
1983         }
1984
1985         dup_handle = dup (pack->handle);
1986         if (dup_handle < 0)
1987         {
1988                 Con_Printf ("FS_OpenPackedFile: can't dup package's handle (pack: %s)\n", pack->filename);
1989                 return NULL;
1990         }
1991
1992         file = (qfile_t *)Mem_Alloc (fs_mempool, sizeof (*file));
1993         memset (file, 0, sizeof (*file));
1994         file->handle = dup_handle;
1995         file->flags = QFILE_FLAG_PACKED;
1996         file->real_length = pfile->realsize;
1997         file->offset = pfile->offset;
1998         file->position = 0;
1999         file->ungetc = EOF;
2000
2001         if (pfile->flags & PACKFILE_FLAG_DEFLATED)
2002         {
2003                 ztoolkit_t *ztk;
2004
2005                 file->flags |= QFILE_FLAG_DEFLATED;
2006
2007                 // We need some more variables
2008                 ztk = (ztoolkit_t *)Mem_Alloc (fs_mempool, sizeof (*ztk));
2009
2010                 ztk->comp_length = pfile->packsize;
2011
2012                 // Initialize zlib stream
2013                 ztk->zstream.next_in = ztk->input;
2014                 ztk->zstream.avail_in = 0;
2015
2016                 /* From Zlib's "unzip.c":
2017                  *
2018                  * windowBits is passed < 0 to tell that there is no zlib header.
2019                  * Note that in this case inflate *requires* an extra "dummy" byte
2020                  * after the compressed stream in order to complete decompression and
2021                  * return Z_STREAM_END.
2022                  * In unzip, i don't wait absolutely Z_STREAM_END because I known the
2023                  * size of both compressed and uncompressed data
2024                  */
2025                 if (qz_inflateInit2 (&ztk->zstream, -MAX_WBITS) != Z_OK)
2026                 {
2027                         Con_Printf ("FS_OpenPackedFile: inflate init error (file: %s)\n", pfile->name);
2028                         close(dup_handle);
2029                         Mem_Free(file);
2030                         return NULL;
2031                 }
2032
2033                 ztk->zstream.next_out = file->buff;
2034                 ztk->zstream.avail_out = sizeof (file->buff);
2035
2036                 file->ztk = ztk;
2037         }
2038
2039         return file;
2040 }
2041
2042 /*
2043 ====================
2044 FS_CheckNastyPath
2045
2046 Return true if the path should be rejected due to one of the following:
2047 1: path elements that are non-portable
2048 2: path elements that would allow access to files outside the game directory,
2049    or are just not a good idea for a mod to be using.
2050 ====================
2051 */
2052 int FS_CheckNastyPath (const char *path, qboolean isgamedir)
2053 {
2054         // all: never allow an empty path, as for gamedir it would access the parent directory and a non-gamedir path it is just useless
2055         if (!path[0])
2056                 return 2;
2057
2058         // Windows: don't allow \ in filenames (windows-only), period.
2059         // (on Windows \ is a directory separator, but / is also supported)
2060         if (strstr(path, "\\"))
2061                 return 1; // non-portable
2062
2063         // Mac: don't allow Mac-only filenames - : is a directory separator
2064         // instead of /, but we rely on / working already, so there's no reason to
2065         // support a Mac-only path
2066         // Amiga and Windows: : tries to go to root of drive
2067         if (strstr(path, ":"))
2068                 return 1; // non-portable attempt to go to root of drive
2069
2070         // Amiga: // is parent directory
2071         if (strstr(path, "//"))
2072                 return 1; // non-portable attempt to go to parent directory
2073
2074         // all: don't allow going to parent directory (../ or /../)
2075         if (strstr(path, ".."))
2076                 return 2; // attempt to go outside the game directory
2077
2078         // Windows and UNIXes: don't allow absolute paths
2079         if (path[0] == '/')
2080                 return 2; // attempt to go outside the game directory
2081
2082         // all: don't allow . characters before the last slash (it should only be used in filenames, not path elements), this catches all imaginable cases of ./, ../, .../, etc
2083         if (strchr(path, '.'))
2084         {
2085                 if (isgamedir)
2086                 {
2087                         // gamedir is entirely path elements, so simply forbid . entirely
2088                         return 2;
2089                 }
2090                 if (strchr(path, '.') < strrchr(path, '/'))
2091                         return 2; // possible attempt to go outside the game directory
2092         }
2093
2094         // all: forbid trailing slash on gamedir
2095         if (isgamedir && path[strlen(path)-1] == '/')
2096                 return 2;
2097
2098         // all: forbid leading dot on any filename for any reason
2099         if (strstr(path, "/."))
2100                 return 2; // attempt to go outside the game directory
2101
2102         // after all these checks we're pretty sure it's a / separated filename
2103         // and won't do much if any harm
2104         return false;
2105 }
2106
2107
2108 /*
2109 ====================
2110 FS_FindFile
2111
2112 Look for a file in the packages and in the filesystem
2113
2114 Return the searchpath where the file was found (or NULL)
2115 and the file index in the package if relevant
2116 ====================
2117 */
2118 static searchpath_t *FS_FindFile (const char *name, int* index, qboolean quiet)
2119 {
2120         searchpath_t *search;
2121         pack_t *pak;
2122
2123         // search through the path, one element at a time
2124         for (search = fs_searchpaths;search;search = search->next)
2125         {
2126                 // is the element a pak file?
2127                 if (search->pack && !search->pack->vpack)
2128                 {
2129                         int (*strcmp_funct) (const char* str1, const char* str2);
2130                         int left, right, middle;
2131
2132                         pak = search->pack;
2133                         strcmp_funct = pak->ignorecase ? strcasecmp : strcmp;
2134
2135                         // Look for the file (binary search)
2136                         left = 0;
2137                         right = pak->numfiles - 1;
2138                         while (left <= right)
2139                         {
2140                                 int diff;
2141
2142                                 middle = (left + right) / 2;
2143                                 diff = strcmp_funct (pak->files[middle].name, name);
2144
2145                                 // Found it
2146                                 if (!diff)
2147                                 {
2148                                         if (fs_empty_files_in_pack_mark_deletions.integer && pak->files[middle].realsize == 0)
2149                                         {
2150                                                 // yes, but the first one is empty so we treat it as not being there
2151                                                 if (!quiet && developer_extra.integer)
2152                                                         Con_DPrintf("FS_FindFile: %s is marked as deleted\n", name);
2153
2154                                                 if (index != NULL)
2155                                                         *index = -1;
2156                                                 return NULL;
2157                                         }
2158
2159                                         if (!quiet && developer_extra.integer)
2160                                                 Con_DPrintf("FS_FindFile: %s in %s\n",
2161                                                                         pak->files[middle].name, pak->filename);
2162
2163                                         if (index != NULL)
2164                                                 *index = middle;
2165                                         return search;
2166                                 }
2167
2168                                 // If we're too far in the list
2169                                 if (diff > 0)
2170                                         right = middle - 1;
2171                                 else
2172                                         left = middle + 1;
2173                         }
2174                 }
2175                 else
2176                 {
2177                         char netpath[MAX_OSPATH];
2178                         dpsnprintf(netpath, sizeof(netpath), "%s%s", search->filename, name);
2179                         if (FS_SysFileExists (netpath))
2180                         {
2181                                 if (!quiet && developer_extra.integer)
2182                                         Con_DPrintf("FS_FindFile: %s\n", netpath);
2183
2184                                 if (index != NULL)
2185                                         *index = -1;
2186                                 return search;
2187                         }
2188                 }
2189         }
2190
2191         if (!quiet && developer_extra.integer)
2192                 Con_DPrintf("FS_FindFile: can't find %s\n", name);
2193
2194         if (index != NULL)
2195                 *index = -1;
2196         return NULL;
2197 }
2198
2199
2200 /*
2201 ===========
2202 FS_OpenReadFile
2203
2204 Look for a file in the search paths and open it in read-only mode
2205 ===========
2206 */
2207 qfile_t *FS_OpenReadFile (const char *filename, qboolean quiet, qboolean nonblocking, int symlinkLevels)
2208 {
2209         searchpath_t *search;
2210         int pack_ind;
2211
2212         search = FS_FindFile (filename, &pack_ind, quiet);
2213
2214         // Not found?
2215         if (search == NULL)
2216                 return NULL;
2217
2218         // Found in the filesystem?
2219         if (pack_ind < 0)
2220         {
2221                 // this works with vpacks, so we are fine
2222                 char path [MAX_OSPATH];
2223                 dpsnprintf (path, sizeof (path), "%s%s", search->filename, filename);
2224                 return FS_SysOpen (path, "rb", nonblocking);
2225         }
2226
2227         // So, we found it in a package...
2228
2229         // Is it a PK3 symlink?
2230         // TODO also handle directory symlinks by parsing the whole structure...
2231         // but heck, file symlinks are good enough for now
2232         if(search->pack->files[pack_ind].flags & PACKFILE_FLAG_SYMLINK)
2233         {
2234                 if(symlinkLevels <= 0)
2235                 {
2236                         Con_Printf("symlink: %s: too many levels of symbolic links\n", filename);
2237                         return NULL;
2238                 }
2239                 else
2240                 {
2241                         char linkbuf[MAX_QPATH];
2242                         fs_offset_t count;
2243                         qfile_t *linkfile = FS_OpenPackedFile (search->pack, pack_ind);
2244                         const char *mergeslash;
2245                         char *mergestart;
2246
2247                         if(!linkfile)
2248                                 return NULL;
2249                         count = FS_Read(linkfile, linkbuf, sizeof(linkbuf) - 1);
2250                         FS_Close(linkfile);
2251                         if(count < 0)
2252                                 return NULL;
2253                         linkbuf[count] = 0;
2254                         
2255                         // Now combine the paths...
2256                         mergeslash = strrchr(filename, '/');
2257                         mergestart = linkbuf;
2258                         if(!mergeslash)
2259                                 mergeslash = filename;
2260                         while(!strncmp(mergestart, "../", 3))
2261                         {
2262                                 mergestart += 3;
2263                                 while(mergeslash > filename)
2264                                 {
2265                                         --mergeslash;
2266                                         if(*mergeslash == '/')
2267                                                 break;
2268                                 }
2269                         }
2270                         // Now, mergestart will point to the path to be appended, and mergeslash points to where it should be appended
2271                         if(mergeslash == filename)
2272                         {
2273                                 // Either mergeslash == filename, then we just replace the name (done below)
2274                         }
2275                         else
2276                         {
2277                                 // Or, we append the name after mergeslash;
2278                                 // or rather, we can also shift the linkbuf so we can put everything up to and including mergeslash first
2279                                 int spaceNeeded = mergeslash - filename + 1;
2280                                 int spaceRemoved = mergestart - linkbuf;
2281                                 if(count - spaceRemoved + spaceNeeded >= MAX_QPATH)
2282                                 {
2283                                         Con_DPrintf("symlink: too long path rejected\n");
2284                                         return NULL;
2285                                 }
2286                                 memmove(linkbuf + spaceNeeded, linkbuf + spaceRemoved, count - spaceRemoved);
2287                                 memcpy(linkbuf, filename, spaceNeeded);
2288                                 linkbuf[count - spaceRemoved + spaceNeeded] = 0;
2289                                 mergestart = linkbuf;
2290                         }
2291                         if (!quiet && developer_loading.integer)
2292                                 Con_DPrintf("symlink: %s -> %s\n", filename, mergestart);
2293                         if(FS_CheckNastyPath (mergestart, false))
2294                         {
2295                                 Con_DPrintf("symlink: nasty path %s rejected\n", mergestart);
2296                                 return NULL;
2297                         }
2298                         return FS_OpenReadFile(mergestart, quiet, nonblocking, symlinkLevels - 1);
2299                 }
2300         }
2301
2302         return FS_OpenPackedFile (search->pack, pack_ind);
2303 }
2304
2305
2306 /*
2307 =============================================================================
2308
2309 MAIN PUBLIC FUNCTIONS
2310
2311 =============================================================================
2312 */
2313
2314 /*
2315 ====================
2316 FS_OpenRealFile
2317
2318 Open a file in the userpath. The syntax is the same as fopen
2319 Used for savegame scanning in menu, and all file writing.
2320 ====================
2321 */
2322 qfile_t* FS_OpenRealFile (const char* filepath, const char* mode, qboolean quiet)
2323 {
2324         char real_path [MAX_OSPATH];
2325
2326         if (FS_CheckNastyPath(filepath, false))
2327         {
2328                 Con_Printf("FS_OpenRealFile(\"%s\", \"%s\", %s): nasty filename rejected\n", filepath, mode, quiet ? "true" : "false");
2329                 return NULL;
2330         }
2331
2332         dpsnprintf (real_path, sizeof (real_path), "%s/%s", fs_gamedir, filepath); // this is never a vpack
2333
2334         // If the file is opened in "write", "append", or "read/write" mode,
2335         // create directories up to the file.
2336         if (mode[0] == 'w' || mode[0] == 'a' || strchr (mode, '+'))
2337                 FS_CreatePath (real_path);
2338         return FS_SysOpen (real_path, mode, false);
2339 }
2340
2341
2342 /*
2343 ====================
2344 FS_OpenVirtualFile
2345
2346 Open a file. The syntax is the same as fopen
2347 ====================
2348 */
2349 qfile_t* FS_OpenVirtualFile (const char* filepath, qboolean quiet)
2350 {
2351         if (FS_CheckNastyPath(filepath, false))
2352         {
2353                 Con_Printf("FS_OpenVirtualFile(\"%s\", %s): nasty filename rejected\n", filepath, quiet ? "true" : "false");
2354                 return NULL;
2355         }
2356
2357         return FS_OpenReadFile (filepath, quiet, false, 16);
2358 }
2359
2360
2361 /*
2362 ====================
2363 FS_FileFromData
2364
2365 Open a file. The syntax is the same as fopen
2366 ====================
2367 */
2368 qfile_t* FS_FileFromData (const unsigned char *data, const size_t size, qboolean quiet)
2369 {
2370         qfile_t* file;
2371         file = (qfile_t *)Mem_Alloc (fs_mempool, sizeof (*file));
2372         memset (file, 0, sizeof (*file));
2373         file->flags = QFILE_FLAG_DATA;
2374         file->ungetc = EOF;
2375         file->real_length = size;
2376         file->data = data;
2377         return file;
2378 }
2379
2380 /*
2381 ====================
2382 FS_Close
2383
2384 Close a file
2385 ====================
2386 */
2387 int FS_Close (qfile_t* file)
2388 {
2389         if(file->flags & QFILE_FLAG_DATA)
2390         {
2391                 Mem_Free(file);
2392                 return 0;
2393         }
2394
2395         if (close (file->handle))
2396                 return EOF;
2397
2398         if (file->filename)
2399         {
2400                 if (file->flags & QFILE_FLAG_REMOVE)
2401                         remove(file->filename);
2402
2403                 Mem_Free((void *) file->filename);
2404         }
2405
2406         if (file->ztk)
2407         {
2408                 qz_inflateEnd (&file->ztk->zstream);
2409                 Mem_Free (file->ztk);
2410         }
2411
2412         Mem_Free (file);
2413         return 0;
2414 }
2415
2416 void FS_RemoveOnClose(qfile_t* file)
2417 {
2418         file->flags |= QFILE_FLAG_REMOVE;
2419 }
2420
2421 /*
2422 ====================
2423 FS_Write
2424
2425 Write "datasize" bytes into a file
2426 ====================
2427 */
2428 fs_offset_t FS_Write (qfile_t* file, const void* data, size_t datasize)
2429 {
2430         fs_offset_t result;
2431
2432         // If necessary, seek to the exact file position we're supposed to be
2433         if (file->buff_ind != file->buff_len)
2434                 lseek (file->handle, file->buff_ind - file->buff_len, SEEK_CUR);
2435
2436         // Purge cached data
2437         FS_Purge (file);
2438
2439         // Write the buffer and update the position
2440         result = write (file->handle, data, (fs_offset_t)datasize);
2441         file->position = lseek (file->handle, 0, SEEK_CUR);
2442         if (file->real_length < file->position)
2443                 file->real_length = file->position;
2444
2445         if (result < 0)
2446                 return 0;
2447
2448         return result;
2449 }
2450
2451
2452 /*
2453 ====================
2454 FS_Read
2455
2456 Read up to "buffersize" bytes from a file
2457 ====================
2458 */
2459 fs_offset_t FS_Read (qfile_t* file, void* buffer, size_t buffersize)
2460 {
2461         fs_offset_t count, done;
2462
2463         if (buffersize == 0)
2464                 return 0;
2465
2466         // Get rid of the ungetc character
2467         if (file->ungetc != EOF)
2468         {
2469                 ((char*)buffer)[0] = file->ungetc;
2470                 buffersize--;
2471                 file->ungetc = EOF;
2472                 done = 1;
2473         }
2474         else
2475                 done = 0;
2476
2477         if(file->flags & QFILE_FLAG_DATA)
2478         {
2479                 size_t left = file->real_length - file->position;
2480                 if(buffersize > left)
2481                         buffersize = left;
2482                 memcpy(buffer, file->data + file->position, buffersize);
2483                 file->position += buffersize;
2484                 return buffersize;
2485         }
2486
2487         // First, we copy as many bytes as we can from "buff"
2488         if (file->buff_ind < file->buff_len)
2489         {
2490                 count = file->buff_len - file->buff_ind;
2491                 count = ((fs_offset_t)buffersize > count) ? count : (fs_offset_t)buffersize;
2492                 done += count;
2493                 memcpy (buffer, &file->buff[file->buff_ind], count);
2494                 file->buff_ind += count;
2495
2496                 buffersize -= count;
2497                 if (buffersize == 0)
2498                         return done;
2499         }
2500
2501         // NOTE: at this point, the read buffer is always empty
2502
2503         // If the file isn't compressed
2504         if (! (file->flags & QFILE_FLAG_DEFLATED))
2505         {
2506                 fs_offset_t nb;
2507
2508                 // We must take care to not read after the end of the file
2509                 count = file->real_length - file->position;
2510
2511                 // If we have a lot of data to get, put them directly into "buffer"
2512                 if (buffersize > sizeof (file->buff) / 2)
2513                 {
2514                         if (count > (fs_offset_t)buffersize)
2515                                 count = (fs_offset_t)buffersize;
2516                         lseek (file->handle, file->offset + file->position, SEEK_SET);
2517                         nb = read (file->handle, &((unsigned char*)buffer)[done], count);
2518                         if (nb > 0)
2519                         {
2520                                 done += nb;
2521                                 file->position += nb;
2522
2523                                 // Purge cached data
2524                                 FS_Purge (file);
2525                         }
2526                 }
2527                 else
2528                 {
2529                         if (count > (fs_offset_t)sizeof (file->buff))
2530                                 count = (fs_offset_t)sizeof (file->buff);
2531                         lseek (file->handle, file->offset + file->position, SEEK_SET);
2532                         nb = read (file->handle, file->buff, count);
2533                         if (nb > 0)
2534                         {
2535                                 file->buff_len = nb;
2536                                 file->position += nb;
2537
2538                                 // Copy the requested data in "buffer" (as much as we can)
2539                                 count = (fs_offset_t)buffersize > file->buff_len ? file->buff_len : (fs_offset_t)buffersize;
2540                                 memcpy (&((unsigned char*)buffer)[done], file->buff, count);
2541                                 file->buff_ind = count;
2542                                 done += count;
2543                         }
2544                 }
2545
2546                 return done;
2547         }
2548
2549         // If the file is compressed, it's more complicated...
2550         // We cycle through a few operations until we have read enough data
2551         while (buffersize > 0)
2552         {
2553                 ztoolkit_t *ztk = file->ztk;
2554                 int error;
2555
2556                 // NOTE: at this point, the read buffer is always empty
2557
2558                 // If "input" is also empty, we need to refill it
2559                 if (ztk->in_ind == ztk->in_len)
2560                 {
2561                         // If we are at the end of the file
2562                         if (file->position == file->real_length)
2563                                 return done;
2564
2565                         count = (fs_offset_t)(ztk->comp_length - ztk->in_position);
2566                         if (count > (fs_offset_t)sizeof (ztk->input))
2567                                 count = (fs_offset_t)sizeof (ztk->input);
2568                         lseek (file->handle, file->offset + (fs_offset_t)ztk->in_position, SEEK_SET);
2569                         if (read (file->handle, ztk->input, count) != count)
2570                         {
2571                                 Con_Printf ("FS_Read: unexpected end of file\n");
2572                                 break;
2573                         }
2574
2575                         ztk->in_ind = 0;
2576                         ztk->in_len = count;
2577                         ztk->in_position += count;
2578                 }
2579
2580                 ztk->zstream.next_in = &ztk->input[ztk->in_ind];
2581                 ztk->zstream.avail_in = (unsigned int)(ztk->in_len - ztk->in_ind);
2582
2583                 // Now that we are sure we have compressed data available, we need to determine
2584                 // if it's better to inflate it in "file->buff" or directly in "buffer"
2585
2586                 // Inflate the data in "file->buff"
2587                 if (buffersize < sizeof (file->buff) / 2)
2588                 {
2589                         ztk->zstream.next_out = file->buff;
2590                         ztk->zstream.avail_out = sizeof (file->buff);
2591                         error = qz_inflate (&ztk->zstream, Z_SYNC_FLUSH);
2592                         if (error != Z_OK && error != Z_STREAM_END)
2593                         {
2594                                 Con_Printf ("FS_Read: Can't inflate file\n");
2595                                 break;
2596                         }
2597                         ztk->in_ind = ztk->in_len - ztk->zstream.avail_in;
2598
2599                         file->buff_len = (fs_offset_t)sizeof (file->buff) - ztk->zstream.avail_out;
2600                         file->position += file->buff_len;
2601
2602                         // Copy the requested data in "buffer" (as much as we can)
2603                         count = (fs_offset_t)buffersize > file->buff_len ? file->buff_len : (fs_offset_t)buffersize;
2604                         memcpy (&((unsigned char*)buffer)[done], file->buff, count);
2605                         file->buff_ind = count;
2606                 }
2607
2608                 // Else, we inflate directly in "buffer"
2609                 else
2610                 {
2611                         ztk->zstream.next_out = &((unsigned char*)buffer)[done];
2612                         ztk->zstream.avail_out = (unsigned int)buffersize;
2613                         error = qz_inflate (&ztk->zstream, Z_SYNC_FLUSH);
2614                         if (error != Z_OK && error != Z_STREAM_END)
2615                         {
2616                                 Con_Printf ("FS_Read: Can't inflate file\n");
2617                                 break;
2618                         }
2619                         ztk->in_ind = ztk->in_len - ztk->zstream.avail_in;
2620
2621                         // How much data did it inflate?
2622                         count = (fs_offset_t)(buffersize - ztk->zstream.avail_out);
2623                         file->position += count;
2624
2625                         // Purge cached data
2626                         FS_Purge (file);
2627                 }
2628
2629                 done += count;
2630                 buffersize -= count;
2631         }
2632
2633         return done;
2634 }
2635
2636
2637 /*
2638 ====================
2639 FS_Print
2640
2641 Print a string into a file
2642 ====================
2643 */
2644 int FS_Print (qfile_t* file, const char *msg)
2645 {
2646         return (int)FS_Write (file, msg, strlen (msg));
2647 }
2648
2649 /*
2650 ====================
2651 FS_Printf
2652
2653 Print a string into a file
2654 ====================
2655 */
2656 int FS_Printf(qfile_t* file, const char* format, ...)
2657 {
2658         int result;
2659         va_list args;
2660
2661         va_start (args, format);
2662         result = FS_VPrintf (file, format, args);
2663         va_end (args);
2664
2665         return result;
2666 }
2667
2668
2669 /*
2670 ====================
2671 FS_VPrintf
2672
2673 Print a string into a file
2674 ====================
2675 */
2676 int FS_VPrintf (qfile_t* file, const char* format, va_list ap)
2677 {
2678         int len;
2679         fs_offset_t buff_size = MAX_INPUTLINE;
2680         char *tempbuff;
2681
2682         for (;;)
2683         {
2684                 tempbuff = (char *)Mem_Alloc (tempmempool, buff_size);
2685                 len = dpvsnprintf (tempbuff, buff_size, format, ap);
2686                 if (len >= 0 && len < buff_size)
2687                         break;
2688                 Mem_Free (tempbuff);
2689                 buff_size *= 2;
2690         }
2691
2692         len = write (file->handle, tempbuff, len);
2693         Mem_Free (tempbuff);
2694
2695         return len;
2696 }
2697
2698
2699 /*
2700 ====================
2701 FS_Getc
2702
2703 Get the next character of a file
2704 ====================
2705 */
2706 int FS_Getc (qfile_t* file)
2707 {
2708         unsigned char c;
2709
2710         if (FS_Read (file, &c, 1) != 1)
2711                 return EOF;
2712
2713         return c;
2714 }
2715
2716
2717 /*
2718 ====================
2719 FS_UnGetc
2720
2721 Put a character back into the read buffer (only supports one character!)
2722 ====================
2723 */
2724 int FS_UnGetc (qfile_t* file, unsigned char c)
2725 {
2726         // If there's already a character waiting to be read
2727         if (file->ungetc != EOF)
2728                 return EOF;
2729
2730         file->ungetc = c;
2731         return c;
2732 }
2733
2734
2735 /*
2736 ====================
2737 FS_Seek
2738
2739 Move the position index in a file
2740 ====================
2741 */
2742 int FS_Seek (qfile_t* file, fs_offset_t offset, int whence)
2743 {
2744         ztoolkit_t *ztk;
2745         unsigned char* buffer;
2746         fs_offset_t buffersize;
2747
2748         // Compute the file offset
2749         switch (whence)
2750         {
2751                 case SEEK_CUR:
2752                         offset += file->position - file->buff_len + file->buff_ind;
2753                         break;
2754
2755                 case SEEK_SET:
2756                         break;
2757
2758                 case SEEK_END:
2759                         offset += file->real_length;
2760                         break;
2761
2762                 default:
2763                         return -1;
2764         }
2765         if (offset < 0 || offset > file->real_length)
2766                 return -1;
2767
2768         if(file->flags & QFILE_FLAG_DATA)
2769         {
2770                 file->position = offset;
2771                 return 0;
2772         }
2773
2774         // If we have the data in our read buffer, we don't need to actually seek
2775         if (file->position - file->buff_len <= offset && offset <= file->position)
2776         {
2777                 file->buff_ind = offset + file->buff_len - file->position;
2778                 return 0;
2779         }
2780
2781         // Purge cached data
2782         FS_Purge (file);
2783
2784         // Unpacked or uncompressed files can seek directly
2785         if (! (file->flags & QFILE_FLAG_DEFLATED))
2786         {
2787                 if (lseek (file->handle, file->offset + offset, SEEK_SET) == -1)
2788                         return -1;
2789                 file->position = offset;
2790                 return 0;
2791         }
2792
2793         // Seeking in compressed files is more a hack than anything else,
2794         // but we need to support it, so here we go.
2795         ztk = file->ztk;
2796
2797         // If we have to go back in the file, we need to restart from the beginning
2798         if (offset <= file->position)
2799         {
2800                 ztk->in_ind = 0;
2801                 ztk->in_len = 0;
2802                 ztk->in_position = 0;
2803                 file->position = 0;
2804                 lseek (file->handle, file->offset, SEEK_SET);
2805
2806                 // Reset the Zlib stream
2807                 ztk->zstream.next_in = ztk->input;
2808                 ztk->zstream.avail_in = 0;
2809                 qz_inflateReset (&ztk->zstream);
2810         }
2811
2812         // We need a big buffer to force inflating into it directly
2813         buffersize = 2 * sizeof (file->buff);
2814         buffer = (unsigned char *)Mem_Alloc (tempmempool, buffersize);
2815
2816         // Skip all data until we reach the requested offset
2817         while (offset > file->position)
2818         {
2819                 fs_offset_t diff = offset - file->position;
2820                 fs_offset_t count, len;
2821
2822                 count = (diff > buffersize) ? buffersize : diff;
2823                 len = FS_Read (file, buffer, count);
2824                 if (len != count)
2825                 {
2826                         Mem_Free (buffer);
2827                         return -1;
2828                 }
2829         }
2830
2831         Mem_Free (buffer);
2832         return 0;
2833 }
2834
2835
2836 /*
2837 ====================
2838 FS_Tell
2839
2840 Give the current position in a file
2841 ====================
2842 */
2843 fs_offset_t FS_Tell (qfile_t* file)
2844 {
2845         return file->position - file->buff_len + file->buff_ind;
2846 }
2847
2848
2849 /*
2850 ====================
2851 FS_FileSize
2852
2853 Give the total size of a file
2854 ====================
2855 */
2856 fs_offset_t FS_FileSize (qfile_t* file)
2857 {
2858         return file->real_length;
2859 }
2860
2861
2862 /*
2863 ====================
2864 FS_Purge
2865
2866 Erases any buffered input or output data
2867 ====================
2868 */
2869 void FS_Purge (qfile_t* file)
2870 {
2871         file->buff_len = 0;
2872         file->buff_ind = 0;
2873         file->ungetc = EOF;
2874 }
2875
2876
2877 /*
2878 ============
2879 FS_LoadFile
2880
2881 Filename are relative to the quake directory.
2882 Always appends a 0 byte.
2883 ============
2884 */
2885 unsigned char *FS_LoadFile (const char *path, mempool_t *pool, qboolean quiet, fs_offset_t *filesizepointer)
2886 {
2887         qfile_t *file;
2888         unsigned char *buf = NULL;
2889         fs_offset_t filesize = 0;
2890
2891         file = FS_OpenVirtualFile(path, quiet);
2892         if (file)
2893         {
2894                 filesize = file->real_length;
2895                 if(filesize < 0)
2896                 {
2897                         Con_Printf("FS_LoadFile(\"%s\", pool, %s, filesizepointer): trying to open a non-regular file\n", path, quiet ? "true" : "false");
2898                         FS_Close(file);
2899                         return NULL;
2900                 }
2901
2902                 buf = (unsigned char *)Mem_Alloc (pool, filesize + 1);
2903                 buf[filesize] = '\0';
2904                 FS_Read (file, buf, filesize);
2905                 FS_Close (file);
2906                 if (developer_loadfile.integer)
2907                         Con_Printf("loaded file \"%s\" (%u bytes)\n", path, (unsigned int)filesize);
2908         }
2909
2910         if (filesizepointer)
2911                 *filesizepointer = filesize;
2912         return buf;
2913 }
2914
2915
2916 /*
2917 ============
2918 FS_WriteFile
2919
2920 The filename will be prefixed by the current game directory
2921 ============
2922 */
2923 qboolean FS_WriteFile (const char *filename, void *data, fs_offset_t len)
2924 {
2925         qfile_t *file;
2926
2927         file = FS_OpenRealFile(filename, "wb", false);
2928         if (!file)
2929         {
2930                 Con_Printf("FS_WriteFile: failed on %s\n", filename);
2931                 return false;
2932         }
2933
2934         Con_DPrintf("FS_WriteFile: %s (%u bytes)\n", filename, (unsigned int)len);
2935         FS_Write (file, data, len);
2936         FS_Close (file);
2937         return true;
2938 }
2939
2940
2941 /*
2942 =============================================================================
2943
2944 OTHERS PUBLIC FUNCTIONS
2945
2946 =============================================================================
2947 */
2948
2949 /*
2950 ============
2951 FS_StripExtension
2952 ============
2953 */
2954 void FS_StripExtension (const char *in, char *out, size_t size_out)
2955 {
2956         char *last = NULL;
2957         char currentchar;
2958
2959         if (size_out == 0)
2960                 return;
2961
2962         while ((currentchar = *in) && size_out > 1)
2963         {
2964                 if (currentchar == '.')
2965                         last = out;
2966                 else if (currentchar == '/' || currentchar == '\\' || currentchar == ':')
2967                         last = NULL;
2968                 *out++ = currentchar;
2969                 in++;
2970                 size_out--;
2971         }
2972         if (last)
2973                 *last = 0;
2974         else
2975                 *out = 0;
2976 }
2977
2978
2979 /*
2980 ==================
2981 FS_DefaultExtension
2982 ==================
2983 */
2984 void FS_DefaultExtension (char *path, const char *extension, size_t size_path)
2985 {
2986         const char *src;
2987
2988         // if path doesn't have a .EXT, append extension
2989         // (extension should include the .)
2990         src = path + strlen(path) - 1;
2991
2992         while (*src != '/' && src != path)
2993         {
2994                 if (*src == '.')
2995                         return;                 // it has an extension
2996                 src--;
2997         }
2998
2999         strlcat (path, extension, size_path);
3000 }
3001
3002
3003 /*
3004 ==================
3005 FS_FileType
3006
3007 Look for a file in the packages and in the filesystem
3008 ==================
3009 */
3010 int FS_FileType (const char *filename)
3011 {
3012         searchpath_t *search;
3013         char fullpath[MAX_OSPATH];
3014
3015         search = FS_FindFile (filename, NULL, true);
3016         if(!search)
3017                 return FS_FILETYPE_NONE;
3018
3019         if(search->pack && !search->pack->vpack)
3020                 return FS_FILETYPE_FILE; // TODO can't check directories in paks yet, maybe later
3021
3022         dpsnprintf(fullpath, sizeof(fullpath), "%s%s", search->filename, filename);
3023         return FS_SysFileType(fullpath);
3024 }
3025
3026
3027 /*
3028 ==================
3029 FS_FileExists
3030
3031 Look for a file in the packages and in the filesystem
3032 ==================
3033 */
3034 qboolean FS_FileExists (const char *filename)
3035 {
3036         return (FS_FindFile (filename, NULL, true) != NULL);
3037 }
3038
3039
3040 /*
3041 ==================
3042 FS_SysFileExists
3043
3044 Look for a file in the filesystem only
3045 ==================
3046 */
3047 int FS_SysFileType (const char *path)
3048 {
3049 #if WIN32
3050 // Sajt - some older sdks are missing this define
3051 # ifndef INVALID_FILE_ATTRIBUTES
3052 #  define INVALID_FILE_ATTRIBUTES ((DWORD)-1)
3053 # endif
3054
3055         DWORD result = GetFileAttributes(path);
3056
3057         if(result == INVALID_FILE_ATTRIBUTES)
3058                 return FS_FILETYPE_NONE;
3059
3060         if(result & FILE_ATTRIBUTE_DIRECTORY)
3061                 return FS_FILETYPE_DIRECTORY;
3062
3063         return FS_FILETYPE_FILE;
3064 #else
3065         struct stat buf;
3066
3067         if (stat (path,&buf) == -1)
3068                 return FS_FILETYPE_NONE;
3069
3070 #ifndef S_ISDIR
3071 #define S_ISDIR(a) (((a) & S_IFMT) == S_IFDIR)
3072 #endif
3073         if(S_ISDIR(buf.st_mode))
3074                 return FS_FILETYPE_DIRECTORY;
3075
3076         return FS_FILETYPE_FILE;
3077 #endif
3078 }
3079
3080 qboolean FS_SysFileExists (const char *path)
3081 {
3082         return FS_SysFileType (path) != FS_FILETYPE_NONE;
3083 }
3084
3085 void FS_mkdir (const char *path)
3086 {
3087 #if WIN32
3088         _mkdir (path);
3089 #else
3090         mkdir (path, 0777);
3091 #endif
3092 }
3093
3094 /*
3095 ===========
3096 FS_Search
3097
3098 Allocate and fill a search structure with information on matching filenames.
3099 ===========
3100 */
3101 fssearch_t *FS_Search(const char *pattern, int caseinsensitive, int quiet)
3102 {
3103         fssearch_t *search;
3104         searchpath_t *searchpath;
3105         pack_t *pak;
3106         int i, basepathlength, numfiles, numchars, resultlistindex, dirlistindex;
3107         stringlist_t resultlist;
3108         stringlist_t dirlist;
3109         const char *slash, *backslash, *colon, *separator;
3110         char *basepath;
3111         char temp[MAX_OSPATH];
3112
3113         for (i = 0;pattern[i] == '.' || pattern[i] == ':' || pattern[i] == '/' || pattern[i] == '\\';i++)
3114                 ;
3115
3116         if (i > 0)
3117         {
3118                 Con_Printf("Don't use punctuation at the beginning of a search pattern!\n");
3119                 return NULL;
3120         }
3121
3122         stringlistinit(&resultlist);
3123         stringlistinit(&dirlist);
3124         search = NULL;
3125         slash = strrchr(pattern, '/');
3126         backslash = strrchr(pattern, '\\');
3127         colon = strrchr(pattern, ':');
3128         separator = max(slash, backslash);
3129         separator = max(separator, colon);
3130         basepathlength = separator ? (separator + 1 - pattern) : 0;
3131         basepath = (char *)Mem_Alloc (tempmempool, basepathlength + 1);
3132         if (basepathlength)
3133                 memcpy(basepath, pattern, basepathlength);
3134         basepath[basepathlength] = 0;
3135
3136         // search through the path, one element at a time
3137         for (searchpath = fs_searchpaths;searchpath;searchpath = searchpath->next)
3138         {
3139                 // is the element a pak file?
3140                 if (searchpath->pack && !searchpath->pack->vpack)
3141                 {
3142                         // look through all the pak file elements
3143                         pak = searchpath->pack;
3144                         for (i = 0;i < pak->numfiles;i++)
3145                         {
3146                                 strlcpy(temp, pak->files[i].name, sizeof(temp));
3147                                 while (temp[0])
3148                                 {
3149                                         if (matchpattern(temp, (char *)pattern, true))
3150                                         {
3151                                                 for (resultlistindex = 0;resultlistindex < resultlist.numstrings;resultlistindex++)
3152                                                         if (!strcmp(resultlist.strings[resultlistindex], temp))
3153                                                                 break;
3154                                                 if (resultlistindex == resultlist.numstrings)
3155                                                 {
3156                                                         stringlistappend(&resultlist, temp);
3157                                                         if (!quiet && developer_loading.integer)
3158                                                                 Con_Printf("SearchPackFile: %s : %s\n", pak->filename, temp);
3159                                                 }
3160                                         }
3161                                         // strip off one path element at a time until empty
3162                                         // this way directories are added to the listing if they match the pattern
3163                                         slash = strrchr(temp, '/');
3164                                         backslash = strrchr(temp, '\\');
3165                                         colon = strrchr(temp, ':');
3166                                         separator = temp;
3167                                         if (separator < slash)
3168                                                 separator = slash;
3169                                         if (separator < backslash)
3170                                                 separator = backslash;
3171                                         if (separator < colon)
3172                                                 separator = colon;
3173                                         *((char *)separator) = 0;
3174                                 }
3175                         }
3176                 }
3177                 else
3178                 {
3179                         stringlist_t matchedSet, foundSet;
3180                         const char *start = pattern;
3181
3182                         stringlistinit(&matchedSet);
3183                         stringlistinit(&foundSet);
3184                         // add a first entry to the set
3185                         stringlistappend(&matchedSet, "");
3186                         // iterate through pattern's path
3187                         while (*start)
3188                         {
3189                                 const char *asterisk, *wildcard, *nextseparator, *prevseparator;
3190                                 char subpath[MAX_OSPATH];
3191                                 char subpattern[MAX_OSPATH];
3192
3193                                 // find the next wildcard
3194                                 wildcard = strchr(start, '?');
3195                                 asterisk = strchr(start, '*');
3196                                 if (asterisk && (!wildcard || asterisk < wildcard))
3197                                 {
3198                                         wildcard = asterisk;
3199                                 }
3200
3201                                 if (wildcard)
3202                                 {
3203                                         nextseparator = strchr( wildcard, '/' );
3204                                 }
3205                                 else
3206                                 {
3207                                         nextseparator = NULL;
3208                                 }
3209
3210                                 if( !nextseparator ) {
3211                                         nextseparator = start + strlen( start );
3212                                 }
3213
3214                                 // prevseparator points past the '/' right before the wildcard and nextseparator at the one following it (or at the end of the string)
3215                                 // copy everything up except nextseperator
3216                                 strlcpy(subpattern, pattern, min(sizeof(subpattern), (size_t) (nextseparator - pattern + 1)));
3217                                 // find the last '/' before the wildcard
3218                                 prevseparator = strrchr( subpattern, '/' );
3219                                 if (!prevseparator)
3220                                         prevseparator = subpattern;
3221                                 else
3222                                         prevseparator++;
3223                                 // copy everything from start to the previous including the '/' (before the wildcard)
3224                                 // everything up to start is already included in the path of matchedSet's entries
3225                                 strlcpy(subpath, start, min(sizeof(subpath), (size_t) ((prevseparator - subpattern) - (start - pattern) + 1)));
3226
3227                                 // for each entry in matchedSet try to open the subdirectories specified in subpath
3228                                 for( dirlistindex = 0 ; dirlistindex < matchedSet.numstrings ; dirlistindex++ ) {
3229                                         strlcpy( temp, matchedSet.strings[ dirlistindex ], sizeof(temp) );
3230                                         strlcat( temp, subpath, sizeof(temp) );
3231                                         listdirectory( &foundSet, searchpath->filename, temp );
3232                                 }
3233                                 if( dirlistindex == 0 ) {
3234                                         break;
3235                                 }
3236                                 // reset the current result set
3237                                 stringlistfreecontents( &matchedSet );
3238                                 // match against the pattern
3239                                 for( dirlistindex = 0 ; dirlistindex < foundSet.numstrings ; dirlistindex++ ) {
3240                                         const char *direntry = foundSet.strings[ dirlistindex ];
3241                                         if (matchpattern(direntry, subpattern, true)) {
3242                                                 stringlistappend( &matchedSet, direntry );
3243                                         }
3244                                 }
3245                                 stringlistfreecontents( &foundSet );
3246
3247                                 start = nextseparator;
3248                         }
3249
3250                         for (dirlistindex = 0;dirlistindex < matchedSet.numstrings;dirlistindex++)
3251                         {
3252                                 const char *temp = matchedSet.strings[dirlistindex];
3253                                 if (matchpattern(temp, (char *)pattern, true))
3254                                 {
3255                                         for (resultlistindex = 0;resultlistindex < resultlist.numstrings;resultlistindex++)
3256                                                 if (!strcmp(resultlist.strings[resultlistindex], temp))
3257                                                         break;
3258                                         if (resultlistindex == resultlist.numstrings)
3259                                         {
3260                                                 stringlistappend(&resultlist, temp);
3261                                                 if (!quiet && developer_loading.integer)
3262                                                         Con_Printf("SearchDirFile: %s\n", temp);
3263                                         }
3264                                 }
3265                         }
3266                         stringlistfreecontents( &matchedSet );
3267                 }
3268         }
3269
3270         if (resultlist.numstrings)
3271         {
3272                 stringlistsort(&resultlist);
3273                 numfiles = resultlist.numstrings;
3274                 numchars = 0;
3275                 for (resultlistindex = 0;resultlistindex < resultlist.numstrings;resultlistindex++)
3276                         numchars += (int)strlen(resultlist.strings[resultlistindex]) + 1;
3277                 search = (fssearch_t *)Z_Malloc(sizeof(fssearch_t) + numchars + numfiles * sizeof(char *));
3278                 search->filenames = (char **)((char *)search + sizeof(fssearch_t));
3279                 search->filenamesbuffer = (char *)((char *)search + sizeof(fssearch_t) + numfiles * sizeof(char *));
3280                 search->numfilenames = (int)numfiles;
3281                 numfiles = 0;
3282                 numchars = 0;
3283                 for (resultlistindex = 0;resultlistindex < resultlist.numstrings;resultlistindex++)
3284                 {
3285                         size_t textlen;
3286                         search->filenames[numfiles] = search->filenamesbuffer + numchars;
3287                         textlen = strlen(resultlist.strings[resultlistindex]) + 1;
3288                         memcpy(search->filenames[numfiles], resultlist.strings[resultlistindex], textlen);
3289                         numfiles++;
3290                         numchars += (int)textlen;
3291                 }
3292         }
3293         stringlistfreecontents(&resultlist);
3294
3295         Mem_Free(basepath);
3296         return search;
3297 }
3298
3299 void FS_FreeSearch(fssearch_t *search)
3300 {
3301         Z_Free(search);
3302 }
3303
3304 extern int con_linewidth;
3305 int FS_ListDirectory(const char *pattern, int oneperline)
3306 {
3307         int numfiles;
3308         int numcolumns;
3309         int numlines;
3310         int columnwidth;
3311         int linebufpos;
3312         int i, j, k, l;
3313         const char *name;
3314         char linebuf[MAX_INPUTLINE];
3315         fssearch_t *search;
3316         search = FS_Search(pattern, true, true);
3317         if (!search)
3318                 return 0;
3319         numfiles = search->numfilenames;
3320         if (!oneperline)
3321         {
3322                 // FIXME: the names could be added to one column list and then
3323                 // gradually shifted into the next column if they fit, and then the
3324                 // next to make a compact variable width listing but it's a lot more
3325                 // complicated...
3326                 // find width for columns
3327                 columnwidth = 0;
3328                 for (i = 0;i < numfiles;i++)
3329                 {
3330                         l = (int)strlen(search->filenames[i]);
3331                         if (columnwidth < l)
3332                                 columnwidth = l;
3333                 }
3334                 // count the spacing character
3335                 columnwidth++;
3336                 // calculate number of columns
3337                 numcolumns = con_linewidth / columnwidth;
3338                 // don't bother with the column printing if it's only one column
3339                 if (numcolumns >= 2)
3340                 {
3341                         numlines = (numfiles + numcolumns - 1) / numcolumns;
3342                         for (i = 0;i < numlines;i++)
3343                         {
3344                                 linebufpos = 0;
3345                                 for (k = 0;k < numcolumns;k++)
3346                                 {
3347                                         l = i * numcolumns + k;
3348                                         if (l < numfiles)
3349                                         {
3350                                                 name = search->filenames[l];
3351                                                 for (j = 0;name[j] && linebufpos + 1 < (int)sizeof(linebuf);j++)
3352                                                         linebuf[linebufpos++] = name[j];
3353                                                 // space out name unless it's the last on the line
3354                                                 if (k + 1 < numcolumns && l + 1 < numfiles)
3355                                                         for (;j < columnwidth && linebufpos + 1 < (int)sizeof(linebuf);j++)
3356                                                                 linebuf[linebufpos++] = ' ';
3357                                         }
3358                                 }
3359                                 linebuf[linebufpos] = 0;
3360                                 Con_Printf("%s\n", linebuf);
3361                         }
3362                 }
3363                 else
3364                         oneperline = true;
3365         }
3366         if (oneperline)
3367                 for (i = 0;i < numfiles;i++)
3368                         Con_Printf("%s\n", search->filenames[i]);
3369         FS_FreeSearch(search);
3370         return (int)numfiles;
3371 }
3372
3373 static void FS_ListDirectoryCmd (const char* cmdname, int oneperline)
3374 {
3375         const char *pattern;
3376         if (Cmd_Argc() > 3)
3377         {
3378                 Con_Printf("usage:\n%s [path/pattern]\n", cmdname);
3379                 return;
3380         }
3381         if (Cmd_Argc() == 2)
3382                 pattern = Cmd_Argv(1);
3383         else
3384                 pattern = "*";
3385         if (!FS_ListDirectory(pattern, oneperline))
3386                 Con_Print("No files found.\n");
3387 }
3388
3389 void FS_Dir_f(void)
3390 {
3391         FS_ListDirectoryCmd("dir", true);
3392 }
3393
3394 void FS_Ls_f(void)
3395 {
3396         FS_ListDirectoryCmd("ls", false);
3397 }
3398
3399 void FS_Which_f(void)
3400 {
3401         const char *filename;
3402         int index;
3403         searchpath_t *sp;
3404         if (Cmd_Argc() != 2)
3405         {
3406                 Con_Printf("usage:\n%s <file>\n", Cmd_Argv(0));
3407                 return;
3408         }  
3409         filename = Cmd_Argv(1);
3410         sp = FS_FindFile(filename, &index, true);
3411         if (!sp) {
3412                 Con_Printf("%s isn't anywhere\n", filename);
3413                 return;
3414         }
3415         if (sp->pack)
3416         {
3417                 if(sp->pack->vpack)
3418                         Con_Printf("%s is in virtual package %sdir\n", filename, sp->pack->shortname);
3419                 else
3420                         Con_Printf("%s is in package %s\n", filename, sp->pack->shortname);
3421         }
3422         else
3423                 Con_Printf("%s is file %s%s\n", filename, sp->filename, filename);
3424 }
3425
3426
3427 const char *FS_WhichPack(const char *filename)
3428 {
3429         int index;
3430         searchpath_t *sp = FS_FindFile(filename, &index, true);
3431         if(sp && sp->pack)
3432                 return sp->pack->shortname;
3433         else
3434                 return 0;
3435 }
3436
3437 /*
3438 ====================
3439 FS_IsRegisteredQuakePack
3440
3441 Look for a proof of purchase file file in the requested package
3442
3443 If it is found, this file should NOT be downloaded.
3444 ====================
3445 */
3446 qboolean FS_IsRegisteredQuakePack(const char *name)
3447 {
3448         searchpath_t *search;
3449         pack_t *pak;
3450
3451         // search through the path, one element at a time
3452         for (search = fs_searchpaths;search;search = search->next)
3453         {
3454                 if (search->pack && !search->pack->vpack && !strcasecmp(FS_FileWithoutPath(search->filename), name))
3455                         // TODO do we want to support vpacks in here too?
3456                 {
3457                         int (*strcmp_funct) (const char* str1, const char* str2);
3458                         int left, right, middle;
3459
3460                         pak = search->pack;
3461                         strcmp_funct = pak->ignorecase ? strcasecmp : strcmp;
3462
3463                         // Look for the file (binary search)
3464                         left = 0;
3465                         right = pak->numfiles - 1;
3466                         while (left <= right)
3467                         {
3468                                 int diff;
3469
3470                                 middle = (left + right) / 2;
3471                                 diff = !strcmp_funct (pak->files[middle].name, "gfx/pop.lmp");
3472
3473                                 // Found it
3474                                 if (!diff)
3475                                         return true;
3476
3477                                 // If we're too far in the list
3478                                 if (diff > 0)
3479                                         right = middle - 1;
3480                                 else
3481                                         left = middle + 1;
3482                         }
3483
3484                         // we found the requested pack but it is not registered quake
3485                         return false;
3486                 }
3487         }
3488
3489         return false;
3490 }
3491
3492 int FS_CRCFile(const char *filename, size_t *filesizepointer)
3493 {
3494         int crc = -1;
3495         unsigned char *filedata;
3496         fs_offset_t filesize;
3497         if (filesizepointer)
3498                 *filesizepointer = 0;
3499         if (!filename || !*filename)
3500                 return crc;
3501         filedata = FS_LoadFile(filename, tempmempool, true, &filesize);
3502         if (filedata)
3503         {
3504                 if (filesizepointer)
3505                         *filesizepointer = filesize;
3506                 crc = CRC_Block(filedata, filesize);
3507                 Mem_Free(filedata);
3508         }
3509         return crc;
3510 }
3511
3512 unsigned char *FS_Deflate(const unsigned char *data, size_t size, size_t *deflated_size, int level, mempool_t *mempool)
3513 {
3514         z_stream strm;
3515         unsigned char *out = NULL;
3516         unsigned char *tmp;
3517
3518         *deflated_size = 0;
3519 #ifndef LINK_TO_ZLIB
3520         if(!zlib_dll)
3521                 return NULL;
3522 #endif
3523
3524         memset(&strm, 0, sizeof(strm));
3525         strm.zalloc = Z_NULL;
3526         strm.zfree = Z_NULL;
3527         strm.opaque = Z_NULL;
3528
3529         if(level < 0)
3530                 level = Z_DEFAULT_COMPRESSION;
3531
3532         if(qz_deflateInit2(&strm, level, Z_DEFLATED, -MAX_WBITS, Z_MEMLEVEL_DEFAULT, Z_BINARY) != Z_OK)
3533         {
3534                 Con_Printf("FS_Deflate: deflate init error!\n");
3535                 return NULL;
3536         }
3537
3538         strm.next_in = (unsigned char*)data;
3539         strm.avail_in = size;
3540
3541         tmp = (unsigned char *) Mem_Alloc(tempmempool, size);
3542         if(!tmp)
3543         {
3544                 Con_Printf("FS_Deflate: not enough memory in tempmempool!\n");
3545                 qz_deflateEnd(&strm);
3546                 return NULL;
3547         }
3548
3549         strm.next_out = tmp;
3550         strm.avail_out = size;
3551
3552         if(qz_deflate(&strm, Z_FINISH) != Z_STREAM_END)
3553         {
3554                 Con_Printf("FS_Deflate: deflate failed!\n");
3555                 qz_deflateEnd(&strm);
3556                 Mem_Free(tmp);
3557                 return NULL;
3558         }
3559         
3560         if(qz_deflateEnd(&strm) != Z_OK)
3561         {
3562                 Con_Printf("FS_Deflate: deflateEnd failed\n");
3563                 Mem_Free(tmp);
3564                 return NULL;
3565         }
3566
3567         if(strm.total_out >= size)
3568         {
3569                 Con_Printf("FS_Deflate: deflate is useless on this data!\n");
3570                 Mem_Free(tmp);
3571                 return NULL;
3572         }
3573
3574         out = (unsigned char *) Mem_Alloc(mempool, strm.total_out);
3575         if(!out)
3576         {
3577                 Con_Printf("FS_Deflate: not enough memory in target mempool!\n");
3578                 Mem_Free(tmp);
3579                 return NULL;
3580         }
3581
3582         if(deflated_size)
3583                 *deflated_size = (size_t)strm.total_out;
3584
3585         memcpy(out, tmp, strm.total_out);
3586         Mem_Free(tmp);
3587         
3588         return out;
3589 }
3590
3591 static void AssertBufsize(sizebuf_t *buf, int length)
3592 {
3593         if(buf->cursize + length > buf->maxsize)
3594         {
3595                 int oldsize = buf->maxsize;
3596                 unsigned char *olddata;
3597                 olddata = buf->data;
3598                 buf->maxsize += length;
3599                 buf->data = (unsigned char *) Mem_Alloc(tempmempool, buf->maxsize);
3600                 if(olddata)
3601                 {
3602                         memcpy(buf->data, olddata, oldsize);
3603                         Mem_Free(olddata);
3604                 }
3605         }
3606 }
3607
3608 unsigned char *FS_Inflate(const unsigned char *data, size_t size, size_t *inflated_size, mempool_t *mempool)
3609 {
3610         int ret;
3611         z_stream strm;
3612         unsigned char *out = NULL;
3613         unsigned char tmp[2048];
3614         unsigned int have;
3615         sizebuf_t outbuf;
3616
3617         *inflated_size = 0;
3618 #ifndef LINK_TO_ZLIB
3619         if(!zlib_dll)
3620                 return NULL;
3621 #endif
3622
3623         memset(&outbuf, 0, sizeof(outbuf));
3624         outbuf.data = (unsigned char *) Mem_Alloc(tempmempool, sizeof(tmp));
3625         outbuf.maxsize = sizeof(tmp);
3626
3627         memset(&strm, 0, sizeof(strm));
3628         strm.zalloc = Z_NULL;
3629         strm.zfree = Z_NULL;
3630         strm.opaque = Z_NULL;
3631
3632         if(qz_inflateInit2(&strm, -MAX_WBITS) != Z_OK)
3633         {
3634                 Con_Printf("FS_Inflate: inflate init error!\n");
3635                 Mem_Free(outbuf.data);
3636                 return NULL;
3637         }
3638
3639         strm.next_in = (unsigned char*)data;
3640         strm.avail_in = size;
3641
3642         do
3643         {
3644                 strm.next_out = tmp;
3645                 strm.avail_out = sizeof(tmp);
3646                 ret = qz_inflate(&strm, Z_NO_FLUSH);
3647                 // it either returns Z_OK on progress, Z_STREAM_END on end
3648                 // or an error code
3649                 switch(ret)
3650                 {
3651                         case Z_STREAM_END:
3652                         case Z_OK:
3653                                 break;
3654                                 
3655                         case Z_STREAM_ERROR:
3656                                 Con_Print("FS_Inflate: stream error!\n");
3657                                 break;
3658                         case Z_DATA_ERROR:
3659                                 Con_Print("FS_Inflate: data error!\n");
3660                                 break;
3661                         case Z_MEM_ERROR:
3662                                 Con_Print("FS_Inflate: mem error!\n");
3663                                 break;
3664                         case Z_BUF_ERROR:
3665                                 Con_Print("FS_Inflate: buf error!\n");
3666                                 break;
3667                         default:
3668                                 Con_Print("FS_Inflate: unknown error!\n");
3669                                 break;
3670                                 
3671                 }
3672                 if(ret != Z_OK && ret != Z_STREAM_END)
3673                 {
3674                         Con_Printf("Error after inflating %u bytes\n", (unsigned)strm.total_in);
3675                         Mem_Free(outbuf.data);
3676                         qz_inflateEnd(&strm);
3677                         return NULL;
3678                 }
3679                 have = sizeof(tmp) - strm.avail_out;
3680                 AssertBufsize(&outbuf, max(have, sizeof(tmp)));
3681                 SZ_Write(&outbuf, tmp, have);
3682         } while(ret != Z_STREAM_END);
3683
3684         qz_inflateEnd(&strm);
3685
3686         out = (unsigned char *) Mem_Alloc(mempool, outbuf.cursize);
3687         if(!out)
3688         {
3689                 Con_Printf("FS_Inflate: not enough memory in target mempool!\n");
3690                 Mem_Free(outbuf.data);
3691                 return NULL;
3692         }
3693
3694         memcpy(out, outbuf.data, outbuf.cursize);
3695         Mem_Free(outbuf.data);
3696
3697         if(inflated_size)
3698                 *inflated_size = (size_t)outbuf.cursize;
3699         
3700         return out;
3701 }