mirror of
https://github.com/rscmbbng/Border-Check.git
synced 2024-12-25 21:41:28 +01:00
fixed some identations and added browser automatic launcher
This commit is contained in:
psy
parent
7b83722aaa
commit
f1548d52eb
66
main.py
66
main.py
@ -26,12 +26,12 @@ DEBUG = 1
|
||||
|
||||
class bc(object):
|
||||
"""
|
||||
BC main Class
|
||||
"""
|
||||
BC main Class
|
||||
"""
|
||||
def __init__(self):
|
||||
"""
|
||||
Init defaults
|
||||
"""
|
||||
Init defaults
|
||||
"""
|
||||
self.browser = "" # "F" Firefox / "C" Chrome
|
||||
self.browser_path = ""
|
||||
self.url = ""
|
||||
@ -39,14 +39,14 @@ Init defaults
|
||||
|
||||
def set_options(self, options):
|
||||
"""
|
||||
Set program options
|
||||
"""
|
||||
Set program options
|
||||
"""
|
||||
self.options = options
|
||||
|
||||
def create_options(self, args=None):
|
||||
"""
|
||||
Create options for OptionParser
|
||||
"""
|
||||
Create options for OptionParser
|
||||
"""
|
||||
self.optionParser = BCOptions()
|
||||
self.options = self.optionParser.get_options(args)
|
||||
if not self.options:
|
||||
@ -55,8 +55,8 @@ Create options for OptionParser
|
||||
|
||||
def try_running(self, func, error, args=None):
|
||||
"""
|
||||
Try running a function and print some error if it fails and exists with a fatal error.
|
||||
"""
|
||||
Try running a function and print some error if it fails and exists with a fatal error.
|
||||
"""
|
||||
options = self.options
|
||||
args = args or []
|
||||
try:
|
||||
@ -72,8 +72,8 @@ Try running a function and print some error if it fails and exists with a fatal
|
||||
|
||||
def check_browser(self):
|
||||
"""
|
||||
Check for browser used by system
|
||||
"""
|
||||
Check browsers used by system
|
||||
"""
|
||||
if sys.platform == 'darwin':
|
||||
f_osx = os.path.join(os.path.expanduser('~'), 'Library/Application Support/Firefox/Profiles')
|
||||
c_osx = os.path.join(os.path.expanduser('~'), 'Library/Application Support/Google/Chrome/Default/History')
|
||||
@ -81,7 +81,7 @@ Check for browser used by system
|
||||
try:
|
||||
if os.path.exists(f_osx):
|
||||
if len(os.listdir(f_osx)) > 2:
|
||||
print 'you have multiple profiles, choosing the last one used'
|
||||
print 'You have multiple profiles, choosing the last one used'
|
||||
#filtering the directory that was last modified
|
||||
all_subdirs = [os.path.join(f_osx,d)for d in os.listdir(f_osx)]
|
||||
try:
|
||||
@ -89,37 +89,30 @@ Check for browser used by system
|
||||
except:
|
||||
pass
|
||||
latest_subdir = max(all_subdirs, key=os.path.getmtime)
|
||||
|
||||
osx_profile = os.path.join(f_osx, latest_subdir)
|
||||
osx_history_path = os.path.join(osx_profile, 'places.sqlite')
|
||||
self.browser_path = osx_history_path
|
||||
|
||||
else:
|
||||
for folder in os.listdir(f_osx):
|
||||
if folder.endswith('.default'):
|
||||
osx_default = os.path.join(f_osx, folder)
|
||||
osx_history_path = os.path.join(osx_default, 'places.sqlite')
|
||||
print "setting:", osx_history_path, "as history file"
|
||||
print "Setting:", osx_history_path, "as history file"
|
||||
self.browser_path = osx_history_path
|
||||
|
||||
self.browser = "F"
|
||||
|
||||
elif os.path.exists(c_osx):
|
||||
self.browser = "C"
|
||||
self.browser_path = c_osx
|
||||
|
||||
elif os.path.exists(chromium_osx):
|
||||
self.browser = "CHROMIUM"
|
||||
self.browser_path = chromium_osx
|
||||
|
||||
except:
|
||||
print "no firefox or chrome installed"
|
||||
print "Warning: No firefox or chrome installed."
|
||||
|
||||
elif sys.platform.startswith('linux'):
|
||||
f_lin = os.path.join(os.path.expanduser('~'), '.mozilla/firefox/') #add the next folder
|
||||
c_lin = os.path.join(os.path.expanduser('~'), '.config/google-chrome/History')
|
||||
chromium_lin = os.path.join(os.path.expanduser('~'), '.config/chromium/Default/History')
|
||||
|
||||
if os.path.exists(f_lin):
|
||||
#missing multiple profile support
|
||||
for folder in os.listdir(f_lin):
|
||||
@ -128,20 +121,17 @@ Check for browser used by system
|
||||
lin_history_path = os.path.join(lin_default, 'places.sqlite')
|
||||
self.browser = "F"
|
||||
self.browser_path = lin_history_path
|
||||
|
||||
elif os.path.exists(c_lin):
|
||||
self.browser = "C"
|
||||
self.browser_path = c_lin
|
||||
|
||||
elif os.path.exists(chromium_lin):
|
||||
self.browser = "CHROMIUM"
|
||||
self.browser_path = chromium_lin
|
||||
|
||||
|
||||
def getURL(self):
|
||||
"""
|
||||
Set urls to visit
|
||||
"""
|
||||
Set urls to visit
|
||||
"""
|
||||
print "Browser database:", self.browser_path, "\n"
|
||||
conn = sqlite3.connect(self.browser_path)
|
||||
c = conn.cursor()
|
||||
@ -149,10 +139,6 @@ Set urls to visit
|
||||
if self.browser == "F": #Firefox history database
|
||||
c.execute('select url, last_visit_date from moz_places ORDER BY last_visit_date DESC')
|
||||
elif self.browser == "C": #Chrome history database
|
||||
# Linux: /home/$USER/.config/google-chrome/
|
||||
# Linux: /home/$USER/.config/chromium/
|
||||
# Windows Vista (and Win 7): C:\Users\[USERNAME]\AppData\Local\Google\Chrome\
|
||||
# Windows XP: C:\Documents and Settings\[USERNAME]\Local Settings\Application Data\Google\Chrome\
|
||||
c.execute('select urls.url, urls.title, urls.visit_count, urls.typed_count, urls.last_visit_time, urls.hidden, visits.visit_time, visits.from_visit, visits.transition from urls, visits where urls.id = visits.url')
|
||||
else: # Browser not allowed
|
||||
print "\nSorry, you haven't a compatible browser\n\n"
|
||||
@ -164,8 +150,8 @@ Set urls to visit
|
||||
|
||||
def getGEO(self):
|
||||
"""
|
||||
Get Geolocation database (http://dev.maxmind.com/geoip/legacy/geolite/)
|
||||
"""
|
||||
Get Geolocation database (http://dev.maxmind.com/geoip/legacy/geolite/)
|
||||
"""
|
||||
# Download and extract database
|
||||
try:
|
||||
urllib.urlretrieve('http://xsser.sf.net/map/GeoLiteCity.dat.gz',
|
||||
@ -183,9 +169,8 @@ Get Geolocation database (http://dev.maxmind.com/geoip/legacy/geolite/)
|
||||
|
||||
def run(self, opts=None):
|
||||
"""
|
||||
Run BorderCheck
|
||||
"""
|
||||
#eprint = sys.stderr.write
|
||||
Run BorderCheck
|
||||
"""
|
||||
# set options
|
||||
if opts:
|
||||
options = self.create_options(opts)
|
||||
@ -203,17 +188,16 @@ Run BorderCheck
|
||||
print "url:", self.url
|
||||
# start web mode
|
||||
print("Running webserver\n")
|
||||
BorderCheckWebserver(self)
|
||||
BorderCheckWebserver(self) #child process or another thread
|
||||
|
||||
while True:
|
||||
url = urlparse(self.url[0]).netloc
|
||||
url = url.replace('www.','') #--> doing a tracert to for example.com and www.example.com yields different results most of the times.
|
||||
url = url.replace('www.','') #--> doing a tracert to example.com and www.example.com yields different results.
|
||||
url_ip = socket.gethostbyname(url)
|
||||
print url_ip
|
||||
if url != self.old_url:
|
||||
count = 0
|
||||
print url
|
||||
|
||||
a = subprocess.Popen(['lft', '-S', '-n', '-E', url_ip], stdout=subprocess.PIPE) # -> using tcp
|
||||
#a = subprocess.Popen(['lft', '-S', '-n', '-u', url_ip], stdout=subprocess.PIPE) # -> using udp
|
||||
logfile = open('logfile', 'a')
|
||||
@ -240,7 +224,7 @@ Run BorderCheck
|
||||
exit()
|
||||
|
||||
self.old_url = url
|
||||
print"old url =", self.old_url
|
||||
print "old url = ", self.old_url
|
||||
logfile.close()
|
||||
time.sleep(5)
|
||||
|
||||
@ -249,4 +233,4 @@ if __name__ == "__main__":
|
||||
options = app.create_options()
|
||||
if options:
|
||||
app.set_options(options)
|
||||
app.run()
|
||||
app.run()
|
||||
|
||||
179
webserver.py
179
webserver.py
@ -12,6 +12,7 @@ from runpy import run_module
|
||||
from urlparse import urlparse
|
||||
from cgi import parse_qs #, parse_header, parse_multipart
|
||||
import cgi
|
||||
import webbrowser
|
||||
|
||||
port = 8080
|
||||
wwwroot = "web/"
|
||||
@ -21,106 +22,103 @@ class ForkingTCPServer(ForkingMixIn, HTTPServer): pass
|
||||
class ThreadingTCPServer(ThreadingMixIn, HTTPServer): pass
|
||||
|
||||
def print_exception(type=None, value=None, tb=None, limit=None):
|
||||
if type is None:
|
||||
type, value, tb = sys.exc_info()
|
||||
import traceback
|
||||
ret = "<html><body><h2>Traceback (most recent call last):<h2 />"
|
||||
ret += "<pre>"
|
||||
list = traceback.format_tb(tb, limit) + \
|
||||
traceback.format_exception_only(type, value)
|
||||
ret += "exception error"
|
||||
ret += "%s: %s<br/>\n" % ( ("\n".join(list[:-1])), (list[-1]))
|
||||
ret +="</body></html>"
|
||||
del tb
|
||||
return ret
|
||||
if type is None:
|
||||
type, value, tb = sys.exc_info()
|
||||
import traceback
|
||||
ret = "<html><body><h2>Traceback (most recent call last):<h2 />"
|
||||
ret += "<pre>"
|
||||
list = traceback.format_tb(tb, limit) + \
|
||||
traceback.format_exception_only(type, value)
|
||||
ret += "exception error"
|
||||
ret += "%s: %s<br/>\n" % ( ("\n".join(list[:-1])), (list[-1]))
|
||||
ret +="</body></html>"
|
||||
del tb
|
||||
return ret
|
||||
|
||||
class HttpHandler(BaseHTTPRequestHandler):
|
||||
|
||||
# TODO: whitelist out there
|
||||
def client_not_allowed(self, addr):
|
||||
return False
|
||||
if addr == "127.0.0.1":
|
||||
return False
|
||||
print ("Client not allowed ",addr)
|
||||
return True
|
||||
def client_not_allowed(self, addr):
|
||||
return False
|
||||
if addr == "127.0.0.1":
|
||||
return False
|
||||
print ("Client not allowed ",addr)
|
||||
return True
|
||||
|
||||
def serve(self):
|
||||
output = ""
|
||||
uri = self.path
|
||||
tmp = uri.find ('?')
|
||||
args = parse_qs(urlparse(uri)[4])
|
||||
def serve(self):
|
||||
output = ""
|
||||
uri = self.path
|
||||
tmp = uri.find ('?')
|
||||
args = parse_qs(urlparse(uri)[4])
|
||||
|
||||
#from ipdb import set_trace;set_trace()
|
||||
if tmp != -1:
|
||||
uri = uri[0:tmp]
|
||||
for a in uri[tmp:-1].split("&"):
|
||||
sep = a.find ("=")
|
||||
if sep != -1:
|
||||
print "%s)(%s"%(a[0:sep],a[sep:-1])
|
||||
args[a[0:sep]]=a[sep:-1]
|
||||
if tmp != -1:
|
||||
uri = uri[0:tmp]
|
||||
for a in uri[tmp:-1].split("&"):
|
||||
sep = a.find ("=")
|
||||
if sep != -1:
|
||||
print "%s)(%s"%(a[0:sep],a[sep:-1])
|
||||
args[a[0:sep]]=a[sep:-1]
|
||||
|
||||
file = wwwroot + "/" + uri
|
||||
if self.client_not_allowed (self.client_address[0]):
|
||||
self.wfile.write ("HTTP/1.0 503 Not allowed\r\n\r\nYou are not whitelisted")
|
||||
return
|
||||
content = ""
|
||||
try:
|
||||
ctype,pdict = cgi.parse_header(self.headers.getheader('content-type'))
|
||||
print "CTYPE IS ",ctype
|
||||
if ctype == 'multipart/form-data':
|
||||
query = cgi.parse_multipart(self.rfile, pdict)
|
||||
content = query.get('upfile')
|
||||
except:
|
||||
pass
|
||||
print "Request from %s:%d"%self.client_address+" "+uri
|
||||
if uri[-1] == '/' or os.path.isdir(file):
|
||||
file = file + "/index.py"
|
||||
if os.path.isfile(file+".py"):
|
||||
file = file + ".py"
|
||||
if file.find("py") != -1:
|
||||
modname = file.replace(".py", "")
|
||||
cwd = modname[0:modname.rfind('/')]+"/"
|
||||
modname = modname.replace("/", ".")
|
||||
while modname.find("..") != -1:
|
||||
modname = modname.replace("..",".")
|
||||
globals = {
|
||||
"output": output,
|
||||
"http": http,
|
||||
"uri": uri,
|
||||
"args": args,
|
||||
"cwd": cwd,
|
||||
"headers": self.headers,
|
||||
"content": content
|
||||
}
|
||||
try:
|
||||
a = run_module(modname, init_globals=globals)
|
||||
output = a["output"]
|
||||
except:
|
||||
output = print_exception()
|
||||
else:
|
||||
try:
|
||||
f = open (file, "r")
|
||||
output = f.read ()
|
||||
f.close ()
|
||||
except:
|
||||
output = "404"
|
||||
if output == "404":
|
||||
self.wfile.write ("HTTP/1.0 404 Not found\r\n\r\n")
|
||||
else:
|
||||
self.wfile.write ("HTTP/1.0 200 OK\r\n\r\n")
|
||||
self.wfile.write (output)
|
||||
file = wwwroot + "/" + uri
|
||||
if self.client_not_allowed (self.client_address[0]):
|
||||
self.wfile.write ("HTTP/1.0 503 Not allowed\r\n\r\nYou are not whitelisted")
|
||||
return
|
||||
content = ""
|
||||
try:
|
||||
ctype,pdict = cgi.parse_header(self.headers.getheader('content-type'))
|
||||
print "CTYPE IS ",ctype
|
||||
if ctype == 'multipart/form-data':
|
||||
query = cgi.parse_multipart(self.rfile, pdict)
|
||||
content = query.get('upfile')
|
||||
except:
|
||||
pass
|
||||
print "Request from %s:%d"%self.client_address + " " + uri
|
||||
if uri[-1] == '/' or os.path.isdir(file):
|
||||
file = file + "/index.py"
|
||||
if os.path.isfile(file + ".py"):
|
||||
file = file + ".py"
|
||||
if file.find("py") != -1:
|
||||
modname = file.replace(".py", "")
|
||||
cwd = modname[0:modname.rfind('/')] + "/"
|
||||
modname = modname.replace("/", ".")
|
||||
while modname.find("..") != -1:
|
||||
modname = modname.replace("..",".")
|
||||
globals = {
|
||||
"output": output,
|
||||
"http": http,
|
||||
"uri": uri,
|
||||
"args": args,
|
||||
"cwd": cwd,
|
||||
"content": content
|
||||
}
|
||||
try:
|
||||
a = run_module(modname, init_globals=globals)
|
||||
output = a["output"]
|
||||
except:
|
||||
output = print_exception()
|
||||
else:
|
||||
try:
|
||||
f = open (file, "r")
|
||||
output = f.read ()
|
||||
f.close ()
|
||||
except:
|
||||
output = "404"
|
||||
if output == "404":
|
||||
self.wfile.write ("HTTP/1.0 404 Not found\r\n\r\n")
|
||||
else:
|
||||
self.wfile.write ("HTTP/1.0 200 OK\r\n\r\n")
|
||||
self.wfile.write (output)
|
||||
|
||||
def do_POST (self):
|
||||
self.serve ()
|
||||
def do_POST (self):
|
||||
self.serve ()
|
||||
|
||||
def do_GET (self):
|
||||
self.serve ()
|
||||
def do_GET (self):
|
||||
self.serve ()
|
||||
|
||||
class BorderCheckWebserver():
|
||||
def __init__(self, ref, *args):
|
||||
HttpHandler.ref = ref
|
||||
HttpHandler.ref = ref
|
||||
httpd = HTTPServer(('', port), HttpHandler)
|
||||
print "http://127.0.0.1:%d/ : Serving directory '%s/www'" % (port, os.getcwd())
|
||||
webbrowser.open('http://127.0.0.1:8080', new=1)
|
||||
try:
|
||||
httpd.serve_forever()
|
||||
except KeyboardInterrupt:
|
||||
@ -130,8 +128,7 @@ if __name__=="__main__":
|
||||
wwwroot = "www"
|
||||
httpd = HTTPServer(('', port), HttpHandler)
|
||||
print "http://127.0.0.1:%d/ : Serving directory '%s/www'" % (port, os.getcwd())
|
||||
|
||||
try:
|
||||
httpd.serve_forever()
|
||||
httpd.serve_forever()
|
||||
except KeyboardInterrupt:
|
||||
print 'Server killed on user request (keyboard interrupt).'
|
||||
print 'Server killed on user request (keyboard interrupt).'
|
||||
|
||||
Loading…
Reference in New Issue
Block a user