From bd01eec38122ce4ba4b5cb36c979989b71ab36a9 Mon Sep 17 00:00:00 2001
From: Matthias Sefrin <729145+segfault79@users.noreply.github.com>
Date: Thu, 15 Dec 2022 17:37:07 +0100
Subject: [PATCH] Update flask_lambda.py

Add default values for HTTP_X_FORWARDED* headers. Needed e.g. when API Gateway is configured with a private endpoint.
---
 flask_lambda.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/flask_lambda.py b/flask_lambda.py
index 7ab97af..725b998 100644
--- a/flask_lambda.py
+++ b/flask_lambda.py
@@ -52,6 +52,10 @@ def make_environ(event):
         http_hdr_name = 'HTTP_%s' % hdr_name
         environ[http_hdr_name] = hdr_value
 
+    # Apply defaults for API Gateway private endpoint/lambda proxy integration mode.
+    defaults = {'HTTP_X_FORWARDED_PORT': '443', 'HTTP_X_FORWARDED_PROTO': 'https'}
+    environ = dict(defaults, **environ)
+
     qs = event['queryStringParameters']
 
     environ['REQUEST_METHOD'] = event['httpMethod']