578 lines
		
	
	
	
		
			16 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable file
		
	
	
	
	
			
		
		
	
	
			578 lines
		
	
	
	
		
			16 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable file
		
	
	
	
	
| #!/usr/bin/python
 | |
| 
 | |
| # tinc-gui -- GUI for controlling a running tincd
 | |
| # Copyright (C) 2009-2012 Guus Sliepen <guus@tinc-vpn.org>
 | |
| #
 | |
| # This program is free software; you can redistribute it and/or modify
 | |
| # it under the terms of the GNU General Public License as published by
 | |
| # the Free Software Foundation; either version 2 of the License, or
 | |
| # (at your option) any later version.
 | |
| #
 | |
| # This program is distributed in the hope that it will be useful,
 | |
| # but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | |
| # GNU General Public License for more details.
 | |
| #
 | |
| # You should have received a copy of the GNU General Public License along
 | |
| # with this program; if not, write to the Free Software Foundation, Inc.,
 | |
| # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 | |
| 
 | |
| import string
 | |
| import socket
 | |
| import wx
 | |
| import sys
 | |
| import os
 | |
| import platform
 | |
| import time
 | |
| from wx.lib.mixins.listctrl import ColumnSorterMixin
 | |
| from wx.lib.mixins.listctrl import ListCtrlAutoWidthMixin
 | |
| 
 | |
| if platform.system == 'Windows':
 | |
| 	import _winreg
 | |
| 
 | |
| # Classes to interface with a running tinc daemon
 | |
| 
 | |
| REQ_STOP = 0
 | |
| REQ_RELOAD = 1
 | |
| REQ_RESTART = 2
 | |
| REQ_DUMP_NODES = 3
 | |
| REQ_DUMP_EDGES = 4
 | |
| REQ_DUMP_SUBNETS = 5
 | |
| REQ_DUMP_CONNECTIONS = 6
 | |
| REQ_DUMP_GRAPH = 7
 | |
| REQ_PURGE = 8
 | |
| REQ_SET_DEBUG = 9
 | |
| REQ_RETRY = 10
 | |
| REQ_CONNECT = 11
 | |
| REQ_DISCONNECT = 12
 | |
| 
 | |
| ID = 0
 | |
| ACK = 4
 | |
| CONTROL = 18
 | |
| 
 | |
| class Node:
 | |
| 	def parse(self, args):
 | |
| 		self.name = args[0]
 | |
| 		self.address = args[1]
 | |
| 		self.port = args[3]
 | |
| 		self.cipher = int(args[4])
 | |
| 		self.digest = int(args[5])
 | |
| 		self.maclength = int(args[6])
 | |
| 		self.compression = int(args[7])
 | |
| 		self.options = int(args[8], 0x10)
 | |
| 		self.status = int(args[9], 0x10)
 | |
| 		self.nexthop = args[10]
 | |
| 		self.via = args[11]
 | |
| 		self.distance = int(args[12])
 | |
| 		self.pmtu = int(args[13])
 | |
| 		self.minmtu = int(args[14])
 | |
| 		self.maxmtu = int(args[15])
 | |
| 		self.last_state_change = float(args[16])
 | |
| 
 | |
| 		self.subnets = {}
 | |
| 
 | |
| class Edge:
 | |
| 	def parse(self, args):
 | |
| 		self.fr = args[0]
 | |
| 		self.to = args[1]
 | |
| 		self.address = args[2]
 | |
| 		self.port = args[4]
 | |
| 		self.options = int(args[5], 16)
 | |
| 		self.weight = int(args[6])
 | |
| 
 | |
| class Subnet:
 | |
| 	def parse(self, args):
 | |
| 		if args[0].find('#') >= 0:
 | |
| 			(address, self.weight) = args[0].split('#', 1)
 | |
| 		else:
 | |
| 			self.weight = 10
 | |
| 			address = args[0]
 | |
| 
 | |
| 		if address.find('/') >= 0:
 | |
| 			(self.address, self.prefixlen) = address.split('/', 1)
 | |
| 		else:
 | |
| 			self.address = address
 | |
| 			self.prefixlen = '48'
 | |
| 
 | |
| 		self.owner = args[1]	
 | |
| 
 | |
| class Connection:
 | |
