You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1840 lines
46 KiB
1840 lines
46 KiB
![]()
4 years ago
|
/* 7zArcIn.c -- 7z Input functions
|
||
|
2014-06-16 : Igor Pavlov : Public domain */
|
||
|
|
||
|
#include "Precomp.h"
|
||
|
|
||
|
#include <string.h>
|
||
|
|
||
|
#include "7z.h"
|
||
|
#include "7zBuf.h"
|
||
|
#include "7zCrc.h"
|
||
|
#include "CpuArch.h"
|
||
|
|
||
|
#define MY_ALLOC(T, p, size, alloc) { if ((size) == 0) p = 0; else \
|
||
|
if ((p = (T *)IAlloc_Alloc(alloc, (size) * sizeof(T))) == 0) return SZ_ERROR_MEM; }
|
||
|
|
||
|
#define k7zMajorVersion 0
|
||
|
|
||
|
enum EIdEnum
|
||
|
{
|
||
|
k7zIdEnd,
|
||
|
k7zIdHeader,
|
||
|
k7zIdArchiveProperties,
|
||
|
k7zIdAdditionalStreamsInfo,
|
||
|
k7zIdMainStreamsInfo,
|
||
|
k7zIdFilesInfo,
|
||
|
k7zIdPackInfo,
|
||
|
k7zIdUnpackInfo,
|
||
|
k7zIdSubStreamsInfo,
|
||
|
k7zIdSize,
|
||
|
k7zIdCRC,
|
||
|
k7zIdFolder,
|
||
|
k7zIdCodersUnpackSize,
|
||
|
k7zIdNumUnpackStream,
|
||
|
k7zIdEmptyStream,
|
||
|
k7zIdEmptyFile,
|
||
|
k7zIdAnti,
|
||
|
k7zIdName,
|
||
|
k7zIdCTime,
|
||
|
k7zIdATime,
|
||
|
k7zIdMTime,
|
||
|
k7zIdWinAttrib,
|
||
|
k7zIdComment,
|
||
|
k7zIdEncodedHeader,
|
||
|
k7zIdStartPos,
|
||
|
k7zIdDummy
|
||
|
// k7zNtSecure,
|
||
|
// k7zParent,
|
||
|
// k7zIsReal
|
||
|
};
|
||
|
|
||
|
Byte k7zSignature[k7zSignatureSize] = {'7', 'z', 0xBC, 0xAF, 0x27, 0x1C};
|
||
|
|
||
|
#define NUM_FOLDER_CODERS_MAX 32
|
||
|
#define NUM_CODER_STREAMS_MAX 32
|
||
|
|
||
|
/*
|
||
|
static int SzFolder_FindBindPairForInStream(const CSzFolder *p, UInt32 inStreamIndex)
|
||
|
{
|
||
|
UInt32 i;
|
||
|
for (i = 0; i < p->NumBindPairs; i++)
|
||
|
if (p->BindPairs[i].InIndex == inStreamIndex)
|
||
|
return i;
|
||
|
return -1;
|
||
|
}
|
||
|
*/
|
||
|
|
||
|
#define SzBitUi32s_Init(p) { (p)->Defs = 0; (p)->Vals = 0; }
|
||
|
|
||
|
static SRes SzBitUi32s_Alloc(CSzBitUi32s *p, size_t num, ISzAlloc *alloc)
|
||
|
{
|
||
|
MY_ALLOC(Byte, p->Defs, (num + 7) >> 3, alloc);
|
||
|
MY_ALLOC(UInt32, p->Vals, num, alloc);
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
void SzBitUi32s_Free(CSzBitUi32s *p, ISzAlloc *alloc)
|
||
|
{
|
||
|
IAlloc_Free(alloc, p->Defs); p->Defs = 0;
|
||
|
IAlloc_Free(alloc, p->Vals); p->Vals = 0;
|
||
|
}
|
||
|
|
||
|
#define SzBitUi64s_Init(p) { (p)->Defs = 0; (p)->Vals = 0; }
|
||
|
|
||
|
void SzBitUi64s_Free(CSzBitUi64s *p, ISzAlloc *alloc)
|
||
|
{
|
||
|
IAlloc_Free(alloc, p->Defs); p->Defs = 0;
|
||
|
IAlloc_Free(alloc, p->Vals); p->Vals = 0;
|
||
|
}
|
||
|
|
||
|
static void SzAr_Init(CSzAr *p)
|
||
|
{
|
||
|
p->NumPackStreams = 0;
|
||
|
p->NumFolders = 0;
|
||
|
p->PackPositions = 0;
|
||
|
SzBitUi32s_Init(&p->FolderCRCs);
|
||
|
// p->Folders = 0;
|
||
|
p->FoCodersOffsets = 0;
|
||
|
p->FoSizesOffsets = 0;
|
||
|
p->FoStartPackStreamIndex = 0;
|
||
|
|
||
|
p->CodersData = 0;
|
||
|
// p->CoderUnpackSizes = 0;
|
||
|
p->UnpackSizesData = 0;
|
||
|
}
|
||
|
|
||
|
static void SzAr_Free(CSzAr *p, ISzAlloc *alloc)
|
||
|
{
|
||
|
IAlloc_Free(alloc, p->UnpackSizesData);
|
||
|
IAlloc_Free(alloc, p->CodersData);
|
||
|
// IAlloc_Free(alloc, p->CoderUnpackSizes);
|
||
|
|
||
|
IAlloc_Free(alloc, p->PackPositions);
|
||
|
|
||
|
// IAlloc_Free(alloc, p->Folders);
|
||
|
IAlloc_Free(alloc, p->FoCodersOffsets);
|
||
|
IAlloc_Free(alloc, p->FoSizesOffsets);
|
||
|
IAlloc_Free(alloc, p->FoStartPackStreamIndex);
|
||
|
|
||
|
SzBitUi32s_Free(&p->FolderCRCs, alloc);
|
||
|
|
||
|
SzAr_Init(p);
|
||
|
}
|
||
|
|
||
|
|
||
|
void SzArEx_Init(CSzArEx *p)
|
||
|
{
|
||
|
SzAr_Init(&p->db);
|
||
|
p->NumFiles = 0;
|
||
|
p->dataPos = 0;
|
||
|
// p->Files = 0;
|
||
|
p->UnpackPositions = 0;
|
||
|
// p->IsEmptyFiles = 0;
|
||
|
p->IsDirs = 0;
|
||
|
// p->FolderStartPackStreamIndex = 0;
|
||
|
// p->PackStreamStartPositions = 0;
|
||
|
p->FolderStartFileIndex = 0;
|
||
|
p->FileIndexToFolderIndexMap = 0;
|
||
|
p->FileNameOffsets = 0;
|
||
|
p->FileNames = 0;
|
||
|
SzBitUi32s_Init(&p->CRCs);
|
||
|
SzBitUi32s_Init(&p->Attribs);
|
||
|
// SzBitUi32s_Init(&p->Parents);
|
||
|
SzBitUi64s_Init(&p->MTime);
|
||
|
SzBitUi64s_Init(&p->CTime);
|
||
|
}
|
||
|
|
||
|
void SzArEx_Free(CSzArEx *p, ISzAlloc *alloc)
|
||
|
{
|
||
|
// IAlloc_Free(alloc, p->FolderStartPackStreamIndex);
|
||
|
// IAlloc_Free(alloc, p->PackStreamStartPositions);
|
||
|
IAlloc_Free(alloc, p->FolderStartFileIndex);
|
||
|
IAlloc_Free(alloc, p->FileIndexToFolderIndexMap);
|
||
|
|
||
|
IAlloc_Free(alloc, p->FileNameOffsets);
|
||
|
IAlloc_Free(alloc, p->FileNames);
|
||
|
|
||
|
SzBitUi64s_Free(&p->CTime, alloc);
|
||
|
SzBitUi64s_Free(&p->MTime, alloc);
|
||
|
SzBitUi32s_Free(&p->CRCs, alloc);
|
||
|
// SzBitUi32s_Free(&p->Parents, alloc);
|
||
|
SzBitUi32s_Free(&p->Attribs, alloc);
|
||
|
IAlloc_Free(alloc, p->IsDirs);
|
||
|
// IAlloc_Free(alloc, p->IsEmptyFiles);
|
||
|
IAlloc_Free(alloc, p->UnpackPositions);
|
||
|
// IAlloc_Free(alloc, p->Files);
|
||
|
|
||
|
SzAr_Free(&p->db, alloc);
|
||
|
SzArEx_Init(p);
|
||
|
}
|
||
|
|
||
|
static int TestSignatureCandidate(Byte *testBytes)
|
||
|
{
|
||
|
size_t i;
|
||
|
for (i = 0; i < k7zSignatureSize; i++)
|
||
|
if (testBytes[i] != k7zSignature[i])
|
||
|
return 0;
|
||
|
return 1;
|
||
|
}
|
||
|
|
||
|
#define SzData_Clear(p) { (p)->Data = 0; (p)->Size = 0; }
|
||
|
|
||
|
static SRes SzReadByte(CSzData *sd, Byte *b)
|
||
|
{
|
||
|
if (sd->Size == 0)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
sd->Size--;
|
||
|
*b = *sd->Data++;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
#define SZ_READ_BYTE_SD(_sd_, dest) if ((_sd_)->Size == 0) return SZ_ERROR_ARCHIVE; (_sd_)->Size--; dest = *(_sd_)->Data++;
|
||
|
#define SZ_READ_BYTE(dest) SZ_READ_BYTE_SD(sd, dest)
|
||
|
#define SZ_READ_BYTE_2(dest) if (sd.Size == 0) return SZ_ERROR_ARCHIVE; sd.Size--; dest = *sd.Data++;
|
||
|
|
||
|
#define SKIP_DATA(sd, size) { sd->Size -= (size_t)(size); sd->Data += (size_t)(size); }
|
||
|
#define SKIP_DATA2(sd, size) { sd.Size -= (size_t)(size); sd.Data += (size_t)(size); }
|
||
|
|
||
|
#define SZ_READ_32(dest) if (sd.Size < 4) return SZ_ERROR_ARCHIVE; \
|
||
|
dest = GetUi32(sd.Data); SKIP_DATA2(sd, 4);
|
||
|
|
||
|
static MY_NO_INLINE SRes ReadNumber(CSzData *sd, UInt64 *value)
|
||
|
{
|
||
|
Byte firstByte, mask;
|
||
|
unsigned i;
|
||
|
UInt32 v;
|
||
|
|
||
|
SZ_READ_BYTE(firstByte);
|
||
|
if ((firstByte & 0x80) == 0)
|
||
|
{
|
||
|
*value = firstByte;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
SZ_READ_BYTE(v);
|
||
|
if ((firstByte & 0x40) == 0)
|
||
|
{
|
||
|
*value = (((UInt32)firstByte & 0x3F) << 8) | v;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
SZ_READ_BYTE(mask);
|
||
|
*value = v | ((UInt32)mask << 8);
|
||
|
mask = 0x20;
|
||
|
for (i = 2; i < 8; i++)
|
||
|
{
|
||
|
Byte b;
|
||
|
if ((firstByte & mask) == 0)
|
||
|
{
|
||
|
UInt64 highPart = firstByte & (mask - 1);
|
||
|
*value |= (highPart << (8 * i));
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
SZ_READ_BYTE(b);
|
||
|
*value |= ((UInt64)b << (8 * i));
|
||
|
mask >>= 1;
|
||
|
}
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
static MY_NO_INLINE const Byte *SzReadNumbers(const Byte *data, const Byte *dataLim, UInt64 *values, UInt32 num)
|
||
|
{
|
||
|
for (; num != 0; num--)
|
||
|
{
|
||
|
Byte firstByte;
|
||
|
Byte mask;
|
||
|
|
||
|
unsigned i;
|
||
|
UInt32 v;
|
||
|
UInt64 value;
|
||
|
|
||
|
if (data == dataLim)
|
||
|
return NULL;
|
||
|
firstByte = *data++;
|
||
|
|
||
|
if ((firstByte & 0x80) == 0)
|
||
|
{
|
||
|
*values++ = firstByte;
|
||
|
continue;
|
||
|
}
|
||
|
if (data == dataLim)
|
||
|
return NULL;
|
||
|
v = *data++;
|
||
|
if ((firstByte & 0x40) == 0)
|
||
|
{
|
||
|
*values++ = (((UInt32)firstByte & 0x3F) << 8) | v;
|
||
|
continue;
|
||
|
}
|
||
|
if (data == dataLim)
|
||
|
return NULL;
|
||
|
value = v | ((UInt32)*data++ << 8);
|
||
|
mask = 0x20;
|
||
|
for (i = 2; i < 8; i++)
|
||
|
{
|
||
|
if ((firstByte & mask) == 0)
|
||
|
{
|
||
|
UInt64 highPart = firstByte & (mask - 1);
|
||
|
value |= (highPart << (8 * i));
|
||
|
break;
|
||
|
}
|
||
|
if (data == dataLim)
|
||
|
return NULL;
|
||
|
value |= ((UInt64)*data++ << (8 * i));
|
||
|
mask >>= 1;
|
||
|
}
|
||
|
*values++ = value;
|
||
|
}
|
||
|
return data;
|
||
|
}
|
||
|
*/
|
||
|
|
||
|
static MY_NO_INLINE SRes SzReadNumber32(CSzData *sd, UInt32 *value)
|
||
|
{
|
||
|
Byte firstByte;
|
||
|
UInt64 value64;
|
||
|
if (sd->Size == 0)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
firstByte = *sd->Data;
|
||
|
if ((firstByte & 0x80) == 0)
|
||
|
{
|
||
|
*value = firstByte;
|
||
|
sd->Data++;
|
||
|
sd->Size--;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
RINOK(ReadNumber(sd, &value64));
|
||
|
if (value64 >= (UInt32)0x80000000 - 1)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
if (value64 >= ((UInt64)(1) << ((sizeof(size_t) - 1) * 8 + 4)))
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
*value = (UInt32)value64;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
#define ReadID(sd, value) ReadNumber(sd, value)
|
||
|
|
||
|
static SRes SkipData(CSzData *sd)
|
||
|
{
|
||
|
UInt64 size;
|
||
|
RINOK(ReadNumber(sd, &size));
|
||
|
if (size > sd->Size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
SKIP_DATA(sd, size);
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
static SRes WaitId(CSzData *sd, UInt64 id)
|
||
|
{
|
||
|
for (;;)
|
||
|
{
|
||
|
UInt64 type;
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
if (type == id)
|
||
|
return SZ_OK;
|
||
|
if (type == k7zIdEnd)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
RINOK(SkipData(sd));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static SRes RememberBitVector(CSzData *sd, UInt32 numItems, const Byte **v)
|
||
|
{
|
||
|
UInt32 numBytes = (numItems + 7) >> 3;
|
||
|
if (numBytes > sd->Size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
*v = sd->Data;
|
||
|
SKIP_DATA(sd, numBytes);
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
static UInt32 CountDefinedBits(const Byte *bits, UInt32 numItems)
|
||
|
{
|
||
|
Byte b = 0;
|
||
|
unsigned m = 0;
|
||
|
UInt32 sum = 0;
|
||
|
for (; numItems != 0; numItems--)
|
||
|
{
|
||
|
if (m == 0)
|
||
|
{
|
||
|
b = *bits++;
|
||
|
m = 8;
|
||
|
}
|
||
|
m--;
|
||
|
sum += ((b >> m) & 1);
|
||
|
}
|
||
|
return sum ;
|
||
|
}
|
||
|
|
||
|
static MY_NO_INLINE SRes ReadBitVector(CSzData *sd, UInt32 numItems, Byte **v, ISzAlloc *alloc)
|
||
|
{
|
||
|
Byte allAreDefined;
|
||
|
UInt32 i;
|
||
|
Byte *v2;
|
||
|
UInt32 numBytes = (numItems + 7) >> 3;
|
||
|
RINOK(SzReadByte(sd, &allAreDefined));
|
||
|
if (allAreDefined == 0)
|
||
|
{
|
||
|
if (numBytes > sd->Size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
MY_ALLOC(Byte, *v, numBytes, alloc);
|
||
|
memcpy(*v, sd->Data, numBytes);
|
||
|
SKIP_DATA(sd, numBytes);
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
MY_ALLOC(Byte, *v, numBytes, alloc);
|
||
|
v2 = *v;
|
||
|
for (i = 0; i < numBytes; i++)
|
||
|
v2[i] = 0xFF;
|
||
|
{
|
||
|
unsigned numBits = (unsigned)numItems & 7;
|
||
|
if (numBits != 0)
|
||
|
v2[numBytes - 1] = (Byte)((((UInt32)1 << numBits) - 1) << (8 - numBits));
|
||
|
}
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
static MY_NO_INLINE SRes ReadUi32s(CSzData *sd2, UInt32 numItems, CSzBitUi32s *crcs, ISzAlloc *alloc)
|
||
|
{
|
||
|
UInt32 i;
|
||
|
CSzData sd;
|
||
|
UInt32 *vals;
|
||
|
const Byte *defs;
|
||
|
MY_ALLOC(UInt32, crcs->Vals, numItems, alloc);
|
||
|
sd = *sd2;
|
||
|
defs = crcs->Defs;
|
||
|
vals = crcs->Vals;
|
||
|
for (i = 0; i < numItems; i++)
|
||
|
if (SzBitArray_Check(defs, i))
|
||
|
{
|
||
|
SZ_READ_32(vals[i]);
|
||
|
}
|
||
|
else
|
||
|
vals[i] = 0;
|
||
|
*sd2 = sd;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
static SRes ReadBitUi32s(CSzData *sd, UInt32 numItems, CSzBitUi32s *crcs, ISzAlloc *alloc)
|
||
|
{
|
||
|
SzBitUi32s_Free(crcs, alloc);
|
||
|
RINOK(ReadBitVector(sd, numItems, &crcs->Defs, alloc));
|
||
|
return ReadUi32s(sd, numItems, crcs, alloc);
|
||
|
}
|
||
|
|
||
|
static SRes SkipBitUi32s(CSzData *sd, UInt32 numItems)
|
||
|
{
|
||
|
Byte allAreDefined;
|
||
|
UInt32 numDefined = numItems;
|
||
|
RINOK(SzReadByte(sd, &allAreDefined));
|
||
|
if (!allAreDefined)
|
||
|
{
|
||
|
size_t numBytes = (numItems + 7) >> 3;
|
||
|
if (numBytes > sd->Size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
numDefined = CountDefinedBits(sd->Data, numItems);
|
||
|
SKIP_DATA(sd, numBytes);
|
||
|
}
|
||
|
if (numDefined > (sd->Size >> 2))
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
SKIP_DATA(sd, (size_t)numDefined * 4);
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
static SRes ReadPackInfo(CSzAr *p, CSzData *sd, ISzAlloc *alloc)
|
||
|
{
|
||
|
RINOK(SzReadNumber32(sd, &p->NumPackStreams));
|
||
|
|
||
|
RINOK(WaitId(sd, k7zIdSize));
|
||
|
MY_ALLOC(UInt64, p->PackPositions, (size_t)p->NumPackStreams + 1, alloc);
|
||
|
{
|
||
|
UInt64 sum = 0;
|
||
|
UInt32 i;
|
||
|
UInt32 numPackStreams = p->NumPackStreams;
|
||
|
for (i = 0; i < numPackStreams; i++)
|
||
|
{
|
||
|
UInt64 packSize;
|
||
|
p->PackPositions[i] = sum;
|
||
|
RINOK(ReadNumber(sd, &packSize));
|
||
|
sum += packSize;
|
||
|
if (sum < packSize)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
}
|
||
|
p->PackPositions[i] = sum;
|
||
|
}
|
||
|
|
||
|
for (;;)
|
||
|
{
|
||
|
UInt64 type;
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
if (type == k7zIdEnd)
|
||
|
return SZ_OK;
|
||
|
if (type == k7zIdCRC)
|
||
|
{
|
||
|
/* CRC of packed streams is unused now */
|
||
|
RINOK(SkipBitUi32s(sd, p->NumPackStreams));
|
||
|
continue;
|
||
|
}
|
||
|
RINOK(SkipData(sd));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
static SRes SzReadSwitch(CSzData *sd)
|
||
|
{
|
||
|
Byte external;
|
||
|
RINOK(SzReadByte(sd, &external));
|
||
|
return (external == 0) ? SZ_OK: SZ_ERROR_UNSUPPORTED;
|
||
|
}
|
||
|
*/
|
||
|
|
||
|
#define SZ_NUM_IN_STREAMS_IN_FOLDER_MAX 16
|
||
|
|
||
|
SRes SzGetNextFolderItem(CSzFolder *f, CSzData *sd, CSzData *sdSizes)
|
||
|
{
|
||
|
UInt32 numCoders, numBindPairs, numPackStreams, i;
|
||
|
UInt32 numInStreams = 0, numOutStreams = 0;
|
||
|
const Byte *dataStart = sd->Data;
|
||
|
Byte inStreamUsed[SZ_NUM_IN_STREAMS_IN_FOLDER_MAX];
|
||
|
|
||
|
RINOK(SzReadNumber32(sd, &numCoders));
|
||
|
if (numCoders > SZ_NUM_CODERS_IN_FOLDER_MAX)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
f->NumCoders = numCoders;
|
||
|
|
||
|
for (i = 0; i < numCoders; i++)
|
||
|
{
|
||
|
Byte mainByte;
|
||
|
CSzCoderInfo *coder = f->Coders + i;
|
||
|
unsigned idSize, j;
|
||
|
UInt64 id;
|
||
|
RINOK(SzReadByte(sd, &mainByte));
|
||
|
if ((mainByte & 0xC0) != 0)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
idSize = (unsigned)(mainByte & 0xF);
|
||
|
if (idSize > sizeof(id))
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
if (idSize > sd->Size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
id = 0;
|
||
|
for (j = 0; j < idSize; j++)
|
||
|
{
|
||
|
id = ((id << 8) | *sd->Data);
|
||
|
sd->Data++;
|
||
|
sd->Size--;
|
||
|
}
|
||
|
if (id > (UInt32)0xFFFFFFFF)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
coder->MethodID = (UInt32)id;
|
||
|
|
||
|
coder->NumInStreams = 1;
|
||
|
coder->NumOutStreams = 1;
|
||
|
coder->PropsOffset = 0;
|
||
|
coder->PropsSize = 0;
|
||
|
|
||
|
if ((mainByte & 0x10) != 0)
|
||
|
{
|
||
|
UInt32 numStreams;
|
||
|
RINOK(SzReadNumber32(sd, &numStreams));
|
||
|
if (numStreams > NUM_CODER_STREAMS_MAX)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
coder->NumInStreams = (Byte)numStreams;
|
||
|
RINOK(SzReadNumber32(sd, &numStreams));
|
||
|
if (numStreams > NUM_CODER_STREAMS_MAX)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
coder->NumOutStreams = (Byte)numStreams;
|
||
|
}
|
||
|
if ((mainByte & 0x20) != 0)
|
||
|
{
|
||
|
UInt32 propsSize = 0;
|
||
|
RINOK(SzReadNumber32(sd, &propsSize));
|
||
|
if (propsSize >= 0x40)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
if (propsSize > sd->Size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
coder->PropsOffset = sd->Data - dataStart;
|
||
|
coder->PropsSize = (Byte)propsSize;
|
||
|
sd->Data += (size_t)propsSize;
|
||
|
sd->Size -= (size_t)propsSize;
|
||
|
}
|
||
|
numInStreams += coder->NumInStreams;
|
||
|
numOutStreams += coder->NumOutStreams;
|
||
|
}
|
||
|
|
||
|
if (numOutStreams == 0)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
|
||
|
f->NumBindPairs = numBindPairs = numOutStreams - 1;
|
||
|
if (numInStreams < numBindPairs)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
if (numInStreams > SZ_NUM_IN_STREAMS_IN_FOLDER_MAX)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
f->MainOutStream = 0;
|
||
|
f->NumPackStreams = numPackStreams = numInStreams - numBindPairs;
|
||
|
if (numPackStreams > SZ_NUM_PACK_STREAMS_IN_FOLDER_MAX)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
for (i = 0; i < numInStreams; i++)
|
||
|
inStreamUsed[i] = False;
|
||
|
if (numBindPairs != 0)
|
||
|
{
|
||
|
Byte outStreamUsed[SZ_NUM_CODERS_OUT_STREAMS_IN_FOLDER_MAX];
|
||
|
|
||
|
if (numBindPairs > SZ_NUM_BINDS_IN_FOLDER_MAX)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
|
||
|
for (i = 0; i < numOutStreams; i++)
|
||
|
outStreamUsed[i] = False;
|
||
|
|
||
|
for (i = 0; i < numBindPairs; i++)
|
||
|
{
|
||
|
CSzBindPair *bp = f->BindPairs + i;
|
||
|
RINOK(SzReadNumber32(sd, &bp->InIndex));
|
||
|
if (bp->InIndex >= numInStreams)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
inStreamUsed[bp->InIndex] = True;
|
||
|
RINOK(SzReadNumber32(sd, &bp->OutIndex));
|
||
|
if (bp->OutIndex >= numInStreams)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
outStreamUsed[bp->OutIndex] = True;
|
||
|
}
|
||
|
for (i = 0; i < numOutStreams; i++)
|
||
|
if (!outStreamUsed[i])
|
||
|
{
|
||
|
f->MainOutStream = i;
|
||
|
break;
|
||
|
}
|
||
|
if (i == numOutStreams)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
}
|
||
|
|
||
|
if (numPackStreams == 1)
|
||
|
{
|
||
|
for (i = 0; i < numInStreams; i++)
|
||
|
if (!inStreamUsed[i])
|
||
|
break;
|
||
|
if (i == numInStreams)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
f->PackStreams[0] = i;
|
||
|
}
|
||
|
else
|
||
|
for (i = 0; i < numPackStreams; i++)
|
||
|
{
|
||
|
RINOK(SzReadNumber32(sd, f->PackStreams + i));
|
||
|
}
|
||
|
|
||
|
for (i = 0; i < numOutStreams; i++)
|
||
|
{
|
||
|
RINOK(ReadNumber(sdSizes, f->CodersUnpackSizes + i));
|
||
|
}
|
||
|
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
static MY_NO_INLINE SRes SkipNumbers(CSzData *sd2, UInt32 num)
|
||
|
{
|
||
|
CSzData sd;
|
||
|
sd = *sd2;
|
||
|
for (; num != 0; num--)
|
||
|
{
|
||
|
Byte firstByte, mask;
|
||
|
unsigned i;
|
||
|
SZ_READ_BYTE_2(firstByte);
|
||
|
if ((firstByte & 0x80) == 0)
|
||
|
continue;
|
||
|
if ((firstByte & 0x40) == 0)
|
||
|
{
|
||
|
if (sd.Size == 0)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
sd.Size--;
|
||
|
sd.Data++;
|
||
|
continue;
|
||
|
}
|
||
|
mask = 0x20;
|
||
|
for (i = 2; i < 8 && (firstByte & mask) != 0; i++)
|
||
|
mask >>= 1;
|
||
|
if (i > sd.Size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
SKIP_DATA2(sd, i);
|
||
|
}
|
||
|
*sd2 = sd;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
#define k_InStreamUsed_MAX 64
|
||
|
#define k_OutStreamUsed_MAX 64
|
||
|
|
||
|
static SRes ReadUnpackInfo(CSzAr *p,
|
||
|
CSzData *sd2,
|
||
|
UInt32 numFoldersMax, const CBuf *tempBufs, UInt32 numTempBufs,
|
||
|
ISzAlloc *alloc)
|
||
|
{
|
||
|
CSzData sd;
|
||
|
Byte inStreamUsed[k_InStreamUsed_MAX];
|
||
|
Byte outStreamUsed[k_OutStreamUsed_MAX];
|
||
|
UInt32 fo, numFolders, numCodersOutStreams, packStreamIndex;
|
||
|
const Byte *startBufPtr;
|
||
|
Byte external;
|
||
|
|
||
|
RINOK(WaitId(sd2, k7zIdFolder));
|
||
|
RINOK(SzReadNumber32(sd2, &numFolders));
|
||
|
if (p->NumFolders > numFoldersMax)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
p->NumFolders = numFolders;
|
||
|
|
||
|
SZ_READ_BYTE_SD(sd2, external);
|
||
|
if (external == 0)
|
||
|
sd = *sd2;
|
||
|
else
|
||
|
{
|
||
|
UInt32 index;
|
||
|
SzReadNumber32(sd2, &index);
|
||
|
if (index >= numTempBufs)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
sd.Data = tempBufs[index].data;
|
||
|
sd.Size = tempBufs[index].size;
|
||
|
}
|
||
|
|
||
|
MY_ALLOC(size_t, p->FoCodersOffsets, (size_t)numFolders + 1, alloc);
|
||
|
MY_ALLOC(size_t, p->FoSizesOffsets, (size_t)numFolders + 1, alloc);
|
||
|
MY_ALLOC(UInt32, p->FoStartPackStreamIndex, (size_t)numFolders + 1, alloc);
|
||
|
|
||
|
startBufPtr = sd.Data;
|
||
|
|
||
|
packStreamIndex = 0;
|
||
|
numCodersOutStreams = 0;
|
||
|
|
||
|
for (fo = 0; fo < numFolders; fo++)
|
||
|
{
|
||
|
UInt32 numCoders, ci, numInStreams = 0, numOutStreams = 0;
|
||
|
|
||
|
p->FoCodersOffsets[fo] = sd.Data - startBufPtr;
|
||
|
RINOK(SzReadNumber32(&sd, &numCoders));
|
||
|
if (numCoders > NUM_FOLDER_CODERS_MAX)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
|
||
|
for (ci = 0; ci < numCoders; ci++)
|
||
|
{
|
||
|
Byte mainByte;
|
||
|
unsigned idSize;
|
||
|
UInt32 coderInStreams, coderOutStreams;
|
||
|
|
||
|
SZ_READ_BYTE_2(mainByte);
|
||
|
if ((mainByte & 0xC0) != 0)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
idSize = (mainByte & 0xF);
|
||
|
if (idSize > 8)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
if (idSize > sd.Size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
SKIP_DATA2(sd, idSize);
|
||
|
|
||
|
coderInStreams = 1;
|
||
|
coderOutStreams = 1;
|
||
|
if ((mainByte & 0x10) != 0)
|
||
|
{
|
||
|
RINOK(SzReadNumber32(&sd, &coderInStreams));
|
||
|
RINOK(SzReadNumber32(&sd, &coderOutStreams));
|
||
|
if (coderInStreams > NUM_CODER_STREAMS_MAX ||
|
||
|
coderOutStreams > NUM_CODER_STREAMS_MAX)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
}
|
||
|
numInStreams += coderInStreams;
|
||
|
numOutStreams += coderOutStreams;
|
||
|
if ((mainByte & 0x20) != 0)
|
||
|
{
|
||
|
UInt32 propsSize;
|
||
|
RINOK(SzReadNumber32(&sd, &propsSize));
|
||
|
if (propsSize > sd.Size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
SKIP_DATA2(sd, propsSize);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
{
|
||
|
UInt32 indexOfMainStream = 0;
|
||
|
UInt32 numPackStreams = 1;
|
||
|
if (numOutStreams != 1 || numInStreams != 1)
|
||
|
{
|
||
|
UInt32 i;
|
||
|
UInt32 numBindPairs = numOutStreams - 1;
|
||
|
if (numOutStreams == 0 || numInStreams < numBindPairs)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
|
||
|
if (numInStreams > k_InStreamUsed_MAX ||
|
||
|
numOutStreams > k_OutStreamUsed_MAX)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
|
||
|
for (i = 0; i < numInStreams; i++)
|
||
|
inStreamUsed[i] = False;
|
||
|
for (i = 0; i < numOutStreams; i++)
|
||
|
outStreamUsed[i] = False;
|
||
|
|
||
|
for (i = 0; i < numBindPairs; i++)
|
||
|
{
|
||
|
UInt32 index;
|
||
|
RINOK(SzReadNumber32(&sd, &index));
|
||
|
if (index >= numInStreams || inStreamUsed[index])
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
inStreamUsed[index] = True;
|
||
|
RINOK(SzReadNumber32(&sd, &index));
|
||
|
if (index >= numInStreams || outStreamUsed[index])
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
outStreamUsed[index] = True;
|
||
|
}
|
||
|
|
||
|
numPackStreams = numInStreams - numBindPairs;
|
||
|
|
||
|
if (numPackStreams != 1)
|
||
|
for (i = 0; i < numPackStreams; i++)
|
||
|
{
|
||
|
UInt32 temp;
|
||
|
RINOK(SzReadNumber32(&sd, &temp));
|
||
|
if (temp >= numInStreams)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
}
|
||
|
|
||
|
for (i = 0; i < numOutStreams; i++)
|
||
|
if (!outStreamUsed[i])
|
||
|
{
|
||
|
indexOfMainStream = i;
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
if (i == numOutStreams)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
}
|
||
|
p->FoStartPackStreamIndex[fo] = packStreamIndex;
|
||
|
p->FoSizesOffsets[fo] = (numOutStreams << 8) | indexOfMainStream;
|
||
|
numCodersOutStreams += numOutStreams;
|
||
|
if (numCodersOutStreams < numOutStreams)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
packStreamIndex += numPackStreams;
|
||
|
if (packStreamIndex < numPackStreams)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
if (packStreamIndex > p->NumPackStreams)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
{
|
||
|
size_t dataSize = sd.Data - startBufPtr;
|
||
|
p->FoStartPackStreamIndex[fo] = packStreamIndex;
|
||
|
p->FoCodersOffsets[fo] = dataSize;
|
||
|
MY_ALLOC(Byte, p->CodersData, dataSize, alloc);
|
||
|
memcpy(p->CodersData, startBufPtr, dataSize);
|
||
|
}
|
||
|
|
||
|
if (external != 0)
|
||
|
{
|
||
|
if (sd.Size != 0)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
sd = *sd2;
|
||
|
}
|
||
|
|
||
|
RINOK(WaitId(&sd, k7zIdCodersUnpackSize));
|
||
|
|
||
|
// MY_ALLOC(UInt64, p->CoderUnpackSizes, (size_t)numCodersOutStreams, alloc);
|
||
|
{
|
||
|
size_t dataSize = sd.Size;
|
||
|
/*
|
||
|
UInt32 i;
|
||
|
for (i = 0; i < numCodersOutStreams; i++)
|
||
|
{
|
||
|
RINOK(ReadNumber(&sd, p->CoderUnpackSizes + i));
|
||
|
}
|
||
|
*/
|
||
|
RINOK(SkipNumbers(&sd, numCodersOutStreams));
|
||
|
dataSize -= sd.Size;
|
||
|
MY_ALLOC(Byte, p->UnpackSizesData, dataSize, alloc);
|
||
|
memcpy(p->UnpackSizesData, sd.Data - dataSize, dataSize);
|
||
|
p->UnpackSizesDataSize = dataSize;
|
||
|
/*
|
||
|
const Byte *data = SzReadNumbers(sd.Data, sd.Data + sd.Size, p->CoderUnpackSizes, numCodersOutStreams);
|
||
|
if (data == NULL)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
sd.Size = sd.Data + sd.Size - data;
|
||
|
sd.Data = data;
|
||
|
*/
|
||
|
}
|
||
|
|
||
|
for (;;)
|
||
|
{
|
||
|
UInt64 type;
|
||
|
RINOK(ReadID(&sd, &type));
|
||
|
if (type == k7zIdEnd)
|
||
|
{
|
||
|
*sd2 = sd;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
if (type == k7zIdCRC)
|
||
|
{
|
||
|
RINOK(ReadBitUi32s(&sd, numFolders, &p->FolderCRCs, alloc));
|
||
|
continue;
|
||
|
}
|
||
|
RINOK(SkipData(&sd));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
typedef struct
|
||
|
{
|
||
|
UInt32 NumTotalSubStreams;
|
||
|
UInt32 NumSubDigests;
|
||
|
CSzData sdNumSubStreams;
|
||
|
CSzData sdSizes;
|
||
|
CSzData sdCRCs;
|
||
|
} CSubStreamInfo;
|
||
|
|
||
|
#define SzUi32IndexMax (((UInt32)1 << 31) - 2)
|
||
|
|
||
|
static SRes ReadSubStreamsInfo(CSzAr *p, CSzData *sd, CSubStreamInfo *ssi)
|
||
|
{
|
||
|
UInt64 type = 0;
|
||
|
UInt32 i;
|
||
|
UInt32 numSubDigests = 0;
|
||
|
UInt32 numFolders = p->NumFolders;
|
||
|
UInt32 numUnpackStreams = numFolders;
|
||
|
UInt32 numUnpackSizesInData = 0;
|
||
|
|
||
|
for (;;)
|
||
|
{
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
if (type == k7zIdNumUnpackStream)
|
||
|
{
|
||
|
ssi->sdNumSubStreams.Data = sd->Data;
|
||
|
numUnpackStreams = 0;
|
||
|
numSubDigests = 0;
|
||
|
for (i = 0; i < numFolders; i++)
|
||
|
{
|
||
|
UInt32 numStreams;
|
||
|
RINOK(SzReadNumber32(sd, &numStreams));
|
||
|
if (numUnpackStreams > numUnpackStreams + numStreams)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
numUnpackStreams += numStreams;
|
||
|
if (numStreams != 0)
|
||
|
numUnpackSizesInData += (numStreams - 1);
|
||
|
if (numStreams != 1 || !SzBitWithVals_Check(&p->FolderCRCs, i))
|
||
|
numSubDigests += numStreams;
|
||
|
}
|
||
|
ssi->sdNumSubStreams.Size = sd->Data - ssi->sdNumSubStreams.Data;
|
||
|
continue;
|
||
|
}
|
||
|
if (type == k7zIdCRC || type == k7zIdSize || type == k7zIdEnd)
|
||
|
break;
|
||
|
RINOK(SkipData(sd));
|
||
|
}
|
||
|
|
||
|
if (!ssi->sdNumSubStreams.Data)
|
||
|
{
|
||
|
numSubDigests = numFolders;
|
||
|
if (p->FolderCRCs.Defs)
|
||
|
numSubDigests = numFolders - CountDefinedBits(p->FolderCRCs.Defs, numFolders);
|
||
|
}
|
||
|
|
||
|
ssi->NumTotalSubStreams = numUnpackStreams;
|
||
|
ssi->NumSubDigests = numSubDigests;
|
||
|
|
||
|
if (type == k7zIdSize)
|
||
|
{
|
||
|
ssi->sdSizes.Data = sd->Data;
|
||
|
RINOK(SkipNumbers(sd, numUnpackSizesInData));
|
||
|
ssi->sdSizes.Size = sd->Data - ssi->sdSizes.Data;
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
}
|
||
|
|
||
|
for (;;)
|
||
|
{
|
||
|
if (type == k7zIdEnd)
|
||
|
return SZ_OK;
|
||
|
if (type == k7zIdCRC)
|
||
|
{
|
||
|
ssi->sdCRCs.Data = sd->Data;
|
||
|
RINOK(SkipBitUi32s(sd, numSubDigests));
|
||
|
ssi->sdCRCs.Size = sd->Data - ssi->sdCRCs.Data;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
RINOK(SkipData(sd));
|
||
|
}
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static SRes SzReadStreamsInfo(CSzAr *p,
|
||
|
CSzData *sd,
|
||
|
UInt32 numFoldersMax, const CBuf *tempBufs, UInt32 numTempBufs,
|
||
|
UInt64 *dataOffset,
|
||
|
CSubStreamInfo *ssi,
|
||
|
ISzAlloc *alloc)
|
||
|
{
|
||
|
UInt64 type;
|
||
|
|
||
|
SzData_Clear(&ssi->sdSizes);
|
||
|
SzData_Clear(&ssi->sdCRCs);
|
||
|
SzData_Clear(&ssi->sdNumSubStreams);
|
||
|
|
||
|
*dataOffset = 0;
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
if (type == k7zIdPackInfo)
|
||
|
{
|
||
|
RINOK(ReadNumber(sd, dataOffset));
|
||
|
RINOK(ReadPackInfo(p, sd, alloc));
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
}
|
||
|
if (type == k7zIdUnpackInfo)
|
||
|
{
|
||
|
RINOK(ReadUnpackInfo(p, sd, numFoldersMax, tempBufs, numTempBufs, alloc));
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
}
|
||
|
if (type == k7zIdSubStreamsInfo)
|
||
|
{
|
||
|
RINOK(ReadSubStreamsInfo(p, sd, ssi));
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
ssi->NumTotalSubStreams = p->NumFolders;
|
||
|
// ssi->NumSubDigests = 0;
|
||
|
}
|
||
|
|
||
|
return (type == k7zIdEnd ? SZ_OK : SZ_ERROR_UNSUPPORTED);
|
||
|
}
|
||
|
|
||
|
static SRes SzReadAndDecodePackedStreams(
|
||
|
ILookInStream *inStream,
|
||
|
CSzData *sd,
|
||
|
CBuf *tempBufs,
|
||
|
UInt32 numFoldersMax,
|
||
|
UInt64 baseOffset,
|
||
|
CSzAr *p,
|
||
|
ISzAlloc *allocTemp)
|
||
|
{
|
||
|
UInt64 dataStartPos;
|
||
|
UInt32 fo;
|
||
|
CSubStreamInfo ssi;
|
||
|
CSzData sdCodersUnpSizes;
|
||
|
|
||
|
RINOK(SzReadStreamsInfo(p, sd, numFoldersMax, NULL, 0, &dataStartPos, &ssi, allocTemp));
|
||
|
|
||
|
dataStartPos += baseOffset;
|
||
|
if (p->NumFolders == 0)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
|
||
|
sdCodersUnpSizes.Data = p->UnpackSizesData;
|
||
|
sdCodersUnpSizes.Size = p->UnpackSizesDataSize;
|
||
|
for (fo = 0; fo < p->NumFolders; fo++)
|
||
|
Buf_Init(tempBufs + fo);
|
||
|
for (fo = 0; fo < p->NumFolders; fo++)
|
||
|
{
|
||
|
CBuf *tempBuf = tempBufs + fo;
|
||
|
// folder = p->Folders;
|
||
|
// unpackSize = SzAr_GetFolderUnpackSize(p, 0);
|
||
|
UInt32 mix = (UInt32)p->FoSizesOffsets[fo];
|
||
|
UInt32 mainIndex = mix & 0xFF;
|
||
|
UInt32 numOutStreams = mix >> 8;
|
||
|
UInt32 si;
|
||
|
UInt64 unpackSize = 0;
|
||
|
p->FoSizesOffsets[fo] = sdCodersUnpSizes.Data - p->UnpackSizesData;
|
||
|
for (si = 0; si < numOutStreams; si++)
|
||
|
{
|
||
|
UInt64 curSize;
|
||
|
RINOK(ReadNumber(&sdCodersUnpSizes, &curSize));
|
||
|
if (si == mainIndex)
|
||
|
{
|
||
|
unpackSize = curSize;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
if (si == numOutStreams)
|
||
|
return SZ_ERROR_FAIL;
|
||
|
if ((size_t)unpackSize != unpackSize)
|
||
|
return SZ_ERROR_MEM;
|
||
|
if (!Buf_Create(tempBuf, (size_t)unpackSize, allocTemp))
|
||
|
return SZ_ERROR_MEM;
|
||
|
}
|
||
|
p->FoSizesOffsets[fo] = sdCodersUnpSizes.Data - p->UnpackSizesData;
|
||
|
|
||
|
for (fo = 0; fo < p->NumFolders; fo++)
|
||
|
{
|
||
|
const CBuf *tempBuf = tempBufs + fo;
|
||
|
RINOK(LookInStream_SeekTo(inStream, dataStartPos));
|
||
|
RINOK(SzAr_DecodeFolder(p, fo, inStream, dataStartPos, tempBuf->data, tempBuf->size, allocTemp));
|
||
|
if (SzBitWithVals_Check(&p->FolderCRCs, fo))
|
||
|
if (CrcCalc(tempBuf->data, tempBuf->size) != p->FolderCRCs.Vals[fo])
|
||
|
return SZ_ERROR_CRC;
|
||
|
}
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
static SRes SzReadFileNames(const Byte *data, size_t size, UInt32 numFiles, size_t *offsets)
|
||
|
{
|
||
|
size_t pos = 0;
|
||
|
*offsets++ = 0;
|
||
|
if (numFiles == 0)
|
||
|
return (size == 0) ? SZ_OK : SZ_ERROR_ARCHIVE;
|
||
|
if (data[size - 2] != 0 || data[size - 1] != 0)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
do
|
||
|
{
|
||
|
const Byte *p;
|
||
|
if (pos == size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
for (p = data + pos;
|
||
|
#ifdef _WIN32
|
||
|
*(const UInt16 *)p != 0
|
||
|
#else
|
||
|
p[0] != 0 || p[1] != 0
|
||
|
#endif
|
||
|
; p += 2);
|
||
|
pos = p - data + 2;
|
||
|
*offsets++ = (pos >> 1);
|
||
|
}
|
||
|
while (--numFiles);
|
||
|
return (pos == size) ? SZ_OK : SZ_ERROR_ARCHIVE;
|
||
|
}
|
||
|
|
||
|
static MY_NO_INLINE SRes ReadTime(CSzBitUi64s *p, UInt32 num,
|
||
|
CSzData *sd2,
|
||
|
const CBuf *tempBufs, UInt32 numTempBufs,
|
||
|
ISzAlloc *alloc)
|
||
|
{
|
||
|
CSzData sd;
|
||
|
UInt32 i;
|
||
|
CNtfsFileTime *vals;
|
||
|
Byte *defs;
|
||
|
Byte external;
|
||
|
RINOK(ReadBitVector(sd2, num, &p->Defs, alloc));
|
||
|
RINOK(SzReadByte(sd2, &external));
|
||
|
if (external == 0)
|
||
|
sd = *sd2;
|
||
|
else
|
||
|
{
|
||
|
UInt32 index;
|
||
|
SzReadNumber32(sd2, &index);
|
||
|
if (index >= numTempBufs)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
sd.Data = tempBufs[index].data;
|
||
|
sd.Size = tempBufs[index].size;
|
||
|
}
|
||
|
MY_ALLOC(CNtfsFileTime, p->Vals, num, alloc);
|
||
|
vals = p->Vals;
|
||
|
defs = p->Defs;
|
||
|
for (i = 0; i < num; i++)
|
||
|
if (SzBitArray_Check(defs, i))
|
||
|
{
|
||
|
if (sd.Size < 8)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
vals[i].Low = GetUi32(sd.Data);
|
||
|
vals[i].High = GetUi32(sd.Data + 4);
|
||
|
SKIP_DATA2(sd, 8);
|
||
|
}
|
||
|
else
|
||
|
vals[i].High = vals[i].Low = 0;
|
||
|
if (external == 0)
|
||
|
*sd2 = sd;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
#define NUM_ADDITIONAL_STREAMS_MAX 8
|
||
|
|
||
|
static SRes SzReadHeader2(
|
||
|
CSzArEx *p, /* allocMain */
|
||
|
CSzData *sd,
|
||
|
// Byte **emptyStreamVector, /* allocTemp */
|
||
|
// Byte **emptyFileVector, /* allocTemp */
|
||
|
// Byte **lwtVector, /* allocTemp */
|
||
|
ILookInStream *inStream,
|
||
|
CBuf *tempBufs,
|
||
|
UInt32 *numTempBufs,
|
||
|
ISzAlloc *allocMain,
|
||
|
ISzAlloc *allocTemp
|
||
|
)
|
||
|
{
|
||
|
UInt64 type;
|
||
|
UInt32 numFiles = 0;
|
||
|
UInt32 numEmptyStreams = 0;
|
||
|
UInt32 i;
|
||
|
CSubStreamInfo ssi;
|
||
|
const Byte *emptyStreams = 0;
|
||
|
const Byte *emptyFiles = 0;
|
||
|
|
||
|
SzData_Clear(&ssi.sdSizes);
|
||
|
SzData_Clear(&ssi.sdCRCs);
|
||
|
SzData_Clear(&ssi.sdNumSubStreams);
|
||
|
|
||
|
ssi.NumSubDigests = 0;
|
||
|
ssi.NumTotalSubStreams = 0;
|
||
|
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
|
||
|
if (type == k7zIdArchiveProperties)
|
||
|
{
|
||
|
for (;;)
|
||
|
{
|
||
|
UInt64 type;
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
if (type == k7zIdEnd)
|
||
|
break;
|
||
|
RINOK(SkipData(sd));
|
||
|
}
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
}
|
||
|
|
||
|
// if (type == k7zIdAdditionalStreamsInfo) return SZ_ERROR_UNSUPPORTED;
|
||
|
|
||
|
if (type == k7zIdAdditionalStreamsInfo)
|
||
|
{
|
||
|
CSzAr tempAr;
|
||
|
SRes res;
|
||
|
UInt32 numTempFolders;
|
||
|
|
||
|
SzAr_Init(&tempAr);
|
||
|
res = SzReadAndDecodePackedStreams(inStream, sd, tempBufs, NUM_ADDITIONAL_STREAMS_MAX,
|
||
|
p->startPosAfterHeader, &tempAr, allocTemp);
|
||
|
numTempFolders = tempAr.NumFolders;
|
||
|
SzAr_Free(&tempAr, allocTemp);
|
||
|
if (res != SZ_OK)
|
||
|
return res;
|
||
|
*numTempBufs = numTempFolders;
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
}
|
||
|
|
||
|
if (type == k7zIdMainStreamsInfo)
|
||
|
{
|
||
|
RINOK(SzReadStreamsInfo(&p->db, sd, (UInt32)1 << 30, tempBufs, *numTempBufs,
|
||
|
&p->dataPos, &ssi, allocMain));
|
||
|
p->dataPos += p->startPosAfterHeader;
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
}
|
||
|
|
||
|
if (type == k7zIdEnd)
|
||
|
{
|
||
|
// *sd2 = sd;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
if (type != k7zIdFilesInfo)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
|
||
|
RINOK(SzReadNumber32(sd, &numFiles));
|
||
|
p->NumFiles = numFiles;
|
||
|
|
||
|
for (;;)
|
||
|
{
|
||
|
UInt64 type;
|
||
|
UInt64 size;
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
if (type == k7zIdEnd)
|
||
|
break;
|
||
|
RINOK(ReadNumber(sd, &size));
|
||
|
if (size > sd->Size)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
if ((UInt64)(int)type != type)
|
||
|
{
|
||
|
SKIP_DATA(sd, size);
|
||
|
}
|
||
|
else switch((int)type)
|
||
|
{
|
||
|
case k7zIdName:
|
||
|
{
|
||
|
size_t namesSize;
|
||
|
const Byte *namesData;
|
||
|
Byte external;
|
||
|
|
||
|
SZ_READ_BYTE(external);
|
||
|
if (external == 0)
|
||
|
{
|
||
|
namesSize = (size_t)size - 1;
|
||
|
namesData = sd->Data;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
UInt32 index;
|
||
|
SzReadNumber32(sd, &index);
|
||
|
if (index >= *numTempBufs)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
namesData = (tempBufs)[index].data;
|
||
|
namesSize = (tempBufs)[index].size;
|
||
|
}
|
||
|
|
||
|
if ((namesSize & 1) != 0)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
MY_ALLOC(Byte, p->FileNames, namesSize, allocMain);
|
||
|
MY_ALLOC(size_t, p->FileNameOffsets, numFiles + 1, allocMain);
|
||
|
memcpy(p->FileNames, namesData, namesSize);
|
||
|
RINOK(SzReadFileNames(p->FileNames, namesSize, numFiles, p->FileNameOffsets))
|
||
|
if (external == 0)
|
||
|
{
|
||
|
SKIP_DATA(sd, namesSize);
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
case k7zIdEmptyStream:
|
||
|
{
|
||
|
RINOK(RememberBitVector(sd, numFiles, &emptyStreams));
|
||
|
numEmptyStreams = CountDefinedBits(emptyStreams, numFiles);
|
||
|
break;
|
||
|
}
|
||
|
case k7zIdEmptyFile:
|
||
|
{
|
||
|
RINOK(RememberBitVector(sd, numEmptyStreams, &emptyFiles));
|
||
|
break;
|
||
|
}
|
||
|
case k7zIdWinAttrib:
|
||
|
{
|
||
|
Byte external;
|
||
|
CSzData sdSwitch;
|
||
|
CSzData *sdPtr;
|
||
|
SzBitUi32s_Free(&p->Attribs, allocMain);
|
||
|
RINOK(ReadBitVector(sd, numFiles, &p->Attribs.Defs, allocMain));
|
||
|
|
||
|
SZ_READ_BYTE(external);
|
||
|
if (external == 0)
|
||
|
sdPtr = sd;
|
||
|
else
|
||
|
{
|
||
|
UInt32 index;
|
||
|
SzReadNumber32(sd, &index);
|
||
|
if (index >= *numTempBufs)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
sdSwitch.Data = (tempBufs)[index].data;
|
||
|
sdSwitch.Size = (tempBufs)[index].size;
|
||
|
sdPtr = &sdSwitch;
|
||
|
}
|
||
|
RINOK(ReadUi32s(sdPtr, numFiles, &p->Attribs, allocMain));
|
||
|
break;
|
||
|
}
|
||
|
/*
|
||
|
case k7zParent:
|
||
|
{
|
||
|
SzBitUi32s_Free(&p->Parents, allocMain);
|
||
|
RINOK(ReadBitVector(sd, numFiles, &p->Parents.Defs, allocMain));
|
||
|
RINOK(SzReadSwitch(sd));
|
||
|
RINOK(ReadUi32s(sd, numFiles, &p->Parents, allocMain));
|
||
|
break;
|
||
|
}
|
||
|
*/
|
||
|
case k7zIdMTime: RINOK(ReadTime(&p->MTime, numFiles, sd, tempBufs, *numTempBufs, allocMain)); break;
|
||
|
case k7zIdCTime: RINOK(ReadTime(&p->CTime, numFiles, sd, tempBufs, *numTempBufs, allocMain)); break;
|
||
|
default:
|
||
|
{
|
||
|
SKIP_DATA(sd, size);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (numFiles - numEmptyStreams != ssi.NumTotalSubStreams)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
|
||
|
for (;;)
|
||
|
{
|
||
|
UInt64 type;
|
||
|
RINOK(ReadID(sd, &type));
|
||
|
if (type == k7zIdEnd)
|
||
|
break;
|
||
|
RINOK(SkipData(sd));
|
||
|
}
|
||
|
|
||
|
{
|
||
|
UInt32 emptyFileIndex = 0;
|
||
|
|
||
|
UInt32 folderIndex = 0;
|
||
|
UInt32 indexInFolder = 0;
|
||
|
UInt64 unpackPos = 0;
|
||
|
const Byte *digestsDefs = 0;
|
||
|
const Byte *digestsVals = 0;
|
||
|
UInt32 digestsValsIndex = 0;
|
||
|
UInt32 digestIndex;
|
||
|
Byte allDigestsDefined = 0;
|
||
|
UInt32 curNumSubStreams = (UInt32)(Int32)-1;
|
||
|
Byte isDirMask = 0;
|
||
|
Byte crcMask = 0;
|
||
|
Byte mask = 0x80;
|
||
|
// size_t unpSizesOffset = 0;
|
||
|
CSzData sdCodersUnpSizes;
|
||
|
sdCodersUnpSizes.Data = p->db.UnpackSizesData;
|
||
|
sdCodersUnpSizes.Size = p->db.UnpackSizesDataSize;
|
||
|
|
||
|
MY_ALLOC(UInt32, p->FolderStartFileIndex, p->db.NumFolders + 1, allocMain);
|
||
|
MY_ALLOC(UInt32, p->FileIndexToFolderIndexMap, p->NumFiles, allocMain);
|
||
|
MY_ALLOC(UInt64, p->UnpackPositions, p->NumFiles + 1, allocMain);
|
||
|
MY_ALLOC(Byte, p->IsDirs, (p->NumFiles + 7) >> 3, allocMain);
|
||
|
|
||
|
RINOK(SzBitUi32s_Alloc(&p->CRCs, p->NumFiles, allocMain));
|
||
|
|
||
|
if (ssi.sdCRCs.Size != 0)
|
||
|
{
|
||
|
RINOK(SzReadByte(&ssi.sdCRCs, &allDigestsDefined));
|
||
|
if (allDigestsDefined)
|
||
|
digestsVals = ssi.sdCRCs.Data;
|
||
|
else
|
||
|
{
|
||
|
size_t numBytes = (ssi.NumSubDigests + 7) >> 3;
|
||
|
digestsDefs = ssi.sdCRCs.Data;
|
||
|
digestsVals = digestsDefs + numBytes;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
digestIndex = 0;
|
||
|
for (i = 0; i < numFiles; i++, mask >>= 1)
|
||
|
{
|
||
|
if (mask == 0)
|
||
|
{
|
||
|
UInt32 byteIndex = (i - 1) >> 3;
|
||
|
p->IsDirs[byteIndex] = isDirMask;
|
||
|
p->CRCs.Defs[byteIndex] = crcMask;
|
||
|
isDirMask = 0;
|
||
|
crcMask = 0;
|
||
|
mask = 0x80;
|
||
|
}
|
||
|
|
||
|
p->UnpackPositions[i] = unpackPos;
|
||
|
p->CRCs.Vals[i] = 0;
|
||
|
// p->CRCs.Defs[i] = 0;
|
||
|
if (emptyStreams && SzBitArray_Check(emptyStreams , i))
|
||
|
{
|
||
|
if (!emptyFiles || !SzBitArray_Check(emptyFiles, emptyFileIndex))
|
||
|
isDirMask |= mask;
|
||
|
emptyFileIndex++;
|
||
|
if (indexInFolder == 0)
|
||
|
{
|
||
|
p->FileIndexToFolderIndexMap[i] = (UInt32)-1;
|
||
|
continue;
|
||
|
}
|
||
|
}
|
||
|
if (indexInFolder == 0)
|
||
|
{
|
||
|
/*
|
||
|
v3.13 incorrectly worked with empty folders
|
||
|
v4.07: Loop for skipping empty folders
|
||
|
*/
|
||
|
for (;;)
|
||
|
{
|
||
|
if (folderIndex >= p->db.NumFolders)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
p->FolderStartFileIndex[folderIndex] = i;
|
||
|
if (curNumSubStreams == (UInt32)(Int32)-1);
|
||
|
{
|
||
|
curNumSubStreams = 1;
|
||
|
if (ssi.sdNumSubStreams.Data != 0)
|
||
|
{
|
||
|
RINOK(SzReadNumber32(&ssi.sdNumSubStreams, &curNumSubStreams));
|
||
|
}
|
||
|
}
|
||
|
if (curNumSubStreams != 0)
|
||
|
break;
|
||
|
curNumSubStreams = (UInt32)(Int32)-1;
|
||
|
folderIndex++; // check it
|
||
|
}
|
||
|
}
|
||
|
p->FileIndexToFolderIndexMap[i] = folderIndex;
|
||
|
if (emptyStreams && SzBitArray_Check(emptyStreams , i))
|
||
|
continue;
|
||
|
|
||
|
indexInFolder++;
|
||
|
if (indexInFolder >= curNumSubStreams)
|
||
|
{
|
||
|
UInt64 folderUnpackSize = 0;
|
||
|
UInt64 startFolderUnpackPos;
|
||
|
{
|
||
|
UInt32 mix = (UInt32)p->db.FoSizesOffsets[folderIndex];
|
||
|
UInt32 mainIndex = mix & 0xFF;
|
||
|
UInt32 numOutStreams = mix >> 8;
|
||
|
UInt32 si;
|
||
|
p->db.FoSizesOffsets[folderIndex] = sdCodersUnpSizes.Data - p->db.UnpackSizesData;
|
||
|
for (si = 0; si < numOutStreams; si++)
|
||
|
{
|
||
|
UInt64 curSize;
|
||
|
RINOK(ReadNumber(&sdCodersUnpSizes, &curSize));
|
||
|
if (si == mainIndex)
|
||
|
{
|
||
|
folderUnpackSize = curSize;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
if (si == numOutStreams)
|
||
|
return SZ_ERROR_FAIL;
|
||
|
}
|
||
|
|
||
|
// UInt64 folderUnpackSize = SzAr_GetFolderUnpackSize(&p->db, folderIndex);
|
||
|
startFolderUnpackPos = p->UnpackPositions[p->FolderStartFileIndex[folderIndex]];
|
||
|
if (folderUnpackSize < unpackPos - startFolderUnpackPos)
|
||
|
return SZ_ERROR_ARCHIVE;
|
||
|
unpackPos = startFolderUnpackPos + folderUnpackSize;
|
||
|
|
||
|
if (curNumSubStreams == 1 && SzBitWithVals_Check(&p->db.FolderCRCs, i))
|
||
|
{
|
||
|
p->CRCs.Vals[i] = p->db.FolderCRCs.Vals[folderIndex];
|
||
|
crcMask |= mask;
|
||
|
}
|
||
|
else if (allDigestsDefined || (digestsDefs && SzBitArray_Check(digestsDefs, digestIndex)))
|
||
|
{
|
||
|
p->CRCs.Vals[i] = GetUi32(digestsVals + (size_t)digestsValsIndex * 4);
|
||
|
digestsValsIndex++;
|
||
|
crcMask |= mask;
|
||
|
}
|
||
|
folderIndex++;
|
||
|
indexInFolder = 0;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
UInt64 v;
|
||
|
RINOK(ReadNumber(&ssi.sdSizes, &v));
|
||
|
unpackPos += v;
|
||
|
if (allDigestsDefined || (digestsDefs && SzBitArray_Check(digestsDefs, digestIndex)))
|
||
|
{
|
||
|
p->CRCs.Vals[i] = GetUi32(digestsVals + (size_t)digestsValsIndex * 4);
|
||
|
digestsValsIndex++;
|
||
|
crcMask |= mask;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
if (mask != 0x80)
|
||
|
{
|
||
|
UInt32 byteIndex = (i - 1) >> 3;
|
||
|
p->IsDirs[byteIndex] = isDirMask;
|
||
|
p->CRCs.Defs[byteIndex] = crcMask;
|
||
|
}
|
||
|
p->UnpackPositions[i] = unpackPos;
|
||
|
p->FolderStartFileIndex[folderIndex] = i;
|
||
|
p->db.FoSizesOffsets[folderIndex] = sdCodersUnpSizes.Data - p->db.UnpackSizesData;
|
||
|
}
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
static SRes SzReadHeader(
|
||
|
CSzArEx *p,
|
||
|
CSzData *sd,
|
||
|
ILookInStream *inStream,
|
||
|
ISzAlloc *allocMain
|
||
|
,ISzAlloc *allocTemp
|
||
|
)
|
||
|
{
|
||
|
// Byte *emptyStreamVector = 0;
|
||
|
// Byte *emptyFileVector = 0;
|
||
|
// Byte *lwtVector = 0;
|
||
|
UInt32 i;
|
||
|
UInt32 numTempBufs = 0;
|
||
|
SRes res;
|
||
|
CBuf tempBufs[NUM_ADDITIONAL_STREAMS_MAX];
|
||
|
|
||
|
for (i = 0; i < NUM_ADDITIONAL_STREAMS_MAX; i++)
|
||
|
Buf_Init(tempBufs + i);
|
||
|
// SzBitUi32s_Init(&digests);
|
||
|
|
||
|
res = SzReadHeader2(p, sd,
|
||
|
// &emptyStreamVector,
|
||
|
// &emptyFileVector,
|
||
|
// &lwtVector,
|
||
|
inStream,
|
||
|
tempBufs, &numTempBufs,
|
||
|
allocMain, allocTemp
|
||
|
);
|
||
|
|
||
|
for (i = 0; i < numTempBufs; i++)
|
||
|
Buf_Free(tempBufs + i, allocTemp);
|
||
|
|
||
|
// IAlloc_Free(allocTemp, emptyStreamVector);
|
||
|
// IAlloc_Free(allocTemp, emptyFileVector);
|
||
|
// IAlloc_Free(allocTemp, lwtVector);
|
||
|
|
||
|
RINOK(res);
|
||
|
{
|
||
|
if (sd->Size != 0)
|
||
|
return SZ_ERROR_FAIL;
|
||
|
}
|
||
|
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
static UInt64 SzAr_GetFolderUnpackSize(const CSzAr *p, UInt32 folderIndex)
|
||
|
{
|
||
|
const CSzFolder2 *f = p->Folders + folderIndex;
|
||
|
|
||
|
// return p->CoderUnpackSizes[f->StartCoderUnpackSizesIndex + f->IndexOfMainOutStream];
|
||
|
|
||
|
UInt32 si;
|
||
|
CSzData sdCodersUnpSizes;
|
||
|
sdCodersUnpSizes.Data = p->UnpackSizesData + f->UnpackSizeDataOffset;
|
||
|
sdCodersUnpSizes.Size = p->UnpackSizesDataSize - f->UnpackSizeDataOffset;
|
||
|
for (si = 0; si < numOutStreams; si++)
|
||
|
{
|
||
|
UInt64 curSize;
|
||
|
ReadNumber(&sdCodersUnpSizes, &curSize);
|
||
|
if (si == mainIndex)
|
||
|
return curSize;
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
*/
|
||
|
|
||
|
static SRes SzArEx_Open2(
|
||
|
CSzArEx *p,
|
||
|
ILookInStream *inStream,
|
||
|
ISzAlloc *allocMain,
|
||
|
ISzAlloc *allocTemp)
|
||
|
{
|
||
|
Byte header[k7zStartHeaderSize];
|
||
|
Int64 startArcPos;
|
||
|
UInt64 nextHeaderOffset, nextHeaderSize;
|
||
|
size_t nextHeaderSizeT;
|
||
|
UInt32 nextHeaderCRC;
|
||
|
CBuf buf;
|
||
|
SRes res;
|
||
|
|
||
|
startArcPos = 0;
|
||
|
RINOK(inStream->Seek(inStream, &startArcPos, SZ_SEEK_CUR));
|
||
|
|
||
|
RINOK(LookInStream_Read2(inStream, header, k7zStartHeaderSize, SZ_ERROR_NO_ARCHIVE));
|
||
|
|
||
|
if (!TestSignatureCandidate(header))
|
||
|
return SZ_ERROR_NO_ARCHIVE;
|
||
|
if (header[6] != k7zMajorVersion)
|
||
|
return SZ_ERROR_UNSUPPORTED;
|
||
|
|
||
|
nextHeaderOffset = GetUi64(header + 12);
|
||
|
nextHeaderSize = GetUi64(header + 20);
|
||
|
nextHeaderCRC = GetUi32(header + 28);
|
||
|
|
||
|
p->startPosAfterHeader = startArcPos + k7zStartHeaderSize;
|
||
|
|
||
|
if (CrcCalc(header + 12, 20) != GetUi32(header + 8))
|
||
|
return SZ_ERROR_CRC;
|
||
|
|
||
|
nextHeaderSizeT = (size_t)nextHeaderSize;
|
||
|
if (nextHeaderSizeT != nextHeaderSize)
|
||
|
return SZ_ERROR_MEM;
|
||
|
if (nextHeaderSizeT == 0)
|
||
|
return SZ_OK;
|
||
|
if (nextHeaderOffset > nextHeaderOffset + nextHeaderSize ||
|
||
|
nextHeaderOffset > nextHeaderOffset + nextHeaderSize + k7zStartHeaderSize)
|
||
|
return SZ_ERROR_NO_ARCHIVE;
|
||
|
|
||
|
{
|
||
|
Int64 pos = 0;
|
||
|
RINOK(inStream->Seek(inStream, &pos, SZ_SEEK_END));
|
||
|
if ((UInt64)pos < startArcPos + nextHeaderOffset ||
|
||
|
(UInt64)pos < startArcPos + k7zStartHeaderSize + nextHeaderOffset ||
|
||
|
(UInt64)pos < startArcPos + k7zStartHeaderSize + nextHeaderOffset + nextHeaderSize)
|
||
|
return SZ_ERROR_INPUT_EOF;
|
||
|
}
|
||
|
|
||
|
RINOK(LookInStream_SeekTo(inStream, startArcPos + k7zStartHeaderSize + nextHeaderOffset));
|
||
|
|
||
|
if (!Buf_Create(&buf, nextHeaderSizeT, allocTemp))
|
||
|
return SZ_ERROR_MEM;
|
||
|
|
||
|
res = LookInStream_Read(inStream, buf.data, nextHeaderSizeT);
|
||
|
if (res == SZ_OK)
|
||
|
{
|
||
|
res = SZ_ERROR_ARCHIVE;
|
||
|
if (CrcCalc(buf.data, nextHeaderSizeT) == nextHeaderCRC)
|
||
|
{
|
||
|
CSzData sd;
|
||
|
UInt64 type;
|
||
|
sd.Data = buf.data;
|
||
|
sd.Size = buf.size;
|
||
|
res = ReadID(&sd, &type);
|
||
|
if (res == SZ_OK && type == k7zIdEncodedHeader)
|
||
|
{
|
||
|
CSzAr tempAr;
|
||
|
CBuf tempBuf;
|
||
|
Buf_Init(&tempBuf);
|
||
|
|
||
|
SzAr_Init(&tempAr);
|
||
|
res = SzReadAndDecodePackedStreams(inStream, &sd, &tempBuf, 1, p->startPosAfterHeader, &tempAr, allocTemp);
|
||
|
SzAr_Free(&tempAr, allocTemp);
|
||
|
|
||
|
if (res != SZ_OK)
|
||
|
{
|
||
|
Buf_Free(&tempBuf, allocTemp);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Buf_Free(&buf, allocTemp);
|
||
|
buf.data = tempBuf.data;
|
||
|
buf.size = tempBuf.size;
|
||
|
sd.Data = buf.data;
|
||
|
sd.Size = buf.size;
|
||
|
res = ReadID(&sd, &type);
|
||
|
}
|
||
|
}
|
||
|
if (res == SZ_OK)
|
||
|
{
|
||
|
if (type == k7zIdHeader)
|
||
|
{
|
||
|
CSzData sd2;
|
||
|
int ttt;
|
||
|
for (ttt = 0; ttt < 1; ttt++)
|
||
|
// for (ttt = 0; ttt < 40000; ttt++)
|
||
|
{
|
||
|
SzArEx_Free(p, allocMain);
|
||
|
sd2 = sd;
|
||
|
res = SzReadHeader(p, &sd2, inStream, allocMain, allocTemp
|
||
|
);
|
||
|
if (res != SZ_OK)
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
// res = SzReadHeader(p, &sd, allocMain, allocTemp);
|
||
|
}
|
||
|
else
|
||
|
res = SZ_ERROR_UNSUPPORTED;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
Buf_Free(&buf, allocTemp);
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
// #include <stdio.h>
|
||
|
|
||
|
SRes SzArEx_Open(CSzArEx *p, ILookInStream *inStream,
|
||
|
ISzAlloc *allocMain, ISzAlloc *allocTemp)
|
||
|
{
|
||
|
SRes res = SzArEx_Open2(p, inStream, allocMain, allocTemp);
|
||
|
if (res != SZ_OK)
|
||
|
SzArEx_Free(p, allocMain);
|
||
|
// printf ("\nrrr=%d\n", rrr);
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
SRes SzArEx_Extract(
|
||
|
const CSzArEx *p,
|
||
|
ILookInStream *inStream,
|
||
|
UInt32 fileIndex,
|
||
|
UInt32 *blockIndex,
|
||
|
Byte **tempBuf,
|
||
|
size_t *outBufferSize,
|
||
|
size_t *offset,
|
||
|
size_t *outSizeProcessed,
|
||
|
ISzAlloc *allocMain,
|
||
|
ISzAlloc *allocTemp)
|
||
|
{
|
||
|
UInt32 folderIndex = p->FileIndexToFolderIndexMap[fileIndex];
|
||
|
SRes res = SZ_OK;
|
||
|
*offset = 0;
|
||
|
*outSizeProcessed = 0;
|
||
|
if (folderIndex == (UInt32)-1)
|
||
|
{
|
||
|
IAlloc_Free(allocMain, *tempBuf);
|
||
|
*blockIndex = folderIndex;
|
||
|
*tempBuf = 0;
|
||
|
*outBufferSize = 0;
|
||
|
return SZ_OK;
|
||
|
}
|
||
|
|
||
|
if (*tempBuf == 0 || *blockIndex != folderIndex)
|
||
|
{
|
||
|
// UInt64 unpackSizeSpec = SzAr_GetFolderUnpackSize(&p->db, folderIndex);
|
||
|
UInt64 unpackSizeSpec =
|
||
|
p->UnpackPositions[p->FolderStartFileIndex[folderIndex + 1]] -
|
||
|
p->UnpackPositions[p->FolderStartFileIndex[folderIndex]];
|
||
|
size_t unpackSize = (size_t)unpackSizeSpec;
|
||
|
|
||
|
if (unpackSize != unpackSizeSpec)
|
||
|
return SZ_ERROR_MEM;
|
||
|
*blockIndex = folderIndex;
|
||
|
IAlloc_Free(allocMain, *tempBuf);
|
||
|
*tempBuf = 0;
|
||
|
|
||
|
// RINOK(LookInStream_SeekTo(inStream, startOffset));
|
||
|
|
||
|
if (res == SZ_OK)
|
||
|
{
|
||
|
*outBufferSize = unpackSize;
|
||
|
if (unpackSize != 0)
|
||
|
{
|
||
|
*tempBuf = (Byte *)IAlloc_Alloc(allocMain, unpackSize);
|
||
|
if (*tempBuf == 0)
|
||
|
res = SZ_ERROR_MEM;
|
||
|
}
|
||
|
if (res == SZ_OK)
|
||
|
{
|
||
|
res = SzAr_DecodeFolder(&p->db, folderIndex,
|
||
|
inStream,
|
||
|
p->dataPos,
|
||
|
*tempBuf, unpackSize, allocTemp);
|
||
|
if (res == SZ_OK)
|
||
|
{
|
||
|
if (SzBitWithVals_Check(&p->db.FolderCRCs, folderIndex))
|
||
|
{
|
||
|
if (CrcCalc(*tempBuf, unpackSize) != p->db.FolderCRCs.Vals[folderIndex])
|
||
|
res = SZ_ERROR_CRC;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
if (res == SZ_OK)
|
||
|
{
|
||
|
UInt64 unpackPos = p->UnpackPositions[fileIndex];
|
||
|
*offset = (size_t)(unpackPos - p->UnpackPositions[p->FolderStartFileIndex[folderIndex]]);
|
||
|
*outSizeProcessed = (size_t)(p->UnpackPositions[fileIndex + 1] - unpackPos);
|
||
|
if (*offset + *outSizeProcessed > *outBufferSize)
|
||
|
return SZ_ERROR_FAIL;
|
||
|
if (SzBitWithVals_Check(&p->CRCs, fileIndex) && CrcCalc(*tempBuf + *offset, *outSizeProcessed) != p->CRCs.Vals[fileIndex])
|
||
|
res = SZ_ERROR_CRC;
|
||
|
}
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
|
||
|
size_t SzArEx_GetFileNameUtf16(const CSzArEx *p, size_t fileIndex, UInt16 *dest)
|
||
|
{
|
||
|
size_t offs = p->FileNameOffsets[fileIndex];
|
||
|
size_t len = p->FileNameOffsets[fileIndex + 1] - offs;
|
||
|
if (dest != 0)
|
||
|
{
|
||
|
size_t i;
|
||
|
const Byte *src = p->FileNames + offs * 2;
|
||
|
for (i = 0; i < len; i++)
|
||
|
dest[i] = GetUi16(src + i * 2);
|
||
|
}
|
||
|
return len;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
size_t SzArEx_GetFullNameLen(const CSzArEx *p, size_t fileIndex)
|
||
|
{
|
||
|
size_t len;
|
||
|
if (!p->FileNameOffsets)
|
||
|
return 1;
|
||
|
len = 0;
|
||
|
for (;;)
|
||
|
{
|
||
|
UInt32 parent = (UInt32)(Int32)-1;
|
||
|
len += p->FileNameOffsets[fileIndex + 1] - p->FileNameOffsets[fileIndex];
|
||
|
if SzBitWithVals_Check(&p->Parents, fileIndex)
|
||
|
parent = p->Parents.Vals[fileIndex];
|
||
|
if (parent == (UInt32)(Int32)-1)
|
||
|
return len;
|
||
|
fileIndex = parent;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
UInt16 *SzArEx_GetFullNameUtf16_Back(const CSzArEx *p, size_t fileIndex, UInt16 *dest)
|
||
|
{
|
||
|
Bool needSlash;
|
||
|
if (!p->FileNameOffsets)
|
||
|
{
|
||
|
*(--dest) = 0;
|
||
|
return dest;
|
||
|
}
|
||
|
needSlash = False;
|
||
|
for (;;)
|
||
|
{
|
||
|
UInt32 parent = (UInt32)(Int32)-1;
|
||
|
size_t curLen = p->FileNameOffsets[fileIndex + 1] - p->FileNameOffsets[fileIndex];
|
||
|
SzArEx_GetFileNameUtf16(p, fileIndex, dest - curLen);
|
||
|
if (needSlash)
|
||
|
*(dest - 1) = '/';
|
||
|
needSlash = True;
|
||
|
dest -= curLen;
|
||
|
|
||
|
if SzBitWithVals_Check(&p->Parents, fileIndex)
|
||
|
parent = p->Parents.Vals[fileIndex];
|
||
|
if (parent == (UInt32)(Int32)-1)
|
||
|
return dest;
|
||
|
fileIndex = parent;
|
||
|
}
|
||
|
}
|
||
|
*/
|