From f1548d52eb36191aa545e04ab74d51e8a26c0b4e Mon Sep 17 00:00:00 2001 From: psy Date: Thu, 5 Sep 2013 08:45:09 -0700 Subject: [PATCH] fixed some identations and added browser automatic launcher --- main.py | 66 +++++++------------ webserver.py | 179 +++++++++++++++++++++++++-------------------------- 2 files changed, 113 insertions(+), 132 deletions(-) diff --git a/main.py b/main.py index fb96f65..47850e6 100644 --- a/main.py +++ b/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() \ No newline at end of file + app.run() diff --git a/webserver.py b/webserver.py index 301a29d..9b47fad 100644 --- a/webserver.py +++ b/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 = "

Traceback (most recent call last):

" - ret += "
"
-	list = traceback.format_tb(tb, limit) + \
-		traceback.format_exception_only(type, value)
-	ret += "exception error"
-	ret += "%s: %s
\n" % ( ("\n".join(list[:-1])), (list[-1]))
-	ret +=""
-	del tb
-	return ret
+    if type is None:
+        type, value, tb = sys.exc_info()
+        import traceback
+        ret = "
Traceback (most recent call last):
"
+        ret += "
"
+        list = traceback.format_tb(tb, limit) + \
+            traceback.format_exception_only(type, value)
+        ret += "exception error"
+        ret += "%s: %s
\n" % ( ("\n".join(list[:-1])), (list[-1]))
+        ret +=""
+        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).'