| 	def parse(self, args):
 | |
| 		self.name = args[0]
 | |
| 		self.address = args[1]
 | |
| 		self.port = args[3]
 | |
| 		self.options = int(args[4], 0x10)
 | |
| 		self.socket = int(args[5])
 | |
| 		self.status = int(args[6], 0x10)
 | |
| 		self.weight = 123
 | |
| 
 | |
| class VPN:
 | |
| 	confdir = '/etc/tinc'
 | |
| 	piddir = '/var/run/'
 | |
| 
 | |
| 	def connect(self):
 | |
| 		f = open(self.pidfile)
 | |
| 		info = string.split(f.readline())
 | |
| 		f.close()
 | |
| 		s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 | |
| 		s.connect((info[2], int(info[4])))
 | |
| 		self.sf = s.makefile()
 | |
| 		s.close()
 | |
| 		hello = string.split(self.sf.readline())
 | |
| 		self.name = hello[1]
 | |
| 		self.sf.write('0 ^' + info[1] + ' 17\r\n')
 | |
| 		self.sf.flush()
 | |
| 		resp = string.split(self.sf.readline())
 | |
| 		self.port = info[4]
 | |
| 		self.nodes = {}
 | |
| 		self.edges = {}
 | |
| 		self.subnets = {}
 | |
| 		self.connections = {}
 | |
| 		self.refresh()
 | |
| 
 | |
| 	def refresh(self):
 | |
| 		self.sf.write('18 3\r\n18 4\r\n18 5\r\n18 6\r\n')
 | |
| 		self.sf.flush()
 | |
| 
 | |
| 		for node in self.nodes.values():
 | |
| 			node.visited = False
 | |
| 		for edge in self.edges.values():
 | |
| 			edge.visited = False
 | |
| 		for subnet in self.subnets.values():
 | |
| 			subnet.visited = False
 | |
| 		for connections in self.connections.values():
 | |
| 			connections.visited = False
 | |
| 
 | |
| 		while True:
 | |
| 			resp = string.split(self.sf.readline())
 | |
| 			if len(resp) < 2:
 | |
| 				break
 | |
| 			if resp[0] != '18':
 | |
| 				break
 | |
| 			if resp[1] == '3':
 | |
| 				if len(resp) < 19:
 | |
| 					continue
 | |
| 				node = self.nodes.get(resp[2]) or Node()
 | |
| 				node.parse(resp[2:])
 | |
| 				node.visited = True
 | |
| 				self.nodes[resp[2]] = node
 | |
| 			elif resp[1] == '4':
 | |
| 				if len(resp) < 9:
 | |
| 					continue
 | |
| 				edge = self.nodes.get((resp[2], resp[3])) or Edge()
 | |
| 				edge.parse(resp[2:])
 | |
| 				edge.visited = True
 | |
| 				self.edges[(resp[2], resp[3])] = edge
 | |
| 			elif resp[1] == '5':
 | |
| 				if len(resp) < 4:
 | |
| 					continue
 | |
| 				subnet = self.subnets.get((resp[2], resp[3])) or Subnet()
 | |
| 				subnet.parse(resp[2:])
 | |
| 				subnet.visited = True
 | |
| 				self.subnets[(resp[2], resp[3])] = subnet
 | |
| 				self.nodes[subnet.owner].subnets[resp[2]] = subnet
 | |
| 			elif resp[1] == '6':
 | |
| 				if len(resp) < 9:
 | |
| 					break
 | |
| 				connection = self.connections.get((resp[2], resp[3], resp[5])) or Connection()
 | |
| 				connection.parse(resp[2:])
 | |
| 				connection.visited = True
 | |
| 				self.connections[(resp[2], resp[3], resp[5])] = connection
 | |
| 			else:
 | |
| 				break
 | |
| 
 | |
| 		for key, subnet in self.subnets.items():
 | |
| 			if not subnet.visited:
 | |
| 				del self.subnets[key]
 | |
| 
 | |
| 		for key, edge in self.edges.items():
 | |
| 			if not edge.visited:
 | |
| 				del self.edges[key]
 | |
| 
 | |
| 		for key, node in self.nodes.items():
 | |
| 			if not node.visited:
 | |
| 				del self.nodes[key]
 | |
| 			else:
 | |
