70 lines
2.1 KiB
Dart
70 lines
2.1 KiB
Dart
|
import 'dart:convert';
|
||
|
|
import 'package:omemo_dart/omemo_dart.dart';
|
||
|
|
import 'package:test/test.dart';
|
||
|
|
|
||
|
|
void main() {
|
||
|
|
test('Test using OMEMO sessions with only one device per user', () async {
|
||
|
|
const aliceJid = 'alice@server.example';
|
||
|
|
const bobJid = 'bob@other.server.example';
|
||
|
|
|
||
|
|
// Alice and Bob generate their sessions
|
||
|
|
final aliceSession = await OmemoSessionManager.generateNewIdentity(opkAmount: 1);
|
||
|
|
final bobSession = await OmemoSessionManager.generateNewIdentity(opkAmount: 1);
|
||
|
|
|
||
|
|
// Perform the X3DH
|
||
|
|
final x3dhAliceResult = await x3dhFromBundle(
|
||
|
|
await bobSession.device.toBundle(),
|
||
|
|
aliceSession.device.ik,
|
||
|
|
);
|
||
|
|
final x3dhBobResult = await x3dhFromInitialMessage(
|
||
|
|
X3DHMessage(
|
||
|
|
aliceSession.device.ik.pk,
|
||
|
|
x3dhAliceResult.ek.pk,
|
||
|
|
'2',
|
||
|
|
),
|
||
|
|
bobSession.device.spk,
|
||
|
|
bobSession.device.opks.values.elementAt(0),
|
||
|
|
bobSession.device.ik,
|
||
|
|
);
|
||
|
|
|
||
|
|
// Build the ratchets
|
||
|
|
final aliceRatchet = await OmemoDoubleRatchet.initiateNewSession(
|
||
|
|
bobSession.device.spk.pk,
|
||
|
|
x3dhAliceResult.sk,
|
||
|
|
x3dhAliceResult.ad,
|
||
|
|
);
|
||
|
|
final bobRatchet = await OmemoDoubleRatchet.acceptNewSession(
|
||
|
|
bobSession.device.spk,
|
||
|
|
x3dhBobResult.sk,
|
||
|
|
x3dhBobResult.ad,
|
||
|
|
);
|
||
|
|
|
||
|
|
// Add the ratchets to the session managers
|
||
|
|
await aliceSession.addSession(bobJid, bobSession.device.id, aliceRatchet);
|
||
|
|
await bobSession.addSession(aliceJid, aliceSession.device.id, bobRatchet);
|
||
|
|
|
||
|
|
// Alice encrypts a message for Bob
|
||
|
|
const messagePlaintext = 'Hello Bob!';
|
||
|
|
final aliceMessage = await aliceSession.encryptToJid(bobJid, messagePlaintext);
|
||
|
|
expect(aliceMessage.encryptedKeys.length, 1);
|
||
|
|
|
||
|
|
// Alice sends the message to Bob
|
||
|
|
// ...
|
||
|
|
|
||
|
|
// Bob decrypts it
|
||
|
|
final bobMessage = await bobSession.decryptMessage(
|
||
|
|
aliceMessage.ciphertext,
|
||
|
|
aliceJid,
|
||
|
|
aliceSession.device.id,
|
||
|
|
[
|
||
|
|
EncryptedKey(
|
||
|
|
bobSession.device.id,
|
||
|
|
base64.encode(aliceMessage.encryptedKeys[bobSession.device.id]!),
|
||
|
|
),
|
||
|
|
],
|
||
|
|
);
|
||
|
|
|
||
|
|
expect(messagePlaintext, bobMessage);
|
||
|
|
});
|
||
|
|
}
|