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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
import pytest
from support.network import get_host, websocket_request
@pytest.mark.parametrize(
"hostname, port_type, status",
[
# Valid hosts
("localhost", "server_port", 101),
("localhost", "default_port", 101),
("127.0.0.1", "server_port", 101),
("127.0.0.1", "default_port", 101),
("[::1]", "server_port", 101),
("[::1]", "default_port", 101),
("192.168.8.1", "server_port", 101),
("192.168.8.1", "default_port", 101),
("[fdf8:f535:82e4::53]", "server_port", 101),
("[fdf8:f535:82e4::53]", "default_port", 101),
# Invalid hosts
("mozilla.org", "server_port", 400),
("mozilla.org", "wrong_port", 400),
("mozilla.org", "default_port", 400),
("localhost", "wrong_port", 400),
("127.0.0.1", "wrong_port", 400),
("[::1]", "wrong_port", 400),
("192.168.8.1", "wrong_port", 400),
("[fdf8:f535:82e4::53]", "wrong_port", 400),
],
ids=[
# Valid hosts
"localhost with same port as RemoteAgent",
"localhost with default port",
"127.0.0.1 (loopback) with same port as RemoteAgent",
"127.0.0.1 (loopback) with default port",
"[::1] (ipv6 loopback) with same port as RemoteAgent",
"[::1] (ipv6 loopback) with default port",
"ipv4 address with same port as RemoteAgent",
"ipv4 address with default port",
"ipv6 address with same port as RemoteAgent",
"ipv6 address with default port",
# Invalid hosts
"random hostname with the same port as RemoteAgent",
"random hostname with a different port than RemoteAgent",
"random hostname with default port",
"localhost with a different port than RemoteAgent",
"127.0.0.1 (loopback) with a different port than RemoteAgent",
"[::1] (ipv6 loopback) with a different port than RemoteAgent",
"ipv4 address with a different port than RemoteAgent",
"ipv6 address with a different port than RemoteAgent",
],
)
def test_host_header(browser, hostname, port_type, status):
# Request a default browser
current_browser = browser(use_bidi=True)
server_host = current_browser.remote_agent_host
server_port = current_browser.remote_agent_port
test_host = get_host(port_type, hostname, server_port)
response = websocket_request(server_host, server_port, host=test_host)
assert response.status == status
@pytest.mark.parametrize(
"hostname, port_type, status",
[
# Allowed hosts
("testhost", "server_port", 101),
("testhost", "default_port", 101),
("testhost", "wrong_port", 400),
# IP addresses
("192.168.8.1", "server_port", 101),
("192.168.8.1", "default_port", 101),
("[fdf8:f535:82e4::53]", "server_port", 101),
("[fdf8:f535:82e4::53]", "default_port", 101),
("127.0.0.1", "server_port", 101),
("127.0.0.1", "default_port", 101),
("[::1]", "server_port", 101),
("[::1]", "default_port", 101),
# Localhost
("localhost", "server_port", 400),
("localhost", "default_port", 400),
],
ids=[
# Allowed hosts
"allowed host with same port as RemoteAgent",
"allowed host with default port",
"allowed host with wrong port",
# IP addresses
"ipv4 address with same port as RemoteAgent",
"ipv4 address with default port",
"ipv6 address with same port as RemoteAgent",
"ipv6 address with default port",
"127.0.0.1 (loopback) with same port as RemoteAgent",
"127.0.0.1 (loopback) with default port",
"[::1] (ipv6 loopback) with same port as RemoteAgent",
"[::1] (ipv6 loopback) with default port",
# Localhost
"localhost with same port as RemoteAgent",
"localhost with default port",
],
)
def test_allowed_hosts(browser, hostname, port_type, status):
# Request a browser with custom allowed hosts.
current_browser = browser(
use_bidi=True,
extra_args=["--remote-allow-hosts", "testhost"],
)
server_host = current_browser.remote_agent_host
server_port = current_browser.remote_agent_port
test_host = get_host(port_type, hostname, server_port)
response = websocket_request(server_host, server_port, host=test_host)
assert response.status == status
@pytest.mark.parametrize(
"origin, status",
[
(None, 101),
("", 400),
("sometext", 400),
("http://localhost:1234", 400),
],
)
def test_origin_header(browser, origin, status):
# Request a default browser.
current_browser = browser(use_bidi=True)
server_host = current_browser.remote_agent_host
server_port = current_browser.remote_agent_port
response = websocket_request(server_host, server_port, origin=origin)
assert response.status == status
@pytest.mark.parametrize(
"origin, status",
[
(None, 101),
("", 400),
("sometext", 400),
("http://localhost:1234", 101),
("https://localhost:1234", 400),
],
)
def test_allowed_origins(browser, origin, status):
# Request a browser with custom allowed origins.
current_browser = browser(
use_bidi=True,
extra_args=["--remote-allow-origins", "http://localhost:1234"],
)
server_port = current_browser.remote_agent_port
# Both `localhost` and `127.0.0.1` have to accept connections.
for target_host in ["127.0.0.1", "localhost"]:
print(f"Connecting to the WebSocket via host {target_host}")
response = websocket_request(target_host, server_port, origin=origin)
assert response.status == status
|