| 				for key, subnet in node.subnets.items():
 | |
| 					if not subnet.visited:
 | |
| 						del node.subnets[key]
 | |
| 
 | |
| 		for key, connection in self.connections.items():
 | |
| 			if not connection.visited:
 | |
| 				del self.connections[key]
 | |
| 
 | |
| 	def close(self):
 | |
| 		self.sf.close()
 | |
| 
 | |
| 	def disconnect(self, name):
 | |
| 		self.sf.write('18 12 ' + name + '\r\n')
 | |
| 		self.sf.flush()
 | |
| 		resp = string.split(self.sf.readline())
 | |
| 
 | |
| 	def debug(self, level = -1):
 | |
| 		self.sf.write('18 9 ' + str(level) + '\r\n')
 | |
| 		self.sf.flush()
 | |
| 		resp = string.split(self.sf.readline())
 | |
| 		return int(resp[2])
 | |
| 
 | |
| 	def __init__(self, netname = None, pidfile = None):
 | |
| 		if platform.system == 'Windows':
 | |
| 			try:
 | |
| 				reg = _winreg.ConnectRegistry(None, HKEY_LOCAL_MACHINE)
 | |
| 				key = _winreg.OpenKey(reg, "SOFTWARE\\tinc")
 | |
| 				VPN.confdir = _winreg.QueryValue(key, None)
 | |
| 			except WindowsError:
 | |
| 				pass
 | |
| 
 | |
| 		if netname:
 | |
| 			self.netname = netname
 | |
| 			self.confbase = os.path.join(VPN.confdir, netname)
 | |
| 		else:
 | |
| 			self.confbase = VPN.confdir
 | |
| 
 | |
| 		self.tincconf = os.path.join(self.confbase, 'tinc.conf')
 | |
| 
 | |
| 		if pidfile != None:
 | |
| 			self.pidfile = pidfile
 | |
| 		else:
 | |
| 			if platform.system == 'Windows':
 | |
| 				self.pidfile = os.path.join(self.confbase, 'pid')
 | |
| 			else:
 | |
| 				if netname:
 | |
| 					self.pidfile = os.path.join(VPN.piddir, 'tinc.' + netname + '.pid')
 | |
| 				else:
 | |
| 					self.pidfile = os.path.join(VPN.piddir, 'tinc.pid')
 | |
| 
 | |
| # GUI starts here
 | |
| 
 | |
| argv0 = sys.argv[0]
 | |
| del sys.argv[0]
 | |
| netname = None
 | |
| pidfile = None
 | |
| 
 | |
| def usage(exitcode = 0):
 | |
| 	print('Usage: ' + argv0 + ' [options]')
 | |
| 	print('\nValid options are:')
 | |
| 	print('  -n, --net=NETNAME       Connect to net NETNAME.')
 | |
| 	print('      --pidfile=FILENAME  Read control cookie from FILENAME.')
 | |
| 	print('      --help              Display this help and exit.')
 | |
| 	print('\nReport bugs to tinc@tinc-vpn.org.')
 | |
| 	sys.exit(exitcode)
 | |
| 
 | |
| while sys.argv:
 | |
| 	if sys.argv[0] in ('-n', '--net'):
 | |
| 		del sys.argv[0]
 | |
| 		netname = sys.argv[0]
 | |
| 	elif sys.argv[0] in ('--pidfile'):
 | |
| 		del sys.argv[0]
 | |
| 		pidfile = sys.argv[0]
 | |
| 	elif sys.argv[0] in ('--help'):
 | |
| 		usage(0)
 | |
| 	else:
 | |
| 		print(argv0 + ': unrecognized option \'' + sys.argv[0] + '\'')
 | |
| 		usage(1)
 | |
| 
 | |
| 	del sys.argv[0]
 | |
| 
 | |
| if netname == None:
 | |
| 	netname = os.getenv("NETNAME")
 | |
| 
 | |
| if netname == ".":
 | |
| 	netname = None
 | |
| 
 | |
| vpn = VPN(netname, pidfile)
 | |
| vpn.connect()
 | |
| 
 | |
| class SuperListCtrl(wx.ListCtrl, ColumnSorterMixin, ListCtrlAutoWidthMixin):
 | |
|     def __init__(self, parent, style):
 | |
