Exemplo n.º 1
0
	def send_session_reply(self, station, session_info):
		logger.debug("Sending session reply")
		
		stream = StreamOut(self.settings)
		stream.add(session_info)
		data = stream.get()
		
		rand_value = random.randint(0, 0xFFFFFFFF)
		
		fragments = (len(data) - 1) // 800 + 1
		for i in range(fragments):
			stream = StreamOut(self.settings)
			stream.u8(self.MESSAGE_SESSION_REPLY)
			stream.pad(11)
			stream.u32(rand_value)
			stream.u16(self.session_reply_id)
			stream.u8(i)
			stream.u8(fragments)
			
			body = data[i * 800 : (i + 1) * 800]
			stream.u32(len(body))
			stream.write(body)
			
			message = PIAMessage()
			message.protocol_id = self.get_protocol_type()
			message.payload = stream.get()
			self.transport.send(station, message)
			
		self.session_reply_id += 1
Exemplo n.º 2
0
    def encode(self):
        payload = self.encrypt(self.payload)

        stream = StreamOut(self.settings)
        stream.u32(0x32AB9864)

        header_version = self.settings.get("pia.header_version")
        crypto_enabled = self.settings.get("pia.crypto_enabled")
        if header_version > 0:
            stream.u8((crypto_enabled << 7) | header_version)
        else:
            stream.u8(crypto_enabled + 1)

        stream.u8(self.connection_id)
        stream.u16(self.sequence_id)

        if header_version == 0:
            stream.u16(self.session_timer)
            stream.u16(self.rtt_timer)

        if self.settings.get(
                "pia.encryption_method") == EncryptionMethod.AES_GCM:
            stream.u64(self.nonce)
            stream.write(self.signature)

        stream.write(payload)
        stream.write(self.calc_signature(stream.get()))

        return stream.get()