diff options
author | Martin Robinson <mrobinson@igalia.com> | 2023-10-05 19:47:39 +0200 |
---|---|---|
committer | Martin Robinson <mrobinson@igalia.com> | 2023-11-03 15:38:18 +0000 |
commit | f4d3af296c05260dfbb3deea4f8fa400cb6887d3 (patch) | |
tree | 169db2cc68e01a755b30500dd525f1a2ec2da861 /components/shared/bluetooth/blocklist.rs | |
parent | 863529d9622c68f0a9535225237eb5e5c5b8c757 (diff) | |
download | servo-f4d3af296c05260dfbb3deea4f8fa400cb6887d3.tar.gz servo-f4d3af296c05260dfbb3deea4f8fa400cb6887d3.zip |
Move `*_traits` and other shared types to `shared`
This is the start of the organization of types that are in their own
crates in order to break dependency cycles between other crates. The
idea here is that putting these packages into their own directory is the
first step toward cleaning them up. They have grown organically and it
is difficult to explain to new folks where to put new shared types. Many
of these crates contain more than traits or don't contain traits at all.
Notably, `script_traits` isn't touched because it is vendored from
Gecko. Eventually this will move to `third_party`.
Diffstat (limited to 'components/shared/bluetooth/blocklist.rs')
-rw-r--r-- | components/shared/bluetooth/blocklist.rs | 112 |
1 files changed, 112 insertions, 0 deletions
diff --git a/components/shared/bluetooth/blocklist.rs b/components/shared/bluetooth/blocklist.rs new file mode 100644 index 00000000000..ca413c41a1e --- /dev/null +++ b/components/shared/bluetooth/blocklist.rs @@ -0,0 +1,112 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at https://mozilla.org/MPL/2.0/. */ + +use std::cell::RefCell; +use std::collections::HashMap; +use std::string::String; + +use embedder_traits::resources::{self, Resource}; +use regex::Regex; + +const EXCLUDE_READS: &str = "exclude-reads"; +const EXCLUDE_WRITES: &str = "exclude-writes"; +const VALID_UUID_REGEX: &str = "^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}"; + +thread_local!(pub static BLUETOOTH_BLOCKLIST: RefCell<BluetoothBlocklist> = + RefCell::new(BluetoothBlocklist(parse_blocklist()))); + +pub fn uuid_is_blocklisted(uuid: &str, exclude_type: Blocklist) -> bool { + BLUETOOTH_BLOCKLIST.with(|blist| match exclude_type { + Blocklist::All => blist.borrow().is_blocklisted(uuid), + Blocklist::Reads => blist.borrow().is_blocklisted_for_reads(uuid), + Blocklist::Writes => blist.borrow().is_blocklisted_for_writes(uuid), + }) +} + +pub struct BluetoothBlocklist(Option<HashMap<String, Blocklist>>); + +#[derive(Eq, PartialEq)] +pub enum Blocklist { + All, // Read and Write + Reads, + Writes, +} + +impl BluetoothBlocklist { + // https://webbluetoothcg.github.io/web-bluetooth/#blocklisted + pub fn is_blocklisted(&self, uuid: &str) -> bool { + match self.0 { + Some(ref map) => map.get(uuid).map_or(false, |et| et.eq(&Blocklist::All)), + None => false, + } + } + + // https://webbluetoothcg.github.io/web-bluetooth/#blocklisted-for-reads + pub fn is_blocklisted_for_reads(&self, uuid: &str) -> bool { + match self.0 { + Some(ref map) => map.get(uuid).map_or(false, |et| { + et.eq(&Blocklist::All) || et.eq(&Blocklist::Reads) + }), + None => false, + } + } + + // https://webbluetoothcg.github.io/web-bluetooth/#blocklisted-for-writes + pub fn is_blocklisted_for_writes(&self, uuid: &str) -> bool { + match self.0 { + Some(ref map) => map.get(uuid).map_or(false, |et| { + et.eq(&Blocklist::All) || et.eq(&Blocklist::Writes) + }), + None => false, + } + } +} + +// https://webbluetoothcg.github.io/web-bluetooth/#parsing-the-blocklist +fn parse_blocklist() -> Option<HashMap<String, Blocklist>> { + // Step 1 missing, currently we parse ./resources/gatt_blocklist.txt. + let valid_uuid_regex = Regex::new(VALID_UUID_REGEX).unwrap(); + let content = resources::read_string(Resource::BluetoothBlocklist); + // Step 3 + let mut result = HashMap::new(); + // Step 2 and 4 + for line in content.lines() { + // Step 4.1 + if line.is_empty() || line.starts_with('#') { + continue; + } + let mut exclude_type = Blocklist::All; + let mut words = line.split_whitespace(); + let uuid = match words.next() { + Some(uuid) => uuid, + None => continue, + }; + if !valid_uuid_regex.is_match(uuid) { + return None; + } + match words.next() { + // Step 4.2 We already have an initialized exclude_type variable with Blocklist::All. + None => {}, + // Step 4.3 + Some(EXCLUDE_READS) => { + exclude_type = Blocklist::Reads; + }, + Some(EXCLUDE_WRITES) => { + exclude_type = Blocklist::Writes; + }, + // Step 4.4 + _ => { + return None; + }, + } + // Step 4.5 + if result.contains_key(uuid) { + return None; + } + // Step 4.6 + result.insert(uuid.to_string(), exclude_type); + } + // Step 5 + Some(result) +} |