|         wx.ListCtrl.__init__(self, parent, -1, style=wx.LC_REPORT | wx.LC_HRULES | wx.LC_VRULES)
 | |
| 	ListCtrlAutoWidthMixin.__init__(self)
 | |
|         ColumnSorterMixin.__init__(self, 14)
 | |
| 
 | |
|     def GetListCtrl(self):
 | |
|         return self
 | |
| 
 | |
| 
 | |
| class SettingsPage(wx.Panel):
 | |
| 	def OnDebugLevel(self, event):
 | |
| 		vpn.debug(self.debug.GetValue())
 | |
| 
 | |
| 	def __init__(self, parent, id):
 | |
| 		wx.Panel.__init__(self, parent, id)
 | |
| 		grid = wx.FlexGridSizer(cols = 2)
 | |
| 		grid.AddGrowableCol(0, 1)
 | |
| 
 | |
| 		namelabel = wx.StaticText(self, -1, 'Name:')
 | |
| 		self.name = wx.TextCtrl(self, -1, vpn.name)
 | |
| 		grid.Add(namelabel)
 | |
| 		grid.Add(self.name)
 | |
| 
 | |
| 		portlabel = wx.StaticText(self, -1, 'Port:')
 | |
| 		self.port = wx.TextCtrl(self, -1, vpn.port)
 | |
| 		grid.Add(portlabel)
 | |
| 		grid.Add(self.port)
 | |
| 
 | |
| 		debuglabel = wx.StaticText(self, -1, 'Debug level:')
 | |
| 		self.debug = wx.SpinCtrl(self, min = 0, max = 5, initial = vpn.debug())
 | |
| 		self.debug.Bind(wx.EVT_SPINCTRL, self.OnDebugLevel)
 | |
| 		grid.Add(debuglabel)
 | |
| 		grid.Add(self.debug)
 | |
| 
 | |
| 		modelabel = wx.StaticText(self, -1, 'Mode:')
 | |
| 		self.mode = wx.ComboBox(self, -1, style = wx.CB_READONLY, value = 'Router', choices = ['Router', 'Switch', 'Hub'])
 | |
| 		grid.Add(modelabel)
 | |
| 		grid.Add(self.mode)
 | |
| 
 | |
| 		self.SetSizer(grid)
 | |
| 
 | |
| class ConnectionsPage(wx.Panel):
 | |
| 	def __init__(self, parent, id):
 | |
| 		wx.Panel.__init__(self, parent, id)
 | |
| 		self.list = wx.ListCtrl(self, id, style=wx.LC_REPORT | wx.LC_HRULES | wx.LC_VRULES)
 | |
| 		self.list.InsertColumn(0, 'Name')
 | |
| 		self.list.InsertColumn(1, 'Address')
 | |
| 		self.list.InsertColumn(2, 'Port')
 | |
| 		self.list.InsertColumn(3, 'Options')
 | |
| 		self.list.InsertColumn(4, 'Weight')
 | |
| 
 | |
| 		hbox = wx.BoxSizer(wx.HORIZONTAL)
 | |
| 		hbox.Add(self.list, 1, wx.EXPAND)
 | |
| 		self.SetSizer(hbox)
 | |
| 		self.refresh()
 | |
| 
 | |
| 	class ContextMenu(wx.Menu):
 | |
| 		def __init__(self, item):
 | |
| 			wx.Menu.__init__(self)
 | |
| 
 | |
| 			self.item = item
 | |
| 
 | |
| 			disconnect = wx.MenuItem(self, -1, 'Disconnect')
 | |
| 			self.AppendItem(disconnect)
 | |
| 			self.Bind(wx.EVT_MENU, self.OnDisconnect, id=disconnect.GetId())
 | |
| 
 | |
| 		def OnDisconnect(self, event):
 | |
| 			vpn.disconnect(self.item[0])
 | |
| 
 | |
| 	def OnContext(self, event):
 | |
| 		i = event.GetIndex()
 | |
| 		self.PopupMenu(self.ContextMenu(self.list.itemDataMap[event.GetIndex()]), event.GetPosition())
 | |
| 
 | |
| 	def refresh(self):
 | |
| 		self.list.itemDataMap = {}
 | |
| 		i = 0
 | |
