darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-18 05:31:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

Added a unit test to validate traffic from more than 3 hops

Browse Source
This commit is contained in:
venusbb
2017-07-11 09:59:41 +01:00
parent 5d57189187
commit d3db4a6a01
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app/authentication/auth.py
View File

@@ -47,6 +47,7 @@ def restrict_ip_sms():
ip = '' ip = ''
if request.headers.get("X-Forwarded-For"): if request.headers.get("X-Forwarded-For"):
# X-Forwarded-For looks like "203.0.113.195, 70.41.3.18, 150.172.238.178" # X-Forwarded-For looks like "203.0.113.195, 70.41.3.18, 150.172.238.178"
# Counting backwards and look at the IP at the 3rd last hop - hence, hop(end-3)
ip_route = request.headers.get("X-Forwarded-For") ip_route = request.headers.get("X-Forwarded-For")
ip_list = ip_route.split(',') ip_list = ip_route.split(',')
if len(ip_list) >= 3: if len(ip_list) >= 3: