1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/web_test/renderer/test_websocket_handshake_throttle_provider.h"
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/string_piece.h"
#include "base/task/single_thread_task_runner.h"
#include "base/time/time.h"
#include "base/timer/timer.h"
#include "content/public/renderer/render_frame.h"
#include "third_party/blink/public/platform/web_url.h"
#include "url/gurl.h"
namespace content {
namespace {
using CompletionCallback = blink::WebSocketHandshakeThrottle::OnCompletion;
// Checks for a valid "content-shell-websocket-delay-ms" parameter and returns
// it as a TimeDelta if it exists. Otherwise returns a zero TimeDelta.
base::TimeDelta ExtractDelayFromUrl(const GURL& url) {
if (!url.has_query())
return base::TimeDelta();
url::Component query = url.parsed_for_possibly_invalid_spec().query;
url::Component key;
url::Component value;
base::StringPiece spec = url.possibly_invalid_spec();
while (url::ExtractQueryKeyValue(spec.data(), &query, &key, &value)) {
base::StringPiece key_piece = spec.substr(key.begin, key.len);
if (key_piece != "content-shell-websocket-delay-ms")
continue;
base::StringPiece value_piece = spec.substr(value.begin, value.len);
int value_int;
if (!base::StringToInt(value_piece, &value_int) || value_int < 0)
return base::TimeDelta();
return base::Milliseconds(value_int);
}
// Parameter was not found.
return base::TimeDelta();
}
// A simple WebSocketHandshakeThrottle that calls callbacks->IsSuccess() after n
// milli-seconds if the URL query contains
// content-shell-websocket-delay-ms=n. Otherwise it calls IsSuccess()
// immediately.
class TestWebSocketHandshakeThrottle
: public blink::WebSocketHandshakeThrottle {
public:
explicit TestWebSocketHandshakeThrottle(
scoped_refptr<base::SingleThreadTaskRunner> task_runner) {
timer_.SetTaskRunner(std::move(task_runner));
}
~TestWebSocketHandshakeThrottle() override = default;
void ThrottleHandshake(const blink::WebURL& url,
const blink::WebSecurityOrigin& creator_origin,
CompletionCallback completion_callback) override {
DCHECK(completion_callback);
auto wrapper = base::BindOnce(
[](CompletionCallback callback) {
std::move(callback).Run(absl::nullopt);
},
std::move(completion_callback));
timer_.Start(FROM_HERE, ExtractDelayFromUrl(url), std::move(wrapper));
}
private:
base::OneShotTimer timer_;
};
} // namespace
std::unique_ptr<blink::WebSocketHandshakeThrottleProvider>
TestWebSocketHandshakeThrottleProvider::Clone(
scoped_refptr<base::SingleThreadTaskRunner> task_runner) {
return std::make_unique<TestWebSocketHandshakeThrottleProvider>();
}
std::unique_ptr<blink::WebSocketHandshakeThrottle>
TestWebSocketHandshakeThrottleProvider::CreateThrottle(
int render_frame_id,
scoped_refptr<base::SingleThreadTaskRunner> task_runner) {
return std::make_unique<TestWebSocketHandshakeThrottle>(
std::move(task_runner));
}
} // namespace content
|
