7zin.cpp

7-Zip 3.11的源码

    RINOK(ReadID(type));
  }
 
  CObjectVector<CByteBuffer> dataVector;
  
  if (type == NID::kAdditionalStreamsInfo)
  {
    /*
    CRecordVector<UINT64> packSizes;
    CRecordVector<bool> packCRCsDefined;
    CRecordVector<UINT32> packCRCs;
    CObjectVector<CFolder> folders;

    CRecordVector<UINT64> numUnPackStreamsInFolders;
    CRecordVector<UINT64> unPackSizes;
    CRecordVector<bool> digestsDefined;
    CRecordVector<UINT32> digests;

    RINOK(ReadStreamsInfo(NULL, 
        database.ArchiveInfo.DataStartPosition2,
        packSizes, 
        packCRCsDefined, 
        packCRCs, 
        folders,
        numUnPackStreamsInFolders,
        unPackSizes,
        digestsDefined, 
        digests));

    database.ArchiveInfo.DataStartPosition2 += database.ArchiveInfo.StartPositionAfterHeader;

    UINT32 packIndex = 0;
    CDecoder decoder;
    UINT64 dataStartPos = database.ArchiveInfo.DataStartPosition2;
    for(int i = 0; i < folders.Size(); i++)
    {
      const CFolder &folder = folders[i];
      dataVector.Add(CByteBuffer());
      CByteBuffer &data = dataVector.Back();
      UINT64 unPackSize = folder.GetUnPackSize();
      data.SetCapacity(unPackSize);
      
      CComObjectNoLock<CSequentialOutStreamImp2> *outStreamSpec = 
        new  CComObjectNoLock<CSequentialOutStreamImp2>;
      CMyComPtr<ISequentialOutStream> outStream = outStreamSpec;
      outStreamSpec->Init(data, unPackSize);
      
      RINOK(decoder.Decode(_stream, dataStartPos, 
          &packSizes[packIndex], folder, outStream, NULL, getTextPassword));
      
      if (folder.UnPackCRCDefined)
        if (!CCRC::VerifyDigest(folder.UnPackCRC, data, unPackSize))
          throw CInArchiveException(CInArchiveException::kIncorrectHeader);
      for (int j = 0; j < folder.PackStreams.Size(); j++)
        dataStartPos += packSizes[packIndex++];
    }
    */
    HRESULT result = ReadAndDecodePackedStreams(
        database.ArchiveInfo.StartPositionAfterHeader, 
        database.ArchiveInfo.DataStartPosition2,
        dataVector
        #ifndef _NO_CRYPTO
        , getTextPassword
        #endif
        );
    RINOK(result);

    database.ArchiveInfo.DataStartPosition2 += database.ArchiveInfo.StartPositionAfterHeader;
    RINOK(ReadID(type));
  }

  CRecordVector<UINT64> unPackSizes;
  CRecordVector<bool> digestsDefined;
  CRecordVector<UINT32> digests;
  
  if (type == NID::kMainStreamsInfo)
  {
    RINOK(ReadStreamsInfo(&dataVector,
        database.ArchiveInfo.DataStartPosition,
        database.PackSizes, 
        database.PackCRCsDefined, 
        database.PackCRCs, 
        database.Folders,
        database.NumUnPackStreamsVector,
        unPackSizes,
        digestsDefined,
        digests));
    database.ArchiveInfo.DataStartPosition += database.ArchiveInfo.StartPositionAfterHeader;
    RINOK(ReadID(type));
  }
  else
  {
    for(int i = 0; i < database.Folders.Size(); i++)
    {
      database.NumUnPackStreamsVector.Add(1);
      CFolder &folder = database.Folders[i];
      unPackSizes.Add(folder.GetUnPackSize());
      digestsDefined.Add(folder.UnPackCRCDefined);
      digests.Add(folder.UnPackCRC);
    }
  }

  UINT64 numUnPackStreamsTotal = 0;

  database.Files.Clear();

  if (type == NID::kEnd)
    return S_OK;
  if (type != NID::kFilesInfo)
    throw CInArchiveException(CInArchiveException::kIncorrectHeader);
  
  UINT64 numFiles;
  RINOK(ReadNumber(numFiles));
  database.Files.Reserve((size_t)numFiles);
  UINT64 i;
  for(i = 0; i < numFiles; i++)
    database.Files.Add(CFileItem());

  // int sizePrev = -1;
  // int posPrev = 0;

  database.ArchiveInfo.FileInfoPopIDs.Add(NID::kSize);
  if (!database.PackSizes.IsEmpty())
    database.ArchiveInfo.FileInfoPopIDs.Add(NID::kPackInfo);
  if (numFiles > 0)
  {
    if (!digests.IsEmpty())
    { 
      database.ArchiveInfo.FileInfoPopIDs.Add(NID::kCRC);
    }
  }

  CBoolVector emptyStreamVector;
  emptyStreamVector.Reserve((size_t)numFiles);
  for(i = 0; i < numFiles; i++)
    emptyStreamVector.Add(false);
  CBoolVector emptyFileVector;
  CBoolVector antiFileVector;
  UINT32 numEmptyStreams = 0;

  while(true)
  {
    /*
    if (sizePrev >= 0)
      if (sizePrev != _inByteBack->GetProcessedSize() - posPrev)
        throw 2;
    */
    UINT64 type;
    RINOK(ReadID(type));
    if (type == NID::kEnd)
      break;
    UINT64 size;
    RINOK(ReadNumber(size));
    
    // sizePrev = size;
    // posPrev = _inByteBack->GetProcessedSize();

    database.ArchiveInfo.FileInfoPopIDs.Add(type);
    switch(type)
    {
      case NID::kName:
      {
        CStreamSwitch streamSwitch;
        RINOK(streamSwitch.Set(this, &dataVector));
        RINOK(ReadFileNames(database.Files))
        break;
      }
      case NID::kWinAttributes:
      {
        CBoolVector boolVector;
        RINOK(ReadBoolVector2(database.Files.Size(), boolVector))
        CStreamSwitch streamSwitch;
        RINOK(streamSwitch.Set(this, &dataVector));
        for(i = 0; i < numFiles; i++)
        {
          CFileItem &file = database.Files[(UINT32)i];
          if (file.AreAttributesDefined = boolVector[(UINT32)i])
          {
            RINOK(SafeReadBytes2(&file.Attributes, 
                sizeof(file.Attributes)));
          }
        }
        break;
      }
      case NID::kEmptyStream:
      {
        RINOK(ReadBoolVector((UINT32)numFiles, emptyStreamVector))
        UINT32 i;
        for (i = 0; i < (UINT32)emptyStreamVector.Size(); i++)
          if (emptyStreamVector[i])
            numEmptyStreams++;
        emptyFileVector.Reserve(numEmptyStreams);
        antiFileVector.Reserve(numEmptyStreams);
        for (i = 0; i < numEmptyStreams; i++)
        {
          emptyFileVector.Add(false);
          antiFileVector.Add(false);
        }
        break;
      }
      case NID::kEmptyFile:
      {
        RINOK(ReadBoolVector(numEmptyStreams, emptyFileVector))
        break;
      }
      case NID::kAnti:
      {
        RINOK(ReadBoolVector(numEmptyStreams, antiFileVector))
        break;
      }
      case NID::kCreationTime:
      case NID::kLastWriteTime:
      case NID::kLastAccessTime:
      {
        RINOK(ReadTime(dataVector, database.Files, type))
        break;
      }
      default:
      {
        database.ArchiveInfo.FileInfoPopIDs.DeleteBack();
        RINOK(SkeepData(size));
      }
    }
  }

  UINT32 emptyFileIndex = 0;
  UINT32 sizeIndex = 0;
  for(i = 0; i < numFiles; i++)
  {
    CFileItem &file = database.Files[(UINT32)i];
    file.HasStream = !emptyStreamVector[(UINT32)i];
    if(file.HasStream)
    {
      file.IsDirectory = false;
      file.IsAnti = false;
      file.UnPackSize = unPackSizes[sizeIndex];
      file.FileCRC = digests[sizeIndex];
      file.FileCRCIsDefined = digestsDefined[sizeIndex];
      sizeIndex++;
    }
    else
    {
      file.IsDirectory = !emptyFileVector[emptyFileIndex];
      file.IsAnti = antiFileVector[emptyFileIndex];
      emptyFileIndex++;
      file.UnPackSize = 0;
      file.FileCRCIsDefined = false;
    }
  }
  return S_OK;
}


