Support custom signing routines in SignaturePrivateKey (#424)

This enables support for signing with non-extractable private keys (e.g. TPM-bound).

Author: rcombs

include/mls/crypto.h

@@ -254,12 +254,15 @@ struct PublicJWK
 
 struct SignaturePrivateKey
 {
+  using SignerFunc = std::function<bytes(const std::vector<uint8_t>&)>;
+
   static SignaturePrivateKey generate(CipherSuite suite);
   static SignaturePrivateKey parse(CipherSuite suite, const bytes& data);
   static SignaturePrivateKey parse_der(CipherSuite suite, const bytes& data);
   static SignaturePrivateKey derive(CipherSuite suite, const bytes& secret);
   static SignaturePrivateKey from_jwk(CipherSuite suite,
                                       const std::string& json_str);
+  static SignaturePrivateKey from_func(SignerFunc func, bytes pub_data);
 
   SignaturePrivateKey() = default;
 
@@ -276,7 +279,11 @@ struct SignaturePrivateKey
   TLS_SERIALIZABLE(data)
 
 private:
-  SignaturePrivateKey(bytes priv_data, bytes pub_data);
+  SignerFunc _sign_func;
+
+  SignaturePrivateKey(bytes priv_data,
+                      bytes pub_data,
+                      SignerFunc func = SignerFunc());
 };
 
 } // namespace MLS_NAMESPACE

src/crypto.cpp

@@ -505,19 +505,31 @@ SignaturePrivateKey::sign(const CipherSuite& suite,
 {
   auto label_plus = mls_1_0_plus(label);
   const auto content = tls::marshal(SignContent{ label_plus, message });
+
+  if (_sign_func) {
+    return _sign_func(content);
+  }
+
   const auto priv = suite.sig().deserialize_private(data);
   return suite.sig().sign(content, *priv);
 }
 
-SignaturePrivateKey::SignaturePrivateKey(bytes priv_data, bytes pub_data)
+SignaturePrivateKey::SignaturePrivateKey(bytes priv_data,
+                                         bytes pub_data,
+                                         SignerFunc func)
   : data(std::move(priv_data))
   , public_key{ std::move(pub_data) }
+  , _sign_func{ std::move(func) }
 {
 }
 
 void
 SignaturePrivateKey::set_public_key(CipherSuite suite)
 {
+  if (_sign_func) {
+    throw std::runtime_error("not implemented");
+  }
+
   const auto priv = suite.sig().deserialize_private(data);
   auto pub = priv->public_key();
   public_key.data = suite.sig().serialize(*pub);
@@ -533,9 +545,19 @@ SignaturePrivateKey::from_jwk(CipherSuite suite, const std::string& json_str)
   return { priv_data, pub_data };
 }
 
+SignaturePrivateKey
+SignaturePrivateKey::from_func(SignerFunc func, bytes pub_data)
+{
+  return { bytes(), std::move(pub_data), std::move(func) };
+}
+
 std::string
 SignaturePrivateKey::to_jwk(CipherSuite suite) const
 {
+  if (_sign_func) {
+    throw std::runtime_error("not implemented");
+  }
+
   const auto priv = suite.sig().deserialize_private(data);
   return suite.sig().export_jwk_private(*priv);
 }