111 lines
3.8 KiB
C++
111 lines
3.8 KiB
C++
// Copyright 2016 The Chromium Authors
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "net/http/http_cache_lookup_manager.h"
|
|
|
|
#include <memory>
|
|
|
|
#include "base/containers/contains.h"
|
|
#include "base/functional/bind.h"
|
|
#include "base/values.h"
|
|
#include "net/base/load_flags.h"
|
|
#include "net/base/network_anonymization_key.h"
|
|
#include "net/http/http_request_info.h"
|
|
namespace net {
|
|
|
|
// Returns parameters associated with the start of a server push lookup
|
|
// transaction.
|
|
base::Value::Dict NetLogPushLookupTransactionParams(
|
|
const NetLogSource& net_log,
|
|
const ServerPushDelegate::ServerPushHelper* push_helper) {
|
|
base::Value::Dict dict;
|
|
net_log.AddToEventParameters(dict);
|
|
dict.Set("push_url", push_helper->GetURL().possibly_invalid_spec());
|
|
return dict;
|
|
}
|
|
|
|
HttpCacheLookupManager::LookupTransaction::LookupTransaction(
|
|
std::unique_ptr<ServerPushHelper> server_push_helper,
|
|
NetLog* net_log)
|
|
: push_helper_(std::move(server_push_helper)),
|
|
request_(std::make_unique<HttpRequestInfo>()),
|
|
net_log_(NetLogWithSource::Make(
|
|
net_log,
|
|
NetLogSourceType::SERVER_PUSH_LOOKUP_TRANSACTION)) {}
|
|
|
|
HttpCacheLookupManager::LookupTransaction::~LookupTransaction() = default;
|
|
|
|
int HttpCacheLookupManager::LookupTransaction::StartLookup(
|
|
HttpCache* cache,
|
|
CompletionOnceCallback callback,
|
|
const NetLogWithSource& session_net_log) {
|
|
net_log_.BeginEvent(NetLogEventType::SERVER_PUSH_LOOKUP_TRANSACTION, [&] {
|
|
return NetLogPushLookupTransactionParams(session_net_log.source(),
|
|
push_helper_.get());
|
|
});
|
|
|
|
request_->url = push_helper_->GetURL();
|
|
// Note: since HTTP/2 Server Push has been disabled and this code will likely
|
|
// be removed, just use empty NIKs and NAKs here. For more info, see
|
|
// https://crbug.com/1355929.
|
|
request_->network_isolation_key = NetworkIsolationKey();
|
|
request_->network_anonymization_key = NetworkAnonymizationKey();
|
|
request_->method = "GET";
|
|
request_->load_flags = LOAD_ONLY_FROM_CACHE | LOAD_SKIP_CACHE_VALIDATION;
|
|
cache->CreateTransaction(DEFAULT_PRIORITY, &transaction_);
|
|
return transaction_->Start(request_.get(), std::move(callback), net_log_);
|
|
}
|
|
|
|
void HttpCacheLookupManager::LookupTransaction::OnLookupComplete(int result) {
|
|
if (result == OK) {
|
|
DCHECK(push_helper_.get());
|
|
push_helper_->Cancel();
|
|
}
|
|
net_log_.EndEventWithNetErrorCode(
|
|
NetLogEventType::SERVER_PUSH_LOOKUP_TRANSACTION, result);
|
|
}
|
|
|
|
HttpCacheLookupManager::HttpCacheLookupManager(HttpCache* http_cache)
|
|
: http_cache_(http_cache) {}
|
|
|
|
HttpCacheLookupManager::~HttpCacheLookupManager() = default;
|
|
|
|
void HttpCacheLookupManager::OnPush(
|
|
std::unique_ptr<ServerPushHelper> push_helper,
|
|
const NetLogWithSource& session_net_log) {
|
|
GURL pushed_url = push_helper->GetURL();
|
|
|
|
// There's a pending lookup transaction sent over already.
|
|
if (base::Contains(lookup_transactions_, pushed_url))
|
|
return;
|
|
|
|
auto lookup = std::make_unique<LookupTransaction>(std::move(push_helper),
|
|
session_net_log.net_log());
|
|
// TODO(zhongyi): add events in session net log to log the creation of
|
|
// LookupTransaction.
|
|
|
|
int rv = lookup->StartLookup(
|
|
http_cache_,
|
|
base::BindOnce(&HttpCacheLookupManager::OnLookupComplete,
|
|
weak_factory_.GetWeakPtr(), pushed_url),
|
|
session_net_log);
|
|
|
|
if (rv == ERR_IO_PENDING) {
|
|
lookup_transactions_[pushed_url] = std::move(lookup);
|
|
} else {
|
|
lookup->OnLookupComplete(rv);
|
|
}
|
|
}
|
|
|
|
void HttpCacheLookupManager::OnLookupComplete(const GURL& url, int rv) {
|
|
auto it = lookup_transactions_.find(url);
|
|
DCHECK(it != lookup_transactions_.end());
|
|
|
|
it->second->OnLookupComplete(rv);
|
|
|
|
lookup_transactions_.erase(it);
|
|
}
|
|
|
|
} // namespace net
|