use of org.matrix.androidsdk.data.RoomState in project matrix-android-sdk by matrix-org.
the class CryptoTest method test09_testAliceInACryptedRoomAfterInitialSync.
@Test
public void test09_testAliceInACryptedRoomAfterInitialSync() throws Exception {
Log.e(LOG_TAG, "test09_testAliceInACryptedRoomAfterInitialSync");
Context context = InstrumentationRegistry.getContext();
final HashMap<String, Object> results = new HashMap<>();
doE2ETestWithAliceInARoom();
mAliceSession.getCrypto().setWarnOnUnknownDevices(false);
final String message = "Hello myself!";
Credentials aliceCredentials = mAliceSession.getCredentials();
mAliceSession.clear(context);
Uri uri = Uri.parse(CryptoTestHelper.TESTS_HOME_SERVER_URL);
HomeServerConnectionConfig hs = new HomeServerConnectionConfig(uri);
hs.setCredentials(aliceCredentials);
IMXStore store = new MXFileStore(hs, context);
final CountDownLatch lock1 = new CountDownLatch(1);
final MXSession aliceSession2 = new MXSession(hs, new MXDataHandler(store, aliceCredentials), context);
MXStoreListener listener = new MXStoreListener() {
@Override
public void postProcess(String accountId) {
}
@Override
public void onStoreReady(String accountId) {
results.put("onStoreReady", "onStoreReady");
lock1.countDown();
}
@Override
public void onStoreCorrupted(String accountId, String description) {
lock1.countDown();
}
@Override
public void onStoreOOM(String accountId, String description) {
lock1.countDown();
}
};
aliceSession2.getDataHandler().getStore().addMXStoreListener(listener);
aliceSession2.getDataHandler().getStore().open();
lock1.await(1000, TimeUnit.MILLISECONDS);
assertTrue(results.containsKey("onStoreReady"));
final CountDownLatch lock1b = new CountDownLatch(2);
MXEventListener eventListener = new MXEventListener() {
@Override
public void onInitialSyncComplete(String toToken) {
results.put("onInitialSyncComplete", "onInitialSyncComplete");
lock1b.countDown();
}
@Override
public void onCryptoSyncComplete() {
results.put("onCryptoSyncComplete", "onCryptoSyncComplete");
lock1b.countDown();
}
};
aliceSession2.getDataHandler().addListener(eventListener);
aliceSession2.startEventStream(null);
lock1b.await(1000, TimeUnit.MILLISECONDS);
assertTrue(results.containsKey("onInitialSyncComplete"));
assertTrue(results.containsKey("onCryptoSyncComplete"));
Room roomFromAlicePOV2 = aliceSession2.getDataHandler().getRoom(mRoomId);
assertTrue(roomFromAlicePOV2.isEncrypted());
final CountDownLatch lock2 = new CountDownLatch(1);
if (false) {
// The android client does not echo its own message
MXEventListener aliceEventListener = new MXEventListener() {
@Override
public void onLiveEvent(Event event, RoomState roomState) {
if (TextUtils.equals(event.getType(), Event.EVENT_TYPE_MESSAGE)) {
try {
if (checkEncryptedEvent(event, mRoomId, message, aliceSession2)) {
lock2.countDown();
}
} catch (Exception e) {
}
}
}
};
roomFromAlicePOV2.addEventListener(aliceEventListener);
}
// the IOS client echoes the message
// the android client does not
roomFromAlicePOV2.sendEvent(buildTextEvent(message, aliceSession2), new ApiCallback<Void>() {
@Override
public void onSuccess(Void info) {
results.put("sendEvent", "sendEvent");
lock2.countDown();
}
@Override
public void onNetworkError(Exception e) {
lock2.countDown();
}
@Override
public void onMatrixError(MatrixError e) {
lock2.countDown();
}
@Override
public void onUnexpectedError(Exception e) {
lock2.countDown();
}
});
lock2.await(1000, TimeUnit.MILLISECONDS);
assertTrue(results.containsKey("sendEvent"));
aliceSession2.clear(context);
}
use of org.matrix.androidsdk.data.RoomState in project matrix-android-sdk by matrix-org.
the class CryptoTest method test15_testReplayAttack.
@Test
public void test15_testReplayAttack() throws Exception {
Log.e(LOG_TAG, "test15_testReplayAttack");
final HashMap<String, Object> results = new HashMap<>();
doE2ETestWithAliceAndBobInARoom(true);
mBobSession.getCrypto().setWarnOnUnknownDevices(false);
mAliceSession.getCrypto().setWarnOnUnknownDevices(false);
String messageFromAlice = "Hello I'm Alice!";
final Room roomFromBobPOV = mBobSession.getDataHandler().getRoom(mRoomId);
final Room roomFromAlicePOV = mAliceSession.getDataHandler().getRoom(mRoomId);
assertTrue(roomFromBobPOV.isEncrypted());
assertTrue(roomFromAlicePOV.isEncrypted());
final CountDownLatch lock1 = new CountDownLatch(2);
MXEventListener bobEventListener = new MXEventListener() {
@Override
public void onLiveEvent(Event event, RoomState roomState) {
if (TextUtils.equals(event.getType(), Event.EVENT_TYPE_MESSAGE) && !TextUtils.equals(event.getSender(), mBobSession.getMyUserId())) {
results.put("bobEcho", event);
event.setClearData(null);
mBobSession.getDataHandler().decryptEvent(event, roomFromBobPOV.getLiveTimeLine().getTimelineId());
results.put("decrypted", event);
lock1.countDown();
}
}
};
roomFromBobPOV.addEventListener(bobEventListener);
mBobSession.getDataHandler().addListener(new MXEventListener() {
@Override
public void onToDeviceEvent(Event event) {
results.put("onToDeviceEvent", event);
lock1.countDown();
}
});
roomFromAlicePOV.sendEvent(buildTextEvent(messageFromAlice, mAliceSession), new ApiCallback<Void>() {
@Override
public void onSuccess(Void info) {
}
@Override
public void onNetworkError(Exception e) {
}
@Override
public void onMatrixError(MatrixError e) {
}
@Override
public void onUnexpectedError(Exception e) {
}
});
lock1.await(1000, TimeUnit.MILLISECONDS);
assertTrue(results.containsKey("onToDeviceEvent"));
assertTrue(results.containsKey("bobEcho"));
assertTrue(results.containsKey("decrypted"));
Event decryptedEvent = (Event) results.get("decrypted");
assertTrue(null == decryptedEvent.getClearEvent());
assertTrue(TextUtils.equals(decryptedEvent.getCryptoError().errcode, MXCryptoError.DUPLICATED_MESSAGE_INDEX_ERROR_CODE));
// Decrypting it with no replay attack mitigation must still work
mBobSession.getDataHandler().decryptEvent(decryptedEvent, null);
assertTrue(checkEncryptedEvent(decryptedEvent, mRoomId, messageFromAlice, mAliceSession));
}
use of org.matrix.androidsdk.data.RoomState in project matrix-android-sdk by matrix-org.
the class CryptoTest method test17_testLateRoomKey.
@Test
public void test17_testLateRoomKey() throws Exception {
Log.e(LOG_TAG, "test17_testLateRoomKey");
final HashMap<String, Object> results = new HashMap<>();
doE2ETestWithAliceAndBobInARoom(true);
mBobSession.getCrypto().setWarnOnUnknownDevices(false);
mAliceSession.getCrypto().setWarnOnUnknownDevices(false);
String messageFromAlice = "Hello I'm Alice!";
final Room roomFromBobPOV = mBobSession.getDataHandler().getRoom(mRoomId);
final Room roomFromAlicePOV = mAliceSession.getDataHandler().getRoom(mRoomId);
assertTrue(roomFromBobPOV.isEncrypted());
assertTrue(roomFromAlicePOV.isEncrypted());
final CountDownLatch lock1 = new CountDownLatch(2);
MXEventListener bobEventListener = new MXEventListener() {
@Override
public void onToDeviceEvent(Event event) {
if (!results.containsKey("onToDeviceEvent")) {
results.put("onToDeviceEvent", event);
lock1.countDown();
}
}
};
mBobSession.getDataHandler().addListener(bobEventListener);
final ArrayList<Event> receivedEvents = new ArrayList<>();
EventTimeline.EventTimelineListener eventTimelineListener = new EventTimeline.EventTimelineListener() {
public void onEvent(Event event, EventTimeline.Direction direction, RoomState roomState) {
if (TextUtils.equals(event.getType(), Event.EVENT_TYPE_MESSAGE)) {
receivedEvents.add(event);
lock1.countDown();
}
}
};
roomFromBobPOV.getLiveTimeLine().addEventTimelineListener(eventTimelineListener);
roomFromAlicePOV.sendEvent(buildTextEvent(messageFromAlice, mAliceSession), new ApiCallback<Void>() {
@Override
public void onSuccess(Void info) {
}
@Override
public void onNetworkError(Exception e) {
}
@Override
public void onMatrixError(MatrixError e) {
}
@Override
public void onUnexpectedError(Exception e) {
}
});
lock1.await(2000, TimeUnit.MILLISECONDS);
assertTrue(results.containsKey("onToDeviceEvent"));
assertTrue(1 == receivedEvents.size());
Event event = receivedEvents.get(0);
assertTrue(checkEncryptedEvent(event, mRoomId, messageFromAlice, mAliceSession));
// Reinject a modified version of the received room_key event from Alice.
// From Bob pov, that mimics Alice resharing her keys but with an advanced outbound group session.
Event toDeviceEvent = (Event) results.get("onToDeviceEvent");
String sessionId = toDeviceEvent.getContentAsJsonObject().get("session_id").getAsString();
String senderKey = toDeviceEvent.senderKey();
// remove the session
mBobSession.getCrypto().getOlmDevice().removeInboundGroupSession(sessionId, senderKey);
event.setClearData(null);
// check that the message cannot be decrypted
assertTrue(!mBobSession.getDataHandler().decryptEvent(event, null));
// check the error code
assertTrue(TextUtils.equals(event.getCryptoError().errcode, MXCryptoError.UNKNOWN_INBOUND_SESSION_ID_ERROR_CODE));
receivedEvents.clear();
final CountDownLatch lock2 = new CountDownLatch(1);
roomFromBobPOV.addEventListener(new MXEventListener() {
@Override
public void onEventDecrypted(Event event) {
results.put("onEventDecrypted", "onEventDecrypted");
receivedEvents.add(event);
lock2.countDown();
}
});
event.setClearData(null);
// reinject the session key
mBobSession.getDataHandler().onToDeviceEvent(toDeviceEvent);
// the message should be decrypted later
lock2.await(1000, TimeUnit.MILLISECONDS);
assertTrue(results.containsKey("onEventDecrypted"));
assertTrue(1 == receivedEvents.size());
assertTrue(checkEncryptedEvent(receivedEvents.get(0), mRoomId, messageFromAlice, mAliceSession));
assertTrue(null == receivedEvents.get(0).getCryptoError());
}
use of org.matrix.androidsdk.data.RoomState in project matrix-android-sdk by matrix-org.
the class CryptoTest method doE2ETestWithAliceAndBobInARoomWithCryptedMessages.
private void doE2ETestWithAliceAndBobInARoomWithCryptedMessages(boolean cryptedBob) throws Exception {
doE2ETestWithAliceAndBobInARoom(cryptedBob);
if (null != mBobSession.getCrypto()) {
mBobSession.getCrypto().setWarnOnUnknownDevices(false);
}
if (null != mAliceSession.getCrypto()) {
mAliceSession.getCrypto().setWarnOnUnknownDevices(false);
}
final Room roomFromBobPOV = mBobSession.getDataHandler().getRoom(mRoomId);
final Room roomFromAlicePOV = mAliceSession.getDataHandler().getRoom(mRoomId);
mMessagesCount = 0;
final ArrayList<CountDownLatch> list = new ArrayList<>();
MXEventListener bobEventsListener = new MXEventListener() {
@Override
public void onLiveEvent(Event event, RoomState roomState) {
if (TextUtils.equals(event.getType(), Event.EVENT_TYPE_MESSAGE) && !TextUtils.equals(event.getSender(), mBobSession.getMyUserId())) {
mMessagesCount++;
list.get(0).countDown();
}
}
};
roomFromBobPOV.addEventListener(bobEventsListener);
ApiCallback<Void> callback = new ApiCallback<Void>() {
@Override
public void onSuccess(Void info) {
list.get(0).countDown();
}
@Override
public void onNetworkError(Exception e) {
}
@Override
public void onMatrixError(MatrixError e) {
}
@Override
public void onUnexpectedError(Exception e) {
}
};
final HashMap<String, Object> results = new HashMap<>();
CountDownLatch lock = new CountDownLatch(3);
list.clear();
list.add(lock);
mBobSession.getDataHandler().addListener(new MXEventListener() {
@Override
public void onToDeviceEvent(Event event) {
results.put("onToDeviceEvent", event);
list.get(0).countDown();
}
});
roomFromAlicePOV.sendEvent(buildTextEvent(messagesFromAlice.get(0), mAliceSession), callback);
lock.await(1000, TimeUnit.MILLISECONDS);
assertTrue(results.containsKey("onToDeviceEvent"));
assertTrue(mMessagesCount == 1);
lock = new CountDownLatch(1);
list.clear();
list.add(lock);
roomFromBobPOV.sendEvent(buildTextEvent(messagesFromBob.get(0), mBobSession), callback);
// android does not echo the messages sent from itself
mMessagesCount++;
lock.await(1000, TimeUnit.MILLISECONDS);
assertTrue(mMessagesCount == 2);
lock = new CountDownLatch(1);
list.clear();
list.add(lock);
roomFromBobPOV.sendEvent(buildTextEvent(messagesFromBob.get(1), mBobSession), callback);
// android does not echo the messages sent from itself
mMessagesCount++;
lock.await(1000, TimeUnit.MILLISECONDS);
assertTrue(mMessagesCount == 3);
lock = new CountDownLatch(1);
list.clear();
list.add(lock);
roomFromBobPOV.sendEvent(buildTextEvent(messagesFromBob.get(2), mBobSession), callback);
// android does not echo the messages sent from itself
mMessagesCount++;
lock.await(1000, TimeUnit.MILLISECONDS);
assertTrue(mMessagesCount == 4);
lock = new CountDownLatch(2);
list.clear();
list.add(lock);
roomFromAlicePOV.sendEvent(buildTextEvent(messagesFromAlice.get(1), mAliceSession), callback);
lock.await(1000, TimeUnit.MILLISECONDS);
assertTrue(mMessagesCount == 5);
}
use of org.matrix.androidsdk.data.RoomState in project matrix-android-sdk by matrix-org.
the class MXDataHandler method deleteRoomEvent.
/**
* Delete an event.
*
* @param event The event to be stored.
*/
public void deleteRoomEvent(Event event) {
if (isAlive()) {
Room room = getRoom(event.roomId);
if (null != room) {
mStore.deleteEvent(event);
Event lastEvent = mStore.getLatestEvent(event.roomId);
RoomState beforeLiveRoomState = room.getState().deepCopy();
RoomSummary summary = mStore.getSummary(event.roomId);
if (null == summary) {
summary = new RoomSummary(null, lastEvent, beforeLiveRoomState, mCredentials.userId);
} else {
summary.setLatestReceivedEvent(lastEvent, beforeLiveRoomState);
}
if (TextUtils.equals(summary.getReadReceiptEventId(), event.eventId)) {
summary.setReadReceiptEventId(lastEvent.eventId);
}
if (TextUtils.equals(summary.getReadMarkerEventId(), event.eventId)) {
summary.setReadMarkerEventId(lastEvent.eventId);
}
mStore.storeSummary(summary);
}
} else {
Log.e(LOG_TAG, "deleteRoomEvent : the session is not anymore active");
}
}
Aggregations