[bindings] Add private key callback (#3847)

aws · Mar 1, 2023 · 274f321 · 274f321
1 parent 8062414
commit 274f321
Show file tree

Hide file tree

Showing 9 changed files with 645 additions and 20 deletions.
diff --git a/bindings/rust/s2n-tls/Cargo.toml b/bindings/rust/s2n-tls/Cargo.toml
@@ -24,3 +24,4 @@ pin-project-lite = "0.2"
 [dev-dependencies]
 bytes = { version = "1" }
 futures-test = "0.3"
+openssl = { version = "0.10" }
diff --git a/bindings/rust/s2n-tls/src/callbacks.rs b/bindings/rust/s2n-tls/src/callbacks.rs
@@ -32,6 +32,9 @@ pub use async_cb::*;
 mod client_hello;
 pub use client_hello::*;
 
+mod pkey;
+pub use pkey::*;
+
 /// Convert the connection pointer provided to a callback into a Connection
 /// and Context useable with the Rust bindings.
 ///

diff --git a/bindings/rust/s2n-tls/src/callbacks/async_cb.rs b/bindings/rust/s2n-tls/src/callbacks/async_cb.rs
@@ -102,7 +102,7 @@ impl ConnectionFuture for OptionalFuture {
 #[derive(PartialEq)]
 enum MarkDone {
     ClientHello,
-    // None,
+    None,
 }
 
 pin_project! {
@@ -141,14 +141,11 @@ impl AsyncCallback {
         CallbackResult::Success
     }
 
-    // pub(crate) fn trigger(
-    //    future: ConnectionFutureResult,
-    //    conn: &mut Connection,
-    //) -> CallbackResult {
-    //    let future = OptionalFuture::new(future);
-    //    let cleanup = MarkDone::None;
-    //    let callback = AsyncCallback { future, cleanup };
-    //    conn.set_async_callback(callback);
-    //    CallbackResult::Success
-    //}
+    pub(crate) fn trigger(future: ConnectionFutureResult, conn: &mut Connection) -> CallbackResult {
+        let future = OptionalFuture::new(future);
+        let cleanup = MarkDone::None;
+        let callback = AsyncCallback { future, cleanup };
+        conn.set_async_callback(callback);
+        CallbackResult::Success
+    }
 }