twonly-app-dependencies/libsignal_protocol_dart/test/ratchet/ratcheting_session_test.dart

import 'dart:typed_data';

import 'package:libsignal_protocol_dart/src/ecc/curve.dart';
import 'package:libsignal_protocol_dart/src/ecc/ec_key_pair.dart';
import 'package:libsignal_protocol_dart/src/identity_key.dart';
import 'package:libsignal_protocol_dart/src/identity_key_pair.dart';
import 'package:libsignal_protocol_dart/src/ratchet/bob_signal_protocol_parameters.dart';
import 'package:libsignal_protocol_dart/src/ratchet/ratcheting_session.dart';
import 'package:libsignal_protocol_dart/src/state/session_state.dart';
import 'package:optional/optional.dart';
import 'package:test/test.dart';

void main() {
  test('testRatchetingSessionAsBob', () {
    final bobPublic = Uint8List.fromList([
      0x05,
      0x2c,
      0xb4,
      0x97,
      0x76,
      0xb8,
      0x77,
      0x02,
      0x05,
      0x74,
      0x5a,
      0x3a,
      0x6e,
      0x24,
      0xf5,
      0x79,
      0xcd,
      0xb4,
      0xba,
      0x7a,
      0x89,
      0x04,
      0x10,
      0x05,
      0x92,
      0x8e,
      0xbb,
      0xad,
      0xc9,
      0xc0,
      0x5a,
      0xd4,
      0x58
    ]);

    final bobPrivate = Uint8List.fromList([
      0xa1,
      0xca,
      0xb4,
      0x8f,
      0x7c,
      0x89,
      0x3f,
      0xaf,
      0xa9,
      0x88,
      0x0a,
      0x28,
      0xc3,
      0xb4,
      0x99,
      0x9d,
      0x28,
      0xd6,
      0x32,
      0x95,
      0x62,
      0xd2,
      0x7a,
      0x4e,
      0xa4,
      0xe2,
      0x2e,
      0x9f,
      0xf1,
      0xbd,
      0xd6,
      0x5a
    ]);

    final bobIdentityPublic = Uint8List.fromList([
      0x05,
      0xf1,
      0xf4,
      0x38,
      0x74,
      0xf6,
      0x96,
      0x69,
      0x56,
      0xc2,
      0xdd,
      0x47,
      0x3f,
      0x8f,
      0xa1,
      0x5a,
      0xde,
      0xb7,
      0x1d,
      0x1c,
      0xb9,
      0x91,
      0xb2,
      0x34,
      0x16,
      0x92,
      0x32,
      0x4c,
      0xef,
      0xb1,
      0xc5,
      0xe6,
      0x26
    ]);

    final bobIdentityPrivate = Uint8List.fromList([
      0x48,
      0x75,
      0xcc,
      0x69,
      0xdd,
      0xf8,
      0xea,
      0x07,
      0x19,
      0xec,
      0x94,
      0x7d,
      0x61,
      0x08,
      0x11,
      0x35,
      0x86,
      0x8d,
      0x5f,
      0xd8,
      0x01,
      0xf0,
      0x2c,
      0x02,
      0x25,
      0xe5,
      0x16,
      0xdf,
      0x21,
      0x56,
      0x60,
      0x5e
    ]);

    final aliceBasePublic = Uint8List.fromList([
      0x05,
      0x47,
      0x2d,
      0x1f,
      0xb1,
      0xa9,
      0x86,
      0x2c,
      0x3a,
      0xf6,
      0xbe,
      0xac,
      0xa8,
      0x92,
      0x02,
      0x77,
      0xe2,
      0xb2,
      0x6f,
      0x4a,
      0x79,
      0x21,
      0x3e,
      0xc7,
      0xc9,
      0x06,
      0xae,
      0xb3,
      0x5e,
      0x03,
      0xcf,
      0x89,
      0x50
    ]);

    final aliceEphemeralPublic = Uint8List.fromList([
      0x05,
      0x6c,
      0x3e,
      0x0d,
      0x1f,
      0x52,
      0x02,
      0x83,
      0xef,
      0xcc,
      0x55,
      0xfc,
      0xa5,
      0xe6,
      0x70,
      0x75,
      0xb9,
      0x04,
      0x00,
      0x7f,
      0x18,
      0x81,
      0xd1,
      0x51,
      0xaf,
      0x76,
      0xdf,
      0x18,
      0xc5,
      0x1d,
      0x29,
      0xd3,
      0x4b
    ]);

    final aliceIdentityPublic = Uint8List.fromList([
      0x05,
      0xb4,
      0xa8,
      0x45,
      0x56,
      0x60,
      0xad,
      0xa6,
      0x5b,
      0x40,
      0x10,
      0x07,
      0xf6,
      0x15,
      0xe6,
      0x54,
      0x04,
      0x17,
      0x46,
      0x43,
      0x2e,
      0x33,
      0x39,
      0xc6,
      0x87,
      0x51,
      0x49,
      0xbc,
      0xee,
      0xfc,
      0xb4,
      0x2b,
      0x4a
    ]);

    final bobSignedPreKeyPublic = Uint8List.fromList([
      0x05,
      0xac,
      0x24,
      0x8a,
      0x8f,
      0x26,
      0x3b,
      0xe6,
      0x86,
      0x35,
      0x76,
      0xeb,
      0x03,
      0x62,
      0xe2,
      0x8c,
      0x82,
      0x8f,
      0x01,
      0x07,
      0xa3,
      0x37,
      0x9d,
      0x34,
      0xba,
      0xb1,
      0x58,
      0x6b,
      0xf8,
      0xc7,
      0x70,
      0xcd,
      0x67
    ]);

    final bobSignedPreKeyPrivate = Uint8List.fromList([
      0x58,
      0x39,
      0x00,
      0x13,
      0x1f,
      0xb7,
      0x27,
      0x99,
      0x8b,
      0x78,
      0x03,
      0xfe,
      0x6a,
      0xc2,
      0x2c,
      0xc5,
      0x91,
      0xf3,
      0x42,
      0xe4,
      0xe4,
      0x2a,
      0x8c,
      0x8d,
      0x5d,
      0x78,
      0x19,
      0x42,
      0x09,
      0xb8,
      0xd2,
      0x53
    ]);

    final senderChain = Uint8List.fromList([
      0x97,
      0x97,
      0xca,
      0xca,
      0x53,
      0xc9,
      0x89,
      0xbb,
      0xe2,
      0x29,
      0xa4,
      0x0c,
      0xa7,
      0x72,
      0x70,
      0x10,
      0xeb,
      0x26,
      0x04,
      0xfc,
      0x14,
      0x94,
      0x5d,
      0x77,
      0x95,
      0x8a,
      0x0a,
      0xed,
      0xa0,
      0x88,
      0xb4,
      0x4d
    ]);

    final bobIdentityKeyPublic = IdentityKey.fromBytes(bobIdentityPublic, 0);
    final bobIdentityKeyPrivate = Curve.decodePrivatePoint(bobIdentityPrivate);
    final bobIdentityKey =
        IdentityKeyPair(bobIdentityKeyPublic, bobIdentityKeyPrivate);
    final bobEphemeralPublicKey = Curve.decodePoint(bobPublic, 0);
    final bobEphemeralPrivateKey = Curve.decodePrivatePoint(bobPrivate);
    final bobEphemeralKey =
        ECKeyPair(bobEphemeralPublicKey, bobEphemeralPrivateKey);
    // ignore: unused_local_variable
    final bobBaseKey = bobEphemeralKey;
    final bobSignedPreKey = ECKeyPair(
        Curve.decodePoint(bobSignedPreKeyPublic, 0),
        Curve.decodePrivatePoint(bobSignedPreKeyPrivate));

    final aliceBasePublicKey = Curve.decodePoint(aliceBasePublic, 0);
    // ignore: unused_local_variable
    final aliceEphemeralPublicKey = Curve.decodePoint(aliceEphemeralPublic, 0);
    final aliceIdentityPublicKey =
        IdentityKey.fromBytes(aliceIdentityPublic, 0);

    final parameters = BobSignalProtocolParameters(
      ourIdentityKey: bobIdentityKey,
      ourSignedPreKey: bobSignedPreKey,
      ourRatchetKey: bobEphemeralKey,
      ourOneTimePreKey: const Optional<ECKeyPair>.empty(),
      theirIdentityKey: aliceIdentityPublicKey,
      theirBaseKey: aliceBasePublicKey,
    );

    final session = SessionState();

    RatchetingSession.initializeSessionBob(session, parameters);

    expect(
        0,
        session
            .getLocalIdentityKey()
            .publicKey
            .compareTo(bobIdentityKey.getPublicKey().publicKey));
    expect(
        0,
        session
            .getRemoteIdentityKey()
            ?.publicKey
            .compareTo(aliceIdentityPublicKey.publicKey));
    expect(session.getSenderChainKey().key, senderChain);
  });
}