PeterSurda/cloud-init-cherrypy
1
0
Fork 0
forked from Sysdeploy/cloud-init-cherrypy
Code Issues Pull Requests Projects Releases Wiki Activity

Add forward DNS lookup check

Browse Source
This commit is contained in:
Peter Šurda 2021-07-31 20:45:50 +08:00
parent 0ee0ac4936
commit 12ee0ab3ec
Signed by: PeterSurda
GPG Key ID: 0C5F50C0B5F37D87
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@ -10,7 +10,7 @@ import sys
from ipaddress import AddressValueError, IPv4Address, IPv6Address from ipaddress import AddressValueError, IPv4Address, IPv6Address
import cherrypy import cherrypy
from cherrypy.lib.static import serve_file # from cherrypy.lib.static import serve_file
from jinja2 import Template from jinja2 import Template
import yaml import yaml
@ -59,8 +59,13 @@ class CloudInitApp:
try: try:
self.hostinfo = socket.gethostbyaddr(self.remoteip) self.hostinfo = socket.gethostbyaddr(self.remoteip)
forward_lookup = socket.gethostbyname(self.hostinfo[0])
if forward_lookup != self.remoteip:
self.hostinfo = ('localhost', )
except socket.herror: except socket.herror:
self.hostinfo = ('localhost', ) self.hostinfo = ('localhost', )
except socket.gaierror:
self.hostinfo = (self.remoteip, )
def _redirect_if_needed(self): def _redirect_if_needed(self):
filepath = os.path.join(PATH, "data", self.hostinfo[0], filepath = os.path.join(PATH, "data", self.hostinfo[0],