diff options
Diffstat (limited to 'chrome/installer/util/lzma_util.cc')
| -rw-r--r-- | chrome/installer/util/lzma_util.cc | 210 |
1 files changed, 210 insertions, 0 deletions
diff --git a/chrome/installer/util/lzma_util.cc b/chrome/installer/util/lzma_util.cc new file mode 100644 index 0000000..45c30c4 --- /dev/null +++ b/chrome/installer/util/lzma_util.cc @@ -0,0 +1,210 @@ +// Copyright 2008, Google Inc. +// All rights reserved. +// +// Redistribution and use in source and binary forms, with or without +// modification, are permitted provided that the following conditions are +// met: +// +// * Redistributions of source code must retain the above copyright +// notice, this list of conditions and the following disclaimer. +// * Redistributions in binary form must reproduce the above +// copyright notice, this list of conditions and the following disclaimer +// in the documentation and/or other materials provided with the +// distribution. +// * Neither the name of Google Inc. nor the names of its +// contributors may be used to endorse or promote products derived from +// this software without specific prior written permission. +// +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT +// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, +// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +#include "lzma_util.h" + +#include "base/file_util.h" +#include "base/logging.h" +#include "base/string_util.h" + +extern "C" { +#include "Archive/7z/7zExtract.h" +#include "Archive/7z/7zIn.h" +#include "7zCrc.h" +} + + +namespace installer { + +typedef struct _CFileInStream { + ISzInStream InStream; + HANDLE File; +} CFileInStream; + + +size_t LzmaReadFile(HANDLE file, void *data, size_t size) { + if (size == 0) + return 0; + + size_t processedSize = 0; + do { + DWORD processedLoc = 0; + BOOL res = ReadFile(file, data, (DWORD) size, &processedLoc, NULL); + data = (void *)((unsigned char *) data + processedLoc); + size -= processedLoc; + processedSize += processedLoc; + if (!res || processedLoc == 0) + break; + } while (size > 0); + + return processedSize; +} + +SZ_RESULT SzFileSeekImp(void *object, CFileSize pos) { + CFileInStream *s = (CFileInStream *) object; + LARGE_INTEGER value; + value.LowPart = (DWORD) pos; + value.HighPart = (LONG) ((UInt64) pos >> 32); + value.LowPart = SetFilePointer(s->File, value.LowPart, &value.HighPart, + FILE_BEGIN); + if (value.LowPart == 0xFFFFFFFF) { + if(GetLastError() != NO_ERROR) { + return SZE_FAIL; + } + } + return SZ_OK; +} + +SZ_RESULT SzFileReadImp(void *object, void **buffer, + size_t maxRequiredSize, size_t *processedSize) { + const int kBufferSize = 1 << 12; + static Byte g_Buffer[kBufferSize]; + if (maxRequiredSize > kBufferSize) { + maxRequiredSize = kBufferSize; + } + + CFileInStream *s = (CFileInStream *) object; + size_t processedSizeLoc; + processedSizeLoc = LzmaReadFile(s->File, g_Buffer, maxRequiredSize); + *buffer = g_Buffer; + if (processedSize != 0) { + *processedSize = processedSizeLoc; + } + return SZ_OK; +} + + +LzmaUtil::LzmaUtil() : archive_handle_(NULL) {} + +LzmaUtil::~LzmaUtil() { + if (archive_handle_) CloseArchive(); +} + +DWORD LzmaUtil::OpenArchive(const std::wstring& archivePath) { + DWORD ret = NO_ERROR; + archive_handle_ = CreateFile(archivePath.c_str(), GENERIC_READ, + FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); + if (archive_handle_ == INVALID_HANDLE_VALUE) { + ret = GetLastError(); + } + return ret; +} + +// Unpacks the archive to the given location. +DWORD LzmaUtil::UnPack(const std::wstring& location) { + CFileInStream archiveStream; + ISzAlloc allocImp; + ISzAlloc allocTempImp; + CArchiveDatabaseEx db; + DWORD ret = NO_ERROR; + + archiveStream.File = archive_handle_; + archiveStream.InStream.Read = SzFileReadImp; + archiveStream.InStream.Seek = SzFileSeekImp; + allocImp.Alloc = SzAlloc; + allocImp.Free = SzFree; + allocTempImp.Alloc = SzAllocTemp; + allocTempImp.Free = SzFreeTemp; + + CrcGenerateTable(); + SzArDbExInit(&db); + if ((ret = SzArchiveOpen(&archiveStream.InStream, &db, + &allocImp, &allocTempImp)) != SZ_OK) { + return ret; + } + + Byte *outBuffer = 0; // it must be 0 before first call for each new archive + UInt32 blockIndex = 0xFFFFFFFF; // can have any value if outBuffer = 0 + size_t outBufferSize = 0; // can have any value if outBuffer = 0 + for (unsigned int i = 0; i < db.Database.NumFiles; i++) { + DWORD written; + size_t offset; + size_t outSizeProcessed; + CFileItem *f = db.Database.Files + i; + + if ((ret = SzExtract(&archiveStream.InStream, &db, i, &blockIndex, + &outBuffer, &outBufferSize, &offset, &outSizeProcessed, + &allocImp, &allocTempImp)) != SZ_OK) { + break; + } + + // Append location to the file path in archive, to get full path. + std::wstring wfileName(location); + file_util::AppendToPath(&wfileName, UTF8ToWide(f->Name)); + + // If archive entry is directory create it and move on to the next entry. + if (f->IsDirectory) { + file_util::CreateDirectory(wfileName); + continue; + } + + HANDLE hFile; + std::wstring directory = file_util::GetDirectoryFromPath(wfileName); + file_util::CreateDirectory(directory); + + hFile = CreateFile(wfileName.c_str(), GENERIC_WRITE, 0, NULL, + CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); + if (hFile == INVALID_HANDLE_VALUE) { + ret = GetLastError(); + break; + } + + WriteFile(hFile, outBuffer + offset, (DWORD) outSizeProcessed, + &written, NULL); + if (written != outSizeProcessed) { + ret = GetLastError(); + CloseHandle(hFile); + break; + } + + if (f->IsLastWriteTimeDefined) { + if (!SetFileTime(hFile, NULL, NULL, + (const FILETIME *)&(f->LastWriteTime))) { + ret = GetLastError(); + CloseHandle(hFile); + break; + } + } + if (!CloseHandle(hFile)) { + ret = GetLastError(); + break; + } + } // for loop + + allocImp.Free(outBuffer); + SzArDbExFree(&db, allocImp.Free); + return ret; +} + +void LzmaUtil::CloseArchive() { + CloseHandle(archive_handle_); + archive_handle_ = NULL; +} + +} // namespace installer |