1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021 |
- #![allow(non_snake_case)]
- use crate::{
- analysis::{blocked_in, Analyzer},
- bridge_verification_info::BridgeVerificationInfo,
- *,
- };
- use lox_library::{
- bridge_table::{self, BridgeLine, BridgeTable},
- cred::Lox,
- proto::*,
- scalar_u32, BridgeAuth, BridgeDb,
- };
- use base64::{engine::general_purpose, Engine as _};
- use curve25519_dalek::{ristretto::RistrettoBasepointTable, Scalar};
- use rand::RngCore;
- use sha1::{Digest, Sha1};
- use std::{
- collections::{BTreeMap, HashMap, HashSet},
- sync::{Arc, Mutex},
- };
- use x25519_dalek::{PublicKey, StaticSecret};
- struct TestHarness {
- bdb: BridgeDb,
- pub ba: BridgeAuth,
- }
- impl TestHarness {
- fn new() -> Self {
- TestHarness::new_buckets(5, 5)
- }
- fn new_buckets(num_buckets: u16, hot_spare: u16) -> Self {
- // Create a BridegDb
- let mut bdb = BridgeDb::new();
- // Create a BridgeAuth
- let mut ba = BridgeAuth::new(bdb.pubkey);
- // Make 3 x num_buckets open invitation bridges, in sets of 3
- for _ in 0..num_buckets {
- let bucket = [random(), random(), random()];
- let _ = ba.add_openinv_bridges(bucket, &mut bdb);
- }
- // Add hot_spare more hot spare buckets
- for _ in 0..hot_spare {
- let bucket = [random(), random(), random()];
- let _ = ba.add_spare_bucket(bucket, &mut bdb);
- }
- // Create the encrypted bridge table
- ba.enc_bridge_table();
- Self { bdb, ba }
- }
- fn advance_days(&mut self, days: u16) {
- self.ba.advance_days(days);
- }
- fn get_new_credential(&mut self) -> Lox {
- let inv = self.bdb.invite().unwrap();
- let (req, state) = open_invite::request(&inv);
- let resp = self.ba.handle_open_invite(req).unwrap();
- let (cred, _bridgeline) =
- open_invite::handle_response(state, resp, &self.ba.lox_pub).unwrap();
- cred
- }
- fn level_up(&mut self, cred: &Lox) -> Lox {
- let current_level = scalar_u32(&cred.trust_level).unwrap();
- if current_level == 0 {
- self.advance_days(trust_promotion::UNTRUSTED_INTERVAL.try_into().unwrap());
- let (promreq, promstate) =
- trust_promotion::request(cred, &self.ba.lox_pub, self.ba.today()).unwrap();
- let promresp = self.ba.handle_trust_promotion(promreq).unwrap();
- let migcred = trust_promotion::handle_response(promstate, promresp).unwrap();
- let (migreq, migstate) =
- migration::request(cred, &migcred, &self.ba.lox_pub, &self.ba.migration_pub)
- .unwrap();
- let migresp = self.ba.handle_migration(migreq).unwrap();
- let new_cred = migration::handle_response(migstate, migresp, &self.ba.lox_pub).unwrap();
- new_cred
- } else {
- self.advance_days(
- level_up::LEVEL_INTERVAL[usize::try_from(current_level).unwrap()]
- .try_into()
- .unwrap(),
- );
- let (id, key) = bridge_table::from_scalar(cred.bucket).unwrap();
- let encbuckets = self.ba.enc_bridge_table();
- let bucket =
- bridge_table::BridgeTable::decrypt_bucket(id, &key, encbuckets.get(&id).unwrap())
- .unwrap();
- let reachcred = bucket.1.unwrap();
- let (lvreq, lvstate) = level_up::request(
- cred,
- &reachcred,
- &self.ba.lox_pub,
- &self.ba.reachability_pub,
- self.ba.today(),
- )
- .unwrap();
- let lvresp = self.ba.handle_level_up(lvreq).unwrap();
- let new_cred = level_up::handle_response(lvstate, lvresp, &self.ba.lox_pub).unwrap();
- new_cred
- }
- }
- fn get_bucket(&mut self, cred: &Lox) -> [BridgeLine; bridge_table::MAX_BRIDGES_PER_BUCKET] {
- let (id, key) = bridge_table::from_scalar(cred.bucket).unwrap();
- let encbuckets = self.ba.enc_bridge_table();
- let bucket =
- bridge_table::BridgeTable::decrypt_bucket(id, &key, encbuckets.get(&id).unwrap())
- .unwrap();
- bucket.0
- }
- }
- pub fn random() -> BridgeLine {
- let mut rng = rand::thread_rng();
- let mut res: BridgeLine = BridgeLine::default();
- // Pick a random 4-byte address
- let mut addr: [u8; 4] = [0; 4];
- rng.fill_bytes(&mut addr);
- // If the leading byte is 224 or more, that's not a valid IPv4
- // address. Choose an IPv6 address instead (but don't worry too
- // much about it being well formed).
- if addr[0] >= 224 {
- rng.fill_bytes(&mut res.addr);
- } else {
- // Store an IPv4 address as a v4-mapped IPv6 address
- res.addr[10] = 255;
- res.addr[11] = 255;
- res.addr[12..16].copy_from_slice(&addr);
- };
- let ports: [u16; 4] = [443, 4433, 8080, 43079];
- let portidx = (rng.next_u32() % 4) as usize;
- res.port = ports[portidx];
- res.uid_fingerprint = rng.next_u64();
- rng.fill_bytes(&mut res.fingerprint);
- let mut cert: [u8; 52] = [0; 52];
- rng.fill_bytes(&mut cert);
- let infostr: String = format!(
- "obfs4 cert={}, iat-mode=0",
- general_purpose::STANDARD_NO_PAD.encode(cert)
- );
- res.info[..infostr.len()].copy_from_slice(infostr.as_bytes());
- res
- }
- #[tokio::test]
- async fn test_extra_infos() {
- let bridge_to_test =
- array_bytes::hex2array("72E12B89136B45BBC81D1EF0AC7DDDBB91B148DB").unwrap();
- // Open test database
- let db: Db = sled::open("test_db").unwrap();
- // Delete all data in test DB
- db.clear().unwrap();
- assert!(!db.contains_key("bridges").unwrap());
- assert!(!db.contains_key(bridge_to_test).unwrap());
- // Download and process recent extra-infos files
- update_extra_infos(
- &db,
- "https://collector.torproject.org/recent/bridge-descriptors/extra-infos/",
- )
- .await
- .unwrap();
- // Check that DB contains information on a bridge with high uptime
- assert!(db.contains_key("bridges").unwrap());
- let bridges: HashSet<[u8; 20]> =
- bincode::deserialize(&db.get("bridges").unwrap().unwrap()).unwrap();
- assert!(bridges.contains(&bridge_to_test));
- assert!(db.contains_key(bridge_to_test).unwrap());
- let _bridge_info: BridgeInfo =
- bincode::deserialize(&db.get(bridge_to_test).unwrap().unwrap()).unwrap();
- }
- #[test]
- fn test_negative_reports() {
- let mut th = TestHarness::new();
- // Get new level 1 credential
- let cred = th.get_new_credential();
- let cred = th.level_up(&cred);
- let bridges = th.get_bucket(&cred);
- // Create BridgeVerificationInfo for each bridge
- let mut buckets = HashSet::<Scalar>::new();
- buckets.insert(cred.bucket);
- let bridge_info_1 = BridgeVerificationInfo {
- bridge_line: bridges[0],
- buckets: buckets.clone(),
- pubkey: None,
- };
- let bridge_info_2 = BridgeVerificationInfo {
- bridge_line: bridges[1],
- buckets: buckets.clone(),
- pubkey: None,
- };
- let bridge_info_3 = BridgeVerificationInfo {
- bridge_line: bridges[2],
- buckets: buckets.clone(),
- pubkey: None,
- };
- // Create reports
- let report_1 =
- NegativeReport::from_bridgeline(bridges[0], "ru".to_string(), BridgeDistributor::Lox);
- let report_2 =
- NegativeReport::from_lox_bucket(bridges[1].fingerprint, cred.bucket, "ru".to_string());
- let report_3 =
- NegativeReport::from_lox_credential(bridges[2].fingerprint, &cred, "ru".to_string());
- // Backdated reports
- let date = get_date();
- let mut rng = rand::thread_rng();
- let mut nonce = [0; 32];
- rng.fill_bytes(&mut nonce);
- let report_4 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(
- &bridges[0],
- date - 1,
- nonce,
- )),
- "ru".to_string(),
- date - 1,
- nonce,
- BridgeDistributor::Lox,
- );
- let mut nonce = [0; 32];
- rng.fill_bytes(&mut nonce);
- let report_5 = NegativeReport::new(
- bridges[1].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(
- &bridges[1],
- date - 2,
- nonce,
- )),
- "ru".to_string(),
- date - 2,
- nonce,
- BridgeDistributor::Lox,
- );
- let mut nonce = [0; 32];
- rng.fill_bytes(&mut nonce);
- let report_6 = NegativeReport::new(
- bridges[2].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(
- &bridges[2],
- date - 3,
- nonce,
- )),
- "ru".to_string(),
- date - 3,
- nonce,
- BridgeDistributor::Lox,
- );
- // Verify reports
- assert!(report_1.verify(&bridge_info_1));
- assert!(report_2.verify(&bridge_info_2));
- assert!(report_3.verify(&bridge_info_3));
- assert!(report_4.verify(&bridge_info_1));
- assert!(report_5.verify(&bridge_info_2));
- assert!(report_6.verify(&bridge_info_3));
- // Check that deserialization fails under invalid conditions
- // Date in the future
- let mut invalid_report_1 =
- NegativeReport::from_bridgeline(bridges[0], "ru".to_string(), BridgeDistributor::Lox)
- .to_serializable_report();
- invalid_report_1.date = invalid_report_1.date + 2;
- // Date too far in past
- let mut invalid_report_2 =
- NegativeReport::from_bridgeline(bridges[1], "ru".to_string(), BridgeDistributor::Lox)
- .to_serializable_report();
- invalid_report_2.date = invalid_report_2.date - MAX_BACKDATE - 1;
- // Invalid country code
- let invalid_report_3 =
- NegativeReport::from_bridgeline(bridges[2], "xx".to_string(), BridgeDistributor::Lox)
- .to_serializable_report();
- assert!(invalid_report_1.to_report().is_err());
- assert!(invalid_report_2.to_report().is_err());
- assert!(invalid_report_3.to_report().is_err());
- // Check that verification fails with incorrect data
- let date = get_date();
- let mut rng = rand::thread_rng();
- // Incorrect BridgeLine hash
- let mut nonce = [0; 32];
- rng.fill_bytes(&mut nonce);
- let invalid_report_4 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(
- &BridgeLine::default(),
- date,
- nonce,
- )),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- // Incorrect bucket hash
- let mut nonce = [0; 32];
- rng.fill_bytes(&mut nonce);
- let invalid_report_5 = NegativeReport::new(
- bridges[1].fingerprint,
- ProofOfBridgeKnowledge::HashOfBucket(HashOfBucket::new(&Scalar::ZERO, date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- assert!(!invalid_report_4.verify(&bridge_info_1));
- assert!(!invalid_report_5.verify(&bridge_info_2));
- // Test that reports with duplicate nonces are rejected
- // (Also test encryption and decryption.)
- // Open test database
- let db: Db = sled::open("test_db").unwrap();
- // Delete all data in test DB
- db.clear().unwrap();
- assert!(!db.contains_key("nrs-to-process").unwrap());
- let mut nonce = [0; 32];
- rng.fill_bytes(&mut nonce);
- // A valid report
- let valid_report_1 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(&bridges[0], date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- let valid_report_1_copy_1 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(&bridges[0], date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- let valid_report_1_copy_2 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(&bridges[0], date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- // Report which reuses this nonce
- let invalid_report_1 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(&bridges[0], date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- // This is the same report
- assert_eq!(valid_report_1, invalid_report_1);
- // Report which reuses this nonce for a different bridge
- let invalid_report_2 = NegativeReport::new(
- bridges[1].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(&bridges[1], date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- // Report which uses this nonce but on a different day
- let valid_report_2 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(
- &bridges[0],
- date - 1,
- nonce,
- )),
- "ru".to_string(),
- date - 1,
- nonce,
- BridgeDistributor::Lox,
- );
- // Report with different nonce
- let mut nonce = [0; 32];
- rng.fill_bytes(&mut nonce);
- let valid_report_3 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBridgeLine(HashOfBridgeLine::new(&bridges[0], date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- let map_key_1 = format!(
- "{}_{}_{}",
- array_bytes::bytes2hex("", valid_report_1.fingerprint),
- "ru".to_string(),
- date
- );
- // Generate key for today
- let secret = StaticSecret::random_from_rng(&mut rng);
- let public = PublicKey::from(&secret);
- let secret_yesterday = StaticSecret::random_from_rng(&mut rng);
- let public_yesterday = PublicKey::from(&secret_yesterday);
- assert!(!db.contains_key("nr-keys").unwrap());
- // Fail to add to database because we can't decrypt
- handle_encrypted_negative_report(&db, valid_report_1_copy_1.encrypt(&public));
- assert!(!db.contains_key("nrs-to-process").unwrap());
- // Store yesterday's key but not today's
- let mut nr_keys = BTreeMap::<u32, StaticSecret>::new();
- nr_keys.insert(date - 1, secret_yesterday);
- db.insert("nr-keys", bincode::serialize(&nr_keys).unwrap())
- .unwrap();
- // Fail to add to database because we still can't decrypt
- handle_encrypted_negative_report(&db, valid_report_1_copy_2.encrypt(&public));
- assert!(!db.contains_key("nrs-to-process").unwrap());
- // Store today's key
- nr_keys.insert(date, secret);
- db.insert("nr-keys", bincode::serialize(&nr_keys).unwrap())
- .unwrap();
- handle_encrypted_negative_report(&db, valid_report_1.encrypt(&public));
- let nrs_to_process: BTreeMap<String, Vec<SerializableNegativeReport>> =
- bincode::deserialize(&db.get("nrs-to-process").unwrap().unwrap()).unwrap();
- let negative_reports = nrs_to_process.get(&map_key_1).unwrap();
- assert_eq!(negative_reports.len(), 1);
- handle_encrypted_negative_report(&db, invalid_report_1.encrypt(&public)); // no change
- let nrs_to_process: BTreeMap<String, Vec<SerializableNegativeReport>> =
- bincode::deserialize(&db.get("nrs-to-process").unwrap().unwrap()).unwrap();
- let negative_reports = nrs_to_process.get(&map_key_1).unwrap();
- assert_eq!(negative_reports.len(), 1);
- let map_key_2 = format!(
- "{}_{}_{}",
- array_bytes::bytes2hex("", invalid_report_2.fingerprint),
- "ru".to_string(),
- date
- );
- handle_encrypted_negative_report(&db, invalid_report_2.encrypt(&public)); // no change
- let nrs_to_process: BTreeMap<String, Vec<SerializableNegativeReport>> =
- bincode::deserialize(&db.get("nrs-to-process").unwrap().unwrap()).unwrap();
- assert!(!nrs_to_process.contains_key(&map_key_2));
- let map_key_3 = format!(
- "{}_{}_{}",
- array_bytes::bytes2hex("", valid_report_2.fingerprint),
- "ru".to_string(),
- date - 1
- );
- handle_encrypted_negative_report(&db, valid_report_2.encrypt(&public_yesterday));
- let nrs_to_process: BTreeMap<String, Vec<SerializableNegativeReport>> =
- bincode::deserialize(&db.get("nrs-to-process").unwrap().unwrap()).unwrap();
- let negative_reports = nrs_to_process.get(&map_key_3).unwrap();
- assert_eq!(negative_reports.len(), 1);
- handle_encrypted_negative_report(&db, valid_report_3.encrypt(&public));
- let nrs_to_process: BTreeMap<String, Vec<SerializableNegativeReport>> =
- bincode::deserialize(&db.get("nrs-to-process").unwrap().unwrap()).unwrap();
- let negative_reports = nrs_to_process.get(&map_key_1).unwrap();
- assert_eq!(negative_reports.len(), 2);
- // Same tests, but use hash of bucket
- // Delete all data in test DB
- db.clear().unwrap();
- assert!(!db.contains_key("nrs-to-process").unwrap());
- // Re-generate keys and save in database
- let public = new_negative_report_key(&db, date).unwrap();
- let public_yesterday = new_negative_report_key(&db, date - 1).unwrap();
- let mut nonce = [0; 32];
- rng.fill_bytes(&mut nonce);
- // A valid report
- let valid_report_1 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBucket(HashOfBucket::new(&cred.bucket, date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- // Report which reuses this nonce
- let invalid_report_1 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBucket(HashOfBucket::new(&cred.bucket, date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- // This is the same report
- assert_eq!(valid_report_1, invalid_report_1);
- // Report which reuses this nonce for a different bridge
- let invalid_report_2 = NegativeReport::new(
- bridges[1].fingerprint,
- ProofOfBridgeKnowledge::HashOfBucket(HashOfBucket::new(&cred.bucket, date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- // Report which uses this nonce but on a different day
- let valid_report_2 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBucket(HashOfBucket::new(&cred.bucket, date - 1, nonce)),
- "ru".to_string(),
- date - 1,
- nonce,
- BridgeDistributor::Lox,
- );
- // Report with different nonce
- let mut nonce = [0; 32];
- rng.fill_bytes(&mut nonce);
- let valid_report_3 = NegativeReport::new(
- bridges[0].fingerprint,
- ProofOfBridgeKnowledge::HashOfBucket(HashOfBucket::new(&cred.bucket, date, nonce)),
- "ru".to_string(),
- date,
- nonce,
- BridgeDistributor::Lox,
- );
- let map_key_1 = format!(
- "{}_{}_{}",
- array_bytes::bytes2hex("", valid_report_1.fingerprint),
- "ru".to_string(),
- date
- );
- handle_encrypted_negative_report(&db, valid_report_1.encrypt(&public));
- let nrs_to_process: BTreeMap<String, Vec<SerializableNegativeReport>> =
- bincode::deserialize(&db.get("nrs-to-process").unwrap().unwrap()).unwrap();
- let negative_reports = nrs_to_process.get(&map_key_1).unwrap();
- assert_eq!(negative_reports.len(), 1);
- handle_encrypted_negative_report(&db, invalid_report_1.encrypt(&public)); // no change
- let nrs_to_process: BTreeMap<String, Vec<SerializableNegativeReport>> =
- bincode::deserialize(&db.get("nrs-to-process").unwrap().unwrap()).unwrap();
- let negative_reports = nrs_to_process.get(&map_key_1).unwrap();
- assert_eq!(negative_reports.len(), 1);
- let map_key_2 = format!(
- "{}_{}_{}",
- array_bytes::bytes2hex("", invalid_report_2.fingerprint),
- "ru".to_string(),
- date
- );
- handle_encrypted_negative_report(&db, invalid_report_2.encrypt(&public)); // no change
- let nrs_to_process: BTreeMap<String, Vec<SerializableNegativeReport>> =
- bincode::deserialize(&db.get("nrs-to-process").unwrap().unwrap()).unwrap();
- assert!(!nrs_to_process.contains_key(&map_key_2));
- let map_key_3 = format!(
- "{}_{}_{}",
- array_bytes::bytes2hex("", valid_report_2.fingerprint),
- "ru".to_string(),
- date - 1
- );
- handle_encrypted_negative_report(&db, valid_report_2.encrypt(&public_yesterday));
- let nrs_to_process: BTreeMap<String, Vec<SerializableNegativeReport>> =
- bincode::deserialize(&db.get("nrs-to-process").unwrap().unwrap()).unwrap();
- let negative_reports = nrs_to_process.get(&map_key_3).unwrap();
- assert_eq!(negative_reports.len(), 1);
- handle_encrypted_negative_report(&db, valid_report_3.encrypt(&public));
- let nrs_to_process: BTreeMap<String, Vec<SerializableNegativeReport>> =
- bincode::deserialize(&db.get("nrs-to-process").unwrap().unwrap()).unwrap();
- let negative_reports = nrs_to_process.get(&map_key_1).unwrap();
- assert_eq!(negative_reports.len(), 2);
- }
- #[test]
- fn test_positive_reports() {
- let mut th = TestHarness::new();
- // Get new level 3 credential
- let cred = th.get_new_credential();
- let cred = th.level_up(&cred);
- let cred = th.level_up(&cred);
- let cred = th.level_up(&cred);
- let bridges = th.get_bucket(&cred);
- // Create BridgeVerificationInfo for each bridge
- let mut buckets = HashSet::<Scalar>::new();
- buckets.insert(cred.bucket);
- let bridge_info_1 = BridgeVerificationInfo {
- bridge_line: bridges[0],
- buckets: buckets.clone(),
- pubkey: None,
- };
- let bridge_info_2 = BridgeVerificationInfo {
- bridge_line: bridges[1],
- buckets: buckets.clone(),
- pubkey: None,
- };
- let bridge_info_3 = BridgeVerificationInfo {
- bridge_line: bridges[2],
- buckets: buckets.clone(),
- pubkey: None,
- };
- // Create reports
- let report_1 = PositiveReport::from_lox_credential(
- bridges[0].fingerprint,
- None,
- &cred,
- &th.ba.lox_pub,
- "ru".to_string(),
- )
- .unwrap();
- let report_2 = PositiveReport::from_lox_credential(
- bridges[1].fingerprint,
- None,
- &cred,
- &th.ba.lox_pub,
- "ru".to_string(),
- )
- .unwrap();
- let report_3 = PositiveReport::from_lox_credential(
- bridges[2].fingerprint,
- None,
- &cred,
- &th.ba.lox_pub,
- "ru".to_string(),
- )
- .unwrap();
- // Compute Htable
- let H = lox_library::proto::positive_report::compute_H(report_1.date);
- let Htable = RistrettoBasepointTable::create(&H);
- assert!(report_1.verify(&mut th.ba, &bridge_info_1, &Htable));
- assert!(report_2.verify(&mut th.ba, &bridge_info_2, &Htable));
- assert!(report_3.verify(&mut th.ba, &bridge_info_3, &Htable));
- // Check that user cannot use credential for other bridge
- // Get new credential
- let cred_2 = th.get_new_credential();
- let bridges_2 = th.get_bucket(&cred_2);
- let mut buckets_2 = HashSet::<Scalar>::new();
- buckets_2.insert(cred_2.bucket);
- let bridge_info_4 = BridgeVerificationInfo {
- bridge_line: bridges_2[0],
- buckets: buckets_2.clone(),
- pubkey: None,
- };
- // Use new credential to create positive report even we don't trust it
- let invalid_report_1 = PositiveReport::from_lox_credential(
- bridges_2[0].fingerprint,
- None,
- &cred_2,
- &th.ba.lox_pub,
- "ru".to_string(),
- );
- // Use first credential for bridge from second bucket
- let invalid_report_2 = PositiveReport::from_lox_credential(
- bridges_2[0].fingerprint,
- None,
- &cred,
- &th.ba.lox_pub,
- "ru".to_string(),
- );
- // Use second credential for bridge from first bucket
- let invalid_report_3 = PositiveReport::from_lox_credential(
- bridges[0].fingerprint,
- None,
- &cred_2,
- &th.ba.lox_pub,
- "ru".to_string(),
- );
- // Check that all of these fail
- assert!(invalid_report_1.is_err());
- assert!(!invalid_report_2
- .unwrap()
- .verify(&mut th.ba, &bridge_info_4, &Htable));
- assert!(invalid_report_3.is_err());
- // Check that deserialization fails under invalid conditions
- // Date in the future
- let mut invalid_report_4 = PositiveReport::from_lox_credential(
- bridges[0].fingerprint,
- None,
- &cred,
- &th.ba.lox_pub,
- "ru".to_string(),
- )
- .unwrap()
- .to_serializable_report();
- invalid_report_4.date = invalid_report_4.date + 2;
- // Invalid country code
- let invalid_report_5 = PositiveReport::from_lox_credential(
- bridges[0].fingerprint,
- None,
- &cred,
- &th.ba.lox_pub,
- "xx".to_string(),
- )
- .unwrap()
- .to_serializable_report();
- assert!(invalid_report_4.to_report().is_err());
- assert!(invalid_report_5.to_report().is_err());
- // Test storing to-be-processed positive reports to database
- // Create reports
- let report_1 = PositiveReport::from_lox_credential(
- bridges[0].fingerprint,
- None,
- &cred,
- &th.ba.lox_pub,
- "ru".to_string(),
- )
- .unwrap();
- let report_2 = PositiveReport::from_lox_credential(
- bridges[0].fingerprint,
- None,
- &cred,
- &th.ba.lox_pub,
- "ru".to_string(),
- )
- .unwrap();
- let report_3 = PositiveReport::from_lox_credential(
- bridges[1].fingerprint,
- None,
- &cred,
- &th.ba.lox_pub,
- "ru".to_string(),
- )
- .unwrap();
- // Open test database
- let db: Db = sled::open("test_db").unwrap();
- // Delete all data in test DB
- db.clear().unwrap();
- assert!(!db.contains_key("prs-to-process").unwrap());
- let map_key_1 = format!(
- "{}_{}_{}",
- array_bytes::bytes2hex("", report_1.fingerprint),
- &report_1.country,
- &report_1.date
- );
- let map_key_2 = format!(
- "{}_{}_{}",
- array_bytes::bytes2hex("", report_3.fingerprint),
- &report_3.country,
- &report_3.date
- );
- save_positive_report_to_process(&db, report_1);
- let prs_to_process: BTreeMap<String, Vec<SerializablePositiveReport>> =
- bincode::deserialize(&db.get("prs-to-process").unwrap().unwrap()).unwrap();
- let positive_reports = prs_to_process.get(&map_key_1).unwrap();
- assert_eq!(positive_reports.len(), 1);
- assert!(!prs_to_process.contains_key(&map_key_2));
- save_positive_report_to_process(&db, report_2);
- let prs_to_process: BTreeMap<String, Vec<SerializablePositiveReport>> =
- bincode::deserialize(&db.get("prs-to-process").unwrap().unwrap()).unwrap();
- let positive_reports = prs_to_process.get(&map_key_1).unwrap();
- assert_eq!(positive_reports.len(), 2);
- assert!(!prs_to_process.contains_key(&map_key_2));
- save_positive_report_to_process(&db, report_3);
- let prs_to_process: BTreeMap<String, Vec<SerializablePositiveReport>> =
- bincode::deserialize(&db.get("prs-to-process").unwrap().unwrap()).unwrap();
- // Check that this has not changed
- let positive_reports = prs_to_process.get(&map_key_1).unwrap();
- assert_eq!(positive_reports.len(), 2);
- // New report added to its own collection
- let positive_reports = prs_to_process.get(&map_key_2).unwrap();
- assert_eq!(positive_reports.len(), 1);
- }
- #[test]
- fn test_analysis() {
- // Test stage 1 analysis
- {
- let mut date = get_date();
- // New bridge info
- let mut bridge_info = BridgeInfo::new([0; 20], &String::default());
- bridge_info
- .info_by_country
- .insert("ru".to_string(), BridgeCountryInfo::new());
- let analyzer = analysis::NormalAnalyzer::new(5, 0.25);
- let confidence = 0.95;
- let mut blocking_countries = HashSet::<String>::new();
- // No data today
- assert_eq!(
- blocked_in(&analyzer, &bridge_info, confidence, date),
- blocking_countries
- );
- // 1 connection, 0 negative reports
- date += 1;
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::BridgeIps,
- date,
- 8,
- );
- assert_eq!(
- blocked_in(&analyzer, &bridge_info, confidence, date),
- blocking_countries
- );
- // 0 connections, 0 negative reports
- date += 1;
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::BridgeIps,
- date,
- 0,
- );
- assert_eq!(
- blocked_in(&analyzer, &bridge_info, confidence, date),
- blocking_countries
- );
- // 0 connections, 1 negative report
- // (exceeds scaled threshold)
- date += 1;
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::NegativeReports,
- date,
- 1,
- );
- blocking_countries.insert("ru".to_string());
- assert_eq!(
- blocked_in(&analyzer, &bridge_info, confidence, date),
- blocking_countries
- );
- }
- {
- let mut date = get_date();
- // New bridge info
- let mut bridge_info = BridgeInfo::new([0; 20], &String::default());
- bridge_info
- .info_by_country
- .insert("ru".to_string(), BridgeCountryInfo::new());
- let analyzer = analysis::NormalAnalyzer::new(5, 0.25);
- let confidence = 0.95;
- let mut blocking_countries = HashSet::<String>::new();
- // No data today
- assert_eq!(
- blocked_in(&analyzer, &bridge_info, confidence, date),
- blocking_countries
- );
- // 1 connection, 1 negative report
- date += 1;
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::BridgeIps,
- date,
- 8,
- );
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::NegativeReports,
- date,
- 1,
- );
- assert_eq!(
- blocked_in(&analyzer, &bridge_info, confidence, date),
- blocking_countries
- );
- // 8 connections, 2 negative reports
- date += 1;
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::BridgeIps,
- date,
- 8,
- );
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::NegativeReports,
- date,
- 2,
- );
- assert_eq!(
- blocked_in(&analyzer, &bridge_info, confidence, date),
- blocking_countries
- );
- // 8 connections, 3 negative reports
- // (exceeds scaled threshold)
- date += 1;
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::BridgeIps,
- date,
- 8,
- );
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::NegativeReports,
- date,
- 3,
- );
- blocking_countries.insert("ru".to_string());
- assert_eq!(
- blocked_in(&analyzer, &bridge_info, confidence, date),
- blocking_countries
- );
- }
- {
- let mut date = get_date();
- // New bridge info
- let mut bridge_info = BridgeInfo::new([0; 20], &String::default());
- bridge_info
- .info_by_country
- .insert("ru".to_string(), BridgeCountryInfo::new());
- let analyzer = analysis::NormalAnalyzer::new(5, 0.25);
- let confidence = 0.95;
- let mut blocking_countries = HashSet::<String>::new();
- // 24 connections, 5 negative reports
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::BridgeIps,
- date,
- 24,
- );
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::NegativeReports,
- date,
- 5,
- );
- assert_eq!(
- blocked_in(&analyzer, &bridge_info, confidence, date),
- blocking_countries
- );
- // 24 connections, 6 negative reports
- // (exceeds max threshold)
- date += 1;
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::BridgeIps,
- date,
- 24,
- );
- bridge_info.info_by_country.get_mut("ru").unwrap().add_info(
- BridgeInfoType::NegativeReports,
- date,
- 6,
- );
- blocking_countries.insert("ru".to_string());
- assert_eq!(
- blocked_in(&analyzer, &bridge_info, confidence, date),
- blocking_countries
- );
- }
- // TODO: Test stage 2 analysis
- // TODO: Test stage 3 analysis
- }
|