80 lines
2.4 KiB
Dart
80 lines
2.4 KiB
Dart
import 'dart:convert';
|
|
import 'package:cryptography/cryptography.dart';
|
|
import 'package:omemo_dart/omemo_dart.dart';
|
|
import 'package:test/test.dart';
|
|
|
|
void main() {
|
|
test('X3DH with correct signature', () async {
|
|
// Generate keys
|
|
final ikAlice = await OmemoKeyPair.generateNewPair(KeyPairType.ed25519);
|
|
final ikBob = await OmemoKeyPair.generateNewPair(KeyPairType.ed25519);
|
|
final spkBob = await OmemoKeyPair.generateNewPair(KeyPairType.x25519);
|
|
final opkBob = await OmemoKeyPair.generateNewPair(KeyPairType.x25519);
|
|
final bundleBob = OmemoBundle(
|
|
'1',
|
|
await spkBob.pk.asBase64(),
|
|
'3',
|
|
base64Encode(
|
|
await sig(ikBob, await spkBob.pk.getBytes()),
|
|
),
|
|
//'Q5in+/L4kJixEX692h6mJkPMyp4I3SlQ84L0E7ipPzqfPHOMiraUlqG2vG/O8wvFjLsKYZpPBraga9IvwhqVDA==',
|
|
await ikBob.pk.asBase64(),
|
|
{
|
|
'2': await opkBob.pk.asBase64(),
|
|
},
|
|
);
|
|
|
|
// Alice does X3DH
|
|
final resultAlice = await x3dhFromBundle(bundleBob, ikAlice);
|
|
|
|
// Alice sends the inital message to Bob
|
|
// ...
|
|
|
|
// Bob does X3DH
|
|
final resultBob = await x3dhFromInitialMessage(
|
|
X3DHMessage(
|
|
ikAlice.pk,
|
|
resultAlice.ek.pk,
|
|
'2',
|
|
),
|
|
spkBob,
|
|
opkBob,
|
|
ikBob,
|
|
);
|
|
|
|
expect(resultAlice.sk, resultBob.sk);
|
|
expect(resultAlice.ad, resultBob.ad);
|
|
});
|
|
|
|
test('X3DH with incorrect signature', () async {
|
|
// Generate keys
|
|
final ikAlice = await OmemoKeyPair.generateNewPair(KeyPairType.ed25519);
|
|
final ikBob = await OmemoKeyPair.generateNewPair(KeyPairType.ed25519);
|
|
final spkBob = await OmemoKeyPair.generateNewPair(KeyPairType.x25519);
|
|
final opkBob = await OmemoKeyPair.generateNewPair(KeyPairType.x25519);
|
|
final bundleBob = OmemoBundle(
|
|
'1',
|
|
await spkBob.pk.asBase64(),
|
|
'3',
|
|
// NOTE: A bit flakey, but it is highly unlikely that the same keypair as this one
|
|
// gets generated.
|
|
'Q5in+/L4kJixEX692h6mJkPMyp4I3SlQ84L0E7ipPzqfPHOMiraUlqG2vG/O8wvFjLsKYZpPBraga9IvwhqVDA==',
|
|
await ikBob.pk.asBase64(),
|
|
{
|
|
'2': await opkBob.pk.asBase64(),
|
|
},
|
|
);
|
|
|
|
// Alice does X3DH
|
|
var exception = false;
|
|
try {
|
|
await x3dhFromBundle(bundleBob, ikAlice);
|
|
} catch(e) {
|
|
exception = true;
|
|
expect(e is InvalidSignatureException, true, reason: 'Expected InvalidSignatureException, but got $e');
|
|
}
|
|
|
|
expect(exception, true, reason: 'Expected test failure');
|
|
});
|
|
}
|