TextMessage_Test.java

package uk.ac.cam.db538.cryptosms.data;

import java.io.IOException;
import java.util.ArrayList;
import java.util.zip.DataFormatException;

import android.content.Context;
import android.test.mock.MockContext;

import junit.framework.TestCase;
import uk.ac.cam.db538.cryptosms.CustomAsserts;
import uk.ac.cam.db538.cryptosms.crypto.Encryption;
import uk.ac.cam.db538.cryptosms.crypto.EncryptionNone;
import uk.ac.cam.db538.cryptosms.crypto.EncryptionInterface.EncryptionException;
import uk.ac.cam.db538.cryptosms.data.Message.MessageException;
import uk.ac.cam.db538.cryptosms.storage.Common;
import uk.ac.cam.db538.cryptosms.storage.Conversation;
import uk.ac.cam.db538.cryptosms.storage.MessageData;
import uk.ac.cam.db538.cryptosms.storage.SessionKeys;
import uk.ac.cam.db538.cryptosms.storage.StorageFileException;
import uk.ac.cam.db538.cryptosms.storage.SessionKeys.SimNumber;
import uk.ac.cam.db538.cryptosms.utils.CompressedText;
import uk.ac.cam.db538.cryptosms.utils.LowLevel;

public class TextMessage_Test extends TestCase {
	
	protected void setUp() throws Exception {
		super.setUp();
		EncryptionNone.initEncryption();
		Common.clearStorageFile();
	}

	protected void tearDown() throws Exception {
		super.tearDown();
	}
	
	private final CompressedText textLong = CompressedText.createFromString("Lorem ipsum dolor sit amet, consectetur adipiscing elit. Donec nisl nisl, ullamcorper eu fermentum ac, eleifend quis orci. Pellentesque ut nulla erat, ac imperdiet tellus. Morbi vulputate mauris in nibh ornare imperdiet. Duis venenatis ligula quis mi vestibulum posuere. Etiam quis felis non justo dignissim viverra. Vivamus dapibus risus id risus sodales tincidunt eu vel orci. Lorem ipsum dolor sit amet, consectetur adipiscing elit. Vivamus sed purus sed odio aliquam gravida vestibulum vitae augue. Nullam neque elit, dapibus eu tincidunt eget, accumsan eu mi. Duis eget sagittis nisi. Aenean consectetur est nisl, non molestie lorem. Sed tincidunt turpis sit amet augue convallis mollis ut vehicula nisi. Morbi lobortis interdum mattis. Donec tincidunt suscipit vehicula. Praesent pharetra molestie arcu. Pellentesque habitant morbi tristique senectus et netus et malesuada fames ac turpis egestas. Aenean facilisis auctor ipsum, id bibendum sapien molestie sed. Donec risus enim, rhoncus vel ultrices a, lobortis non mauris. Aenean sit amet magna a dolor sollicitudin lacinia sit amet eu sapien. Sed semper pretium varius.");
	private final CompressedText textShort = CompressedText.createFromString("I'll be there at 5pm. Promise...");
	private final byte headerFirstShort = (byte) 0xA0;
	private final byte headerFirstLong = (byte) 0xB0;
	private final byte headerNext = (byte) 0xC0;
	
	private final byte nextID_Out = (byte)237;

	public void testTextMessage() throws StorageFileException, IOException, DataFormatException {
		Conversation conv = Conversation.createConversation();
		TextMessage msg = new TextMessage(MessageData.createMessageData(conv));
		
		// this should work
		try {
			msg.setText(textLong);
		} catch (MessageException e) {
			fail("Should work");
		}
		CompressedText textDecompressed = CompressedText.decode(textLong.getData(), textLong.getCharset(), textLong.isCompressed());
		assertEquals(textDecompressed, textLong);
		CustomAsserts.assertArrayEquals(msg.getStoredData(), textLong.getData());
		CompressedText textRead = msg.getText();
		assertEquals(textRead, textLong);
		
		// too long message
		// EXTREME!!!
		/*
		Random random = new Random();
		CompressedText tooLong;
		String str = "";
		while ((tooLong = CompressedText.createFromString(str)).getLength() <= 34870)
			if (tooLong.getLength() < 28000)
				for (int i = 0; i < 2000; ++i)
					str += (char) random.nextInt(40000);
			else
				str += (char) random.nextInt(40000);
		
		try {
			msg.setText(tooLong);
			fail("Should have thrown an exception");
		} catch (MessageException e) {
		}*/
	}
	
	public void testSMSData() throws StorageFileException, IOException, MessageException, EncryptionException {
		Context context = new MockContext();
		Conversation conv = Conversation.createConversation();
		TextMessage msg = new TextMessage(MessageData.createMessageData(conv));
		msg.setText(textShort);
		
		try {
			msg.getBytes(context);
			fail("Should have thrown exception");
		} catch (MessageException e) {
		}

		// generate keys
		SessionKeys keys = SessionKeys.createSessionKeys(conv);
		keys.setSimNumber(new SimNumber("+441234567890", false));
		keys.setNextID_Out(nextID_Out);
		keys.saveToFile();
		
		// get the data
		ArrayList<byte[]> data = msg.getBytes(context);
		assertEquals(data.size(), 1);
		assertEquals(data.get(0).length, 133);
		assertEquals(data.get(0)[0], headerFirstShort); // header
		assertEquals(data.get(0)[1], nextID_Out); // ID
		assertEquals(LowLevel.getUnsignedShort(data.get(0), 2), textShort.getDataLength()); // data length
		byte[] dataDecrypted = Encryption.getEncryption().decryptSymmetric(LowLevel.cutData(data.get(0), 4, 129), keys.getSessionKey_Out());
		assertEquals(dataDecrypted.length, 81);
		CustomAsserts.assertArrayEquals(LowLevel.cutData(dataDecrypted, 0, textShort.getDataLength()), textShort.getData());
		
		msg.setText(textLong);
		data = msg.getBytes(context);
		int texts = TextMessage.computeNumberOfMessageParts(textLong);
		int totalBytes = 81 + (texts - 1) * 130;
		assertEquals(data.size(), texts);
		byte[] dataEncrypted = new byte[Encryption.ENCRYPTION_OVERHEAD + totalBytes];
		// first text
		assertEquals(data.get(0).length, 133);
		assertEquals(data.get(0)[0], headerFirstLong); // header
		assertEquals(data.get(0)[1], nextID_Out); // ID
		assertEquals(LowLevel.getUnsignedShort(data.get(0), 2), textLong.getDataLength()); // data length
		System.arraycopy(data.get(0), 4, dataEncrypted, 0, 81 + Encryption.ENCRYPTION_OVERHEAD);
		for (int i = 1; i < texts; ++i) {
			assertEquals(data.get(i).length, 133);
			assertEquals(data.get(i)[0], headerNext); // header
			assertEquals(data.get(i)[1], nextID_Out); // header
			assertEquals(data.get(i)[2], LowLevel.getBytesUnsignedByte(i)); // index
			System.arraycopy(data.get(i), 3, dataEncrypted, 81 + Encryption.ENCRYPTION_OVERHEAD + (i - 1) * 130, 130);
		}
		dataDecrypted = Encryption.getEncryption().decryptSymmetric(dataEncrypted, keys.getSessionKey_Out());
		assertEquals(dataDecrypted.length, totalBytes);
		CustomAsserts.assertArrayEquals(LowLevel.cutData(dataDecrypted, 0, textLong.getDataLength()), textLong.getData());
		
		
	}
}