123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620 |
- #!/usr/bin/python
- # tinc-gui -- GUI for controlling a running tincd
- # Copyright (C) 2009-2014 Guus Sliepen <guus@tinc-vpn.org>
- # 2014 Dennis Joachimsthaler <dennis@efjot.de>
- #
- # 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):
- # read the pidfile
- f = open(self.pidfile)
- info = string.split(f.readline())
- f.close()
- # 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.");
- if ':' in info[2]:
- af = socket.AF_INET6
- else:
- af = socket.AF_INET
- s = socket.socket(af, 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':
- sam = _winreg.KEY_READ
- if platform.machine().endswith('64'):
- sam = sam | _winreg.KEY_WOW64_64KEY
- try:
- reg = _winreg.ConnectRegistry(None, _winreg.HKEY_LOCAL_MACHINE)
- try:
- key = _winreg.OpenKey(reg, "SOFTWARE\\tinc", 0, sam)
- except WindowsError:
- key = _winreg.OpenKey(reg, "SOFTWARE\\Wow6432Node\\tinc", 0, sam)
- 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, 16)
- 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(1, 1)
- namelabel = wx.StaticText(self, -1, 'Name:')
- self.name = wx.TextCtrl(self, -1, vpn.name)
- grid.Add(namelabel)
- grid.Add(self.name, 1, wx.EXPAND)
- 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 = SuperListCtrl(self, id)
- 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):
- sortstate = self.list.GetSortState()
- 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)
- self.list.SetItemData(i, i)
- i += 1
- while self.list.GetItemCount() > i:
- self.list.DeleteItem(self.list.GetItemCount() - 1)
- self.list.SortListItems(sortstate[0], sortstate[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):
- sortstate = self.list.GetSortState()
- 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, since)
- self.list.SetItemData(i, i)
- i += 1
- while self.list.GetItemCount() > i:
- self.list.DeleteItem(self.list.GetItemCount() - 1)
- self.list.SortListItems(sortstate[0], sortstate[1])
- class EdgesPage(wx.Panel):
- def __init__(self, parent, id):
- wx.Panel.__init__(self, parent, id)
- self.list = SuperListCtrl(self, id)
- 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):
- sortstate = self.list.GetSortState()
- 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)
- self.list.SetItemData(i, i)
- i += 1
- while self.list.GetItemCount() > i:
- self.list.DeleteItem(self.list.GetItemCount() - 1)
- self.list.SortListItems(sortstate[0], sortstate[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):
- sortstate = self.list.GetSortState()
- 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)
- self.list.SetItemData(i, i)
- i += 1
- while self.list.GetItemCount() > i:
- self.list.DeleteItem(self.list.GetItemCount() - 1)
- self.list.SortListItems(sortstate[0], sortstate[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()
|