1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
import pytest
from nio import EncryptionError
from nio.crypto import (
InboundSession,
OlmAccount,
OutboundGroupSession,
OutboundSession,
Session,
)
BOB_ID = "@bob:example.org"
BOB_DEVICE = "AGMTSWVYML"
BOB_CURVE = "T9tOKF+TShsn6mk1zisW2IBsBbTtzDNvw99RBFMJOgI"
BOB_ONETIME = "6QlQw3mGUveS735k/JDaviuoaih5eEi6S1J65iHjfgU"
TEST_ROOM = "!test:example.org"
class TestClass:
def test_account(self):
account = OlmAccount()
assert (
account.identity_keys
== OlmAccount.from_pickle(account.pickle()).identity_keys
)
def test_session(self):
account = OlmAccount()
session = OutboundSession(account, BOB_CURVE, BOB_ONETIME)
assert (
session.id
== Session.from_pickle(session.pickle(), session.creation_time).id
)
assert not session.expired
def test_olm_session_encryption(self):
alice = OlmAccount()
bob = OlmAccount()
plaintext = "It's a secret to everybody"
bob_curve = bob.identity_keys["curve25519"]
bob.generate_one_time_keys(1)
bob_onetime = list(bob.one_time_keys["curve25519"].values())[0]
session = OutboundSession(alice, bob_curve, bob_onetime)
creation_time = session.use_time
# Encrypt a message and check that the use time increased.
message = session.encrypt(plaintext)
assert session.use_time >= creation_time
inbound = InboundSession(bob, message)
creation_time = inbound.use_time
# Decrypt a message and check that the use time increased.
decrypted_plaintext = inbound.decrypt(message)
assert inbound.use_time >= creation_time
assert decrypted_plaintext == plaintext
pickle = inbound.pickle("")
unpickled = Session.from_pickle(
pickle, inbound.creation_time, "", inbound.use_time
)
use_time = unpickled.use_time
message = unpickled.encrypt(plaintext)
assert unpickled.use_time > use_time
pickle = session.pickle("")
unpickled = Session.from_pickle(
pickle, session.creation_time, "", session.use_time
)
use_time = unpickled.use_time
decrypted_plaintext = unpickled.decrypt(message)
assert unpickled.use_time >= use_time
assert decrypted_plaintext == plaintext
def test_outbound_group_session(self):
session = OutboundGroupSession()
assert not session.expired
assert not session.shared
assert session.message_count == 0
with pytest.raises(EncryptionError):
session.encrypt("Hello")
session.mark_as_shared()
assert session.shared
session.encrypt("Hello")
assert session.message_count == 1
session.message_count = 101
assert session.expired
with pytest.raises(EncryptionError):
session.encrypt("Hello")
|
