mozilla
/
OpenWPM
зеркало из https://github.com/openwpm/OpenWPM.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность
OpenWPM/test/test_extension.py

412 строки
12 KiB
Python
Исходник Ответственный История

import os
from datetime import datetime
from pathlib import Path
from sqlite3 import Row
from typing import List, Tuple
import pytest
from openwpm.config import BrowserParams, ManagerParams
from openwpm.utilities import db_utils
from . import utilities
from .openwpmtest import OpenWPMTest
# Expected Navigator and Screen properties
PROPERTIES = {
"window.navigator.appCodeName",
"window.navigator.appName",
"window.navigator.appVersion",
"window.navigator.buildID",
"window.navigator.cookieEnabled",
"window.navigator.doNotTrack",
"window.navigator.geolocation",
"window.navigator.language",
"window.navigator.languages",
"window.navigator.onLine",
"window.navigator.oscpu",
"window.navigator.platform",
"window.navigator.product",
"window.navigator.productSub",
"window.navigator.userAgent",
"window.navigator.vendorSub",
"window.navigator.vendor",
"window.screen.pixelDepth",
"window.screen.colorDepth",
}
# Canvas Fingerprinting DB calls and property sets
CANVAS_TEST_URL = u"%s/canvas_fingerprinting.html" % utilities.BASE_TEST_URL
CANVAS_CALLS = {
(CANVAS_TEST_URL, u"CanvasRenderingContext2D.fillStyle", u"set", u"#f60", None),
(
CANVAS_TEST_URL,
u"CanvasRenderingContext2D.textBaseline",
u"set",
u"alphabetic",
None,
),
(CANVAS_TEST_URL, u"CanvasRenderingContext2D.textBaseline", u"set", u"top", None),
(CANVAS_TEST_URL, u"CanvasRenderingContext2D.font", u"set", u"14px 'Arial'", None),
(CANVAS_TEST_URL, u"CanvasRenderingContext2D.fillStyle", u"set", u"#069", None),
(
CANVAS_TEST_URL,
u"CanvasRenderingContext2D.fillStyle",
u"set",
u"rgba(102, 204, 0, 0.7)",
None,
),
(CANVAS_TEST_URL, u"HTMLCanvasElement.getContext", u"call", u"", u'["2d"]'),
(
CANVAS_TEST_URL,
u"CanvasRenderingContext2D.fillRect",
u"call",
u"",
u"[125,1,62,20]",
),
(CANVAS_TEST_URL, u"HTMLCanvasElement.toDataURL", u"call", u"", None),
(
CANVAS_TEST_URL,
u"CanvasRenderingContext2D.fillText",
u"call",
u"",
u'["BrowserLeaks,com <canvas> 1.0",4,17]',
),
(
CANVAS_TEST_URL,
u"CanvasRenderingContext2D.fillText",
u"call",
u"",
u'["BrowserLeaks,com <canvas> 1.0",2,15]',
),
}
WEBRTC_TEST_URL = u"%s/webrtc_localip.html" % utilities.BASE_TEST_URL
WEBRTC_CALLS = {
(
WEBRTC_TEST_URL,
u"RTCPeerConnection.createOffer",
u"call",
u"",
u'["FUNCTION","FUNCTION"]',
),
(WEBRTC_TEST_URL, u"RTCPeerConnection.createDataChannel", u"call", u"", u'[""]'),
(
WEBRTC_TEST_URL,
u"RTCPeerConnection.createDataChannel",
u"call",
u"",
u'["","{\\"reliable\\":false}"]',
),
(WEBRTC_TEST_URL, u"RTCPeerConnection.onicecandidate", u"set", u"FUNCTION", None),
}
# we expect these strings to be present in the WebRTC SDP
WEBRTC_SDP_OFFER_STRINGS = (
"a=ice-options",
"o=mozilla...THIS_IS_SDPARTA",
"IN IP4",
"a=fingerprint:sha-256",
"a=ice-options:",
"a=msid-semantic",
"m=application",
"a=sendrecv",
"a=ice-pwd:",
"a=ice-ufrag:",
"a=mid:0",
"a=sctp-port:",
"a=setup:",
)
# AudioContext and AudioNode symbols we expect from our test script
AUDIO_SYMBOLS = {
u"AudioContext.createOscillator",
u"AudioContext.createAnalyser",
u"AudioContext.createGain",
u"AudioContext.createScriptProcessor",
u"GainNode.gain",
u"OscillatorNode.type",
u"OscillatorNode.connect",
u"AnalyserNode.connect",
u"ScriptProcessorNode.connect",
u"AudioContext.destination",
u"GainNode.connect",
u"ScriptProcessorNode.onaudioprocess",
u"OscillatorNode.start",
u"AnalyserNode.frequencyBinCount",
u"AnalyserNode.getFloatFrequencyData",
u"AnalyserNode.disconnect",
u"ScriptProcessorNode.disconnect",
u"GainNode.disconnect",
u"OscillatorNode.stop",
}
JS_STACK_TEST_URL = u"%s/js_call_stack.html" % utilities.BASE_TEST_URL
JS_STACK_TEST_SCRIPT_URL = u"%s/stack.js" % utilities.BASE_TEST_URL
JS_STACK_CALLS = {
(
JS_STACK_TEST_URL,
u"1",
u"1",
u"",
u"line 10 > eval",
u"",
u"window.navigator.appName",
u"get",
),
(
JS_STACK_TEST_SCRIPT_URL,
u"3",
u"5",
u"js_check_navigator",
u"",
u"",
u"window.navigator.userAgent",
u"get",
),
(
JS_STACK_TEST_SCRIPT_URL,
u"1",
u"1",
u"",
u"line 4 > eval",
u"",
u"window.navigator.platform",
u"get",
),
(
JS_STACK_TEST_SCRIPT_URL,
u"1",
u"1",
u"",
u"line 11 > eval",
u"",
u"window.navigator.buildID",
u"get",
),
(
JS_STACK_TEST_SCRIPT_URL,
u"3",
u"1",
u"anonymous",
u"line 14 > Function",
u"",
u"window.navigator.appVersion",
u"get",
),
(
JS_STACK_TEST_URL,
u"7",
u"9",
u"check_navigator",
u"",
u"",
u"window.navigator.userAgent",
u"get",
),
(
JS_STACK_TEST_URL,
u"1",
u"1",
u"",
u"line 8 > eval",
u"",
u"window.navigator.appCodeName",
u"get",
),
}
JS_COOKIE_TEST_URL = u"%s/js_cookie.html" % utilities.BASE_TEST_URL
DOCUMENT_COOKIE_READ = (
JS_COOKIE_TEST_URL,
u"8",
u"9",
u"set_cookie",
u"",
u"set_cookie@" + JS_COOKIE_TEST_URL + ":8:9"
"\nonload@" + JS_COOKIE_TEST_URL + ":1:1",
u"window.document.cookie",
u"get",
u"test_cookie=Test-0123456789",
)
DOCUMENT_COOKIE_WRITE = (
JS_COOKIE_TEST_URL,
u"7",
u"9",
u"set_cookie",
u"",
u"set_cookie@" + JS_COOKIE_TEST_URL + ":7:9"
"\nonload@" + JS_COOKIE_TEST_URL + ":1:1",
u"window.document.cookie",
u"set",
u"test_cookie=Test-0123456789; " "expires=Tue, 31 Dec 2030 00:00:00 UTC; path=/",
)
DOCUMENT_COOKIE_READ_WRITE = {DOCUMENT_COOKIE_READ, DOCUMENT_COOKIE_WRITE}
class TestExtension(OpenWPMTest):
def get_config(
self, data_dir: Path = None
) -> Tuple[ManagerParams, List[BrowserParams]]:
manager_params, browser_params = self.get_test_config(data_dir)
browser_params[0].js_instrument = True
return manager_params, browser_params
def test_property_enumeration(self) -> None:
test_url = utilities.BASE_TEST_URL + "/property_enumeration.html"
db = self.visit(test_url)
rows = db_utils.query_db(db, "SELECT script_url, symbol FROM javascript")
observed_symbols = set()
for script_url, symbol in rows:
assert script_url == test_url
observed_symbols.add(symbol)
assert PROPERTIES == observed_symbols
def test_canvas_fingerprinting(self) -> None:
db = self.visit("/canvas_fingerprinting.html")
# Check that all calls and methods are recorded
rows = db_utils.get_javascript_entries(db)
observed_rows = set()
for row in rows:
assert isinstance(row, Row)
item = (
row["script_url"],
row["symbol"],
row["operation"],
row["value"],
row["arguments"],
)
observed_rows.add(item)
assert CANVAS_CALLS == observed_rows
def test_extension_gets_correct_visit_id(self) -> None:
url_a = utilities.BASE_TEST_URL + "/simple_a.html"
url_b = utilities.BASE_TEST_URL + "/simple_b.html"
self.visit(url_a)
db = self.visit(url_b)
qry_res = db_utils.query_db(db, "SELECT visit_id, site_url FROM site_visits")
# Construct dict mapping site_url to visit_id
visit_ids = dict()
for row in qry_res:
visit_ids[row[1]] = row[0]
simple_a_visit_id = db_utils.query_db(
db,
"SELECT visit_id FROM javascript WHERE symbol=?",
("window.navigator.userAgent",),
)
simple_b_visit_id = db_utils.query_db(
db,
"SELECT visit_id FROM javascript WHERE symbol=?",
("window.navigator.platform",),
)
assert visit_ids[url_a] == simple_a_visit_id[0][0]
assert visit_ids[url_b] == simple_b_visit_id[0][0]
def check_webrtc_sdp_offer(self, sdp_str: str) -> None:
"""Make sure the SDP offer includes expected fields/strings.
SDP offer contains randomly generated strings (e.g. GUID). That's why
we don't expect a fixed string but only check the presence of certain
protocol fields.
"""
for expected_str in WEBRTC_SDP_OFFER_STRINGS:
assert expected_str in sdp_str
def test_webrtc_localip(self) -> None:
db = self.visit("/webrtc_localip.html")
# Check that all calls and methods are recorded
rows = db_utils.get_javascript_entries(db)
observed_rows = set()
for row in rows:
assert isinstance(row, Row)
if row["symbol"] == "RTCPeerConnection.setLocalDescription" and (
row["operation"] == "call"
):
sdp_offer = row["arguments"]
self.check_webrtc_sdp_offer(sdp_offer)
else:
item = (
row["script_url"],
row["symbol"],
row["operation"],
row["value"],
row["arguments"],
)
observed_rows.add(item)
assert WEBRTC_CALLS == observed_rows
@pytest.mark.skipif(
"CI" in os.environ and os.environ["CI"] == "true",
reason="Flaky on CI",
)
def test_audio_fingerprinting(self):
db = self.visit("/audio_fingerprinting.html")
# Check that all calls and methods are recorded
rows = db_utils.get_javascript_entries(db)
observed_symbols = set()
for item in rows:
observed_symbols.add(item[1])
assert AUDIO_SYMBOLS == observed_symbols
def test_js_call_stack(self):
db = self.visit("/js_call_stack.html")
# Check that all stack info are recorded
rows = db_utils.get_javascript_entries(db, all_columns=True)
observed_rows = set()
for row in rows:
item = (
row["script_url"],
row["script_line"],
row["script_col"],
row["func_name"],
row["script_loc_eval"],
row["call_stack"],
row["symbol"],
row["operation"],
)
observed_rows.add(item)
assert JS_STACK_CALLS == observed_rows
def test_js_time_stamp(self):
# Check that timestamp is recorded correctly for the javascript table
MAX_TIMEDELTA = 60 # max time diff in seconds
db = self.visit("/js_call_stack.html")
utc_now = datetime.utcnow() # OpenWPM stores timestamp in UTC time
rows = db_utils.get_javascript_entries(db, all_columns=True)
assert len(rows) # make sure we have some JS events captured
for row in rows:
js_time = datetime.strptime(row["time_stamp"], "%Y-%m-%dT%H:%M:%S.%fZ")
# compare UTC now and the timestamp recorded at the visit
assert (utc_now - js_time).seconds < MAX_TIMEDELTA
assert not db_utils.any_command_failed(db)
def test_document_cookie_instrumentation(self):
db = self.visit(utilities.BASE_TEST_URL + "/js_cookie.html")
rows = db_utils.get_javascript_entries(db, all_columns=True)
captured_cookie_calls = set()
for row in rows:
item = (
row["script_url"],
row["script_line"],
row["script_col"],
row["func_name"],
row["script_loc_eval"],
row["call_stack"],
row["symbol"],
row["operation"],
row["value"],
)
captured_cookie_calls.add(item)
assert captured_cookie_calls == DOCUMENT_COOKIE_READ_WRITE
Ссылка в новой задаче Показать git blame Копировать постоянную ссылку