diff options
Diffstat (limited to 'dep/src/libmpq/mpq.c')
| -rw-r--r-- | dep/src/libmpq/mpq.c | 1033 |
1 files changed, 0 insertions, 1033 deletions
diff --git a/dep/src/libmpq/mpq.c b/dep/src/libmpq/mpq.c deleted file mode 100644 index 4e3d7db0b4d..00000000000 --- a/dep/src/libmpq/mpq.c +++ /dev/null @@ -1,1033 +0,0 @@ -/* - * mpq.c -- functions for developers using libmpq. - * - * Copyright (c) 2003-2008 Maik Broemme <mbroemme@plusserver.de> - * - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. - */ - -/* mpq-tools configuration includes. */ -#include "config.h" - -/* libmpq main includes. */ -#include "mpq.h" -#include "mpq-internal.h" - -/* libmpq generic includes. */ -#include "common.h" - -/* generic includes. */ -#include <fcntl.h> -#include <stdlib.h> -#include <string.h> -#include <sys/stat.h> -#if HAVE_UNISTD_H - #include <unistd.h> -#endif - -#ifdef _MSC_VER - #define fseeko _fseeki64 -#endif - -/* this function returns the library version information. */ -const char *libmpq__version(void) { - - /* return version information. */ - return VERSION; -} - -/* this function read a file and verify if it is a valid mpq archive, then it read and decrypt the hash table. */ -int32_t libmpq__archive_open(mpq_archive_s **mpq_archive, const char *mpq_filename, libmpq__off_t archive_offset) { - - /* some common variables. */ - uint32_t rb = 0; - uint32_t i = 0; - uint32_t count = 0; - int32_t result = 0; - uint32_t header_search = FALSE; - - if (archive_offset == -1) { - archive_offset = 0; - header_search = TRUE; - } - - if ((*mpq_archive = calloc(1, sizeof(mpq_archive_s))) == NULL) { - - /* archive struct could not be allocated */ - return LIBMPQ_ERROR_MALLOC; - } - - /* check if file exists and is readable */ - if (((*mpq_archive)->fp = fopen(mpq_filename, "rb")) == NULL) { - - /* file could not be opened. */ - result = LIBMPQ_ERROR_OPEN; - goto error; - } - - /* assign some default values. */ - (*mpq_archive)->mpq_header.mpq_magic = 0; - (*mpq_archive)->files = 0; - - /* loop through file and search for mpq signature. */ - while (TRUE) { - - /* reset header values. */ - (*mpq_archive)->mpq_header.mpq_magic = 0; - - /* seek in file. */ - if (fseeko((*mpq_archive)->fp, archive_offset, SEEK_SET) < 0) { - - /* seek in file failed. */ - result = LIBMPQ_ERROR_SEEK; - goto error; - } - - /* read header from file. */ - if ((rb = fread(&(*mpq_archive)->mpq_header, 1, sizeof(mpq_header_s), (*mpq_archive)->fp)) != sizeof(mpq_header_s)) { - - /* no valid mpq archive. */ - result = LIBMPQ_ERROR_FORMAT; - goto error; - } - - /* check if we found a valid mpq header. */ - if ((*mpq_archive)->mpq_header.mpq_magic == LIBMPQ_HEADER) { - - /* check if we process old mpq archive version. */ - if ((*mpq_archive)->mpq_header.version == LIBMPQ_ARCHIVE_VERSION_ONE) { - - /* check if the archive is protected. */ - if ((*mpq_archive)->mpq_header.header_size != sizeof(mpq_header_s)) { - - /* correct header size. */ - (*mpq_archive)->mpq_header.header_size = sizeof(mpq_header_s); - } - } - - /* check if we process new mpq archive version. */ - if ((*mpq_archive)->mpq_header.version == LIBMPQ_ARCHIVE_VERSION_TWO) { - - /* check if the archive is protected. */ - if ((*mpq_archive)->mpq_header.header_size != sizeof(mpq_header_s) + sizeof(mpq_header_ex_s)) { - - /* correct header size. */ - (*mpq_archive)->mpq_header.header_size = sizeof(mpq_header_s) + sizeof(mpq_header_ex_s); - } - } - - /* break the loop, because header was found. */ - break; - } - - /* move to the next possible offset. */ - if (!header_search) { - - /* no valid mpq archive. */ - result = LIBMPQ_ERROR_FORMAT; - goto error; - } - archive_offset += 512; - } - - /* store block size for later use. */ - (*mpq_archive)->block_size = 512 << (*mpq_archive)->mpq_header.block_size; - - /* store archive offset and size for later use. */ - (*mpq_archive)->archive_offset = archive_offset; - - /* check if we process new mpq archive version. */ - if ((*mpq_archive)->mpq_header.version == LIBMPQ_ARCHIVE_VERSION_TWO) { - - /* seek in file. */ - if (fseeko((*mpq_archive)->fp, sizeof(mpq_header_s) + archive_offset, SEEK_SET) < 0) { - - /* seek in file failed. */ - result = LIBMPQ_ERROR_SEEK; - goto error; - } - - /* read header from file. */ - if ((rb = fread(&(*mpq_archive)->mpq_header_ex, 1, sizeof(mpq_header_ex_s), (*mpq_archive)->fp)) != sizeof(mpq_header_ex_s)) { - - /* no valid mpq archive. */ - result = LIBMPQ_ERROR_FORMAT; - goto error; - } - } - - /* allocate memory for the block table, hash table, file and block table to file mapping. */ - if (((*mpq_archive)->mpq_block = calloc((*mpq_archive)->mpq_header.block_table_count, sizeof(mpq_block_s))) == NULL || - ((*mpq_archive)->mpq_block_ex = calloc((*mpq_archive)->mpq_header.block_table_count, sizeof(mpq_block_ex_s))) == NULL || - ((*mpq_archive)->mpq_hash = calloc((*mpq_archive)->mpq_header.hash_table_count, sizeof(mpq_hash_s))) == NULL || - ((*mpq_archive)->mpq_file = calloc((*mpq_archive)->mpq_header.block_table_count, sizeof(mpq_file_s))) == NULL || - ((*mpq_archive)->mpq_map = calloc((*mpq_archive)->mpq_header.block_table_count, sizeof(mpq_map_s))) == NULL) { - - /* memory allocation problem. */ - result = LIBMPQ_ERROR_MALLOC; - goto error; - } - - /* seek in file. */ - if (fseeko((*mpq_archive)->fp, (*mpq_archive)->mpq_header.hash_table_offset + (((long long)((*mpq_archive)->mpq_header_ex.hash_table_offset_high)) << 32) + (*mpq_archive)->archive_offset, SEEK_SET) < 0) { - - /* seek in file failed. */ - result = LIBMPQ_ERROR_SEEK; - goto error; - } - - /* read the hash table into the buffer. */ - if ((rb = fread((*mpq_archive)->mpq_hash, 1, (*mpq_archive)->mpq_header.hash_table_count * sizeof(mpq_hash_s), (*mpq_archive)->fp)) < 0) { - - /* something on read failed. */ - result = LIBMPQ_ERROR_READ; - goto error; - } - - /* decrypt the hashtable. */ - libmpq__decrypt_block((uint32_t *)((*mpq_archive)->mpq_hash), (*mpq_archive)->mpq_header.hash_table_count * sizeof(mpq_hash_s), libmpq__hash_string("(hash table)", 0x300)); - - /* seek in file. */ - if (fseeko((*mpq_archive)->fp, (*mpq_archive)->mpq_header.block_table_offset + (((long long)((*mpq_archive)->mpq_header_ex.block_table_offset_high)) << 32) + (*mpq_archive)->archive_offset, SEEK_SET) < 0) { - - /* seek in file failed. */ - result = LIBMPQ_ERROR_SEEK; - goto error; - } - - /* read the block table into the buffer. */ - if ((rb = fread((*mpq_archive)->mpq_block, 1, (*mpq_archive)->mpq_header.block_table_count * sizeof(mpq_block_s), (*mpq_archive)->fp)) < 0) { - - /* something on read failed. */ - result = LIBMPQ_ERROR_READ; - goto error; - } - - /* decrypt block table. */ - libmpq__decrypt_block((uint32_t *)((*mpq_archive)->mpq_block), (*mpq_archive)->mpq_header.block_table_count * sizeof(mpq_block_s), libmpq__hash_string("(block table)", 0x300)); - - /* check if extended block table is present, regardless of version 2 it is only present in archives > 4GB. */ - if ((*mpq_archive)->mpq_header_ex.extended_offset > 0) { - - /* seek in file. */ - if (fseeko((*mpq_archive)->fp, (*mpq_archive)->mpq_header_ex.extended_offset + archive_offset, SEEK_SET) < 0) { - - /* seek in file failed. */ - result = LIBMPQ_ERROR_SEEK; - goto error; - } - - /* read header from file. */ - if ((rb = fread((*mpq_archive)->mpq_block_ex, 1, (*mpq_archive)->mpq_header.block_table_count * sizeof(mpq_block_ex_s), (*mpq_archive)->fp)) < 0) { - - /* no valid mpq archive. */ - result = LIBMPQ_ERROR_FORMAT; - goto error; - } - } - - /* loop through all files in mpq archive and check if they are valid. */ - for (i = 0; i < (*mpq_archive)->mpq_header.block_table_count; i++) { - - /* save block difference between valid and invalid blocks. */ - (*mpq_archive)->mpq_map[i].block_table_diff = i - count; - - /* check if file exists, sizes and offsets are correct. */ - if (((*mpq_archive)->mpq_block[i].flags & LIBMPQ_FLAG_EXISTS) == 0) { - - /* file does not exist, so nothing to do with that block. */ - continue; - } - - /* create final indices tables. */ - (*mpq_archive)->mpq_map[count].block_table_indices = i; - - /* increase file counter. */ - count++; - } - - /* save the number of files. */ - (*mpq_archive)->files = count; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; - -error: - if ((*mpq_archive)->fp) - fclose((*mpq_archive)->fp); - - free((*mpq_archive)->mpq_map); - free((*mpq_archive)->mpq_file); - free((*mpq_archive)->mpq_hash); - free((*mpq_archive)->mpq_block); - free((*mpq_archive)->mpq_block_ex); - free(*mpq_archive); - - *mpq_archive = NULL; - - return result; -} - -/* this function close the file descriptor, free the decryption buffer and the file list. */ -int32_t libmpq__archive_close(mpq_archive_s *mpq_archive) { - - /* try to close the file */ - if ((fclose(mpq_archive->fp)) < 0) { - - /* don't free anything here, so the caller can try calling us - * again. - */ - return LIBMPQ_ERROR_CLOSE; - } - - /* free header, tables and list. */ - free(mpq_archive->mpq_map); - free(mpq_archive->mpq_file); - free(mpq_archive->mpq_hash); - free(mpq_archive->mpq_block); - free(mpq_archive->mpq_block_ex); - free(mpq_archive); - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the packed size of all files in the archive. */ -int32_t libmpq__archive_packed_size(mpq_archive_s *mpq_archive, libmpq__off_t *packed_size) { - - /* some common variables. */ - uint32_t i; - - /* loop through all files in archive and count packed size. */ - for (i = 0; i < mpq_archive->files; i++) { - *packed_size += mpq_archive->mpq_block[mpq_archive->mpq_map[i].block_table_indices].packed_size; - } - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the unpacked size of all files in the archive. */ -int32_t libmpq__archive_unpacked_size(mpq_archive_s *mpq_archive, libmpq__off_t *unpacked_size) { - - /* some common variables. */ - uint32_t i; - - /* loop through all files in archive and count unpacked size. */ - for (i = 0; i < mpq_archive->files; i++) { - *unpacked_size += mpq_archive->mpq_block[mpq_archive->mpq_map[i].block_table_indices].unpacked_size; - } - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the archive offset (beginning of archive in file). */ -int32_t libmpq__archive_offset(mpq_archive_s *mpq_archive, libmpq__off_t *offset) { - - /* return archive offset. */ - *offset = mpq_archive->archive_offset; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the archive offset. */ -int32_t libmpq__archive_version(mpq_archive_s *mpq_archive, uint32_t *version) { - - /* return archive version. */ - *version = mpq_archive->mpq_header.version + 1; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the number of valid files in archive. */ -int32_t libmpq__archive_files(mpq_archive_s *mpq_archive, uint32_t *files) { - - /* return archive version. */ - *files = mpq_archive->files; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the packed size of the given files in the archive. */ -int32_t libmpq__file_packed_size(mpq_archive_s *mpq_archive, uint32_t file_number, libmpq__off_t *packed_size) { - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* get the packed size of file. */ - *packed_size = mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].packed_size; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the unpacked size of the given file in the archive. */ -int32_t libmpq__file_unpacked_size(mpq_archive_s *mpq_archive, uint32_t file_number, libmpq__off_t *unpacked_size) { - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* get the unpacked size of file. */ - *unpacked_size = mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the file offset (beginning of file in archive). */ -int32_t libmpq__file_offset(mpq_archive_s *mpq_archive, uint32_t file_number, libmpq__off_t *offset) { - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* return file offset relative to archive start. */ - *offset = mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].offset + (((long long)mpq_archive->mpq_block_ex[mpq_archive->mpq_map[file_number].block_table_indices].offset_high) << 32); - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the number of blocks for the given file in the archive. */ -int32_t libmpq__file_blocks(mpq_archive_s *mpq_archive, uint32_t file_number, uint32_t *blocks) { - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* return the number of blocks for the given file. */ - *blocks = (mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_SINGLE) != 0 ? 1 : (mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size + mpq_archive->block_size - 1) / mpq_archive->block_size; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return if the file is encrypted or not. */ -int32_t libmpq__file_encrypted(mpq_archive_s *mpq_archive, uint32_t file_number, uint32_t *encrypted) { - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* return the encryption status of file. */ - *encrypted = (mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_ENCRYPTED) != 0 ? TRUE : FALSE; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return if the file is compressed or not. */ -int32_t libmpq__file_compressed(mpq_archive_s *mpq_archive, uint32_t file_number, uint32_t *compressed) { - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* return the compression status of file. */ - *compressed = (mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_COMPRESS_MULTI) != 0 ? TRUE : FALSE; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return if the file is imploded or not. */ -int32_t libmpq__file_imploded(mpq_archive_s *mpq_archive, uint32_t file_number, uint32_t *imploded) { - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* return the implosion status of file. */ - *imploded = (mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_COMPRESS_PKZIP) != 0 ? TRUE : FALSE; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return filenumber by the given name. */ -int32_t libmpq__file_number(mpq_archive_s *mpq_archive, const char *filename, uint32_t *number) { - - /* some common variables. */ - uint32_t i, hash1, hash2, hash3, ht_count; - - /* if the list of file names doesn't include this one, we'll have - * to figure out the file number the "hard" way. - */ - ht_count = mpq_archive->mpq_header.hash_table_count; - - hash1 = libmpq__hash_string (filename, 0x0) & (ht_count - 1); - hash2 = libmpq__hash_string (filename, 0x100); - hash3 = libmpq__hash_string (filename, 0x200); - - /* loop through all files in mpq archive. - * hash1 gives us a clue about the starting position of this - * search. - */ - for (i = hash1; mpq_archive->mpq_hash[i].block_table_index != LIBMPQ_HASH_FREE; i = (i + 1) & (ht_count - 1)) { - - /* if the other two hashes match, we found our file number. */ - if (mpq_archive->mpq_hash[i].hash_a == hash2 && - mpq_archive->mpq_hash[i].hash_b == hash3) { - - /* return the file number. */ - *number = mpq_archive->mpq_hash[i].block_table_index - mpq_archive->mpq_map[mpq_archive->mpq_hash[i].block_table_index].block_table_diff; - - /* we found our file, return zero. */ - return LIBMPQ_SUCCESS; - } - - /* check if we have cycled through the whole hash table */ - if (((i + 1) & (ht_count - 1)) == hash1) { - break; - } - } - - /* if no matching entry found, so return error. */ - return LIBMPQ_ERROR_EXIST; -} - -/* this function read the given file from archive into a buffer. */ -int32_t libmpq__file_read(mpq_archive_s *mpq_archive, uint32_t file_number, uint8_t *out_buf, libmpq__off_t out_size, libmpq__off_t *transferred) { - - /* some common variables. */ - uint32_t i; - uint32_t blocks = 0; - int32_t result = 0; - libmpq__off_t file_offset = 0; - libmpq__off_t unpacked_size = 0; - libmpq__off_t transferred_block = 0; - libmpq__off_t transferred_total = 0; - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* get target size of block. */ - libmpq__file_unpacked_size(mpq_archive, file_number, &unpacked_size); - - /* check if target buffer is to small. */ - if (unpacked_size > out_size) { - - /* output buffer size is to small or block size is unknown. */ - return LIBMPQ_ERROR_SIZE; - } - - /* fetch file offset. */ - libmpq__file_offset(mpq_archive, file_number, &file_offset); - - /* get block count for file. */ - libmpq__file_blocks(mpq_archive, file_number, &blocks); - - /* open the packed block offset table. */ - if ((result = libmpq__block_open_offset(mpq_archive, file_number)) < 0) { - - /* something on opening packed block offset table failed. */ - return result; - } - - /* loop through all blocks. */ - for (i = 0; i < blocks; i++) { - - /* cleanup size variable. */ - unpacked_size = 0; - - /* get unpacked block size. */ - libmpq__block_unpacked_size(mpq_archive, file_number, i, &unpacked_size); - - /* read block. */ - if ((result = libmpq__block_read(mpq_archive, file_number, i, out_buf + transferred_total, unpacked_size, &transferred_block)) < 0) { - - /* close the packed block offset table. */ - libmpq__block_close_offset(mpq_archive, file_number); - - /* something on reading block failed. */ - return result; - } - - transferred_total += transferred_block; - - } - - /* close the packed block offset table. */ - libmpq__block_close_offset(mpq_archive, file_number); - - /* check for null pointer. */ - if (transferred != NULL) { - - /* store transferred bytes. */ - *transferred = transferred_total; - } - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function open a file in the given archive and caches the block offset information. */ -int32_t libmpq__block_open_offset(mpq_archive_s *mpq_archive, uint32_t file_number) { - - /* some common variables. */ - uint32_t i; - uint32_t packed_size; - int32_t rb = 0; - int32_t result = 0; - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - if (mpq_archive->mpq_file[file_number]) { - - /* file already opened, so increment counter */ - mpq_archive->mpq_file[file_number]->open_count++; - return LIBMPQ_SUCCESS; - } - - /* check if file is not stored in a single sector. */ - if ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_SINGLE) == 0) { - - /* get packed size based on block size and block count. */ - packed_size = sizeof(uint32_t) * (((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size + mpq_archive->block_size - 1) / mpq_archive->block_size) + 1); - } else { - - /* file is stored in single sector and we need only two entries for the packed block offset table. */ - packed_size = sizeof(uint32_t) * 2; - } - - /* check if data has one extra entry. */ - if ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_EXTRA) != 0) { - - /* add one uint32_t. */ - packed_size += sizeof(uint32_t); - } - - /* allocate memory for the file. */ - if ((mpq_archive->mpq_file[file_number] = calloc(1, sizeof(mpq_file_s))) == NULL) { - - /* memory allocation problem. */ - result = LIBMPQ_ERROR_MALLOC; - goto error; - } - - /* allocate memory for the packed block offset table. */ - if ((mpq_archive->mpq_file[file_number]->packed_offset = calloc(1, packed_size)) == NULL) { - - /* memory allocation problem. */ - result = LIBMPQ_ERROR_MALLOC; - goto error; - } - - /* initialize counter to one opening */ - mpq_archive->mpq_file[file_number]->open_count = 1; - - /* check if we need to load the packed block offset table, we will maintain this table for unpacked files too. */ - if ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_COMPRESSED) != 0 && - (mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_SINGLE) == 0) { - - /* seek to block position. */ - if (fseeko(mpq_archive->fp, mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].offset + (((long long)mpq_archive->mpq_block_ex[mpq_archive->mpq_map[file_number].block_table_indices].offset_high) << 32) + mpq_archive->archive_offset, SEEK_SET) < 0) { - - /* seek in file failed. */ - result = LIBMPQ_ERROR_SEEK; - goto error; - } - - /* read block positions from begin of file. */ - if ((rb = fread(mpq_archive->mpq_file[file_number]->packed_offset, 1, packed_size, mpq_archive->fp)) < 0) { - - /* something on read from archive failed. */ - result = LIBMPQ_ERROR_READ; - goto error; - } - - /* check if the archive is protected some way, sometimes the file appears not to be encrypted, but it is. */ - if (mpq_archive->mpq_file[file_number]->packed_offset[0] != rb) { - - /* file is encrypted. */ - mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags |= LIBMPQ_FLAG_ENCRYPTED; - } - - /* check if packed offset block is encrypted, we have to decrypt it. */ - if (mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_ENCRYPTED) { - - /* check if we don't know the file seed, try to find it. */ - if ((mpq_archive->mpq_file[file_number]->seed = libmpq__decrypt_key((uint8_t *)mpq_archive->mpq_file[file_number]->packed_offset, packed_size, mpq_archive->block_size)) < 0) { - - /* sorry without seed, we cannot extract file. */ - result = LIBMPQ_ERROR_DECRYPT; - goto error; - } - - /* decrypt block in input buffer. */ - if (libmpq__decrypt_block(mpq_archive->mpq_file[file_number]->packed_offset, packed_size, mpq_archive->mpq_file[file_number]->seed - 1) < 0 ) { - - /* something on decrypt failed. */ - result = LIBMPQ_ERROR_DECRYPT; - goto error; - } - - /* check if the block positions are correctly decrypted. */ - if (mpq_archive->mpq_file[file_number]->packed_offset[0] != packed_size) { - - /* sorry without seed, we cannot extract file. */ - result = LIBMPQ_ERROR_DECRYPT; - goto error; - } - } - } else { - - /* check if file is not stored in a single sector. */ - if ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_SINGLE) == 0) { - - /* loop through all blocks and create packed block offset table based on block size. */ - for (i = 0; i < ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size + mpq_archive->block_size - 1) / mpq_archive->block_size + 1); i++) { - - /* check if we process the last block. */ - if (i == ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size + mpq_archive->block_size - 1) / mpq_archive->block_size)) { - - /* store size of last block. */ - mpq_archive->mpq_file[file_number]->packed_offset[i] = mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size; - } else { - - /* store default block size. */ - mpq_archive->mpq_file[file_number]->packed_offset[i] = i * mpq_archive->block_size; - } - } - } else { - - /* store offsets. */ - mpq_archive->mpq_file[file_number]->packed_offset[0] = 0; - mpq_archive->mpq_file[file_number]->packed_offset[1] = mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].packed_size; - } - } - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; - -error: - - /* free packed block offset table and file pointer. */ - free(mpq_archive->mpq_file[file_number]->packed_offset); - free(mpq_archive->mpq_file[file_number]); - - /* return error constant. */ - return result; -} - -/* this function free the file pointer to the opened file in archive. */ -int32_t libmpq__block_close_offset(mpq_archive_s *mpq_archive, uint32_t file_number) { - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - if (mpq_archive->mpq_file[file_number] == NULL) { - - /* packed block offset table is not opened. */ - return LIBMPQ_ERROR_OPEN; - } - - mpq_archive->mpq_file[file_number]->open_count--; - - if (mpq_archive->mpq_file[file_number]->open_count != 0) { - - /* still in use */ - return LIBMPQ_SUCCESS; - } - - /* free packed block offset table and file pointer. */ - free(mpq_archive->mpq_file[file_number]->packed_offset); - free(mpq_archive->mpq_file[file_number]); - - /* mark it as unopened - libmpq__block_open_offset checks for this to decide whether to increment the counter */ - mpq_archive->mpq_file[file_number] = NULL; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the unpacked size of the given file and block in the archive. */ -int32_t libmpq__block_unpacked_size(mpq_archive_s *mpq_archive, uint32_t file_number, uint32_t block_number, libmpq__off_t *unpacked_size) { - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* check if given block number is not out of range. */ - if (block_number < 0 || block_number >= ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_SINGLE) != 0 ? 1 : (mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size + mpq_archive->block_size - 1) / mpq_archive->block_size)) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* check if packed block offset table is opened. */ - if (mpq_archive->mpq_file[file_number] == NULL || - mpq_archive->mpq_file[file_number]->packed_offset == NULL) { - - /* packed block offset table is not opened. */ - return LIBMPQ_ERROR_OPEN; - } - - /* check if block is stored as single sector. */ - if ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_SINGLE) != 0) { - - /* return the unpacked size of the block in the mpq archive. */ - *unpacked_size = mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size; - } - - /* check if block is not stored as single sector. */ - if ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_SINGLE) == 0) { - - /* check if we not process the last block. */ - if (block_number < ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size + mpq_archive->block_size - 1) / mpq_archive->block_size) - 1) { - - /* return the block size as unpacked size. */ - *unpacked_size = mpq_archive->block_size; - } else { - - /* return the unpacked size of the last block in the mpq archive. */ - *unpacked_size = mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size - mpq_archive->block_size * block_number; - } - } - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function return the decryption seed for the given file and block. */ -int32_t libmpq__block_seed(mpq_archive_s *mpq_archive, uint32_t file_number, uint32_t block_number, uint32_t *seed) { - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* check if given block number is not out of range. */ - if (block_number < 0 || block_number >= ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_SINGLE) != 0 ? 1 : (mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size + mpq_archive->block_size - 1) / mpq_archive->block_size)) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* check if packed block offset table is opened. */ - if (mpq_archive->mpq_file[file_number] == NULL || - mpq_archive->mpq_file[file_number]->packed_offset == NULL) { - - /* packed block offset table is not opened. */ - return LIBMPQ_ERROR_OPEN; - } - - /* return the decryption key. */ - *seed = mpq_archive->mpq_file[file_number]->seed + block_number; - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} - -/* this function read the given block from archive into a buffer. */ -int32_t libmpq__block_read(mpq_archive_s *mpq_archive, uint32_t file_number, uint32_t block_number, uint8_t *out_buf, libmpq__off_t out_size, libmpq__off_t *transferred) { - - /* some common variables. */ - uint8_t *in_buf; - uint32_t seed = 0; - uint32_t encrypted = 0; - uint32_t compressed = 0; - uint32_t imploded = 0; - int32_t tb = 0; - libmpq__off_t block_offset = 0; - off_t in_size = 0; - libmpq__off_t unpacked_size = 0; - - /* check if given file number is not out of range. */ - if (file_number < 0 || file_number > mpq_archive->files - 1) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* check if given block number is not out of range. */ - if (block_number < 0 || block_number >= ((mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].flags & LIBMPQ_FLAG_SINGLE) != 0 ? 1 : (mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].unpacked_size + mpq_archive->block_size - 1) / mpq_archive->block_size)) { - - /* file number is out of range. */ - return LIBMPQ_ERROR_EXIST; - } - - /* check if packed block offset table is opened. */ - if (mpq_archive->mpq_file[file_number] == NULL || - mpq_archive->mpq_file[file_number]->packed_offset == NULL) { - - /* packed block offset table is not opened. */ - return LIBMPQ_ERROR_OPEN; - } - - /* get target size of block. */ - libmpq__block_unpacked_size(mpq_archive, file_number, block_number, &unpacked_size); - - /* check if target buffer is to small. */ - if (unpacked_size > out_size) { - - /* output buffer size is to small or block size is unknown. */ - return LIBMPQ_ERROR_SIZE; - } - - /* fetch some required values like input buffer size and block offset. */ - block_offset = mpq_archive->mpq_block[mpq_archive->mpq_map[file_number].block_table_indices].offset + (((long long)mpq_archive->mpq_block_ex[mpq_archive->mpq_map[file_number].block_table_indices].offset_high) << 32) + mpq_archive->mpq_file[file_number]->packed_offset[block_number]; - in_size = mpq_archive->mpq_file[file_number]->packed_offset[block_number + 1] - mpq_archive->mpq_file[file_number]->packed_offset[block_number]; - - /* seek in file. */ - if (fseeko(mpq_archive->fp, block_offset + mpq_archive->archive_offset, SEEK_SET) < 0) { - - /* something with seek in file failed. */ - return LIBMPQ_ERROR_SEEK; - } - - /* allocate memory for the read buffer. */ - if ((in_buf = calloc(1, in_size)) == NULL) { - - /* memory allocation problem. */ - return LIBMPQ_ERROR_MALLOC; - } - - /* read block from file. */ - if (fread(in_buf, 1, in_size, mpq_archive->fp) < 0) { - - /* free buffers. */ - free(in_buf); - - /* something on reading block failed. */ - return LIBMPQ_ERROR_READ; - } - - /* get encryption status. */ - libmpq__file_encrypted(mpq_archive, file_number, &encrypted); - - /* check if file is encrypted. */ - if (encrypted == 1) { - - /* get decryption key. */ - libmpq__block_seed(mpq_archive, file_number, block_number, &seed); - - /* decrypt block. */ - if (libmpq__decrypt_block((uint32_t *)in_buf, in_size, seed) < 0) { - - /* free buffers. */ - free(in_buf); - - /* something on decrypting block failed. */ - return LIBMPQ_ERROR_DECRYPT; - } - } - - /* get compression status. */ - libmpq__file_compressed(mpq_archive, file_number, &compressed); - - /* check if file is compressed. */ - if (compressed == 1) { - - /* decompress block. */ - if ((tb = libmpq__decompress_block(in_buf, in_size, out_buf, out_size, LIBMPQ_FLAG_COMPRESS_MULTI)) < 0) { - - /* free temporary buffer. */ - free(in_buf); - - /* something on decompressing block failed. */ - return LIBMPQ_ERROR_UNPACK; - } - } - - /* get implosion status. */ - libmpq__file_imploded(mpq_archive, file_number, &imploded); - - /* check if file is imploded. */ - if (imploded == 1) { - - /* explode block. */ - if ((tb = libmpq__decompress_block(in_buf, in_size, out_buf, out_size, LIBMPQ_FLAG_COMPRESS_PKZIP)) < 0) { - - /* free temporary buffer. */ - free(in_buf); - - /* something on decompressing block failed. */ - return LIBMPQ_ERROR_UNPACK; - } - } - - /* check if file is neither compressed nor imploded. */ - if (compressed == 0 && imploded == 0) { - - /* copy block. */ - if ((tb = libmpq__decompress_block(in_buf, in_size, out_buf, out_size, LIBMPQ_FLAG_COMPRESS_NONE)) < 0) { - - /* free temporary buffer. */ - free(in_buf); - - /* something on decompressing block failed. */ - return LIBMPQ_ERROR_UNPACK; - } - } - - /* free read buffer. */ - free(in_buf); - - /* check for null pointer. */ - if (transferred != NULL) { - - /* store transferred bytes. */ - *transferred = tb; - } - - /* if no error was found, return zero. */ - return LIBMPQ_SUCCESS; -} |