tinc/gui/Tinc.py

#!/usr/bin/python

import string
import socket

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 __init__(self):
		print('New node')

	def __exit__(self):
		print('Deleting node ' + self.name)

	def parse(self, args):
		self.name = args[0]
		self.address = args[2]
		if args[3] != 'port':
			args.insert(3, 'port')
			args.insert(4, '')
		self.port = args[4]
		self.cipher = int(args[6])
		self.digest = int(args[8])
		self.maclength = int(args[10])
		self.compression = int(args[12])
		self.options = int(args[14], 0x10)
		self.status = int(args[16], 0x10)
		self.nexthop = args[18]
		self.via = args[20]
		self.distance = int(args[22])
		self.pmtu = int(args[24])
		self.minmtu = int(args[26])
		self.maxmtu = int(args[28][:-1])

		self.subnets = {}

class Edge:
	def parse(self, args):
		self.fr = args[0]
		self.to = args[2]
		self.address = args[4]
		self.port = args[6]
		self.options = int(args[8])
		self.weight = int(args[10])

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[2]	

class Connection:
	def parse(self, args):
		self.name = args[0]
		self.address = args[2]
		if args[3] != 'port':
			args.insert(3, 'port')
			args.insert(4, '')
		self.port = args[4]
		self.options = int(args[6], 0x10)
		self.socket = int(args[8])
		self.status = int(args[10], 0x10)
		self.weight = 123

class VPN:
	confdir = '/etc/tinc'
	cookiedir = '/var/run/'

	def connect(self):
		f = open(self.cookiefile)
		cookie = string.split(f.readline())
		f.close()
		s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
		s.connect(('127.0.0.1', int(cookie[1])))
		self.sf = s.makefile()
		s.close()
		hello = string.split(self.sf.readline())
		self.name = hello[1]
		self.sf.write('0 ^' + cookie[0] + ' 17\r\n')
		self.sf.flush()
		resp = string.split(self.sf.readline())
		self.port = cookie[1]
		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) < 3:
					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) < 5:
					continue
				edge = self.nodes.get((resp[2], resp[4])) or Edge()
				edge.parse(resp[2:])
				edge.visited = True
				self.edges[(resp[2], resp[4])] = edge
			elif resp[1] == '5':
				if len(resp) < 5:
					continue
				subnet = self.subnets.get((resp[2], resp[4])) or Subnet()
				subnet.parse(resp[2:])
				subnet.visited = True
				self.subnets[(resp[2], resp[4])] = subnet
				self.nodes[subnet.owner].subnets[resp[2]] = subnet
			elif resp[1] == '6':
				if len(resp) < 5:
					break
				connection = self.connections.get((resp[2], resp[4])) or Connection()
				connection.parse(resp[2:])
				connection.visited = True
				self.connections[(resp[2], resp[4])] = 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, controlcookie = None):
		self.tincconf = VPN.confdir + '/'

		if netname:
			self.netname = netname
			self.tincconf += netname + '/'

		self.tincconf += 'tinc.conf'

		if controlcookie is not None:
			self.cookiefile = controlcookie
		else:
			self.cookiefile = VPN.cookiedir + 'tinc.'
			if netname:
				self.cookiefile += netname + '.'
			self.cookiefile += 'cookie'