Brian Johnson
f9b8cc532d
The templates should be removed along with the feature flag when we switch over to unique_ptr
104 lines
3.6 KiB
C++
104 lines
3.6 KiB
C++
/* Copyright (c) 2019 The Brave Authors. All rights reserved.
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
* You can obtain one at https://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef BRAVE_BROWSER_NET_RESOURCE_CONTEXT_DATA_H_
|
|
#define BRAVE_BROWSER_NET_RESOURCE_CONTEXT_DATA_H_
|
|
|
|
#include <cstdint>
|
|
#include <memory>
|
|
#include <optional>
|
|
#include <set>
|
|
#include <string>
|
|
|
|
#include "base/containers/unique_ptr_adapters.h"
|
|
#include "base/memory/ref_counted.h"
|
|
#include "base/memory/scoped_refptr.h"
|
|
#include "base/supports_user_data.h"
|
|
#include "content/public/browser/browser_thread.h"
|
|
#include "content/public/browser/content_browser_client.h"
|
|
#include "mojo/public/cpp/bindings/pending_receiver.h"
|
|
#include "mojo/public/cpp/bindings/pending_remote.h"
|
|
#include "services/network/public/mojom/url_loader_factory.mojom.h"
|
|
#include "services/network/public/mojom/websocket.mojom.h"
|
|
|
|
template <template <typename> class T>
|
|
class BraveProxyingURLLoaderFactory;
|
|
template <template <typename> class T>
|
|
class BraveProxyingWebSocket;
|
|
template <template <typename> class T>
|
|
class BraveRequestHandler;
|
|
|
|
namespace content {
|
|
class BrowserContext;
|
|
}
|
|
|
|
// Used for both URLLoaders and WebSocket proxies.
|
|
class RequestIDGenerator
|
|
: public base::RefCountedThreadSafe<RequestIDGenerator> {
|
|
public:
|
|
RequestIDGenerator();
|
|
RequestIDGenerator(const RequestIDGenerator&) = delete;
|
|
RequestIDGenerator& operator=(const RequestIDGenerator&) = delete;
|
|
int64_t Generate() {
|
|
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
|
|
return ++id_;
|
|
}
|
|
|
|
private:
|
|
friend class base::RefCountedThreadSafe<RequestIDGenerator>;
|
|
~RequestIDGenerator() {}
|
|
|
|
// Although this initialization can be done in a thread other than the IO
|
|
// thread, we expect at least one memory barrier before actually calling
|
|
// Generate in the IO thread, so we don't protect the variable with a lock.
|
|
int64_t id_ = 0;
|
|
};
|
|
|
|
// Owns proxying factories for URLLoaders and websocket proxies. There is
|
|
// one |ResourceContextData| per profile.
|
|
template <template <typename> class T>
|
|
class ResourceContextData : public base::SupportsUserData::Data {
|
|
public:
|
|
ResourceContextData(const ResourceContextData&) = delete;
|
|
ResourceContextData& operator=(const ResourceContextData&) = delete;
|
|
~ResourceContextData() override;
|
|
|
|
static void StartProxying(
|
|
content::BrowserContext* browser_context,
|
|
content::GlobalRenderFrameHostToken render_frame_token,
|
|
network::URLLoaderFactoryBuilder& factory_builder,
|
|
scoped_refptr<base::SequencedTaskRunner> navigation_response_task_runner);
|
|
|
|
static BraveProxyingWebSocket<T>* CreateProxyingWebSocket(
|
|
content::ContentBrowserClient::WebSocketFactory factory,
|
|
const GURL& url,
|
|
const net::SiteForCookies& site_for_cookies,
|
|
const std::optional<std::string>& user_agent,
|
|
content::BrowserContext* browser_context,
|
|
content::GlobalRenderFrameHostToken render_frame_token,
|
|
const url::Origin& origin);
|
|
|
|
void RemoveProxy(BraveProxyingURLLoaderFactory<T>* proxy);
|
|
void RemoveProxyWebSocket(BraveProxyingWebSocket<T>* proxy);
|
|
|
|
private:
|
|
ResourceContextData();
|
|
|
|
std::unique_ptr<BraveRequestHandler<T>> request_handler_;
|
|
scoped_refptr<RequestIDGenerator> request_id_generator_;
|
|
|
|
std::set<std::unique_ptr<BraveProxyingURLLoaderFactory<T>>,
|
|
base::UniquePtrComparator>
|
|
proxies_;
|
|
|
|
std::set<std::unique_ptr<BraveProxyingWebSocket<T>>,
|
|
base::UniquePtrComparator>
|
|
websocket_proxies_;
|
|
|
|
base::WeakPtrFactory<ResourceContextData> weak_factory_;
|
|
};
|
|
|
|
#endif // BRAVE_BROWSER_NET_RESOURCE_CONTEXT_DATA_H_
|