# -*- coding: utf-8 -*- from array import array import datetime import os import unittest from messaging.mms.message import MMSMessage # test data extracted from heyman's # http://github.com/heyman/mms-decoder DATA_DIR = os.path.join(os.path.dirname(__file__), 'mms-data') class TestMmsDecoding(unittest.TestCase): def test_decoding_from_data(self): path = os.path.join(DATA_DIR, 'iPhone.mms') data = array("B", open(path, 'rb').read()) mms = MMSMessage.from_data(data) headers = { 'From': '', 'Transaction-Id': '1262957356-3', 'MMS-Version': '1.2', 'To': '1337/TYPE=PLMN', 'Message-Type': 'm-send-req', 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '0.smil', 'Type': 'application/smil'}), } self.assertEqual(mms.headers, headers) def test_decoding_iPhone_mms(self): path = os.path.join(DATA_DIR, 'iPhone.mms') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'From': '', 'Transaction-Id': '1262957356-3', 'MMS-Version': '1.2', 'To': '1337/TYPE=PLMN', 'Message-Type': 'm-send-req', 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '0.smil', 'Type': 'application/smil'}), } smil_data = '\n\n\n \n\n\n\n\n\n\n

\n\n\n\n' self.assertEqual(mms.headers, headers) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(len(mms.data_parts), 2) self.assertEqual(mms.data_parts[0].content_type, 'application/smil') self.assertEqual(mms.data_parts[0].data, smil_data) self.assertEqual(mms.data_parts[1].content_type, 'image/jpeg') self.assertEqual(mms.data_parts[1].content_type_parameters, {'Name': 'IMG_6807.jpg'}) def test_decoding_SIMPLE_mms(self): path = os.path.join(DATA_DIR, 'SIMPLE.MMS') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'Transaction-Id': '1234', 'MMS-Version': '1.0', 'Message-Type': 'm-retrieve-conf', 'Date': datetime.datetime(2002, 12, 20, 21, 26, 56), 'Content-Type': ('application/vnd.wap.multipart.related', {}), 'Subject': 'Simple message', } text_data = "This is a simple MMS message with a single text body part." self.assertEqual(mms.headers, headers) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(len(mms.data_parts), 1) self.assertEqual(mms.data_parts[0].content_type, 'text/plain') self.assertEqual(mms.data_parts[0].data, text_data) def test_decoding_BTMMS_mms(self): path = os.path.join(DATA_DIR, 'BTMMS.MMS') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'Transaction-Id': '1234', 'MMS-Version': '1.0', 'Message-Type': 'm-retrieve-conf', 'Date': datetime.datetime(2003, 1, 21, 1, 57, 4), 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '', 'Type': 'application/smil'}), 'Subject': 'BT Ignite MMS', } smil_data = '

\r\n\r\n\r\n

\r\n\r\n

\r\n

\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n' text_data = 'BT Ignite\r\n\r\nMMS Services' self.assertEqual(mms.headers, headers) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(len(mms.data_parts), 4) self.assertEqual(mms.data_parts[0].content_type, 'application/smil') self.assertEqual(mms.data_parts[0].data, smil_data) self.assertEqual(mms.data_parts[1].content_type, 'image/gif') self.assertEqual(mms.data_parts[2].content_type, 'audio/amr') self.assertEqual(mms.data_parts[3].content_type, 'text/plain') self.assertEqual(mms.data_parts[3].data, text_data) def test_decoding_TOMSLOT_mms(self): path = os.path.join(DATA_DIR, 'TOMSLOT.MMS') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'From': '616c6c616e40746f6d736c6f742e636f6d'.decode('hex'), 'Transaction-Id': '1234', 'MMS-Version': '1.0', 'Message-Type': 'm-retrieve-conf', 'Date': datetime.datetime(2003, 2, 16, 3, 48, 33), 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '', 'Type': 'application/smil'}), 'Subject': 'Tom Slot Band', } smil_data = '\r\n\t\r\n\t\t\r\n\t\t\r\n\t\t\r\n\t\t\t\r\n\t\t\t\r\n\t\t\t\r\n\t\t\r\n\t\r\n\t\r\n\t\t\r\n\t\t\t

\r\n\t\t\r\n\t\t\r\n\t\t\t

\r\n\t\t\t\r\n\t\t\t\t\r\n\t\t\t\r\n\t\t\t

