From bf75c7a25ff1e7175d235bc5c330aada42fbba6b Mon Sep 17 00:00:00 2001 From: Jon Bergli Heier Date: Wed, 1 Mar 2017 19:19:49 +0100 Subject: Removed custom ipv6 code. twisted now has built-in ipv6 support; get rid of the ipv6 code as we don't need it anymore. --- fot.py | 8 +--- ipv6.py | 127 ---------------------------------------------------------------- 2 files changed, 2 insertions(+), 133 deletions(-) delete mode 100644 ipv6.py diff --git a/fot.py b/fot.py index 674b042..7c48583 100755 --- a/fot.py +++ b/fot.py @@ -10,7 +10,7 @@ import os, re, platform, sys, traceback from ConfigParser import ConfigParser -import login, ipv6 +import login config = ConfigParser() @@ -239,11 +239,7 @@ def start_server(server): print '%s' % server factory = BotFactory(server, config.get(server, 'nickname')) - if config.has_option(server, 'host6'): - bind6 = (config.get(server, 'bind6'), 0) if config.has_option(server, 'bind6') else None - ipv6.connectTCP6(config.get(server, 'host6'), config.getint(server, 'port6'), factory, bindAddress = bind6) - else: - reactor.connectTCP(config.get(server, 'host'), config.getint(server, 'port'), factory) + reactor.connectTCP(config.get(server, 'host'), config.getint(server, 'port'), factory, bindAddress = (config.get(server, 'bind'), 0)) print 'Starting per-network instances...' for server in (server for server in config.sections() if server.startswith('server/')): diff --git a/ipv6.py b/ipv6.py deleted file mode 100644 index 9dd1f32..0000000 --- a/ipv6.py +++ /dev/null @@ -1,127 +0,0 @@ - -import socket -from twisted.internet import tcp -from twisted.internet import default -from twisted.internet import protocol -from twisted.internet import reactor - -from twisted.names.client import Resolver -resolver = Resolver('/etc/resolv.conf') - -class IPv6Address(object): - def __init__(self, type, host, port, flowInfo, scope): - self.type = type - self.host = host - self.port = port - self.flowInfo = flowInfo - self.scope = scope - - def __eq__(self, other): - if isinstance(other, IPv6Address): - a = (self.type, self.host, self.port, self.flowInfo, self.scope) - b = (other.type, other.host, other.port, other.flowInfo, other.scope) - return a == b - return False - - def __str__(self): - return 'IPv6Address(%s, %r, %d, %d, %d)' % ( - self.type, self.host, self.port, self.flowInfo, self.scope) - -def isIPv6Address(ip): - try: - socket.inet_pton(socket.AF_INET6, ip) - except: - return 0 - return 1 - -class Client(tcp.Client): - addressFamily = socket.AF_INET6 - - def resolveAddress(self): - if isIPv6Address(self.addr[0]): - self._setRealAddress(self.addr[0]) - else: - resolver.lookupIPV6Address(self.addr[0]).addCallbacks( - self.resolvedAddress, self.failIfNotConnected - ) - - def resolvedAddress(self, addr): - aaaas = [x for x in addr[0] if x.type == 28] # 28 = AAAA record - if not len(aaaas): - self.failIfNotConnected(Exception('Unable to resolve IPv6 address.')) - return - aaaa = aaaas[0].payload - self._setRealAddress(aaaa._address) - - def getHost(self): - return IPv6Address('TCP', *self.socket.getsockname()) - - def getPeer(self): - return IPv6Address('TCP', *self.socket.getpeername()) - - -class Connector(tcp.Connector): - def _makeTransport(self): - return Client(self.host, self.port, self.bindAddress, self, self.reactor) - - def getDestination(self): - return IPv6Address('TCP', self.host, self.port) - -class Server(tcp.Server): - def getHost(self): - return IPv6Address('TCP', *self.socket.getsockname()) - - def getPeer(self): - return IPv6Address('TCP', *self.client) - -class Port(tcp.Port): - addressFamily = socket.AF_INET6 - - transport = Server - - def _buildAddr(self, address): - return IPv6Address('TCP', *address) - - def getHost(self): - return IPv6Address('TCP', *self.socket.getsockname()) - - def getPeer(self): - return IPv6Address('TCP', *self.socket.getpeername()) - -def connectTCP6(host, port, factory, timeout=30, bindAddress=None, reactor=None): - if reactor is None: - from twisted.internet import reactor - return reactor.connectWith( - Connector, host, port, factory, timeout, bindAddress - ) - - -def listenTCP6(port, factory, backlog=5, interface='::', reactor=None): - if reactor is None: - from twisted.internet import reactor - return reactor.listenWith(Port, port, factory, backlog, interface) - -def main(): - from twisted.internet import reactor - - class TrivialProtocol(protocol.Protocol): - def connectionMade(self): - print 'I (', self.transport.getHost(), ') am connected! (to ', self.transport.getPeer(), ')' - self.transport.write('Hello, world!\n') - - def dataReceived(self, data): - print 'Received: ' + repr(data) - - class TrivialServerFactory(protocol.ServerFactory): - protocol = TrivialProtocol - class TrivialClientFactory(protocol.ClientFactory): - protocol = TrivialProtocol - - p = listenTCP6(6666, TrivialServerFactory()) - c = connectTCP6('::1', 6666, TrivialClientFactory()) - - reactor.run() - -if __name__ == '__main__': - main() - -- cgit v1.2.3