contrib/extractor/loadlib/loadlib.cpp

   1 /*
   2  * Copyright (C) 2005-2012 MaNGOS <http://getmangos.com/>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19 #define _CRT_SECURE_NO_DEPRECATE
  20
  21 #include "loadlib.h"
  22
  23 // list of mpq files for lookup most recent file version
  24 ArchiveSet gOpenArchives;
  25
  26 ArchiveSetBounds GetArchivesBounds()
  27 {
  28     return ArchiveSetBounds(gOpenArchives.begin(), gOpenArchives.end());
  29 }
  30
  31 bool OpenArchive(char const* mpqFileName, HANDLE* mpqHandlePtr /*= NULL*/)
  32 {
  33     HANDLE mpqHandle;
  34
  35     if (!SFileOpenArchive(mpqFileName, 0, MPQ_OPEN_READ_ONLY, &mpqHandle))
  36         return false;
  37
  38     gOpenArchives.push_back(mpqHandle);
  39
  40     if (mpqHandlePtr)
  41         *mpqHandlePtr = mpqHandle;
  42
  43     return true;
  44 }
  45
  46 bool OpenNewestFile(char const* filename, HANDLE* fileHandlerPtr)
  47 {
  48     for(ArchiveSet::const_reverse_iterator i=gOpenArchives.rbegin(); i!=gOpenArchives.rend();++i)
  49     {
  50         // always prefer get updated file version
  51         if (SFileOpenFileEx(*i, filename, SFILE_OPEN_PATCHED_FILE, fileHandlerPtr))
  52             return true;
  53     }
  54
  55     return false;
  56 }
  57
  58 bool ExtractFile( char const* mpq_name, std::string const& filename )
  59 {
  60     for(ArchiveSet::const_reverse_iterator i=gOpenArchives.rbegin(); i!=gOpenArchives.rend();++i)
  61     {
  62         HANDLE fileHandle;
  63         if (!SFileOpenFileEx(*i, mpq_name, SFILE_OPEN_PATCHED_FILE, &fileHandle))
  64             continue;
  65
  66         if (SFileGetFileSize(fileHandle, NULL) == 0)              // some files removed in next updates and its reported  size 0
  67         {
  68             SFileCloseFile(fileHandle);
  69             return true;
  70         }
  71
  72         SFileCloseFile(fileHandle);
  73
  74         if (!SFileExtractFile(*i, mpq_name, filename.c_str(), SFILE_OPEN_PATCHED_FILE))
  75         {
  76             printf("Can't extract file: %s\n", mpq_name);
  77             return false;
  78         }
  79
  80         return true;
  81     }
  82
  83     printf("Extracting file not found: %s\n", filename.c_str());
  84     return false;
  85 }
  86
  87
  88 void CloseArchives()
  89 {
  90     for(ArchiveSet::const_iterator i = gOpenArchives.begin(); i != gOpenArchives.end();++i)
  91         SFileCloseArchive(*i);
  92     gOpenArchives.clear();
  93 }
  94
  95 FileLoader::FileLoader()
  96 {
  97     data = 0;
  98     data_size = 0;
  99     version = 0;
 100 }
 101
 102 FileLoader::~FileLoader()
 103 {
 104     free();
 105 }
 106
 107 bool FileLoader::loadFile(char *filename, bool log)
 108 {
 109     free();
 110
 111     HANDLE fileHandle = 0;
 112
 113     if (!OpenNewestFile(filename, &fileHandle))
 114     {
 115         if (log)
 116             printf("No such file %s\n", filename);
 117         return false;
 118     }
 119
 120     data_size = SFileGetFileSize(fileHandle, NULL);
 121
 122     data = new uint8 [data_size];
 123     if (!data)
 124     {
 125         SFileCloseFile(fileHandle);
 126         return false;
 127     }
 128
 129     if (!SFileReadFile(fileHandle, data, data_size, NULL, NULL))
 130     {
 131         if (log)
 132             printf("Can't read file %s\n", filename);
 133         SFileCloseFile(fileHandle);
 134         return false;
 135     }
 136
 137     SFileCloseFile(fileHandle);
 138
 139     // ToDo: Fix WDT errors...
 140     if (!prepareLoadedData())
 141     {
 142         //printf("Error loading %s\n\n", filename);
 143         //free();
 144         return true;
 145     }
 146
 147     return true;
 148 }
 149
 150 bool FileLoader::prepareLoadedData()
 151 {
 152     // Check version
 153     version = (file_MVER *) data;
 154
 155     if (version->fcc != 'MVER')
 156         return false;
 157
 158     if (version->ver != FILE_FORMAT_VERSION)
 159         return false;
 160     return true;
 161 }
 162
 163 void FileLoader::free()
 164 {
 165     if (data) delete[] data;
 166     data = 0;
 167     data_size = 0;
 168     version = 0;
 169 }