| 
 | |
| 		for key, connection in vpn.connections.items():
 | |
| 			if self.list.GetItemCount() <= i:
 | |
| 				self.list.InsertStringItem(i, connection.name)
 | |
| 			else:
 | |
| 				self.list.SetStringItem(i, 0, connection.name)
 | |
| 			self.list.SetStringItem(i, 1, connection.address)
 | |
| 			self.list.SetStringItem(i, 2, connection.port)
 | |
| 			self.list.SetStringItem(i, 3, str(connection.options))
 | |
| 			self.list.SetStringItem(i, 4, str(connection.weight))
 | |
| 			self.list.itemDataMap[i] = (connection.name, connection.address, connection.port, connection.options, connection.weight)
 | |
| 			self.list.Bind(wx.EVT_LIST_ITEM_RIGHT_CLICK, self.OnContext)
 | |
| 			i += 1
 | |
| 
 | |
| 		while self.list.GetItemCount() > i:
 | |
| 			self.list.DeleteItem(self.list.GetItemCount() - 1)
 | |
| 
 | |
| 
 | |
| class NodesPage(wx.Panel):
 | |
| 	def __init__(self, parent, id):
 | |
| 		wx.Panel.__init__(self, parent, id)
 | |
| 		self.list = SuperListCtrl(self, id)
 | |
| 		self.list.InsertColumn( 0, 'Name')
 | |
| 		self.list.InsertColumn( 1, 'Address')
 | |
| 		self.list.InsertColumn( 2, 'Port')
 | |
| 		self.list.InsertColumn( 3, 'Cipher')
 | |
| 		self.list.InsertColumn( 4, 'Digest')
 | |
| 		self.list.InsertColumn( 5, 'MACLength')
 | |
| 		self.list.InsertColumn( 6, 'Compression')
 | |
| 		self.list.InsertColumn( 7, 'Options')
 | |
| 		self.list.InsertColumn( 8, 'Status')
 | |
| 		self.list.InsertColumn( 9, 'Nexthop')
 | |
| 		self.list.InsertColumn(10, 'Via')
 | |
| 		self.list.InsertColumn(11, 'Distance')
 | |
| 		self.list.InsertColumn(12, 'PMTU')
 | |
| 		self.list.InsertColumn(13, 'Min MTU')
 | |
| 		self.list.InsertColumn(14, 'Max MTU')
 | |
| 		self.list.InsertColumn(15, 'Since')
 | |
| 
 | |
| 		hbox = wx.BoxSizer(wx.HORIZONTAL)
 | |
| 		hbox.Add(self.list, 1, wx.EXPAND)
 | |
| 		self.SetSizer(hbox)
 | |
| 		self.refresh()
 | |
| 
 | |
| 	def refresh(self):
 | |
| 		self.list.itemDataMap = {}
 | |
| 		i = 0
 | |
| 
 | |
| 		for key, node in vpn.nodes.items():
 | |
| 			if self.list.GetItemCount() <= i:
 | |
| 				self.list.InsertStringItem(i, node.name)
 | |
| 			else:
 | |
| 				self.list.SetStringItem(i,  0, node.name)
 | |
| 			self.list.SetStringItem(i,  1, node.address)
 | |
| 			self.list.SetStringItem(i,  2, node.port)
 | |
| 			self.list.SetStringItem(i,  3, str(node.cipher))
 | |
| 			self.list.SetStringItem(i,  4, str(node.digest))
 | |
| 			self.list.SetStringItem(i,  5, str(node.maclength))
 | |
| 			self.list.SetStringItem(i,  6, str(node.compression))
 | |
| 			self.list.SetStringItem(i,  7, format(node.options, "x"))
 | |
| 			self.list.SetStringItem(i,  8, format(node.status, "04x"))
 | |
| 			self.list.SetStringItem(i,  9, node.nexthop)
 | |
| 			self.list.SetStringItem(i, 10, node.via)
 | |
| 			self.list.SetStringItem(i, 11, str(node.distance))
 | |
| 			self.list.SetStringItem(i, 12, str(node.pmtu))
 | |
| 			self.list.SetStringItem(i, 13, str(node.minmtu))
 | |
| 			self.list.SetStringItem(i, 14, str(node.maxmtu))
 | |
