Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
Pau Espin Pedrol | dfe38ad | 2017-11-09 13:57:39 +0100 | [diff] [blame] | 2 | from osmo_gsm_tester.testenv import * |
Pau Espin Pedrol | 40c7bc7 | 2020-05-05 13:41:42 +0200 | [diff] [blame] | 3 | hlr = tenv.hlr() |
| 4 | bts = tenv.bts() |
Pau Espin Pedrol | 4385780 | 2018-09-13 15:07:27 +0200 | [diff] [blame] | 5 | pcu = bts.pcu() |
Pau Espin Pedrol | 40c7bc7 | 2020-05-05 13:41:42 +0200 | [diff] [blame] | 6 | mgw_msc = tenv.mgw() |
| 7 | mgw_bsc = tenv.mgw() |
| 8 | stp = tenv.stp() |
| 9 | ggsn = tenv.ggsn() |
| 10 | sgsn = tenv.sgsn(hlr, ggsn) |
| 11 | msc = tenv.msc(hlr, mgw_msc, stp) |
| 12 | bsc = tenv.bsc(msc, mgw_bsc, stp) |
Pau Espin Pedrol | 4385780 | 2018-09-13 15:07:27 +0200 | [diff] [blame] | 13 | |
Pau Espin Pedrol | 40c7bc7 | 2020-05-05 13:41:42 +0200 | [diff] [blame] | 14 | modems = tenv.modems(int(prompt('How many modems?'))) |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 15 | |
Pau Espin Pedrol | 4385780 | 2018-09-13 15:07:27 +0200 | [diff] [blame] | 16 | bsc.bts_add(bts) |
| 17 | sgsn.bts_add(bts) |
| 18 | |
| 19 | hlr.start() |
| 20 | stp.start() |
| 21 | ggsn.start() |
| 22 | sgsn.start() |
| 23 | msc.start() |
| 24 | mgw_msc.start() |
| 25 | mgw_bsc.start() |
| 26 | bsc.start() |
| 27 | |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 28 | bts.start() |
Pau Espin Pedrol | 4385780 | 2018-09-13 15:07:27 +0200 | [diff] [blame] | 29 | print('Waiting for bts to connect to bsc...') |
| 30 | wait(bsc.bts_is_connected, bts) |
| 31 | print('Waiting for bts to be ready...') |
| 32 | wait(bts.ready_for_pcu) |
| 33 | pcu.start() |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 34 | |
| 35 | for m in modems: |
Pau Espin Pedrol | 4385780 | 2018-09-13 15:07:27 +0200 | [diff] [blame] | 36 | hlr.subscriber_add(m) |
| 37 | m.connect(msc.mcc_mnc()) |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 38 | |
| 39 | while True: |
Pau Espin Pedrol | d12b84c | 2020-02-18 12:56:21 +0100 | [diff] [blame] | 40 | cmd = prompt('Enter command: (q)uit (d)ebug (s)ms (g)et-registered (w)ait-registered, call-list [<ms_msisdn>], call-dial <src_msisdn> <dst_msisdn>, call-wait-incoming <src_msisdn> <dst_msisdn>, call-answer <mt_msisdn> <call_id>, call-hangup <ms_msisdn> <call_id>, ussd <command>, data-attach, data-wait, data-detach, data-activate') |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 41 | cmd = cmd.strip().lower() |
| 42 | |
| 43 | if not cmd: |
| 44 | continue |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 45 | |
| 46 | params = cmd.split() |
| 47 | |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 48 | if 'quit'.startswith(cmd): |
| 49 | break |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 50 | |
Pau Espin Pedrol | d12b84c | 2020-02-18 12:56:21 +0100 | [diff] [blame] | 51 | elif 'debug'.startswith(cmd): |
| 52 | import pdb; pdb.set_trace() |
| 53 | |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 54 | elif 'wait-registered'.startswith(cmd): |
Neels Hofmeyr | a88b0c7 | 2017-05-07 02:15:48 +0200 | [diff] [blame] | 55 | try: |
Pau Espin Pedrol | 0e57aad | 2017-05-29 14:25:22 +0200 | [diff] [blame] | 56 | for m in modems: |
Pau Espin Pedrol | 3a81a7e | 2020-06-10 16:52:53 +0200 | [diff] [blame] | 57 | wait(m.is_registered, msc.mcc_mnc()) |
Pau Espin Pedrol | 4385780 | 2018-09-13 15:07:27 +0200 | [diff] [blame] | 58 | wait(msc.subscriber_attached, *modems) |
Neels Hofmeyr | a88b0c7 | 2017-05-07 02:15:48 +0200 | [diff] [blame] | 59 | except Timeout: |
| 60 | print('Timeout while waiting for registration.') |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 61 | |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 62 | elif 'get-registered'.startswith(cmd): |
Pau Espin Pedrol | 4385780 | 2018-09-13 15:07:27 +0200 | [diff] [blame] | 63 | print(msc.imsi_list_attached()) |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 64 | print('RESULT: %s' % |
Pau Espin Pedrol | 4385780 | 2018-09-13 15:07:27 +0200 | [diff] [blame] | 65 | ('All modems are registered.' if msc.subscriber_attached(*modems) |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 66 | else 'Some modem(s) not registered yet.')) |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 67 | |
Neels Hofmeyr | 67cea22 | 2017-05-06 15:06:04 +0200 | [diff] [blame] | 68 | elif 'sms'.startswith(cmd): |
| 69 | for mo in modems: |
| 70 | for mt in modems: |
Pau Espin Pedrol | 4b7c585 | 2020-10-14 14:48:21 +0200 | [diff] [blame] | 71 | mo.sms_send(mt.msisdn(), 'to ' + mt.name()) |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 72 | |
| 73 | elif cmd.startswith('call-list'): |
| 74 | if len(params) != 1 and len(params) != 2: |
| 75 | print('wrong format') |
| 76 | continue |
| 77 | for ms in modems: |
Pau Espin Pedrol | 4b7c585 | 2020-10-14 14:48:21 +0200 | [diff] [blame] | 78 | if len(params) == 1 or str(ms.msisdn()) == params[1]: |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 79 | print('call-list: %r %r' % (ms.name(), ms.call_id_list())) |
| 80 | |
| 81 | elif cmd.startswith('call-dial'): |
| 82 | if len(params) != 3: |
| 83 | print('wrong format') |
| 84 | continue |
| 85 | src_msisdn, dst_msisdn = params[1:] |
| 86 | for mo in modems: |
Pau Espin Pedrol | 4b7c585 | 2020-10-14 14:48:21 +0200 | [diff] [blame] | 87 | if str(mo.msisdn()) == src_msisdn: |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 88 | print('dialing %s->%s' % (src_msisdn, dst_msisdn)) |
| 89 | call_id = mo.call_dial(dst_msisdn) |
| 90 | print('dial success: call_id=%r' % call_id) |
| 91 | |
| 92 | elif cmd.startswith('call-wait-incoming'): |
| 93 | if len(params) != 3: |
| 94 | print('wrong format') |
| 95 | continue |
| 96 | src_msisdn, dst_msisdn = params[1:] |
| 97 | for mt in modems: |
Pau Espin Pedrol | 4b7c585 | 2020-10-14 14:48:21 +0200 | [diff] [blame] | 98 | if str(mt.msisdn()) == dst_msisdn: |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 99 | print('waiting for incoming %s->%s' % (src_msisdn, dst_msisdn)) |
| 100 | call_id = mt.call_wait_incoming(src_msisdn) |
| 101 | print('incoming call success: call_id=%r' % call_id) |
| 102 | |
| 103 | elif cmd.startswith('call-answer'): |
| 104 | if len(params) != 3: |
| 105 | print('wrong format') |
| 106 | continue |
| 107 | mt_msisdn, call_id = params[1:] |
| 108 | for mt in modems: |
Pau Espin Pedrol | 4b7c585 | 2020-10-14 14:48:21 +0200 | [diff] [blame] | 109 | if str(mt.msisdn()) == mt_msisdn: |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 110 | print('answering %s %r' % (mt.name(), call_id)) |
| 111 | mt.call_answer(call_id) |
| 112 | |
| 113 | elif cmd.startswith('call-hangup'): |
| 114 | if len(params) != 3: |
| 115 | print('wrong format') |
| 116 | continue |
| 117 | ms_msisdn, call_id = params[1:] |
| 118 | for ms in modems: |
Pau Espin Pedrol | 4b7c585 | 2020-10-14 14:48:21 +0200 | [diff] [blame] | 119 | if str(ms.msisdn()) == ms_msisdn: |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 120 | print('hanging up %s %r' % (ms.name(), call_id)) |
| 121 | ms.call_hangup(call_id) |
| 122 | |
Pau Espin Pedrol | d27de2b | 2017-11-06 17:42:41 +0100 | [diff] [blame] | 123 | elif cmd.startswith('ussd'): |
| 124 | if len(params) != 2: |
| 125 | print('wrong format') |
| 126 | continue |
| 127 | ussd_cmd = params[1] |
| 128 | for ms in modems: |
| 129 | print('modem %s: ussd %s' % (ms.name(), ussd_cmd)) |
| 130 | response = ms.ussd_send(ussd_cmd) |
| 131 | print('modem %s: response=%r' % (ms.name(), response)) |
| 132 | |
Pau Espin Pedrol | 4385780 | 2018-09-13 15:07:27 +0200 | [diff] [blame] | 133 | elif cmd.startswith('data-attach'): |
| 134 | if len(params) != 1: |
| 135 | print('wrong format') |
| 136 | continue |
| 137 | for ms in modems: |
| 138 | print('modem %s: attach' % ms.name()) |
| 139 | ms.attach() |
| 140 | wait(ms.is_attached) |
| 141 | print('modem %s: attached' % ms.name()) |
| 142 | |
| 143 | elif cmd.startswith('data-detach'): |
| 144 | if len(params) != 1: |
| 145 | print('wrong format') |
| 146 | continue |
| 147 | for ms in modems: |
| 148 | print('modem %s: detach' % ms.name()) |
| 149 | ms.attach() |
| 150 | wait(lambda: not ms.is_attached()) |
| 151 | print('modem %s: detached' % ms.name()) |
| 152 | |
| 153 | elif cmd.startswith('data-activate'): |
| 154 | if len(params) != 1: |
| 155 | print('wrong format') |
| 156 | continue |
| 157 | for ms in modems: |
| 158 | print('modem %s: activate' % ms.name()) |
| 159 | response = ms.activate_context() |
| 160 | print('modem %s: response=%r' % (ms.name(), response)) |
| 161 | |
Pau Espin Pedrol | 4897e7a | 2017-10-06 18:41:20 +0200 | [diff] [blame] | 162 | else: |
| 163 | print('Unknown command: %s' % cmd) |