2009-12-16 20:18:21 +00:00
|
|
|
#!/usr/bin/python
|
|
|
|
|
2012-07-22 10:55:04 +00:00
|
|
|
# tinc-gui -- GUI for controlling a running tincd
|
2014-02-07 19:38:48 +00:00
|
|
|
# Copyright (C) 2009-2014 Guus Sliepen <guus@tinc-vpn.org>
|
|
|
|
# 2014 Dennis Joachimsthaler <dennis@efjot.de>
|
2012-07-22 10:55:04 +00:00
|
|
|
#
|
|
|
|
# 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.
|
|
|
|
|
2011-06-25 15:39:02 +00:00
|
|
|
import string
|
|
|
|
import socket
|
2009-12-16 20:18:21 +00:00
|
|
|
import wx
|
|
|
|
import sys
|
2012-07-22 10:55:04 +00:00
|
|
|
import os
|
|
|
|
import platform
|
2012-09-27 20:12:15 +00:00
|
|
|
import time
|
2009-12-16 20:18:21 +00:00
|
|
|
from wx.lib.mixins.listctrl import ColumnSorterMixin
|
|
|
|
from wx.lib.mixins.listctrl import ListCtrlAutoWidthMixin
|
|
|
|
|
2014-01-17 15:10:10 +00:00
|
|
|
if platform.system() == 'Windows':
|
2012-07-22 10:55:04 +00:00
|
|
|
import _winreg
|
|
|
|
|
2011-06-25 15:39:02 +00:00
|
|
|
# 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]
|
2012-09-27 20:12:15 +00:00
|
|
|
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])
|
2011-06-25 15:39:02 +00:00
|
|
|
|
|
|
|
self.subnets = {}
|
|
|
|
|
|
|
|
class Edge:
|
|
|
|
def parse(self, args):
|
|
|
|
self.fr = args[0]
|
2012-09-27 20:12:15 +00:00
|
|
|
self.to = args[1]
|
|
|
|
self.address = args[2]
|
|
|
|
self.port = args[4]
|
|
|
|
self.options = int(args[5], 16)
|
|
|
|
self.weight = int(args[6])
|
2011-06-25 15:39:02 +00:00
|
|
|
|
|
|
|
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'
|
|
|
|
|
2012-09-27 20:12:15 +00:00
|
|
|
self.owner = args[1]
|
2011-06-25 15:39:02 +00:00
|
|
|
|
|
|
|
class Connection:
|
|
|
|
def parse(self, args):
|
|
|
|
self.name = args[0]
|
2012-09-27 20:12:15 +00:00
|
|
|
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)
|
2011-06-25 15:39:02 +00:00
|
|
|
self.weight = 123
|
|
|
|
|
|
|
|
class VPN:
|
|
|
|
confdir = '/etc/tinc'
|
2011-06-26 10:51:25 +00:00
|
|
|
piddir = '/var/run/'
|
2011-06-25 15:39:02 +00:00
|
|
|
|
|
|
|
def connect(self):
|
2013-02-07 14:27:16 +00:00
|
|
|
# read the pidfile
|
2011-06-26 10:51:25 +00:00
|
|
|
f = open(self.pidfile)
|
|
|
|
info = string.split(f.readline())
|
2011-06-25 15:39:02 +00:00
|
|
|
f.close()
|
2013-02-07 14:27:16 +00:00
|
|
|
|
|
|
|
# check if there is a UNIX socket as well
|
|
|
|
if self.pidfile.endswith(".pid"):
|
|
|
|
unixfile = self.pidfile.replace(".pid", ".socket");
|
|
|
|
else:
|
|
|
|
unixfile = self.pidfile + ".socket";
|
|
|
|
|
|
|
|
if os.path.exists(unixfile):
|
|
|
|
# use it if it exists
|
|
|
|
print(unixfile + " exists!");
|
|
|
|
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
|
|
s.connect(unixfile)
|
|
|
|
else:
|
|
|
|
# otherwise connect via TCP
|
|
|
|
print(unixfile + " does not exist.");
|
2014-01-19 20:15:23 +00:00
|
|
|
if ':' in info[2]:
|
|
|
|
af = socket.AF_INET6
|
|
|
|
else:
|
|
|
|
af = socket.AF_INET
|
|
|
|
s = socket.socket(af, socket.SOCK_STREAM)
|
2013-02-07 14:27:16 +00:00
|
|
|
s.connect((info[2], int(info[4])))
|
|
|
|
|
2011-06-25 15:39:02 +00:00
|
|
|
self.sf = s.makefile()
|
|
|
|
s.close()
|
|
|
|
hello = string.split(self.sf.readline())
|
|
|
|
self.name = hello[1]
|
2011-06-26 10:51:25 +00:00
|
|
|
self.sf.write('0 ^' + info[1] + ' 17\r\n')
|
2011-06-25 15:39:02 +00:00
|
|
|
self.sf.flush()
|
|
|
|
resp = string.split(self.sf.readline())
|
2011-06-26 10:51:25 +00:00
|
|
|
self.port = info[4]
|
2011-06-25 15:39:02 +00:00
|
|
|
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':
|
2012-09-27 20:12:15 +00:00
|
|
|
if len(resp) < 19:
|
2011-06-25 15:39:02 +00:00
|
|
|
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':
|
2012-09-27 20:12:15 +00:00
|
|
|
if len(resp) < 9:
|
2011-06-25 15:39:02 +00:00
|
|
|
continue
|
2012-09-27 20:12:15 +00:00
|
|
|
edge = self.nodes.get((resp[2], resp[3])) or Edge()
|
2011-06-25 15:39:02 +00:00
|
|
|
edge.parse(resp[2:])
|
|
|
|
edge.visited = True
|
2012-09-27 20:12:15 +00:00
|
|
|
self.edges[(resp[2], resp[3])] = edge
|
2011-06-25 15:39:02 +00:00
|
|
|
elif resp[1] == '5':
|
2012-09-27 20:12:15 +00:00
|
|
|
if len(resp) < 4:
|
2011-06-25 15:39:02 +00:00
|
|
|
continue
|
2012-09-27 20:12:15 +00:00
|
|
|
subnet = self.subnets.get((resp[2], resp[3])) or Subnet()
|
2011-06-25 15:39:02 +00:00
|
|
|
subnet.parse(resp[2:])
|
|
|
|
subnet.visited = True
|
2012-09-27 20:12:15 +00:00
|
|
|
self.subnets[(resp[2], resp[3])] = subnet
|
2011-06-25 15:39:02 +00:00
|
|
|
self.nodes[subnet.owner].subnets[resp[2]] = subnet
|
|
|
|
elif resp[1] == '6':
|
2012-09-27 20:12:15 +00:00
|
|
|
if len(resp) < 9:
|
2011-06-25 15:39:02 +00:00
|
|
|
break
|
2012-09-27 20:12:15 +00:00
|
|
|
connection = self.connections.get((resp[2], resp[3], resp[5])) or Connection()
|
2011-06-25 15:39:02 +00:00
|
|
|
connection.parse(resp[2:])
|
|
|
|
connection.visited = True
|
2012-09-27 20:12:15 +00:00
|
|
|
self.connections[(resp[2], resp[3], resp[5])] = connection
|
2011-06-25 15:39:02 +00:00
|
|
|
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])
|
|
|
|
|
2011-06-26 10:51:25 +00:00
|
|
|
def __init__(self, netname = None, pidfile = None):
|
2014-01-17 15:10:10 +00:00
|
|
|
if platform.system() == 'Windows':
|
2014-01-17 17:15:40 +00:00
|
|
|
sam = _winreg.KEY_READ
|
|
|
|
if platform.machine().endswith('64'):
|
|
|
|
sam = sam | _winreg.KEY_WOW64_64KEY
|
2012-07-22 10:55:04 +00:00
|
|
|
try:
|
2014-01-17 15:10:10 +00:00
|
|
|
reg = _winreg.ConnectRegistry(None, _winreg.HKEY_LOCAL_MACHINE)
|
2014-01-17 17:15:40 +00:00
|
|
|
try:
|
|
|
|
key = _winreg.OpenKey(reg, "SOFTWARE\\tinc", 0, sam)
|
|
|
|
except WindowsError:
|
|
|
|
key = _winreg.OpenKey(reg, "SOFTWARE\\Wow6432Node\\tinc", 0, sam)
|
2012-07-22 10:55:04 +00:00
|
|
|
VPN.confdir = _winreg.QueryValue(key, None)
|
|
|
|
except WindowsError:
|
|
|
|
pass
|
2011-06-25 15:39:02 +00:00
|
|
|
|
|
|
|
if netname:
|
|
|
|
self.netname = netname
|
2012-07-22 10:55:04 +00:00
|
|
|
self.confbase = os.path.join(VPN.confdir, netname)
|
|
|
|
else:
|
|
|
|
self.confbase = VPN.confdir
|
2011-06-25 15:39:02 +00:00
|
|
|
|
2012-07-22 10:55:04 +00:00
|
|
|
self.tincconf = os.path.join(self.confbase, 'tinc.conf')
|
2011-06-25 15:39:02 +00:00
|
|
|
|
2012-07-22 10:55:04 +00:00
|
|
|
if pidfile != None:
|
2011-06-26 10:51:25 +00:00
|
|
|
self.pidfile = pidfile
|
2011-06-25 15:39:02 +00:00
|
|
|
else:
|
2014-01-17 15:10:10 +00:00
|
|
|
if platform.system() == 'Windows':
|
2012-07-22 10:55:04 +00:00
|
|
|
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')
|
2011-06-25 15:39:02 +00:00
|
|
|
|
|
|
|
# GUI starts here
|
|
|
|
|
2011-06-26 10:51:25 +00:00
|
|
|
argv0 = sys.argv[0]
|
2009-12-16 20:18:21 +00:00
|
|
|
del sys.argv[0]
|
2012-07-22 10:55:04 +00:00
|
|
|
netname = None
|
2011-06-26 10:51:25 +00:00
|
|
|
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)
|
|
|
|
|
2012-07-22 10:55:04 +00:00
|
|
|
while sys.argv:
|
2009-12-16 20:18:21 +00:00
|
|
|
if sys.argv[0] in ('-n', '--net'):
|
2011-06-26 10:51:25 +00:00
|
|
|
del sys.argv[0]
|
2012-07-22 10:55:04 +00:00
|
|
|
netname = sys.argv[0]
|
2011-06-26 10:51:25 +00:00
|
|
|
elif sys.argv[0] in ('--pidfile'):
|
|
|
|
del sys.argv[0]
|
|
|
|
pidfile = sys.argv[0]
|
|
|
|
elif sys.argv[0] in ('--help'):
|
|
|
|
usage(0)
|
2009-12-16 20:18:21 +00:00
|
|
|
else:
|
2011-06-26 10:51:25 +00:00
|
|
|
print(argv0 + ': unrecognized option \'' + sys.argv[0] + '\'')
|
|
|
|
usage(1)
|
2009-12-16 20:18:21 +00:00
|
|
|
|
|
|
|
del sys.argv[0]
|
|
|
|
|
2012-07-22 10:55:04 +00:00
|
|
|
if netname == None:
|
|
|
|
netname = os.getenv("NETNAME")
|
|
|
|
|
|
|
|
if netname == ".":
|
|
|
|
netname = None
|
|
|
|
|
|
|
|
vpn = VPN(netname, pidfile)
|
2009-12-16 20:18:21 +00:00
|
|
|
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)
|
2012-09-28 15:03:14 +00:00
|
|
|
ColumnSorterMixin.__init__(self, 16)
|
2009-12-16 20:18:21 +00:00
|
|
|
|
|
|
|
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)
|
2012-09-28 15:03:14 +00:00
|
|
|
grid.AddGrowableCol(1, 1)
|
2009-12-16 20:18:21 +00:00
|
|
|
|
|
|
|
namelabel = wx.StaticText(self, -1, 'Name:')
|
|
|
|
self.name = wx.TextCtrl(self, -1, vpn.name)
|
|
|
|
grid.Add(namelabel)
|
2012-09-28 15:03:14 +00:00
|
|
|
grid.Add(self.name, 1, wx.EXPAND)
|
2009-12-16 20:18:21 +00:00
|
|
|
|
|
|
|
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)
|
2012-09-28 15:03:14 +00:00
|
|
|
self.list = SuperListCtrl(self, id)
|
2009-12-16 20:18:21 +00:00
|
|
|
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):
|
2012-09-28 15:03:14 +00:00
|
|
|
sortstate = self.list.GetSortState()
|
2009-12-16 20:18:21 +00:00
|
|
|
self.list.itemDataMap = {}
|
|
|
|
i = 0
|
|
|
|
|
|
|
|
for key, connection in vpn.connections.items():
|
2011-05-17 08:57:30 +00:00
|
|
|
if self.list.GetItemCount() <= i:
|
2009-12-16 20:18:21 +00:00
|
|
|
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)
|
2012-09-28 15:03:14 +00:00
|
|
|
self.list.SetItemData(i, i)
|
2009-12-16 20:18:21 +00:00
|
|
|
i += 1
|
|
|
|
|
2011-05-17 08:57:30 +00:00
|
|
|
while self.list.GetItemCount() > i:
|
|
|
|
self.list.DeleteItem(self.list.GetItemCount() - 1)
|
2009-12-16 20:18:21 +00:00
|
|
|
|
2012-09-28 15:03:14 +00:00
|
|
|
self.list.SortListItems(sortstate[0], sortstate[1])
|
2009-12-16 20:18:21 +00:00
|
|
|
|
|
|
|
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')
|
2012-09-27 20:12:15 +00:00
|
|
|
self.list.InsertColumn(15, 'Since')
|
2009-12-16 20:18:21 +00:00
|
|
|
|
|
|
|
hbox = wx.BoxSizer(wx.HORIZONTAL)
|
|
|
|
hbox.Add(self.list, 1, wx.EXPAND)
|
|
|
|
self.SetSizer(hbox)
|
|
|
|
self.refresh()
|
|
|
|
|
|
|
|
def refresh(self):
|
2012-09-28 15:03:14 +00:00
|
|
|
sortstate = self.list.GetSortState()
|
2009-12-16 20:18:21 +00:00
|
|
|
self.list.itemDataMap = {}
|
|
|
|
i = 0
|
|
|
|
|
|
|
|
for key, node in vpn.nodes.items():
|
2011-05-17 08:57:30 +00:00
|
|
|
if self.list.GetItemCount() <= i:
|
2009-12-16 20:18:21 +00:00
|
|
|
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))
|
2012-09-27 20:12:15 +00:00
|
|
|
self.list.SetStringItem(i, 7, format(node.options, "x"))
|
|
|
|
self.list.SetStringItem(i, 8, format(node.status, "04x"))
|
2009-12-16 20:18:21 +00:00
|
|
|
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))
|
2012-09-27 20:12:15 +00:00
|
|
|
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)
|
2012-09-28 15:03:14 +00:00
|
|
|
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, since)
|
2009-12-16 20:18:21 +00:00
|
|
|
self.list.SetItemData(i, i)
|
|
|
|
i += 1
|
|
|
|
|
2011-05-17 08:57:30 +00:00
|
|
|
while self.list.GetItemCount() > i:
|
|
|
|
self.list.DeleteItem(self.list.GetItemCount() - 1)
|
2009-12-16 20:18:21 +00:00
|
|
|
|
2012-09-28 15:03:14 +00:00
|
|
|
self.list.SortListItems(sortstate[0], sortstate[1])
|
|
|
|
|
2009-12-16 20:18:21 +00:00
|
|
|
class EdgesPage(wx.Panel):
|
|
|
|
def __init__(self, parent, id):
|
|
|
|
wx.Panel.__init__(self, parent, id)
|
2012-09-28 15:03:14 +00:00
|
|
|
self.list = SuperListCtrl(self, id)
|
2009-12-16 20:18:21 +00:00
|
|
|
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):
|
2012-09-28 15:03:14 +00:00
|
|
|
sortstate = self.list.GetSortState()
|
2009-12-16 20:18:21 +00:00
|
|
|
self.list.itemDataMap = {}
|
|
|
|
i = 0
|
|
|
|
|
|
|
|
for key, edge in vpn.edges.items():
|
2011-05-17 08:57:30 +00:00
|
|
|
if self.list.GetItemCount() <= i:
|
2009-12-16 20:18:21 +00:00
|
|
|
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)
|
2012-09-27 20:12:15 +00:00
|
|
|
self.list.SetStringItem(i, 4, format(edge.options, "x"))
|
2009-12-16 20:18:21 +00:00
|
|
|
self.list.SetStringItem(i, 5, str(edge.weight))
|
|
|
|
self.list.itemDataMap[i] = (edge.fr, edge.to, edge.address, edge.port, edge.options, edge.weight)
|
2012-09-28 15:03:14 +00:00
|
|
|
self.list.SetItemData(i, i)
|
2009-12-16 20:18:21 +00:00
|
|
|
i += 1
|
|
|
|
|
2011-05-17 08:57:30 +00:00
|
|
|
while self.list.GetItemCount() > i:
|
|
|
|
self.list.DeleteItem(self.list.GetItemCount() - 1)
|
2009-12-16 20:18:21 +00:00
|
|
|
|
2012-09-28 15:03:14 +00:00
|
|
|
self.list.SortListItems(sortstate[0], sortstate[1])
|
|
|
|
|
2009-12-16 20:18:21 +00:00
|
|
|
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):
|
2012-09-28 15:03:14 +00:00
|
|
|
sortstate = self.list.GetSortState()
|
2009-12-16 20:18:21 +00:00
|
|
|
self.list.itemDataMap = {}
|
|
|
|
i = 0
|
|
|
|
|
|
|
|
for key, subnet in vpn.subnets.items():
|
2011-05-17 08:57:30 +00:00
|
|
|
if self.list.GetItemCount() <= i:
|
2009-12-16 20:18:21 +00:00
|
|
|
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)
|
2012-09-28 15:03:14 +00:00
|
|
|
self.list.SetItemData(i, i)
|
|
|
|
i += 1
|
2009-12-16 20:18:21 +00:00
|
|
|
|
2011-05-17 08:57:30 +00:00
|
|
|
while self.list.GetItemCount() > i:
|
|
|
|
self.list.DeleteItem(self.list.GetItemCount() - 1)
|
2009-12-16 20:18:21 +00:00
|
|
|
|
2012-09-28 15:03:14 +00:00
|
|
|
self.list.SortListItems(sortstate[0], sortstate[1])
|
|
|
|
|
2009-12-16 20:18:21 +00:00
|
|
|
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):
|
2012-07-22 10:55:04 +00:00
|
|
|
app.ExitMainLoop()
|
2009-12-16 20:18:21 +00:00
|
|
|
|
|
|
|
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()
|