def sendKeys(self, fresh=True, countPreKeys=_COUNT_PREKEYS):
identityKeyPair = KeyHelper.generateIdentityKeyPair(
) if fresh else self.store.getIdentityKeyPair()
registrationId = KeyHelper.generateRegistrationId(
) if fresh else self.store.getLocalRegistrationId()
preKeys = KeyHelper.generatePreKeys(KeyHelper.getRandomSequence(),
countPreKeys)
signedPreKey = KeyHelper.generateSignedPreKey(
identityKeyPair, KeyHelper.getRandomSequence(65536))
preKeysDict = {}
for preKey in preKeys:
keyPair = preKey.getKeyPair()
preKeysDict[self.adjustId(preKey.getId())] = self.adjustArray(
keyPair.getPublicKey().serialize()[1:])
signedKeyTuple = (
self.adjustId(signedPreKey.getId()),
self.adjustArray(
signedPreKey.getKeyPair().getPublicKey().serialize()[1:]),
self.adjustArray(signedPreKey.getSignature()))
setKeysIq = SetKeysIqProtocolEntity(
self.adjustArray(identityKeyPair.getPublicKey().serialize()[1:]),
signedKeyTuple, preKeysDict, Curve.DJB_TYPE,
self.adjustId(registrationId))
onResult = lambda _, __: self.persistKeys(
registrationId, identityKeyPair, preKeys, signedPreKey, fresh)
self._sendIq(setKeysIq, onResult, self.onSentKeysError)
def _generate_axolotl_keys(self):
identityKeyPair = KeyHelper.generateIdentityKeyPair()
registrationId = KeyHelper.generateRegistrationId()
preKeys = KeyHelper.generatePreKeys(KeyHelper.getRandomSequence(),
DEFAULT_PREKEY_AMOUNT)
self.storeLocalData(registrationId, identityKeyPair)
signedPreKey = KeyHelper.generateSignedPreKey(
identityKeyPair, KeyHelper.getRandomSequence(65536))
self.storeSignedPreKey(signedPreKey.getId(), signedPreKey)
for preKey in preKeys:
self.storePreKey(preKey.getId(), preKey)
def _generate_axolotl_keys(self):
identityKeyPair = KeyHelper.generateIdentityKeyPair()
registrationId = KeyHelper.generateRegistrationId()
preKeys = KeyHelper.generatePreKeys(KeyHelper.getRandomSequence(),
DEFAULT_PREKEY_AMOUNT)
self.storeLocalData(registrationId, identityKeyPair)
signedPreKey = KeyHelper.generateSignedPreKey(
identityKeyPair, KeyHelper.getRandomSequence(65536))
self.storeSignedPreKey(signedPreKey.getId(), signedPreKey)
for preKey in preKeys:
self.storePreKey(preKey.getId(), preKey)
def init_store(self):
"""Create new identity key pair and initialize database"""
logger.info('Creating new identity...')
identityKeyPair = KeyHelper.generateIdentityKeyPair()
registrationId = KeyHelper.generateRegistrationId()
preKeys = KeyHelper.generatePreKeys(KeyHelper.getRandomSequence(), self.COUNT_PREKEYS)
signedPreKey = KeyHelper.generateSignedPreKey(identityKeyPair,
KeyHelper.getRandomSequence(65536))
self.store.storeLocalData(registrationId, identityKeyPair)
self.store.storeSignedPreKey(signedPreKey.getId(), signedPreKey)
for preKey in preKeys:
self.store.storePreKey(preKey.getId(), preKey)
def init_store(self):
"""Create new identity key pair and initialize database"""
logger.info('Creating new identity...')
identityKeyPair = KeyHelper.generateIdentityKeyPair()
registrationId = KeyHelper.generateRegistrationId()
preKeys = KeyHelper.generatePreKeys(KeyHelper.getRandomSequence(),
self.COUNT_PREKEYS)
signedPreKey = KeyHelper.generateSignedPreKey(
identityKeyPair, KeyHelper.getRandomSequence(65536))
self.store.storeLocalData(registrationId, identityKeyPair)
self.store.storeSignedPreKey(signedPreKey.getId(), signedPreKey)
for preKey in preKeys:
self.store.storePreKey(preKey.getId(), preKey)
def bundle(self):
prekeys = [(k.getId(),
b64encode(k.getKeyPair().getPublicKey().serialize()))
for k in self.store.loadPreKeys()]
identityKeyPair = self.store.getIdentityKeyPair()
signedPreKey = KeyHelper.generateSignedPreKey(
identityKeyPair, KeyHelper.getRandomSequence(65536))
self.store.storeSignedPreKey(signedPreKey.getId(), signedPreKey)
result = {
'signedPreKeyId':
signedPreKey.getId(),
'signedPreKeyPublic':
b64encode(signedPreKey.getKeyPair().getPublicKey().serialize()),
'signedPreKeySignature':
b64encode(signedPreKey.getSignature()),
'identityKey':
b64encode(identityKeyPair.getPublicKey().serialize()),
'prekeys':
prekeys
}
return result
def sendKeys(self, fresh = True, countPreKeys = _COUNT_PREKEYS):
identityKeyPair = KeyHelper.generateIdentityKeyPair() if fresh else self.store.getIdentityKeyPair()
registrationId = KeyHelper.generateRegistrationId() if fresh else self.store.getLocalRegistrationId()
preKeys = KeyHelper.generatePreKeys(KeyHelper.getRandomSequence(), countPreKeys)
signedPreKey = KeyHelper.generateSignedPreKey(identityKeyPair, KeyHelper.getRandomSequence(65536))
preKeysDict = {}
for preKey in preKeys:
keyPair = preKey.getKeyPair()
preKeysDict[self.adjustId(preKey.getId())] = self.adjustArray(keyPair.getPublicKey().serialize()[1:])
signedKeyTuple = (self.adjustId(signedPreKey.getId()),
self.adjustArray(signedPreKey.getKeyPair().getPublicKey().serialize()[1:]),
self.adjustArray(signedPreKey.getSignature()))
setKeysIq = SetKeysIqProtocolEntity(self.adjustArray(identityKeyPair.getPublicKey().serialize()[1:]), signedKeyTuple, preKeysDict, Curve.DJB_TYPE, self.adjustId(registrationId))
onResult = lambda _, __: self.persistKeys(registrationId, identityKeyPair, preKeys, signedPreKey, fresh)
self._sendIq(setKeysIq, onResult, self.onSentKeysError)
def processInitKeyExchangeMessage(self):
try:
sequence = KeyHelper.getRandomSequence(65534) + 1
flags = KeyExchangeMessage.INITIATE_FLAG
baseKey = Curve.generateKeyPair()
ratchetKey = Curve.generateKeyPair()
identityKey = self.identityKeyStore.getIdentityKeyPair()
baseKeySignature = Curve.calculateSignature(identityKey.getPrivateKey(), baseKey.getPublicKey().serialize())
sessionRecord = self.sessionStore.loadSession(self.recipientId, self.deviceId)
sessionRecord.getSessionState().setPendingKeyExchange(sequence, baseKey, ratchetKey, identityKey)
self.sessionStore.storeSession(self.recipientId, self.deviceId, sessionRecord)
return KeyExchangeMessage(2, sequence, flags, baseKey.getPublicKey(), baseKeySignature,
ratchetKey.getPublicKey(), identityKey.getPublicKey())
except InvalidKeyException as e:
raise AssertionError(e)
def level_prekeys(self, force=False):
logger.debug("level_prekeys(force=%s)" % force)
pending_prekeys = self._store.loadPreKeys()
logger.debug("len(pending_prekeys) = %d" % len(pending_prekeys))
if force or len(pending_prekeys) < self.THRESHOLD_REGEN:
count_gen = self.COUNT_GEN_PREKEYS - len(pending_prekeys)
logger.info("Generating %d prekeys" % count_gen)
## arbitrary, should keep track of generated prekey ids and create from there
prekeys = KeyHelper.generatePreKeys(KeyHelper.getRandomSequence(2**32 // 2), count_gen)
logger.info("Storing %d prekeys" % len(prekeys))
for i in range(0, len(prekeys)):
key = prekeys[i]
if logger.level <= logging.DEBUG:
sys.stdout.write("Storing prekey %d/%d \r" % (i + 1, len(prekeys)))
sys.stdout.flush()
self._store.storePreKey(key.getId(), key)
return prekeys
return []
def bundle(self):
"""
.. highlight: python
Returns all data needed to announce bundle information.
::
bundle_dict = {
'signedPreKeyPublic': bytes,
'prekeys': [(int, bytes) (int, bytes)],
'identityKey': bytes,
'signedPreKeyId': int,
'signedPreKeySignature': bytes
}
"""
prekeys = [(k.getId(),
b64encode(k.getKeyPair().getPublicKey().serialize()))
for k in self.store.loadPreKeys()]
identityKeyPair = self.store.getIdentityKeyPair()
signedPreKey = KeyHelper.generateSignedPreKey(
identityKeyPair, KeyHelper.getRandomSequence(65536))
self.store.storeSignedPreKey(signedPreKey.getId(), signedPreKey)
result = {
'signedPreKeyId':
signedPreKey.getId(),
'signedPreKeyPublic':
b64encode(signedPreKey.getKeyPair().getPublicKey().serialize()),
'signedPreKeySignature':
b64encode(signedPreKey.getSignature()),
'identityKey':
b64encode(identityKeyPair.getPublicKey().serialize()),
'prekeys':
prekeys
}
return result
def bundle(self):
prekeys = [
(k.getId(), b64encode(k.getKeyPair().getPublicKey().serialize()))
for k in self.store.loadPreKeys()
]
identityKeyPair = self.store.getIdentityKeyPair()
signedPreKey = KeyHelper.generateSignedPreKey(
identityKeyPair, KeyHelper.getRandomSequence(65536))
self.store.storeSignedPreKey(signedPreKey.getId(), signedPreKey)
result = {
'signedPreKeyId': signedPreKey.getId(),
'signedPreKeyPublic':
b64encode(signedPreKey.getKeyPair().getPublicKey().serialize()),
'signedPreKeySignature': b64encode(signedPreKey.getSignature()),
'identityKey':
b64encode(identityKeyPair.getPublicKey().serialize()),
'prekeys': prekeys
}
return result
def bundle(self):
"""
.. highlight: python
Returns all data needed to announce bundle information.
::
bundle_dict = {
'signedPreKeyPublic': bytes,
'prekeys': [(int, bytes) (int, bytes)],
'identityKey': bytes,
'signedPreKeyId': int,
'signedPreKeySignature': bytes
}
"""
prekeys = [
(k.getId(), b64encode(k.getKeyPair().getPublicKey().serialize()))
for k in self.store.loadPreKeys()
]
identityKeyPair = self.store.getIdentityKeyPair()
signedPreKey = KeyHelper.generateSignedPreKey(
identityKeyPair, KeyHelper.getRandomSequence(65536))
self.store.storeSignedPreKey(signedPreKey.getId(), signedPreKey)
result = {
'signedPreKeyId': signedPreKey.getId(),
'signedPreKeyPublic':
b64encode(signedPreKey.getKeyPair().getPublicKey().serialize()),
'signedPreKeySignature': b64encode(signedPreKey.getSignature()),
'identityKey':
b64encode(identityKeyPair.getPublicKey().serialize()),
'prekeys': prekeys
}
return result
def test_outOfOrder(self):
aliceStore = InMemorySenderKeyStore();
bobStore = InMemorySenderKeyStore();
aliceSessionBuilder = GroupSessionBuilder(aliceStore)
bobSessionBuilder = GroupSessionBuilder(bobStore)
aliceGroupCipher = GroupCipher(aliceStore, GROUP_SENDER)
bobGroupCipher = GroupCipher(bobStore, GROUP_SENDER);
sentAliceDistributionMessage = aliceSessionBuilder.create(GROUP_SENDER);
receivedAliceDistributionMessage = SenderKeyDistributionMessage(serialized = sentAliceDistributionMessage.serialize());
bobSessionBuilder.process(GROUP_SENDER, receivedAliceDistributionMessage)
ciphertexts = []
for i in range(0, 100):
ciphertexts.append(aliceGroupCipher.encrypt("up the punks"))
while len(ciphertexts) > 0:
index = KeyHelper.getRandomSequence(2147483647) % len(ciphertexts)
ciphertext = ciphertexts.pop(index)
plaintext = bobGroupCipher.decrypt(ciphertext)
self.assertEqual(plaintext, "up the punks")
