add string methods

Author: jacksonwalters

benches/benchmark_decrypt.rs

@@ -1,7 +1,7 @@
 use criterion::{criterion_group, criterion_main, Criterion};
-use module_lwe::decrypt::decrypt;
-use module_lwe::keygen::keygen;
-use module_lwe::encrypt::encrypt;
+use module_lwe::decrypt::{decrypt,decrypt_string};
+use module_lwe::keygen::{keygen,keygen_string};
+use module_lwe::encrypt::{encrypt,encrypt_string};
 use module_lwe::utils::Parameters;
 
 fn bench_decrypt(c: &mut Criterion) {
@@ -15,5 +15,18 @@ fn bench_decrypt(c: &mut Criterion) {
     });
 }
 
-criterion_group!(benches, bench_decrypt);
+fn bench_decrypt_string(c: &mut Criterion) {
+    let params = Parameters::default();
+    let keypair = keygen_string(&params, None);
+    let sk_string = keypair.get("secret").unwrap();
+    let pk_string = keypair.get("public").unwrap();
+    let message = String::from("hello");
+    let ciphertext_string = encrypt_string(&pk_string, &message, &params, None);
+    
+    c.bench_function("decrypt_string", |b| {
+        b.iter(|| decrypt_string(&sk_string, &ciphertext_string, &params))
+    });
+}
+
+criterion_group!(benches, bench_decrypt, bench_decrypt_string);
 criterion_main!(benches);

benches/benchmark_encrypt.rs

@@ -1,6 +1,6 @@
 use criterion::{criterion_group, criterion_main, Criterion};
-use module_lwe::encrypt::encrypt;
-use module_lwe::keygen::keygen;
+use module_lwe::encrypt::{encrypt,encrypt_string};
+use module_lwe::keygen::{keygen,keygen_string};
 use module_lwe::utils::Parameters;
 
 fn bench_encrypt(c: &mut Criterion) {
@@ -13,5 +13,16 @@ fn bench_encrypt(c: &mut Criterion) {
     });
 }
 
-criterion_group!(benches, bench_encrypt);
+fn bench_encrypt_string(c: &mut Criterion) {
+    let params = Parameters::default();
+    let keypair = keygen_string(&params, None);
+    let pk_string = keypair.get("public").unwrap();
+    let message = String::from("hello");
+
+    c.bench_function("encrypt_string", |b| {
+        b.iter(|| encrypt_string(&pk_string, &message, &params, None))
+    });
+}
+
+criterion_group!(benches, bench_encrypt, bench_encrypt_string);
 criterion_main!(benches);

benches/benchmark_keygen.rs

@@ -1,5 +1,5 @@
 use criterion::{criterion_group, criterion_main, Criterion};
-use module_lwe::keygen::keygen;
+use module_lwe::keygen::{keygen,keygen_string};
 use module_lwe::utils::Parameters;
 
 fn bench_keygen(c: &mut Criterion) {
@@ -9,5 +9,13 @@ fn bench_keygen(c: &mut Criterion) {
     });
 }
 
-criterion_group!(benches, bench_keygen);
+fn bench_keygen_string(c: &mut Criterion) {
+    let params = Parameters::default();
+    
+    c.bench_function("keygen_string", |b| {
+        b.iter(|| keygen_string(&params, None))
+    });
+}
+
+criterion_group!(benches, bench_keygen, bench_keygen_string);
 criterion_main!(benches);