Prevent a segmentation fault when using the -map option without specifying a map.

[quix0rs-blobwars.git] / src / pak.cpp

diff --git a/src/pak.cpp b/src/pak.cpp
index d70fe571864cc15227742e257cd782486f9b5567..16beab2b1ee0c05b047be699b780c0f999524b33 100644 (file)
--- a/src/pak.cpp
+++ b/src/pak.cpp
@@ -119,7 +119,7 @@ void recurseDirectory(const char *dirName)
                        {
                                printf("Couldn't open %s for reading!\n", filename);
                                closedir(dirp);
-                               gzclose(pak);
+                               fclose(pak);
                                exit(1);
                        }
                        
@@ -141,7 +141,7 @@ void recurseDirectory(const char *dirName)
                        {
                                printf("Couldn't open %s for reading!\n", filename);
                                closedir(dirp);
-                               gzclose(pak);
+                               fclose(pak);
                                exit(1);
                        }
                        else
@@ -154,7 +154,12 @@ void recurseDirectory(const char *dirName)
                                
                                fileData[files].set(filename, fSize, cSize, ftell(pak));
 
-                               fwrite(output, 1, cSize, pak);
+                               if (fwrite(output, 1, cSize, pak) != cSize)
+                               {
+                                       fprintf(stderr, "Error writing to pakfile: %s\n", strerror(errno));
+                                       fclose(pak);
+                                       exit(1);
+                               }
 
                                files++;
                                
@@ -183,6 +188,11 @@ int main(int argc, char *argv[])
        }
 
        pak = fopen(argv[argc - 1], "wb");
+       if (!pak)
+       {
+               fprintf(stderr, "Error opening %s: %s\n", argv[argc - 1], strerror(errno));
+               return 1;
+       }
        
        for (int i = 1 ; i < (argc - 1) ; i++)
        {
@@ -211,13 +221,28 @@ int main(int argc, char *argv[])
                        break;
                }
 
-               fwrite(&fileData[i], sizeof(FileData), 1, pak);
+               if (fwrite(&fileData[i], sizeof(FileData), 1, pak) != 1)
+               {
+                       fprintf(stderr, "Error writing to %s: %s\n", argv[argc - 1], strerror(errno));
+                       fclose(pak);
+                       return 1;
+               }
        }
        
        unsigned int numberOfFiles = totalFiles;
 
-       fwrite(&pos, sizeof(unsigned int), 1, pak);
-       fwrite(&numberOfFiles, sizeof(unsigned int), 1, pak);
+       if (fwrite(&pos, sizeof(unsigned int), 1, pak) != 1)
+       {
+               fprintf(stderr, "Error writing to %s: %s\n", argv[argc - 1], strerror(errno));
+               fclose(pak);
+               return 1;
+       }
+       if (fwrite(&numberOfFiles, sizeof(unsigned int), 1, pak) != 1)
+       {
+               fprintf(stderr, "Error writing to %s: %s\n", argv[argc - 1], strerror(errno));
+               fclose(pak);
+               return 1;
+       }
 
        fclose(pak);