void CArchiveDatabaseEx::FillFolderStartPackStream()
{
  FolderStartPackStreamIndex.Clear();
  FolderStartPackStreamIndex.Reserve(Folders.Size());
  UINT64 startPos = 0;
  for(UINT64 i = 0; i < Folders.Size(); i++)
  {
    FolderStartPackStreamIndex.Add((UINT32)startPos);
    startPos += Folders[(UINT32)i].PackStreams.Size();
  }
}

void CArchiveDatabaseEx::FillStartPos()
{
  PackStreamStartPositions.Clear();
  PackStreamStartPositions.Reserve(PackSizes.Size());
  UINT64 startPos = 0;
  for(UINT64 i = 0; i < PackSizes.Size(); i++)
  {
    PackStreamStartPositions.Add(startPos);
    startPos += PackSizes[(UINT32)i];
  }
}

void CArchiveDatabaseEx::FillFolderStartFileIndex()
{
  FolderStartFileIndex.Clear();
  FolderStartFileIndex.Reserve(Folders.Size());
  FileIndexToFolderIndexMap.Clear();
  FileIndexToFolderIndexMap.Reserve(Files.Size());
  
  int folderIndex = 0;
  int indexInFolder = 0;
  for (int i = 0; i < Files.Size(); i++)
  {
    const CFileItem &file = Files[i];
    bool emptyStream = !file.HasStream;
    if (emptyStream && indexInFolder == 0)
    {
      FileIndexToFolderIndexMap.Add(-1);
      continue;
    }
    if (indexInFolder == 0)
    {
      if (folderIndex >= Folders.Size())
        throw CInArchiveException(CInArchiveException::kIncorrectHeader);
      FolderStartFileIndex.Add(i);
    }
    FileIndexToFolderIndexMap.Add(folderIndex);
    if (emptyStream)
      continue;
    indexInFolder++;
    if (indexInFolder >= NumUnPackStreamsVector[folderIndex])
    {
      folderIndex++;
      indexInFolder = 0;
    }
  }
}

