key package KAT from OpenMLS

tls_codec_derive/tests/decode.rs

@@ -138,3 +138,82 @@ fn simple_struct() {
     let deserialized = ExtensionTypeVec::tls_deserialize(&mut b).unwrap();
     assert_eq!(extension, deserialized);
 }
+
+// KAT from MLS
+
+#[derive(TlsSerialize, TlsDeserialize, Clone, PartialEq)]
+#[repr(u8)]
+enum ProtocolVersion {
+    Reserved = 0,
+    Mls10 = 1,
+}
+
+#[derive(TlsSerialize, TlsDeserialize, Clone, PartialEq)]
+struct CipherSuite(u16);
+
+#[derive(TlsSerialize, TlsDeserialize, Clone, PartialEq)]
+struct HPKEPublicKey(TlsVecU16<u8>);
+
+#[derive(TlsSerialize, TlsDeserialize, Clone, PartialEq)]
+struct CredentialType(u16);
+
+#[derive(TlsSerialize, TlsDeserialize, Clone, PartialEq)]
+struct SignatureScheme(u16);
+
+#[derive(TlsSerialize, TlsDeserialize, Clone, PartialEq)]
+struct BasicCredential {
+    identity: TlsVecU16<u8>,
+    signature_scheme: SignatureScheme,
+    signature_key: TlsVecU16<u8>,
+}
+
+#[derive(TlsSerialize, TlsDeserialize, Clone, PartialEq)]
+struct Credential {
+    credential_type: CredentialType,
+    credential: BasicCredential,
+}
+
+#[derive(TlsSerialize, TlsDeserialize, Clone, PartialEq)]
+struct Extension {
+    extension_type: ExtensionType,
+    extension_data: TlsVecU32<u8>,
+}
+
+#[derive(TlsSerialize, TlsDeserialize, Clone, PartialEq)]
+struct KeyPackage {
+    version: ProtocolVersion,
+    cipher_suite: CipherSuite,
+    hpke_init_key: HPKEPublicKey,
+    credential: Credential,
+    extensions: TlsVecU32<Extension>,
+    signature: TlsVecU16<u8>,
+}
+
+#[test]
+fn kat_mls_key_package() {
+    let key_package_bytes = &[
+        0x01u8, 0x00, 0x01, 0x00, 0x20, 0xF2, 0xBC, 0xD8, 0x95, 0x19, 0xDD, 0x1D, 0x06, 0x9F, 0x8B,
+        0x9E, 0xB2, 0xEC, 0xBA, 0xA9, 0xF1, 0x67, 0xAA, 0xCC, 0x52, 0xE7, 0x4D, 0x8D, 0xFE, 0xCC,
+        0xAA, 0xA3, 0xF9, 0xCF, 0x92, 0xAA, 0x35, 0x00, 0x01, 0x00, 0x0D, 0x4F, 0x70, 0x65, 0x6E,
+        0x4D, 0x4C, 0x53, 0x20, 0x72, 0x6F, 0x63, 0x6B, 0x73, 0x08, 0x07, 0x00, 0x20, 0xD8, 0x0F,
+        0x6A, 0x71, 0xFD, 0x5F, 0xB5, 0xEF, 0x27, 0x13, 0xE0, 0xA1, 0xD4, 0xC9, 0x28, 0x5D, 0xD2,
+        0x4A, 0x5A, 0x5B, 0x21, 0xCC, 0xF5, 0x13, 0x4F, 0xDF, 0xE8, 0x25, 0xB2, 0xA6, 0x17, 0x18,
+        0x00, 0x00, 0x00, 0x29, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0D, 0x02, 0x01, 0xC8, 0x02, 0x00,
+        0x01, 0x06, 0x00, 0x01, 0x00, 0x02, 0x00, 0x03, 0x00, 0x02, 0x00, 0x00, 0x00, 0x10, 0x00,
+        0x00, 0x00, 0x00, 0x60, 0xA5, 0xEF, 0xCE, 0x00, 0x00, 0x00, 0x00, 0x61, 0x14, 0xBB, 0xDE,
+        0x00, 0x40, 0x73, 0xE2, 0xC6, 0x9A, 0x23, 0x49, 0x24, 0x04, 0x8A, 0xC3, 0x19, 0x20, 0x72,
+        0x1A, 0x14, 0xBB, 0x92, 0x52, 0x33, 0x29, 0xB0, 0xDD, 0x36, 0x08, 0x35, 0xA1, 0x6F, 0xCA,
+        0xA7, 0x64, 0xB2, 0x2A, 0xC3, 0x5C, 0x47, 0x6D, 0x0C, 0x6A, 0x8E, 0x28, 0x86, 0x96, 0x94,
+        0xB6, 0xC4, 0xE0, 0x2C, 0xBE, 0x00, 0xB6, 0x2D, 0x50, 0xA7, 0x39, 0xF9, 0x30, 0xA7, 0x3F,
+        0x09, 0xFC, 0xA1, 0xF4, 0x53, 0x03,
+    ];
+    let key_package = KeyPackage::tls_deserialize(&mut (key_package_bytes as &[u8]))
+        .expect("Error deserializing key package.");
+    let serialized_key_package = key_package
+        .tls_serialize_detached()
+        .expect("Error serializing key package.");
+    assert_eq!(
+        key_package_bytes as &[u8],
+        serialized_key_package.as_slice()
+    );
+}