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
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 64
      main.py
  2. 13
      webserver.py

64
main.py
View File

@ -26,12 +26,12 @@ DEBUG = 1
class bc(object): class bc(object):
""" """
BC main Class BC main Class
""" """
def __init__(self): def __init__(self):
""" """
Init defaults Init defaults
""" """
self.browser = "" # "F" Firefox / "C" Chrome self.browser = "" # "F" Firefox / "C" Chrome
self.browser_path = "" self.browser_path = ""
self.url = "" self.url = ""
@ -39,14 +39,14 @@ Init defaults
def set_options(self, options): def set_options(self, options):
""" """
Set program options Set program options
""" """
self.options = options self.options = options
def create_options(self, args=None): def create_options(self, args=None):
""" """
Create options for OptionParser Create options for OptionParser
""" """
self.optionParser = BCOptions() self.optionParser = BCOptions()
self.options = self.optionParser.get_options(args) self.options = self.optionParser.get_options(args)
if not self.options: if not self.options:
@ -55,8 +55,8 @@ Create options for OptionParser
def try_running(self, func, error, args=None): 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 options = self.options
args = args or [] args = args or []
try: 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): def check_browser(self):
""" """
Check for browser used by system Check browsers used by system
""" """
if sys.platform == 'darwin': if sys.platform == 'darwin':
f_osx = os.path.join(os.path.expanduser('~'), 'Library/Application Support/Firefox/Profiles') 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') 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: try:
if os.path.exists(f_osx): if os.path.exists(f_osx):
if len(os.listdir(f_osx)) > 2: 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 #filtering the directory that was last modified
all_subdirs = [os.path.join(f_osx,d)for d in os.listdir(f_osx)] all_subdirs = [os.path.join(f_osx,d)for d in os.listdir(f_osx)]
try: try:
@ -89,37 +89,30 @@ Check for browser used by system
except: except:
pass pass
latest_subdir = max(all_subdirs, key=os.path.getmtime) latest_subdir = max(all_subdirs, key=os.path.getmtime)
osx_profile = os.path.join(f_osx, latest_subdir) osx_profile = os.path.join(f_osx, latest_subdir)
osx_history_path = os.path.join(osx_profile, 'places.sqlite') osx_history_path = os.path.join(osx_profile, 'places.sqlite')
self.browser_path = osx_history_path self.browser_path = osx_history_path
else: else:
for folder in os.listdir(f_osx): for folder in os.listdir(f_osx):
if folder.endswith('.default'): if folder.endswith('.default'):
osx_default = os.path.join(f_osx, folder) osx_default = os.path.join(f_osx, folder)
osx_history_path = os.path.join(osx_default, 'places.sqlite') 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_path = osx_history_path
self.browser = "F" self.browser = "F"
elif os.path.exists(c_osx): elif os.path.exists(c_osx):
self.browser = "C" self.browser = "C"
self.browser_path = c_osx self.browser_path = c_osx
elif os.path.exists(chromium_osx): elif os.path.exists(chromium_osx):
self.browser = "CHROMIUM" self.browser = "CHROMIUM"
self.browser_path = chromium_osx self.browser_path = chromium_osx
except: except:
print "no firefox or chrome installed" print "Warning: No firefox or chrome installed."
elif sys.platform.startswith('linux'): elif sys.platform.startswith('linux'):
f_lin = os.path.join(os.path.expanduser('~'), '.mozilla/firefox/') #add the next folder 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') c_lin = os.path.join(os.path.expanduser('~'), '.config/google-chrome/History')
chromium_lin = os.path.join(os.path.expanduser('~'), '.config/chromium/Default/History') chromium_lin = os.path.join(os.path.expanduser('~'), '.config/chromium/Default/History')
if os.path.exists(f_lin): if os.path.exists(f_lin):
#missing multiple profile support #missing multiple profile support
for folder in os.listdir(f_lin): 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') lin_history_path = os.path.join(lin_default, 'places.sqlite')
self.browser = "F" self.browser = "F"
self.browser_path = lin_history_path self.browser_path = lin_history_path
elif os.path.exists(c_lin): elif os.path.exists(c_lin):
self.browser = "C" self.browser = "C"
self.browser_path = c_lin self.browser_path = c_lin
elif os.path.exists(chromium_lin): elif os.path.exists(chromium_lin):
self.browser = "CHROMIUM" self.browser = "CHROMIUM"
self.browser_path = chromium_lin self.browser_path = chromium_lin
def getURL(self): def getURL(self):
""" """
Set urls to visit Set urls to visit
""" """
print "Browser database:", self.browser_path, "\n" print "Browser database:", self.browser_path, "\n"
conn = sqlite3.connect(self.browser_path) conn = sqlite3.connect(self.browser_path)
c = conn.cursor() c = conn.cursor()
@ -149,10 +139,6 @@ Set urls to visit
if self.browser == "F": #Firefox history database if self.browser == "F": #Firefox history database
c.execute('select url, last_visit_date from moz_places ORDER BY last_visit_date DESC') c.execute('select url, last_visit_date from moz_places ORDER BY last_visit_date DESC')
elif self.browser == "C": #Chrome history database 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') 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 else: # Browser not allowed
print "\nSorry, you haven't a compatible browser\n\n" print "\nSorry, you haven't a compatible browser\n\n"
@ -164,8 +150,8 @@ Set urls to visit
def getGEO(self): 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 # Download and extract database
try: try:
urllib.urlretrieve('http://xsser.sf.net/map/GeoLiteCity.dat.gz', 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): def run(self, opts=None):
""" """
Run BorderCheck Run BorderCheck
""" """
#eprint = sys.stderr.write
# set options # set options
if opts: if opts:
options = self.create_options(opts) options = self.create_options(opts)
@ -203,17 +188,16 @@ Run BorderCheck
print "url:", self.url print "url:", self.url
# start web mode # start web mode
print("Running webserver\n") print("Running webserver\n")
BorderCheckWebserver(self) BorderCheckWebserver(self) #child process or another thread
while True: while True:
url = urlparse(self.url[0]).netloc 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) url_ip = socket.gethostbyname(url)
print url_ip print url_ip
if url != self.old_url: if url != self.old_url:
count = 0 count = 0
print url print url
a = subprocess.Popen(['lft', '-S', '-n', '-E', url_ip], stdout=subprocess.PIPE) # -> using tcp 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 #a = subprocess.Popen(['lft', '-S', '-n', '-u', url_ip], stdout=subprocess.PIPE) # -> using udp
logfile = open('logfile', 'a') logfile = open('logfile', 'a')
@ -240,7 +224,7 @@ Run BorderCheck
exit() exit()
self.old_url = url self.old_url = url
print"old url =", self.old_url print "old url = ", self.old_url
logfile.close() logfile.close()
time.sleep(5) time.sleep(5)

