x59/iip-go
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

Fix ip response

Browse source
This commit is contained in:
Yehuda Deutsch 2019-08-12 19:34:52 +03:00
parent 48d77068ae
commit ca8d7c6edd
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
app.py
View file

@ -11,8 +11,13 @@ class GetClientIpMixin:
for header_name in ('x-real-ip', 'x-forwarded-for'): for header_name in ('x-real-ip', 'x-forwarded-for'):
candidate_ip = None candidate_ip = None
header_value = request.headers.get(header_name) header_value = request.headers.get(header_name)
if not header_value:
continue
for ip in header_value.split(',')[::-1]: for ip in header_value.split(',')[::-1]:
candidate_ip = ip_address(ip.strip()) try:
candidate_ip = ip_address(ip.strip())
except ValueError:
continue
if candidate_ip.is_global: if candidate_ip.is_global:
break break
if candidate_ip: if candidate_ip:
@ -48,8 +53,8 @@ if __name__ == '__main__':
import os import os
from argparse import ArgumentParser from argparse import ArgumentParser
default_host = os.environ.get('APP_HOST', 'localhost') default_host = os.environ.get('APP_HOST', '0.0.0.0')
default_port = int(os.environ.get('APP_PORT', 8081)) default_port = int(os.environ.get('APP_PORT', 8080))
default_trusted = os.environ.get('TRUSTED_PROXIES') or None default_trusted = os.environ.get('TRUSTED_PROXIES') or None
parser = ArgumentParser('app.py') parser = ArgumentParser('app.py')
@ -58,5 +63,5 @@ if __name__ == '__main__':
parser.add_argument('--proxies', type=str, help='A list of trusted proxies, comma separated. you can use CIDRs.', default=default_trusted) parser.add_argument('--proxies', type=str, help='A list of trusted proxies, comma separated. you can use CIDRs.', default=default_trusted)
args = parser.parse_args() args = parser.parse_args()
txip = Application(args) zero_if = Application(args)
txip.run() zero_if.run()