From 9c5f4ebe031566e6b1cecbffbbb57f06e0169f31 Mon Sep 17 00:00:00 2001
From: David Pedersen <david.pdrsn@gmail.com>
Date: Sun, 30 Apr 2023 11:54:41 +0200
Subject: [PATCH] Don't remove the `Sec-WebSocket-Key` header in
 `WebSocketUpgrade` (#1972)

---
 axum/CHANGELOG.md      | 4 +++-
 axum/src/extract/ws.rs | 5 +++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/axum/CHANGELOG.md b/axum/CHANGELOG.md
index b0da2436..b669231c 100644
--- a/axum/CHANGELOG.md
+++ b/axum/CHANGELOG.md
@@ -7,7 +7,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 # Unreleased
 
-- None.
+- **fixed:** Don't remove the `Sec-WebSocket-Key` header in `WebSocketUpgrade` ([#1972])
+
+[#1972]: https://github.com/tokio-rs/axum/pull/1972
 
 # 0.6.17 (25. April, 2023)
 
diff --git a/axum/src/extract/ws.rs b/axum/src/extract/ws.rs
index d785a800..c2c1d279 100644
--- a/axum/src/extract/ws.rs
+++ b/axum/src/extract/ws.rs
@@ -391,8 +391,9 @@ where
 
         let sec_websocket_key = parts
             .headers
-            .remove(header::SEC_WEBSOCKET_KEY)
-            .ok_or(WebSocketKeyHeaderMissing)?;
+            .get(header::SEC_WEBSOCKET_KEY)
+            .ok_or(WebSocketKeyHeaderMissing)?
+            .clone();
 
         let on_upgrade = parts
             .extensions