13
webserver.py
View File

@ -12,6 +12,7 @@ from runpy import run_module
from urlparse import urlparse from urlparse import urlparse
from cgi import parse_qs #, parse_header, parse_multipart from cgi import parse_qs #, parse_header, parse_multipart
import cgi import cgi
import webbrowser
port = 8080 port = 8080
wwwroot = "web/" wwwroot = "web/"
@ -35,8 +36,6 @@ def print_exception(type=None, value=None, tb=None, limit=None):
return ret return ret
class HttpHandler(BaseHTTPRequestHandler): class HttpHandler(BaseHTTPRequestHandler):
# TODO: whitelist out there
def client_not_allowed(self, addr): def client_not_allowed(self, addr):
return False return False
if addr == "127.0.0.1": if addr == "127.0.0.1":
@ -50,7 +49,6 @@ class HttpHandler(BaseHTTPRequestHandler):
tmp = uri.find ('?') tmp = uri.find ('?')
args = parse_qs(urlparse(uri)[4]) args = parse_qs(urlparse(uri)[4])
#from ipdb import set_trace;set_trace()
if tmp != -1: if tmp != -1:
uri = uri[0:tmp] uri = uri[0:tmp]
for a in uri[tmp:-1].split("&"): for a in uri[tmp:-1].split("&"):
@ -72,14 +70,14 @@ class HttpHandler(BaseHTTPRequestHandler):
content = query.get('upfile') content = query.get('upfile')
except: except:
pass pass
print "Request from %s:%d"%self.client_address+" "+uri print "Request from %s:%d"%self.client_address + " " + uri
if uri[-1] == '/' or os.path.isdir(file): if uri[-1] == '/' or os.path.isdir(file):
file = file + "/index.py" file = file + "/index.py"
if os.path.isfile(file+".py"): if os.path.isfile(file + ".py"):
file = file + ".py" file = file + ".py"
if file.find("py") != -1: if file.find("py") != -1:
modname = file.replace(".py", "") modname = file.replace(".py", "")
cwd = modname[0:modname.rfind('/')]+"/" cwd = modname[0:modname.rfind('/')] + "/"
modname = modname.replace("/", ".") modname = modname.replace("/", ".")
while modname.find("..") != -1: while modname.find("..") != -1:
modname = modname.replace("..",".") modname = modname.replace("..",".")
@ -89,7 +87,6 @@ class HttpHandler(BaseHTTPRequestHandler):
"uri": uri, "uri": uri,
"args": args, "args": args,
"cwd": cwd, "cwd": cwd,
"headers": self.headers,
"content": content "content": content
} }
try: try:
@ -121,6 +118,7 @@ class BorderCheckWebserver():
HttpHandler.ref = ref HttpHandler.ref = ref
httpd = HTTPServer(('', port), HttpHandler) httpd = HTTPServer(('', port), HttpHandler)
print "http://127.0.0.1:%d/ : Serving directory '%s/www'" % (port, os.getcwd()) 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: try:
httpd.serve_forever() httpd.serve_forever()
except KeyboardInterrupt: except KeyboardInterrupt:
@ -130,7 +128,6 @@ if __name__=="__main__":
wwwroot = "www" wwwroot = "www"
httpd = HTTPServer(('', port), HttpHandler) httpd = HTTPServer(('', port), HttpHandler)
print "http://127.0.0.1:%d/ : Serving directory '%s/www'" % (port, os.getcwd()) print "http://127.0.0.1:%d/ : Serving directory '%s/www'" % (port, os.getcwd())
try: try:
httpd.serve_forever() httpd.serve_forever()
except KeyboardInterrupt: except KeyboardInterrupt:

Write Preview
Loading…
Cancel
Save