rra
/
Border-Check
mirror of https://github.com/rscmbbng/Border-Check
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

fixed some identations and added browser automatic launcher

pull/3/head
psy 11 years ago
parent
7b83722aaa
commit
f1548d52eb
2 changed files with 113 additions and 132 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 66
      main.py
  2. 179
      webserver.py

66
main.py
View File

@ -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
View File

@ -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).'

Write Preview
Loading…
Cancel
Save