\r\n\t\r\n\r\n' text_data = 'Presented by NowMMS\r\n' self.assertEqual(mms.headers, headers) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(len(mms.data_parts), 8) self.assertEqual(mms.data_parts[0].content_type, 'application/smil') self.assertEqual(mms.data_parts[0].data, smil_data) self.assertEqual(mms.data_parts[1].content_type, 'image/jpeg') self.assertEqual(mms.data_parts[2].content_type, 'image/jpeg') self.assertEqual(mms.data_parts[3].content_type, 'image/jpeg') self.assertEqual(mms.data_parts[4].content_type, 'image/jpeg') self.assertEqual(mms.data_parts[5].content_type, 'image/jpeg') self.assertEqual(mms.data_parts[6].content_type, 'text/plain') self.assertEqual(mms.data_parts[6].data, text_data) self.assertEqual(mms.data_parts[7].content_type, 'audio/amr') def test_decoding_images_are_cut_off_debug_mms(self): path = os.path.join(DATA_DIR, 'images_are_cut_off_debug.mms') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'From': '', 'Read-Reply': False, 'Transaction-Id': '2112410527', 'MMS-Version': '1.0', 'To': '7464707440616a616a672e63646d'.decode('hex'), 'Delivery-Report': False, 'Message-Type': 'm-send-req', 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '', 'Type': 'application/smil'}), 'Subject': 'Picture3', } smil_data = '

' self.assertEqual(mms.headers, headers) self.assertEqual(len(mms.data_parts), 2) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(mms.data_parts[0].content_type, 'image/jpeg') self.assertEqual(mms.data_parts[0].content_type_parameters, {'Name': 'Picture3.jpg'}) self.assertEqual(mms.data_parts[1].content_type, 'application/smil') self.assertEqual(mms.data_parts[1].data, smil_data) def test_decoding_openwave_mms(self): path = os.path.join(DATA_DIR, 'openwave.mms') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'From': '2b31363530353535303030302f545950453d504c4d4e'.decode('hex'), 'Message-Class': 'Personal', 'Transaction-Id': '1067263672', 'MMS-Version': '1.0', 'Priority': 'Normal', 'To': '112/TYPE=PLMN', 'Delivery-Report': False, 'Message-Type': 'm-send-req', 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '', 'Type': 'application/smil'}), 'Subject': 'rubrik', } smil_data = '\n \n \n \n \n \n \n \n \n \n \n \n \n\n' text_data = 'rubrik' self.assertEqual(mms.headers, headers) self.assertEqual(len(mms.data_parts), 2) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(mms.data_parts[0].content_type, 'application/smil') self.assertEqual(mms.data_parts[0].data, smil_data) self.assertEqual(mms.data_parts[1].data, text_data) def test_decoding_SonyEricssonT310_R201_mms(self): path = os.path.join(DATA_DIR, 'SonyEricssonT310-R201.mms') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'Sender-Visibility': 'Show', 'From': '', 'Read-Reply': False, 'Message-Class': 'Personal', 'Transaction-Id': '1-8db', 'MMS-Version': '1.0', 'Priority': 'Normal', 'To': '55225/TYPE=PLMN', 'Delivery-Report': False, 'Message-Type': 'm-send-req', 'Date': datetime.datetime(2004, 3, 18, 7, 30, 34), 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '', 'Type': 'application/smil'}), } text_data = 'Hej hopp' smil_data = '

\r\n\r\n\r\n

\r\n\r\n

\r\n\r\n\r\n\r\n' self.assertEqual(mms.headers, headers) self.assertEqual(len(mms.data_parts), 4) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(mms.data_parts[0].content_type, 'image/gif') self.assertEqual(mms.data_parts[0].content_type_parameters, {'Name': 'Tony.gif'}) self.assertEqual(mms.data_parts[1].content_type, 'text/plain') self.assertEqual(mms.data_parts[1].data, text_data) self.assertEqual(mms.data_parts[2].content_type, 'audio/midi') self.assertEqual(mms.data_parts[2].content_type_parameters, {'Name': 'OldhPhone.mid'}) self.assertEqual(mms.data_parts[3].content_type, 'application/smil') self.assertEqual(mms.data_parts[3].data, smil_data) def test_decoding_gallery2test_mms(self): path = os.path.join(DATA_DIR, 'gallery2test.mms') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'From': '2b31363530353535303030302f545950453d504c4d4e'.decode('hex'), 'Message-Class': 'Personal', 'Transaction-Id': '1118775337', 'MMS-Version': '1.0', 'Priority': 'Normal', 'To': 'Jg', 'Delivery-Report': False, 'Message-Type': 'm-send-req', 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '', 'Type': 'application/smil'}), 'Subject': 'Jgj', } text_data = 'Jgj' smil_data = '\n \n \n \n \n \n \n \n \n \n gnu-head

