mac compatibility

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.pythonPath": "/usr/bin/python3"
+}

auth.py

@@ -8,7 +8,7 @@ import bcrypt
 def login(config, user, password, sysid):
     fprint("Attempting to login as " + user)
     filename = sysid + "login.csv"
-    #return True
+    
     #hashpasswd = bcrypt.hashpw(password.encode('utf-8'), user).decode()
     with open(find_data_file(filename), "w", newline="") as f:
         writer = csv.writer(f)

block.py

@@ -10,10 +10,8 @@ import time
 import csv
 import ssh
 
-def get_blocklist(config, settings):
+def get_blocklist(config):
     setup_child()
-    appendbad = settings["appendbad"]
-    whitelist = settings["whitelist"]
     fprint("Downloading deny list from server")
     data = ssh.check_for_file(config, "BadIPs.csv", "receive")
     #fprint(data.stdout)
@@ -23,29 +21,14 @@ def get_blocklist(config, settings):
         data2.append(row)
     data2 = [i for i in data2 if i]
     #fprint(data2)
-    for line in appendbad:
-        data2.append(line)
+    data2.append(["N/A", "TCP", "N/A", "N/A", "20.112.52.29", "5000", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
+    data2.append(["N/A", "TCP", "N/A", "N/A", "20.81.111.85", "80", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
+    data2.append(["N/A", "TCP", "N/A", "N/A", "100.115.71.78", "5000", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
+    data2.append(["N/A", "TCP", "N/A", "N/A", "100.115.71.78", "5000", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
+    data2.append(["N/A", "TCP", "N/A", "N/A", "174.143.130.167", "443", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
+    data2.append(["N/A", "TCP", "N/A", "N/A", "216.47.134.203", "443", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
+    data2.append(["N/A", "TCP", "N/A", "N/A", "34.111.83.189", "443", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
     
-    for line in whitelist:
-        for line2 in data2:
-            if line[0] in line2 and line[1] in line2:
-                fprint("Whitelisting P1 " + line[0] + ":" + str(line[1]))
-                data2.remove(line2)
-                
-                if line[2] in settings["badapps"]:
-                    tmp = settings["badapps"]
-                    tmp.remove(line[2])
-                    settings["badapps"] = tmp
-                if line[0] in settings["badips"]:
-                    tmp = settings["badips"]
-                    tmp.remove(line[0])
-                    settings["badips"] = tmp
-                for line3 in settings["badlines"]:
-                    if line[0] in line3 and line[1] in line3:
-                        fprint("Whitelisting P2" + line[0] + ":" + str(line[1]))
-                        tmp = settings["badlines"]
-                        tmp.remove(line3)
-                        settings["badlines"] = tmp
     
     fprint(data2)
     with open(find_data_file("blocklist.csv"), "w", newline="") as f:
@@ -69,7 +52,7 @@ def block_conn(config, datafile, res):
     
     baddata = res
     fprint("Local loaded successfully")
-    goodct = 0
+    
     #fprint(mydata)
     for line in mydata:
         #fprint(line)
@@ -78,25 +61,19 @@ def block_conn(config, datafile, res):
             continue
 
         srcip = line[2].split(":")[0]
-        srcport = line[2].split(":")[1]
         destip = line[3].split(":")[0]
-        destport = line[3].split(":")[1]
         pid = line[5]
         try: 
             pid = int(pid)
         except ValueError:
             continue
-        found = False
+
         for line in baddata:
             #fprint(destip + " " + line[4])
             badsrcip = line[2]
-            badsrcport = line[3]
             baddestip = line[4]
-            baddestport = line[5]
             badpid = line[11]
-            
-            if ((srcip == badsrcip and srcport == badsrcport) or (destip == baddestip and destport == baddestport)) and not pid in badapps:
-                found = True
+            if srcip == badsrcip or destip == baddestip and not pid in badapps:
                 fprint("FLAG " + srcip + " " + destip + " " + str(pid))
                 badapps.append(pid)
                 badips.append(baddestip)
@@ -104,8 +81,4 @@ def block_conn(config, datafile, res):
                 #fprint(badapps)
                 #fprint("FLAG " + srcip + " " + destip + " " + str(pid))
                 #kill(pid)
-            
-        if not found:
-            goodct = goodct + 1
-    
-    return badapps, badips, badlines, goodct
+    return badapps, badips, badlines

config.yml

@@ -1,8 +1,8 @@
 core:
-  autostart: true
+  autostart: false
   clockspeed: 20
   interval: 10
-  level: 3
+  level: 2
   localadmin: true
 sftp:
   filepath:

ippigeon.py

@@ -10,7 +10,6 @@ from util import find_data_file
 from util import fprint
 from util import kill
 from util import run_cmd
-from notification import send_notification
 import taskbartool
 import util
 import netstat
@@ -19,8 +18,8 @@ import auth
 import panel
 import block
 
-
-history = list()
+badapps = [756, 278670]
+badips = ["208.59.79.12",]
 displaydata = None
 settings = None
 netdata_res = None
@@ -60,87 +59,69 @@ if linux or macos:
     #fprint(hostname)
 
 def netstat_done(res):
-    fprint("netstat done, processing", settings)
-    procdata_res = pool.apply_async(netstat.process, (res,), callback=process_done)
-    #netstat.process(res)
+    fprint("netstat done, processing")
+    #procdata_res = pool.apply_async(netstat.process, (res,), callback=process_done)
+    netstat.process(res)
 
 def process_done(res):
     if settings["running"] == True:
-        fprint("uploading to sftp...", settings)
-        #ssh.sftp_send_data(config, datafile, 'send')
-        
+        fprint("uploading to sftp...")
+        #ssh.sftp_send_data(res, config, datafile)
         procdata_res = pool.apply_async(ssh.sftp_send_data, (config, datafile, 'send'), callback=upload_done)
     
 def upload_done(res):
     settings["block"] = True
-    tmpstat = settings["stats"]
-    tmpstat[2] += 1
-    settings["stats"] = tmpstat
 
 def login_done(res):
     if not res:
-        fprint("Login failure", settings)
+        fprint("Login failure")
         settings["message"] = "Login failure"
     else:
-        fprint("Login result in main: " + str(res), settings)
+        fprint("Login result in main: " + str(res))
         settings["loggedin"] = res
         settings["continueui"] = True
 
 def blockdata_done(res):
     global settings
-    fprint("FINISHED downloading block data", settings)
+    fprint("FINISHED downloading block data")
     tmpkill = settings["kill"]
     settings["kill"] = False
     #block_res = pool.apply_async(block.block_conn, (config, datafile, res, settings))
-    block_pids, block_ips, block_data, goodct = block.block_conn(config, datafile, res)
-    tmpstat = settings["stats"]
-    tmpstat[1] += goodct
-    if tmpstat[0] > 0 and goodct > 0:
-        tmpstat[4] = 1.0 / (goodct * 100.0 / tmpstat[0])
-    else: 
-        tmpstat[4] = 0.0
-    tmpstat[3] += 1
-    settings["stats"] = tmpstat
-
+    block_pids, block_ips, block_data = block.block_conn(config, datafile, res)
     tmplist = settings["badapps"]
     
     for x in block_pids:
-        fprint(x, settings)
+        fprint(x)
         if not x in tmplist:
             tmplist.append(x)
     settings["badapps"] = tmplist
-    fprint(settings["badapps"], settings)
+    fprint(settings["badapps"])
     
     tmplist = settings["badips"]
     for x in block_ips:
-        fprint(x, settings)
+        fprint(x)
         if not x in tmplist:
             tmplist.append(x)
     settings["badips"] = tmplist
-    fprint(settings["badips"], settings)
+    fprint(settings["badips"])
 
     settings["kill"] = tmpkill
 
     tmplist = settings["badlines"]
     for x in block_data:
-        fprint(x, settings)
+        fprint(x)
         if not x in tmplist:
             tmplist.append(x)
     settings["badlines"] = tmplist
-    fprint(settings["badlines"], settings)
+    fprint(settings["badlines"])
 
     settings["newdata"] = True
         
-def readstat_done(res):
-    settings["stats"] = res
-    fprint("Read stats!" + str(settings["stats"]), settings)
-
-
 def killall():
     kids = active_children()
     for kid in kids:
         kid.kill()
-    fprint("Every child has been killed", settings)
+    fprint("Every child has been killed")
     os.kill(os.getpid(), 9) # dirty kill of self
 
 
@@ -157,17 +138,14 @@ def mainloop(pool):
     #print(killme)
     if killme.value > 0:
         #print("killing")
-        util.clear_fwll() # clear the firewall rules before shutdown
         killall()
     #print(res.get(timeout=1))
     if counter == 0: # runs every INTERVAL
         #fprint("start loop")
-        if settings["stats"][1] > 0:
-            util.write_stats(settings["stats"])
         if netdata_res is None or netdata_res.ready():
             #rawdata = netdata_res.get()
             #procdata_res = pool.apply_async(process_netstat, (rawdata))
-            fprint("netstat starting", settings)
+            fprint("netstat starting")
             netdata_res = pool.apply_async(netstat.start, callback=netstat_done)
             #fprint(netdata_res.successful())
     
@@ -175,11 +153,11 @@ def mainloop(pool):
     
     if settings["continueui"] == True:
         settings["continueui"] = False
-        #if ppanel is not None:
+        if ppanel is not None:
             # login panel is already open
-        #    ppanel.terminate()
-        #    ppanel = Process(target=panel.openwindow, args=(displaydata,settings,killme))
-        #    ppanel.start()  
+            ppanel.terminate()
+            ppanel = Process(target=panel.openwindow, args=(displaydata,settings,killme))
+            ppanel.start()  
 
     if settings["showui"] == True:
         settings["showui"] = False
@@ -193,9 +171,8 @@ def mainloop(pool):
         
     
     if settings["block"] == True and settings["running"] == True:
-        #blockdata_res = pool.apply_async(block.get_blocklist, (config, settings), callback=blockdata_done)
-        res = block.get_blocklist(config, settings)
-        blockdata_done(res)
+        blockdata_res = pool.apply_async(block.get_blocklist, (config,), callback=blockdata_done)
+        #block.get_blocklist(config)
         settings["block"] = False
 
     if config["core"]["level"] == 0:
@@ -218,33 +195,18 @@ def mainloop(pool):
         tmplist = settings["badapps"]
         settings["badapps"] = list()
         for x in tmplist:
-            send_notification("Killing PID " + str(x))
             kill(x)
 
     if settings["fwll"] == True:
-        global history
         tmplist = settings["badlines"]
-        tmpstat = settings["stats"]
-        tmpstat[0] += len(tmplist) 
-        settings["stats"] = tmpstat
         settings["badlines"] = list()
         for line in tmplist:
             badproto = line[1]
             badip = line[4]
             badport = line[5]
-            
-            if (badip, badport) not in history:
-                fprint("Firewalling " + badip + ":" + str(badport), settings)
-                send_notification("Firewalling " + badip + ":" + str(badport))
-                if win32:
-                    cmd = 'New-NetFirewallRule -DisplayName "IPPigeon Security Rule ' + badip + ':' + str(badport) + '" -Group "IPPigeon" -Direction Outbound -LocalPort Any -Protocol ' + badproto + ' -Action Block -RemoteAddress ' + badip + ' -RemotePort ' + str(badport)
-                    run_cmd(cmd)
-                if linux:
-                    cmd = "nft add rule ip ippigeon output ip daddr " + badip + " " + badproto.lower() + " dport " + str(badport) + " drop"
-                    run_cmd(cmd)
-            else:
-                history.append((badip, badport))
-        settings["badapps"] = list()
+            fprint("Firewalling " + badip + ":" + str(badport))
+            cmd = 'New-NetFirewallRule -DisplayName "IPPigeon Security Rule ' + badip + ':' + str(badport) + '" -Group "IPPigeon" -Direction Outbound -LocalPort Any -Protocol ' + badproto + ' -Action Block -RemoteAddress ' + badip + ' -RemotePort ' + str(badport)
+            run_cmd(cmd)
         
     
     if settings["applyconfig"] == True:
@@ -254,7 +216,7 @@ def mainloop(pool):
         with open(find_data_file('config.yml'), 'w') as filewrite:
             #global config
             yaml.dump(config, filewrite)
-        fprint("Config saved!", settings)
+        fprint("Config saved!")
 
     sleep(interval / (interval * config["core"]["clockspeed"]))
     counter += 1
@@ -296,7 +258,6 @@ if __name__ == '__main__':
             interval = config['core']['interval']
             displaydata = manager.list(range(2)) # data to be printed
             settings = manager.dict() # configuration 
-            # setup shared data variables
             settings["login"] = False
             settings["loggedin"] = False
             settings["showui"] = False
@@ -307,46 +268,24 @@ if __name__ == '__main__':
             settings["badlines"] = list()
             settings["block"] = False
             settings["kill"] = False
-            settings["config"] = config 
+            settings["config"] = config
             settings["applyconfig"] = False
             settings["fwll"] = 0
             settings["running"] = config["core"]["autostart"]
             settings["newdata"] = False
-            settings["appendbad"] = list()
-            settings["logMsg"] = list()
-            settings["whitelist"] = list()
-            tmp = list()
-            tmp.append(["N/A", "TCP", "N/A", "N/A", "20.112.52.29", "5000", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
-            tmp.append(["N/A", "TCP", "N/A", "N/A", "20.81.111.85", "80", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
-            tmp.append(["N/A", "TCP", "N/A", "N/A", "100.115.71.78", "5000", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
-            tmp.append(["N/A", "TCP", "N/A", "N/A", "100.115.71.78", "5000", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
-            tmp.append(["N/A", "TCP", "N/A", "N/A", "174.143.130.167", "443", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
-            tmp.append(["N/A", "TCP", "N/A", "N/A", "216.47.134.203", "443", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
-            tmp.append(["N/A", "TCP", "N/A", "N/A", "34.111.83.189", "443", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
-            settings["appendbad"] = tmp
-            # connections blocked, total connections allowed, count of data sent, data recieved, ratio blocked-unblocked
-            settings["stats"] = [0, 0, 0, 0, 0.0]
-
             killme = manager.Value('d', 0)
             #killme = False
             # launch background UI app as process
             util.clear_fwll()
             p = Process(target=taskbartool.background, args=(displaydata,settings,killme))
-            
             p.start()   
             #p.join() # not a foreground job, so let's not join it
             keeprunning = True
 
             # initial setup
             #netdata_res = pool.apply_async(netstat, callback=netstat_done)
-            if linux:
-                # clear existing nftables entries, create new table
-                run_cmd("nft delete table ip ippigeon")
-                run_cmd("nft create table ip ippigeon")
-                run_cmd("nft add chain ip ippigeon output { type filter hook output priority 0 \; policy accept\; }")
-                #run_cmd("nft add chain ippigeon filter")
-                
-            pool.apply_async(util.read_stats, callback=readstat_done)
+
+
             # launch loop - non-blocking!
             counter = 0
             while(keeprunning):

netstat.py

@@ -5,7 +5,6 @@ from util import run_cmd
 from util import win32
 from util import linux
 from util import macos
-from ippigeon import settings
 import util
 import time
 import csv
@@ -14,7 +13,7 @@ import csv
 
 def process(data):
     setup_child()
-    fprint("netstat processing", settings)
+    fprint("netstat processing")
     if win32:
         #output = data.stdout
         #print(output)
@@ -70,60 +69,22 @@ def process(data):
         with open(find_data_file(util.datafile), "w", newline="") as f:
             writer = csv.writer(f)
             writer.writerows(output2)
-        fprint("done creating csv", settings)
+        fprint("done creating csv")
     
-    if linux:
-        output = data.stdout.decode().split('\n') # split stdout into lines
-        output = [i for i in output if i]
-        if output[0].find("Not all processes could be identified") >= 0:
-            fprint("Not enough permissions", settings)
-            raise PermissionError("Unable to acquire netstat data without admin!")
-        output2 = list()
-        output2.append([util.sysid, util.userid, util.sysdom, util.time()]) # add metadata
-        for line in output:
-            string_split = line.split(" ")
-            string_split = [i for i in string_split if i]
-            #fprint("Input: " + str(string_split))
-
-            if string_split[0].find("Active") >= 0 or string_split[0].find("Proto") >= 0:
-                continue
-            
-            if len(string_split) == 6: # no connection status
-                #fprint(string_split)
-                string_split.append(string_split[-1])
-                string_split[-2] = "UNKNOWN"
-                #fprint(string_split)
-            
-            procname = string_split[6]
-            if procname != "-":
-                string_split2 = procname.split("/")
-                procname = string_split2[1]
-                pid = string_split2[0]
-            else:
-                pid = "Unknown"
-            
-            output2.append([procname, string_split[0], string_split[3], string_split[4], string_split[5], pid]) 
-        #fprint(output2)
-        
-        with open(find_data_file(util.datafile), "w", newline="") as f:
-            writer = csv.writer(f)
-            writer.writerows(output2)
-        fprint("done creating csv", settings)
-
-    if macos:
+    if linux or macos:
         output = data.stdout.decode().split('\n') # split stdout into lines
         #output = data.stdout.decode().split(',')
         #fprint("output data: " + str(output))
         output = [i for i in output if i]
         if output[0].find("Not all processes could be identified") >= 0:
-            fprint("Not enough permissions", settings)
+            fprint("Not enough permissions")
             raise PermissionError("Unable to acquire netstat data without admin!")
         output2 = list()
         output2.append([util.sysid, util.userid, util.sysdom, util.time()]) # add metadata
         for line in output:
             string_split = line.split(" ")
             string_split = [i for i in string_split if i]
-            #fprint("Input: " + str(string_split))
+            fprint("Input: " + str(string_split))
 
             if string_split[1].find("Multipath") >= 0:
                 break
@@ -131,39 +92,38 @@ def process(data):
                 continue
             
             if len(string_split) == 10: # no connection status
-                #fprint(string_split)
+                fprint(string_split)
                 string_split.append(string_split[-1])
                 string_split[-7] = "UNKNOWN"
                 string_split[-4] = string_split[-5]
-                #fprint(string_split)
+                fprint(string_split)
             
            
             
-            output2.append(["Unknown", string_split[0], string_split[3], string_split[4], string_split[5], string_split[8]]) 
-        #fprint("FINAL CSV: " + str(output2))
+            output2.append(["", string_split[0], string_split[3], string_split[4], string_split[5], string_split[8]]) 
+        fprint("FINAL CSV: " + str(output2))
         
         with open(find_data_file(util.datafile), "w", newline="") as f:
             writer = csv.writer(f)
             writer.writerows(output2)
-        fprint("done creating csv", settings)
-
+        fprint("done creating csv")
+        
 
 def start():
     setup_child()
-    fprint("netstat started", settings)
+    fprint("netstat started")
     
     if win32:
         data = run_cmd("netstat -n -o -b")
-        fprint("data acquired", settings)
+        fprint("data acquired")
         return data
 
     if linux:
         data = run_cmd("netstat -atunpw")
-        fprint("data acquired", settings)
+        fprint("data acquired")
         return data
 
     if macos:
         data = run_cmd("netstat -anv")
-        fprint("data acquired", settings)
+        fprint("data acquired")
         return data
-

notification.py

@@ -1,19 +0,0 @@
-from plyer import notification #for getting notification on your PC
-import sys
-from playsound import playsound #pip install playsound==1.2.2
-from util import find_data_file
-relative_path = (sys.path[0])
-sound = find_data_file('music.mp3')
-
-def send_notification(msg="Test notification"):
-    notification.notify(
-        title = "IP Pigeon",
-        message = msg,
-        app_name = "IPPigeon",
-        app_icon = find_data_file("icon.ico"),
-        timeout = 5 #number of seconds message is displayed
-    )
-    playsound(sound)
-
-if __name__ == '__main__':
-    send_notification() 

panel.py

@@ -1,7 +1,5 @@
 import glob
 import wx
-import logging
-import sys
 import wx.lib.buttons as buttons
 import numpy as np
 import pandas as pd
@@ -42,14 +40,14 @@ class OtherFrame(wx.Frame):
     Class used for creating frames other than the main one
     """
     def __init__(self):
-        wx.Frame.__init__(self, None, -1, 'Server Panel', size=(1000, 675))
+        wx.Frame.__init__(self, None, -1, 'Server Panel', size=(1000, 620))
         panel = ServerPanel(self)
         self.SetIcon(wx.Icon(find_data_file("icon.ico"), wx.BITMAP_TYPE_ICO))
 
         self.Show()
 
     def on_edit(self, event):
-        fprint('in on_edit', settings)
+        fprint('in on_edit')
 
     def on_window(self, event):
        return
@@ -103,9 +101,6 @@ class LoginPanel(wx.Panel):
         settings["username"] = basicText.GetValue()
         settings["password"] = pwdText.GetValue()
         settings["login"] = True
-        #self.Destroy()
-        self.Close()
-        self.Parent.Close()
         #OtherFrame()
 
     def on_help(self, event):
@@ -147,11 +142,7 @@ class ServerPanel(wx.Panel):
         super().__init__(parent, size=(500, 500))
         #self.SetBackgroundColour((44, 51, 51))
         self.main_sizer = wx.BoxSizer(wx.VERTICAL)
-        self.secondary_sizer = wx.BoxSizer(wx.HORIZONTAL)
-        self.sub_sizer_left = wx.BoxSizer(wx.VERTICAL)
-        self.sub_sizer_right = wx.BoxSizer(wx.VERTICAL)
-        self.sub_sizer_stats = wx.BoxSizer(wx.VERTICAL)
-        self.sub_sizer_filter = wx.BoxSizer(wx.VERTICAL)
+        secondary_sizer = wx.BoxSizer(wx.HORIZONTAL)
         self.row_obj_dict = {}
         self.list_ctrl = wx.ListCtrl(
             self, size=(-1, 400),
@@ -168,140 +159,64 @@ class ServerPanel(wx.Panel):
         else:
             txt = "Status: Not running"
 
-        if settings["loggedin"] == True:
-            logintxt = "Logged in"
-        else:
-            logintxt = "Not logged in"
-        
-        stattxt = "Statistics:\nBlocked Connections: " + str(settings["stats"][0]) + " \nAllowed Connections: " + str(settings["stats"][1]) + " \nUpload count: " + str(settings["stats"][2]) + " \nDownload count: " + str(settings["stats"][3]) + " \nBlock ratio: " + str("{:.4f}".format(settings["stats"][4]))
         self.statustext = wx.StaticText(self, label=txt) # pos=(20,20))
-        self.logintext = wx.StaticText(self, label=logintxt) # pos=(20,20))
-        self.stattext = wx.StaticText(self, label=stattxt)
         #self.main_sizer.Add(tb)
         #main_sizer.SetBackgroundColour((44, 51, 51))
         # self.pnl1.SetBackgroundColour(wx.BLACK)
         self.handle_columns()
         #self.updatedata()
-        
-        self.main_sizer.Add(self.list_ctrl, 1, wx.ALL | wx.EXPAND, 20)
-        self.main_sizer.Add(self.secondary_sizer, 0, wx.ALL | wx.EXPAND, 20)
-        self.secondary_sizer.Add(self.sub_sizer_left, 0, wx.ALL | wx.CENTER, 20)
-        self.secondary_sizer.Add(self.sub_sizer_right, 0, wx.ALL | wx.CENTER, 20)
-        self.secondary_sizer.Add(self.sub_sizer_filter, 0, wx.ALL | wx.CENTER, 20)
-        self.secondary_sizer.Add(self.sub_sizer_stats, 0, wx.ALL | wx.CENTER, 20)
-        self.start_button = wx.Button(self, label='Start IPPigeon')
-        self.start_button.SetBackgroundColour((205, 215, 206))
-        self.start_button.Bind(wx.EVT_BUTTON, self.on_start)
-        #start_button.Bind(wx.EVT_ENTER_WINDOW, self.on_start)
-        self.stop_button = wx.Button(self, label='Stop IPPigeon')
-        self.stop_button.SetBackgroundColour('#F08080')
-        self.secondary_frame_button = wx.Button(self, label='Settings')
-        self.secondary_frame_button.Bind(wx.EVT_BUTTON, self.on_window)
-        # wx.BORDER(stop_button, wx.BORDER_NONE)
-        self.stop_button.Bind(wx.EVT_BUTTON, self.on_stop)
-
-        self.login_button = wx.Button(self, label='Login')
-        #self.login_button.SetBackgroundColour((205, 215, 206))
-        self.login_button.Bind(wx.EVT_BUTTON, self.on_login)
-        self.quit_button = wx.Button(self, label='Quit IPPigeon')
-        #self.login_button.SetBackgroundColour((205, 215, 206))
-        self.quit_button.Bind(wx.EVT_BUTTON, self.on_quit)
-
-        self.blacklist_button = wx.Button(self, label='Add to blacklist')
-        #self.login_button.SetBackgroundColour((205, 215, 206))
-        self.blacklist_button.Bind(wx.EVT_BUTTON, self.on_blacklist)
-
-        self.whitelist_button = wx.Button(self, label='Add to whitelist')
-        #self.login_button.SetBackgroundColour((205, 215, 206))
-        self.whitelist_button.Bind(wx.EVT_BUTTON, self.on_whitelist)
-        
-        self.log_button = wx.Button(self, label='Logs')
-        self.log_button.Bind(wx.EVT_BUTTON, self.on_log)
-
-        #self.log_window = wx.LogWindow(self, 'Log Window', True)
-    
-        #show_log_button = wx.Button(self, wx.NewId(), 'Show Log')
-        #show_log_button.Bind(wx.EVT_BUTTON, self._show_log)    
-
-        self.sub_sizer_right.Add(self.statustext, 0,  wx.CENTER | wx.ALL | 100, 5)
-        self.sub_sizer_right.Add(self.logintext, 0,  wx.CENTER | wx.ALL | 100, 5)
-        self.sub_sizer_left.Add(self.start_button, 0, wx.CENTER | wx.ALL | 100, 5)
-        self.sub_sizer_left.Add(self.stop_button, 0, wx.CENTER | wx.ALL | 100, 5)
-        self.sub_sizer_left.Add(self.secondary_frame_button, 0, wx.CENTER | wx.ALL | 100, 5)
-        self.sub_sizer_left.Add(self.log_button, 0, wx.CENTER | wx.ALL | 100, 5)
-        self.sub_sizer_right.Add(self.login_button, 0, wx.CENTER | wx.ALL | 100, 5)
-        self.sub_sizer_right.Add(self.quit_button, 0, wx.CENTER | wx.ALL | 100, 5)
-        self.sub_sizer_stats.Add(self.stattext, 0, wx.CENTER | wx.ALL | 100, 5)
-        self.sub_sizer_filter.Add(self.blacklist_button, 0, wx.CENTER | wx.ALL | 100, 5)
-        self.sub_sizer_filter.Add(self.whitelist_button, 0, wx.CENTER | wx.ALL | 100, 5)
-        self.SetSizer(self.main_sizer)
-        self.start_button.Enable(enable=settings["loggedin"])
-        self.stop_button.Enable(enable=settings["loggedin"])
-        self.secondary_frame_button.Enable(enable=settings["loggedin"])
-        
         wx.CallAfter(self.updatedata)
         
         wx.GetApp().ProcessPendingEvents() 
+        self.main_sizer.Add(self.list_ctrl, 0, wx.ALL | wx.EXPAND, 20)
+        start_button = wx.Button(self, label='Start IPPigeon')
+        start_button.SetBackgroundColour((205, 215, 206))
+        start_button.Bind(wx.EVT_BUTTON, self.on_start)
+        #start_button.Bind(wx.EVT_ENTER_WINDOW, self.on_start)
+        stop_button = wx.Button(self, label='Stop IPPigeon')
+        stop_button.SetBackgroundColour('#F08080')
+
+        secondary_frame_button = wx.Button(self, label='Settings')
+        secondary_frame_button.Bind(wx.EVT_BUTTON, self.on_window)
+        # wx.BORDER(stop_button, wx.BORDER_NONE)
+        stop_button.Bind(wx.EVT_BUTTON, self.on_stop)
+        self.main_sizer.Add(self.statustext, 0,  wx.CENTER | wx.ALL | 100, 5)
+        self.main_sizer.Add(start_button, 0, wx.CENTER | wx.ALL | 100, 5)
+        self.main_sizer.Add(stop_button, 0, wx.CENTER | wx.ALL | 100, 5)
+        self.main_sizer.Add(secondary_frame_button, 0, wx.CENTER | wx.ALL | 100, 5)
+        self.SetSizer(self.main_sizer)
 
     def handle_columns(self):
         for col in range(len(COLUMN_NAMES)):
             self.list_ctrl.InsertColumn(0, COLUMN_NAMES[col], width=COLUMN_SIZES[col])
 
-    def checklogin(self):
-        self.start_button.Enable(enable=settings["loggedin"])
-        self.stop_button.Enable(enable=settings["loggedin"])
-        self.secondary_frame_button.Enable(enable=settings["loggedin"])
-        self.quit_button.Enable(enable=settings["loggedin"])
-        if self.list_ctrl.GetFirstSelected() < 0:
-            self.blacklist_button.Enable(enable=False)
-            self.whitelist_button.Enable(enable=False)
-        else:
-            self.blacklist_button.Enable(enable=settings["loggedin"])
-            self.whitelist_button.Enable(enable=settings["loggedin"])
     def updatedata(self):
         global settings
         if settings["running"] == True:
             txt = "Status: Running (" + str(settings["config"]["core"]["level"]) + ")"
         else:
             txt = "Status: Not running"
-        #self.list_ctrl.SetSize(self.GetSize()[0] - 50, self.GetSize()[1] - 200)
-        self.checklogin()
-        if settings["loggedin"] == True:
-            logintxt = "   Logged in"
-            self.login_button.SetLabelText("Logout")
-        else:
-            self.login_button.SetLabelText("Login")
-            logintxt = "Not logged in"
-
-        stattxt = "Statistics:\nBlocked Connections: " + str(settings["stats"][0]) + " \nAllowed Connections: " + str(settings["stats"][1]) + " \nUpload count: " + str(settings["stats"][2]) + " \nDownload count: " + str(settings["stats"][3]) + " \nBlock ratio: " + str("{:.4f}".format(settings["stats"][4]))
         
-        self.statustext.SetLabel(txt)
-        self.logintext.SetLabel(logintxt)
-        self.stattext.SetLabel(stattxt)
+        
         if settings["newdata"] == True:
             settings["newdata"] = False
         else:
-            wx.CallLater(100, self.updatedata)
+            wx.CallLater(1000, self.updatedata)
             return
-        fprint("updatedata called", settings)
+        fprint("updatedata called")
         loaddata()
-        list_total  = self.list_ctrl.GetItemCount()
-        list_top    = self.list_ctrl.GetTopItem()
-        list_pp     = self.list_ctrl.GetCountPerPage()
-        list_bottom = min(list_top + list_pp, list_total - 1)
         if self.list_ctrl.DeleteAllItems():
-            fprint("Items deleted", settings)
+            fprint("Items deleted")
         else:
-            fprint("Unable to delete", settings)
+            fprint("Unable to delete")
         for i in range(len(TEST_FILE)):
             if str(TEST_FILE.iloc[i, 4]).find("TIME_WAIT") >= 0 or str(TEST_FILE.iloc[i, 4]).find("FIN_WAIT_2") >= 0:
                 continue
             idx = 0
-            for app in settings["badapps"]:
+            for ip in settings["badips"]:
                 #fprint(pid)
                 idx = i
-                if TEST_FILE.iloc[i, 5] == app: # "bad" pid, highlight in red at the top
-                    fprint("BAD APP UI: " + str(app), settings)
+                if str(TEST_FILE.iloc[i, 3]).find(ip) >= 0 and str(TEST_FILE.iloc[i, 4]).find("TIME_WAIT") < 0: # "bad" pid, highlight in red at the top
                     idx = self.list_ctrl.InsertItem(0, TEST_FILE.iloc[i, 0])
                     self.list_ctrl.SetItemBackgroundColour(idx, wx.Colour(200, 51, 51))
                     break
@@ -314,77 +229,26 @@ class ServerPanel(wx.Panel):
             for j in range(1, 6):
                 #fprint(str(idx) + " " + str(TEST_FILE.iloc[i, 0]))
                 self.list_ctrl.SetItem(idx, j, str(TEST_FILE.iloc[i, j]))
-                
 
                 #fprint(i, j, TEST_FILE.iloc[i, j])
         #self.SetSizer(self.main_sizer)
-        
-        self.list_ctrl.EnsureVisible((list_bottom - 1))
-        wx.CallLater(100, self.updatedata)
+        self.statustext.SetLabel(txt)
+        wx.CallLater(1000, self.updatedata)
 
     def on_start(self, event):
-        fprint('in on_start', settings)
+        fprint('in on_start')
         settings["running"] = True
 
     def on_stop(self, event):
-        fprint('in on_stop', settings)
+        fprint('in on_stop')
         settings["running"] = False
         util.clear_fwll()
 
     def on_window(self, event):
-        fprint("open settings", settings)
+        fprint("open settings")
         dg = GetData(parent = None)
         dg.ShowModal()
 
-    def on_login(self, event):
-        fprint("open login", settings)
-        if settings["loggedin"] == True:
-            settings["loggedin"] = False
-            return
-        dg2 = ServerFrame()
-        
-        #dg2.ShowModal()
-    def on_quit(self, event):
-        global killme
-        killme.value += 1
-        self.Close()
-        self.Parent.Close()
-
-    def on_blacklist(self, event): 
-        global settings
-        tmp = settings["appendbad"]
-        idx = self.list_ctrl.GetFirstSelected()
-        if idx < 0:
-            return
-        proto = self.list_ctrl.GetItem(idx, 1).GetText()
-        dest = self.list_ctrl.GetItem(idx, 3).GetText()
-        destip, destport = dest.split(":")
-        fprint([proto, destip, destport], settings)
-        tmp.append(["N/A", proto, "N/A", "N/A", destip, destport, "N/A", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A"])
-        settings["appendbad"] = tmp
-        
-    def on_whitelist(self, event): 
-        global settings
-        tmp = settings["whitelist"]
-        idx = self.list_ctrl.GetFirstSelected()
-        if idx < 0:
-            return
-        proto = self.list_ctrl.GetItem(idx, 1).GetText()
-        dest = self.list_ctrl.GetItem(idx, 3).GetText()
-        pid = self.list_ctrl.GetItem(idx, 5).GetText()
-        destip, destport = dest.split(":")
-        fprint([proto, destip, destport], settings)
-        tmp.append([destip, destport, int(pid)])
-        settings["whitelist"] = tmp
-
-    def on_log(self, event):
-        fprint("open log", settings)
-        dg = logData(parent = None)
-        #fprint("attempting logLoop", settings)
-        logData.logLoop(dg)
-        #fprint("did logLoop work?", settings)
-        dg.ShowModal()
-
     def ShowImage(self, imageFile):
         if imageFile == "":
             self.bitmap = wx.StaticBitmap(self, -1, size=(0, 0))
@@ -412,72 +276,20 @@ def openwindow(data, sets, kill):
     killme = kill
     displaydata = data
     settings = sets
-    #if settings["loggedin"]:
-    fprint("Creating server panel", settings)
-    loaddata()
-    app = wx.App(False)
-    frame = OtherFrame()
-    frame.SetIcon(wx.Icon(find_data_file("icon.ico"), wx.BITMAP_TYPE_ICO))
-    app.MainLoop()
-    #else:
-    #fprint("Creating login panel")
-    
-    #app = wx.App(False)
-    #frame = ServerFrame()
-    #frame.SetIcon(wx.Icon(find_data_file("icon.ico"), wx.BITMAP_TYPE_ICO))
-    #app.MainLoop()
-
-
-
-class RedirectText(object):
-    def __init__(self,aWxTextCtrl):
-        self.out = aWxTextCtrl
-
-    def write(self,string):
-        self.out.WriteText(string)
-
-class logData(wx.Dialog):
-    def __init__(self, parent):
-        wx.Dialog.__init__(self, parent, wx.ID_ANY, "Logs: ", size = (800,500))
-        self.p = wx.Panel(self, wx.ID_ANY)
-        #self.logBox = wx.BoxSizer(wx.VERTICAL)
-        self.logsTitle  = wx.StaticText(self.p, label="Logs", pos=(20,20))
-        self.logs = wx.TextCtrl(self.p, pos=(20,40), size= (700, 400), style=wx.TE_MULTILINE|wx.TE_READONLY|wx.HSCROLL)
-        self.logs.WriteText("init\n")
-        #keepLoop = False
-
-    def logLoop(self):
-        #self.logs.WriteText("before while loop\n")
-        i=0
-        #self.logs.Clear()
-        tmp = settings["logMsg"]
-        if (tmp is not None):
-            while(i < len(settings["logMsg"])):
-                #for i in range(len(settings["logMsg"])):
-                #self.logs.WriteText("in while loop\n")
-                self.logs.WriteText(settings["logMsg"][i])
-                self.logs.WriteText("\n")
-                tmp.remove(settings["logMsg"][i])
-                i+=1
-        else:
-            #self.logs.WriteText("settings is none :(")
-            tmp.add("init string for list")
-            logData.logLoop(self)
-        settings["logMsg"] = tmp
-        wx.CallLater(100, self.logLoop)
-        #self.logs.WriteText("after while loop\n")
+    if settings["loggedin"]:
+        fprint("Creating server panel")
+        loaddata()
+        app = wx.App(False)
+        frame = OtherFrame()
+        frame.SetIcon(wx.Icon(find_data_file("icon.ico"), wx.BITMAP_TYPE_ICO))
+        app.MainLoop()
+    else:
+        fprint("Creating login panel")
         
-
-
-        #redir = RedirectText(self.logs)
-        #sys.stdout = redir
-        #self.logs.WriteText(redir)
-
-        #self.logs.LoadFile('output.log')
-    
-    #def doLogging(self):
-        #self.logs.WriteText("hi from doLogging\n")
-        #self.logs.WriteText(settings["logMsg"])
+        app = wx.App(False)
+        frame = ServerFrame()
+        frame.SetIcon(wx.Icon(find_data_file("icon.ico"), wx.BITMAP_TYPE_ICO))
+        app.MainLoop()
 
 class GetData(wx.Dialog):
     def __init__(self, parent):
@@ -526,7 +338,7 @@ class GetData(wx.Dialog):
 
     def OnQuit(self, event):
         # save changes
-        fprint("saving changes...", settings)
+        fprint("saving changes...")
         config = settings["config"]
         config["sftp"]["host"] = self.hostnametext.GetValue()
         config["sftp"]["user"] = self.usertext.GetValue()

requirements.txt

@@ -5,6 +5,4 @@ cx_Freeze
 pandas
 pyyaml
 numpy
-bcrypt
-plyer
-playsound
+bcrypt

setup.py

@@ -5,7 +5,7 @@ debug = True
 debug = not debug
 # Dependencies are automatically detected, but it might need fine tuning.
 # "packages": ["os"] is used as example only
-build_exe_options = {"packages": ["os", "plyer.platforms.win.notification"], "excludes": ["tkinter"], "include_msvcr": True, "include_files": ["icon.png", "config.yml", "keyfile-admin.pem", "WXPython.png", "WXPython_30x30.png", "settings.png", "settings_30x30.png", "icon.ico", "music.mp3"], "optimize": 1}
+build_exe_options = {"packages": ["os"], "excludes": ["tkinter"], "include_msvcr": True, "include_files": ["icon.png", "config.yml", "keyfile-admin.pem", "WXPython.png", "WXPython_30x30.png", "settings.png", "settings_30x30.png", "icon.ico"], "optimize": 1}
 
 # base="Win32GUI" should be used only for Windows GUI app
 base = None

ssh.py

@@ -3,50 +3,38 @@ from fabric import Connection
 from util import find_data_file
 from util import setup_child
 from util import fprint
-from util import macos
 from invoke import exceptions
 import sys
 
-c = None
-settings = None
-
 def sftp_send_data(config, filename, filetype):
     setup_child()
-    if not macos:
-        fprint("Connecting over SSH to " + config['sftp']['host'], settings)
-        global c
-        if c is None:
-            c = Connection(host=config['sftp']['host'], user=config['sftp']['user'], port=config['sftp']['port'], connect_kwargs={"key_filename": find_data_file(config['sftp']['keyfile']),})
-
-        fprint("Sending data over SFTP: " + filename, settings)
-        fprint(c.put(find_data_file(filename), remote=config['sftp']['filepath'][filetype]), settings)
-        fprint("Data sent over SFTP successfully", settings)
+    fprint("Connecting over SSH to " + config['sftp']['host'])
+    c = Connection(host=config['sftp']['host'], user=config['sftp']['user'], port=config['sftp']['port'], connect_kwargs={"key_filename": find_data_file(config['sftp']['keyfile']),})
+    fprint("Sending data over SFTP: " + filename)
+    fprint(c.put(find_data_file(filename), remote=config['sftp']['filepath'][filetype]))
+    fprint("Data sent over SFTP successfully")
     #command = 'ls ' + config['sftp']['filepath'][filetype]
     #fprint(c.run(command))
 
 def check_for_file(config, filename, location):
     setup_child()
-    fprint("Connecting over SSH to " + config['sftp']['host'], settings)
-    global c
-    if c is None:
-        c = Connection(host=config['sftp']['host'], user=config['sftp']['user'], port=config['sftp']['port'], connect_kwargs={"key_filename": find_data_file(config['sftp']['keyfile']),})
-    fprint("Checking for existence of file " + config['sftp']['filepath'][location] + "/" + filename, settings)
+    fprint("Connecting over SSH to " + config['sftp']['host'])
+    c = Connection(host=config['sftp']['host'], user=config['sftp']['user'], port=config['sftp']['port'], connect_kwargs={"key_filename": find_data_file(config['sftp']['keyfile']),})
+    fprint("Checking for existence of file " + config['sftp']['filepath'][location] + "/" + filename)
     try:
         res = c.run("ls -l " + config['sftp']['filepath'][location] + "/" + filename, hide=True)
-        fprint("File " + filename + " exists!", settings)
+        fprint("File " + filename + " exists!")
         return c.run("cat " + config['sftp']['filepath'][location] + "/" + filename, hide=True)
     except exceptions.UnexpectedExit:
         return False
 
 def run_ssh(config, command, location):
     setup_child()
-    fprint("Connecting over SSH to " + config['sftp']['host'], settings)
-    global c
-    if c is None:
-        c = Connection(host=config['sftp']['host'], user=config['sftp']['user'], port=config['sftp']['port'], connect_kwargs={"key_filename": find_data_file(config['sftp']['keyfile']),})
-    fprint("cd to " + config['sftp']['filepath'][location], settings)
+    fprint("Connecting over SSH to " + config['sftp']['host'])
+    c = Connection(host=config['sftp']['host'], user=config['sftp']['user'], port=config['sftp']['port'], connect_kwargs={"key_filename": find_data_file(config['sftp']['keyfile']),})
+    fprint("cd to " + config['sftp']['filepath'][location])
     with c.cd(config['sftp']['filepath'][location]):
-        fprint("Running ssh command: " + command, settings)
+        fprint("Running ssh command: " + command)
         res = c.run(command, hide=True, asynchronous=True)
         return res
 

taskbartool.py

@@ -42,7 +42,7 @@ class TaskBarIcon(wx.adv.TaskBarIcon):
         self.SetIcon(icon, TRAY_TOOLTIP)
 
     def on_left_down(self, event):      
-        fprint('Tray icon was left-clicked.', settings)
+        fprint ('Tray icon was left-clicked.')
 
     def on_open(self, event):
         settings["showui"] = True
@@ -81,7 +81,7 @@ def background(data, sets, kill):
     app = TaskbarApp(False)
     displaydata = data
     settings = sets
-    fprint("Creating taskbar icon", settings)
+    fprint("Creating taskbar icon")
     app.MainLoop()
         
 TRAY_ICON = find_data_file('icon.png')

util.py

@@ -6,16 +6,11 @@ from sys import platform
 import time as t
 from time import sleep
 import uuid
-import csv
 
 win32 = platform == "win32"
 linux = platform == "linux" or platform == "linux2"
 macos = platform == "darwin"
 datafile = ""
-logMsg = ""
-logCont = ""
-
-settings = None
 
 if win32:
     sysid = hex(uuid.getnode())
@@ -48,28 +43,21 @@ def time():
 def kill(pid):
     setup_child()
     if pid > 0:
-        fprint("Killing PID " + str(pid), settings)
+        fprint("Killing PID " + str(pid))
         os.kill(int(pid), 9)
-        fprint("Signal 9 sent to PID " + str(pid), settings)
+        fprint("Signal 9 sent to PID " + str(pid))
 
-def fprint(msg, settings = None):
+def fprint(msg):
     #if not getattr(sys, "frozen", False):
     setup_child()
     try:
         frm = inspect.stack()[1]
         
         mod = inspect.getmodule(frm[0])
-        logMsg = '[' + mod.__name__ + ":" + frm.function + ']:' + str(msg)
-        
-        print(logMsg)
-        if (settings is not None):
-            tmpList = settings["logMsg"]
-            tmpList.append(logMsg)
-            settings["logMsg"] = tmpList
+        print('[' + mod.__name__ + ":" + frm.function + ']:', str(msg))
     except Exception as e:
         try:
             print('[????:' + frm.function + ']:', str(msg))
-            print('[util:fprint]: ' + str(e))
         except:
             print('[????]:', str(msg))
         
@@ -98,23 +86,21 @@ def run_cmd(cmd):
         #else:
         #    print("alt")
         startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW    # , "-WindowStyle", "hidden"
-        fprint("running PS command: " + cmd, settings)
+        fprint("running PS command: " + cmd)
         completed = subprocess.run(["powershell", "-Command", cmd], capture_output=True, startupinfo=startupinfo)
-        fprint("ran PS command successfully", settings)
+        fprint("ran PS command successfully")
         #completed = subprocess.run(["powershell", "-WindowStyle", "hidden", "-Command", cmd], capture_output=True, startupinfo=startupinfo)
         return completed
     if linux or macos:
-        fprint("running sh command: " + cmd, settings)
+        fprint("running sh command: " + cmd)
         completed = subprocess.run(["sh", "-c", cmd], capture_output=True)
-        fprint("ran sh command successfully", settings)
+        fprint("ran sh command successfully")
         return completed
 
-def setup_child(sets=None):
+def setup_child():
     if not getattr(sys, "frozen", False):
         sys.stdout = Logger(filename=find_data_file("output.log"))
         sys.stderr = Logger(filename=find_data_file("output.log"))
-    if sets is not None:
-        settings = sets
 
 class Logger(object):
     def __init__(self, filename="output.log"):
@@ -136,35 +122,3 @@ class Logger(object):
 def clear_fwll():
     if win32:
         run_cmd('Remove-NetFirewallRule -Group "IPPigeon"')
-    if linux:
-        run_cmd("nft delete table ip ippigeon")
-
-
-def write_stats(stats):
-    fprint("Writing stats", settings)
-    tmp = list()
-    tmp.append(["connections blocked", "connections allowed", "data uploaded", "data recieved", "block ratio"])
-    tmp.append(stats)
-    
-    with open(find_data_file("stats.csv"), "w", newline="") as f:
-        writer = csv.writer(f)
-        writer.writerows(tmp)
-    fprint("Done writing stats", settings)
-
-def read_stats():
-    with open(find_data_file("stats.csv"), newline='') as csvfile:
-        csvreader = csv.reader(csvfile, delimiter=',', quotechar='|')
-        header = True
-        fprint(csvreader, settings)
-        data = list()
-        for line in csvreader:
-            fprint(line, settings)
-            if header:
-                header = False
-                continue
-            data = line
-        for idx in range(len(data) - 1):
-            data[idx] = int(data[idx])
-        data[len(data) - 1] = float(data[len(data) - 1])
-        return data
-            

wizard.py

@@ -0,0 +1,34 @@
+import wx
+from wx.adv import Wizard, WizardPageSimple
+
+class TitlePage(WizardPageSimple):
+
+    def __init__(self, parent, title):
+        WizardPageSimple.__init__(self, parent)
+
+        sizer = wx.BoxSizer(wx.VERTICAL)
+        self.SetSizer(sizer)
+
+        title = wx.StaticText(self, wx.ID_ANY, title)
+        title.SetFont(wx.Font(18, wx.SWISS, wx.NORMAL, wx.BOLD))
+        sizer.Add(title, 0, wx.ALIGN_CENTER|wx.ALL, 5)
+        sizer.Add(wx.StaticLine(self, wx.ID_ANY), 0, wx.EXPAND|wx.ALL, 5)
+
+
+def main():
+    wizard = Wizard(None, wx.ID_ANY, "Simple Wizard")
+    page1 = TitlePage(wizard, "Page 1")
+    page2 = TitlePage(wizard, "Page 2")
+    page3 = TitlePage(wizard, "Page 3")
+
+    WizardPageSimple.Chain(page1, page2)
+    WizardPageSimple.Chain(page2, page3)
+    wizard.FitToPage(page1)
+
+    wizard.RunWizard(page1)
+    wizard.Destroy()
+
+if __name__ == "__main__":
+    app = wx.App()
+    main()
+    app.MainLoop()

wizardTutorial.py

@@ -0,0 +1,111 @@
+import wx
+########################################################################
+class WizardPage(wx.Panel):
+    """"""
+    #----------------------------------------------------------------------
+    def __init__(self, parent, title=None):
+        """Constructor"""
+        wx.Panel.__init__(self, parent)
+        
+        sizer = wx.BoxSizer(wx.VERTICAL)
+        self.SetSizer(sizer)
+        
+        if title:
+            title = wx.StaticText(self, -1, title)
+            title.SetFont(wx.Font(18, wx.SWISS, wx.NORMAL, wx.BOLD))
+            sizer.Add(title, 0, wx.ALIGN_CENTRE|wx.ALL, 5)
+            sizer.Add(wx.StaticLine(self, -1), 0, wx.EXPAND|wx.ALL, 5)
+    
+    
+########################################################################
+class WizardPanel(wx.Panel):
+    """"""
+    #----------------------------------------------------------------------
+    def __init__(self, parent):
+        """Constructor"""
+        wx.Panel.__init__(self, parent=parent)
+        self.pages = []
+        self.page_num = 0
+        
+        self.mainSizer = wx.BoxSizer(wx.VERTICAL)
+        self.panelSizer = wx.BoxSizer(wx.VERTICAL)
+        btnSizer = wx.BoxSizer(wx.HORIZONTAL)
+                
+        # add prev/next buttons
+        self.prevBtn = wx.Button(self, label="Previous")
+        self.prevBtn.Bind(wx.EVT_BUTTON, self.onPrev)
+        btnSizer.Add(self.prevBtn, 0, wx.ALL|wx.ALIGN_RIGHT, 5)
+        
+        self.nextBtn = wx.Button(self, label="Next")
+        self.nextBtn.Bind(wx.EVT_BUTTON, self.onNext)
+        btnSizer.Add(self.nextBtn, 0, wx.ALL|wx.ALIGN_RIGHT, 5)
+        
+        # finish layout
+        self.mainSizer.Add(self.panelSizer, 1, wx.EXPAND)
+        self.mainSizer.Add(btnSizer, 0, wx.ALIGN_RIGHT)
+        self.SetSizer(self.mainSizer)
+        
+        
+    #----------------------------------------------------------------------
+    def addPage(self, title=None):
+        """"""
+        panel = WizardPage(self, title)
+        self.panelSizer.Add(panel, 2, wx.EXPAND)
+        self.pages.append(panel)
+        if len(self.pages) > 1:
+            # hide all panels after the first one
+            panel.Hide()
+            self.Layout()
+            
+    #----------------------------------------------------------------------
+    def onNext(self, event):
+        """"""
+        pageCount = len(self.pages)
+        if pageCount-1 != self.page_num:
+            self.pages[self.page_num].Hide()
+            self.page_num += 1
+            self.pages[self.page_num].Show()
+            self.panelSizer.Layout()
+        else:
+            print("End of pages!")
+            
+        if self.nextBtn.GetLabel() == "Finish":
+            # close the app
+            self.GetParent().Close()
+            
+        if pageCount == self.page_num+1:
+            # change label
+            self.nextBtn.SetLabel("Finish")
+    
+    #----------------------------------------------------------------------
+    def onPrev(self, event):
+        """"""
+        pageCount = len(self.pages)
+        if self.page_num-1 != -1:
+            self.pages[self.page_num].Hide()
+            self.page_num -= 1
+            self.pages[self.page_num].Show()
+            self.panelSizer.Layout()
+        else:
+            print("You're already on the first page!")
+        
+    
+########################################################################
+class MainFrame(wx.Frame):
+    """"""
+    #----------------------------------------------------------------------
+    def __init__(self):
+        """Constructor"""
+        wx.Frame.__init__(self, None, title="Generic Wizard", size=(800,600))
+        
+        self.panel = WizardPanel(self)
+        self.panel.addPage("Page 1")
+        self.panel.addPage("Page 2")
+        self.panel.addPage("Page 3")
+        
+        self.Show()
+    
+if __name__ == "__main__":
+    app = wx.App()
+    frame = MainFrame()
+    app.MainLoop()