blob: daccbe06cc8c048548a9991a99c2912633430cac [file] [log] [blame]
// Copyright 2015 The Chromium 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 "net/url_request/report_sender.h"
#include <utility>
#include "net/base/elements_upload_data_stream.h"
#include "net/base/load_flags.h"
#include "net/base/request_priority.h"
#include "net/base/upload_bytes_element_reader.h"
#include "net/http/http_status_code.h"
#include "net/url_request/url_request_context.h"
#include "net/url_request/url_request_status.h"
namespace {
const void* const kUserDataKey = &kUserDataKey;
class CallbackInfo : public base::SupportsUserData::Data {
public:
CallbackInfo(const net::ReportSender::SuccessCallback& success_callback,
const net::ReportSender::ErrorCallback& error_callback)
: success_callback_(success_callback), error_callback_(error_callback) {}
~CallbackInfo() override = default;
const net::ReportSender::SuccessCallback& success_callback() const {
return success_callback_;
}
const net::ReportSender::ErrorCallback& error_callback() const {
return error_callback_;
}
private:
net::ReportSender::SuccessCallback success_callback_;
net::ReportSender::ErrorCallback error_callback_;
};
} // namespace
namespace net {
const int ReportSender::kLoadFlags = LOAD_BYPASS_CACHE | LOAD_DISABLE_CACHE;
ReportSender::ReportSender(URLRequestContext* request_context,
net::NetworkTrafficAnnotationTag traffic_annotation)
: request_context_(request_context),
traffic_annotation_(traffic_annotation) {}
ReportSender::~ReportSender() = default;
void ReportSender::Send(const GURL& report_uri,
base::StringPiece content_type,
base::StringPiece report,
const SuccessCallback& success_callback,
const ErrorCallback& error_callback) {
DCHECK(!content_type.empty());
std::unique_ptr<URLRequest> url_request = request_context_->CreateRequest(
report_uri, DEFAULT_PRIORITY, this, traffic_annotation_);
url_request->SetUserData(
&kUserDataKey,
std::make_unique<CallbackInfo>(success_callback, error_callback));
url_request->SetLoadFlags(kLoadFlags);
url_request->set_allow_credentials(false);
HttpRequestHeaders extra_headers;
extra_headers.SetHeader(HttpRequestHeaders::kContentType, content_type);
url_request->SetExtraRequestHeaders(extra_headers);
url_request->set_method("POST");
std::vector<char> report_data(report.begin(), report.end());
std::unique_ptr<UploadElementReader> reader(
new UploadOwnedBytesElementReader(&report_data));
url_request->set_upload(
ElementsUploadDataStream::CreateWithReader(std::move(reader), 0));
URLRequest* raw_url_request = url_request.get();
inflight_requests_[raw_url_request] = std::move(url_request);
raw_url_request->Start();
}
void ReportSender::OnResponseStarted(URLRequest* request, int net_error) {
DCHECK_NE(ERR_IO_PENDING, net_error);
CallbackInfo* callback_info =
static_cast<CallbackInfo*>(request->GetUserData(&kUserDataKey));
DCHECK(callback_info);
if (net_error != OK) {
DVLOG(1) << "Failed to send report for " << request->url().host();
if (!callback_info->error_callback().is_null())
callback_info->error_callback().Run(request->url(), net_error, -1);
} else if (request->GetResponseCode() != net::HTTP_OK) {
if (!callback_info->error_callback().is_null())
callback_info->error_callback().Run(request->url(), OK,
request->GetResponseCode());
} else {
if (!callback_info->success_callback().is_null())
callback_info->success_callback().Run();
}
CHECK_GT(inflight_requests_.erase(request), 0u);
}
void ReportSender::OnReadCompleted(URLRequest* request, int bytes_read) {
NOTREACHED();
}
} // namespace net