Index: storage/browser/blob/blob_storage_context.cc |
diff --git a/storage/browser/blob/blob_storage_context.cc b/storage/browser/blob/blob_storage_context.cc |
index ec66525e23f1746afa84a746f13b14ef5dc3b11b..43ba3eb632655fdadf96521b52dfb95a585327e5 100644 |
--- a/storage/browser/blob/blob_storage_context.cc |
+++ b/storage/browser/blob/blob_storage_context.cc |
@@ -4,12 +4,10 @@ |
#include "storage/browser/blob/blob_storage_context.h" |
-#include <stddef.h> |
-#include <stdint.h> |
- |
#include <algorithm> |
#include <limits> |
#include <memory> |
+#include <set> |
#include <utility> |
#include "base/bind.h" |
@@ -19,57 +17,330 @@ |
#include "base/memory/ptr_util.h" |
#include "base/message_loop/message_loop.h" |
#include "base/metrics/histogram.h" |
+#include "base/numerics/safe_conversions.h" |
+#include "base/numerics/safe_math.h" |
+#include "base/task_runner.h" |
#include "base/threading/thread_task_runner_handle.h" |
#include "base/trace_event/trace_event.h" |
+#include "storage/browser/blob/blob_async_transport_request_builder.h" |
#include "storage/browser/blob/blob_data_builder.h" |
-#include "storage/browser/blob/blob_data_handle.h" |
#include "storage/browser/blob/blob_data_item.h" |
#include "storage/browser/blob/blob_data_snapshot.h" |
#include "storage/browser/blob/shareable_blob_data_item.h" |
+#include "storage/common/data_element.h" |
#include "url/gurl.h" |
namespace storage { |
-using BlobRegistryEntry = BlobStorageRegistry::Entry; |
-using BlobState = BlobStorageRegistry::BlobState; |
+namespace { |
+using TransportState = InternalBlobData::TransportState; |
+using ItemCopyEntry = InternalBlobData::ItemCopyEntry; |
+using QuotaAllocationTask = BlobMemoryController::QuotaAllocationTask; |
+ |
+bool IsBytes(DataElement::Type type) { |
+ return type == DataElement::TYPE_BYTES || |
+ type == DataElement::TYPE_BYTES_DESCRIPTION; |
+} |
+} // namespace |
+ |
+BlobStorageContext::BlobFlattener::BlobFlattener( |
+ const BlobDataBuilder& input_builder, |
+ InternalBlobData* output_blob, |
+ BlobStorageRegistry* registry) { |
+ const std::string& uuid = input_builder.uuid_; |
+ std::set<std::string> dependent_blob_uuids; |
+ bool contains_pending_content = false; |
+ |
+ size_t num_files_with_unknown_size = 0; |
+ size_t num_building_dependent_blobs = 0; |
+ |
+ base::CheckedNumeric<uint64_t> checked_total_size = 0; |
+ base::CheckedNumeric<uint64_t> checked_memory_quota_needed = 0; |
+ base::CheckedNumeric<uint64_t> checked_file_quota_needed = 0; |
+ |
+ for (scoped_refptr<BlobDataItem> input_item : input_builder.items_) { |
+ const DataElement& input_element = input_item->data_element(); |
+ DataElement::Type type = input_element.type(); |
+ uint64_t length = input_element.length(); |
+ |
+ if (IsBytes(type)) { |
+ DCHECK_NE(0 + DataElement::kUnknownSize, input_element.length()); |
+ contains_pending_content |= type == DataElement::TYPE_BYTES_DESCRIPTION; |
+ checked_memory_quota_needed += length; |
+ checked_total_size += length; |
+ scoped_refptr<ShareableBlobDataItem> item = new ShareableBlobDataItem( |
+ std::move(input_item), ShareableBlobDataItem::QUOTA_NEEDED); |
+ pending_memory_items.push_back(item); |
+ user_items.push_back(item.get()); |
+ output_blob->AppendSharedBlobItem(uuid, std::move(item)); |
+ continue; |
+ } |
+ if (type == DataElement::TYPE_BLOB) { |
+ InternalBlobData* ref_entry = |
+ registry->GetEntry(input_element.blob_uuid()); |
+ |
+ if (!ref_entry || input_element.blob_uuid() == uuid) { |
+ status = BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS; |
+ return; |
+ } |
+ |
+ if (BlobStatusIsError(ref_entry->status())) { |
+ status = BlobStatus::ERR_REFERENCED_BLOB_BROKEN; |
+ return; |
+ } |
+ |
+ if (ref_entry->total_size() == DataElement::kUnknownSize) { |
+ // We can't reference a blob with unknown size. |
+ status = BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS; |
+ return; |
+ } |
+ |
+ if (dependent_blob_uuids.find(input_element.blob_uuid()) == |
+ dependent_blob_uuids.end()) { |
+ dependent_blobs.push_back( |
+ std::make_pair(input_element.blob_uuid(), ref_entry)); |
+ dependent_blob_uuids.insert(input_element.blob_uuid()); |
+ if (BlobStatusIsPending(ref_entry->status())) { |
+ num_building_dependent_blobs++; |
+ } |
+ } |
+ |
+ length = length == DataElement::kUnknownSize ? ref_entry->total_size() |
+ : input_element.length(); |
+ checked_total_size += length; |
+ |
+ // If we're referencing the whole blob, then we don't need to slice. |
+ if (input_element.offset() == 0 && length == ref_entry->total_size()) { |
+ for (const auto& shareable_item : ref_entry->items()) { |
+ output_blob->AppendSharedBlobItem(uuid, shareable_item); |
+ } |
+ continue; |
+ } |
+ |
+ // Validate our reference has good offset & length. |
+ if (input_element.offset() + length > ref_entry->total_size()) { |
+ status = BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS; |
+ return; |
+ } |
+ |
+ BlobSlice slice(*ref_entry, input_element.offset(), length); |
+ |
+ if (slice.first_source_item) { |
+ copies.push_back(ItemCopyEntry(slice.first_source_item, |
+ slice.first_item_slice_offset, |
+ slice.dest_items.front())); |
+ pending_memory_items.push_back(slice.dest_items.front()); |
+ } |
+ if (slice.last_source_item) { |
+ copies.push_back( |
+ ItemCopyEntry(slice.last_source_item, 0, slice.dest_items.back())); |
+ pending_memory_items.push_back(slice.dest_items.back()); |
+ } |
+ checked_memory_quota_needed += slice.copying_memory_size; |
+ |
+ for (auto& shareable_item : slice.dest_items) { |
+ output_blob->AppendSharedBlobItem(uuid, std::move(shareable_item)); |
+ } |
+ continue; |
+ } |
+ |
+ // If the source item is a temporary file item, then we need to keep track |
+ // of that and mark is as needing quota. |
+ scoped_refptr<ShareableBlobDataItem> item; |
+ if (type == DataElement::TYPE_FILE && |
+ BlobDataBuilder::IsFutureFileItem(input_element)) { |
+ item = new ShareableBlobDataItem(std::move(input_item), |
+ ShareableBlobDataItem::QUOTA_NEEDED); |
+ contains_pending_content = true; |
+ pending_file_items.push_back(item); |
+ user_items.push_back(item.get()); |
+ checked_file_quota_needed += length; |
+ } else { |
+ item = new ShareableBlobDataItem( |
+ std::move(input_item), |
+ ShareableBlobDataItem::POPULATED_WITHOUT_QUOTA); |
+ } |
+ if (length == DataElement::kUnknownSize) |
+ num_files_with_unknown_size++; |
+ |
+ checked_total_size += length; |
+ output_blob->AppendSharedBlobItem(uuid, std::move(item)); |
+ } |
+ |
+ if (num_files_with_unknown_size > 1 && input_builder.items_.size() > 1) { |
+ LOG(ERROR) << "We only allow an unknown size when it's a single file item"; |
+ status = BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS; |
+ return; |
+ } |
+ if (!checked_total_size.IsValid() || !checked_memory_quota_needed.IsValid() || |
+ !checked_file_quota_needed.IsValid()) { |
+ LOG(ERROR) << "Blob sizes aren't valid."; |
+ status = BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS; |
+ return; |
+ } |
+ total_size = checked_total_size.ValueOrDie(); |
+ memory_quota_needed = checked_memory_quota_needed.ValueOrDie(); |
+ file_quota_needed = checked_file_quota_needed.ValueOrDie(); |
+ if (memory_quota_needed || file_quota_needed) { |
+ status = BlobStatus::PENDING_QUOTA; |
+ } else if (contains_pending_content) { |
+ status = BlobStatus::PENDING_TRANSPORT; |
+ } else { |
+ // We either have dependent blobs or we're done. Either way, we're still |
+ // pending final construction. |
+ status = BlobStatus::PENDING_INTERNALS; |
+ } |
+} |
+ |
+BlobStorageContext::BlobFlattener::~BlobFlattener() {} |
+ |
+BlobStorageContext::BlobSlice::BlobSlice(const InternalBlobData& source, |
+ uint64_t slice_offset, |
+ uint64_t slice_size) { |
+ const auto& source_items = source.items(); |
+ const auto& offsets = source.offsets(); |
+ LOG(ERROR) << "doing a slice at " << slice_offset << " with size " |
+ << slice_size; |
+ DCHECK_LE(slice_offset + slice_size, source.total_size()); |
+ size_t item_index = |
+ std::upper_bound(offsets.begin(), offsets.end(), slice_offset) - |
+ offsets.begin(); |
+ uint64_t item_offset = |
+ item_index == 0 ? slice_offset : slice_offset - offsets[item_index - 1]; |
+ size_t num_items = source_items.size(); |
+ |
+ size_t first_item_index = item_index; |
+ copying_memory_size = 0; |
+ |
+ // Read starting from 'first_item_index' and 'item_offset'. |
+ for (uint64_t total_sliced = 0; |
+ item_index < num_items && total_sliced < slice_size; item_index++) { |
+ const scoped_refptr<BlobDataItem>& source_item = |
+ source_items[item_index]->item(); |
+ uint64_t source_length = source_item->length(); |
+ DCHECK_NE(source_length, std::numeric_limits<uint64_t>::max()); |
+ DCHECK_NE(source_length, 0ull); |
+ |
+ uint64_t read_size = |
+ std::min(source_length - item_offset, slice_size - total_sliced); |
+ total_sliced += read_size; |
+ |
+ if (read_size == source_length) { |
+ // We can share the entire item. |
+ LOG(ERROR) << "we can share"; |
+ dest_items.push_back(source_items[item_index]); |
+ continue; |
+ } |
+ |
+ scoped_refptr<BlobDataItem> data_item; |
+ ShareableBlobDataItem::State state = |
+ ShareableBlobDataItem::POPULATED_WITHOUT_QUOTA; |
+ switch (source_item->type()) { |
+ case DataElement::TYPE_BYTES_DESCRIPTION: |
+ case DataElement::TYPE_BYTES: { |
+ if (item_index == first_item_index) { |
+ first_item_slice_offset = item_offset; |
+ first_source_item = source_items[item_index]; |
+ } else { |
+ last_source_item = source_items[item_index]; |
+ } |
+ copying_memory_size += read_size; |
+ // Since we don't have quota yet for memory, we create temporary items |
+ // for this data. When our blob is finished constructing, all dependent |
+ // blobs are done, and we have enough memory quota, we'll copy the data |
+ // over. |
+ std::unique_ptr<DataElement> element(new DataElement()); |
+ element->SetToBytesDescription(base::checked_cast<size_t>(read_size)); |
+ data_item = new BlobDataItem(std::move(element)); |
+ state = ShareableBlobDataItem::QUOTA_NEEDED; |
+ break; |
+ } |
+ case DataElement::TYPE_FILE: { |
+ std::unique_ptr<DataElement> element(new DataElement()); |
+ element->SetToFilePathRange( |
+ source_item->path(), source_item->offset() + item_offset, read_size, |
+ source_item->expected_modification_time()); |
+ data_item = |
+ new BlobDataItem(std::move(element), source_item->data_handle_); |
+ |
+ if (BlobDataBuilder::IsFutureFileItem(source_item->data_element())) { |
+ // Since we don't have file path / reference for our future file, we |
+ // create another file item with this temporary file name. When our |
+ // blob is finished constructing, all dependent blobs are done, and we |
+ // can copy the handle over. |
+ LOG(ERROR) << "we're slicing a temp file item!"; |
+ if (item_index == first_item_index) { |
+ first_item_slice_offset = item_offset; |
+ first_source_item = source_items[item_index]; |
+ } else { |
+ last_source_item = source_items[item_index]; |
+ } |
+ state = ShareableBlobDataItem::QUOTA_NEEDED; |
+ } |
+ break; |
+ } |
+ case DataElement::TYPE_FILE_FILESYSTEM: { |
+ std::unique_ptr<DataElement> element(new DataElement()); |
+ element->SetToFileSystemUrlRange( |
+ source_item->filesystem_url(), source_item->offset() + item_offset, |
+ read_size, source_item->expected_modification_time()); |
+ data_item = new BlobDataItem(std::move(element)); |
+ break; |
+ } |
+ case DataElement::TYPE_DISK_CACHE_ENTRY: { |
+ std::unique_ptr<DataElement> element(new DataElement()); |
+ element->SetToDiskCacheEntryRange(source_item->offset() + item_offset, |
+ read_size); |
+ data_item = |
+ new BlobDataItem(std::move(element), source_item->data_handle_, |
+ source_item->disk_cache_entry(), |
+ source_item->disk_cache_stream_index(), |
+ source_item->disk_cache_side_stream_index()); |
+ break; |
+ } |
+ case DataElement::TYPE_BLOB: |
+ case DataElement::TYPE_UNKNOWN: |
+ CHECK(false) << "Illegal blob item type: " << source_item->type(); |
+ } |
+ dest_items.push_back( |
+ new ShareableBlobDataItem(std::move(data_item), state)); |
+ item_offset = 0; |
+ } |
+} |
+ |
+BlobStorageContext::BlobSlice::~BlobSlice() {} |
-BlobStorageContext::BlobStorageContext() : memory_usage_(0) {} |
+BlobStorageContext::BlobStorageContext() |
+ : memory_controller_(base::FilePath(), scoped_refptr<base::TaskRunner>()), |
+ ptr_factory_(this) {} |
+BlobStorageContext::BlobStorageContext( |
+ const base::FilePath& storage_directory, |
+ scoped_refptr<base::TaskRunner> file_runner) |
+ : memory_controller_(storage_directory, file_runner), ptr_factory_(this) {} |
BlobStorageContext::~BlobStorageContext() { |
} |
std::unique_ptr<BlobDataHandle> BlobStorageContext::GetBlobDataFromUUID( |
const std::string& uuid) { |
- BlobRegistryEntry* entry = registry_.GetEntry(uuid); |
- if (!entry) { |
+ InternalBlobData* entry = registry_.GetEntry(uuid); |
+ if (!entry) |
return nullptr; |
- } |
- return base::WrapUnique( |
- new BlobDataHandle(uuid, entry->content_type, entry->content_disposition, |
- this, base::ThreadTaskRunnerHandle::Get().get())); |
+ return CreateHandle(uuid, entry); |
} |
std::unique_ptr<BlobDataHandle> BlobStorageContext::GetBlobDataFromPublicURL( |
const GURL& url) { |
std::string uuid; |
- BlobRegistryEntry* entry = registry_.GetEntryFromURL(url, &uuid); |
- if (!entry) { |
+ InternalBlobData* entry = registry_.GetEntryFromURL(url, &uuid); |
+ if (!entry) |
return nullptr; |
- } |
- return base::WrapUnique( |
- new BlobDataHandle(uuid, entry->content_type, entry->content_disposition, |
- this, base::ThreadTaskRunnerHandle::Get().get())); |
+ return CreateHandle(uuid, entry); |
} |
std::unique_ptr<BlobDataHandle> BlobStorageContext::AddFinishedBlob( |
const BlobDataBuilder& external_builder) { |
TRACE_EVENT0("Blob", "Context::AddFinishedBlob"); |
- CreatePendingBlob(external_builder.uuid(), external_builder.content_type_, |
- external_builder.content_disposition_); |
- CompletePendingBlob(external_builder); |
- std::unique_ptr<BlobDataHandle> handle = |
- GetBlobDataFromUUID(external_builder.uuid_); |
- DecrementBlobRefCount(external_builder.uuid_); |
- return handle; |
+ return BuildBlob(external_builder, PopulatationAllowedCallback()); |
} |
std::unique_ptr<BlobDataHandle> BlobStorageContext::AddFinishedBlob( |
@@ -78,123 +349,201 @@ std::unique_ptr<BlobDataHandle> BlobStorageContext::AddFinishedBlob( |
return AddFinishedBlob(*builder); |
} |
+std::unique_ptr<BlobDataHandle> BlobStorageContext::AddBrokenBlob( |
+ const std::string& uuid, |
+ const std::string& content_type, |
+ const std::string& content_disposition, |
+ BlobStatus reason) { |
+ DCHECK(!registry_.HasEntry(uuid)); |
+ DCHECK(BlobStatusIsError(reason)); |
+ InternalBlobData* entry = |
+ registry_.CreateEntry(uuid, content_type, content_disposition); |
+ entry->status_ = reason; |
+ FinishBuilding(entry); |
+ return CreateHandle(uuid, entry); |
+} |
+ |
bool BlobStorageContext::RegisterPublicBlobURL(const GURL& blob_url, |
const std::string& uuid) { |
- if (!registry_.CreateUrlMapping(blob_url, uuid)) { |
+ if (!registry_.CreateUrlMapping(blob_url, uuid)) |
return false; |
- } |
IncrementBlobRefCount(uuid); |
return true; |
} |
void BlobStorageContext::RevokePublicBlobURL(const GURL& blob_url) { |
std::string uuid; |
- if (!registry_.DeleteURLMapping(blob_url, &uuid)) { |
+ if (!registry_.DeleteURLMapping(blob_url, &uuid)) |
return; |
- } |
DecrementBlobRefCount(uuid); |
} |
-void BlobStorageContext::CreatePendingBlob( |
- const std::string& uuid, |
- const std::string& content_type, |
- const std::string& content_disposition) { |
- DCHECK(!registry_.GetEntry(uuid) && !uuid.empty()); |
- registry_.CreateEntry(uuid, content_type, content_disposition); |
-} |
+std::unique_ptr<BlobDataHandle> BlobStorageContext::BuildBlob( |
+ const BlobDataBuilder& content, |
+ const PopulatationAllowedCallback& can_populate_memory) { |
+ DCHECK(!registry_.HasEntry(content.uuid_)); |
-void BlobStorageContext::CompletePendingBlob( |
- const BlobDataBuilder& external_builder) { |
- BlobRegistryEntry* entry = registry_.GetEntry(external_builder.uuid()); |
- DCHECK(entry); |
- DCHECK(!entry->data.get()) << "Blob already constructed: " |
- << external_builder.uuid(); |
- // We want to handle storing our broken blob as well. |
- switch (entry->state) { |
- case BlobState::PENDING: { |
- entry->data_builder.reset(new InternalBlobData::Builder()); |
- InternalBlobData::Builder* internal_data_builder = |
- entry->data_builder.get(); |
- |
- bool broken = false; |
- for (const auto& blob_item : external_builder.items_) { |
- IPCBlobCreationCancelCode error_code; |
- if (!AppendAllocatedBlobItem(external_builder.uuid_, blob_item, |
- internal_data_builder, &error_code)) { |
- broken = true; |
- memory_usage_ -= entry->data_builder->GetNonsharedMemoryUsage(); |
- entry->state = BlobState::BROKEN; |
- entry->broken_reason = error_code; |
- entry->data_builder.reset(new InternalBlobData::Builder()); |
- break; |
- } |
- } |
- entry->data = entry->data_builder->Build(); |
- entry->data_builder.reset(); |
- entry->state = broken ? BlobState::BROKEN : BlobState::COMPLETE; |
- break; |
- } |
- case BlobState::BROKEN: { |
- InternalBlobData::Builder builder; |
- entry->data = builder.Build(); |
- break; |
+ InternalBlobData* entry = registry_.CreateEntry( |
+ content.uuid_, content.content_type_, content.content_disposition_); |
+ |
+ // This flattens all blob references in the transportion content out and |
+ // stores the complete item representation in the internal data. |
+ BlobFlattener flattener(content, entry, ®istry_); |
+ |
+ // If we contain invalid references we're invalid input. |
+ if (flattener.status == BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS) { |
+ LOG(ERROR) << "invalid construction args"; |
+ BreakAndFinishPendingBlob(content.uuid_, |
+ BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS); |
+ return CreateHandle(content.uuid_, entry); |
+ } |
+ |
+ // If we contain broken references (or we reference ourself) we want to fail. |
+ if (flattener.status == BlobStatus::ERR_REFERENCED_BLOB_BROKEN) { |
+ LOG(ERROR) << "references don't exist"; |
+ BreakAndFinishPendingBlob(content.uuid_, |
+ BlobStatus::ERR_REFERENCED_BLOB_BROKEN); |
+ return CreateHandle(content.uuid_, entry); |
+ } |
+ |
+ LOG(ERROR) << "We need " << flattener.memory_quota_needed |
+ << " memory quota, and " << flattener.file_quota_needed |
+ << " file quota."; |
+ |
+ if (!memory_controller_.CanReserveQuota(flattener.memory_quota_needed + |
+ flattener.file_quota_needed)) { |
+ LOG(ERROR) << "we can't fit."; |
+ BreakAndFinishPendingBlob(content.uuid_, BlobStatus::ERR_OUT_OF_MEMORY); |
+ return CreateHandle(content.uuid_, entry); |
+ } |
+ |
+ entry->status_ = flattener.status; |
+ entry->size_ = flattener.total_size; |
+ std::unique_ptr<BlobDataHandle> handle = CreateHandle(content.uuid_, entry); |
+ |
+ size_t num_building_dependent_blobs = 0; |
+ std::vector<std::unique_ptr<BlobDataHandle>> dependent_blobs; |
+ // We hold a handle to all blobs we're using. This is important, as our memory |
+ // accounting can be delayed until OnEnoughSizeForBlobData is called, and we |
+ // only free memory on canceling when we've done this accounting. If a |
+ // dependent blob is dereferenced, then we're the last blob holding onto that |
+ // data item, and we need to account for that. So we prevent that case by |
+ // holding onto all blobs. |
+ for (const std::pair<std::string, InternalBlobData*>& pending_blob : |
+ flattener.dependent_blobs) { |
+ LOG(ERROR) << "adding dependent blob"; |
+ dependent_blobs.push_back( |
+ CreateHandle(pending_blob.first, pending_blob.second)); |
+ if (BlobStatusIsPending(pending_blob.second->status())) { |
+ pending_blob.second->building_state_->build_completion_callbacks |
+ .push_back(base::Bind(&BlobStorageContext::OnDependentBlobFinished, |
+ ptr_factory_.GetWeakPtr(), content.uuid_)); |
+ num_building_dependent_blobs++; |
} |
- case BlobState::COMPLETE: |
- DCHECK(false) << "Blob already constructed: " << external_builder.uuid(); |
- return; |
} |
- UMA_HISTOGRAM_COUNTS("Storage.Blob.ItemCount", entry->data->items().size()); |
- UMA_HISTOGRAM_BOOLEAN("Storage.Blob.Broken", |
- entry->state == BlobState::BROKEN); |
- if (entry->state == BlobState::BROKEN) { |
- UMA_HISTOGRAM_ENUMERATION( |
- "Storage.Blob.BrokenReason", static_cast<int>(entry->broken_reason), |
- (static_cast<int>(IPCBlobCreationCancelCode::LAST) + 1)); |
+ DCHECK(flattener.status != BlobStatus::PENDING_TRANSPORT || |
+ !can_populate_memory) |
+ << "There is no pending content for the user to populate, so the " |
+ "callback should be null."; |
+ DCHECK(flattener.status != BlobStatus::PENDING_TRANSPORT || |
+ can_populate_memory) |
+ << "If we have pending content then there needs to be a callback " |
+ "present."; |
+ |
+ TransportState transport_state = TransportState::DONE; |
+ if (!flattener.user_items.empty()) { |
+ transport_state = flattener.file_quota_needed > 0 |
+ ? TransportState::PENDING_FILE_QUOTA |
+ : TransportState::PENDING_MEMORY_QUOTA; |
} |
- size_t total_memory = 0, nonshared_memory = 0; |
- entry->data->GetMemoryUsage(&total_memory, &nonshared_memory); |
- UMA_HISTOGRAM_COUNTS("Storage.Blob.TotalSize", total_memory / 1024); |
- UMA_HISTOGRAM_COUNTS("Storage.Blob.TotalUnsharedSize", |
- nonshared_memory / 1024); |
- TRACE_COUNTER1("Blob", "MemoryStoreUsageBytes", memory_usage_); |
- auto runner = base::ThreadTaskRunnerHandle::Get(); |
- for (const auto& callback : entry->build_completion_callbacks) { |
- runner->PostTask(FROM_HERE, |
- base::Bind(callback, entry->state == BlobState::COMPLETE, |
- entry->broken_reason)); |
+ entry->building_state_.reset(new InternalBlobData::BuildingState( |
+ transport_state, |
+ /* user_data_population_callback */ can_populate_memory, |
+ num_building_dependent_blobs, |
+ /* memory_quota_needed */ flattener.memory_quota_needed > 0, |
+ /* file_quota_needed */ flattener.file_quota_needed > 0)); |
+ |
+ InternalBlobData::BuildingState* building_state = |
+ entry->building_state_.get(); |
+ std::swap(building_state->copies, flattener.copies); |
+ std::swap(building_state->dependent_blobs, dependent_blobs); |
+ std::swap(building_state->user_items, flattener.user_items); |
+ |
+ if (flattener.memory_quota_needed) { |
+ // We can finish our blob before returning from this method. |
+ base::WeakPtr<QuotaAllocationTask> pending_request = |
+ memory_controller_.ReserveMemoryQuota( |
+ std::move(flattener.pending_memory_items), |
+ base::Bind(&BlobStorageContext::OnEnoughSizeForMemory, |
+ ptr_factory_.GetWeakPtr(), content.uuid_)); |
+ if (entry->building_state_) |
+ entry->building_state_->memory_quota_request = std::move(pending_request); |
} |
- entry->build_completion_callbacks.clear(); |
+ |
+ if (flattener.file_quota_needed) { |
+ // We can finish our blob before returning from this method. |
+ base::WeakPtr<QuotaAllocationTask> pending_request = |
+ memory_controller_.ReserveFileQuota( |
+ std::move(flattener.pending_file_items), |
+ base::Bind(&BlobStorageContext::OnEnoughSizeForTransportFiles, |
+ ptr_factory_.GetWeakPtr(), content.uuid_)); |
+ if (entry->building_state_) |
+ entry->building_state_->file_quota_request = std::move(pending_request); |
+ } |
+ |
+ if (entry->CanFinishBuilding()) |
+ FinishBuilding(entry); |
+ |
+ return handle; |
} |
-void BlobStorageContext::CancelPendingBlob(const std::string& uuid, |
- IPCBlobCreationCancelCode reason) { |
- BlobRegistryEntry* entry = registry_.GetEntry(uuid); |
- DCHECK(entry && entry->state == BlobState::PENDING); |
- entry->state = BlobState::BROKEN; |
- entry->broken_reason = reason; |
- CompletePendingBlob(BlobDataBuilder(uuid)); |
+void BlobStorageContext::BreakAndFinishPendingBlob(const std::string& uuid, |
+ BlobStatus reason) { |
+ InternalBlobData* entry = registry_.GetEntry(uuid); |
+ DCHECK(entry); |
+ DCHECK(BlobStatusIsError(reason)); |
+ PopulatationAllowedCallback user_data_population_callback; |
+ if (entry->building_state_ && |
+ entry->building_state_->user_data_population_callback) { |
+ user_data_population_callback = |
+ entry->building_state_->user_data_population_callback; |
+ entry->building_state_->user_data_population_callback.Reset(); |
+ } |
+ ClearAndFreeMemory(uuid, entry); |
+ entry->status_ = reason; |
+ if (user_data_population_callback) { |
+ user_data_population_callback.Run( |
+ reason, std::vector<BlobMemoryController::FileCreationInfo>()); |
+ } |
+ entry->items_.clear(); |
+ entry->offsets_.clear(); |
+ entry->size_ = 0; |
+ FinishBuilding(entry); |
+} |
+ |
+void BlobStorageContext::FinishedPopulatingPendingBlob( |
+ const std::string& uuid) { |
+ InternalBlobData* entry = registry_.GetEntry(uuid); |
+ CHECK(entry) << "There is no blob entry with uuid " << uuid; |
+ DCHECK(BlobStatusIsPending(entry->status())); |
+ FinishedPopulatingPendingBlob(entry); |
} |
void BlobStorageContext::IncrementBlobRefCount(const std::string& uuid) { |
- BlobRegistryEntry* entry = registry_.GetEntry(uuid); |
+ InternalBlobData* entry = registry_.GetEntry(uuid); |
DCHECK(entry); |
- ++(entry->refcount); |
+ ++(entry->refcount_); |
} |
void BlobStorageContext::DecrementBlobRefCount(const std::string& uuid) { |
- BlobRegistryEntry* entry = registry_.GetEntry(uuid); |
+ InternalBlobData* entry = registry_.GetEntry(uuid); |
DCHECK(entry); |
- DCHECK_GT(entry->refcount, 0u); |
- if (--(entry->refcount) == 0) { |
- size_t memory_freeing = 0; |
- if (entry->state == BlobState::COMPLETE) { |
- memory_freeing = entry->data->GetUnsharedMemoryUsage(); |
- entry->data->RemoveBlobFromShareableItems(uuid); |
- } |
- DCHECK_LE(memory_freeing, memory_usage_); |
- memory_usage_ -= memory_freeing; |
+ DCHECK_GT(entry->refcount_, 0u); |
+ if (--(entry->refcount_) == 0) { |
+ LOG(ERROR) << "Clearing blob " << uuid; |
+ ClearAndFreeMemory(uuid, entry); |
registry_.DeleteEntry(uuid); |
} |
} |
@@ -202,263 +551,249 @@ void BlobStorageContext::DecrementBlobRefCount(const std::string& uuid) { |
std::unique_ptr<BlobDataSnapshot> BlobStorageContext::CreateSnapshot( |
const std::string& uuid) { |
std::unique_ptr<BlobDataSnapshot> result; |
- BlobRegistryEntry* entry = registry_.GetEntry(uuid); |
- if (entry->state != BlobState::COMPLETE) { |
+ InternalBlobData* entry = registry_.GetEntry(uuid); |
+ if (entry->status() != BlobStatus::DONE) |
return result; |
- } |
- const InternalBlobData& data = *entry->data; |
std::unique_ptr<BlobDataSnapshot> snapshot(new BlobDataSnapshot( |
- uuid, entry->content_type, entry->content_disposition)); |
- snapshot->items_.reserve(data.items().size()); |
- for (const auto& shareable_item : data.items()) { |
+ uuid, entry->content_type(), entry->content_disposition())); |
+ snapshot->items_.reserve(entry->items().size()); |
+ for (const auto& shareable_item : entry->items()) { |
snapshot->items_.push_back(shareable_item->item()); |
} |
+ memory_controller_.NotifyMemoryItemsUsed(entry->items()); |
return snapshot; |
} |
-bool BlobStorageContext::IsBroken(const std::string& uuid) const { |
- const BlobRegistryEntry* entry = registry_.GetEntry(uuid); |
- if (!entry) { |
- return true; |
- } |
- return entry->state == BlobState::BROKEN; |
-} |
- |
-bool BlobStorageContext::IsBeingBuilt(const std::string& uuid) const { |
- const BlobRegistryEntry* entry = registry_.GetEntry(uuid); |
- if (!entry) { |
- return false; |
- } |
- return entry->state == BlobState::PENDING; |
+BlobStatus BlobStorageContext::GetBlobStatus(const std::string& uuid) const { |
+ const InternalBlobData* entry = registry_.GetEntry(uuid); |
+ if (!entry) |
+ return BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS; |
+ return entry->status(); |
} |
void BlobStorageContext::RunOnConstructionComplete( |
const std::string& uuid, |
- const BlobConstructedCallback& done) { |
- BlobRegistryEntry* entry = registry_.GetEntry(uuid); |
+ const BlobStatusCallback& done) { |
+ InternalBlobData* entry = registry_.GetEntry(uuid); |
DCHECK(entry); |
- switch (entry->state) { |
- case BlobState::COMPLETE: |
- done.Run(true, IPCBlobCreationCancelCode::UNKNOWN); |
- return; |
- case BlobState::BROKEN: |
- done.Run(false, entry->broken_reason); |
- return; |
- case BlobState::PENDING: |
- entry->build_completion_callbacks.push_back(done); |
- return; |
+ if (BlobStatusIsPending(entry->status())) { |
+ entry->building_state_->build_completion_callbacks.push_back(done); |
+ return; |
} |
- NOTREACHED(); |
+ done.Run(entry->status()); |
} |
-bool BlobStorageContext::AppendAllocatedBlobItem( |
- const std::string& target_blob_uuid, |
- scoped_refptr<BlobDataItem> blob_item, |
- InternalBlobData::Builder* target_blob_builder, |
- IPCBlobCreationCancelCode* error_code) { |
- DCHECK(error_code); |
- *error_code = IPCBlobCreationCancelCode::UNKNOWN; |
- bool error = false; |
- |
- // The blob data is stored in the canonical way which only contains a |
- // list of Data, File, and FileSystem items. Aggregated TYPE_BLOB items |
- // are expanded into the primitive constituent types and reused if possible. |
- // 1) The Data item is denoted by the raw data and length. |
- // 2) The File item is denoted by the file path, the range and the expected |
- // modification time. |
- // 3) The FileSystem File item is denoted by the FileSystem URL, the range |
- // and the expected modification time. |
- // 4) The Blob item is denoted by the source blob and an offset and size. |
- // Internal items that are fully used by the new blob (not cut by the |
- // offset or size) are shared between the blobs. Otherwise, the relevant |
- // portion of the item is copied. |
- |
- DCHECK(blob_item->data_element_ptr()); |
- const DataElement& data_element = blob_item->data_element(); |
- uint64_t length = data_element.length(); |
- uint64_t offset = data_element.offset(); |
- UMA_HISTOGRAM_COUNTS("Storage.Blob.StorageSizeBeforeAppend", |
- memory_usage_ / 1024); |
- switch (data_element.type()) { |
- case DataElement::TYPE_BYTES: |
- UMA_HISTOGRAM_COUNTS("Storage.BlobItemSize.Bytes", length / 1024); |
- DCHECK(!offset); |
- if (memory_usage_ + length > kBlobStorageMaxMemoryUsage) { |
- error = true; |
- *error_code = IPCBlobCreationCancelCode::OUT_OF_MEMORY; |
- break; |
- } |
- memory_usage_ += length; |
- target_blob_builder->AppendSharedBlobItem(new ShareableBlobDataItem( |
- target_blob_uuid, blob_item, |
- ShareableBlobDataItem::POPULATED_WITH_QUOTA)); |
- break; |
- case DataElement::TYPE_FILE: { |
- bool full_file = (length == std::numeric_limits<uint64_t>::max()); |
- UMA_HISTOGRAM_BOOLEAN("Storage.BlobItemSize.File.Unknown", full_file); |
- if (!full_file) { |
- UMA_HISTOGRAM_COUNTS("Storage.BlobItemSize.File", |
- (length - offset) / 1024); |
- } |
- target_blob_builder->AppendSharedBlobItem(new ShareableBlobDataItem( |
- target_blob_uuid, blob_item, |
- ShareableBlobDataItem::POPULATED_WITHOUT_QUOTA)); |
- break; |
- } |
- case DataElement::TYPE_FILE_FILESYSTEM: { |
- bool full_file = (length == std::numeric_limits<uint64_t>::max()); |
- UMA_HISTOGRAM_BOOLEAN("Storage.BlobItemSize.FileSystem.Unknown", |
- full_file); |
- if (!full_file) { |
- UMA_HISTOGRAM_COUNTS("Storage.BlobItemSize.FileSystem", |
- (length - offset) / 1024); |
- } |
- target_blob_builder->AppendSharedBlobItem(new ShareableBlobDataItem( |
- target_blob_uuid, blob_item, |
- ShareableBlobDataItem::POPULATED_WITHOUT_QUOTA)); |
- break; |
- } |
- case DataElement::TYPE_BLOB: { |
- UMA_HISTOGRAM_COUNTS("Storage.BlobItemSize.Blob", |
- (length - offset) / 1024); |
- // We grab the handle to ensure it stays around while we copy it. |
- std::unique_ptr<BlobDataHandle> src = |
- GetBlobDataFromUUID(data_element.blob_uuid()); |
- if (!src || src->IsBroken() || src->IsBeingBuilt()) { |
- error = true; |
- *error_code = IPCBlobCreationCancelCode::REFERENCED_BLOB_BROKEN; |
- break; |
- } |
- BlobRegistryEntry* other_entry = |
- registry_.GetEntry(data_element.blob_uuid()); |
- DCHECK(other_entry->data); |
- if (!AppendBlob(target_blob_uuid, *other_entry->data, offset, length, |
- target_blob_builder)) { |
- error = true; |
- *error_code = IPCBlobCreationCancelCode::OUT_OF_MEMORY; |
+std::unique_ptr<BlobDataHandle> BlobStorageContext::CreateHandle( |
+ const std::string& uuid, |
+ InternalBlobData* entry) { |
+ return base::WrapUnique(new BlobDataHandle( |
+ uuid, entry->content_type_, entry->content_disposition_, entry->size_, |
+ this, base::ThreadTaskRunnerHandle::Get().get())); |
+} |
+ |
+void BlobStorageContext::FinishedPopulatingPendingBlob(InternalBlobData* entry) { |
+ |
+ SetItemStateToPopulated(&entry->building_state_->user_items); |
+ entry->building_state_->transport_state = TransportState::DONE; |
+ entry->status_ = BlobStatus::PENDING_INTERNALS; |
+ if (entry->CanFinishBuilding()) |
+ FinishBuilding(entry); |
+} |
+ |
+void BlobStorageContext::FinishBuilding(InternalBlobData* entry) { |
+ DCHECK(entry); |
+ |
+ if (BlobStatusIsPending(entry->status_)) { |
+ for (const ItemCopyEntry& copy : entry->building_state_->copies) { |
+ // Our source item can be a file if it was a slice of an unpopulated file, |
+ // or a slice of data that was then paged to disk. |
+ size_t dest_size = static_cast<size_t>(copy.dest_item->item()->length()); |
+ DataElement::Type dest_type = copy.dest_item->item()->type(); |
+ switch (copy.source_item->item()->type()) { |
+ case DataElement::TYPE_BYTES: { |
+ DCHECK_EQ(dest_type, DataElement::TYPE_BYTES_DESCRIPTION); |
+ const char* src_data = |
+ copy.source_item->item()->bytes() + copy.source_item_offset; |
+ copy.dest_item->item()->item_->SetToBytes(src_data, dest_size); |
+ } break; |
+ case DataElement::TYPE_FILE: { |
+ LOG(ERROR) << "Source item has been paged or we started as a file, " |
+ "so we're grabbing file ref " |
+ << dest_type; |
+ |
+ // If we expected a memory item (and our source was paged to disk) we |
+ // free that memory. |
+ if (dest_type == DataElement::TYPE_BYTES_DESCRIPTION) |
+ copy.dest_item->set_memory_allocation(nullptr); |
+ |
+ const DataElement& source_element = |
+ copy.source_item->item()->data_element(); |
+ std::unique_ptr<DataElement> new_element(new DataElement()); |
+ new_element->SetToFilePathRange( |
+ source_element.path(), |
+ source_element.offset() + copy.source_item_offset, dest_size, |
+ source_element.expected_modification_time()); |
+ scoped_refptr<BlobDataItem> new_item(new BlobDataItem( |
+ std::move(new_element), copy.source_item->item()->data_handle_)); |
+ copy.dest_item->set_item(std::move(new_item)); |
+ } break; |
+ case DataElement::TYPE_UNKNOWN: |
+ case DataElement::TYPE_BLOB: |
+ case DataElement::TYPE_BYTES_DESCRIPTION: |
+ case DataElement::TYPE_FILE_FILESYSTEM: |
+ case DataElement::TYPE_DISK_CACHE_ENTRY: |
+ NOTREACHED(); |
+ break; |
} |
- break; |
+ copy.dest_item->set_state(ShareableBlobDataItem::POPULATED_WITH_QUOTA); |
} |
- case DataElement::TYPE_DISK_CACHE_ENTRY: { |
- UMA_HISTOGRAM_COUNTS("Storage.BlobItemSize.CacheEntry", |
- (length - offset) / 1024); |
- target_blob_builder->AppendSharedBlobItem(new ShareableBlobDataItem( |
- target_blob_uuid, blob_item, |
- ShareableBlobDataItem::POPULATED_WITHOUT_QUOTA)); |
- break; |
+ entry->status_ = BlobStatus::DONE; |
+ } |
+ |
+ LOG(ERROR) << "Finished blob."; |
+ |
+ std::vector<BlobStatusCallback> callbacks; |
+ if (entry->building_state_.get()) { |
+ std::swap(callbacks, entry->building_state_->build_completion_callbacks); |
+ entry->building_state_.reset(); |
+ } |
+ |
+ memory_controller_.NotifyMemoryItemsUsed(entry->items()); |
+ |
+ auto runner = base::ThreadTaskRunnerHandle::Get(); |
+ for (const auto& callback : callbacks) |
+ runner->PostTask(FROM_HERE, base::Bind(callback, entry->status())); |
+ |
+ for (const auto& shareable_item : entry->items()) { |
+ DCHECK_NE(DataElement::TYPE_BYTES_DESCRIPTION, |
+ shareable_item->item()->type()); |
+ if (!shareable_item->IsPopulated()) { |
+ PrintTo(*shareable_item, &LOG_STREAM(ERROR)); |
+ DCHECK(shareable_item->IsPopulated()) << shareable_item->state(); |
} |
- case DataElement::TYPE_BYTES_DESCRIPTION: |
- case DataElement::TYPE_UNKNOWN: |
- NOTREACHED(); |
- break; |
} |
- UMA_HISTOGRAM_COUNTS("Storage.Blob.StorageSizeAfterAppend", |
- memory_usage_ / 1024); |
- return !error; |
} |
-bool BlobStorageContext::AppendBlob( |
- const std::string& target_blob_uuid, |
- const InternalBlobData& blob, |
- uint64_t offset, |
- uint64_t length, |
- InternalBlobData::Builder* target_blob_builder) { |
- DCHECK_GT(length, 0ull); |
- |
- const std::vector<scoped_refptr<ShareableBlobDataItem>>& items = blob.items(); |
- auto iter = items.begin(); |
- if (offset) { |
- for (; iter != items.end(); ++iter) { |
- const BlobDataItem& item = *(iter->get()->item()); |
- if (offset >= item.length()) |
- offset -= item.length(); |
- else |
- break; |
- } |
+void BlobStorageContext::RequestUserPopulation( |
+ InternalBlobData* entry, |
+ std::vector<BlobMemoryController::FileCreationInfo> files) { |
+ LOG(ERROR) << "Checking if we need user population"; |
+ |
+ InternalBlobData::BuildingState* building_state = |
+ entry->building_state_.get(); |
+ if (building_state->user_data_population_callback) { |
+ LOG(ERROR) << "yes, asking user for population"; |
+ PopulatationAllowedCallback user_data_population_callback = |
+ building_state->user_data_population_callback; |
+ building_state->user_data_population_callback.Reset(); |
+ user_data_population_callback.Run(BlobStatus::PENDING_TRANSPORT, |
+ std::move(files)); |
+ return; |
} |
+ LOG(ERROR) << "no populatio gneed"; |
+ DCHECK(files.empty()); |
+ FinishedPopulatingPendingBlob(entry); |
+} |
- for (; iter != items.end() && length > 0; ++iter) { |
- scoped_refptr<ShareableBlobDataItem> shareable_item = iter->get(); |
- const BlobDataItem& item = *(shareable_item->item()); |
- uint64_t item_length = item.length(); |
- DCHECK_GT(item_length, offset); |
- uint64_t current_length = item_length - offset; |
- uint64_t new_length = current_length > length ? length : current_length; |
- |
- bool reusing_blob_item = offset == 0 && new_length == item.length(); |
- UMA_HISTOGRAM_BOOLEAN("Storage.Blob.ReusedItem", reusing_blob_item); |
- if (reusing_blob_item) { |
- shareable_item->referencing_blobs_mutable()->insert(target_blob_uuid); |
- target_blob_builder->AppendSharedBlobItem(shareable_item); |
- length -= new_length; |
- continue; |
- } |
+void BlobStorageContext::OnEnoughSizeForMemory(const std::string& uuid, |
+ bool success) { |
+ if (!success) { |
+ BreakAndFinishPendingBlob(uuid, BlobStatus::ERR_OUT_OF_MEMORY); |
+ return; |
+ } |
+ InternalBlobData* entry = registry_.GetEntry(uuid); |
+ if (!entry || !entry->building_state_.get()) |
+ return; |
+ InternalBlobData::BuildingState& building_state = *entry->building_state_; |
+ building_state.memory_quota_granted = true; |
+ DCHECK(!building_state.memory_quota_request); |
+ |
+ // We can crequest memory quota for both blob copies and transport memory. |
+ if (building_state.transport_state == TransportState::PENDING_MEMORY_QUOTA) { |
+ entry->status_ = BlobStatus::PENDING_TRANSPORT; |
+ building_state.transport_state = TransportState::PENDING_TRANSPORT; |
+ RequestUserPopulation( |
+ entry, std::vector<BlobMemoryController::FileCreationInfo>()); |
+ } else if (building_state.transport_state == TransportState::DONE) { |
+ // Transport data could have been already transported by files. |
+ entry->status_ = BlobStatus::PENDING_INTERNALS; |
+ } |
- // We need to do copying of the items when we have a different offset or |
- // length |
- switch (item.type()) { |
- case DataElement::TYPE_BYTES: { |
- UMA_HISTOGRAM_COUNTS("Storage.BlobItemSize.BlobSlice.Bytes", |
- new_length / 1024); |
- if (memory_usage_ + new_length > kBlobStorageMaxMemoryUsage) { |
- return false; |
- } |
- DCHECK(!item.offset()); |
- std::unique_ptr<DataElement> element(new DataElement()); |
- element->SetToBytes(item.bytes() + offset, |
- static_cast<int64_t>(new_length)); |
- memory_usage_ += new_length; |
- target_blob_builder->AppendSharedBlobItem(new ShareableBlobDataItem( |
- target_blob_uuid, new BlobDataItem(std::move(element)), |
- ShareableBlobDataItem::POPULATED_WITH_QUOTA)); |
- } break; |
- case DataElement::TYPE_FILE: { |
- DCHECK_NE(item.length(), std::numeric_limits<uint64_t>::max()) |
- << "We cannot use a section of a file with an unknown length"; |
- UMA_HISTOGRAM_COUNTS("Storage.BlobItemSize.BlobSlice.File", |
- new_length / 1024); |
- std::unique_ptr<DataElement> element(new DataElement()); |
- element->SetToFilePathRange(item.path(), item.offset() + offset, |
- new_length, |
- item.expected_modification_time()); |
- target_blob_builder->AppendSharedBlobItem(new ShareableBlobDataItem( |
- target_blob_uuid, |
- new BlobDataItem(std::move(element), item.data_handle_), |
- ShareableBlobDataItem::POPULATED_WITHOUT_QUOTA)); |
- } break; |
- case DataElement::TYPE_FILE_FILESYSTEM: { |
- UMA_HISTOGRAM_COUNTS("Storage.BlobItemSize.BlobSlice.FileSystem", |
- new_length / 1024); |
- std::unique_ptr<DataElement> element(new DataElement()); |
- element->SetToFileSystemUrlRange(item.filesystem_url(), |
- item.offset() + offset, new_length, |
- item.expected_modification_time()); |
- target_blob_builder->AppendSharedBlobItem(new ShareableBlobDataItem( |
- target_blob_uuid, new BlobDataItem(std::move(element)), |
- ShareableBlobDataItem::POPULATED_WITHOUT_QUOTA)); |
- } break; |
- case DataElement::TYPE_DISK_CACHE_ENTRY: { |
- std::unique_ptr<DataElement> element(new DataElement()); |
- element->SetToDiskCacheEntryRange(item.offset() + offset, |
- new_length); |
- target_blob_builder->AppendSharedBlobItem(new ShareableBlobDataItem( |
- target_blob_uuid, |
- new BlobDataItem(std::move(element), item.data_handle_, |
- item.disk_cache_entry(), |
- item.disk_cache_stream_index(), |
- item.disk_cache_side_stream_index()), |
- ShareableBlobDataItem::POPULATED_WITHOUT_QUOTA)); |
- } break; |
- case DataElement::TYPE_BYTES_DESCRIPTION: |
- case DataElement::TYPE_BLOB: |
- case DataElement::TYPE_UNKNOWN: |
- CHECK(false) << "Illegal blob item type: " << item.type(); |
+ if (entry->CanFinishBuilding()) |
+ FinishBuilding(entry); |
+} |
+ |
+void BlobStorageContext::OnEnoughSizeForTransportFiles( |
+ const std::string& uuid, |
+ bool success, |
+ std::vector<BlobMemoryController::FileCreationInfo> files) { |
+ if (!success) { |
+ BreakAndFinishPendingBlob(uuid, BlobStatus::ERR_OUT_OF_MEMORY); |
+ return; |
+ } |
+ InternalBlobData* entry = registry_.GetEntry(uuid); |
+ if (!entry || !entry->building_state_.get()) |
+ return; |
+ InternalBlobData::BuildingState& building_state = *entry->building_state_; |
+ building_state.file_quota_granted = true; |
+ DCHECK(building_state.transport_state == TransportState::PENDING_FILE_QUOTA); |
+ DCHECK(!building_state.file_quota_request); |
+ |
+ entry->status_ = BlobStatus::PENDING_TRANSPORT; |
+ building_state.transport_state = TransportState::PENDING_TRANSPORT; |
+ RequestUserPopulation(entry, std::move(files)); |
+ |
+ if (entry->CanFinishBuilding()) |
+ FinishBuilding(entry); |
+} |
+ |
+void BlobStorageContext::OnDependentBlobFinished( |
+ const std::string& owning_blob_uuid, |
+ BlobStatus status) { |
+ InternalBlobData* entry = registry_.GetEntry(owning_blob_uuid); |
+ if (!entry || !entry->building_state_) |
+ return; |
+ |
+ if (BlobStatusIsError(status)) { |
+ DCHECK_NE(BlobStatus::ERR_BLOB_DEREFERENCED_WHILE_BUILDING, status) |
+ << "Referenced blob should never be dereferenced while we " |
+ << "are depending on it, as our system holds a handle."; |
+ BreakAndFinishPendingBlob(owning_blob_uuid, |
+ BlobStatus::ERR_REFERENCED_BLOB_BROKEN); |
+ return; |
+ } |
+ DCHECK_GT(entry->building_state_->num_building_dependent_blobs, 0u); |
+ --entry->building_state_->num_building_dependent_blobs; |
+ |
+ if (entry->CanFinishBuilding()) |
+ FinishBuilding(entry); |
+} |
+ |
+void BlobStorageContext::ClearAndFreeMemory(const std::string& uuid, |
+ InternalBlobData* entry) { |
+ if (entry->building_state_) { |
+ InternalBlobData::BuildingState* building_state = |
+ entry->building_state_.get(); |
+ if (building_state->memory_quota_request) { |
+ building_state->memory_quota_request->Cancel(); |
+ } |
+ if (building_state->file_quota_request) { |
+ building_state->file_quota_request->Cancel(); |
} |
- length -= new_length; |
- offset = 0; |
} |
- return true; |
+ entry->RemoveBlobFromShareableItems(uuid); |
+ |
+ entry->items_.clear(); |
+ entry->offsets_.clear(); |
+} |
+ |
+void BlobStorageContext::SetItemStateToPopulated( |
+ std::vector<ShareableBlobDataItem*>* items) { |
+ for (ShareableBlobDataItem* shareable_item : *items) { |
+ DCHECK(shareable_item->state() == ShareableBlobDataItem::QUOTA_GRANTED); |
+ shareable_item->set_state(ShareableBlobDataItem::POPULATED_WITH_QUOTA); |
+ } |
} |
} // namespace storage |