167 lines
3.8 KiB
C++
167 lines
3.8 KiB
C++
// Copyright (c) 2010 The Chromium Embedded Framework Authors. All rights
|
|
// reserved. Use of this source code is governed by a BSD-style license that
|
|
// can be found in the LICENSE file.
|
|
|
|
#include "include/cef_wrapper.h"
|
|
#include "libcef_dll/cef_logging.h"
|
|
#include <algorithm>
|
|
|
|
#if defined(__linux__)
|
|
#include <wctype.h>
|
|
#endif
|
|
|
|
namespace {
|
|
|
|
class CefZipFile : public CefThreadSafeBase<CefZipArchive::File>
|
|
{
|
|
public:
|
|
CefZipFile(size_t size) : data_(size) {}
|
|
~CefZipFile() {}
|
|
|
|
// Returns the read-only data contained in the file.
|
|
virtual const unsigned char* GetData() { return &data_[0]; }
|
|
|
|
// Returns the size of the data in the file.
|
|
virtual size_t GetDataSize() { return data_.size(); }
|
|
|
|
// Returns a CefStreamReader object for streaming the contents of the file.
|
|
virtual CefRefPtr<CefStreamReader> GetStreamReader() {
|
|
CefRefPtr<CefReadHandler> handler(
|
|
new CefByteReadHandler(GetData(), GetDataSize(), this));
|
|
return CefStreamReader::CreateForHandler(handler);
|
|
}
|
|
|
|
std::vector<unsigned char>* GetDataVector() { return &data_; }
|
|
|
|
private:
|
|
std::vector<unsigned char> data_;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
// CefZipArchive implementation
|
|
|
|
CefZipArchive::CefZipArchive()
|
|
{
|
|
}
|
|
|
|
CefZipArchive::~CefZipArchive()
|
|
{
|
|
}
|
|
|
|
size_t CefZipArchive::Load(CefRefPtr<CefStreamReader> stream,
|
|
bool overwriteExisting)
|
|
{
|
|
AutoLock lock_scope(this);
|
|
|
|
CefRefPtr<CefZipReader> reader(CefZipReader::Create(stream));
|
|
if (!reader.get())
|
|
return 0;
|
|
|
|
if (!reader->MoveToFirstFile())
|
|
return 0;
|
|
|
|
std::wstring name;
|
|
CefRefPtr<CefZipFile> contents;
|
|
FileMap::iterator it;
|
|
std::vector<unsigned char>* data;
|
|
size_t count = 0, size, offset;
|
|
|
|
do {
|
|
size = static_cast<size_t>(reader->GetFileSize());
|
|
if (size == 0) {
|
|
// Skip directories and empty files.
|
|
continue;
|
|
}
|
|
|
|
if (!reader->OpenFile(CefString()))
|
|
break;
|
|
|
|
name = reader->GetFileName();
|
|
std::transform(name.begin(), name.end(), name.begin(), towlower);
|
|
|
|
it = contents_.find(name);
|
|
if (it != contents_.end()) {
|
|
if (overwriteExisting)
|
|
contents_.erase(it);
|
|
else // Skip files that already exist.
|
|
continue;
|
|
}
|
|
|
|
contents = new CefZipFile(size);
|
|
data = contents->GetDataVector();
|
|
offset = 0;
|
|
|
|
// Read the file contents.
|
|
do {
|
|
offset += reader->ReadFile(&(*data)[offset], size - offset);
|
|
} while (offset < size && !reader->Eof());
|
|
|
|
DCHECK(offset == size);
|
|
|
|
reader->CloseFile();
|
|
count++;
|
|
|
|
// Add the file to the map.
|
|
contents_.insert(std::make_pair(name, contents.get()));
|
|
} while (reader->MoveToNextFile());
|
|
|
|
return count;
|
|
}
|
|
|
|
void CefZipArchive::Clear()
|
|
{
|
|
AutoLock lock_scope(this);
|
|
contents_.empty();
|
|
}
|
|
|
|
size_t CefZipArchive::GetFileCount()
|
|
{
|
|
AutoLock lock_scope(this);
|
|
return contents_.size();
|
|
}
|
|
|
|
bool CefZipArchive::HasFile(const CefString& fileName)
|
|
{
|
|
std::wstring str = fileName;
|
|
std::transform(str.begin(), str.end(), str.begin(), towlower);
|
|
|
|
AutoLock lock_scope(this);
|
|
FileMap::const_iterator it = contents_.find(CefString(str));
|
|
return (it != contents_.end());
|
|
}
|
|
|
|
CefRefPtr<CefZipArchive::File> CefZipArchive::GetFile(
|
|
const CefString& fileName)
|
|
{
|
|
std::wstring str = fileName;
|
|
std::transform(str.begin(), str.end(), str.begin(), towlower);
|
|
|
|
AutoLock lock_scope(this);
|
|
FileMap::const_iterator it = contents_.find(CefString(str));
|
|
if (it != contents_.end())
|
|
return it->second;
|
|
return NULL;
|
|
}
|
|
|
|
bool CefZipArchive::RemoveFile(const CefString& fileName)
|
|
{
|
|
std::wstring str = fileName;
|
|
std::transform(str.begin(), str.end(), str.begin(), towlower);
|
|
|
|
AutoLock lock_scope(this);
|
|
FileMap::iterator it = contents_.find(CefString(str));
|
|
if (it != contents_.end()) {
|
|
contents_.erase(it);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
size_t CefZipArchive::GetFiles(FileMap& map)
|
|
{
|
|
AutoLock lock_scope(this);
|
|
map = contents_;
|
|
return contents_.size();
|
|
}
|