| 			if node.last_state_change:
 | |
| 				since = time.strftime("%Y-%m-%d %H:%M", time.localtime(node.last_state_change))
 | |
| 			else:
 | |
| 				since = "never"
 | |
| 			self.list.SetStringItem(i, 15, since)
 | |
| 			self.list.itemDataMap[i] = (node.name, node.address, node.port, node.cipher, node.digest, node.maclength, node.compression, node.options, node.status, node.nexthop, node.via, node.distance, node.pmtu, node.minmtu, node.maxmtu)
 | |
| 			self.list.SetItemData(i, i)
 | |
| 			i += 1
 | |
| 
 | |
| 		while self.list.GetItemCount() > i:
 | |
| 			self.list.DeleteItem(self.list.GetItemCount() - 1)
 | |
| 
 | |
| class EdgesPage(wx.Panel):
 | |
| 	def __init__(self, parent, id):
 | |
| 		wx.Panel.__init__(self, parent, id)
 | |
| 		self.list = wx.ListCtrl(self, id, style=wx.LC_REPORT | wx.LC_HRULES | wx.LC_VRULES)
 | |
| 		self.list.InsertColumn(0, 'From')
 | |
| 		self.list.InsertColumn(1, 'To')
 | |
| 		self.list.InsertColumn(2, 'Address')
 | |
| 		self.list.InsertColumn(3, 'Port')
 | |
| 		self.list.InsertColumn(4, 'Options')
 | |
| 		self.list.InsertColumn(5, 'Weight')
 | |
| 
 | |
| 		hbox = wx.BoxSizer(wx.HORIZONTAL)
 | |
| 		hbox.Add(self.list, 1, wx.EXPAND)
 | |
| 		self.SetSizer(hbox)
 | |
| 		self.refresh()
 | |
| 
 | |
| 	def refresh(self):
 | |
| 		self.list.itemDataMap = {}
 | |
| 		i = 0
 | |
| 
 | |
| 		for key, edge in vpn.edges.items():
 | |
| 			if self.list.GetItemCount() <= i:
 | |
| 				self.list.InsertStringItem(i, edge.fr)
 | |
| 			else:
 | |
| 				self.list.SetStringItem(i, 0, edge.fr)
 | |
| 			self.list.SetStringItem(i, 1, edge.to)
 | |
| 			self.list.SetStringItem(i, 2, edge.address)
 | |
| 			self.list.SetStringItem(i, 3, edge.port)
 | |
| 			self.list.SetStringItem(i, 4, format(edge.options, "x"))
 | |
| 			self.list.SetStringItem(i, 5, str(edge.weight))
 | |
| 			self.list.itemDataMap[i] = (edge.fr, edge.to, edge.address, edge.port, edge.options, edge.weight)
 | |
| 			i += 1
 | |
| 
 | |
| 		while self.list.GetItemCount() > i:
 | |
| 			self.list.DeleteItem(self.list.GetItemCount() - 1)
 | |
| 
 | |
| class SubnetsPage(wx.Panel):
 | |
| 	def __init__(self, parent, id):
 | |
| 		wx.Panel.__init__(self, parent, id)
 | |
| 		self.list = SuperListCtrl(self, id)
 | |
| 		self.list.InsertColumn(0, 'Subnet', wx.LIST_FORMAT_RIGHT)
 | |
| 		self.list.InsertColumn(1, 'Weight', wx.LIST_FORMAT_RIGHT)
 | |
| 		self.list.InsertColumn(2, 'Owner')
 | |
| 		hbox = wx.BoxSizer(wx.HORIZONTAL)
 | |
| 		hbox.Add(self.list, 1, wx.EXPAND)
 | |
| 		self.SetSizer(hbox)
 | |
| 		self.refresh()
 | |
| 
 | |
| 	def refresh(self):
 | |
| 		self.list.itemDataMap = {}
 | |
| 		i = 0
 | |
| 
 | |
| 		for key, subnet in vpn.subnets.items():
 | |
| 			if self.list.GetItemCount() <= i:
 | |
| 				self.list.InsertStringItem(i, subnet.address + '/' + subnet.prefixlen)
 | |
| 			else:
 | |
| 				self.list.SetStringItem(i, 0, subnet.address + '/' + subnet.prefixlen)
 | |