HRESULT CInArchive::ReadDatabase(CArchiveDatabaseEx &database
    #ifndef _NO_CRYPTO
    , ICryptoGetTextPassword *getTextPassword
    #endif
    )
{
  database.Clear();
  database.ArchiveInfo.StartPosition = _arhiveBeginStreamPosition;
  RINOK(SafeReadBytes(&database.ArchiveInfo.Version, 
      sizeof(database.ArchiveInfo.Version)));
  if (database.ArchiveInfo.Version.Major != kMajorVersion)
    throw  CInArchiveException(CInArchiveException::kUnsupportedVersion);

  UINT32 crcFromArchive;
  RINOK(SafeReadBytes(&crcFromArchive, sizeof(crcFromArchive)));
  CStartHeader startHeader;
  RINOK(SafeReadBytes(&startHeader, sizeof(startHeader)));
  if (!CCRC::VerifyDigest(crcFromArchive, &startHeader, sizeof(startHeader)))
    throw CInArchiveException(CInArchiveException::kIncorrectHeader);

  database.ArchiveInfo.StartPositionAfterHeader = _position;

  if (startHeader.NextHeaderSize == 0)
    return S_OK;

  RINOK(_stream->Seek(startHeader.NextHeaderOffset, STREAM_SEEK_CUR, &_position));

  CByteBuffer buffer2;
  buffer2.SetCapacity((size_t)startHeader.NextHeaderSize);
  RINOK(SafeReadBytes(buffer2, (UINT32)startHeader.NextHeaderSize));
  if (!CCRC::VerifyDigest(startHeader.NextHeaderCRC, buffer2, (UINT32)startHeader.NextHeaderSize))
    throw CInArchiveException(CInArchiveException::kIncorrectHeader);
  
  CStreamSwitch streamSwitch;
  streamSwitch.Set(this, buffer2);
  
  CObjectVector<CByteBuffer> dataVector;
  
  while (true)
  {
    UINT64 type;
    RINOK(ReadID(type));
    if (type == NID::kHeader)
      break;
    if (type != NID::kEncodedHeader)
      throw CInArchiveException(CInArchiveException::kIncorrectHeader);
    HRESULT result = ReadAndDecodePackedStreams(
        database.ArchiveInfo.StartPositionAfterHeader, 
        database.ArchiveInfo.DataStartPosition2,
        dataVector
        #ifndef _NO_CRYPTO
        , getTextPassword
        #endif
        );
    RINOK(result);
    if (dataVector.Size() == 0)
      return S_OK;
    if (dataVector.Size() > 1)
      throw CInArchiveException(CInArchiveException::kIncorrectHeader);
    streamSwitch.Remove();
    streamSwitch.Set(this, dataVector.Front());
  }

  return ReadHeader(database
    #ifndef _NO_CRYPTO
    , getTextPassword
    #endif
    );
}

}}