Index: third_party/twisted_8_1/twisted/internet/iocpreactor/udp.py |
diff --git a/third_party/twisted_8_1/twisted/internet/iocpreactor/udp.py b/third_party/twisted_8_1/twisted/internet/iocpreactor/udp.py |
deleted file mode 100644 |
index c0e1a8ee702eb98a2b590fbb6629e8b3c5c2fc89..0000000000000000000000000000000000000000 |
--- a/third_party/twisted_8_1/twisted/internet/iocpreactor/udp.py |
+++ /dev/null |
@@ -1,389 +0,0 @@ |
-# Copyright (c) 2008 Twisted Matrix Laboratories. |
-# See LICENSE for details. |
- |
- |
-""" |
-UDP support for IOCP reactor |
-""" |
- |
-from twisted.internet import defer, address, error, interfaces |
-from twisted.internet.abstract import isIPAddress |
-from twisted.python import log, reflect, failure |
- |
-from zope.interface import implements |
-import socket, operator, struct, warnings, errno |
- |
-from twisted.internet.iocpreactor.const import ERROR_IO_PENDING |
-from twisted.internet.iocpreactor.const import ERROR_CONNECTION_REFUSED |
-from twisted.internet.iocpreactor.const import ERROR_PORT_UNREACHABLE |
-from twisted.internet.iocpreactor.interfaces import IReadWriteHandle |
-from twisted.internet.iocpreactor import iocpsupport as _iocp, abstract |
- |
- |
- |
-class Port(abstract.FileHandle): |
- """ |
- UDP port, listening for packets. |
- """ |
- |
- implements(IReadWriteHandle, interfaces.IUDPTransport, |
- interfaces.ISystemHandle) |
- |
- addressFamily = socket.AF_INET |
- socketType = socket.SOCK_DGRAM |
- maxThroughput = 256 * 1024 # max bytes we read in one eventloop iteration |
- dynamicReadBuffers = False |
- |
- # Actual port number being listened on, only set to a non-None |
- # value when we are actually listening. |
- _realPortNumber = None |
- |
- |
- def __init__(self, port, proto, interface='', maxPacketSize=8192, |
- reactor=None): |
- """ |
- Initialize with a numeric port to listen on. |
- """ |
- self.port = port |
- self.protocol = proto |
- self.readBufferSize = maxPacketSize |
- self.interface = interface |
- self.setLogStr() |
- self._connectedAddr = None |
- |
- abstract.FileHandle.__init__(self, reactor) |
- |
- skt = socket.socket(self.addressFamily, self.socketType) |
- addrLen = _iocp.maxAddrLen(skt.fileno()) |
- self.addressBuffer = _iocp.AllocateReadBuffer(addrLen) |
- |
- |
- def __repr__(self): |
- if self._realPortNumber is not None: |
- return ("<%s on %s>" % |
- (self.protocol.__class__, self._realPortNumber)) |
- else: |
- return "<%s not connected>" % (self.protocol.__class__,) |
- |
- |
- def getHandle(self): |
- """ |
- Return a socket object. |
- """ |
- return self.socket |
- |
- |
- def startListening(self): |
- """ |
- Create and bind my socket, and begin listening on it. |
- |
- This is called on unserialization, and must be called after creating a |
- server to begin listening on the specified port. |
- """ |
- self._bindSocket() |
- self._connectToProtocol() |
- |
- |
- def createSocket(self): |
- return self.reactor.createSocket(self.addressFamily, self.socketType) |
- |
- |
- def _bindSocket(self): |
- try: |
- skt = self.createSocket() |
- skt.bind((self.interface, self.port)) |
- except socket.error, le: |
- raise error.CannotListenError, (self.interface, self.port, le) |
- |
- # Make sure that if we listened on port 0, we update that to |
- # reflect what the OS actually assigned us. |
- self._realPortNumber = skt.getsockname()[1] |
- |
- log.msg("%s starting on %s" % |
- (self.protocol.__class__, self._realPortNumber)) |
- |
- self.connected = True |
- self.socket = skt |
- self.getFileHandle = self.socket.fileno |
- |
- |
- def _connectToProtocol(self): |
- self.protocol.makeConnection(self) |
- self.startReading() |
- self.reactor.addActiveHandle(self) |
- |
- |
- def cbRead(self, rc, bytes, evt): |
- if self.reading: |
- self.handleRead(rc, bytes, evt) |
- self.doRead() |
- |
- |
- def handleRead(self, rc, bytes, evt): |
- if rc in (errno.WSAECONNREFUSED, errno.WSAECONNRESET, |
- ERROR_CONNECTION_REFUSED, ERROR_PORT_UNREACHABLE): |
- if self._connectedAddr: |
- self.protocol.connectionRefused() |
- elif rc: |
- log.msg("error in recvfrom -- %s (%s)" % |
- (errno.errorcode.get(rc, 'unknown error'), rc)) |
- else: |
- try: |
- self.protocol.datagramReceived(str(evt.buff[:bytes]), |
- _iocp.makesockaddr(evt.addr_buff)) |
- except: |
- log.err() |
- |
- |
- def doRead(self): |
- read = 0 |
- while self.reading: |
- evt = _iocp.Event(self.cbRead, self) |
- |
- evt.buff = buff = self._readBuffers[0] |
- evt.addr_buff = addr_buff = self.addressBuffer |
- rc, bytes = _iocp.recvfrom(self.getFileHandle(), buff, |
- addr_buff, evt) |
- |
- if (rc == ERROR_IO_PENDING |
- or (not rc and read >= self.maxThroughput)): |
- break |
- else: |
- evt.ignore = True |
- self.handleRead(rc, bytes, evt) |
- read += bytes |
- |
- |
- def write(self, datagram, addr=None): |
- """ |
- Write a datagram. |
- |
- @param addr: should be a tuple (ip, port), can be None in connected |
- mode. |
- """ |
- if self._connectedAddr: |
- assert addr in (None, self._connectedAddr) |
- try: |
- return self.socket.send(datagram) |
- except socket.error, se: |
- no = se.args[0] |
- if no == errno.WSAEINTR: |
- return self.write(datagram) |
- elif no == errno.WSAEMSGSIZE: |
- raise error.MessageLengthError, "message too long" |
- elif no in (errno.WSAECONNREFUSED, errno.WSAECONNRESET, |
- ERROR_CONNECTION_REFUSED, ERROR_PORT_UNREACHABLE): |
- self.protocol.connectionRefused() |
- else: |
- raise |
- else: |
- assert addr != None |
- if not addr[0].replace(".", "").isdigit(): |
- warnings.warn("Please only pass IPs to write(), not hostnames", |
- DeprecationWarning, stacklevel=2) |
- try: |
- return self.socket.sendto(datagram, addr) |
- except socket.error, se: |
- no = se.args[0] |
- if no == errno.WSAEINTR: |
- return self.write(datagram, addr) |
- elif no == errno.WSAEMSGSIZE: |
- raise error.MessageLengthError, "message too long" |
- elif no in (errno.WSAECONNREFUSED, errno.WSAECONNRESET, |
- ERROR_CONNECTION_REFUSED, ERROR_PORT_UNREACHABLE): |
- # in non-connected UDP ECONNREFUSED is platform dependent, |
- # I think and the info is not necessarily useful. |
- # Nevertheless maybe we should call connectionRefused? XXX |
- return |
- else: |
- raise |
- |
- |
- def writeSequence(self, seq, addr): |
- self.write("".join(seq), addr) |
- |
- |
- def connect(self, host, port): |
- """ |
- 'Connect' to remote server. |
- """ |
- if self._connectedAddr: |
- raise RuntimeError( |
- "already connected, reconnecting is not currently supported " |
- "(talk to itamar if you want this)") |
- if not isIPAddress(host): |
- raise ValueError, "please pass only IP addresses, not domain names" |
- self._connectedAddr = (host, port) |
- self.socket.connect((host, port)) |
- |
- |
- def _loseConnection(self): |
- self.stopReading() |
- self.reactor.removeActiveHandle(self) |
- if self.connected: # actually means if we are *listening* |
- from twisted.internet import reactor |
- reactor.callLater(0, self.connectionLost) |
- |
- |
- def stopListening(self): |
- if self.connected: |
- result = self.d = defer.Deferred() |
- else: |
- result = None |
- self._loseConnection() |
- return result |
- |
- |
- def loseConnection(self): |
- warnings.warn("Please use stopListening() to disconnect port", |
- DeprecationWarning, stacklevel=2) |
- self.stopListening() |
- |
- |
- def connectionLost(self, reason=None): |
- """ |
- Cleans up my socket. |
- """ |
- log.msg('(Port %s Closed)' % self._realPortNumber) |
- self._realPortNumber = None |
- self.stopReading() |
- if hasattr(self, "protocol"): |
- # we won't have attribute in ConnectedPort, in cases |
- # where there was an error in connection process |
- self.protocol.doStop() |
- self.connected = False |
- self.disconnected = True |
- self.socket.close() |
- del self.socket |
- del self.getFileHandle |
- if hasattr(self, "d"): |
- self.d.callback(None) |
- del self.d |
- |
- |
- def setLogStr(self): |
- self.logstr = reflect.qual(self.protocol.__class__) + " (UDP)" |
- |
- |
- def logPrefix(self): |
- """ |
- Returns the name of my class, to prefix log entries with. |
- """ |
- return self.logstr |
- |
- |
- def getHost(self): |
- """ |
- Returns an IPv4Address. |
- |
- This indicates the address from which I am connecting. |
- """ |
- return address.IPv4Address('UDP', *(self.socket.getsockname() + |
- ('INET_UDP',))) |
- |
- |
- |
-class MulticastMixin: |
- """ |
- Implement multicast functionality. |
- """ |
- |
- |
- def getOutgoingInterface(self): |
- i = self.socket.getsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_IF) |
- return socket.inet_ntoa(struct.pack("@i", i)) |
- |
- |
- def setOutgoingInterface(self, addr): |
- """ |
- Returns Deferred of success. |
- """ |
- return self.reactor.resolve(addr).addCallback(self._setInterface) |
- |
- |
- def _setInterface(self, addr): |
- i = socket.inet_aton(addr) |
- self.socket.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_IF, i) |
- return 1 |
- |
- |
- def getLoopbackMode(self): |
- return self.socket.getsockopt(socket.IPPROTO_IP, |
- socket.IP_MULTICAST_LOOP) |
- |
- |
- def setLoopbackMode(self, mode): |
- mode = struct.pack("b", operator.truth(mode)) |
- self.socket.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_LOOP, |
- mode) |
- |
- |
- def getTTL(self): |
- return self.socket.getsockopt(socket.IPPROTO_IP, |
- socket.IP_MULTICAST_TTL) |
- |
- |
- def setTTL(self, ttl): |
- ttl = struct.pack("B", ttl) |
- self.socket.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, ttl) |
- |
- |
- def joinGroup(self, addr, interface=""): |
- """ |
- Join a multicast group. Returns Deferred of success. |
- """ |
- return self.reactor.resolve(addr).addCallback(self._joinAddr1, |
- interface, 1) |
- |
- |
- def _joinAddr1(self, addr, interface, join): |
- return self.reactor.resolve(interface).addCallback(self._joinAddr2, |
- addr, join) |
- |
- |
- def _joinAddr2(self, interface, addr, join): |
- addr = socket.inet_aton(addr) |
- interface = socket.inet_aton(interface) |
- if join: |
- cmd = socket.IP_ADD_MEMBERSHIP |
- else: |
- cmd = socket.IP_DROP_MEMBERSHIP |
- try: |
- self.socket.setsockopt(socket.IPPROTO_IP, cmd, addr + interface) |
- except socket.error, e: |
- return failure.Failure(error.MulticastJoinError(addr, interface, |
- *e.args)) |
- |
- |
- def leaveGroup(self, addr, interface=""): |
- """ |
- Leave multicast group, return Deferred of success. |
- """ |
- return self.reactor.resolve(addr).addCallback(self._joinAddr1, |
- interface, 0) |
- |
- |
- |
-class MulticastPort(MulticastMixin, Port): |
- """ |
- UDP Port that supports multicasting. |
- """ |
- |
- implements(interfaces.IMulticastTransport) |
- |
- |
- def __init__(self, port, proto, interface='', maxPacketSize=8192, |
- reactor=None, listenMultiple=False): |
- Port.__init__(self, port, proto, interface, maxPacketSize, reactor) |
- self.listenMultiple = listenMultiple |
- |
- |
- def createSocket(self): |
- skt = Port.createSocket(self) |
- if self.listenMultiple: |
- skt.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) |
- if hasattr(socket, "SO_REUSEPORT"): |
- skt.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1) |
- return skt |
- |
- |