def listen(self):
self.log.info("Listening at: %s:%s", self.hostname, self.port)
if network_util.is_loopback_addr(self.hostname):
# we are in local test mode so bind that socket on the
# specified IP
self.log.info("PeerListener.socket.bind('%s') LOOPBACK", self.hostname)
self._prepare_datagram_socket()
elif '[' in self.hostname:
self.log.info("PeerListener.socket.bind('tcp://[*]:%s') IPV6", self.port)
self.socket.ipv6 = True
self._prepare_datagram_socket(socket.AF_INET6)
else:
self.log.info("PeerListener.socket.bind('tcp://*:%s') IPV4", self.port)
# Temporary while I fix things
self.hostname = '0.0.0.0'
self._prepare_datagram_socket()
self.is_listening = True
def start_listening():
while self.is_listening:
try:
data, addr = self.socket.recvfrom(PEERLISTENER_RECV_FROM_BUFFER_SIZE)
self.log.debug('Got data from %s:%d: %s', addr[0], addr[1], data[:50])
count_incoming_packet(data)
if data[:MSG_PING_ID_SIZE] == MSG_PING_ID:
self.socket.sendto('pong', (addr[0], addr[1]))
count_outgoing_packet('pong')
elif data[:MSG_PONG_ID_SIZE] == MSG_PONG_ID:
self.event_emitter.emit('on_pong_message', (data, addr))
elif data[:MSG_SEND_RELAY_PING_ID_SIZE] == MSG_SEND_RELAY_PING_ID:
self.event_emitter.emit('on_send_relay_ping', (data, addr))
elif data[:MSG_RELAY_PING_ID_SIZE] == MSG_RELAY_PING_ID:
data = data.split(' ')
sender = self.guid
recipient = data[1]
self.socket.sendto('send_relay_pong %s %s' % (sender, recipient), (addr[0], addr[1]))
count_outgoing_packet('send_relay_pong %s %s' % (sender, recipient))
elif data[:MSG_SEND_RELAY_PONG_ID_SIZE] == MSG_SEND_RELAY_PONG_ID:
self.event_emitter.emit('on_send_relay_pong', (data, addr))
elif data[:MSG_HEARTBEAT_ID_SIZE] == MSG_HEARTBEAT_ID:
self.log.debug('We just received a heartbeat.')
elif data[:MSG_RELAYTO_ID_SIZE] == MSG_RELAYTO_ID:
self.log.debug('Relay To Packet')
self.event_emitter.emit('on_relayto', data)
elif data[:MSG_RELAY_ID_SIZE] == MSG_RELAY_ID:
self.log.debug('Relay Packet')
self.event_emitter.emit('on_message', (data, addr))
else:
self.event_emitter.emit('on_message', (data, addr))
except socket.timeout as exc:
err = exc.args[0]
if err == 'timed out':
time.sleep(0.5)
continue
else:
sys.exit(1)
except socket.error:
# No data. This is normal.
pass
# except AttributeError as err:
# print 'Packet was jacked up: %s', err
Thread(target=start_listening).start()
def test_is_loopback_addr(self):
self.assertTrue(network_util.is_loopback_addr("127.0.0.1"))
self.assertTrue(network_util.is_loopback_addr("localhost"))
self.assertFalse(network_util.is_loopback_addr("10.0.0.1"))
self.assertFalse(network_util.is_loopback_addr("192.168.0.1"))
def test_is_loopback_addr(self):
self.assertTrue(network_util.is_loopback_addr("127.0.0.1"))
self.assertTrue(network_util.is_loopback_addr("localhost"))
self.assertFalse(network_util.is_loopback_addr("10.0.0.1"))
self.assertFalse(network_util.is_loopback_addr("192.168.0.1"))
def listen(self):
self.log.info("Listening at: %s:%s", self.hostname, self.port)
if network_util.is_loopback_addr(self.hostname):
# we are in local test mode so bind that socket on the
# specified IP
self.log.info("PeerListener.socket.bind('%s') LOOPBACK",
self.hostname)
self._prepare_datagram_socket()
elif '[' in self.hostname:
self.log.info("PeerListener.socket.bind('tcp://[*]:%s') IPV6",
self.port)
self.socket.ipv6 = True
self._prepare_datagram_socket(socket.AF_INET6)
else:
self.log.info("PeerListener.socket.bind('tcp://*:%s') IPV4",
self.port)
# Temporary while I fix things
self.hostname = '0.0.0.0'
self._prepare_datagram_socket()
self.is_listening = True
def start_listening():
while self.is_listening:
try:
data, addr = self.socket.recvfrom(2048)
self.log.debug('Got data from %s:%d: %s', addr[0], addr[1],
data[:50])
count_incoming_packet(data)
if data[:4] == 'ping':
self.socket.sendto('pong', (addr[0], addr[1]))
count_outgoing_packet('pong')
elif data[:4] == 'pong':
self.ee.emit('on_pong_message', (data, addr))
elif data[:15] == 'send_relay_ping':
self.ee.emit('on_send_relay_ping', (data, addr))
elif data[:10] == 'relay_ping':
data = data.split(' ')
sender = self.guid
recipient = data[1]
self.socket.sendto(
'send_relay_pong %s %s' % (sender, recipient),
(addr[0], addr[1]))
count_outgoing_packet('send_relay_pong %s %s' %
(sender, recipient))
elif data[:15] == 'send_relay_pong':
self.ee.emit('on_send_relay_pong', (data, addr))
elif data[:9] == 'heartbeat':
self.log.debug('We just received a heartbeat.')
elif data[:7] == 'relayto':
self.log.debug('Relay To Packet')
self.ee.emit('on_relayto', data)
elif data[:6] == 'relay ':
self.log.debug('Relay Packet')
self.ee.emit('on_message', (data, addr))
else:
self.ee.emit('on_message', (data, addr))
except socket.timeout as e:
err = e.args[0]
if err == 'timed out':
time.sleep(0.5)
continue
else:
sys.exit(1)
except socket.error:
# No data. This is normal.
pass
# except AttributeError as err:
# print 'Packet was jacked up: %s', err
Thread(target=start_listening).start()
