dilithium, hybrid, ctap implementation, experiments

third_party/dilithium/examples/sign.rs

@@ -0,0 +1,41 @@
+// Command for changing the stack size:
+// cargo run --example sign --features std -- --stack-size-kb (new value in KB)
+
+extern crate dilithium;
+extern crate rng256;
+extern crate structopt;
+
+use dilithium::sign::SecKey;
+use rng256::Rng256;
+use std::thread;
+use structopt::StructOpt;
+
+const DEFAULT_STACK_SIZE_KB: &str = "81";
+
+#[derive(Debug, StructOpt)]
+struct Opts {
+    #[structopt(long, default_value=DEFAULT_STACK_SIZE_KB)]
+    stack_size_kb: usize,
+}
+
+fn run() {
+    let mut rng = rng256::ThreadRng256 {};
+
+    let sk = SecKey::gensk(&mut rng);
+    let mut message = [0; 59];
+    rng.fill_bytes(&mut message);
+    sk.sign(&message);
+}
+
+fn main() {
+    let stack_size_kb = Opts::from_args().stack_size_kb;
+
+    // We bound the stack size for generating keys and signing in Dilithium.
+    let child = thread::Builder::new()
+        .stack_size(stack_size_kb * 1024)
+        .spawn(run)
+        .unwrap();
+
+    // Wait for thread to join
+    child.join().unwrap();
+}