Index: webrtc/base/diskcache.cc |
diff --git a/webrtc/base/diskcache.cc b/webrtc/base/diskcache.cc |
deleted file mode 100644 |
index 2747018d17473547d2c3bab52884c336b865f999..0000000000000000000000000000000000000000 |
--- a/webrtc/base/diskcache.cc |
+++ /dev/null |
@@ -1,347 +0,0 @@ |
-/* |
- * Copyright 2004 The WebRTC Project Authors. All rights reserved. |
- * |
- * Use of this source code is governed by a BSD-style license |
- * that can be found in the LICENSE file in the root of the source |
- * tree. An additional intellectual property rights grant can be found |
- * in the file PATENTS. All contributing project authors may |
- * be found in the AUTHORS file in the root of the source tree. |
- */ |
- |
-#include <time.h> |
- |
-#if defined(WEBRTC_WIN) |
-#include "webrtc/base/win32.h" |
-#endif |
- |
-#include <algorithm> |
-#include <memory> |
- |
-#include "webrtc/base/arraysize.h" |
-#include "webrtc/base/checks.h" |
-#include "webrtc/base/diskcache.h" |
-#include "webrtc/base/fileutils.h" |
-#include "webrtc/base/pathutils.h" |
-#include "webrtc/base/stream.h" |
-#include "webrtc/base/stringencode.h" |
-#include "webrtc/base/stringutils.h" |
- |
-#define TRANSPARENT_CACHE_NAMES RTC_DCHECK_IS_ON |
- |
-namespace rtc { |
- |
-class DiskCache; |
- |
-/////////////////////////////////////////////////////////////////////////////// |
-// DiskCacheAdapter |
-/////////////////////////////////////////////////////////////////////////////// |
- |
-class DiskCacheAdapter : public StreamAdapterInterface { |
-public: |
- DiskCacheAdapter(const DiskCache* cache, const std::string& id, size_t index, |
- StreamInterface* stream) |
- : StreamAdapterInterface(stream), cache_(cache), id_(id), index_(index) |
- { } |
- ~DiskCacheAdapter() override { |
- Close(); |
- cache_->ReleaseResource(id_, index_); |
- } |
- |
-private: |
- const DiskCache* cache_; |
- std::string id_; |
- size_t index_; |
-}; |
- |
-/////////////////////////////////////////////////////////////////////////////// |
-// DiskCache |
-/////////////////////////////////////////////////////////////////////////////// |
- |
-DiskCache::DiskCache() : max_cache_(0), total_size_(0), total_accessors_(0) { |
-} |
- |
-DiskCache::~DiskCache() { |
- RTC_DCHECK(0 == total_accessors_); |
-} |
- |
-bool DiskCache::Initialize(const std::string& folder, size_t size) { |
- if (!folder_.empty() || !Filesystem::CreateFolder(folder)) |
- return false; |
- |
- folder_ = folder; |
- max_cache_ = size; |
- RTC_DCHECK(0 == total_size_); |
- |
- if (!InitializeEntries()) |
- return false; |
- |
- return CheckLimit(); |
-} |
- |
-bool DiskCache::Purge() { |
- if (folder_.empty()) |
- return false; |
- |
- if (total_accessors_ > 0) { |
- LOG_F(LS_WARNING) << "Cache files open"; |
- return false; |
- } |
- |
- if (!PurgeFiles()) |
- return false; |
- |
- map_.clear(); |
- return true; |
-} |
- |
-bool DiskCache::LockResource(const std::string& id) { |
- Entry* entry = GetOrCreateEntry(id, true); |
- if (LS_LOCKED == entry->lock_state) |
- return false; |
- if ((LS_UNLOCKED == entry->lock_state) && (entry->accessors > 0)) |
- return false; |
- if ((total_size_ > max_cache_) && !CheckLimit()) { |
- LOG_F(LS_WARNING) << "Cache overfull"; |
- return false; |
- } |
- entry->lock_state = LS_LOCKED; |
- return true; |
-} |
- |
-StreamInterface* DiskCache::WriteResource(const std::string& id, size_t index) { |
- Entry* entry = GetOrCreateEntry(id, false); |
- if (LS_LOCKED != entry->lock_state) |
- return NULL; |
- |
- size_t previous_size = 0; |
- std::string filename(IdToFilename(id, index)); |
- FileStream::GetSize(filename, &previous_size); |
- RTC_DCHECK(previous_size <= entry->size); |
- if (previous_size > entry->size) { |
- previous_size = entry->size; |
- } |
- |
- std::unique_ptr<FileStream> file(new FileStream); |
- if (!file->Open(filename, "wb", NULL)) { |
- LOG_F(LS_ERROR) << "Couldn't create cache file"; |
- return NULL; |
- } |
- |
- entry->streams = std::max(entry->streams, index + 1); |
- entry->size -= previous_size; |
- total_size_ -= previous_size; |
- |
- entry->accessors += 1; |
- total_accessors_ += 1; |
- return new DiskCacheAdapter(this, id, index, file.release()); |
-} |
- |
-bool DiskCache::UnlockResource(const std::string& id) { |
- Entry* entry = GetOrCreateEntry(id, false); |
- if (LS_LOCKED != entry->lock_state) |
- return false; |
- |
- if (entry->accessors > 0) { |
- entry->lock_state = LS_UNLOCKING; |
- } else { |
- entry->lock_state = LS_UNLOCKED; |
- entry->last_modified = time(0); |
- CheckLimit(); |
- } |
- return true; |
-} |
- |
-StreamInterface* DiskCache::ReadResource(const std::string& id, |
- size_t index) const { |
- const Entry* entry = GetEntry(id); |
- if (LS_UNLOCKED != entry->lock_state) |
- return NULL; |
- if (index >= entry->streams) |
- return NULL; |
- |
- std::unique_ptr<FileStream> file(new FileStream); |
- if (!file->Open(IdToFilename(id, index), "rb", NULL)) |
- return NULL; |
- |
- entry->accessors += 1; |
- total_accessors_ += 1; |
- return new DiskCacheAdapter(this, id, index, file.release()); |
-} |
- |
-bool DiskCache::HasResource(const std::string& id) const { |
- const Entry* entry = GetEntry(id); |
- return (NULL != entry) && (entry->streams > 0); |
-} |
- |
-bool DiskCache::HasResourceStream(const std::string& id, size_t index) const { |
- const Entry* entry = GetEntry(id); |
- if ((NULL == entry) || (index >= entry->streams)) |
- return false; |
- |
- std::string filename = IdToFilename(id, index); |
- |
- return FileExists(filename); |
-} |
- |
-bool DiskCache::DeleteResource(const std::string& id) { |
- Entry* entry = GetOrCreateEntry(id, false); |
- if (!entry) |
- return true; |
- |
- if ((LS_UNLOCKED != entry->lock_state) || (entry->accessors > 0)) |
- return false; |
- |
- bool success = true; |
- for (size_t index = 0; index < entry->streams; ++index) { |
- std::string filename = IdToFilename(id, index); |
- |
- if (!FileExists(filename)) |
- continue; |
- |
- if (!DeleteFile(filename)) { |
- LOG_F(LS_ERROR) << "Couldn't remove cache file: " << filename; |
- success = false; |
- } |
- } |
- |
- total_size_ -= entry->size; |
- map_.erase(id); |
- return success; |
-} |
- |
-bool DiskCache::CheckLimit() { |
-#if RTC_DCHECK_IS_ON |
- // Temporary check to make sure everything is working correctly. |
- size_t cache_size = 0; |
- for (EntryMap::iterator it = map_.begin(); it != map_.end(); ++it) { |
- cache_size += it->second.size; |
- } |
- RTC_DCHECK(cache_size == total_size_); |
-#endif |
- |
- // TODO: Replace this with a non-brain-dead algorithm for clearing out the |
- // oldest resources... something that isn't O(n^2) |
- while (total_size_ > max_cache_) { |
- EntryMap::iterator oldest = map_.end(); |
- for (EntryMap::iterator it = map_.begin(); it != map_.end(); ++it) { |
- if ((LS_UNLOCKED != it->second.lock_state) || (it->second.accessors > 0)) |
- continue; |
- oldest = it; |
- break; |
- } |
- if (oldest == map_.end()) { |
- LOG_F(LS_WARNING) << "All resources are locked!"; |
- return false; |
- } |
- for (EntryMap::iterator it = oldest++; it != map_.end(); ++it) { |
- if (it->second.last_modified < oldest->second.last_modified) { |
- oldest = it; |
- } |
- } |
- if (!DeleteResource(oldest->first)) { |
- LOG_F(LS_ERROR) << "Couldn't delete from cache!"; |
- return false; |
- } |
- } |
- return true; |
-} |
- |
-std::string DiskCache::IdToFilename(const std::string& id, size_t index) const { |
-#ifdef TRANSPARENT_CACHE_NAMES |
- // This escapes colons and other filesystem characters, so the user can't open |
- // special devices (like "COM1:"), or access other directories. |
- size_t buffer_size = id.length()*3 + 1; |
- char* buffer = new char[buffer_size]; |
- encode(buffer, buffer_size, id.data(), id.length(), |
- unsafe_filename_characters(), '%'); |
- // TODO(nisse): RTC_DCHECK(strlen(buffer) < FileSystem::MaxBasenameLength()); |
-#else // !TRANSPARENT_CACHE_NAMES |
- // We might want to just use a hash of the filename at some point, both for |
- // obfuscation, and to avoid both filename length and escaping issues. |
- RTC_NOTREACHED(); |
-#endif // !TRANSPARENT_CACHE_NAMES |
- |
- char extension[32]; |
- sprintfn(extension, arraysize(extension), ".%u", index); |
- |
- Pathname pathname; |
- pathname.SetFolder(folder_); |
- pathname.SetBasename(buffer); |
- pathname.SetExtension(extension); |
- |
-#ifdef TRANSPARENT_CACHE_NAMES |
- delete [] buffer; |
-#endif // TRANSPARENT_CACHE_NAMES |
- |
- return pathname.pathname(); |
-} |
- |
-bool DiskCache::FilenameToId(const std::string& filename, std::string* id, |
- size_t* index) const { |
- Pathname pathname(filename); |
- unsigned tempdex; |
- if (1 != sscanf(pathname.extension().c_str(), ".%u", &tempdex)) |
- return false; |
- |
- *index = static_cast<size_t>(tempdex); |
- |
- size_t buffer_size = pathname.basename().length() + 1; |
- char* buffer = new char[buffer_size]; |
- decode(buffer, buffer_size, pathname.basename().data(), |
- pathname.basename().length(), '%'); |
- id->assign(buffer); |
- delete [] buffer; |
- return true; |
-} |
- |
-DiskCache::Entry* DiskCache::GetOrCreateEntry(const std::string& id, |
- bool create) { |
- EntryMap::iterator it = map_.find(id); |
- if (it != map_.end()) |
- return &it->second; |
- if (!create) |
- return NULL; |
- Entry e; |
- e.lock_state = LS_UNLOCKED; |
- e.accessors = 0; |
- e.size = 0; |
- e.streams = 0; |
- e.last_modified = time(0); |
- it = map_.insert(EntryMap::value_type(id, e)).first; |
- return &it->second; |
-} |
- |
-void DiskCache::ReleaseResource(const std::string& id, size_t index) const { |
- const Entry* entry = GetEntry(id); |
- if (!entry) { |
- LOG_F(LS_WARNING) << "Missing cache entry"; |
- RTC_NOTREACHED(); |
- return; |
- } |
- |
- entry->accessors -= 1; |
- total_accessors_ -= 1; |
- |
- if (LS_UNLOCKED != entry->lock_state) { |
- // This is safe, because locked resources only issue WriteResource, which |
- // is non-const. Think about a better way to handle it. |
- DiskCache* this2 = const_cast<DiskCache*>(this); |
- Entry* entry2 = this2->GetOrCreateEntry(id, false); |
- |
- size_t new_size = 0; |
- std::string filename(IdToFilename(id, index)); |
- FileStream::GetSize(filename, &new_size); |
- entry2->size += new_size; |
- this2->total_size_ += new_size; |
- |
- if ((LS_UNLOCKING == entry->lock_state) && (0 == entry->accessors)) { |
- entry2->last_modified = time(0); |
- entry2->lock_state = LS_UNLOCKED; |
- this2->CheckLimit(); |
- } |
- } |
-} |
- |
-/////////////////////////////////////////////////////////////////////////////// |
- |
-} // namespace rtc |