From 4359dec952b79dddfef88801791425277ffb5510 Mon Sep 17 00:00:00 2001 From: Adam Cattermole Date: Tue, 8 Oct 2024 11:56:29 +0100 Subject: [PATCH] Process headers_to_remove Signed-off-by: Adam Cattermole --- src/service/auth.rs | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/src/service/auth.rs b/src/service/auth.rs index e66ab51..7a177f4 100644 --- a/src/service/auth.rs +++ b/src/service/auth.rs @@ -102,6 +102,17 @@ impl AuthService { match check_response.http_response { Some(CheckResponse_oneof_http_response::ok_response(ok_response)) => { debug!("process_auth_grpc_response: received OkHttpResponse"); + ok_response + .get_headers_to_remove() + .iter() + .for_each(|header| { + hostcalls::set_map_value( + MapType::HttpResponseHeaders, + header.as_str(), + None, + ) + .unwrap() + }); ok_response .get_response_headers_to_add() .iter() @@ -116,8 +127,7 @@ impl AuthService { Ok(()) } Some(CheckResponse_oneof_http_response::denied_response(denied_response)) => { - debug!("process_auth_grpc_response: received DeniedHttpResponse",); - + debug!("process_auth_grpc_response: received DeniedHttpResponse"); let mut response_headers = vec![]; let status_code = denied_response.get_status().code; denied_response.get_headers().iter().for_each(|header| {