def _client_respond(self, client, their_ts):
l = protocol.PTP(data=[])
l.data = []
t = protocol.TLV(type=protocol.PTP_TYPE_SERVERVER, data=protocol.UInt(size=1, data=PTP_SERVERVER))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_SEQUENCE, data=protocol.UInt(size=4, data=self.server_seq))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_UUID, data=protocol.String(data=client['uuid']))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_YOURTS, data=protocol.UInt(size=8, data=their_ts))
l.data.append(t)
packet = l.pack()
if len(packet) > protocol.PTP_MTU: # bad
self.ui.log("Ignoring attempt to send ts %d bytes to client %s. MTU is %d" % \
(len(packet), str(client['sin']), protocol.PTP_MTU))
return
if self.args.debug:
self.ui.log("Sending ts %d bytes to client %s" % (len(packet), str(client['sin'])))
self.ui.log("%s" % repr(protocol.PTP(packet)), indent=' ')
if self.args.hexdump:
self.ui.log(hexdump.hexdump(result='return', data=packet))
self.sock.sendto(packet, client['sin'])
self.server_seq += 1L
def _client_beacons(self):
with self._clock:
for k in self.clients:
client = self.clients[k]
if not 'myseq' in client or not 'sin' in client: continue
if 'uuid' in client and client['uuid'] == self.uuid:
continue # self!
l = protocol.PTP(data=[])
l.data = []
t = protocol.TLV(type=protocol.PTP_TYPE_CLIENTVER,
data=protocol.UInt(size=1,
data=PTP_CLIENTVER))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_SEQUENCE,
data=protocol.UInt(size=4,
data=client['myseq']))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_UUID,
data=protocol.String(data=self.uuid))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_MYTS,
data=protocol.UInt(size=8,
data=int(time.time() *
2**32)))
l.data.append(t)
packet = l.pack()
if len(packet) > protocol.PTP_MTU: # bad
self.ui.log("Ignoring attempt to send ts %d bytes to client %s. MTU is %d" % \
(len(packet), str(client['sin']), protocol.PTP_MTU))
return
if self.args.debug:
self.ui.log("Sending ts %d bytes to client %s" %
(len(packet), str(client['sin'])))
self.ui.log("%s" % repr(protocol.PTP(packet)), indent=' ')
if self.args.hexdump:
self.ui.log(
hexdump.hexdump(result='return', data=packet))
self.sock.sendto(packet, client['sin'])
client['stats']['sent'] += 1
client['myseq'] += 1
self.ui.peer_update('client', client['sin'], client['stats'])
def _client_beacon(self, k, client):
# We need a list of TLVs and then form a PTP fron them
l = protocol.PTP(data=[])
l.data = []
t = protocol.TLV(type=protocol.PTP_TYPE_SERVERVER, data=protocol.UInt(size=1, data=PTP_SERVERVER))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_SEQUENCE, data=protocol.UInt(size=4, data=self.server_seq))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_UUID, data=protocol.String(data=client['uuid']))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_MYTS, data=protocol.UInt(size=8, data=int(time.time()*2**32)))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_YOURADDR, data=protocol.Address(data=client['sin']))
l.data.append(t)
# Now add the list of known clients
count = 0
for sk in self.clients:
if sk == k: continue # skip the client we're sending this to
sc = self.clients[sk]
t = protocol.TLV(type=protocol.PTP_TYPE_CLIENTLIST_EXT,
data=protocol.Address(data=sc['sin']))
l.data.append(t)
count += 1
# If this client address matches the address we're sending this packet to then we
# also send its local address - the clients can then try to talk locally
if sc['sin'][0] == client['sin'][0]:
t = protocol.TLV(type=protocol.PTP_TYPE_CLIENTLIST_INT,
data=protocol.Address(data=sc['ptpaddr']))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_CLIENTLEN, data=protocol.UInt(size=1, data=count))
l.data.append(t)
packet = l.pack()
if len(packet) > protocol.PTP_MTU: # bad
self.ui.log("Ignoring attempt to send %d bytes to client %s. MTU is %d" % \
(len(packet), str(client['sin']), protocol.PTP_MTU))
return
if self.args.debug:
self.ui.log("Sending %d bytes to client %s" % (len(packet), str(client['sin'])))
self.ui.log("%s" % repr(protocol.PTP(packet)), indent=' ')
if self.args.hexdump:
self.ui.log(hexdump.hexdump(result='return', data=packet))
self.sock.sendto(packet, client['sin'])
client['stats']['sent'] += 1
self.server_seq += 1L
self.ui.peer_update('client', client['sin'], client['stats'])
def _server_beacons(self, shutdown=False):
# Tell the server about ourself
l = protocol.PTP(data=[])
l.data = []
t = protocol.TLV(type=protocol.PTP_TYPE_CLIENTVER,
data=protocol.UInt(size=1, data=PTP_CLIENTVER))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_SEQUENCE,
data=protocol.UInt(size=4, data=self.server_seq))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_UUID,
data=protocol.String(data=self.uuid))
l.data.append(t)
t = protocol.TLV(type=protocol.PTP_TYPE_PTPADDR,
data=protocol.Address(data=(self.addr, self.port)))
l.data.append(t)
if shutdown:
t = protocol.TLV(type=protocol.PTP_TYPE_SHUTDOWN,
data=protocol.UInt(size=1, data=1))
l.data.append(t)
else:
t = protocol.TLV(type=protocol.PTP_TYPE_MYTS,
data=protocol.UInt(size=8,
data=int(time.time() * 2**32)))
l.data.append(t)
tmp = copy.deepcopy(self.servers)
for k in tmp:
if 'uuid' in tmp[k]:
del (tmp[k]['uuid'])
t = protocol.TLV(type=protocol.PTP_TYPE_META,
data=protocol.JSON(data=tmp))
l.data.append(t)
packet = l.pack()
if len(packet) > protocol.PTP_MTU: # bad
self.ui.log(
"Ignoring attempt to send %d bytes to servers. MTU is %d" %
(len(packet), protocol.PTP_MTU))
return
if self.args.debug:
self.ui.log("Sending %d bytes to servers:" % len(packet))
self.ui.log("%s" % repr(protocol.PTP(packet)), indent=' ')
if self.args.hexdump:
self.ui.log(hexdump.hexdump(result='return', data=packet))
with self._slock:
for k in self.servers:
server = self.servers[k]
server['stats']['sent'] += 1
self.sock.sendto(packet, server['sin'])
self.ui.peer_update('server', server['sin'], server['stats'])
self.server_seq += 1L