Exemple #1
0
    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
Exemple #2
0
    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
Exemple #3
0
    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'])
Exemple #4
0
    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'])