blob: b7b21053bb7e6eed77f31906009dd85125f2abd6 [file] [log] [blame]
// Copyright 2018 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.
#ifndef ANDROID_WEBVIEW_RENDERER_AW_WEBSOCKET_HANDSHAKE_THROTTLE_PROVIDER_H_
#define ANDROID_WEBVIEW_RENDERER_AW_WEBSOCKET_HANDSHAKE_THROTTLE_PROVIDER_H_
#include <memory>
#include "base/macros.h"
#include "base/threading/thread_checker.h"
#include "components/safe_browsing/common/safe_browsing.mojom.h"
#include "content/public/renderer/websocket_handshake_throttle_provider.h"
namespace service_manager {
class Connector;
}
namespace android_webview {
// This must be constructed on the render thread, and then used and destructed
// on a single thread, which can be different from the render thread.
class AwWebSocketHandshakeThrottleProvider final
: public content::WebSocketHandshakeThrottleProvider {
public:
AwWebSocketHandshakeThrottleProvider(service_manager::Connector* connector);
~AwWebSocketHandshakeThrottleProvider() override;
// Implements content::WebSocketHandshakeThrottleProvider.
std::unique_ptr<content::WebSocketHandshakeThrottleProvider> Clone(
scoped_refptr<base::SingleThreadTaskRunner> task_runner) override;
std::unique_ptr<blink::WebSocketHandshakeThrottle> CreateThrottle(
int render_frame_id,
scoped_refptr<base::SingleThreadTaskRunner> task_runner) override;
private:
// This copy constructor works in conjunction with Clone(), not intended for
// general use.
AwWebSocketHandshakeThrottleProvider(
const AwWebSocketHandshakeThrottleProvider& other);
safe_browsing::mojom::SafeBrowsingPtrInfo safe_browsing_info_;
safe_browsing::mojom::SafeBrowsingPtr safe_browsing_;
THREAD_CHECKER(thread_checker_);
DISALLOW_ASSIGN(AwWebSocketHandshakeThrottleProvider);
};
} // namespace android_webview
#endif // ANDROID_WEBVIEW_RENDERER_AW_WEBSOCKET_HANDSHAKE_THROTTLE_PROVIDER_H_