2018-06-13 12:29:16 +02:00
|
|
|
/**
|
|
|
|
*
|
|
|
|
* Copyright 2018 Paul Schaub
|
|
|
|
*
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
*
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
*
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
* limitations under the License.
|
|
|
|
*/
|
|
|
|
package org.jivesoftware.smackx.omemo;
|
|
|
|
|
2020-04-12 21:54:18 +02:00
|
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
2018-06-13 12:29:16 +02:00
|
|
|
|
2019-08-05 08:28:42 +02:00
|
|
|
import java.io.IOException;
|
2018-06-13 12:29:16 +02:00
|
|
|
import java.util.List;
|
|
|
|
|
|
|
|
import org.jivesoftware.smack.SmackException;
|
2019-10-24 15:45:08 +02:00
|
|
|
import org.jivesoftware.smack.XMPPConnection;
|
2018-06-13 12:29:16 +02:00
|
|
|
import org.jivesoftware.smack.XMPPException;
|
2019-10-24 15:45:08 +02:00
|
|
|
import org.jivesoftware.smack.packet.MessageBuilder;
|
|
|
|
|
2018-06-13 12:29:16 +02:00
|
|
|
import org.jivesoftware.smackx.mam.MamManager;
|
|
|
|
import org.jivesoftware.smackx.mam.element.MamPrefsIQ;
|
|
|
|
import org.jivesoftware.smackx.omemo.exceptions.CryptoFailedException;
|
|
|
|
import org.jivesoftware.smackx.omemo.exceptions.UndecidedOmemoIdentityException;
|
|
|
|
import org.jivesoftware.smackx.omemo.util.MessageOrOmemoMessage;
|
|
|
|
|
|
|
|
import org.igniterealtime.smack.inttest.SmackIntegrationTestEnvironment;
|
|
|
|
import org.igniterealtime.smack.inttest.TestNotPossibleException;
|
2020-04-12 21:54:18 +02:00
|
|
|
import org.igniterealtime.smack.inttest.annotations.SmackIntegrationTest;
|
2018-06-13 12:29:16 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
* This test sends a message from Alice to Bob, while Bob has automatic decryption disabled.
|
|
|
|
* Then Bob fetches his Mam archive and decrypts the result.
|
|
|
|
*/
|
|
|
|
public class OmemoMamDecryptionTest extends AbstractTwoUsersOmemoIntegrationTest {
|
2020-04-04 13:03:31 +02:00
|
|
|
public OmemoMamDecryptionTest(SmackIntegrationTestEnvironment environment)
|
2018-06-13 12:29:16 +02:00
|
|
|
throws XMPPException.XMPPErrorException, SmackException.NotConnectedException, InterruptedException,
|
|
|
|
SmackException.NoResponseException, TestNotPossibleException {
|
|
|
|
super(environment);
|
|
|
|
MamManager bobsMamManager = MamManager.getInstanceFor(conTwo);
|
|
|
|
if (!bobsMamManager.isSupported()) {
|
|
|
|
throw new TestNotPossibleException("Test is not possible, because MAM is not supported on the server.");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
@SmackIntegrationTest
|
|
|
|
public void mamDecryptionTest() throws XMPPException.XMPPErrorException, SmackException.NotLoggedInException,
|
|
|
|
SmackException.NotConnectedException, InterruptedException, SmackException.NoResponseException,
|
2019-08-05 08:28:42 +02:00
|
|
|
CryptoFailedException, UndecidedOmemoIdentityException, IOException {
|
2018-06-13 12:29:16 +02:00
|
|
|
// Make sure, Bobs server stores messages in the archive
|
|
|
|
MamManager bobsMamManager = MamManager.getInstanceFor(bob.getConnection());
|
|
|
|
bobsMamManager.enableMamForAllMessages();
|
|
|
|
bobsMamManager.setDefaultBehavior(MamPrefsIQ.DefaultBehavior.always);
|
|
|
|
|
|
|
|
// Prevent bob from automatically decrypting MAM messages.
|
|
|
|
bob.stopStanzaAndPEPListeners();
|
|
|
|
|
|
|
|
String body = "This message will be stored in MAM!";
|
|
|
|
OmemoMessage.Sent encrypted = alice.encrypt(bob.getOwnJid(), body);
|
2019-10-24 15:45:08 +02:00
|
|
|
|
|
|
|
XMPPConnection alicesConnection = alice.getConnection();
|
|
|
|
MessageBuilder messageBuilder = alicesConnection.getStanzaFactory().buildMessageStanza();
|
|
|
|
alicesConnection.sendStanza(encrypted.buildMessage(messageBuilder, bob.getOwnJid()));
|
2018-06-13 12:29:16 +02:00
|
|
|
|
|
|
|
MamManager.MamQuery query = bobsMamManager.queryArchive(MamManager.MamQueryArgs.builder().limitResultsToJid(alice.getOwnJid()).build());
|
|
|
|
assertEquals(1, query.getMessageCount());
|
|
|
|
|
|
|
|
List<MessageOrOmemoMessage> decryptedMamQuery = bob.decryptMamQueryResult(query);
|
|
|
|
|
|
|
|
assertEquals(1, decryptedMamQuery.size());
|
|
|
|
assertEquals(body, decryptedMamQuery.get(decryptedMamQuery.size() - 1).getOmemoMessage().getBody());
|
|
|
|
}
|
|
|
|
}
|