[Phreak] 15403: Rewrote usetup's file copy and cabinet extraction code to use memory mapped files
19 May
2005
19 May
'05
12:59 a.m.
Rewrote usetup's file copy and cabinet extraction code to use memory
mapped files
This increased performance and the code is cleaner
Still needs a bit more cleaning up, but so far, looks good, needs some
testing
Modified: trunk/reactos/subsys/system/usetup/cabinet.c
Modified: trunk/reactos/subsys/system/usetup/cabinet.h
Modified: trunk/reactos/subsys/system/usetup/filesup.c
_____
Modified: trunk/reactos/subsys/system/usetup/cabinet.c
--- trunk/reactos/subsys/system/usetup/cabinet.c 2005-05-18
22:05:11 UTC (rev 15402)
+++ trunk/reactos/subsys/system/usetup/cabinet.c 2005-05-19
00:59:40 UTC (rev 15403)
@@ -46,30 +46,21 @@
static WCHAR DiskNext[256]; // Label of cabinet in file
CabinetNext
static ULONG FolderUncompSize = 0; // Uncompressed size of folder
static ULONG BytesLeftInBlock = 0; // Number of bytes left in
current block
-static BOOL ReuseBlock = FALSE;
static WCHAR DestPath[MAX_PATH];
static HANDLE FileHandle;
+static HANDLE FileSectionHandle;
+static PUCHAR FileBuffer;
+static ULONG DestFileSize;
+static ULONG FileSize;
static BOOL FileOpen = FALSE;
-static CFHEADER CABHeader;
+static PCFHEADER PCABHeader;
+static PCFFOLDER CabinetFolders;
static ULONG CabinetReserved = 0;
static ULONG FolderReserved = 0;
static ULONG DataReserved = 0;
-static PCFFOLDER_NODE FolderListHead = NULL;
-static PCFFOLDER_NODE FolderListTail = NULL;
-static PCFFOLDER_NODE CurrentFolderNode = NULL;
-static PCFDATA_NODE CurrentDataNode = NULL;
-static PCFFILE_NODE FileListHead = NULL;
-static PCFFILE_NODE FileListTail = NULL;
static ULONG CodecId;
static PCABINET_CODEC_UNCOMPRESS CodecUncompress = NULL;
static BOOL CodecSelected = FALSE;
-static PVOID InputBuffer = NULL;
-static PVOID CurrentIBuffer = NULL; // Current offset in input
buffer
-static ULONG CurrentIBufferSize = 0; // Bytes left in input buffer
-static PVOID OutputBuffer = NULL;
-static PVOID CurrentOBuffer = NULL; // Current offset in output
buffer
-static ULONG CurrentOBufferSize = 0; // Bytes left in output
buffer
-static BOOL RestartSearch = FALSE;
static ULONG LastFileOffset = 0; // Uncompressed offset of
last extracted file
static PCABINET_OVERWRITE OverwriteHandler = NULL;
static PCABINET_EXTRACT ExtractHandler = NULL;
@@ -94,20 +85,23 @@
ULONG
RawCodecUncompress(PVOID OutputBuffer,
- PVOID InputBuffer,
- ULONG InputLength,
- PULONG OutputLength)
+ PVOID InputBuffer,
+ PLONG InputLength,
+ PLONG OutputLength)
/*
* FUNCTION: Uncompresses data in a buffer
* ARGUMENTS:
* OutputBuffer = Pointer to buffer to place uncompressed data
* InputBuffer = Pointer to buffer with data to be uncompressed
- * InputLength = Length of input buffer
- * OutputLength = Address of buffer to place size of uncompressed
data
+ * InputLength = Length of input buffer before, and amount
consumed after
+ * Negative to indicate that this is not the start
of a new block
+ * OutputLength = Length of output buffer before, amount filled
after
+ * Negative to indicate that this is not the end of
the block
*/
{
- memcpy(OutputBuffer, InputBuffer, InputLength);
- *OutputLength = InputLength;
+ LONG In = abs(*InputLength), Out = abs(*OutputLength);
+ memcpy(OutputBuffer, InputBuffer, In < Out ? In : Out);
+ *InputLength = *OutputLength = In < Out ? In : Out;
return CS_SUCCESS;
}
@@ -116,70 +110,80 @@
ULONG
MSZipCodecUncompress(PVOID OutputBuffer,
- PVOID InputBuffer,
- ULONG InputLength,
- PULONG OutputLength)
+ PVOID InputBuffer,
+ PLONG InputLength,
+ PLONG OutputLength)
/*
* FUNCTION: Uncompresses data in a buffer
* ARGUMENTS:
* OutputBuffer = Pointer to buffer to place uncompressed data
* InputBuffer = Pointer to buffer with data to be uncompressed
- * InputLength = Length of input buffer
- * OutputLength = Address of buffer to place size of uncompressed
data
+ * InputLength = Length of input buffer before, and amount
consumed after
+ * Negative to indicate that this is not the start
of a new block
+ * OutputLength = Length of output buffer before, amount filled
after
+ * Negative to indicate that this is not the end of
the block
*/
{
USHORT Magic;
INT Status;
- DPRINT("InputLength (%d).\n", InputLength);
+ DPRINT("MSZipCodecUncompress( OutputBuffer = %x, InputBuffer = %x,
InputLength = %d, OutputLength = %d.\n", OutputBuffer, InputBuffer,
*InputLength, *OutputLength);
+ if( *InputLength > 0 )
+ {
+ Magic = *((PUSHORT)InputBuffer);
- Magic = *((PUSHORT)InputBuffer);
+ if (Magic != MSZIP_MAGIC)
+ {
+ DPRINT("Bad MSZIP block header magic (0x%X)\n",
Magic);
+ return CS_BADSTREAM;
+ }
+
+ ZStream.next_in = (PUCHAR)((ULONG)InputBuffer + 2);
+ ZStream.avail_in = *InputLength - 2;
+ ZStream.next_out = (PUCHAR)OutputBuffer;
+ ZStream.avail_out = abs(*OutputLength);
- if (Magic != MSZIP_MAGIC)
- {
- DPRINT("Bad MSZIP block header magic (0x%X)\n", Magic);
- return CS_BADSTREAM;
- }
+ /* WindowBits is passed < 0 to tell that there is no zlib
header.
+ * Note that in this case inflate *requires* an extra "dummy"
byte
+ * after the compressed stream in order to complete
decompression and
+ * return Z_STREAM_END.
+ */
+ Status = inflateInit2(&ZStream, -MAX_WBITS);
+ if (Status != Z_OK)
+ {
+ DPRINT("inflateInit2() returned (%d).\n", Status);
+ return CS_BADSTREAM;
+ }
+ ZStream.total_in = 2;
+ }
+ else {
+ ZStream.avail_in = -*InputLength;
+ ZStream.next_in = (PUCHAR)InputBuffer;
+ ZStream.next_out = (PUCHAR)OutputBuffer;
+ ZStream.avail_out = abs(*OutputLength);
+ ZStream.total_in = 0;
+ }
+ ZStream.total_out = 0;
+ Status = inflate(&ZStream, Z_SYNC_FLUSH );
+ if (Status != Z_OK && Status != Z_STREAM_END)
+ {
+ DPRINT("inflate() returned (%d) (%s).\n", Status,
ZStream.msg);
+ if (Status == Z_MEM_ERROR)
+ return CS_NOMEMORY;
+ return CS_BADSTREAM;
+ }
- ZStream.next_in = (PUCHAR)((ULONG)InputBuffer + 2);
- ZStream.avail_in = InputLength - 2;
- ZStream.next_out = (PUCHAR)OutputBuffer;
- ZStream.avail_out = CAB_BLOCKSIZE + 12;
-
- /* WindowBits is passed < 0 to tell that there is no zlib header.
- * Note that in this case inflate *requires* an extra "dummy" byte
- * after the compressed stream in order to complete decompression and
- * return Z_STREAM_END.
- */
- Status = inflateInit2(&ZStream, -MAX_WBITS);
- if (Status != Z_OK)
- {
- DPRINT("inflateInit2() returned (%d).\n", Status);
- return CS_BADSTREAM;
- }
-
- while ((ZStream.total_out < CAB_BLOCKSIZE + 12) &&
- (ZStream.total_in < InputLength - 2))
- {
- Status = inflate(&ZStream, Z_NO_FLUSH);
- if (Status == Z_STREAM_END) break;
- if (Status != Z_OK)
- {
- DPRINT("inflate() returned (%d) (%s).\n", Status,
ZStream.msg);
- if (Status == Z_MEM_ERROR)
- return CS_NOMEMORY;
- return CS_BADSTREAM;
- }
- }
-
+ if( *OutputLength > 0 )
+ {
+ Status = inflateEnd(&ZStream);
+ if (Status != Z_OK)
+ {
+ DPRINT("inflateEnd() returned (%d).\n", Status);
+ return CS_BADSTREAM;
+ }
+ }
*OutputLength = ZStream.total_out;
-
- Status = inflateEnd(&ZStream);
- if (Status != Z_OK)
- {
- DPRINT("inflateEnd() returned (%d).\n", Status);
- return CS_BADSTREAM;
- }
+ *InputLength = ZStream.total_in;
return CS_SUCCESS;
}
@@ -197,97 +201,14 @@
RtlFreeHeap(ProcessHeap, 0, address);
}
-
-static DWORD
-SeekInFile(HANDLE hFile,
- LONG lDistanceToMove,
- PLONG lpDistanceToMoveHigh,
- DWORD dwMoveMethod,
- PNTSTATUS Status)
-{
- FILE_POSITION_INFORMATION FilePosition;
- FILE_STANDARD_INFORMATION FileStandard;
- NTSTATUS errCode;
- IO_STATUS_BLOCK IoStatusBlock;
- LARGE_INTEGER Distance;
-
- DPRINT("SeekInFile(hFile %x, lDistanceToMove %d, dwMoveMethod %d)\n",
- hFile,lDistanceToMove,dwMoveMethod);
-
- Distance.u.LowPart = lDistanceToMove;
- if (lpDistanceToMoveHigh)
- {
- Distance.u.HighPart = *lpDistanceToMoveHigh;
- }
- else if (lDistanceToMove >= 0)
- {
- Distance.u.HighPart = 0;
- }
- else
- {
- Distance.u.HighPart = -1;
- }
-
- if (dwMoveMethod == SEEK_CURRENT)
- {
- NtQueryInformationFile(hFile,
- &IoStatusBlock,
- &FilePosition,
- sizeof(FILE_POSITION_INFORMATION),
- FilePositionInformation);
- FilePosition.CurrentByteOffset.QuadPart += Distance.QuadPart;
- }
- else if (dwMoveMethod == SEEK_END)
- {
- NtQueryInformationFile(hFile,
- &IoStatusBlock,
- &FileStandard,
- sizeof(FILE_STANDARD_INFORMATION),
- FileStandardInformation);
- FilePosition.CurrentByteOffset.QuadPart =
- FileStandard.EndOfFile.QuadPart + Distance.QuadPart;
- }
- else if ( dwMoveMethod == SEEK_BEGIN )
- {
- FilePosition.CurrentByteOffset.QuadPart = Distance.QuadPart;
- }
-
-// DPRINT1("GOTO FILE OFFSET: %I64d\n",
FilePosition.CurrentByteOffset.QuadPart);
-
- errCode = NtSetInformationFile(hFile,
- &IoStatusBlock,
- &FilePosition,
- sizeof(FILE_POSITION_INFORMATION),
- FilePositionInformation);
- if (!NT_SUCCESS(errCode))
- {
- if (Status != NULL)
- {
- *Status = errCode;
- }
- return -1;
- }
-
- if (lpDistanceToMoveHigh != NULL)
- {
- *lpDistanceToMoveHigh =
FilePosition.CurrentByteOffset.u.HighPart;
- }
- if (Status != NULL)
- {
- *Status = STATUS_SUCCESS;
- }
- return FilePosition.CurrentByteOffset.u.LowPart;
-}
-
-
static BOOL
ConvertSystemTimeToFileTime(
- CONST SYSTEMTIME * lpSystemTime,
+ CONST SYSTEMTIME * lpSystemTime,
LPFILETIME lpFileTime)
{
TIME_FIELDS TimeFields;
LARGE_INTEGER liTime;
-
+
TimeFields.Year = lpSystemTime->wYear;
TimeFields.Month = lpSystemTime->wMonth;
TimeFields.Day = lpSystemTime->wDay;
@@ -295,7 +216,7 @@
TimeFields.Minute = lpSystemTime->wMinute;
TimeFields.Second = lpSystemTime->wSecond;
TimeFields.Milliseconds = lpSystemTime->wMilliseconds;
-
+
if (RtlTimeFieldsToTime(&TimeFields, &liTime))
{
lpFileTime->dwLowDateTime = liTime.u.LowPart;
@@ -315,21 +236,21 @@
PDOSTIME pdtime = (PDOSTIME) &wFatTime;
PDOSDATE pddate = (PDOSDATE) &wFatDate;
SYSTEMTIME SystemTime;
-
+
if (lpFileTime == NULL)
return FALSE;
-
+
SystemTime.wMilliseconds = 0;
SystemTime.wSecond = pdtime->Second;
SystemTime.wMinute = pdtime->Minute;
SystemTime.wHour = pdtime->Hour;
-
+
SystemTime.wDay = pddate->Day;
SystemTime.wMonth = pddate->Month;
SystemTime.wYear = 1980 + pddate->Year;
-
+
ConvertSystemTimeToFileTime(&SystemTime,lpFileTime);
-
+
return TRUE;
}
@@ -345,9 +266,9 @@
*/
{
ULONG i, j;
-
+
j = i = 0;
-
+
while (Path [i++])
{
if (Path[i - 1] == L'\\') j = i;
@@ -366,10 +287,10 @@
{
PWCHAR FileName;
DWORD i;
-
+
i = 0;
FileName = GetFileName(Path + i);
-
+
if ((FileName != (Path + i)) && (FileName [-1] == L'\\'))
FileName--;
if ((FileName == (Path + i)) && (FileName [0] == L'\\'))
@@ -379,7 +300,7 @@
static BOOL
-SetAttributesOnFile(PCFFILE_NODE File, HANDLE hFile)
+SetAttributesOnFile(PCFFILE File, HANDLE hFile)
/*
* FUNCTION: Sets attributes on a file
* ARGUMENTS:
@@ -393,19 +314,19 @@
NTSTATUS NtStatus;
ULONG Attributes = 0;
- if (File->File.Attributes & CAB_ATTRIB_READONLY)
+ if (File->Attributes & CAB_ATTRIB_READONLY)
Attributes |= FILE_ATTRIBUTE_READONLY;
- if (File->File.Attributes & CAB_ATTRIB_HIDDEN)
+ if (File->Attributes & CAB_ATTRIB_HIDDEN)
Attributes |= FILE_ATTRIBUTE_HIDDEN;
- if (File->File.Attributes & CAB_ATTRIB_SYSTEM)
+ if (File->Attributes & CAB_ATTRIB_SYSTEM)
Attributes |= FILE_ATTRIBUTE_SYSTEM;
- if (File->File.Attributes & CAB_ATTRIB_DIRECTORY)
+ if (File->Attributes & CAB_ATTRIB_DIRECTORY)
Attributes |= FILE_ATTRIBUTE_DIRECTORY;
- if (File->File.Attributes & CAB_ATTRIB_ARCHIVE)
+ if (File->Attributes & CAB_ATTRIB_ARCHIVE)
Attributes |= FILE_ATTRIBUTE_ARCHIVE;
NtStatus = NtQueryInformationFile(hFile,
@@ -435,698 +356,7 @@
return NT_SUCCESS(NtStatus);
}
-
static ULONG
-ReadBlock(PVOID Buffer,
- ULONG Size,
- PULONG BytesRead)
-/*
- * FUNCTION: Read a block of data from file
- * ARGUMENTS:
- * Buffer = Pointer to data buffer
- * Size = Length of data buffer
- * BytesRead = Pointer to ULONG that on return will contain
- * number of bytes read
- * RETURNS:
- * Status of operation
- */
-{
- IO_STATUS_BLOCK IoStatusBlock;
- NTSTATUS NtStatus;
-
- NtStatus = NtReadFile(FileHandle,
- NULL,
- NULL,
- NULL,
- &IoStatusBlock,
- Buffer,
- Size,
- NULL,
- NULL);
- if (!NT_SUCCESS(NtStatus))
- {
- DPRINT("ReadBlock for %d bytes failed (%x)\n", Size, NtStatus);
- *BytesRead = 0;
- return CAB_STATUS_INVALID_CAB;
- }
- *BytesRead = Size;
- return CAB_STATUS_SUCCESS;
-}
-
-
-static PCFFOLDER_NODE
-NewFolderNode()
-/*
- * FUNCTION: Creates a new folder node
- * RETURNS:
- * Pointer to node if there was enough free memory available,
otherwise NULL
- */
-{
- PCFFOLDER_NODE Node;
-
- Node = (PCFFOLDER_NODE)RtlAllocateHeap (ProcessHeap, 0,
sizeof(CFFOLDER_NODE));
- if (!Node)
- return NULL;
-
- RtlZeroMemory(Node, sizeof(CFFOLDER_NODE));
-
- Node->Folder.CompressionType = CAB_COMP_NONE;
-
- Node->Prev = FolderListTail;
-
- if (FolderListTail != NULL)
- {
- FolderListTail->Next = Node;
- }
- else
- {
- FolderListHead = Node;
- }
- FolderListTail = Node;
-
- return Node;
-}
-
-
-static PCFFILE_NODE
-NewFileNode()
-/*
- * FUNCTION: Creates a new file node
- * ARGUMENTS:
- * FolderNode = Pointer to folder node to bind file to
- * RETURNS:
- * Pointer to node if there was enough free memory available,
otherwise NULL
- */
-{
- PCFFILE_NODE Node;
-
- Node = (PCFFILE_NODE)RtlAllocateHeap (ProcessHeap, 0,
sizeof(CFFILE_NODE));
- if (!Node)
- return NULL;
-
- RtlZeroMemory(Node, sizeof(CFFILE_NODE));
-
- Node->Prev = FileListTail;
-
- if (FileListTail != NULL)
- {
- FileListTail->Next = Node;
- }
- else
- {
- FileListHead = Node;
- }
- FileListTail = Node;
-
- return Node;
-}
-
-
-static PCFDATA_NODE
-NewDataNode(PCFFOLDER_NODE FolderNode)
-/*
- * FUNCTION: Creates a new data block node
- * ARGUMENTS:
- * FolderNode = Pointer to folder node to bind data block to
- * RETURNS:
- * Pointer to node if there was enough free memory available,
otherwise NULL
- */
-{
- PCFDATA_NODE Node;
-
- Node = (PCFDATA_NODE)RtlAllocateHeap (ProcessHeap, 0,
sizeof(CFDATA_NODE));
- if (!Node)
- return NULL;
-
- RtlZeroMemory(Node, sizeof(CFDATA_NODE));
-
- Node->Prev = FolderNode->DataListTail;
-
- if (FolderNode->DataListTail != NULL)
- {
- FolderNode->DataListTail->Next = Node;
- }
- else
- {
- FolderNode->DataListHead = Node;
- }
- FolderNode->DataListTail = Node;
-
- return Node;
-}
-
-
-static VOID
-DestroyDataNodes(PCFFOLDER_NODE FolderNode)
-/*
- * FUNCTION: Destroys data block nodes bound to a folder node
- * ARGUMENTS:
- * FolderNode = Pointer to folder node
- */
-{
- PCFDATA_NODE PrevNode;
- PCFDATA_NODE NextNode;
-
- NextNode = FolderNode->DataListHead;
- while (NextNode != NULL)
- {
- PrevNode = NextNode->Next;
- RtlFreeHeap(ProcessHeap, 0, NextNode);
- NextNode = PrevNode;
- }
- FolderNode->DataListHead = NULL;
- FolderNode->DataListTail = NULL;
-}
-
-
-static VOID
-DestroyFileNodes()
-/*
- * FUNCTION: Destroys file nodes
- * ARGUMENTS:
- * FolderNode = Pointer to folder node
- */
-{
- PCFFILE_NODE PrevNode;
- PCFFILE_NODE NextNode;
-
- NextNode = FileListHead;
- while (NextNode != NULL)
- {
- PrevNode = NextNode->Next;
- if (NextNode->FileName)
- RtlFreeHeap(ProcessHeap, 0, NextNode->FileName);
- RtlFreeHeap(ProcessHeap, 0, NextNode);
- NextNode = PrevNode;
- }
- FileListHead = NULL;
- FileListTail = NULL;
-}
-
-
-#if 0
-static VOID
-DestroyDeletedFileNodes()
-/*
- * FUNCTION: Destroys file nodes that are marked for deletion
- */
-{
- PCFFILE_NODE CurNode;
- PCFFILE_NODE NextNode;
-
- CurNode = FileListHead;
- while (CurNode != NULL)
- {
- NextNode = CurNode->Next;
-
- if (CurNode->Delete)
- {
- if (CurNode->Prev != NULL)
- {
- CurNode->Prev->Next = CurNode->Next;
- }
- else
- {
- FileListHead = CurNode->Next;
- if (FileListHead)
- FileListHead->Prev = NULL;
- }
-
- if (CurNode->Next != NULL)
- {
- CurNode->Next->Prev = CurNode->Prev;
- }
- else
- {
- FileListTail = CurNode->Prev;
- if (FileListTail)
- FileListTail->Next = NULL;
- }
-
- DPRINT("Deleting file: '%S'\n", CurNode->FileName);
-
- if (CurNode->FileName)
- RtlFreeHeap(ProcessHeap, 0, CurNode->FileName);
- RtlFreeHeap(ProcessHeap, 0, CurNode);
- }
- CurNode = NextNode;
- }
-}
-#endif
-
-static VOID
-DestroyFolderNodes()
-/*
- * FUNCTION: Destroys folder nodes
- */
-{
- PCFFOLDER_NODE PrevNode;
- PCFFOLDER_NODE NextNode;
-
- NextNode = FolderListHead;
- while (NextNode != NULL)
- {
- PrevNode = NextNode->Next;
- DestroyDataNodes(NextNode);
- RtlFreeHeap(ProcessHeap, 0, NextNode);
- NextNode = PrevNode;
- }
- FolderListHead = NULL;
- FolderListTail = NULL;
-}
-
-#if 0
-static VOID
-DestroyDeletedFolderNodes()
-/*
- * FUNCTION: Destroys folder nodes that are marked for deletion
- */
-{
- PCFFOLDER_NODE CurNode;
- PCFFOLDER_NODE NextNode;
-
- CurNode = FolderListHead;
- while (CurNode != NULL)
- {
- NextNode = CurNode->Next;
-
- if (CurNode->Delete)
- {
- if (CurNode->Prev != NULL)
- {
- CurNode->Prev->Next = CurNode->Next;
- }
- else
- {
- FolderListHead = CurNode->Next;
- if (FolderListHead)
- FolderListHead->Prev = NULL;
- }
-
- if (CurNode->Next != NULL)
- {
- CurNode->Next->Prev = CurNode->Prev;
- }
- else
- {
- FolderListTail = CurNode->Prev;
- if (FolderListTail)
- FolderListTail->Next = NULL;
- }
-
- DestroyDataNodes(CurNode);
- RtlFreeHeap(ProcessHeap, 0, CurNode);
- }
- CurNode = NextNode;
- }
-}
-#endif
-
-static PCFFOLDER_NODE
-LocateFolderNode(ULONG Index)
-/*
- * FUNCTION: Locates a folder node
- * ARGUMENTS:
- * Index = Folder index
- * RETURNS:
- * Pointer to folder node or NULL if the folder node was not found
- */
-{
- PCFFOLDER_NODE Node;
-
- switch (Index)
- {
- case CAB_FILE_SPLIT:
- return FolderListTail;
-
- case CAB_FILE_CONTINUED:
- case CAB_FILE_PREV_NEXT:
- return FolderListHead;
- }
-
- Node = FolderListHead;
- while (Node != NULL)
- {
- if (Node->Index == Index)
- return Node;
- Node = Node->Next;
- }
- return NULL;
-}
-
-
-static ULONG
-GetAbsoluteOffset(PCFFILE_NODE File)
-/*
- * FUNCTION: Returns the absolute offset of a file
- * ARGUMENTS:
- * File = Pointer to CFFILE_NODE structure for file
- * RETURNS:
- * Status of operation
- */
-{
- PCFDATA_NODE Node;
-
- DPRINT("FileName '%S' FileOffset (0x%X) FileSize (%d).\n",
- (PWCHAR)File->FileName,
- (UINT)File->File.FileOffset,
- (UINT)File->File.FileSize);
-
- Node = CurrentFolderNode->DataListHead;
- while (Node != NULL)
- {
- DPRINT("GetAbsoluteOffset(): Comparing (0x%X, 0x%X) (%d).\n",
- (UINT)Node->UncompOffset,
- (UINT)Node->UncompOffset + Node->Data.UncompSize,
- (UINT)Node->Data.UncompSize);
-
- /* Node->Data.UncompSize will be 0 if the block is split
- (ie. it is the last block in this cabinet) */
- if ((Node->Data.UncompSize == 0) ||
- ((File->File.FileOffset >= Node->UncompOffset) &&
- (File->File.FileOffset < Node->UncompOffset +
- Node->Data.UncompSize)))
- {
- File->DataBlock = Node;
- return CAB_STATUS_SUCCESS;
- }
-
- Node = Node->Next;
- }
- return CAB_STATUS_INVALID_CAB;
-}
-
-
-static ULONG
-LocateFile(PWCHAR FileName,
- PCFFILE_NODE *File)
-/*
- * FUNCTION: Locates a file in the cabinet
- * ARGUMENTS:
- * FileName = Pointer to string with name of file to locate
- * File = Address of pointer to CFFILE_NODE structure to fill
- * RETURNS:
- * Status of operation
- * NOTES:
- * Current folder is set to the folder of the file
- */
-{
- PCFFILE_NODE Node;
- ULONG Status;
-
- DPRINT("FileName '%S'\n", FileName);
-
- Node = FileListHead;
- while (Node != NULL)
- {
- if (_wcsicmp(FileName, Node->FileName) == 0)
- {
- CurrentFolderNode =
LocateFolderNode(Node->File.FileControlID);
- if (!CurrentFolderNode)
- {
- DPRINT("Folder with index number (%d) not found.\n",
- (UINT)Node->File.FileControlID);
- return CAB_STATUS_INVALID_CAB;
- }
-
- if (Node->DataBlock == NULL)
- {
- Status = GetAbsoluteOffset(Node);
- }
- else
- Status = CAB_STATUS_SUCCESS;
- *File = Node;
- return Status;
- }
- Node = Node->Next;
- }
- return CAB_STATUS_NOFILE;
-}
-
-
-static ULONG
-ReadString(PWCHAR String, ULONG MaxLength)
-/*
- * FUNCTION: Reads a NULL-terminated string from the cabinet
- * ARGUMENTS:
- * String = Pointer to buffer to place string
- * MaxLength = Maximum length of string
- * RETURNS:
- * Status of operation
- */
-{
- NTSTATUS NtStatus;
- ULONG BytesRead;
- ULONG Offset;
- ULONG Status;
- ULONG Size;
- BOOL Found;
- CHAR buf[MAX_PATH];
- ANSI_STRING as;
- UNICODE_STRING us;
-
- Offset = 0;
- Found = FALSE;
- do
- {
- Size = ((Offset + 32) <= MaxLength)? 32 : MaxLength - Offset;
-
- if (Size == 0)
- {
- DPRINT("Too long a filename.\n");
- return CAB_STATUS_INVALID_CAB;
- }
-
- Status = ReadBlock((PCFDATA)&buf[Offset], Size, &BytesRead);
- if ((Status != CAB_STATUS_SUCCESS) || (BytesRead != Size))
- {
- DPRINT("Cannot read from file (%d).\n", (UINT)Status);
- return CAB_STATUS_INVALID_CAB;
- }
-
- for (Size = Offset; Size < Offset + BytesRead; Size++)
- {
- if (buf[Size] == '\0')
- {
- Found = TRUE;
- break;
- }
- }
-
- Offset += BytesRead;
- } while (!Found);
-
- /* Back up some bytes */
- Size = (BytesRead - Size) - 1;
- SeekInFile(FileHandle, -(LONG)Size, NULL, SEEK_CURRENT, &NtStatus);
- if (!NT_SUCCESS(NtStatus))
- {
- DPRINT("SeekInFile() failed (%x).\n", NtStatus);
- return CAB_STATUS_INVALID_CAB;
- }
-
- RtlInitAnsiString(&as, buf);
- us.Buffer = String;
- us.MaximumLength = MaxLength * sizeof(WCHAR);
- us.Length = 0;
-
- RtlAnsiStringToUnicodeString(&us, &as, FALSE);
-
- return CAB_STATUS_SUCCESS;
-}
-
-
-static ULONG
-ReadFileTable(VOID)
-/*
- * FUNCTION: Reads the file table from the cabinet file
- * RETURNS:
- * Status of operation
- */
-{
- ULONG i;
- ULONG Status;
- ULONG BytesRead;
- PCFFILE_NODE File;
- NTSTATUS NtStatus;
-
- DPRINT("Reading file table at absolute offset (0x%X).\n",
- CABHeader.FileTableOffset);
-
- /* Seek to file table */
- SeekInFile(FileHandle, CABHeader.FileTableOffset, NULL, SEEK_BEGIN,
&NtStatus);
- if (!NT_SUCCESS(NtStatus))
- {
- DPRINT("SeekInFile() failed (%x).\n", NtStatus);
- return CAB_STATUS_INVALID_CAB;
- }
-
- for (i = 0; i < CABHeader.FileCount; i++)
- {
- File = NewFileNode();
- if (!File)
- {
- DPRINT("Insufficient memory.\n");
- return CAB_STATUS_NOMEMORY;
- }
-
- if ((Status = ReadBlock(&File->File, sizeof(CFFILE),
- &BytesRead)) != CAB_STATUS_SUCCESS) {
- DPRINT("Cannot read from file (%d).\n", (UINT)Status);
- return CAB_STATUS_INVALID_CAB;
- }
-
- File->FileName = (PWCHAR)RtlAllocateHeap(ProcessHeap, 0, MAX_PATH
* sizeof(WCHAR));
- if (!File->FileName)
- {
- DPRINT("Insufficient memory.\n");
- return CAB_STATUS_NOMEMORY;
- }
-
- /* Read file name */
- Status = ReadString(File->FileName, MAX_PATH);
- if (Status != CAB_STATUS_SUCCESS)
- return Status;
-
- DPRINT("Found file '%S' at uncompressed offset (0x%X). Size (%d
bytes) ControlId (0x%X).\n",
- (PWCHAR)File->FileName,
- (UINT)File->File.FileOffset,
- (UINT)File->File.FileSize,
- (UINT)File->File.FileControlID);
- }
- return CAB_STATUS_SUCCESS;
-}
-
-
-static ULONG
-ReadDataBlocks(PCFFOLDER_NODE FolderNode)
-/*
- * FUNCTION: Reads all CFDATA blocks for a folder from the cabinet file
- * ARGUMENTS:
- * FolderNode = Pointer to CFFOLDER_NODE structure for folder
- * RETURNS:
- * Status of operation
- */
-{
- ULONG AbsoluteOffset;
- ULONG UncompOffset;
- PCFDATA_NODE Node;
- NTSTATUS NtStatus;
- ULONG BytesRead;
- ULONG Status;
- ULONG i;
-
- DPRINT("Reading data blocks for folder (%d) at absolute offset
(0x%X).\n",
- FolderNode->Index, FolderNode->Folder.DataOffset);
-
- AbsoluteOffset = FolderNode->Folder.DataOffset;
- UncompOffset = FolderNode->UncompOffset;
-
- for (i = 0; i < FolderNode->Folder.DataBlockCount; i++)
- {
- Node = NewDataNode(FolderNode);
- if (!Node)
- {
- DPRINT("Insufficient memory.\n");
- return CAB_STATUS_NOMEMORY;
- }
-
- /* Seek to data block */
- SeekInFile(FileHandle, AbsoluteOffset, NULL, SEEK_BEGIN,
&NtStatus);
[truncated at 1000 lines; 1631 more skipped]
7064
days inactive
7064
days old
0 comments
1 participants
participants (1)
-
Phreak＠svn.reactos.com