]> de.git.xonotic.org Git - xonotic/netradiant.git/blobdiff - tools/quake3/common/vfs.c
Fix minizip regression
[xonotic/netradiant.git] / tools / quake3 / common / vfs.c
index 04af830e1ee7d8c4917223a6ea416d0bdd4d2617..0c3ca69f9535406330c0aa9ba3a3b05cabd2826a 100644 (file)
 // Leonardo Zide (leo@lokigames.com)
 //
 
-#include <stdio.h>
-
-#if defined ( __linux__ ) || defined ( __APPLE__ )
-#include <dirent.h>
-#include <unistd.h>
-#else
-#include <wtypes.h>
-#include <io.h>
-#define R_OK 04
-#define S_ISDIR( mode ) ( mode & _S_IFDIR )
-#define PATH_MAX 260
-#endif
-
 #include <string.h>
 #include <stdlib.h>
 #include <sys/stat.h>
 
 #include "cmdlib.h"
+#include "filematch.h"
 #include "mathlib.h"
-#include <glib.h>
 #include "inout.h"
 #include "vfs.h"
-#include "unzip.h"
+#include <minizip/unzip.h>
 
 typedef struct
 {
        char*   name;
-       unz_s zipinfo;
        unzFile zipfile;
+       unz_file_pos zippos;
        guint32 size;
 } VFS_PAKFILE;
 
@@ -78,8 +65,10 @@ typedef struct
 
 static GSList*  g_unzFiles;
 static GSList*  g_pakFiles;
-static char g_strDirs[VFS_MAXDIRS][PATH_MAX];
+static char g_strDirs[VFS_MAXDIRS][PATH_MAX + 1];
 static int g_numDirs;
+char g_strForbiddenDirs[VFS_MAXDIRS][PATH_MAX + 1];
+int g_numForbiddenDirs = 0;
 static gboolean g_bUsePak = TRUE;
 
 // =============================================================================
@@ -140,6 +129,11 @@ static void vfsInitPakFile( const char *filename ){
                if ( err != UNZ_OK ) {
                        break;
                }
+               unz_file_pos pos;
+               err = unzGetFilePos( uf, &pos );
+               if ( err != UNZ_OK ) {
+                       break;
+               }
 
                file = (VFS_PAKFILE*)safe_malloc( sizeof( VFS_PAKFILE ) );
                g_pakFiles = g_slist_append( g_pakFiles, file );
@@ -150,7 +144,7 @@ static void vfsInitPakFile( const char *filename ){
                file->name = strdup( filename_inzip );
                file->size = file_info.uncompressed_size;
                file->zipfile = uf;
-               memcpy( &file->zipinfo, uf, sizeof( unz_s ) );
+               file->zippos = pos;
 
                if ( ( i + 1 ) < gi.number_entry ) {
                        err = unzGoToNextFile( uf );
@@ -169,14 +163,34 @@ void vfsInitDirectory( const char *path ){
        char filename[PATH_MAX];
        char *dirlist;
        GDir *dir;
+       int j;
+
+       for ( j = 0; j < g_numForbiddenDirs; ++j )
+       {
+               char* dbuf = g_strdup( path );
+               if ( *dbuf && dbuf[strlen( dbuf ) - 1] == '/' ) {
+                       dbuf[strlen( dbuf ) - 1] = 0;
+               }
+               const char *p = strrchr( dbuf, '/' );
+               p = ( p ? ( p + 1 ) : dbuf );
+               if ( matchpattern( p, g_strForbiddenDirs[j], TRUE ) ) {
+                       g_free( dbuf );
+                       break;
+               }
+               g_free( dbuf );
+       }
+       if ( j < g_numForbiddenDirs ) {
+               return;
+       }
 
-       if ( g_numDirs == ( VFS_MAXDIRS - 1 ) ) {
+       if ( g_numDirs == VFS_MAXDIRS ) {
                return;
        }
 
        Sys_Printf( "VFS Init: %s\n", path );
 
-       strcpy( g_strDirs[g_numDirs], path );
+       strncpy( g_strDirs[g_numDirs], path, PATH_MAX );
+       g_strDirs[g_numDirs][PATH_MAX] = 0;
        vfsFixDOSName( g_strDirs[g_numDirs] );
        vfsAddSlash( g_strDirs[g_numDirs] );
        g_numDirs++;
@@ -192,11 +206,35 @@ void vfsInitDirectory( const char *path ){
                                        break;
                                }
 
+                               for ( j = 0; j < g_numForbiddenDirs; ++j )
+                               {
+                                       const char *p = strrchr( name, '/' );
+                                       p = ( p ? ( p + 1 ) : name );
+                                       if ( matchpattern( p, g_strForbiddenDirs[j], TRUE ) ) {
+                                               break;
+                                       }
+                               }
+                               if ( j < g_numForbiddenDirs ) {
+                                       continue;
+                               }
+
                                dirlist = g_strdup( name );
 
                                {
                                        char *ext = strrchr( dirlist, '.' );
-                                       if ( ( ext == NULL ) || ( Q_stricmp( ext, ".pk3" ) != 0 ) ) {
+
+                                       if ( ext && ( !Q_stricmp( ext, ".pk3dir" ) || !Q_stricmp( ext, ".dpkdir" ) ) ) {
+                                               if ( g_numDirs == VFS_MAXDIRS ) {
+                                                       continue;
+                                               }
+                                               snprintf( g_strDirs[g_numDirs], PATH_MAX, "%s/%s", path, name );
+                                               g_strDirs[g_numDirs][PATH_MAX] = '\0';
+                                               vfsFixDOSName( g_strDirs[g_numDirs] );
+                                               vfsAddSlash( g_strDirs[g_numDirs] );
+                                               ++g_numDirs;
+                                       }
+
+                                       if ( ( ext == NULL ) || ( Q_stricmp( ext, ".pk3" ) != 0 || !Q_stricmp( ext, ".dpk" ) != 0 ) ) {
                                                continue;
                                        }
                                }
@@ -281,10 +319,14 @@ int vfsLoadFile( const char *filename, void **bufferptr, int index ){
 
                *bufferptr = safe_malloc( len + 1 );
                if ( *bufferptr == NULL ) {
+                       fclose( f );
                        return -1;
                }
 
-               fread( *bufferptr, 1, len, f );
+               if ( fread( *bufferptr, 1, len, f ) != (size_t) len ) {
+                       fclose( f );
+                       return -1;
+               }
                fclose( f );
 
                // we need to end the buffer with a 0
@@ -318,10 +360,14 @@ int vfsLoadFile( const char *filename, void **bufferptr, int index ){
 
                                *bufferptr = safe_malloc( len + 1 );
                                if ( *bufferptr == NULL ) {
+                                       fclose( f );
                                        return -1;
                                }
 
-                               fread( *bufferptr, 1, len, f );
+                               if ( fread( *bufferptr, 1, len, f ) != (size_t) len ) {
+                                       fclose( f );
+                                       return -1;
+                               }
                                fclose( f );
 
                                // we need to end the buffer with a 0
@@ -343,8 +389,10 @@ int vfsLoadFile( const char *filename, void **bufferptr, int index ){
                }
 
                if ( count == index ) {
-                       memcpy( file->zipfile, &file->zipinfo, sizeof( unz_s ) );
 
+            if ( unzGoToFilePos( file->zipfile, &file->zippos ) != UNZ_OK ) {
+                return -1;
+            }
                        if ( unzOpenCurrentFile( file->zipfile ) != UNZ_OK ) {
                                return -1;
                        }