From 746167e0a8f74c76b02ca0bdda7749188e6fc809 Mon Sep 17 00:00:00 2001
From: Amos Wenger <amos@bearcove.net>
Date: Mon, 11 Mar 2024 13:31:15 +0100
Subject: [PATCH] TLS 1.3 codepath still does the right thing in rustls 0.22.2

---
 Cargo.toml |  2 +-
 src/ffi.rs | 34 ++++++++++++++++++++++++++++------
 2 files changed, 29 insertions(+), 7 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 4c78ceb..d29db5f 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -15,7 +15,7 @@ libc = { version = "0.2.148", features = ["const-extern-fn"] }
 thiserror = "1.0.49"
 tracing = "0.1.37"
 tokio-rustls = "0.25.0"
-rustls = { version = "0.22" }
+rustls = { version = "0.22.2" }
 smallvec = "1.11.1"
 memoffset = "0.9.0"
 pin-project-lite = "0.2.13"
diff --git a/src/ffi.rs b/src/ffi.rs
index 250a33e..472e3d4 100644
--- a/src/ffi.rs
+++ b/src/ffi.rs
@@ -139,10 +139,10 @@ impl CryptoInfo {
 
         Ok(match secrets {
             ConnectionTrafficSecrets::Aes128Gcm { key, iv } => {
-                // see https://github.com/rustls/rustls/issues/1833,
-                // between rustls 0.21 and 0.22, the extract_keys codepath
-                // was changed, so it always returns AesGcm128, even if
-                // the cipher suite is Aes256Gcm.
+                // see https://github.com/rustls/rustls/issues/1833, between
+                // rustls 0.21 and 0.22, the extract_keys codepath was changed,
+                // so, for TLS 1.2, both GCM-128 and GCM-256 return the
+                // Aes128Gcm variant.
 
                 match key.as_ref().len() {
                     16 => CryptoInfo::AesGcm128(ktls::tls12_crypto_info_aes_gcm_128 {
@@ -194,8 +194,30 @@ impl CryptoInfo {
                     _ => unreachable!("GCM key length is not 16 or 32"),
                 }
             }
-            ConnectionTrafficSecrets::Aes256Gcm { .. } => {
-                unreachable!("a bug in rustls 0.22 means this codepath is dead. when we can upgrade to 0.23, we should fix this. see https://github.com/rustls/rustls/issues/1833")
+            ConnectionTrafficSecrets::Aes256Gcm { key, iv } => {
+                CryptoInfo::AesGcm256(ktls::tls12_crypto_info_aes_gcm_256 {
+                    info: ktls::tls_crypto_info {
+                        version,
+                        cipher_type: ktls::TLS_CIPHER_AES_GCM_256 as _,
+                    },
+                    iv: iv
+                        .as_ref()
+                        .get(4..)
+                        .expect("AES-GCM-256 iv is 8 bytes")
+                        .try_into()
+                        .expect("AES-GCM-256 iv is 8 bytes"),
+                    key: key
+                        .as_ref()
+                        .try_into()
+                        .expect("AES-GCM-256 key is 32 bytes"),
+                    salt: iv
+                        .as_ref()
+                        .get(..4)
+                        .expect("AES-GCM-256 salt is 4 bytes")
+                        .try_into()
+                        .expect("AES-GCM-256 salt is 4 bytes"),
+                    rec_seq: seq.to_be_bytes(),
+                })
             }
             ConnectionTrafficSecrets::Chacha20Poly1305 { key, iv } => {
                 CryptoInfo::Chacha20Poly1305(ktls::tls12_crypto_info_chacha20_poly1305 {