189 lines
7.6 KiB
C++
Executable File
189 lines
7.6 KiB
C++
Executable File
//-----------------------------------------------------------------------------
|
|
// Torque Game Engine
|
|
// Copyright (C) GarageGames.com, Inc.
|
|
//-----------------------------------------------------------------------------
|
|
|
|
#include "platform/platform.h"
|
|
#include "core/stream.h"
|
|
#include "core/zipHeaders.h"
|
|
|
|
const U32 ZipLocalFileHeader::csm_localFileHeaderSig = 0x04034b50;
|
|
const U32 ZipDirFileHeader::csm_dirFileHeaderSig = 0x02014b50;
|
|
const U32 ZipEOCDRecord::csm_eocdRecordSig = 0x06054b50;
|
|
|
|
bool
|
|
ZipLocalFileHeader::readFromStream(Stream& io_rStream)
|
|
{
|
|
AssertFatal(io_rStream.getStatus() == Stream::Ok,
|
|
"Error, stream is closed or has an uncleared error.");
|
|
AssertFatal(io_rStream.hasCapability(Stream::StreamPosition),
|
|
"Must be positionable stream to read zip headers...");
|
|
|
|
// Read the initial header fields, marking the initial position...
|
|
//
|
|
U32 initialPosition = io_rStream.getPosition();
|
|
// this is NOT endian-safe code!!!
|
|
//bool success = io_rStream.read(sizeof(m_header), &m_header);
|
|
// this IS.
|
|
bool success = io_rStream.read(&m_header.headerSig);
|
|
if (success == false || m_header.headerSig != csm_localFileHeaderSig) {
|
|
AssertWarn(0, "Unable to retrieve local file header from stream position...");
|
|
io_rStream.setPosition(initialPosition);
|
|
return false;
|
|
}
|
|
|
|
// NOW, read in the rest of the fields. ENDIAN-SAFE.
|
|
// not checking errors to start with...
|
|
success &= io_rStream.read(&m_header.versionToDecompress);
|
|
success &= io_rStream.read(&m_header.bitFlags);
|
|
success &= io_rStream.read(&m_header.compressionMethod);
|
|
success &= io_rStream.read(&m_header.lastModTime);
|
|
success &= io_rStream.read(&m_header.lastModDate);
|
|
success &= io_rStream.read(&m_header.crc32);
|
|
success &= io_rStream.read(&m_header.compressedSize);
|
|
success &= io_rStream.read(&m_header.uncompressedSize);
|
|
success &= io_rStream.read(&m_header.fileNameLength);
|
|
success &= io_rStream.read(&m_header.extraFieldLength);
|
|
if (success == false)
|
|
{
|
|
AssertWarn(0, "Failed retrieving local file header fields from stream...");
|
|
io_rStream.setPosition(initialPosition);
|
|
return false;
|
|
}
|
|
|
|
|
|
// Read the variable length file name from the stream...
|
|
//
|
|
AssertFatal(m_header.fileNameLength < (MaxFileNameLength - 1),
|
|
"Filename too long, increase structure size");
|
|
success = io_rStream.read(m_header.fileNameLength, m_pFileName);
|
|
m_pFileName[m_header.fileNameLength] = '\0';
|
|
if (success == false) {
|
|
AssertWarn(0, "Unable to read file name from stream position...");
|
|
io_rStream.setPosition(initialPosition);
|
|
return false;
|
|
}
|
|
|
|
|
|
// And seek to the end of the header, ignoring the extra field.
|
|
io_rStream.setPosition(initialPosition +
|
|
(sizeof(m_header) +
|
|
m_header.fileNameLength +
|
|
m_header.extraFieldLength));
|
|
return true;
|
|
}
|
|
|
|
|
|
bool
|
|
ZipDirFileHeader::readFromStream(Stream& io_rStream)
|
|
{
|
|
AssertFatal(io_rStream.getStatus() == Stream::Ok,
|
|
"Error, stream is closed or has an uncleared error.");
|
|
AssertFatal(io_rStream.hasCapability(Stream::StreamPosition),
|
|
"Must be positionable stream to read zip headers...");
|
|
|
|
// Read the initial header fields, marking the initial position...
|
|
//
|
|
U32 initialPosition = io_rStream.getPosition();
|
|
// this is NOT endian-safe code!!!
|
|
//bool success = io_rStream.read(sizeof(m_header), &m_header);
|
|
// this IS.
|
|
bool success = io_rStream.read(&m_header.headerSig);
|
|
if (success == false || m_header.headerSig != csm_dirFileHeaderSig) {
|
|
AssertWarn(0, "Unable to retrieve dir file header from stream position...");
|
|
io_rStream.setPosition(initialPosition);
|
|
return false;
|
|
}
|
|
|
|
// NOW, read in the rest of the fields. ENDIAN-SAFE.
|
|
// not checking errors to start with...
|
|
success &= io_rStream.read(&m_header.versionMadeBy);
|
|
success &= io_rStream.read(&m_header.versionToDecompress);
|
|
success &= io_rStream.read(&m_header.bitFlags);
|
|
success &= io_rStream.read(&m_header.compressionMethod);
|
|
success &= io_rStream.read(&m_header.lastModTime);
|
|
success &= io_rStream.read(&m_header.lastModDate);
|
|
success &= io_rStream.read(&m_header.crc32);
|
|
success &= io_rStream.read(&m_header.compressedSize);
|
|
success &= io_rStream.read(&m_header.uncompressedSize);
|
|
success &= io_rStream.read(&m_header.fileNameLength);
|
|
success &= io_rStream.read(&m_header.extraFieldLength);
|
|
success &= io_rStream.read(&m_header.fileCommentLength);
|
|
success &= io_rStream.read(&m_header.diskNumberStart);
|
|
success &= io_rStream.read(&m_header.internalFileAttributes);
|
|
success &= io_rStream.read(&m_header.externalFileAttributes);
|
|
success &= io_rStream.read(&m_header.relativeOffsetOfLocalHeader);
|
|
if (success == false)
|
|
{
|
|
AssertWarn(0, "Failed retrieving dir file header fields from stream...");
|
|
io_rStream.setPosition(initialPosition);
|
|
return false;
|
|
}
|
|
|
|
// Read the variable length file name from the stream...
|
|
//
|
|
AssertFatal(m_header.fileNameLength < (MaxFileNameLength - 1),
|
|
"Filename too long, increase structure size");
|
|
success = io_rStream.read(m_header.fileNameLength, m_pFileName);
|
|
m_pFileName[m_header.fileNameLength] = '\0';
|
|
if (success == false) {
|
|
AssertWarn(0, "Unable to read dir file name from stream position...");
|
|
io_rStream.setPosition(initialPosition);
|
|
return false;
|
|
}
|
|
|
|
|
|
// And seek to the end of the header, ignoring the extra field.
|
|
io_rStream.setPosition(initialPosition +
|
|
(sizeof(m_header) +
|
|
m_header.fileNameLength +
|
|
m_header.extraFieldLength));
|
|
return true;
|
|
}
|
|
|
|
|
|
bool
|
|
ZipEOCDRecord::readFromStream(Stream& io_rStream)
|
|
{
|
|
AssertFatal(io_rStream.getStatus() == Stream::Ok,
|
|
"Error, stream is closed or has an uncleared error.");
|
|
AssertFatal(io_rStream.hasCapability(Stream::StreamPosition),
|
|
"Must be positionable stream to read zip headers...");
|
|
|
|
// Read the initial header fields, marking the initial position...
|
|
//
|
|
U32 initialPosition = io_rStream.getPosition();
|
|
// this is NOT endian-safe code!!!
|
|
//bool success = io_rStream.read(sizeof(m_record), &m_record);
|
|
// this IS.
|
|
bool success = io_rStream.read(&m_record.eocdSig);
|
|
if (success == false || m_record.eocdSig != csm_eocdRecordSig)
|
|
{
|
|
AssertWarn(0, "Unable to retrieve EOCD header from stream position...");
|
|
io_rStream.setPosition(initialPosition);
|
|
return false;
|
|
}
|
|
|
|
// NOW, read in the rest of the fields. ENDIAN-SAFE.
|
|
// not checking errors to start with...
|
|
success &= io_rStream.read(&m_record.diskNumber);
|
|
success &= io_rStream.read(&m_record.eocdDiskNumber);
|
|
success &= io_rStream.read(&m_record.numCDEntriesDisk);
|
|
success &= io_rStream.read(&m_record.numCDEntriesTotal);
|
|
success &= io_rStream.read(&m_record.cdSize);
|
|
success &= io_rStream.read(&m_record.cdOffset);
|
|
success &= io_rStream.read(&m_record.zipFileCommentLength);
|
|
if (success == false)
|
|
{
|
|
AssertWarn(0, "Failed retrieving EOCD header fields from stream...");
|
|
io_rStream.setPosition(initialPosition);
|
|
return false;
|
|
}
|
|
|
|
// And seek to the end of the header, ignoring the extra field.
|
|
io_rStream.setPosition(initialPosition +
|
|
(sizeof(m_record) + m_record.zipFileCommentLength));
|
|
return true;
|
|
}
|
|
|