From 35f164cdb6324c6dfb635f8de93a8221861a5991 Mon Sep 17 00:00:00 2001
From: cel 🌸 <cel@bunny.garden>
Date: Sun, 24 Nov 2024 02:04:45 +0000
Subject: implement starttls

---
 src/connection.rs | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

(limited to 'src/connection.rs')

diff --git a/src/connection.rs b/src/connection.rs
index 89f382f..2b70747 100644
--- a/src/connection.rs
+++ b/src/connection.rs
@@ -27,8 +27,11 @@ impl Connection {
         match self {
             Connection::Encrypted(j) => Ok(j),
             Connection::Unencrypted(mut j) => {
+                j.start_stream().await?;
                 info!("upgrading connection to tls");
-                Ok(j.starttls().await?)
+                j.get_features().await?;
+                let j = j.starttls().await?;
+                Ok(j)
             }
         }
     }
@@ -179,4 +182,14 @@ mod tests {
     async fn connect() {
         Connection::connect("blos.sm").await.unwrap();
     }
+
+    #[test(tokio::test)]
+    async fn test_tls() {
+        Connection::connect("blos.sm")
+            .await
+            .unwrap()
+            .ensure_tls()
+            .await
+            .unwrap();
+    }
 }
-- 
cgit