\n \n \n \n\n' self.assertEqual(mms.headers, headers) self.assertEqual(len(mms.data_parts), 3) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(mms.data_parts[0].content_type, 'application/smil') self.assertEqual(mms.data_parts[0].data, smil_data) self.assertEqual(mms.data_parts[1].content_type, 'text/plain') self.assertEqual(mms.data_parts[1].data, text_data) self.assertEqual(mms.data_parts[2].content_type, 'image/jpeg') # XXX: Shouldn't it be 'Name' instead ? self.assertEqual(mms.data_parts[2].content_type_parameters, {'name': 'gnu-head.jpg'}) def test_decoding_projekt_exempel_mms(self): path = os.path.join(DATA_DIR, 'projekt_exempel.mms') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'Sender-Visibility': 'Show', 'From': '', 'Read-Reply': False, 'Message-Class': 'Personal', 'Transaction-Id': '4-fc60', 'MMS-Version': '1.0', 'Priority': 'Normal', 'To': '12345/TYPE=PLMN', 'Delivery-Report': False, 'Message-Type': 'm-send-req', 'Date': datetime.datetime(2004, 5, 23, 15, 13, 40), 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '', 'Type': 'application/smil'}), 'Subject': 'Hej', } smil_data = '

\r\n\r\n\r\n

\r\n\r\n\r\n\r\n' text_data = 'Jonatan \xc3\xa4r en GNU' self.assertEqual(mms.headers, headers) self.assertEqual(len(mms.data_parts), 3) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(mms.data_parts[0].content_type, 'text/plain') self.assertEqual(mms.data_parts[0].data, text_data) self.assertEqual(mms.data_parts[1].content_type, 'image/gif') self.assertEqual(mms.data_parts[2].content_type, 'application/smil') self.assertEqual(mms.data_parts[2].data, smil_data) self.assertEqual(mms.data_parts[2].content_type_parameters, {'Charset': 'utf-8', 'Name': 'mms.smil'}) def test_decoding_m_mms(self): path = os.path.join(DATA_DIR, 'm.mms') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'From': '676f6c64706f737440686f746d61696c2e636f6d'.decode('hex'), 'Transaction-Id': '0000000001', 'MMS-Version': '1.0', 'Message-Type': 'm-retrieve-conf', 'Date': datetime.datetime(2002, 8, 9, 13, 8, 2), 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '', 'Type': 'application/smil'}), 'Subject': 'GOLD', } text_data1 = 'Audio' text_data2 = 'Text +' text_data3 = 'tagtag.com/gold\r\n' text_data4 = 'globalisierunglobalisierunglobalisierunglobalisierunglobalisierunglobalisierunglobalisierungnureisilabolg' text_data5 = 'KLONE\r\nKLONE\r\n' text_data6 = 'pr\xe4sentiert..' text_data7 = 'GOLD' smil_data = '

\r\n\r\n

\r\n\r\n\r\n\r\n\r\n \r\n \r\n\r\n\r\n\r\n\r\n \r\n \r\n\r\n\r\n\r\n\r\n \r\n \r\n\r\n\r\n\r\n\r\n \r\n \r\n\r\n

