12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- use rand::Rng;
- use rand::SeedableRng;
- use rand::thread_rng;
- use rand_chacha::ChaCha20Rng;
- use spiral_rs::client::*;
- use spiral_rs::server::*;
- use spiral_rs::util::*;
- use spiral_rs::arith::*;
- use spiral_rs::params::*;
- use std::env;
- use std::fs;
- use std::time::Instant;
- fn print_params_summary(params: &Params) {
- let db_elem_size = params.item_size();
- let total_size = params.num_items() * db_elem_size;
- println!("Using a {} x {} byte database ({} bytes total)", params.num_items(), db_elem_size, total_size);
- }
- fn main() {
- let params;
- let args: Vec<String> = env::args().collect();
- if args.len() == 2 {
- let inp_params_fname = &args[1];
- let params_json_str = fs::read_to_string(inp_params_fname).unwrap();
- params = params_from_json(¶ms_json_str);
- } else {
- let target_num_log2: usize = args[1].parse().unwrap();
- let item_size_bytes: usize = args[2].parse().unwrap();
- params = get_params_from_store(target_num_log2, item_size_bytes);
- }
- print_params_summary(¶ms);
- let mut rng = thread_rng();
- let idx_target: usize = rng.gen::<usize>() % params.num_items();
- println!("fetching index {} out of {} items", idx_target, params.num_items());
- println!("initializing client");
- let mut client = Client::init(¶ms, ChaCha20Rng::from_entropy);
- println!("generating public parameters");
- let pub_params = client.generate_keys();
- let pub_params_buf = pub_params.serialize();
- println!("public parameters size: {} bytes", pub_params_buf.len());
- let query = client.generate_query(idx_target);
- let query_buf = query.serialize();
- println!("initial query size: {} bytes", query_buf.len());
- println!("generating db");
- let (corr_item, db) = generate_random_db_and_get_item(¶ms, idx_target);
- println!("processing query");
- let now = Instant::now();
- let response = process_query(¶ms, &pub_params, &query, db.as_slice());
- println!("done processing (took {} us).", now.elapsed().as_micros());
- println!("response size: {} bytes", response.len());
- println!("decoding response");
- let result = client.decode_response(response.as_slice());
- let p_bits = log2_ceil(params.pt_modulus) as usize;
- let corr_result = corr_item.to_vec(p_bits, params.modp_words_per_chunk());
- assert_eq!(result.len(), corr_result.len());
- for z in 0..corr_result.len() {
- assert_eq!(result[z], corr_result[z], "error in response at {:?}", z);
- }
- println!("completed correctly!");
- }
|