Initial commit

Torque/SDK/engine/core/zipHeaders.cc

@@ -0,0 +1,188 @@
+//-----------------------------------------------------------------------------
+// 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;
+}
+