vecna
/
lox
mirror of https://gitlab.torproject.org/vecna/lox.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added serialized sizes for request and response in open invite

This commit is contained in:
onyinyang 2021-06-01 15:07:28 -04:00
parent 795fadeaef
commit 668cf86893
1 changed files with 23 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
crates/lox-library/src/tests.rs
View File

@ -45,14 +45,25 @@ impl TestHarness {
self.ba.advance_days(days); self.ba.advance_days(days);
} }
fn open_invite(&mut self) -> (cred::Lox, bridge_table::BridgeLine) { fn open_invite(&mut self) -> (usize, usize, (cred::Lox, bridge_table::BridgeLine)) {
// Issue an open invitation // Issue an open invitation
let inv = self.bdb.invite(); let inv = self.bdb.invite();
// Use it to get a Lox credential // Use it to get a Lox credential
let (req, state) = open_invite::request(&inv); let (req, state) = open_invite::request(&inv);
let resp = self.ba.handle_open_invite(req).unwrap(); let encoded: Vec<u8> = bincode::serialize(&req).unwrap();
open_invite::handle_response(state, resp, &self.ba.lox_pub).unwrap() let req_len = encoded.len();
let decoded: open_invite::Request = bincode::deserialize(&encoded[..]).unwrap();
let resp = self.ba.handle_open_invite(decoded).unwrap();
let encoded_resp: Vec<u8> = bincode::serialize(&resp).unwrap();
let resp_len = encoded_resp.len();
let decode_resp: open_invite::Response = bincode::deserialize(&encoded_resp[..]).unwrap();
return (
req_len,
resp_len,
(open_invite::handle_response(state, decode_resp, &self.ba.lox_pub).unwrap()),
);
} }
fn trust_promotion(&mut self, cred: &cred::Lox) -> cred::Migration { fn trust_promotion(&mut self, cred: &cred::Lox) -> cred::Migration {
@ -141,13 +152,15 @@ fn test_open_invite() {
let mut th = TestHarness::new(); let mut th = TestHarness::new();
// Join an untrusted user // Join an untrusted user
let (cred, bridgeline) = th.open_invite(); let (ser_req, ser_resp, (cred, bridgeline)) = th.open_invite();
// Check that we can use the credential to read a bucket // Check that we can use the credential to read a bucket
let (id, key) = bridge_table::from_scalar(cred.bucket).unwrap(); let (id, key) = bridge_table::from_scalar(cred.bucket).unwrap();
let encbuckets = th.ba.enc_bridge_table(); let encbuckets = th.ba.enc_bridge_table();
let bucket = let bucket =
bridge_table::BridgeTable::decrypt_bucket(id, &key, &encbuckets[id as usize]).unwrap(); bridge_table::BridgeTable::decrypt_bucket(id, &key, &encbuckets[id as usize]).unwrap();
println!("Request size = {:?}", ser_req);
println!("Response size = {:?}", ser_resp);
println!("cred = {:?}", cred); println!("cred = {:?}", cred);
println!("bucket = {:?}", bucket); println!("bucket = {:?}", bucket);
println!("bridgeline = {:?}", bridgeline); println!("bridgeline = {:?}", bridgeline);
@ -160,7 +173,7 @@ fn test_open_invite() {
fn test_trust_promotion() { fn test_trust_promotion() {
let mut th = TestHarness::new(); let mut th = TestHarness::new();
let cred = th.open_invite().0; let cred = th.open_invite().2 .0;
assert!(th.ba.verify_lox(&cred)); assert!(th.ba.verify_lox(&cred));
// Time passes // Time passes
@ -183,7 +196,7 @@ fn test_trust_promotion() {
fn test_level0_migration() { fn test_level0_migration() {
let mut th = TestHarness::new(); let mut th = TestHarness::new();
let cred = th.open_invite().0; let cred = th.open_invite().2 .0;
assert!(th.ba.verify_lox(&cred)); assert!(th.ba.verify_lox(&cred));
// Time passes // Time passes
@ -210,7 +223,7 @@ fn test_level_up() {
let mut th = TestHarness::new(); let mut th = TestHarness::new();
// Join an untrusted user // Join an untrusted user
let cred = th.open_invite().0; let cred = th.open_invite().2 .0;
// Time passes // Time passes
th.advance_days(47); th.advance_days(47);
@ -250,7 +263,7 @@ fn test_issue_invite() {
let mut th = TestHarness::new(); let mut th = TestHarness::new();
// Join an untrusted user // Join an untrusted user
let cred = th.open_invite().0; let cred = th.open_invite().2 .0;
// Time passes // Time passes
th.advance_days(47); th.advance_days(47);
@ -282,7 +295,7 @@ fn test_redeem_invite() {
let mut th = TestHarness::new(); let mut th = TestHarness::new();
// Join an untrusted user // Join an untrusted user
let cred = th.open_invite().0; let cred = th.open_invite().2 .0;
// Time passes // Time passes
th.advance_days(47); th.advance_days(47);
@ -372,7 +385,7 @@ fn test_blockage_migration() {
let mut th = TestHarness::new(); let mut th = TestHarness::new();
// Join an untrusted user // Join an untrusted user
let cred = th.open_invite().0; let cred = th.open_invite().2 .0;
// Time passes // Time passes
th.advance_days(47); th.advance_days(47);