e80f96cc3b
It is common in CSV files that the columns have uppercase names, so we have adopted this scheme when we started using the card_key_provider. This also means that the API of the card_key_provider_get() and card_key_provider_get_field() function now implicitly requires uppercase field names like 'ICCID', 'ADM1', etc. Unfortunately this may be unreliable, so let's convert the field names to uppercase as soon as we receive them. This makes the API case-insensitive and gives us the assurance that all field names we ever work with are in uppercase. Related: SYS#7725 Change-Id: I9d80752587e2ccff0963c10abd5a2f42f5868d79
153 lines
10 KiB
Python
153 lines
10 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import unittest
|
|
import os
|
|
from pySim.card_key_provider import *
|
|
|
|
class TestCardKeyProviderCsv(unittest.TestCase):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
column_keys = {"KI" : "000424252525535532532A0B0C0D0E0F",
|
|
"OPC" : "000102030405065545645645645D0E0F",
|
|
"KIC1" : "06410203546406456456450B0C0D0E0F",
|
|
"KID1" : "00040267840507667609045645645E0F",
|
|
"KIK1" : "0001020307687607668678678C0D0E0F",
|
|
"KIC2" : "000142457594860706090A0B0688678F",
|
|
"KID2" : "600102030405649468690A0B0C0D648F",
|
|
"KIK2" : "00010203330506070496330B08640E0F",
|
|
"KIC3" : "000104030405064684686A068C0D0E0F",
|
|
"KID3" : "00010243048468070809060B0C0D0E0F",
|
|
"KIK3" : "00010204040506070809488B0C0D0E0F"}
|
|
|
|
csv_file_path = os.path.dirname(os.path.abspath(__file__)) + "/test_card_key_provider.csv"
|
|
card_key_provider_register(CardKeyProviderCsv(csv_file_path, column_keys))
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def test_card_key_provider_get(self):
|
|
test_data = [{'EXPECTED' : {'PIN1': '1234', 'PUK1': '12345678', 'PIN2': '1223', 'PUK2': '12345678',
|
|
'KI': '48a6d5f60567d45299e3ba08594009e7', 'ADM1': '10101010',
|
|
'ADM2': '9999999999999999', 'KIC1': '3eb8567fa0b4b1e63bcab13bff5f2702',
|
|
'KIC2': 'fd6c173a5b3f04b563808da24237fb46',
|
|
'KIC3': '66c8c848e5dff69d70689d155d44f323',
|
|
'KID1': 'd78accce870332dced467c173244dd94',
|
|
'KID2': 'b3bf050969747b2d2c9389e127a3d791',
|
|
'KID3': '40a77deb50d260b3041bbde1b5040625',
|
|
'KIK1': '451b503239d818ea34421aa9c2a8887a',
|
|
'KIK2': '967716f5fca8ae179f87f76524d1ae6b',
|
|
'KIK3': '0884db5eee5409a00fc1bbc57ac52541',
|
|
'OPC': '81817574c1961dd272ad080eb2caf279'}, 'ICCID' :"8988211000000000001"},
|
|
{'EXPECTED' : {'PIN1': '1234', 'PUK1': '12345678', 'PIN2': '1223', 'PUK2': '12345678',
|
|
'KI': 'e94d7fa6fb92375dae86744ff6ecef49', 'ADM1': '10101010',
|
|
'ADM2': '9999999999999999', 'KIC1': '79b4e39387c66253da68f653381ded44',
|
|
'KIC2': '560561b5dba89c1da8d1920049e5e4f7',
|
|
'KIC3': '79ff35e84e39305a119af8c79f84e8e5',
|
|
'KID1': '233baf89122159553d67545ecedcf8e0',
|
|
'KID2': '8fc2874164d7a8e40d72c968bc894ab8',
|
|
'KID3': '2e3320f0dda85054d261be920fbfa065',
|
|
'KIK1': 'd51b1b17630103d1672a3e9e0e4827ed',
|
|
'KIK2': 'd01edbc48be555139506b0d7982bf7ff',
|
|
'KIK3': 'a6487a5170849e8e0a03026afea91f5a',
|
|
'OPC': '6b0d19ef28bd12f2daac31828d426939'}, 'ICCID' :"8988211000000000002"},
|
|
{'EXPECTED' : {'PIN1': '1234', 'PUK1': '12345678', 'PIN2': '1223', 'PUK2': '12345678',
|
|
'KI': '3cdec1552ef433a89f327905213c5a6e', 'ADM1': '10101010',
|
|
'ADM2': '9999999999999999', 'KIC1': '72986b13ce505e12653ad42df5cfca13',
|
|
'KIC2': '8f0d1e58b01e833773e5562c4940674d',
|
|
'KIC3': '9c72ba5a14d54f489edbffd3d8802f03',
|
|
'KID1': 'd23a42995df9ca83f74b2cfd22695526',
|
|
'KID2': '5c3a189d12aa1ac6614883d7de5e6c8c',
|
|
'KID3': 'a6ace0d303a2b38a96b418ab83c16725',
|
|
'KIK1': 'bf2319467d859c12527aa598430caef2',
|
|
'KIK2': '6a4c459934bea7e40787976b8881ab01',
|
|
'KIK3': '91cd02c38b5f68a98cc90a1f2299538f',
|
|
'OPC': '6df46814b1697daca003da23808bbbc3'}, 'ICCID' :"8988211000000000003"}]
|
|
|
|
for t in test_data:
|
|
result = card_key_provider_get(["PIN1","PUK1","PIN2","PUK2","KI","ADM1","ADM2","KIC1",
|
|
"KIC2","KIC3","KID1","KID2","KID3","KIK1","KIK2","KIK3","OPC"],
|
|
"ICCID", t.get('ICCID'))
|
|
self.assertEqual(result, t.get('EXPECTED'))
|
|
result = card_key_provider_get(["PIN1","puk1","PIN2","PUK2","KI","adm1","ADM2","KIC1",
|
|
"KIC2","kic3","KID1","KID2","KID3","kik1","KIK2","KIK3","OPC"],
|
|
"iccid", t.get('ICCID'))
|
|
self.assertEqual(result, t.get('EXPECTED'))
|
|
|
|
|
|
def test_card_key_provider_get_field(self):
|
|
test_data = [{'EXPECTED' : "3eb8567fa0b4b1e63bcab13bff5f2702", 'ICCID' :"8988211000000000001"},
|
|
{'EXPECTED' : "79b4e39387c66253da68f653381ded44", 'ICCID' :"8988211000000000002"},
|
|
{'EXPECTED' : "72986b13ce505e12653ad42df5cfca13", 'ICCID' :"8988211000000000003"}]
|
|
|
|
for t in test_data:
|
|
result = card_key_provider_get_field("KIC1", "ICCID", t.get('ICCID'))
|
|
self.assertEqual(result, t.get('EXPECTED'))
|
|
for t in test_data:
|
|
result = card_key_provider_get_field("kic1", "iccid", t.get('ICCID'))
|
|
self.assertEqual(result, t.get('EXPECTED'))
|
|
|
|
|
|
class TestCardKeyFieldCryptor(unittest.TestCase):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
transport_keys = {"KI" : "000424252525535532532A0B0C0D0E0F",
|
|
"OPC" : "000102030405065545645645645D0E0F",
|
|
"KIC1" : "06410203546406456456450B0C0D0E0F",
|
|
"UICC_SCP03" : "00040267840507667609045645645E0F"}
|
|
self.crypt = CardKeyFieldCryptor(transport_keys)
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def test_encrypt_field(self):
|
|
test_data = [{'EXPECTED' : "0b1e1e56cd62645aeb4c2d72a7c98f27",
|
|
'PLAINTEXT_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "OPC"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'PLAINTEXT_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "NOCRYPT"},
|
|
{'EXPECTED' : "00248276d2734f108f9761e2f98e2a9d",
|
|
'PLAINTEXT_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "UICC_SCP03_KIC1"},
|
|
{'EXPECTED' : "00248276d2734f108f9761e2f98e2a9d",
|
|
'PLAINTEXT_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "UICC_SCP03_KID1"},
|
|
{'EXPECTED' : "00248276d2734f108f9761e2f98e2a9d",
|
|
'PLAINTEXT_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "UICC_SCP03_KIK1"},
|
|
{'EXPECTED' : "0b1e1e56cd62645aeb4c2d72a7c98f27",
|
|
'PLAINTEXT_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "opc"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'PLAINTEXT_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "nocrypt"},
|
|
{'EXPECTED' : "00248276d2734f108f9761e2f98e2a9d",
|
|
'PLAINTEXT_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "uicc_scp03_kic1"},
|
|
{'EXPECTED' : "00248276d2734f108f9761e2f98e2a9d",
|
|
'PLAINTEXT_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "uicc_scp03_kid1"},
|
|
{'EXPECTED' : "00248276d2734f108f9761e2f98e2a9d",
|
|
'PLAINTEXT_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "uicc_scp03_kik1"}]
|
|
|
|
for t in test_data:
|
|
result = self.crypt.encrypt_field(t.get('FIELDNAME'), t.get('PLAINTEXT_VAL'))
|
|
self.assertEqual(result, t.get('EXPECTED'))
|
|
|
|
def test_decrypt_field(self):
|
|
test_data = [{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'ENCRYPTED_VAL' : "0b1e1e56cd62645aeb4c2d72a7c98f27", 'FIELDNAME' : "OPC"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'ENCRYPTED_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "NOCRYPT"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'ENCRYPTED_VAL' : "00248276d2734f108f9761e2f98e2a9d", 'FIELDNAME' : "UICC_SCP03_KIC1"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'ENCRYPTED_VAL' : "00248276d2734f108f9761e2f98e2a9d", 'FIELDNAME' : "UICC_SCP03_KID1"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'ENCRYPTED_VAL' : "00248276d2734f108f9761e2f98e2a9d", 'FIELDNAME' : "UICC_SCP03_KIK1"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'ENCRYPTED_VAL' : "0b1e1e56cd62645aeb4c2d72a7c98f27", 'FIELDNAME' : "opc"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'ENCRYPTED_VAL' : "000102030405060708090a0b0c0d0e0f", 'FIELDNAME' : "nocrypt"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'ENCRYPTED_VAL' : "00248276d2734f108f9761e2f98e2a9d", 'FIELDNAME' : "uicc_scp03_kic1"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'ENCRYPTED_VAL' : "00248276d2734f108f9761e2f98e2a9d", 'FIELDNAME' : "uicc_scp03_kid1"},
|
|
{'EXPECTED' : "000102030405060708090a0b0c0d0e0f",
|
|
'ENCRYPTED_VAL' : "00248276d2734f108f9761e2f98e2a9d", 'FIELDNAME' : "uicc_scp03_kik1"}]
|
|
|
|
for t in test_data:
|
|
result = self.crypt.decrypt_field(t.get('FIELDNAME'), t.get('ENCRYPTED_VAL'))
|
|
self.assertEqual(result, t.get('EXPECTED'))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|