blob: 3a7470a286a14c1f4f43d4f363faa471001ebe04 [file] [log] [blame]
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROMECAST_RENDERER_CAST_WEBSOCKET_HANDSHAKE_THROTTLE_PROVIDER_H_
#define CHROMECAST_RENDERER_CAST_WEBSOCKET_HANDSHAKE_THROTTLE_PROVIDER_H_
#include <memory>
#include "base/task/single_thread_task_runner.h"
#include "base/threading/thread_checker.h"
#include "chromecast/renderer/cast_activity_url_filter_manager.h"
#include "third_party/blink/public/platform/websocket_handshake_throttle_provider.h"
namespace chromecast {
// This class allows cast_shell to provide a WebSocketHandshakeThrottle
// implementation to delay or block WebSocket handshakes.
// 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 CastWebSocketHandshakeThrottleProvider
: public blink::WebSocketHandshakeThrottleProvider {
public:
explicit CastWebSocketHandshakeThrottleProvider(
CastActivityUrlFilterManager* url_filter_manager);
CastWebSocketHandshakeThrottleProvider& operator=(
const CastWebSocketHandshakeThrottleProvider&) = delete;
~CastWebSocketHandshakeThrottleProvider() override;
// blink::WebSocketHandshakeThrottleProvider implementation:
std::unique_ptr<blink::WebSocketHandshakeThrottleProvider> Clone(
scoped_refptr<base::SingleThreadTaskRunner> task_runner) override;
std::unique_ptr<blink::WebSocketHandshakeThrottle> CreateThrottle(
base::optional_ref<const blink::LocalFrameToken> local_frame_token,
scoped_refptr<base::SingleThreadTaskRunner> task_runner) override;
private:
// This copy constructor works in conjunction with Clone(), not intended for
// general use.
CastWebSocketHandshakeThrottleProvider(
const CastWebSocketHandshakeThrottleProvider& other);
CastActivityUrlFilterManager* const cast_activity_url_filter_manager_;
THREAD_CHECKER(thread_checker_);
};
} // namespace chromecast
#endif // CHROMECAST_RENDERER_CAST_WEBSOCKET_HANDSHAKE_THROTTLE_PROVIDER_H_