def test_group_large_message():
sender_address = ProtocolAddress("+14159999111", DEVICE_ID)
group_sender = SenderKeyName("pizza party", sender_address)
alice_identity_key_pair = IdentityKeyPair.generate()
alice_registration_id = 1
alice_store = InMemSignalProtocolStore(alice_identity_key_pair,
alice_registration_id)
bob_identity_key_pair = IdentityKeyPair.generate()
bob_registration_id = 2
bob_store = InMemSignalProtocolStore(bob_identity_key_pair,
bob_registration_id)
sent_distribution_message = create_sender_key_distribution_message(
group_sender, alice_store)
recv_distribution_message = SenderKeyDistributionMessage.try_from(
sent_distribution_message.serialized())
large_message = bytes(1024)
alice_ciphertext = group_encrypt(alice_store, group_sender, large_message)
process_sender_key_distribution_message(group_sender,
recv_distribution_message,
bob_store)
bob_plaintext = group_decrypt(alice_ciphertext, bob_store, group_sender)
assert bob_plaintext == large_message
def test_group_too_far_in_the_future():
sender_address = ProtocolAddress("+14159999111", DEVICE_ID)
group_sender = SenderKeyName("pizza party", sender_address)
alice_identity_key_pair = IdentityKeyPair.generate()
alice_registration_id = 1
alice_store = InMemSignalProtocolStore(alice_identity_key_pair,
alice_registration_id)
bob_identity_key_pair = IdentityKeyPair.generate()
bob_registration_id = 2
bob_store = InMemSignalProtocolStore(bob_identity_key_pair,
bob_registration_id)
sent_distribution_message = create_sender_key_distribution_message(
group_sender, alice_store)
recv_distribution_message = SenderKeyDistributionMessage.try_from(
sent_distribution_message.serialized())
process_sender_key_distribution_message(group_sender,
recv_distribution_message,
bob_store)
for i in range(2001):
group_encrypt(alice_store, group_sender,
f"this is message {i}".encode("utf8"))
alice_ciphertext = group_encrypt(alice_store, group_sender,
"hello????".encode("utf8"))
with pytest.raises(SignalProtocolException,
match="message from too far into the future"):
assert group_decrypt(alice_ciphertext, bob_store, group_sender)
def test_group_out_of_order():
sender_address = ProtocolAddress("+14159999111", DEVICE_ID)
group_sender = SenderKeyName("pizza party", sender_address)
alice_identity_key_pair = IdentityKeyPair.generate()
alice_registration_id = 1
alice_store = InMemSignalProtocolStore(alice_identity_key_pair,
alice_registration_id)
bob_identity_key_pair = IdentityKeyPair.generate()
bob_registration_id = 2
bob_store = InMemSignalProtocolStore(bob_identity_key_pair,
bob_registration_id)
sent_distribution_message = create_sender_key_distribution_message(
group_sender, alice_store)
recv_distribution_message = SenderKeyDistributionMessage.try_from(
sent_distribution_message.serialized())
process_sender_key_distribution_message(group_sender,
recv_distribution_message,
bob_store)
ooo_ciphertexts = []
for i in range(100):
ooo_ciphertexts.append(
group_encrypt(alice_store, group_sender, f"{i}".encode("utf8")))
random.shuffle(ooo_ciphertexts)
plaintexts = []
for ciphertext in ooo_ciphertexts:
plaintexts.append(group_decrypt(ciphertext, bob_store, group_sender))
plaintexts.sort(key=lambda x: int(x))
for i, plaintext in enumerate(plaintexts):
assert plaintext.decode("utf8") == f"{i}"
def test_group_late_join():
sender_address = ProtocolAddress("+14159999111", DEVICE_ID)
group_sender = SenderKeyName("pizza party", sender_address)
alice_identity_key_pair = IdentityKeyPair.generate()
alice_registration_id = 1
alice_store = InMemSignalProtocolStore(alice_identity_key_pair,
alice_registration_id)
bob_identity_key_pair = IdentityKeyPair.generate()
bob_registration_id = 2
bob_store = InMemSignalProtocolStore(bob_identity_key_pair,
bob_registration_id)
sent_distribution_message = create_sender_key_distribution_message(
group_sender, alice_store)
recv_distribution_message = SenderKeyDistributionMessage.try_from(
sent_distribution_message.serialized())
for i in range(100):
group_encrypt(alice_store, group_sender,
f"message {i}/100".encode("utf8"))
process_sender_key_distribution_message(group_sender,
recv_distribution_message,
bob_store)
alice_ciphertext = group_encrypt(alice_store, group_sender,
"welcome Bob!".encode("utf8"))
bob_plaintext = group_decrypt(alice_ciphertext, bob_store, group_sender)
assert bob_plaintext.decode("utf8") == "welcome Bob!"
def test_group_no_send_session():
sender_address = ProtocolAddress("+14159999111", DEVICE_ID)
group_sender = SenderKeyName("pizza party", sender_address)
alice_identity_key_pair = IdentityKeyPair.generate()
alice_registration_id = 1
alice_store = InMemSignalProtocolStore(alice_identity_key_pair,
alice_registration_id)
with pytest.raises(SignalProtocolException, match="invalid send key id"):
group_encrypt(alice_store, group_sender, "hello".encode("utf8"))
def test_group_basic_ratchet():
sender_address = ProtocolAddress("+14159999111", DEVICE_ID)
group_sender = SenderKeyName("pizza party", sender_address)
alice_identity_key_pair = IdentityKeyPair.generate()
alice_registration_id = 1
alice_store = InMemSignalProtocolStore(alice_identity_key_pair,
alice_registration_id)
bob_identity_key_pair = IdentityKeyPair.generate()
bob_registration_id = 2
bob_store = InMemSignalProtocolStore(bob_identity_key_pair,
bob_registration_id)
sent_distribution_message = create_sender_key_distribution_message(
group_sender, alice_store)
recv_distribution_message = SenderKeyDistributionMessage.try_from(
sent_distribution_message.serialized())
process_sender_key_distribution_message(group_sender,
recv_distribution_message,
bob_store)
alice_ciphertext_1 = group_encrypt(alice_store, group_sender,
"message 1".encode("utf8"))
alice_ciphertext_2 = group_encrypt(alice_store, group_sender,
"message 2".encode("utf8"))
alice_ciphertext_3 = group_encrypt(alice_store, group_sender,
"message 3".encode("utf8"))
bob_plaintext1 = group_decrypt(alice_ciphertext_1, bob_store, group_sender)
assert bob_plaintext1.decode("utf8") == "message 1"
with pytest.raises(SignalProtocolException,
match="message with old counter 1 / 0"):
bob_plaintext1 = group_decrypt(alice_ciphertext_1, bob_store,
group_sender)
bob_plaintext2 = group_decrypt(alice_ciphertext_2, bob_store, group_sender)
assert bob_plaintext2.decode("utf8") == "message 2"
bob_plaintext3 = group_decrypt(alice_ciphertext_3, bob_store, group_sender)
assert bob_plaintext3.decode("utf8") == "message 3"
def test_group_no_recv_session():
sender_address = ProtocolAddress("+14159999111", DEVICE_ID)
group_sender = SenderKeyName("pizza party", sender_address)
alice_identity_key_pair = IdentityKeyPair.generate()
alice_registration_id = 1
alice_store = InMemSignalProtocolStore(alice_identity_key_pair,
alice_registration_id)
bob_identity_key_pair = IdentityKeyPair.generate()
bob_registration_id = 2
bob_store = InMemSignalProtocolStore(bob_identity_key_pair,
bob_registration_id)
sent_distribution_message = create_sender_key_distribution_message(
group_sender, alice_store)
recv_distribution_message = SenderKeyDistributionMessage.try_from(
sent_distribution_message.serialized())
alice_ciphertext = group_encrypt(alice_store, group_sender,
"hello".encode("utf8"))
with pytest.raises(SignalProtocolException, match="invalid send key id"):
bob_plaintext = group_decrypt(alice_ciphertext, bob_store,
group_sender)
def test_group_message_key_limit():
sender_address = ProtocolAddress("+14159999111", DEVICE_ID)
group_sender = SenderKeyName("pizza party", sender_address)
alice_identity_key_pair = IdentityKeyPair.generate()
alice_registration_id = 1
alice_store = InMemSignalProtocolStore(alice_identity_key_pair,
alice_registration_id)
bob_identity_key_pair = IdentityKeyPair.generate()
bob_registration_id = 2
bob_store = InMemSignalProtocolStore(bob_identity_key_pair,
bob_registration_id)
sent_distribution_message = create_sender_key_distribution_message(
group_sender, alice_store)
recv_distribution_message = SenderKeyDistributionMessage.try_from(
sent_distribution_message.serialized())
process_sender_key_distribution_message(group_sender,
recv_distribution_message,
bob_store)
ciphertexts = []
for i in range(2010):
ciphertexts.append(
group_encrypt(alice_store, group_sender,
"too many msg".encode("utf8")))
assert (group_decrypt(ciphertexts[1000], bob_store,
group_sender).decode("utf8") == "too many msg")
assert (group_decrypt(ciphertexts[len(ciphertexts) - 1], bob_store,
group_sender).decode("utf8") == "too many msg")
with pytest.raises(SignalProtocolException,
match="message with old counter"):
assert group_decrypt(ciphertexts[0], bob_store,
group_sender).decode("utf8")