diff options
-rw-r--r-- | suites/encryption/lib/testlib.py | 54 | ||||
-rwxr-xr-x | suites/encryption/register_a5_0_authopt.py | 37 | ||||
-rwxr-xr-x | suites/encryption/register_a5_0_authreq.py | 50 | ||||
-rwxr-xr-x | suites/encryption/register_a5_1_authreq.py | 50 |
4 files changed, 66 insertions, 125 deletions
diff --git a/suites/encryption/lib/testlib.py b/suites/encryption/lib/testlib.py new file mode 100644 index 0000000..3948941 --- /dev/null +++ b/suites/encryption/lib/testlib.py @@ -0,0 +1,54 @@ +#!/usr/bin/env python3 +from osmo_gsm_tester.testenv import * + +def encryption_test_setup_run(enable_auth, algo): + hlr = suite.hlr() + bts = suite.bts() + mgw_msc = suite.mgw() + mgw_bsc = suite.mgw() + stp = suite.stp() + msc = suite.msc(hlr, mgw_msc, stp) + bsc = suite.bsc(msc, mgw_bsc, stp) + ms = suite.modem() + + print('start network...') + msc.set_authentication(enable_auth) + msc.set_encryption(algo) + bsc.set_encryption(algo) + hlr.start() + stp.start() + msc.start() + mgw_msc.start() + mgw_bsc.start() + bsc.bts_add(bts) + bsc.start() + bts.start() + wait(bsc.bts_is_connected, bts) + + ms.log_info() + good_ki = ms.ki() + bad_ki = ("%1X" % (int(good_ki[0], 16) ^ 0x01)) + good_ki[1:] + + print('KI changed: ' + good_ki + " => " + bad_ki) + ms.set_ki(bad_ki) + hlr.subscriber_add(ms) + if enable_auth: + print('Attempt connection with wrong KI...') + ms.connect(msc.mcc_mnc()) + + sleep(40) # TODO: read pcap or CTRL interface and look for Rejected? (gsm_a.dtap.msg_mm_type == 0x04) + print('Asserting modem did not register') + # FIXME: this can fail because ofono qmi signals registered before being accepted by network. See OS#2458 + # assert not ms.is_connected(msc.mcc_mnc()) + assert not msc.subscriber_attached(ms) + + hlr.subscriber_delete(ms) + print('KI changed: ' + bad_ki + " => " + good_ki) + ms.set_ki(good_ki) + hlr.subscriber_add(ms, ms.msisdn) + print('Attempt connection with correct KI...') + else: + print('Attempt connection with wrong KI, should work as it is not used...') + ms.connect(msc.mcc_mnc()) + wait(ms.is_connected, msc.mcc_mnc()) + wait(msc.subscriber_attached, ms) diff --git a/suites/encryption/register_a5_0_authopt.py b/suites/encryption/register_a5_0_authopt.py index acbbb1f..1b7f471 100755 --- a/suites/encryption/register_a5_0_authopt.py +++ b/suites/encryption/register_a5_0_authopt.py @@ -1,37 +1,8 @@ #!/usr/bin/env python3 from osmo_gsm_tester.testenv import * -hlr = suite.hlr() -bts = suite.bts() -mgw_msc = suite.mgw() -mgw_bsc = suite.mgw() -stp = suite.stp() -msc = suite.msc(hlr, mgw_msc, stp) -bsc = suite.bsc(msc, mgw_bsc, stp) -ms = suite.modem() +import testlib +suite.test_import_modules_register_for_cleanup(testlib) +from testlib import encryption_test_setup_run -print('start network...') -msc.set_authentication(False) -msc.set_encryption('a5_0') -bsc.set_encryption('a5_0') -hlr.start() -stp.start() -msc.start() -mgw_msc.start() -mgw_bsc.start() -bsc.bts_add(bts) -bsc.start() -bts.start() -wait(bsc.bts_is_connected, bts) - -ms.log_info() -good_ki = ms.ki() -bad_ki = ("%1X" % (int(good_ki[0], 16) ^ 0x01)) + good_ki[1:] - -print('KI changed: ' + good_ki + " => " + bad_ki) -ms.set_ki(bad_ki) -hlr.subscriber_add(ms) -print('Attempt connection with wrong KI, should work as it is not used...') -ms.connect(msc.mcc_mnc()) -wait(ms.is_connected, msc.mcc_mnc()) -wait(msc.subscriber_attached, ms) +encryption_test_setup_run(False, 'a5_0') diff --git a/suites/encryption/register_a5_0_authreq.py b/suites/encryption/register_a5_0_authreq.py index 789ace5..feca525 100755 --- a/suites/encryption/register_a5_0_authreq.py +++ b/suites/encryption/register_a5_0_authreq.py @@ -1,50 +1,8 @@ #!/usr/bin/env python3 from osmo_gsm_tester.testenv import * -hlr = suite.hlr() -bts = suite.bts() -mgw_msc = suite.mgw() -mgw_bsc = suite.mgw() -stp = suite.stp() -msc = suite.msc(hlr, mgw_msc, stp) -bsc = suite.bsc(msc, mgw_bsc, stp) -ms = suite.modem() +import testlib +suite.test_import_modules_register_for_cleanup(testlib) +from testlib import encryption_test_setup_run -print('start network...') -msc.set_authentication(True) -msc.set_encryption('a5_0') -bsc.set_encryption('a5_0') -hlr.start() -stp.start() -msc.start() -mgw_msc.start() -mgw_bsc.start() -bsc.bts_add(bts) -bsc.start() -bts.start() -wait(bsc.bts_is_connected, bts) - -ms.log_info() -good_ki = ms.ki() -bad_ki = ("%1X" % (int(good_ki[0], 16) ^ 0x01)) + good_ki[1:] - -print('KI changed: ' + good_ki + " => " + bad_ki) -ms.set_ki(bad_ki) -hlr.subscriber_add(ms) -print('Attempt connection with wrong KI...') -ms.connect(msc.mcc_mnc()) - -sleep(40) # TODO: read pcap or CTRL interface and look for Rejected? (gsm_a.dtap.msg_mm_type == 0x04) -print('Asserting modem did not register') -# FIXME: this can fail because ofono qmi signals registered before being accepted by network. See OS#2458 -# assert not ms.is_connected(msc.mcc_mnc()) -assert not msc.subscriber_attached(ms) - -hlr.subscriber_delete(ms) -print('KI changed: ' + bad_ki + " => " + good_ki) -ms.set_ki(good_ki) -hlr.subscriber_add(ms, ms.msisdn) -print('Attempt connection with correct KI...') -ms.connect(msc.mcc_mnc()) -wait(ms.is_connected, msc.mcc_mnc()) -wait(msc.subscriber_attached, ms) +encryption_test_setup_run(True, 'a5_0') diff --git a/suites/encryption/register_a5_1_authreq.py b/suites/encryption/register_a5_1_authreq.py index e7aa76c..077819b 100755 --- a/suites/encryption/register_a5_1_authreq.py +++ b/suites/encryption/register_a5_1_authreq.py @@ -1,50 +1,8 @@ #!/usr/bin/env python3 from osmo_gsm_tester.testenv import * -hlr = suite.hlr() -bts = suite.bts() -mgw_msc = suite.mgw() -mgw_bsc = suite.mgw() -stp = suite.stp() -msc = suite.msc(hlr, mgw_msc, stp) -bsc = suite.bsc(msc, mgw_bsc, stp) -ms = suite.modem() +import testlib +suite.test_import_modules_register_for_cleanup(testlib) +from testlib import encryption_test_setup_run -print('start network...') -msc.set_authentication(True) -msc.set_encryption('a5_1') -bsc.set_encryption('a5_1') -hlr.start() -stp.start() -msc.start() -mgw_msc.start() -mgw_bsc.start() -bsc.bts_add(bts) -bsc.start() -bts.start() -wait(bsc.bts_is_connected, bts) - -ms.log_info() -good_ki = ms.ki() -bad_ki = ("%1X" % (int(good_ki[0], 16) ^ 0x01)) + good_ki[1:] - -print('KI changed: ' + good_ki + " => " + bad_ki) -ms.set_ki(bad_ki) -hlr.subscriber_add(ms) -print('Attempt connection with wrong KI...') -ms.connect(msc.mcc_mnc()) - -sleep(40) # TODO: read pcap or CTRL interface and look for Rejected? (gsm_a.dtap.msg_mm_type == 0x04) -print('Asserting modem did not register') -# FIXME: this can fail because ofono qmi signals registered before being accepted by network. See OS#2458 -# assert not ms.is_connected(msc.mcc_mnc()) -assert not msc.subscriber_attached(ms) - -hlr.subscriber_delete(ms) -print('KI changed: ' + bad_ki + " => " + good_ki) -ms.set_ki(good_ki) -hlr.subscriber_add(ms, ms.msisdn) -print('Attempt connection with correct KI...') -ms.connect(msc.mcc_mnc()) -wait(ms.is_connected, msc.mcc_mnc()) -wait(msc.subscriber_attached, ms) +encryption_test_setup_run(True, 'a5_1') |