kat: add KEM schemes

test/katrunner/src/main.rs

@@ -13,7 +13,18 @@ macro_rules! REG_SIGN {
                 scheme_type: AlgType::AlgSignature,
                 scheme_id: $ID,
                 kat_file: $F},
-				execfn: signature_scheme}
+                execfn: test_sign_vector}
+    }
+}
+
+macro_rules! REG_KEM {
+    ($ID:expr,$F:expr) => {
+        Register{
+            kat:Kat{
+                scheme_type: AlgType::AlgKem,
+                scheme_id: $ID,
+                kat_file: $F},
+                execfn: test_kem_vector}
     }
 }
 
@@ -24,8 +35,7 @@ struct Register {
     execfn: ExecFn,
 }
 
-fn signature_scheme(el: &TestVector) {
+fn test_sign_vector(el: &TestVector) {
-
     unsafe {
         let p = pqc_sig_alg_by_id(el.scheme_id as u8);
         assert_ne!(p.is_null(), true);
@@ -43,6 +53,19 @@ fn signature_scheme(el: &TestVector) {
     }
 }
 
+fn test_kem_vector(el: &TestVector) {
+    let mut ss = Vec::new();
+
+    ss.resize(el.kem.ss.len(), 0);
+    unsafe {
+        let p = pqc_kem_alg_by_id(el.scheme_id as u8);
+        assert_ne!(p.is_null(), true);
+        assert_eq!(
+            pqc_kem_decapsulate(p, ss.as_mut_ptr(), el.kem.ct.as_ptr(), el.kem.sk.as_ptr()),
+            true);
+    }
+}
+
 // KAT test register
 const KATS: &'static[Register] = &[
     REG_SIGN!(DILITHIUM2, "round3/dilithium/dilithium2/PQCsignKAT_2544.rsp"),
@@ -50,16 +73,12 @@ const KATS: &'static[Register] = &[
     REG_SIGN!(DILITHIUM5, "round3/dilithium/dilithium5/PQCsignKAT_4880.rsp"),
     //REG_SIGN!(FALCON512,  "round3/falcon/falcon512-KAT.rsp"),
     //REG_SIGN!(FALCON1024, "round3/falcon/falcon1024-KAT.rsp"),
-	//REG_SIGN!(RAINBOWVCLASSIC),
-    //REG_SIGN!(RAINBOWICLASSIC),
-    //REG_SIGN!(RAINBOWIIICLASSIC),
 
     // Some implementations of sphincs are for round3
     REG_SIGN!(SPHINCSSHA256128SSIMPLE,"round3/sphincs/sphincs-sha256-128s-simple/PQCsignKAT_64.rsp"),
     REG_SIGN!(SPHINCSSHA256128SROBUST,"round3/sphincs/sphincs-sha256-128s-robust/PQCsignKAT_64.rsp"),
     REG_SIGN!(SPHINCSSHA256128FSIMPLE,"round3/sphincs/sphincs-sha256-128f-simple/PQCsignKAT_64.rsp"),
     REG_SIGN!(SPHINCSSHA256128FROBUST,"round3/sphincs/sphincs-sha256-128f-robust/PQCsignKAT_64.rsp"),
-
     REG_SIGN!(SPHINCSSHA256192SSIMPLE,"round3/sphincs/sphincs-sha256-192s-simple/PQCsignKAT_96.rsp"),
     REG_SIGN!(SPHINCSSHA256192SROBUST,"round3/sphincs/sphincs-sha256-192s-robust/PQCsignKAT_96.rsp"),
     REG_SIGN!(SPHINCSSHA256192FSIMPLE,"round3/sphincs/sphincs-sha256-192f-simple/PQCsignKAT_96.rsp"),
@@ -80,6 +99,28 @@ const KATS: &'static[Register] = &[
     REG_SIGN!(SPHINCSSHAKE256256FSIMPLE,"round3/sphincs/sphincs-shake256-256f-simple/PQCsignKAT_128.rsp"),
     REG_SIGN!(SPHINCSSHAKE256256SSIMPLE,"round3/sphincs/sphincs-shake256-256s-simple/PQCsignKAT_128.rsp"),
     REG_SIGN!(SPHINCSSHAKE256256SROBUST,"round3/sphincs/sphincs-shake256-256s-robust/PQCsignKAT_128.rsp"),
+
+    // KEM Schemes
+    REG_KEM!(FRODOKEM640SHAKE, "round3/frodokem/PQCkemKAT_19888_shake.rsp"),
+    REG_KEM!(FRODOKEM976SHAKE, "round3/frodokem/PQCkemKAT_31296_shake.rsp"),
+    REG_KEM!(FRODOKEM1344SHAKE, "round3/frodokem/PQCkemKAT_43088_shake.rsp"),
+    REG_KEM!(KYBER768, "round3/kyber/kyber768/PQCkemKAT_2400.rsp"),
+    REG_KEM!(KYBER1024, "round3/kyber/kyber1024/PQCkemKAT_3168.rsp"),
+    REG_KEM!(KYBER512, "round3/kyber/kyber512/PQCkemKAT_1632.rsp"),
+    REG_KEM!(NTRUHPS4096821, "round3/ntru/ntruhps4096821/PQCkemKAT_1590.rsp"),
+    REG_KEM!(NTRUHPS2048509, "round3/ntru/ntruhps2048509/PQCkemKAT_935.rsp"),
+    REG_KEM!(NTRUHRSS701, "round3/ntru/ntruhrss701/PQCkemKAT_1450.rsp"),
+    REG_KEM!(NTRUHPS2048677, "round3/ntru/ntruhps2048677/PQCkemKAT_1234.rsp"),
+    REG_KEM!(NTRULPR761, "round3/ntrup/ntrulpr761/kat_kem.rsp"),
+    REG_KEM!(NTRULPR653, "round3/ntrup/ntrulpr653/kat_kem.rsp"),
+    REG_KEM!(NTRULPR857, "round3/ntrup/ntrulpr857/kat_kem.rsp"),
+    REG_KEM!(LIGHTSABER, "round3/saber/LightSaber/PQCkemKAT_1568.rsp"),
+    REG_KEM!(FIRESABER, "round3/saber/FireSaber/PQCkemKAT_3040.rsp"),
+    REG_KEM!(SABER, "round3/saber/Saber/PQCkemKAT_2304.rsp"),
+    // Those are Round2. KATs are very big, so skip testing until it makes sense to do so.
+    //REG_SIGN!(RAINBOWVCLASSIC),
+    //REG_SIGN!(RAINBOWICLASSIC),
+    //REG_SIGN!(RAINBOWIIICLASSIC),
 ];
 
 fn execute(kat_dir: String) {