]> icculus.org git repositories - divverent/darkplaces.git/blob - fs.c
8571aed1c7cae7ee291eeff9a23913b6d8609a14
[divverent/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 /*
59
60 All of Quake's data access is through a hierchal file system, but the contents
61 of the file system can be transparently merged from several sources.
62
63 The "base directory" is the path to the directory holding the quake.exe and
64 all game directories.  The sys_* files pass this to host_init in
65 quakeparms_t->basedir.  This can be overridden with the "-basedir" command
66 line parm to allow code debugging in a different directory.  The base
67 directory is only used during filesystem initialization.
68
69 The "game directory" is the first tree on the search path and directory that
70 all generated files (savegames, screenshots, demos, config files) will be
71 saved to.  This can be overridden with the "-game" command line parameter.
72 The game directory can never be changed while quake is executing.  This is a
73 precaution against having a malicious server instruct clients to write files
74 over areas they shouldn't.
75
76 */
77
78
79 /*
80 =============================================================================
81
82 CONSTANTS
83
84 =============================================================================
85 */
86
87 // Magic numbers of a ZIP file (big-endian format)
88 #define ZIP_DATA_HEADER 0x504B0304  // "PK\3\4"
89 #define ZIP_CDIR_HEADER 0x504B0102  // "PK\1\2"
90 #define ZIP_END_HEADER  0x504B0506  // "PK\5\6"
91
92 // Other constants for ZIP files
93 #define ZIP_MAX_COMMENTS_SIZE           ((unsigned short)0xFFFF)
94 #define ZIP_END_CDIR_SIZE                       22
95 #define ZIP_CDIR_CHUNK_BASE_SIZE        46
96 #define ZIP_LOCAL_CHUNK_BASE_SIZE       30
97
98 // Zlib constants (from zlib.h)
99 #define Z_SYNC_FLUSH    2
100 #define MAX_WBITS               15
101 #define Z_OK                    0
102 #define Z_STREAM_END    1
103 #define ZLIB_VERSION    "1.2.3"
104
105 // Uncomment the following line if the zlib DLL you have still uses
106 // the 1.1.x series calling convention on Win32 (WINAPI)
107 //#define ZLIB_USES_WINAPI
108
109
110 /*
111 =============================================================================
112
113 TYPES
114
115 =============================================================================
116 */
117
118 // Zlib stream (from zlib.h)
119 // Warning: some pointers we don't use directly have
120 // been cast to "void*" for a matter of simplicity
121 typedef struct
122 {
123         unsigned char                   *next_in;       // next input byte
124         unsigned int    avail_in;       // number of bytes available at next_in
125         unsigned long   total_in;       // total nb of input bytes read so far
126
127         unsigned char                   *next_out;      // next output byte should be put there
128         unsigned int    avail_out;      // remaining free space at next_out
129         unsigned long   total_out;      // total nb of bytes output so far
130
131         char                    *msg;           // last error message, NULL if no error
132         void                    *state;         // not visible by applications
133
134         void                    *zalloc;        // used to allocate the internal state
135         void                    *zfree;         // used to free the internal state
136         void                    *opaque;        // private data object passed to zalloc and zfree
137
138         int                             data_type;      // best guess about the data type: ascii or binary
139         unsigned long   adler;          // adler32 value of the uncompressed data
140         unsigned long   reserved;       // reserved for future use
141 } z_stream;
142
143
144 // inside a package (PAK or PK3)
145 #define QFILE_FLAG_PACKED (1 << 0)
146 // file is compressed using the deflate algorithm (PK3 only)
147 #define QFILE_FLAG_DEFLATED (1 << 1)
148
149 #define FILE_BUFF_SIZE 2048
150 typedef struct
151 {
152         z_stream        zstream;
153         size_t          comp_length;                    // length of the compressed file
154         size_t          in_ind, in_len;                 // input buffer current index and length
155         size_t          in_position;                    // position in the compressed file
156         unsigned char           input [FILE_BUFF_SIZE];
157 } ztoolkit_t;
158
159 struct qfile_s
160 {
161         int                             flags;
162         int                             handle;                                 // file descriptor
163         fs_offset_t             real_length;                    // uncompressed file size (for files opened in "read" mode)
164         fs_offset_t             position;                               // current position in the file
165         fs_offset_t             offset;                                 // offset into the package (0 if external file)
166         int                             ungetc;                                 // single stored character from ungetc, cleared to EOF when read
167
168         // Contents buffer
169         fs_offset_t             buff_ind, buff_len;             // buffer current index and length
170         unsigned char                   buff [FILE_BUFF_SIZE];
171
172         // For zipped files
173         ztoolkit_t*             ztk;
174 };
175
176
177 // ------ PK3 files on disk ------ //
178
179 // You can get the complete ZIP format description from PKWARE website
180
181 typedef struct pk3_endOfCentralDir_s
182 {
183         unsigned int signature;
184         unsigned short disknum;
185         unsigned short cdir_disknum;    // number of the disk with the start of the central directory
186         unsigned short localentries;    // number of entries in the central directory on this disk
187         unsigned short nbentries;               // total number of entries in the central directory on this disk
188         unsigned int cdir_size;                 // size of the central directory
189         unsigned int cdir_offset;               // with respect to the starting disk number
190         unsigned short comment_size;
191 } pk3_endOfCentralDir_t;
192
193
194 // ------ PAK files on disk ------ //
195 typedef struct dpackfile_s
196 {
197         char name[56];
198         int filepos, filelen;
199 } dpackfile_t;
200
201 typedef struct dpackheader_s
202 {
203         char id[4];
204         int dirofs;
205         int dirlen;
206 } dpackheader_t;
207
208
209 // Packages in memory
210 // the offset in packfile_t is the true contents offset
211 #define PACKFILE_FLAG_TRUEOFFS (1 << 0)
212 // file compressed using the deflate algorithm
213 #define PACKFILE_FLAG_DEFLATED (1 << 1)
214
215 typedef struct packfile_s
216 {
217         char name [MAX_QPATH];
218         int flags;
219         fs_offset_t offset;
220         fs_offset_t packsize;   // size in the package
221         fs_offset_t realsize;   // real file size (uncompressed)
222 } packfile_t;
223
224 typedef struct pack_s
225 {
226         char filename [MAX_OSPATH];
227         int handle;
228         int ignorecase;  // PK3 ignores case
229         int numfiles;
230         packfile_t *files;
231 } pack_t;
232
233
234 // Search paths for files (including packages)
235 typedef struct searchpath_s
236 {
237         // only one of filename / pack will be used
238         char filename[MAX_OSPATH];
239         pack_t *pack;
240         struct searchpath_s *next;
241 } searchpath_t;
242
243
244 /*
245 =============================================================================
246
247 FUNCTION PROTOTYPES
248
249 =============================================================================
250 */
251
252 void FS_Dir_f(void);
253 void FS_Ls_f(void);
254
255 static searchpath_t *FS_FindFile (const char *name, int* index, qboolean quiet);
256 static packfile_t* FS_AddFileToPack (const char* name, pack_t* pack,
257                                                                         fs_offset_t offset, fs_offset_t packsize,
258                                                                         fs_offset_t realsize, int flags);
259
260
261 /*
262 =============================================================================
263
264 VARIABLES
265
266 =============================================================================
267 */
268
269 mempool_t *fs_mempool;
270
271 searchpath_t *fs_searchpaths = NULL;
272
273 #define MAX_FILES_IN_PACK       65536
274
275 char fs_gamedir[MAX_OSPATH];
276 char fs_basedir[MAX_OSPATH];
277
278 // list of active game directories (empty if not running a mod)
279 int fs_numgamedirs = 0;
280 char fs_gamedirs[MAX_GAMEDIRS][MAX_QPATH];
281
282 cvar_t scr_screenshot_name = {0, "scr_screenshot_name","dp", "prefix name for saved screenshots (changes based on -game commandline, as well as which game mode is running)"};
283 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"};
284
285
286 /*
287 =============================================================================
288
289 PRIVATE FUNCTIONS - PK3 HANDLING
290
291 =============================================================================
292 */
293
294 // Functions exported from zlib
295 #if defined(WIN32) && defined(ZLIB_USES_WINAPI)
296 # define ZEXPORT WINAPI
297 #else
298 # define ZEXPORT
299 #endif
300
301 static int (ZEXPORT *qz_inflate) (z_stream* strm, int flush);
302 static int (ZEXPORT *qz_inflateEnd) (z_stream* strm);
303 static int (ZEXPORT *qz_inflateInit2_) (z_stream* strm, int windowBits, const char *version, int stream_size);
304 static int (ZEXPORT *qz_inflateReset) (z_stream* strm);
305
306 #define qz_inflateInit2(strm, windowBits) \
307         qz_inflateInit2_((strm), (windowBits), ZLIB_VERSION, sizeof(z_stream))
308
309 static dllfunction_t zlibfuncs[] =
310 {
311         {"inflate",                     (void **) &qz_inflate},
312         {"inflateEnd",          (void **) &qz_inflateEnd},
313         {"inflateInit2_",       (void **) &qz_inflateInit2_},
314         {"inflateReset",        (void **) &qz_inflateReset},
315         {NULL, NULL}
316 };
317
318 // Handle for Zlib DLL
319 static dllhandle_t zlib_dll = NULL;
320
321
322 /*
323 ====================
324 PK3_CloseLibrary
325
326 Unload the Zlib DLL
327 ====================
328 */
329 void PK3_CloseLibrary (void)
330 {
331         Sys_UnloadLibrary (&zlib_dll);
332 }
333
334
335 /*
336 ====================
337 PK3_OpenLibrary
338
339 Try to load the Zlib DLL
340 ====================
341 */
342 qboolean PK3_OpenLibrary (void)
343 {
344         const char* dllnames [] =
345         {
346 #if defined(WIN64)
347                 "zlib64.dll",
348 #elif defined(WIN32)
349 # ifdef ZLIB_USES_WINAPI
350                 "zlibwapi.dll",
351                 "zlib.dll",
352 # else
353                 "zlib1.dll",
354 # endif
355 #elif defined(MACOSX)
356                 "libz.dylib",
357 #else
358                 "libz.so.1",
359                 "libz.so",
360 #endif
361                 NULL
362         };
363
364         // Already loaded?
365         if (zlib_dll)
366                 return true;
367
368         // Load the DLL
369         if (! Sys_LoadLibrary (dllnames, &zlib_dll, zlibfuncs))
370         {
371                 Con_Printf ("Compressed files support disabled\n");
372                 return false;
373         }
374
375         Con_Printf ("Compressed files support enabled\n");
376         return true;
377 }
378
379
380 /*
381 ====================
382 PK3_GetEndOfCentralDir
383
384 Extract the end of the central directory from a PK3 package
385 ====================
386 */
387 qboolean PK3_GetEndOfCentralDir (const char *packfile, int packhandle, pk3_endOfCentralDir_t *eocd)
388 {
389         fs_offset_t filesize, maxsize;
390         unsigned char *buffer, *ptr;
391         int ind;
392
393         // Get the package size
394         filesize = lseek (packhandle, 0, SEEK_END);
395         if (filesize < ZIP_END_CDIR_SIZE)
396                 return false;
397
398         // Load the end of the file in memory
399         if (filesize < ZIP_MAX_COMMENTS_SIZE + ZIP_END_CDIR_SIZE)
400                 maxsize = filesize;
401         else
402                 maxsize = ZIP_MAX_COMMENTS_SIZE + ZIP_END_CDIR_SIZE;
403         buffer = (unsigned char *)Mem_Alloc (tempmempool, maxsize);
404         lseek (packhandle, filesize - maxsize, SEEK_SET);
405         if (read (packhandle, buffer, maxsize) != (fs_offset_t) maxsize)
406         {
407                 Mem_Free (buffer);
408                 return false;
409         }
410
411         // Look for the end of central dir signature around the end of the file
412         maxsize -= ZIP_END_CDIR_SIZE;
413         ptr = &buffer[maxsize];
414         ind = 0;
415         while (BuffBigLong (ptr) != ZIP_END_HEADER)
416         {
417                 if (ind == maxsize)
418                 {
419                         Mem_Free (buffer);
420                         return false;
421                 }
422
423                 ind++;
424                 ptr--;
425         }
426
427         memcpy (eocd, ptr, ZIP_END_CDIR_SIZE);
428         eocd->signature = LittleLong (eocd->signature);
429         eocd->disknum = LittleShort (eocd->disknum);
430         eocd->cdir_disknum = LittleShort (eocd->cdir_disknum);
431         eocd->localentries = LittleShort (eocd->localentries);
432         eocd->nbentries = LittleShort (eocd->nbentries);
433         eocd->cdir_size = LittleLong (eocd->cdir_size);
434         eocd->cdir_offset = LittleLong (eocd->cdir_offset);
435         eocd->comment_size = LittleShort (eocd->comment_size);
436
437         Mem_Free (buffer);
438
439         return true;
440 }
441
442
443 /*
444 ====================
445 PK3_BuildFileList
446
447 Extract the file list from a PK3 file
448 ====================
449 */
450 int PK3_BuildFileList (pack_t *pack, const pk3_endOfCentralDir_t *eocd)
451 {
452         unsigned char *central_dir, *ptr;
453         unsigned int ind;
454         fs_offset_t remaining;
455
456         // Load the central directory in memory
457         central_dir = (unsigned char *)Mem_Alloc (tempmempool, eocd->cdir_size);
458         lseek (pack->handle, eocd->cdir_offset, SEEK_SET);
459         read (pack->handle, central_dir, eocd->cdir_size);
460
461         // Extract the files properties
462         // The parsing is done "by hand" because some fields have variable sizes and
463         // the constant part isn't 4-bytes aligned, which makes the use of structs difficult
464         remaining = eocd->cdir_size;
465         pack->numfiles = 0;
466         ptr = central_dir;
467         for (ind = 0; ind < eocd->nbentries; ind++)
468         {
469                 fs_offset_t namesize, count;
470
471                 // Checking the remaining size
472                 if (remaining < ZIP_CDIR_CHUNK_BASE_SIZE)
473                 {
474                         Mem_Free (central_dir);
475                         return -1;
476                 }
477                 remaining -= ZIP_CDIR_CHUNK_BASE_SIZE;
478
479                 // Check header
480                 if (BuffBigLong (ptr) != ZIP_CDIR_HEADER)
481                 {
482                         Mem_Free (central_dir);
483                         return -1;
484                 }
485
486                 namesize = BuffLittleShort (&ptr[28]);  // filename length
487
488                 // Check encryption, compression, and attributes
489                 // 1st uint8  : general purpose bit flag
490                 //    Check bits 0 (encryption), 3 (data descriptor after the file), and 5 (compressed patched data (?))
491                 // 2nd uint8 : external file attributes
492                 //    Check bits 3 (file is a directory) and 5 (file is a volume (?))
493                 if ((ptr[8] & 0x29) == 0 && (ptr[38] & 0x18) == 0)
494                 {
495                         // Still enough bytes for the name?
496                         if (remaining < namesize || namesize >= (int)sizeof (*pack->files))
497                         {
498                                 Mem_Free (central_dir);
499                                 return -1;
500                         }
501
502                         // WinZip doesn't use the "directory" attribute, so we need to check the name directly
503                         if (ptr[ZIP_CDIR_CHUNK_BASE_SIZE + namesize - 1] != '/')
504                         {
505                                 char filename [sizeof (pack->files[0].name)];
506                                 fs_offset_t offset, packsize, realsize;
507                                 int flags;
508
509                                 // Extract the name (strip it if necessary)
510                                 namesize = min(namesize, (int)sizeof (filename) - 1);
511                                 memcpy (filename, &ptr[ZIP_CDIR_CHUNK_BASE_SIZE], namesize);
512                                 filename[namesize] = '\0';
513
514                                 if (BuffLittleShort (&ptr[10]))
515                                         flags = PACKFILE_FLAG_DEFLATED;
516                                 else
517                                         flags = 0;
518                                 offset = BuffLittleLong (&ptr[42]);
519                                 packsize = BuffLittleLong (&ptr[20]);
520                                 realsize = BuffLittleLong (&ptr[24]);
521                                 FS_AddFileToPack (filename, pack, offset, packsize, realsize, flags);
522                         }
523                 }
524
525                 // Skip the name, additionnal field, and comment
526                 // 1er uint16 : extra field length
527                 // 2eme uint16 : file comment length
528                 count = namesize + BuffLittleShort (&ptr[30]) + BuffLittleShort (&ptr[32]);
529                 ptr += ZIP_CDIR_CHUNK_BASE_SIZE + count;
530                 remaining -= count;
531         }
532
533         // If the package is empty, central_dir is NULL here
534         if (central_dir != NULL)
535                 Mem_Free (central_dir);
536         return pack->numfiles;
537 }
538
539
540 /*
541 ====================
542 FS_LoadPackPK3
543
544 Create a package entry associated with a PK3 file
545 ====================
546 */
547 pack_t *FS_LoadPackPK3 (const char *packfile)
548 {
549         int packhandle;
550         pk3_endOfCentralDir_t eocd;
551         pack_t *pack;
552         int real_nb_files;
553
554         packhandle = open (packfile, O_RDONLY | O_BINARY);
555         if (packhandle < 0)
556                 return NULL;
557
558         if (! PK3_GetEndOfCentralDir (packfile, packhandle, &eocd))
559         {
560                 Con_Printf ("%s is not a PK3 file\n", packfile);
561                 close(packhandle);
562                 return NULL;
563         }
564
565         // Multi-volume ZIP archives are NOT allowed
566         if (eocd.disknum != 0 || eocd.cdir_disknum != 0)
567         {
568                 Con_Printf ("%s is a multi-volume ZIP archive\n", packfile);
569                 close(packhandle);
570                 return NULL;
571         }
572
573         // We only need to do this test if MAX_FILES_IN_PACK is lesser than 65535
574         // since eocd.nbentries is an unsigned 16 bits integer
575 #if MAX_FILES_IN_PACK < 65535
576         if (eocd.nbentries > MAX_FILES_IN_PACK)
577         {
578                 Con_Printf ("%s contains too many files (%hu)\n", packfile, eocd.nbentries);
579                 close(packhandle);
580                 return NULL;
581         }
582 #endif
583
584         // Create a package structure in memory
585         pack = (pack_t *)Mem_Alloc(fs_mempool, sizeof (pack_t));
586         pack->ignorecase = true; // PK3 ignores case
587         strlcpy (pack->filename, packfile, sizeof (pack->filename));
588         pack->handle = packhandle;
589         pack->numfiles = eocd.nbentries;
590         pack->files = (packfile_t *)Mem_Alloc(fs_mempool, eocd.nbentries * sizeof(packfile_t));
591
592         real_nb_files = PK3_BuildFileList (pack, &eocd);
593         if (real_nb_files < 0)
594         {
595                 Con_Printf ("%s is not a valid PK3 file\n", packfile);
596                 close(pack->handle);
597                 Mem_Free(pack);
598                 return NULL;
599         }
600
601         Con_Printf("Added packfile %s (%i files)\n", packfile, real_nb_files);
602         return pack;
603 }
604
605
606 /*
607 ====================
608 PK3_GetTrueFileOffset
609
610 Find where the true file data offset is
611 ====================
612 */
613 qboolean PK3_GetTrueFileOffset (packfile_t *pfile, pack_t *pack)
614 {
615         unsigned char buffer [ZIP_LOCAL_CHUNK_BASE_SIZE];
616         fs_offset_t count;
617
618         // Already found?
619         if (pfile->flags & PACKFILE_FLAG_TRUEOFFS)
620                 return true;
621
622         // Load the local file description
623         lseek (pack->handle, pfile->offset, SEEK_SET);
624         count = read (pack->handle, buffer, ZIP_LOCAL_CHUNK_BASE_SIZE);
625         if (count != ZIP_LOCAL_CHUNK_BASE_SIZE || BuffBigLong (buffer) != ZIP_DATA_HEADER)
626         {
627                 Con_Printf ("Can't retrieve file %s in package %s\n", pfile->name, pack->filename);
628                 return false;
629         }
630
631         // Skip name and extra field
632         pfile->offset += BuffLittleShort (&buffer[26]) + BuffLittleShort (&buffer[28]) + ZIP_LOCAL_CHUNK_BASE_SIZE;
633
634         pfile->flags |= PACKFILE_FLAG_TRUEOFFS;
635         return true;
636 }
637
638
639 /*
640 =============================================================================
641
642 OTHER PRIVATE FUNCTIONS
643
644 =============================================================================
645 */
646
647
648 /*
649 ====================
650 FS_AddFileToPack
651
652 Add a file to the list of files contained into a package
653 ====================
654 */
655 static packfile_t* FS_AddFileToPack (const char* name, pack_t* pack,
656                                                                          fs_offset_t offset, fs_offset_t packsize,
657                                                                          fs_offset_t realsize, int flags)
658 {
659         int (*strcmp_funct) (const char* str1, const char* str2);
660         int left, right, middle;
661         packfile_t *pfile;
662
663         strcmp_funct = pack->ignorecase ? strcasecmp : strcmp;
664
665         // Look for the slot we should put that file into (binary search)
666         left = 0;
667         right = pack->numfiles - 1;
668         while (left <= right)
669         {
670                 int diff;
671
672                 middle = (left + right) / 2;
673                 diff = strcmp_funct (pack->files[middle].name, name);
674
675                 // If we found the file, there's a problem
676                 if (!diff)
677                         Con_Printf ("Package %s contains the file %s several times\n", pack->filename, name);
678
679                 // If we're too far in the list
680                 if (diff > 0)
681                         right = middle - 1;
682                 else
683                         left = middle + 1;
684         }
685
686         // We have to move the right of the list by one slot to free the one we need
687         pfile = &pack->files[left];
688         memmove (pfile + 1, pfile, (pack->numfiles - left) * sizeof (*pfile));
689         pack->numfiles++;
690
691         strlcpy (pfile->name, name, sizeof (pfile->name));
692         pfile->offset = offset;
693         pfile->packsize = packsize;
694         pfile->realsize = realsize;
695         pfile->flags = flags;
696
697         return pfile;
698 }
699
700
701 /*
702 ============
703 FS_CreatePath
704
705 Only used for FS_Open.
706 ============
707 */
708 void FS_CreatePath (char *path)
709 {
710         char *ofs, save;
711
712         for (ofs = path+1 ; *ofs ; ofs++)
713         {
714                 if (*ofs == '/' || *ofs == '\\')
715                 {
716                         // create the directory
717                         save = *ofs;
718                         *ofs = 0;
719                         FS_mkdir (path);
720                         *ofs = save;
721                 }
722         }
723 }
724
725
726 /*
727 ============
728 FS_Path_f
729
730 ============
731 */
732 void FS_Path_f (void)
733 {
734         searchpath_t *s;
735
736         Con_Print("Current search path:\n");
737         for (s=fs_searchpaths ; s ; s=s->next)
738         {
739                 if (s->pack)
740                         Con_Printf("%s (%i files)\n", s->pack->filename, s->pack->numfiles);
741                 else
742                         Con_Printf("%s\n", s->filename);
743         }
744 }
745
746
747 /*
748 =================
749 FS_LoadPackPAK
750
751 Takes an explicit (not game tree related) path to a pak file.
752
753 Loads the header and directory, adding the files at the beginning
754 of the list so they override previous pack files.
755 =================
756 */
757 pack_t *FS_LoadPackPAK (const char *packfile)
758 {
759         dpackheader_t header;
760         int i, numpackfiles;
761         int packhandle;
762         pack_t *pack;
763         dpackfile_t *info;
764
765         packhandle = open (packfile, O_RDONLY | O_BINARY);
766         if (packhandle < 0)
767                 return NULL;
768         read (packhandle, (void *)&header, sizeof(header));
769         if (memcmp(header.id, "PACK", 4))
770         {
771                 Con_Printf ("%s is not a packfile\n", packfile);
772                 close(packhandle);
773                 return NULL;
774         }
775         header.dirofs = LittleLong (header.dirofs);
776         header.dirlen = LittleLong (header.dirlen);
777
778         if (header.dirlen % sizeof(dpackfile_t))
779         {
780                 Con_Printf ("%s has an invalid directory size\n", packfile);
781                 close(packhandle);
782                 return NULL;
783         }
784
785         numpackfiles = header.dirlen / sizeof(dpackfile_t);
786
787         if (numpackfiles > MAX_FILES_IN_PACK)
788         {
789                 Con_Printf ("%s has %i files\n", packfile, numpackfiles);
790                 close(packhandle);
791                 return NULL;
792         }
793
794         info = (dpackfile_t *)Mem_Alloc(tempmempool, sizeof(*info) * numpackfiles);
795         lseek (packhandle, header.dirofs, SEEK_SET);
796         if(header.dirlen != read (packhandle, (void *)info, header.dirlen))
797         {
798                 Con_Printf("%s is an incomplete PAK, not loading\n", packfile);
799                 Mem_Free(info);
800                 close(packhandle);
801                 return NULL;
802         }
803
804         pack = (pack_t *)Mem_Alloc(fs_mempool, sizeof (pack_t));
805         pack->ignorecase = false; // PAK is case sensitive
806         strlcpy (pack->filename, packfile, sizeof (pack->filename));
807         pack->handle = packhandle;
808         pack->numfiles = 0;
809         pack->files = (packfile_t *)Mem_Alloc(fs_mempool, numpackfiles * sizeof(packfile_t));
810
811         // parse the directory
812         for (i = 0;i < numpackfiles;i++)
813         {
814                 fs_offset_t offset = LittleLong (info[i].filepos);
815                 fs_offset_t size = LittleLong (info[i].filelen);
816
817                 FS_AddFileToPack (info[i].name, pack, offset, size, size, PACKFILE_FLAG_TRUEOFFS);
818         }
819
820         Mem_Free(info);
821
822         Con_Printf("Added packfile %s (%i files)\n", packfile, numpackfiles);
823         return pack;
824 }
825
826 /*
827 ================
828 FS_AddPack_Fullpath
829
830 Adds the given pack to the search path.
831 The pack type is autodetected by the file extension.
832
833 Returns true if the file was successfully added to the
834 search path or if it was already included.
835
836 If keep_plain_dirs is set, the pack will be added AFTER the first sequence of
837 plain directories.
838 ================
839 */
840 static qboolean FS_AddPack_Fullpath(const char *pakfile, qboolean *already_loaded, qboolean keep_plain_dirs)
841 {
842         searchpath_t *search;
843         pack_t *pak = NULL;
844         const char *ext = FS_FileExtension(pakfile);
845
846         for(search = fs_searchpaths; search; search = search->next)
847         {
848                 if(search->pack && !strcasecmp(search->pack->filename, pakfile))
849                 {
850                         if(already_loaded)
851                                 *already_loaded = true;
852                         return true; // already loaded
853                 }
854         }
855
856         if(already_loaded)
857                 *already_loaded = false;
858
859         if(!strcasecmp(ext, "pak"))
860                 pak = FS_LoadPackPAK (pakfile);
861         else if(!strcasecmp(ext, "pk3"))
862                 pak = FS_LoadPackPK3 (pakfile);
863         else
864                 Con_Printf("\"%s\" does not have a pack extension\n", pakfile);
865
866         if (pak)
867         {
868                 if(keep_plain_dirs)
869                 {
870                         // find the first item whose next one is a pack or NULL
871                         searchpath_t *insertion_point = 0;
872                         if(fs_searchpaths && !fs_searchpaths->pack)
873                         {
874                                 insertion_point = fs_searchpaths;
875                                 for(;;)
876                                 {
877                                         if(!insertion_point->next)
878                                                 break;
879                                         if(insertion_point->next->pack)
880                                                 break;
881                                         insertion_point = insertion_point->next;
882                                 }
883                         }
884                         // If insertion_point is NULL, this means that either there is no
885                         // item in the list yet, or that the very first item is a pack. In
886                         // that case, we want to insert at the beginning...
887                         if(!insertion_point)
888                         {
889                                 search = (searchpath_t *)Mem_Alloc(fs_mempool, sizeof(searchpath_t));
890                                 search->pack = pak;
891                                 search->next = fs_searchpaths;
892                                 fs_searchpaths = search;
893                         }
894                         else
895                         // otherwise we want to append directly after insertion_point.
896                         {
897                                 search = (searchpath_t *)Mem_Alloc(fs_mempool, sizeof(searchpath_t));
898                                 search->pack = pak;
899                                 search->next = insertion_point->next;
900                                 insertion_point->next = search;
901                         }
902                 }
903                 else
904                 {
905                         search = (searchpath_t *)Mem_Alloc(fs_mempool, sizeof(searchpath_t));
906                         search->pack = pak;
907                         search->next = fs_searchpaths;
908                         fs_searchpaths = search;
909                 }
910                 return true;
911         }
912         else
913         {
914                 Con_Printf("unable to load pak \"%s\"\n", pakfile);
915                 return false;
916         }
917 }
918
919
920 /*
921 ================
922 FS_AddPack
923
924 Adds the given pack to the search path and searches for it in the game path.
925 The pack type is autodetected by the file extension.
926
927 Returns true if the file was successfully added to the
928 search path or if it was already included.
929
930 If keep_plain_dirs is set, the pack will be added AFTER the first sequence of
931 plain directories.
932 ================
933 */
934 qboolean FS_AddPack(const char *pakfile, qboolean *already_loaded, qboolean keep_plain_dirs)
935 {
936         char fullpath[MAX_QPATH];
937         int index;
938         searchpath_t *search;
939
940         if(already_loaded)
941                 *already_loaded = false;
942
943         // then find the real name...
944         search = FS_FindFile(pakfile, &index, true);
945         if(!search || search->pack)
946         {
947                 Con_Printf("could not find pak \"%s\"\n", pakfile);
948                 return false;
949         }
950
951         dpsnprintf(fullpath, sizeof(fullpath), "%s%s", search->filename, pakfile);
952
953         return FS_AddPack_Fullpath(fullpath, already_loaded, keep_plain_dirs);
954 }
955
956
957 /*
958 ================
959 FS_AddGameDirectory
960
961 Sets fs_gamedir, adds the directory to the head of the path,
962 then loads and adds pak1.pak pak2.pak ...
963 ================
964 */
965 void FS_AddGameDirectory (const char *dir)
966 {
967         int i;
968         stringlist_t list;
969         searchpath_t *search;
970         char pakfile[MAX_OSPATH];
971
972         strlcpy (fs_gamedir, dir, sizeof (fs_gamedir));
973
974         stringlistinit(&list);
975         listdirectory(&list, dir);
976         stringlistsort(&list);
977
978         // add any PAK package in the directory
979         for (i = 0;i < list.numstrings;i++)
980         {
981                 if (!strcasecmp(FS_FileExtension(list.strings[i]), "pak"))
982                 {
983                         dpsnprintf (pakfile, sizeof (pakfile), "%s%s", dir, list.strings[i]);
984                         FS_AddPack_Fullpath(pakfile, NULL, false);
985                 }
986         }
987
988         // add any PK3 package in the directory
989         for (i = 0;i < list.numstrings;i++)
990         {
991                 if (!strcasecmp(FS_FileExtension(list.strings[i]), "pk3"))
992                 {
993                         dpsnprintf (pakfile, sizeof (pakfile), "%s%s", dir, list.strings[i]);
994                         FS_AddPack_Fullpath(pakfile, NULL, false);
995                 }
996         }
997
998         stringlistfreecontents(&list);
999
1000         // Add the directory to the search path
1001         // (unpacked files have the priority over packed files)
1002         search = (searchpath_t *)Mem_Alloc(fs_mempool, sizeof(searchpath_t));
1003         strlcpy (search->filename, dir, sizeof (search->filename));
1004         search->next = fs_searchpaths;
1005         fs_searchpaths = search;
1006 }
1007
1008
1009 /*
1010 ================
1011 FS_AddGameHierarchy
1012 ================
1013 */
1014 void FS_AddGameHierarchy (const char *dir)
1015 {
1016         int i;
1017         char userdir[MAX_QPATH];
1018 #ifdef WIN32
1019         TCHAR mydocsdir[MAX_PATH + 1];
1020 #else
1021         const char *homedir;
1022 #endif
1023
1024         // Add the common game directory
1025         FS_AddGameDirectory (va("%s%s/", fs_basedir, dir));
1026
1027         *userdir = 0;
1028
1029         // Add the personal game directory
1030 #ifdef WIN32
1031         if(SHGetFolderPath(NULL, CSIDL_PERSONAL, NULL, 0, mydocsdir) == S_OK)
1032                 dpsnprintf(userdir, sizeof(userdir), "%s/My Games/%s/", mydocsdir, gameuserdirname);
1033         fprintf(stderr, "userdir = %s\n", userdir);
1034 #else
1035         homedir = getenv ("HOME");
1036         if(homedir)
1037                 dpsnprintf(userdir, sizeof(userdir), "%s/.%s/", homedir, gameuserdirname);
1038 #endif
1039
1040 #ifdef WIN32
1041         if(!COM_CheckParm("-mygames"))
1042         {
1043                 int fd = open (va("%s%s/config.cfg", fs_basedir, dir), O_WRONLY | O_CREAT, 0666); // note: no O_TRUNC here!
1044                 if(fd >= 0)
1045                 {
1046                         close(fd);
1047                         *userdir = 0; // we have write access to the game dir, so let's use it
1048                 }
1049         }
1050 #endif
1051
1052         if(COM_CheckParm("-nohome"))
1053                 *userdir = 0;
1054         
1055         if((i = COM_CheckParm("-userdir")) && i < com_argc - 1)
1056                 dpsnprintf(userdir, sizeof(userdir), "%s/", com_argv[i+1]);
1057
1058         if (*userdir)
1059                 FS_AddGameDirectory(va("%s%s/", userdir, dir));
1060 }
1061
1062
1063 /*
1064 ============
1065 FS_FileExtension
1066 ============
1067 */
1068 const char *FS_FileExtension (const char *in)
1069 {
1070         const char *separator, *backslash, *colon, *dot;
1071
1072         separator = strrchr(in, '/');
1073         backslash = strrchr(in, '\\');
1074         if (!separator || separator < backslash)
1075                 separator = backslash;
1076         colon = strrchr(in, ':');
1077         if (!separator || separator < colon)
1078                 separator = colon;
1079
1080         dot = strrchr(in, '.');
1081         if (dot == NULL || (separator && (dot < separator)))
1082                 return "";
1083
1084         return dot + 1;
1085 }
1086
1087
1088 /*
1089 ============
1090 FS_FileWithoutPath
1091 ============
1092 */
1093 const char *FS_FileWithoutPath (const char *in)
1094 {
1095         const char *separator, *backslash, *colon;
1096
1097         separator = strrchr(in, '/');
1098         backslash = strrchr(in, '\\');
1099         if (!separator || separator < backslash)
1100                 separator = backslash;
1101         colon = strrchr(in, ':');
1102         if (!separator || separator < colon)
1103                 separator = colon;
1104         return separator ? separator + 1 : in;
1105 }
1106
1107
1108 /*
1109 ================
1110 FS_ClearSearchPath
1111 ================
1112 */
1113 void FS_ClearSearchPath (void)
1114 {
1115         // unload all packs and directory information, close all pack files
1116         // (if a qfile is still reading a pack it won't be harmed because it used
1117         //  dup() to get its own handle already)
1118         while (fs_searchpaths)
1119         {
1120                 searchpath_t *search = fs_searchpaths;
1121                 fs_searchpaths = search->next;
1122                 if (search->pack)
1123                 {
1124                         // close the file
1125                         close(search->pack->handle);
1126                         // free any memory associated with it
1127                         if (search->pack->files)
1128                                 Mem_Free(search->pack->files);
1129                         Mem_Free(search->pack);
1130                 }
1131                 Mem_Free(search);
1132         }
1133 }
1134
1135
1136 /*
1137 ================
1138 FS_Rescan
1139 ================
1140 */
1141 void FS_Rescan (void)
1142 {
1143         int i;
1144         qboolean fs_modified = false;
1145
1146         FS_ClearSearchPath();
1147
1148         // add the game-specific paths
1149         // gamedirname1 (typically id1)
1150         FS_AddGameHierarchy (gamedirname1);
1151         // update the com_modname (used for server info)
1152         strlcpy(com_modname, gamedirname1, sizeof(com_modname));
1153
1154         // add the game-specific path, if any
1155         // (only used for mission packs and the like, which should set fs_modified)
1156         if (gamedirname2)
1157         {
1158                 fs_modified = true;
1159                 FS_AddGameHierarchy (gamedirname2);
1160         }
1161
1162         // -game <gamedir>
1163         // Adds basedir/gamedir as an override game
1164         // LordHavoc: now supports multiple -game directories
1165         // set the com_modname (reported in server info)
1166         for (i = 0;i < fs_numgamedirs;i++)
1167         {
1168                 fs_modified = true;
1169                 FS_AddGameHierarchy (fs_gamedirs[i]);
1170                 // update the com_modname (used server info)
1171                 strlcpy (com_modname, fs_gamedirs[i], sizeof (com_modname));
1172         }
1173
1174         // set the default screenshot name to either the mod name or the
1175         // gamemode screenshot name
1176         if (strcmp(com_modname, gamedirname1))
1177                 Cvar_SetQuick (&scr_screenshot_name, com_modname);
1178         else
1179                 Cvar_SetQuick (&scr_screenshot_name, gamescreenshotname);
1180
1181         // If "-condebug" is in the command line, remove the previous log file
1182         if (COM_CheckParm ("-condebug") != 0)
1183                 unlink (va("%s/qconsole.log", fs_gamedir));
1184
1185         // look for the pop.lmp file and set registered to true if it is found
1186         if ((gamemode == GAME_NORMAL || gamemode == GAME_HIPNOTIC || gamemode == GAME_ROGUE) && !FS_FileExists("gfx/pop.lmp"))
1187         {
1188                 if (fs_modified)
1189                         Con_Print("Playing shareware version, with modification.\nwarning: most mods require full quake data.\n");
1190                 else
1191                         Con_Print("Playing shareware version.\n");
1192         }
1193         else
1194         {
1195                 Cvar_Set ("registered", "1");
1196                 if (gamemode == GAME_NORMAL || gamemode == GAME_HIPNOTIC || gamemode == GAME_ROGUE)
1197                         Con_Print("Playing registered version.\n");
1198         }
1199
1200         // unload all wads so that future queries will return the new data
1201         W_UnloadAll();
1202 }
1203
1204 void FS_Rescan_f(void)
1205 {
1206         FS_Rescan();
1207 }
1208
1209 /*
1210 ================
1211 FS_ChangeGameDirs
1212 ================
1213 */
1214 extern void Host_SaveConfig (void);
1215 extern void Host_LoadConfig_f (void);
1216 qboolean FS_ChangeGameDirs(int numgamedirs, char gamedirs[][MAX_QPATH], qboolean complain, qboolean failmissing)
1217 {
1218         int i;
1219
1220         if (fs_numgamedirs == numgamedirs)
1221         {
1222                 for (i = 0;i < numgamedirs;i++)
1223                         if (strcasecmp(fs_gamedirs[i], gamedirs[i]))
1224                                 break;
1225                 if (i == numgamedirs)
1226                         return true; // already using this set of gamedirs, do nothing
1227         }
1228
1229         if (numgamedirs > MAX_GAMEDIRS)
1230         {
1231                 if (complain)
1232                         Con_Printf("That is too many gamedirs (%i > %i)\n", numgamedirs, MAX_GAMEDIRS);
1233                 return false; // too many gamedirs
1234         }
1235
1236         for (i = 0;i < numgamedirs;i++)
1237         {
1238                 // if string is nasty, reject it
1239                 if(FS_CheckNastyPath(gamedirs[i], true))
1240                 {
1241                         if (complain)
1242                                 Con_Printf("Nasty gamedir name rejected: %s\n", gamedirs[i]);
1243                         return false; // nasty gamedirs
1244                 }
1245         }
1246
1247         for (i = 0;i < numgamedirs;i++)
1248         {
1249                 if (!FS_CheckGameDir(gamedirs[i]) && failmissing)
1250                 {
1251                         if (complain)
1252                                 Con_Printf("Gamedir missing: %s%s/\n", fs_basedir, gamedirs[i]);
1253                         return false; // missing gamedirs
1254                 }
1255         }
1256
1257         // halt demo playback to close the file
1258         CL_Disconnect();
1259
1260         Host_SaveConfig();
1261
1262         fs_numgamedirs = numgamedirs;
1263         for (i = 0;i < fs_numgamedirs;i++)
1264                 strlcpy(fs_gamedirs[i], gamedirs[i], sizeof(fs_gamedirs[i]));
1265
1266         // reinitialize filesystem to detect the new paks
1267         FS_Rescan();
1268
1269         // exec the new config
1270         Host_LoadConfig_f();
1271
1272         // unload all sounds so they will be reloaded from the new files as needed
1273         S_UnloadAllSounds_f();
1274
1275         // reinitialize renderer (this reloads hud/console background/etc)
1276         R_Modules_Restart();
1277
1278         return true;
1279 }
1280
1281 /*
1282 ================
1283 FS_GameDir_f
1284 ================
1285 */
1286 void FS_GameDir_f (void)
1287 {
1288         int i;
1289         int numgamedirs;
1290         char gamedirs[MAX_GAMEDIRS][MAX_QPATH];
1291
1292         if (Cmd_Argc() < 2)
1293         {
1294                 Con_Printf("gamedirs active:");
1295                 for (i = 0;i < fs_numgamedirs;i++)
1296                         Con_Printf(" %s", fs_gamedirs[i]);
1297                 Con_Printf("\n");
1298                 return;
1299         }
1300
1301         numgamedirs = Cmd_Argc() - 1;
1302         if (numgamedirs > MAX_GAMEDIRS)
1303         {
1304                 Con_Printf("Too many gamedirs (%i > %i)\n", numgamedirs, MAX_GAMEDIRS);
1305                 return;
1306         }
1307
1308         for (i = 0;i < numgamedirs;i++)
1309                 strlcpy(gamedirs[i], Cmd_Argv(i+1), sizeof(gamedirs[i]));
1310
1311         if ((cls.state == ca_connected && !cls.demoplayback) || sv.active)
1312         {
1313                 // actually, changing during game would work fine, but would be stupid
1314                 Con_Printf("Can not change gamedir while client is connected or server is running!\n");
1315                 return;
1316         }
1317
1318         FS_ChangeGameDirs(numgamedirs, gamedirs, true, true);
1319 }
1320
1321
1322 /*
1323 ================
1324 FS_CheckGameDir
1325 ================
1326 */
1327 qboolean FS_CheckGameDir(const char *gamedir)
1328 {
1329         qboolean success;
1330         stringlist_t list;
1331         stringlistinit(&list);
1332         listdirectory(&list, va("%s%s/", fs_basedir, gamedir));
1333         success = list.numstrings > 0;
1334         stringlistfreecontents(&list);
1335         return success;
1336 }
1337
1338
1339 /*
1340 ================
1341 FS_Init
1342 ================
1343 */
1344 void FS_Init (void)
1345 {
1346         int i;
1347
1348         fs_mempool = Mem_AllocPool("file management", 0, NULL);
1349
1350         strlcpy(fs_gamedir, "", sizeof(fs_gamedir));
1351
1352 // If the base directory is explicitly defined by the compilation process
1353 #ifdef DP_FS_BASEDIR
1354         strlcpy(fs_basedir, DP_FS_BASEDIR, sizeof(fs_basedir));
1355 #else
1356         strlcpy(fs_basedir, "", sizeof(fs_basedir));
1357
1358 #ifdef MACOSX
1359         // FIXME: is there a better way to find the directory outside the .app?
1360         if (strstr(com_argv[0], ".app/"))
1361         {
1362                 char *split;
1363
1364                 split = strstr(com_argv[0], ".app/");
1365                 while (split > com_argv[0] && *split != '/')
1366                         split--;
1367                 strlcpy(fs_basedir, com_argv[0], sizeof(fs_basedir));
1368                 fs_basedir[split - com_argv[0]] = 0;
1369         }
1370 #endif
1371 #endif
1372
1373         PK3_OpenLibrary ();
1374
1375         // -basedir <path>
1376         // Overrides the system supplied base directory (under GAMENAME)
1377 // 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)
1378         i = COM_CheckParm ("-basedir");
1379         if (i && i < com_argc-1)
1380         {
1381                 strlcpy (fs_basedir, com_argv[i+1], sizeof (fs_basedir));
1382                 i = (int)strlen (fs_basedir);
1383                 if (i > 0 && (fs_basedir[i-1] == '\\' || fs_basedir[i-1] == '/'))
1384                         fs_basedir[i-1] = 0;
1385         }
1386
1387         // add a path separator to the end of the basedir if it lacks one
1388         if (fs_basedir[0] && fs_basedir[strlen(fs_basedir) - 1] != '/' && fs_basedir[strlen(fs_basedir) - 1] != '\\')
1389                 strlcat(fs_basedir, "/", sizeof(fs_basedir));
1390
1391         if (!FS_CheckGameDir(gamedirname1))
1392                 Sys_Error("base gamedir %s%s/ not found!\n", fs_basedir, gamedirname1);
1393
1394         if (gamedirname2 && !FS_CheckGameDir(gamedirname2))
1395                 Sys_Error("base gamedir %s%s/ not found!\n", fs_basedir, gamedirname2);
1396
1397         // -game <gamedir>
1398         // Adds basedir/gamedir as an override game
1399         // LordHavoc: now supports multiple -game directories
1400         for (i = 1;i < com_argc && fs_numgamedirs < MAX_GAMEDIRS;i++)
1401         {
1402                 if (!com_argv[i])
1403                         continue;
1404                 if (!strcmp (com_argv[i], "-game") && i < com_argc-1)
1405                 {
1406                         i++;
1407                         if (FS_CheckNastyPath(com_argv[i], true))
1408                                 Sys_Error("-game %s%s/ is a dangerous/non-portable path\n", fs_basedir, com_argv[i]);
1409                         if (!FS_CheckGameDir(com_argv[i]))
1410                                 Sys_Error("-game %s%s/ not found!\n", fs_basedir, com_argv[i]);
1411                         // add the gamedir to the list of active gamedirs
1412                         strlcpy (fs_gamedirs[fs_numgamedirs], com_argv[i], sizeof(fs_gamedirs[fs_numgamedirs]));
1413                         fs_numgamedirs++;
1414                 }
1415         }
1416
1417         // generate the searchpath
1418         FS_Rescan();
1419 }
1420
1421 void FS_Init_Commands(void)
1422 {
1423         Cvar_RegisterVariable (&scr_screenshot_name);
1424         Cvar_RegisterVariable (&fs_empty_files_in_pack_mark_deletions);
1425
1426         Cmd_AddCommand ("gamedir", FS_GameDir_f, "changes active gamedir list (can take multiple arguments), not including base directory (example usage: gamedir ctf)");
1427         Cmd_AddCommand ("fs_rescan", FS_Rescan_f, "rescans filesystem for new pack archives and any other changes");
1428         Cmd_AddCommand ("path", FS_Path_f, "print searchpath (game directories and archives)");
1429         Cmd_AddCommand ("dir", FS_Dir_f, "list files in searchpath matching an * filename pattern, one per line");
1430         Cmd_AddCommand ("ls", FS_Ls_f, "list files in searchpath matching an * filename pattern, multiple per line");
1431 }
1432
1433 /*
1434 ================
1435 FS_Shutdown
1436 ================
1437 */
1438 void FS_Shutdown (void)
1439 {
1440         // close all pack files and such
1441         // (hopefully there aren't any other open files, but they'll be cleaned up
1442         //  by the OS anyway)
1443         FS_ClearSearchPath();
1444         Mem_FreePool (&fs_mempool);
1445 }
1446
1447 /*
1448 ====================
1449 FS_SysOpen
1450
1451 Internal function used to create a qfile_t and open the relevant non-packed file on disk
1452 ====================
1453 */
1454 static qfile_t* FS_SysOpen (const char* filepath, const char* mode, qboolean nonblocking)
1455 {
1456         qfile_t* file;
1457         int mod, opt;
1458         unsigned int ind;
1459
1460         // Parse the mode string
1461         switch (mode[0])
1462         {
1463                 case 'r':
1464                         mod = O_RDONLY;
1465                         opt = 0;
1466                         break;
1467                 case 'w':
1468                         mod = O_WRONLY;
1469                         opt = O_CREAT | O_TRUNC;
1470                         break;
1471                 case 'a':
1472                         mod = O_WRONLY;
1473                         opt = O_CREAT | O_APPEND;
1474                         break;
1475                 default:
1476                         Con_Printf ("FS_SysOpen(%s, %s): invalid mode\n", filepath, mode);
1477                         return NULL;
1478         }
1479         for (ind = 1; mode[ind] != '\0'; ind++)
1480         {
1481                 switch (mode[ind])
1482                 {
1483                         case '+':
1484                                 mod = O_RDWR;
1485                                 break;
1486                         case 'b':
1487                                 opt |= O_BINARY;
1488                                 break;
1489                         default:
1490                                 Con_Printf ("FS_SysOpen(%s, %s): unknown character in mode (%c)\n",
1491                                                         filepath, mode, mode[ind]);
1492                 }
1493         }
1494
1495         if (nonblocking)
1496                 opt |= O_NONBLOCK;
1497
1498         file = (qfile_t *)Mem_Alloc (fs_mempool, sizeof (*file));
1499         memset (file, 0, sizeof (*file));
1500         file->ungetc = EOF;
1501
1502         file->handle = open (filepath, mod | opt, 0666);
1503         if (file->handle < 0)
1504         {
1505                 Mem_Free (file);
1506                 return NULL;
1507         }
1508
1509         file->real_length = lseek (file->handle, 0, SEEK_END);
1510
1511         // For files opened in append mode, we start at the end of the file
1512         if (mod & O_APPEND)
1513                 file->position = file->real_length;
1514         else
1515                 lseek (file->handle, 0, SEEK_SET);
1516
1517         return file;
1518 }
1519
1520
1521 /*
1522 ===========
1523 FS_OpenPackedFile
1524
1525 Open a packed file using its package file descriptor
1526 ===========
1527 */
1528 qfile_t *FS_OpenPackedFile (pack_t* pack, int pack_ind)
1529 {
1530         packfile_t *pfile;
1531         int dup_handle;
1532         qfile_t* file;
1533
1534         pfile = &pack->files[pack_ind];
1535
1536         // If we don't have the true offset, get it now
1537         if (! (pfile->flags & PACKFILE_FLAG_TRUEOFFS))
1538                 if (!PK3_GetTrueFileOffset (pfile, pack))
1539                         return NULL;
1540
1541         // No Zlib DLL = no compressed files
1542         if (!zlib_dll && (pfile->flags & PACKFILE_FLAG_DEFLATED))
1543         {
1544                 Con_Printf("WARNING: can't open the compressed file %s\n"
1545                                         "You need the Zlib DLL to use compressed files\n",
1546                                         pfile->name);
1547                 return NULL;
1548         }
1549
1550         // LordHavoc: lseek affects all duplicates of a handle so we do it before
1551         // the dup() call to avoid having to close the dup_handle on error here
1552         if (lseek (pack->handle, pfile->offset, SEEK_SET) == -1)
1553         {
1554                 Con_Printf ("FS_OpenPackedFile: can't lseek to %s in %s (offset: %d)\n",
1555                                         pfile->name, pack->filename, (int) pfile->offset);
1556                 return NULL;
1557         }
1558
1559         dup_handle = dup (pack->handle);
1560         if (dup_handle < 0)
1561         {
1562                 Con_Printf ("FS_OpenPackedFile: can't dup package's handle (pack: %s)\n", pack->filename);
1563                 return NULL;
1564         }
1565
1566         file = (qfile_t *)Mem_Alloc (fs_mempool, sizeof (*file));
1567         memset (file, 0, sizeof (*file));
1568         file->handle = dup_handle;
1569         file->flags = QFILE_FLAG_PACKED;
1570         file->real_length = pfile->realsize;
1571         file->offset = pfile->offset;
1572         file->position = 0;
1573         file->ungetc = EOF;
1574
1575         if (pfile->flags & PACKFILE_FLAG_DEFLATED)
1576         {
1577                 ztoolkit_t *ztk;
1578
1579                 file->flags |= QFILE_FLAG_DEFLATED;
1580
1581                 // We need some more variables
1582                 ztk = (ztoolkit_t *)Mem_Alloc (fs_mempool, sizeof (*ztk));
1583
1584                 ztk->comp_length = pfile->packsize;
1585
1586                 // Initialize zlib stream
1587                 ztk->zstream.next_in = ztk->input;
1588                 ztk->zstream.avail_in = 0;
1589
1590                 /* From Zlib's "unzip.c":
1591                  *
1592                  * windowBits is passed < 0 to tell that there is no zlib header.
1593                  * Note that in this case inflate *requires* an extra "dummy" byte
1594                  * after the compressed stream in order to complete decompression and
1595                  * return Z_STREAM_END.
1596                  * In unzip, i don't wait absolutely Z_STREAM_END because I known the
1597                  * size of both compressed and uncompressed data
1598                  */
1599                 if (qz_inflateInit2 (&ztk->zstream, -MAX_WBITS) != Z_OK)
1600                 {
1601                         Con_Printf ("FS_OpenPackedFile: inflate init error (file: %s)\n", pfile->name);
1602                         close(dup_handle);
1603                         Mem_Free(file);
1604                         return NULL;
1605                 }
1606
1607                 ztk->zstream.next_out = file->buff;
1608                 ztk->zstream.avail_out = sizeof (file->buff);
1609
1610                 file->ztk = ztk;
1611         }
1612
1613         return file;
1614 }
1615
1616 /*
1617 ====================
1618 FS_CheckNastyPath
1619
1620 Return true if the path should be rejected due to one of the following:
1621 1: path elements that are non-portable
1622 2: path elements that would allow access to files outside the game directory,
1623    or are just not a good idea for a mod to be using.
1624 ====================
1625 */
1626 int FS_CheckNastyPath (const char *path, qboolean isgamedir)
1627 {
1628         // all: never allow an empty path, as for gamedir it would access the parent directory and a non-gamedir path it is just useless
1629         if (!path[0])
1630                 return 2;
1631
1632         // Windows: don't allow \ in filenames (windows-only), period.
1633         // (on Windows \ is a directory separator, but / is also supported)
1634         if (strstr(path, "\\"))
1635                 return 1; // non-portable
1636
1637         // Mac: don't allow Mac-only filenames - : is a directory separator
1638         // instead of /, but we rely on / working already, so there's no reason to
1639         // support a Mac-only path
1640         // Amiga and Windows: : tries to go to root of drive
1641         if (strstr(path, ":"))
1642                 return 1; // non-portable attempt to go to root of drive
1643
1644         // Amiga: // is parent directory
1645         if (strstr(path, "//"))
1646                 return 1; // non-portable attempt to go to parent directory
1647
1648         // all: don't allow going to parent directory (../ or /../)
1649         if (strstr(path, ".."))
1650                 return 2; // attempt to go outside the game directory
1651
1652         // Windows and UNIXes: don't allow absolute paths
1653         if (path[0] == '/')
1654                 return 2; // attempt to go outside the game directory
1655
1656         // 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
1657         if (strchr(path, '.'))
1658         {
1659                 if (isgamedir)
1660                 {
1661                         // gamedir is entirely path elements, so simply forbid . entirely
1662                         return 2;
1663                 }
1664                 if (strchr(path, '.') < strrchr(path, '/'))
1665                         return 2; // possible attempt to go outside the game directory
1666         }
1667
1668         // all: forbid trailing slash on gamedir
1669         if (isgamedir && path[strlen(path)-1] == '/')
1670                 return 2;
1671
1672         // all: forbid leading dot on any filename for any reason
1673         if (strstr(path, "/."))
1674                 return 2; // attempt to go outside the game directory
1675
1676         // after all these checks we're pretty sure it's a / separated filename
1677         // and won't do much if any harm
1678         return false;
1679 }
1680
1681
1682 /*
1683 ====================
1684 FS_FindFile
1685
1686 Look for a file in the packages and in the filesystem
1687
1688 Return the searchpath where the file was found (or NULL)
1689 and the file index in the package if relevant
1690 ====================
1691 */
1692 static searchpath_t *FS_FindFile (const char *name, int* index, qboolean quiet)
1693 {
1694         searchpath_t *search;
1695         pack_t *pak;
1696
1697         // search through the path, one element at a time
1698         for (search = fs_searchpaths;search;search = search->next)
1699         {
1700                 // is the element a pak file?
1701                 if (search->pack)
1702                 {
1703                         int (*strcmp_funct) (const char* str1, const char* str2);
1704                         int left, right, middle;
1705
1706                         pak = search->pack;
1707                         strcmp_funct = pak->ignorecase ? strcasecmp : strcmp;
1708
1709                         // Look for the file (binary search)
1710                         left = 0;
1711                         right = pak->numfiles - 1;
1712                         while (left <= right)
1713                         {
1714                                 int diff;
1715
1716                                 middle = (left + right) / 2;
1717                                 diff = strcmp_funct (pak->files[middle].name, name);
1718
1719                                 // Found it
1720                                 if (!diff)
1721                                 {
1722                                         if (fs_empty_files_in_pack_mark_deletions.integer && pak->files[middle].realsize == 0)
1723                                         {
1724                                                 // yes, but the first one is empty so we treat it as not being there
1725                                                 if (!quiet && developer.integer >= 10)
1726                                                         Con_Printf("FS_FindFile: %s is marked as deleted\n", name);
1727
1728                                                 if (index != NULL)
1729                                                         *index = -1;
1730                                                 return NULL;
1731                                         }
1732
1733                                         if (!quiet && developer.integer >= 10)
1734                                                 Con_Printf("FS_FindFile: %s in %s\n",
1735                                                                         pak->files[middle].name, pak->filename);
1736
1737                                         if (index != NULL)
1738                                                 *index = middle;
1739                                         return search;
1740                                 }
1741
1742                                 // If we're too far in the list
1743                                 if (diff > 0)
1744                                         right = middle - 1;
1745                                 else
1746                                         left = middle + 1;
1747                         }
1748                 }
1749                 else
1750                 {
1751                         char netpath[MAX_OSPATH];
1752                         dpsnprintf(netpath, sizeof(netpath), "%s%s", search->filename, name);
1753                         if (FS_SysFileExists (netpath))
1754                         {
1755                                 if (!quiet && developer.integer >= 10)
1756                                         Con_Printf("FS_FindFile: %s\n", netpath);
1757
1758                                 if (index != NULL)
1759                                         *index = -1;
1760                                 return search;
1761                         }
1762                 }
1763         }
1764
1765         if (!quiet && developer.integer >= 10)
1766                 Con_Printf("FS_FindFile: can't find %s\n", name);
1767
1768         if (index != NULL)
1769                 *index = -1;
1770         return NULL;
1771 }
1772
1773
1774 /*
1775 ===========
1776 FS_OpenReadFile
1777
1778 Look for a file in the search paths and open it in read-only mode
1779 ===========
1780 */
1781 qfile_t *FS_OpenReadFile (const char *filename, qboolean quiet, qboolean nonblocking)
1782 {
1783         searchpath_t *search;
1784         int pack_ind;
1785
1786         search = FS_FindFile (filename, &pack_ind, quiet);
1787
1788         // Not found?
1789         if (search == NULL)
1790                 return NULL;
1791
1792         // Found in the filesystem?
1793         if (pack_ind < 0)
1794         {
1795                 char path [MAX_OSPATH];
1796                 dpsnprintf (path, sizeof (path), "%s%s", search->filename, filename);
1797                 return FS_SysOpen (path, "rb", nonblocking);
1798         }
1799
1800         // So, we found it in a package...
1801         return FS_OpenPackedFile (search->pack, pack_ind);
1802 }
1803
1804
1805 /*
1806 =============================================================================
1807
1808 MAIN PUBLIC FUNCTIONS
1809
1810 =============================================================================
1811 */
1812
1813 /*
1814 ====================
1815 FS_Open
1816
1817 Open a file. The syntax is the same as fopen
1818 ====================
1819 */
1820 qfile_t* FS_Open (const char* filepath, const char* mode, qboolean quiet, qboolean nonblocking)
1821 {
1822 #ifdef FS_FIX_PATHS
1823         char fixedFileName[MAX_QPATH];
1824         char *d = fixedFileName;
1825         // try to fix common mistakes (\ instead of /)
1826         for( ; *filepath ; filepath++, d++ )
1827                 if( *filepath != '\\' )
1828                         *d = *filepath;
1829                 else
1830                         *d = '/';
1831         *d = '\0';
1832         filepath = fixedFileName;
1833 #endif
1834
1835         if (FS_CheckNastyPath(filepath, false))
1836         {
1837                 Con_Printf("FS_Open(\"%s\", \"%s\", %s): nasty filename rejected\n", filepath, mode, quiet ? "true" : "false");
1838                 return NULL;
1839         }
1840
1841         // If the file is opened in "write", "append", or "read/write" mode
1842         if (mode[0] == 'w' || mode[0] == 'a' || strchr (mode, '+'))
1843         {
1844                 char real_path [MAX_OSPATH];
1845
1846                 // Open the file on disk directly
1847                 dpsnprintf (real_path, sizeof (real_path), "%s/%s", fs_gamedir, filepath);
1848
1849                 // Create directories up to the file
1850                 FS_CreatePath (real_path);
1851
1852                 return FS_SysOpen (real_path, mode, nonblocking);
1853         }
1854         // Else, we look at the various search paths and open the file in read-only mode
1855         else
1856                 return FS_OpenReadFile (filepath, quiet, nonblocking);
1857 }
1858
1859
1860 /*
1861 ====================
1862 FS_Close
1863
1864 Close a file
1865 ====================
1866 */
1867 int FS_Close (qfile_t* file)
1868 {
1869         if (close (file->handle))
1870                 return EOF;
1871
1872         if (file->ztk)
1873         {
1874                 qz_inflateEnd (&file->ztk->zstream);
1875                 Mem_Free (file->ztk);
1876         }
1877
1878         Mem_Free (file);
1879         return 0;
1880 }
1881
1882
1883 /*
1884 ====================
1885 FS_Write
1886
1887 Write "datasize" bytes into a file
1888 ====================
1889 */
1890 fs_offset_t FS_Write (qfile_t* file, const void* data, size_t datasize)
1891 {
1892         fs_offset_t result;
1893
1894         // If necessary, seek to the exact file position we're supposed to be
1895         if (file->buff_ind != file->buff_len)
1896                 lseek (file->handle, file->buff_ind - file->buff_len, SEEK_CUR);
1897
1898         // Purge cached data
1899         FS_Purge (file);
1900
1901         // Write the buffer and update the position
1902         result = write (file->handle, data, (fs_offset_t)datasize);
1903         file->position = lseek (file->handle, 0, SEEK_CUR);
1904         if (file->real_length < file->position)
1905                 file->real_length = file->position;
1906
1907         if (result < 0)
1908                 return 0;
1909
1910         return result;
1911 }
1912
1913
1914 /*
1915 ====================
1916 FS_Read
1917
1918 Read up to "buffersize" bytes from a file
1919 ====================
1920 */
1921 fs_offset_t FS_Read (qfile_t* file, void* buffer, size_t buffersize)
1922 {
1923         fs_offset_t count, done;
1924
1925         if (buffersize == 0)
1926                 return 0;
1927
1928         // Get rid of the ungetc character
1929         if (file->ungetc != EOF)
1930         {
1931                 ((char*)buffer)[0] = file->ungetc;
1932                 buffersize--;
1933                 file->ungetc = EOF;
1934                 done = 1;
1935         }
1936         else
1937                 done = 0;
1938
1939         // First, we copy as many bytes as we can from "buff"
1940         if (file->buff_ind < file->buff_len)
1941         {
1942                 count = file->buff_len - file->buff_ind;
1943                 count = ((fs_offset_t)buffersize > count) ? count : (fs_offset_t)buffersize;
1944                 done += count;
1945                 memcpy (buffer, &file->buff[file->buff_ind], count);
1946                 file->buff_ind += count;
1947
1948                 buffersize -= count;
1949                 if (buffersize == 0)
1950                         return done;
1951         }
1952
1953         // NOTE: at this point, the read buffer is always empty
1954
1955         // If the file isn't compressed
1956         if (! (file->flags & QFILE_FLAG_DEFLATED))
1957         {
1958                 fs_offset_t nb;
1959
1960                 // We must take care to not read after the end of the file
1961                 count = file->real_length - file->position;
1962
1963                 // If we have a lot of data to get, put them directly into "buffer"
1964                 if (buffersize > sizeof (file->buff) / 2)
1965                 {
1966                         if (count > (fs_offset_t)buffersize)
1967                                 count = (fs_offset_t)buffersize;
1968                         lseek (file->handle, file->offset + file->position, SEEK_SET);
1969                         nb = read (file->handle, &((unsigned char*)buffer)[done], count);
1970                         if (nb > 0)
1971                         {
1972                                 done += nb;
1973                                 file->position += nb;
1974
1975                                 // Purge cached data
1976                                 FS_Purge (file);
1977                         }
1978                 }
1979                 else
1980                 {
1981                         if (count > (fs_offset_t)sizeof (file->buff))
1982                                 count = (fs_offset_t)sizeof (file->buff);
1983                         lseek (file->handle, file->offset + file->position, SEEK_SET);
1984                         nb = read (file->handle, file->buff, count);
1985                         if (nb > 0)
1986                         {
1987                                 file->buff_len = nb;
1988                                 file->position += nb;
1989
1990                                 // Copy the requested data in "buffer" (as much as we can)
1991                                 count = (fs_offset_t)buffersize > file->buff_len ? file->buff_len : (fs_offset_t)buffersize;
1992                                 memcpy (&((unsigned char*)buffer)[done], file->buff, count);
1993                                 file->buff_ind = count;
1994                                 done += count;
1995                         }
1996                 }
1997
1998                 return done;
1999         }
2000
2001         // If the file is compressed, it's more complicated...
2002         // We cycle through a few operations until we have read enough data
2003         while (buffersize > 0)
2004         {
2005                 ztoolkit_t *ztk = file->ztk;
2006                 int error;
2007
2008                 // NOTE: at this point, the read buffer is always empty
2009
2010                 // If "input" is also empty, we need to refill it
2011                 if (ztk->in_ind == ztk->in_len)
2012                 {
2013                         // If we are at the end of the file
2014                         if (file->position == file->real_length)
2015                                 return done;
2016
2017                         count = (fs_offset_t)(ztk->comp_length - ztk->in_position);
2018                         if (count > (fs_offset_t)sizeof (ztk->input))
2019                                 count = (fs_offset_t)sizeof (ztk->input);
2020                         lseek (file->handle, file->offset + (fs_offset_t)ztk->in_position, SEEK_SET);
2021                         if (read (file->handle, ztk->input, count) != count)
2022                         {
2023                                 Con_Printf ("FS_Read: unexpected end of file\n");
2024                                 break;
2025                         }
2026
2027                         ztk->in_ind = 0;
2028                         ztk->in_len = count;
2029                         ztk->in_position += count;
2030                 }
2031
2032                 ztk->zstream.next_in = &ztk->input[ztk->in_ind];
2033                 ztk->zstream.avail_in = (unsigned int)(ztk->in_len - ztk->in_ind);
2034
2035                 // Now that we are sure we have compressed data available, we need to determine
2036                 // if it's better to inflate it in "file->buff" or directly in "buffer"
2037
2038                 // Inflate the data in "file->buff"
2039                 if (buffersize < sizeof (file->buff) / 2)
2040                 {
2041                         ztk->zstream.next_out = file->buff;
2042                         ztk->zstream.avail_out = sizeof (file->buff);
2043                         error = qz_inflate (&ztk->zstream, Z_SYNC_FLUSH);
2044                         if (error != Z_OK && error != Z_STREAM_END)
2045                         {
2046                                 Con_Printf ("FS_Read: Can't inflate file\n");
2047                                 break;
2048                         }
2049                         ztk->in_ind = ztk->in_len - ztk->zstream.avail_in;
2050
2051                         file->buff_len = (fs_offset_t)sizeof (file->buff) - ztk->zstream.avail_out;
2052                         file->position += file->buff_len;
2053
2054                         // Copy the requested data in "buffer" (as much as we can)
2055                         count = (fs_offset_t)buffersize > file->buff_len ? file->buff_len : (fs_offset_t)buffersize;
2056                         memcpy (&((unsigned char*)buffer)[done], file->buff, count);
2057                         file->buff_ind = count;
2058                 }
2059
2060                 // Else, we inflate directly in "buffer"
2061                 else
2062                 {
2063                         ztk->zstream.next_out = &((unsigned char*)buffer)[done];
2064                         ztk->zstream.avail_out = (unsigned int)buffersize;
2065                         error = qz_inflate (&ztk->zstream, Z_SYNC_FLUSH);
2066                         if (error != Z_OK && error != Z_STREAM_END)
2067                         {
2068                                 Con_Printf ("FS_Read: Can't inflate file\n");
2069                                 break;
2070                         }
2071                         ztk->in_ind = ztk->in_len - ztk->zstream.avail_in;
2072
2073                         // How much data did it inflate?
2074                         count = (fs_offset_t)(buffersize - ztk->zstream.avail_out);
2075                         file->position += count;
2076
2077                         // Purge cached data
2078                         FS_Purge (file);
2079                 }
2080
2081                 done += count;
2082                 buffersize -= count;
2083         }
2084
2085         return done;
2086 }
2087
2088
2089 /*
2090 ====================
2091 FS_Print
2092
2093 Print a string into a file
2094 ====================
2095 */
2096 int FS_Print (qfile_t* file, const char *msg)
2097 {
2098         return (int)FS_Write (file, msg, strlen (msg));
2099 }
2100
2101 /*
2102 ====================
2103 FS_Printf
2104
2105 Print a string into a file
2106 ====================
2107 */
2108 int FS_Printf(qfile_t* file, const char* format, ...)
2109 {
2110         int result;
2111         va_list args;
2112
2113         va_start (args, format);
2114         result = FS_VPrintf (file, format, args);
2115         va_end (args);
2116
2117         return result;
2118 }
2119
2120
2121 /*
2122 ====================
2123 FS_VPrintf
2124
2125 Print a string into a file
2126 ====================
2127 */
2128 int FS_VPrintf (qfile_t* file, const char* format, va_list ap)
2129 {
2130         int len;
2131         fs_offset_t buff_size = MAX_INPUTLINE;
2132         char *tempbuff;
2133
2134         for (;;)
2135         {
2136                 tempbuff = (char *)Mem_Alloc (tempmempool, buff_size);
2137                 len = dpvsnprintf (tempbuff, buff_size, format, ap);
2138                 if (len >= 0 && len < buff_size)
2139                         break;
2140                 Mem_Free (tempbuff);
2141                 buff_size *= 2;
2142         }
2143
2144         len = write (file->handle, tempbuff, len);
2145         Mem_Free (tempbuff);
2146
2147         return len;
2148 }
2149
2150
2151 /*
2152 ====================
2153 FS_Getc
2154
2155 Get the next character of a file
2156 ====================
2157 */
2158 int FS_Getc (qfile_t* file)
2159 {
2160         unsigned char c;
2161
2162         if (FS_Read (file, &c, 1) != 1)
2163                 return EOF;
2164
2165         return c;
2166 }
2167
2168
2169 /*
2170 ====================
2171 FS_UnGetc
2172
2173 Put a character back into the read buffer (only supports one character!)
2174 ====================
2175 */
2176 int FS_UnGetc (qfile_t* file, unsigned char c)
2177 {
2178         // If there's already a character waiting to be read
2179         if (file->ungetc != EOF)
2180                 return EOF;
2181
2182         file->ungetc = c;
2183         return c;
2184 }
2185
2186
2187 /*
2188 ====================
2189 FS_Seek
2190
2191 Move the position index in a file
2192 ====================
2193 */
2194 int FS_Seek (qfile_t* file, fs_offset_t offset, int whence)
2195 {
2196         ztoolkit_t *ztk;
2197         unsigned char* buffer;
2198         fs_offset_t buffersize;
2199
2200         // Compute the file offset
2201         switch (whence)
2202         {
2203                 case SEEK_CUR:
2204                         offset += file->position - file->buff_len + file->buff_ind;
2205                         break;
2206
2207                 case SEEK_SET:
2208                         break;
2209
2210                 case SEEK_END:
2211                         offset += file->real_length;
2212                         break;
2213
2214                 default:
2215                         return -1;
2216         }
2217         if (offset < 0 || offset > file->real_length)
2218                 return -1;
2219
2220         // If we have the data in our read buffer, we don't need to actually seek
2221         if (file->position - file->buff_len <= offset && offset <= file->position)
2222         {
2223                 file->buff_ind = offset + file->buff_len - file->position;
2224                 return 0;
2225         }
2226
2227         // Purge cached data
2228         FS_Purge (file);
2229
2230         // Unpacked or uncompressed files can seek directly
2231         if (! (file->flags & QFILE_FLAG_DEFLATED))
2232         {
2233                 if (lseek (file->handle, file->offset + offset, SEEK_SET) == -1)
2234                         return -1;
2235                 file->position = offset;
2236                 return 0;
2237         }
2238
2239         // Seeking in compressed files is more a hack than anything else,
2240         // but we need to support it, so here we go.
2241         ztk = file->ztk;
2242
2243         // If we have to go back in the file, we need to restart from the beginning
2244         if (offset <= file->position)
2245         {
2246                 ztk->in_ind = 0;
2247                 ztk->in_len = 0;
2248                 ztk->in_position = 0;
2249                 file->position = 0;
2250                 lseek (file->handle, file->offset, SEEK_SET);
2251
2252                 // Reset the Zlib stream
2253                 ztk->zstream.next_in = ztk->input;
2254                 ztk->zstream.avail_in = 0;
2255                 qz_inflateReset (&ztk->zstream);
2256         }
2257
2258         // We need a big buffer to force inflating into it directly
2259         buffersize = 2 * sizeof (file->buff);
2260         buffer = (unsigned char *)Mem_Alloc (tempmempool, buffersize);
2261
2262         // Skip all data until we reach the requested offset
2263         while (offset > file->position)
2264         {
2265                 fs_offset_t diff = offset - file->position;
2266                 fs_offset_t count, len;
2267
2268                 count = (diff > buffersize) ? buffersize : diff;
2269                 len = FS_Read (file, buffer, count);
2270                 if (len != count)
2271                 {
2272                         Mem_Free (buffer);
2273                         return -1;
2274                 }
2275         }
2276
2277         Mem_Free (buffer);
2278         return 0;
2279 }
2280
2281
2282 /*
2283 ====================
2284 FS_Tell
2285
2286 Give the current position in a file
2287 ====================
2288 */
2289 fs_offset_t FS_Tell (qfile_t* file)
2290 {
2291         return file->position - file->buff_len + file->buff_ind;
2292 }
2293
2294
2295 /*
2296 ====================
2297 FS_FileSize
2298
2299 Give the total size of a file
2300 ====================
2301 */
2302 fs_offset_t FS_FileSize (qfile_t* file)
2303 {
2304         return file->real_length;
2305 }
2306
2307
2308 /*
2309 ====================
2310 FS_Purge
2311
2312 Erases any buffered input or output data
2313 ====================
2314 */
2315 void FS_Purge (qfile_t* file)
2316 {
2317         file->buff_len = 0;
2318         file->buff_ind = 0;
2319         file->ungetc = EOF;
2320 }
2321
2322
2323 /*
2324 ============
2325 FS_LoadFile
2326
2327 Filename are relative to the quake directory.
2328 Always appends a 0 byte.
2329 ============
2330 */
2331 unsigned char *FS_LoadFile (const char *path, mempool_t *pool, qboolean quiet, fs_offset_t *filesizepointer)
2332 {
2333         qfile_t *file;
2334         unsigned char *buf = NULL;
2335         fs_offset_t filesize = 0;
2336
2337         file = FS_Open (path, "rb", quiet, false);
2338         if (file)
2339         {
2340                 filesize = file->real_length;
2341                 buf = (unsigned char *)Mem_Alloc (pool, filesize + 1);
2342                 buf[filesize] = '\0';
2343                 FS_Read (file, buf, filesize);
2344                 FS_Close (file);
2345         }
2346
2347         if (filesizepointer)
2348                 *filesizepointer = filesize;
2349         return buf;
2350 }
2351
2352
2353 /*
2354 ============
2355 FS_WriteFile
2356
2357 The filename will be prefixed by the current game directory
2358 ============
2359 */
2360 qboolean FS_WriteFile (const char *filename, void *data, fs_offset_t len)
2361 {
2362         qfile_t *file;
2363
2364         file = FS_Open (filename, "wb", false, false);
2365         if (!file)
2366         {
2367                 Con_Printf("FS_WriteFile: failed on %s\n", filename);
2368                 return false;
2369         }
2370
2371         Con_DPrintf("FS_WriteFile: %s\n", filename);
2372         FS_Write (file, data, len);
2373         FS_Close (file);
2374         return true;
2375 }
2376
2377
2378 /*
2379 =============================================================================
2380
2381 OTHERS PUBLIC FUNCTIONS
2382
2383 =============================================================================
2384 */
2385
2386 /*
2387 ============
2388 FS_StripExtension
2389 ============
2390 */
2391 void FS_StripExtension (const char *in, char *out, size_t size_out)
2392 {
2393         char *last = NULL;
2394         char currentchar;
2395
2396         if (size_out == 0)
2397                 return;
2398
2399         while ((currentchar = *in) && size_out > 1)
2400         {
2401                 if (currentchar == '.')
2402                         last = out;
2403                 else if (currentchar == '/' || currentchar == '\\' || currentchar == ':')
2404                         last = NULL;
2405                 *out++ = currentchar;
2406                 in++;
2407                 size_out--;
2408         }
2409         if (last)
2410                 *last = 0;
2411         else
2412                 *out = 0;
2413 }
2414
2415
2416 /*
2417 ==================
2418 FS_DefaultExtension
2419 ==================
2420 */
2421 void FS_DefaultExtension (char *path, const char *extension, size_t size_path)
2422 {
2423         const char *src;
2424
2425         // if path doesn't have a .EXT, append extension
2426         // (extension should include the .)
2427         src = path + strlen(path) - 1;
2428
2429         while (*src != '/' && src != path)
2430         {
2431                 if (*src == '.')
2432                         return;                 // it has an extension
2433                 src--;
2434         }
2435
2436         strlcat (path, extension, size_path);
2437 }
2438
2439
2440 /*
2441 ==================
2442 FS_FileType
2443
2444 Look for a file in the packages and in the filesystem
2445 ==================
2446 */
2447 int FS_FileType (const char *filename)
2448 {
2449         searchpath_t *search;
2450         char fullpath[MAX_QPATH];
2451
2452         search = FS_FindFile (filename, NULL, true);
2453         if(!search)
2454                 return FS_FILETYPE_NONE;
2455
2456         if(search->pack)
2457                 return FS_FILETYPE_FILE; // TODO can't check directories in paks yet, maybe later
2458
2459         dpsnprintf(fullpath, sizeof(fullpath), "%s%s", search->filename, filename);
2460         return FS_SysFileType(fullpath);
2461 }
2462
2463
2464 /*
2465 ==================
2466 FS_FileExists
2467
2468 Look for a file in the packages and in the filesystem
2469 ==================
2470 */
2471 qboolean FS_FileExists (const char *filename)
2472 {
2473         return (FS_FindFile (filename, NULL, true) != NULL);
2474 }
2475
2476
2477 /*
2478 ==================
2479 FS_SysFileExists
2480
2481 Look for a file in the filesystem only
2482 ==================
2483 */
2484 int FS_SysFileType (const char *path)
2485 {
2486 #if WIN32
2487         DWORD result = GetFileAttributes(path);
2488
2489         if(result == INVALID_FILE_ATTRIBUTES)
2490                 return FS_FILETYPE_NONE;
2491
2492         if(result & FILE_ATTRIBUTE_DIRECTORY)
2493                 return FS_FILETYPE_DIRECTORY;
2494
2495         return FS_FILETYPE_FILE;
2496 #else
2497         struct stat buf;
2498
2499         if (stat (path,&buf) == -1)
2500                 return FS_FILETYPE_NONE;
2501
2502         if(S_ISDIR(buf.st_mode))
2503                 return FS_FILETYPE_DIRECTORY;
2504
2505         return FS_FILETYPE_FILE;
2506 #endif
2507 }
2508
2509 qboolean FS_SysFileExists (const char *path)
2510 {
2511         return FS_SysFileType (path) != FS_FILETYPE_NONE;
2512 }
2513
2514 void FS_mkdir (const char *path)
2515 {
2516 #if WIN32
2517         _mkdir (path);
2518 #else
2519         mkdir (path, 0777);
2520 #endif
2521 }
2522
2523 /*
2524 ===========
2525 FS_Search
2526
2527 Allocate and fill a search structure with information on matching filenames.
2528 ===========
2529 */
2530 fssearch_t *FS_Search(const char *pattern, int caseinsensitive, int quiet)
2531 {
2532         fssearch_t *search;
2533         searchpath_t *searchpath;
2534         pack_t *pak;
2535         int i, basepathlength, numfiles, numchars, resultlistindex, dirlistindex;
2536         stringlist_t resultlist;
2537         stringlist_t dirlist;
2538         const char *slash, *backslash, *colon, *separator;
2539         char *basepath;
2540         char netpath[MAX_OSPATH];
2541         char temp[MAX_OSPATH];
2542
2543         for (i = 0;pattern[i] == '.' || pattern[i] == ':' || pattern[i] == '/' || pattern[i] == '\\';i++)
2544                 ;
2545
2546         if (i > 0)
2547         {
2548                 Con_Printf("Don't use punctuation at the beginning of a search pattern!\n");
2549                 return NULL;
2550         }
2551
2552         stringlistinit(&resultlist);
2553         stringlistinit(&dirlist);
2554         search = NULL;
2555         slash = strrchr(pattern, '/');
2556         backslash = strrchr(pattern, '\\');
2557         colon = strrchr(pattern, ':');
2558         separator = max(slash, backslash);
2559         separator = max(separator, colon);
2560         basepathlength = separator ? (separator + 1 - pattern) : 0;
2561         basepath = (char *)Mem_Alloc (tempmempool, basepathlength + 1);
2562         if (basepathlength)
2563                 memcpy(basepath, pattern, basepathlength);
2564         basepath[basepathlength] = 0;
2565
2566         // search through the path, one element at a time
2567         for (searchpath = fs_searchpaths;searchpath;searchpath = searchpath->next)
2568         {
2569                 // is the element a pak file?
2570                 if (searchpath->pack)
2571                 {
2572                         // look through all the pak file elements
2573                         pak = searchpath->pack;
2574                         for (i = 0;i < pak->numfiles;i++)
2575                         {
2576                                 strlcpy(temp, pak->files[i].name, sizeof(temp));
2577                                 while (temp[0])
2578                                 {
2579                                         if (matchpattern(temp, (char *)pattern, true))
2580                                         {
2581                                                 for (resultlistindex = 0;resultlistindex < resultlist.numstrings;resultlistindex++)
2582                                                         if (!strcmp(resultlist.strings[resultlistindex], temp))
2583                                                                 break;
2584                                                 if (resultlistindex == resultlist.numstrings)
2585                                                 {
2586                                                         stringlistappend(&resultlist, temp);
2587                                                         if (!quiet)
2588                                                                 Con_DPrintf("SearchPackFile: %s : %s\n", pak->filename, temp);
2589                                                 }
2590                                         }
2591                                         // strip off one path element at a time until empty
2592                                         // this way directories are added to the listing if they match the pattern
2593                                         slash = strrchr(temp, '/');
2594                                         backslash = strrchr(temp, '\\');
2595                                         colon = strrchr(temp, ':');
2596                                         separator = temp;
2597                                         if (separator < slash)
2598                                                 separator = slash;
2599                                         if (separator < backslash)
2600                                                 separator = backslash;
2601                                         if (separator < colon)
2602                                                 separator = colon;
2603                                         *((char *)separator) = 0;
2604                                 }
2605                         }
2606                 }
2607                 else
2608                 {
2609                         // get a directory listing and look at each name
2610                         dpsnprintf(netpath, sizeof (netpath), "%s%s", searchpath->filename, basepath);
2611                         stringlistinit(&dirlist);
2612                         listdirectory(&dirlist, netpath);
2613                         for (dirlistindex = 0;dirlistindex < dirlist.numstrings;dirlistindex++)
2614                         {
2615                                 dpsnprintf(temp, sizeof(temp), "%s%s", basepath, dirlist.strings[dirlistindex]);
2616                                 if (matchpattern(temp, (char *)pattern, true))
2617                                 {
2618                                         for (resultlistindex = 0;resultlistindex < resultlist.numstrings;resultlistindex++)
2619                                                 if (!strcmp(resultlist.strings[resultlistindex], temp))
2620                                                         break;
2621                                         if (resultlistindex == resultlist.numstrings)
2622                                         {
2623                                                 stringlistappend(&resultlist, temp);
2624                                                 if (!quiet)
2625                                                         Con_DPrintf("SearchDirFile: %s\n", temp);
2626                                         }
2627                                 }
2628                         }
2629                         stringlistfreecontents(&dirlist);
2630                 }
2631         }
2632
2633         if (resultlist.numstrings)
2634         {
2635                 stringlistsort(&resultlist);
2636                 numfiles = resultlist.numstrings;
2637                 numchars = 0;
2638                 for (resultlistindex = 0;resultlistindex < resultlist.numstrings;resultlistindex++)
2639                         numchars += (int)strlen(resultlist.strings[resultlistindex]) + 1;
2640                 search = (fssearch_t *)Z_Malloc(sizeof(fssearch_t) + numchars + numfiles * sizeof(char *));
2641                 search->filenames = (char **)((char *)search + sizeof(fssearch_t));
2642                 search->filenamesbuffer = (char *)((char *)search + sizeof(fssearch_t) + numfiles * sizeof(char *));
2643                 search->numfilenames = (int)numfiles;
2644                 numfiles = 0;
2645                 numchars = 0;
2646                 for (resultlistindex = 0;resultlistindex < resultlist.numstrings;resultlistindex++)
2647                 {
2648                         size_t textlen;
2649                         search->filenames[numfiles] = search->filenamesbuffer + numchars;
2650                         textlen = strlen(resultlist.strings[resultlistindex]) + 1;
2651                         memcpy(search->filenames[numfiles], resultlist.strings[resultlistindex], textlen);
2652                         numfiles++;
2653                         numchars += (int)textlen;
2654                 }
2655         }
2656         stringlistfreecontents(&resultlist);
2657
2658         Mem_Free(basepath);
2659         return search;
2660 }
2661
2662 void FS_FreeSearch(fssearch_t *search)
2663 {
2664         Z_Free(search);
2665 }
2666
2667 extern int con_linewidth;
2668 int FS_ListDirectory(const char *pattern, int oneperline)
2669 {
2670         int numfiles;
2671         int numcolumns;
2672         int numlines;
2673         int columnwidth;
2674         int linebufpos;
2675         int i, j, k, l;
2676         const char *name;
2677         char linebuf[MAX_INPUTLINE];
2678         fssearch_t *search;
2679         search = FS_Search(pattern, true, true);
2680         if (!search)
2681                 return 0;
2682         numfiles = search->numfilenames;
2683         if (!oneperline)
2684         {
2685                 // FIXME: the names could be added to one column list and then
2686                 // gradually shifted into the next column if they fit, and then the
2687                 // next to make a compact variable width listing but it's a lot more
2688                 // complicated...
2689                 // find width for columns
2690                 columnwidth = 0;
2691                 for (i = 0;i < numfiles;i++)
2692                 {
2693                         l = (int)strlen(search->filenames[i]);
2694                         if (columnwidth < l)
2695                                 columnwidth = l;
2696                 }
2697                 // count the spacing character
2698                 columnwidth++;
2699                 // calculate number of columns
2700                 numcolumns = con_linewidth / columnwidth;
2701                 // don't bother with the column printing if it's only one column
2702                 if (numcolumns >= 2)
2703                 {
2704                         numlines = (numfiles + numcolumns - 1) / numcolumns;
2705                         for (i = 0;i < numlines;i++)
2706                         {
2707                                 linebufpos = 0;
2708                                 for (k = 0;k < numcolumns;k++)
2709                                 {
2710                                         l = i * numcolumns + k;
2711                                         if (l < numfiles)
2712                                         {
2713                                                 name = search->filenames[l];
2714                                                 for (j = 0;name[j] && linebufpos + 1 < (int)sizeof(linebuf);j++)
2715                                                         linebuf[linebufpos++] = name[j];
2716                                                 // space out name unless it's the last on the line
2717                                                 if (k + 1 < numcolumns && l + 1 < numfiles)
2718                                                         for (;j < columnwidth && linebufpos + 1 < (int)sizeof(linebuf);j++)
2719                                                                 linebuf[linebufpos++] = ' ';
2720                                         }
2721                                 }
2722                                 linebuf[linebufpos] = 0;
2723                                 Con_Printf("%s\n", linebuf);
2724                         }
2725                 }
2726                 else
2727                         oneperline = true;
2728         }
2729         if (oneperline)
2730                 for (i = 0;i < numfiles;i++)
2731                         Con_Printf("%s\n", search->filenames[i]);
2732         FS_FreeSearch(search);
2733         return (int)numfiles;
2734 }
2735
2736 static void FS_ListDirectoryCmd (const char* cmdname, int oneperline)
2737 {
2738         const char *pattern;
2739         if (Cmd_Argc() > 3)
2740         {
2741                 Con_Printf("usage:\n%s [path/pattern]\n", cmdname);
2742                 return;
2743         }
2744         if (Cmd_Argc() == 2)
2745                 pattern = Cmd_Argv(1);
2746         else
2747                 pattern = "*";
2748         if (!FS_ListDirectory(pattern, oneperline))
2749                 Con_Print("No files found.\n");
2750 }
2751
2752 void FS_Dir_f(void)
2753 {
2754         FS_ListDirectoryCmd("dir", true);
2755 }
2756
2757 void FS_Ls_f(void)
2758 {
2759         FS_ListDirectoryCmd("ls", false);
2760 }
2761
2762 const char *FS_WhichPack(const char *filename)
2763 {
2764         int index;
2765         searchpath_t *sp = FS_FindFile(filename, &index, true);
2766         if(sp && sp->pack)
2767                 return sp->pack->filename;
2768         else
2769                 return 0;
2770 }
2771
2772 /*
2773 ====================
2774 FS_IsRegisteredQuakePack
2775
2776 Look for a proof of purchase file file in the requested package
2777
2778 If it is found, this file should NOT be downloaded.
2779 ====================
2780 */
2781 qboolean FS_IsRegisteredQuakePack(const char *name)
2782 {
2783         searchpath_t *search;
2784         pack_t *pak;
2785
2786         // search through the path, one element at a time
2787         for (search = fs_searchpaths;search;search = search->next)
2788         {
2789                 if (search->pack && !strcasecmp(FS_FileWithoutPath(search->filename), name))
2790                 {
2791                         int (*strcmp_funct) (const char* str1, const char* str2);
2792                         int left, right, middle;
2793
2794                         pak = search->pack;
2795                         strcmp_funct = pak->ignorecase ? strcasecmp : strcmp;
2796
2797                         // Look for the file (binary search)
2798                         left = 0;
2799                         right = pak->numfiles - 1;
2800                         while (left <= right)
2801                         {
2802                                 int diff;
2803
2804                                 middle = (left + right) / 2;
2805                                 diff = !strcmp_funct (pak->files[middle].name, "gfx/pop.lmp");
2806
2807                                 // Found it
2808                                 if (!diff)
2809                                         return true;
2810
2811                                 // If we're too far in the list
2812                                 if (diff > 0)
2813                                         right = middle - 1;
2814                                 else
2815                                         left = middle + 1;
2816                         }
2817
2818                         // we found the requested pack but it is not registered quake
2819                         return false;
2820                 }
2821         }
2822
2823         return false;
2824 }
2825
2826 int FS_CRCFile(const char *filename, size_t *filesizepointer)
2827 {
2828         int crc = -1;
2829         unsigned char *filedata;
2830         fs_offset_t filesize;
2831         if (filesizepointer)
2832                 *filesizepointer = 0;
2833         if (!filename || !*filename)
2834                 return crc;
2835         filedata = FS_LoadFile(filename, tempmempool, true, &filesize);
2836         if (filedata)
2837         {
2838                 if (filesizepointer)
2839                         *filesizepointer = filesize;
2840                 crc = CRC_Block(filedata, filesize);
2841                 Mem_Free(filedata);
2842         }
2843         return crc;
2844 }
2845