Examples with ImportRoomKeysResult org.matrix.androidsdk.crypto.data.ImportRoomKeysResult used on opensource projects

Example 1 with ImportRoomKeysResult

use of org.matrix.androidsdk.crypto.data.ImportRoomKeysResult in project matrix-android-sdk by matrix-org.

the class CryptoTest method test24_testExportImport.

@Test
public void test24_testExportImport() throws Exception {
    Log.e(LOG_TAG, "test24_testExportImport");
    Context context = InstrumentationRegistry.getContext();
    final Map<String, Object> results = new HashMap<>();
    CryptoTestData cryptoTestData = mCryptoTestHelper.doE2ETestWithAliceInARoom();
    MXSession aliceSession = cryptoTestData.getFirstSession();
    String aliceRoomId = cryptoTestData.getRoomId();
    aliceSession.getCrypto().setWarnOnUnknownDevices(false);
    String message = "Hello myself!";
    String password = "hello";
    Room roomFromAlicePOV = aliceSession.getDataHandler().getRoom(aliceRoomId);
    CountDownLatch lock1 = new CountDownLatch(1);
    roomFromAlicePOV.sendEvent(mCryptoTestHelper.buildTextEvent(message, aliceSession, aliceRoomId), new TestApiCallback<Void>(lock1) {

        @Override
        public void onSuccess(Void info) {
            results.put("sendEvent", "sendEvent");
            super.onSuccess(info);
        }
    });
    mTestHelper.await(lock1);
    Assert.assertTrue(results.containsKey("sendEvent"));
    Credentials aliceCredentials = aliceSession.getCredentials();
    Credentials aliceCredentials2 = new Credentials();
    CountDownLatch lock1a = new CountDownLatch(1);
    aliceSession.getCrypto().exportRoomKeys(password, new TestApiCallback<byte[]>(lock1a) {

        @Override
        public void onSuccess(byte[] info) {
            results.put("exportRoomKeys", info);
            super.onSuccess(info);
        }
    });
    mTestHelper.await(lock1a);
    Assert.assertTrue(results.containsKey("exportRoomKeys"));
    // close the session and clear the data
    aliceSession.clear(context);
    aliceCredentials2.userId = aliceCredentials.userId;
    aliceCredentials2.homeServer = aliceCredentials.homeServer;
    aliceCredentials2.accessToken = aliceCredentials.accessToken;
    aliceCredentials2.refreshToken = aliceCredentials.refreshToken;
    aliceCredentials2.deviceId = "AliceNewDevice";
    HomeServerConnectionConfig hs = mTestHelper.createHomeServerConfig(aliceCredentials2);
    IMXStore store = new MXFileStore(hs, false, context);
    MXSession aliceSession2 = new MXSession.Builder(hs, new MXDataHandler(store, aliceCredentials2), context).withLegacyCryptoStore(mCryptoTestHelper.getUSE_LEGACY_CRYPTO_STORE()).build();
    aliceSession2.enableCryptoWhenStarting();
    final CountDownLatch lock1b = new CountDownLatch(1);
    MXStoreListener listener = new MXStoreListener() {

        @Override
        public void postProcess(String accountId) {
        }

        @Override
        public void onStoreReady(String accountId) {
            results.put("onStoreReady", "onStoreReady");
            lock1b.countDown();
        }

        @Override
        public void onStoreCorrupted(String accountId, String description) {
            lock1b.countDown();
        }

        @Override
        public void onStoreOOM(String accountId, String description) {
            lock1b.countDown();
        }
    };
    aliceSession2.getDataHandler().getStore().addMXStoreListener(listener);
    aliceSession2.getDataHandler().getStore().open();
    mTestHelper.await(lock1b);
    Assert.assertTrue(results.containsKey("onStoreReady"));
    final CountDownLatch lock2 = new CountDownLatch(2);
    MXEventListener eventListener = new MXEventListener() {

        @Override
        public void onInitialSyncComplete(String toToken) {
            results.put("onInitialSyncComplete", "onInitialSyncComplete");
            lock2.countDown();
        }

        @Override
        public void onCryptoSyncComplete() {
            results.put("onCryptoSyncComplete", "onCryptoSyncComplete");
            lock2.countDown();
        }
    };
    aliceSession2.getDataHandler().addListener(eventListener);
    aliceSession2.startEventStream(null);
    mTestHelper.await(lock2);
    Assert.assertTrue(results.containsKey("onInitialSyncComplete"));
    Assert.assertTrue(results.containsKey("onCryptoSyncComplete"));
    Room roomFromAlicePOV2 = aliceSession2.getDataHandler().getRoom(aliceRoomId);
    Assert.assertNotNull(roomFromAlicePOV2);
    Assert.assertTrue(roomFromAlicePOV2.getState().isEncrypted());
    Event event = roomFromAlicePOV2.getDataHandler().getStore().getLatestEvent(aliceRoomId);
    Assert.assertNotNull(event);
    Assert.assertTrue(event.isEncrypted());
    Assert.assertNull(event.getClearEvent());
    Assert.assertNotNull(event.getCryptoError());
    Assert.assertEquals(MXCryptoError.UNKNOWN_INBOUND_SESSION_ID_ERROR_CODE, event.getCryptoError().errcode);
    // import the e2e keys
    // test with a wrong password
    CountDownLatch lock3 = new CountDownLatch(1);
    aliceSession2.getCrypto().importRoomKeys((byte[]) results.get("exportRoomKeys"), "wrong password", null, new TestApiCallback<ImportRoomKeysResult>(lock3, false) {

        @Override
        public void onUnexpectedError(Exception e) {
            results.put("importRoomKeys_failed", "importRoomKeys_failed");
            super.onUnexpectedError(e);
        }
    });
    mTestHelper.await(lock3);
    Assert.assertTrue(results.containsKey("importRoomKeys_failed"));
    // check that the message cannot be decrypted
    event = roomFromAlicePOV2.getDataHandler().getStore().getLatestEvent(aliceRoomId);
    Assert.assertNotNull(event);
    Assert.assertTrue(event.isEncrypted());
    Assert.assertNull(event.getClearEvent());
    Assert.assertNotNull(event.getCryptoError());
    Assert.assertEquals(MXCryptoError.UNKNOWN_INBOUND_SESSION_ID_ERROR_CODE, event.getCryptoError().errcode);
    CountDownLatch lock4 = new CountDownLatch(1);
    aliceSession2.getCrypto().importRoomKeys((byte[]) results.get("exportRoomKeys"), password, null, new TestApiCallback<ImportRoomKeysResult>(lock4) {

        @Override
        public void onSuccess(ImportRoomKeysResult info) {
            Assert.assertEquals(1, info.getTotalNumberOfKeys());
            Assert.assertEquals(1, info.getSuccessfullyNumberOfImportedKeys());
            results.put("importRoomKeys", "importRoomKeys");
            super.onSuccess(info);
        }
    });
    mTestHelper.await(lock4);
    Assert.assertTrue(results.containsKey("importRoomKeys"));
    // check that the message CAN be decrypted
    event = roomFromAlicePOV2.getDataHandler().getStore().getLatestEvent(aliceRoomId);
    Assert.assertNotNull(event);
    Assert.assertTrue(event.isEncrypted());
    Assert.assertNotNull(event.getClearEvent());
    Assert.assertNull(event.getCryptoError());
    mCryptoTestHelper.checkEncryptedEvent(event, aliceRoomId, message, aliceSession);
    cryptoTestData.clear(context);
    aliceSession2.clear(context);
}