| 			self.list.SetStringItem(i, 1, subnet.weight)
 | |
| 			self.list.SetStringItem(i, 2, subnet.owner)
 | |
| 			self.list.itemDataMap[i] = (subnet.address + '/' + subnet.prefixlen, subnet.weight, subnet.owner)
 | |
| 			i = i + 1
 | |
| 
 | |
| 		while self.list.GetItemCount() > i:
 | |
| 			self.list.DeleteItem(self.list.GetItemCount() - 1)
 | |
| 
 | |
| class StatusPage(wx.Panel):
 | |
| 	def __init__(self, parent, id):
 | |
| 		wx.Panel.__init__(self, parent, id)
 | |
| 
 | |
| class GraphPage(wx.Window):
 | |
| 	def __init__(self, parent, id):
 | |
| 		wx.Window.__init__(self, parent, id)
 | |
| 
 | |
| class NetPage(wx.Notebook):
 | |
| 	def __init__(self, parent, id):
 | |
| 		wx.Notebook.__init__(self, parent)
 | |
| 		self.settings = SettingsPage(self, id)
 | |
| 		self.connections = ConnectionsPage(self, id)
 | |
| 		self.nodes = NodesPage(self, id)
 | |
| 		self.edges = EdgesPage(self, id)
 | |
| 		self.subnets = SubnetsPage(self, id)
 | |
| 		self.graph = GraphPage(self, id)
 | |
| 		self.status = StatusPage(self, id)
 | |
| 
 | |
| 		self.AddPage(self.settings, 'Settings')
 | |
| 		#self.AddPage(self.status, 'Status')
 | |
| 		self.AddPage(self.connections, 'Connections')
 | |
| 		self.AddPage(self.nodes, 'Nodes')
 | |
| 		self.AddPage(self.edges, 'Edges')
 | |
| 		self.AddPage(self.subnets, 'Subnets')
 | |
| 		#self.AddPage(self.graph, 'Graph')
 | |
| 		
 | |
| 
 | |
| class MainWindow(wx.Frame):
 | |
| 	def OnQuit(self, event):
 | |
| 		app.ExitMainLoop()
 | |
| 
 | |
| 	def OnTimer(self, event):
 | |
| 		vpn.refresh()
 | |
| 		self.np.nodes.refresh()
 | |
| 		self.np.subnets.refresh()
 | |
| 		self.np.edges.refresh()
 | |
| 		self.np.connections.refresh()
 | |
| 
 | |
| 	def __init__(self, parent, id, title):
 | |
| 		wx.Frame.__init__(self, parent, id, title)
 | |
| 
 | |
| 		menubar = wx.MenuBar()
 | |
| 		file = wx.Menu()
 | |
| 		file.Append(1, '&Quit\tCtrl-X', 'Quit tinc GUI')
 | |
| 		menubar.Append(file, '&File')
 | |
| 
 | |
| 		#nb = wx.Notebook(self, -1)
 | |
| 		#nb.SetPadding((0, 0))
 | |
| 		self.np = NetPage(self, -1)
 | |
| 		#nb.AddPage(np, 'VPN')
 | |
| 		
 | |
| 		self.timer = wx.Timer(self, -1)
 | |
| 		self.Bind(wx.EVT_TIMER, self.OnTimer, self.timer)
 | |
| 		self.timer.Start(1000)
 | |
| 		self.Bind(wx.EVT_MENU, self.OnQuit, id=1)
 | |
| 		self.SetMenuBar(menubar)
 | |
| 		self.Show()
 | |
| 
 | |
| app = wx.App()
 | |
| mw = MainWindow(None, -1, 'Tinc GUI')
 | |
| 
 | |
| #def OnTaskBarIcon(event):
 | |
| #	mw.Raise()
 | |
| #
 | |
| #icon = wx.Icon("tincgui.ico", wx.BITMAP_TYPE_PNG)
 | |
| #taskbaricon = wx.TaskBarIcon()
 | |
| #taskbaricon.SetIcon(icon, 'Tinc GUI')
 | |
| #wx.EVT_TASKBAR_RIGHT_UP(taskbaricon, OnTaskBarIcon)
 | |
| 
 | |
| app.MainLoop()
 | |
| vpn.close()
 |