mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-24 06:16:30 +03:00
109 lines
3.8 KiB
C++
109 lines
3.8 KiB
C++
// 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 {
|
|
|
|
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(
|
|
LOAD_BYPASS_CACHE | LOAD_DISABLE_CACHE | LOAD_DO_NOT_SEND_AUTH_DATA |
|
|
LOAD_DO_NOT_SEND_COOKIES | LOAD_DO_NOT_SAVE_COOKIES);
|
|
|
|
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
|