\r\n\r\n\r\n\r\n \r\n \r\n\r\n\r\n\r\n\r\n \r\n \r\n\r\n\r\n\r\n\r\n\r\n \r\n \r\n\r\n\r\n' self.assertEqual(mms.headers, headers) self.assertEqual(len(mms.data_parts), 9) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(mms.data_parts[0].content_type, 'text/plain') self.assertEqual(mms.data_parts[0].data, text_data1) self.assertEqual(mms.data_parts[0].content_type_parameters, {'Charset': 'us-ascii'}) self.assertEqual(mms.data_parts[1].content_type, 'application/smil') self.assertEqual(mms.data_parts[1].data, smil_data) self.assertEqual(mms.data_parts[1].content_type_parameters, {'Charset': 'us-ascii'}) self.assertEqual(mms.data_parts[2].content_type, 'text/plain') self.assertEqual(mms.data_parts[2].data, text_data2) self.assertEqual(mms.data_parts[2].content_type_parameters, {'Charset': 'us-ascii'}) self.assertEqual(mms.data_parts[3].content_type, 'text/plain') self.assertEqual(mms.data_parts[3].data, text_data3) self.assertEqual(mms.data_parts[3].content_type_parameters, {'Charset': 'us-ascii'}) self.assertEqual(mms.data_parts[4].content_type, 'audio/amr') self.assertEqual(mms.data_parts[5].content_type, 'text/plain') self.assertEqual(mms.data_parts[5].data, text_data4) self.assertEqual(mms.data_parts[5].content_type_parameters, {'Charset': 'us-ascii'}) self.assertEqual(mms.data_parts[6].content_type, 'text/plain') self.assertEqual(mms.data_parts[6].data, text_data5) self.assertEqual(mms.data_parts[6].content_type_parameters, {'Charset': 'us-ascii'}) self.assertEqual(mms.data_parts[7].content_type, 'text/plain') self.assertEqual(mms.data_parts[7].data, text_data6) self.assertEqual(mms.data_parts[7].content_type_parameters, {'Charset': 'us-ascii'}) self.assertEqual(mms.data_parts[8].content_type, 'text/plain') self.assertEqual(mms.data_parts[8].data, text_data7) self.assertEqual(mms.data_parts[8].content_type_parameters, {'Charset': 'us-ascii'}) def test_decoding_27d0a048cd79555de05283a22372b0eb_mms(self): path = os.path.join(DATA_DIR, '27d0a048cd79555de05283a22372b0eb.mms') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'Sender-Visibility': 'Show', 'From': '', 'Read-Reply': False, 'Message-Class': 'Personal', 'Transaction-Id': '3-31cb', 'MMS-Version': '1.0', 'Priority': 'Normal', 'To': '123/TYPE=PLMN', 'Delivery-Report': False, 'Message-Type': 'm-send-req', 'Date': datetime.datetime(2004, 5, 23, 14, 14, 58), 'Content-Type': ('application/vnd.wap.multipart.related', {'Start': '', 'Type': 'application/smil'}), 'Subject': 'Angående art-tillhörighet', #'Subject': 'Ang\xc3\xa5ende art-tillh\xc3\xb6righet', } smil_data = '

\r\n\r\n\r\n

\r\n\r\n

\r\n\r\n\r\n\r\n\r\n' text_data = 'Jonatan \xc3\xa4r en gnu.' self.assertEqual(mms.headers, headers) self.assertEqual(len(mms.data_parts), 3) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.related') self.assertEqual(mms.data_parts[0].content_type, 'image/vnd.wap.wbmp') self.assertEqual(mms.data_parts[0].content_type_parameters, {'Name': 'Rain.wbmp'}) self.assertEqual(mms.data_parts[1].content_type, 'text/plain') self.assertEqual(mms.data_parts[1].data, text_data) self.assertEqual(mms.data_parts[1].content_type_parameters, {'Charset': 'utf-8', 'Name': 'mms.txt'}) self.assertEqual(mms.data_parts[2].content_type, 'application/smil') self.assertEqual(mms.data_parts[2].data, smil_data) self.assertEqual(mms.data_parts[2].content_type_parameters, {'Charset': 'utf-8', 'Name': 'mms.smil'}) def test_decoding_SEC_SGHS300M(self): path = os.path.join(DATA_DIR, 'SEC-SGHS300M.mms') mms = MMSMessage.from_file(path) self.assertTrue(isinstance(mms, MMSMessage)) headers = { 'Sender-Visibility': 'Show', 'From': '', 'Read-Reply': False, 'Message-Class': 'Personal', 'Transaction-Id': '31887', 'MMS-Version': '1.0', 'To': '303733383334353636342f545950453d504c4d4e'.decode('hex'), 'Delivery-Report': False, 'Message-Type': 'm-send-req', 'Subject': 'IL', 'Content-Type': ('application/vnd.wap.multipart.mixed', {}), } text_data = 'HV' self.assertEqual(mms.headers, headers) self.assertEqual(len(mms.data_parts), 1) self.assertEqual(mms.content_type, 'application/vnd.wap.multipart.mixed') self.assertEqual(mms.data_parts[0].content_type, 'text/plain') self.assertEqual(mms.data_parts[0].data, text_data) self.assertEqual(mms.data_parts[0].content_type_parameters, {'Charset': 'utf-8'}) def test_encoding_m_sendnotifyresp_ind(self): message = MMSMessage() message.headers['Transaction-Id'] = 'NOK5AIdhfTMYSG4JeIgAAsHtp72AGAAAAAAAA' message.headers['Message-Type'] = 'm-notifyresp-ind' message.headers['Status'] = 'Retrieved' data = [ 140, 131, 152, 78, 79, 75, 53, 65, 73, 100, 104, 102, 84, 77, 89, 83, 71, 52, 74, 101, 73, 103, 65, 65, 115, 72, 116, 112, 55, 50, 65, 71, 65, 65, 65, 65, 65, 65, 65, 65, 0, 141, 144, 149, 129, 132, 163, 1, 35, 129] self.assertEqual(list(message.encode()[:50]), data)