164 lines
4.6 KiB
Python
Executable File
164 lines
4.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
from osmo_gsm_tester.testenv import *
|
|
hlr = tenv.hlr()
|
|
bts = tenv.bts()
|
|
pcu = bts.pcu()
|
|
mgw_msc = tenv.mgw()
|
|
mgw_bsc = tenv.mgw()
|
|
stp = tenv.stp()
|
|
ggsn = tenv.ggsn()
|
|
sgsn = tenv.sgsn(hlr, ggsn)
|
|
msc = tenv.msc(hlr, mgw_msc, stp)
|
|
bsc = tenv.bsc(msc, mgw_bsc, stp)
|
|
|
|
modems = tenv.modems(int(prompt('How many modems?')))
|
|
|
|
bsc.bts_add(bts)
|
|
sgsn.bts_add(bts)
|
|
|
|
hlr.start()
|
|
stp.start()
|
|
ggsn.start()
|
|
sgsn.start()
|
|
msc.start()
|
|
mgw_msc.start()
|
|
mgw_bsc.start()
|
|
bsc.start()
|
|
|
|
bts.start()
|
|
print('Waiting for bts to connect to bsc...')
|
|
wait(bsc.bts_is_connected, bts)
|
|
print('Waiting for bts to be ready...')
|
|
wait(bts.ready_for_pcu)
|
|
pcu.start()
|
|
|
|
for m in modems:
|
|
hlr.subscriber_add(m)
|
|
m.connect(msc.mcc_mnc())
|
|
|
|
while True:
|
|
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')
|
|
cmd = cmd.strip().lower()
|
|
|
|
if not cmd:
|
|
continue
|
|
|
|
params = cmd.split()
|
|
|
|
if 'quit'.startswith(cmd):
|
|
break
|
|
|
|
elif 'debug'.startswith(cmd):
|
|
import pdb; pdb.set_trace()
|
|
|
|
elif 'wait-registered'.startswith(cmd):
|
|
try:
|
|
for m in modems:
|
|
wait(m.is_connected, msc.mcc_mnc())
|
|
wait(msc.subscriber_attached, *modems)
|
|
except Timeout:
|
|
print('Timeout while waiting for registration.')
|
|
|
|
elif 'get-registered'.startswith(cmd):
|
|
print(msc.imsi_list_attached())
|
|
print('RESULT: %s' %
|
|
('All modems are registered.' if msc.subscriber_attached(*modems)
|
|
else 'Some modem(s) not registered yet.'))
|
|
|
|
elif 'sms'.startswith(cmd):
|
|
for mo in modems:
|
|
for mt in modems:
|
|
mo.sms_send(mt.msisdn, 'to ' + mt.name())
|
|
|
|
elif cmd.startswith('call-list'):
|
|
if len(params) != 1 and len(params) != 2:
|
|
print('wrong format')
|
|
continue
|
|
for ms in modems:
|
|
if len(params) == 1 or str(ms.msisdn) == params[1]:
|
|
print('call-list: %r %r' % (ms.name(), ms.call_id_list()))
|
|
|
|
elif cmd.startswith('call-dial'):
|
|
if len(params) != 3:
|
|
print('wrong format')
|
|
continue
|
|
src_msisdn, dst_msisdn = params[1:]
|
|
for mo in modems:
|
|
if str(mo.msisdn) == src_msisdn:
|
|
print('dialing %s->%s' % (src_msisdn, dst_msisdn))
|
|
call_id = mo.call_dial(dst_msisdn)
|
|
print('dial success: call_id=%r' % call_id)
|
|
|
|
elif cmd.startswith('call-wait-incoming'):
|
|
if len(params) != 3:
|
|
print('wrong format')
|
|
continue
|
|
src_msisdn, dst_msisdn = params[1:]
|
|
for mt in modems:
|
|
if str(mt.msisdn) == dst_msisdn:
|
|
print('waiting for incoming %s->%s' % (src_msisdn, dst_msisdn))
|
|
call_id = mt.call_wait_incoming(src_msisdn)
|
|
print('incoming call success: call_id=%r' % call_id)
|
|
|
|
elif cmd.startswith('call-answer'):
|
|
if len(params) != 3:
|
|
print('wrong format')
|
|
continue
|
|
mt_msisdn, call_id = params[1:]
|
|
for mt in modems:
|
|
if str(mt.msisdn) == mt_msisdn:
|
|
print('answering %s %r' % (mt.name(), call_id))
|
|
mt.call_answer(call_id)
|
|
|
|
elif cmd.startswith('call-hangup'):
|
|
if len(params) != 3:
|
|
print('wrong format')
|
|
continue
|
|
ms_msisdn, call_id = params[1:]
|
|
for ms in modems:
|
|
if str(ms.msisdn) == ms_msisdn:
|
|
print('hanging up %s %r' % (ms.name(), call_id))
|
|
ms.call_hangup(call_id)
|
|
|
|
elif cmd.startswith('ussd'):
|
|
if len(params) != 2:
|
|
print('wrong format')
|
|
continue
|
|
ussd_cmd = params[1]
|
|
for ms in modems:
|
|
print('modem %s: ussd %s' % (ms.name(), ussd_cmd))
|
|
response = ms.ussd_send(ussd_cmd)
|
|
print('modem %s: response=%r' % (ms.name(), response))
|
|
|
|
elif cmd.startswith('data-attach'):
|
|
if len(params) != 1:
|
|
print('wrong format')
|
|
continue
|
|
for ms in modems:
|
|
print('modem %s: attach' % ms.name())
|
|
ms.attach()
|
|
wait(ms.is_attached)
|
|
print('modem %s: attached' % ms.name())
|
|
|
|
elif cmd.startswith('data-detach'):
|
|
if len(params) != 1:
|
|
print('wrong format')
|
|
continue
|
|
for ms in modems:
|
|
print('modem %s: detach' % ms.name())
|
|
ms.attach()
|
|
wait(lambda: not ms.is_attached())
|
|
print('modem %s: detached' % ms.name())
|
|
|
|
elif cmd.startswith('data-activate'):
|
|
if len(params) != 1:
|
|
print('wrong format')
|
|
continue
|
|
for ms in modems:
|
|
print('modem %s: activate' % ms.name())
|
|
response = ms.activate_context()
|
|
print('modem %s: response=%r' % (ms.name(), response))
|
|
|
|
else:
|
|
print('Unknown command: %s' % cmd)
|