saip-tool: add commandline option to edit mandatory services list

Change-Id: I120b98d4b0942c26674bc1365c5711101ec95235
This commit is contained in:
Philipp Maier
2025-03-31 16:58:24 +02:00
parent 59faa02f9a
commit d5da431fd4
2 changed files with 36 additions and 0 deletions

View File

@@ -101,6 +101,12 @@ parser_rappi.add_argument('--output-file', required=True, help='Output file name
parser_rappi.add_argument('--aid', required=True, help='Load package AID') parser_rappi.add_argument('--aid', required=True, help='Load package AID')
parser_rappi.add_argument('--inst-aid', required=True, help='Instance AID') parser_rappi.add_argument('--inst-aid', required=True, help='Instance AID')
esrv_flag_choices = [t.name for t in asn1.types['ServicesList'].type.root_members]
parser_esrv = subparsers.add_parser('edit-mand-srv-list', help='Add/Remove service flag from/to mandatory services list')
parser_esrv.add_argument('--output-file', required=True, help='Output file name')
parser_esrv.add_argument('--add-flag', default=[], choices=esrv_flag_choices, action='append', help='Add flag to mandatory services list')
parser_esrv.add_argument('--remove-flag', default=[], choices=esrv_flag_choices, action='append', help='Remove flag from mandatory services list')
parser_info = subparsers.add_parser('tree', help='Display the filesystem tree') parser_info = subparsers.add_parser('tree', help='Display the filesystem tree')
def write_pes(pes: ProfileElementSequence, output_file:str): def write_pes(pes: ProfileElementSequence, output_file:str):
@@ -418,6 +424,25 @@ def do_remove_app_inst(pes:ProfileElementSequence, opts):
return return
print("Load Package AID: %s not found in PE Sequence" % opts.aid) print("Load Package AID: %s not found in PE Sequence" % opts.aid)
def do_edit_mand_srv_list(pes: ProfileElementSequence, opts):
header = pes.pe_by_type.get('header', [])[0]
for s in opts.add_flag:
print("Adding service '%s' to mandatory services list..." % s)
header.mandatory_service_add(s)
for s in opts.remove_flag:
if s in header.decoded['eUICC-Mandatory-services'].keys():
print("Removing service '%s' from mandatory services list..." % s)
header.mandatory_service_remove(s)
else:
print("Service '%s' not present in mandatory services list, cannot remove!" % s)
print("The following services are now set mandatory:")
for s in header.decoded['eUICC-Mandatory-services'].keys():
print("\t%s" % s)
write_pes(pes, opts.output_file)
def do_tree(pes:ProfileElementSequence, opts): def do_tree(pes:ProfileElementSequence, opts):
pes.mf.print_tree() pes.mf.print_tree()
@@ -458,5 +483,7 @@ if __name__ == '__main__':
do_add_app_inst(pes, opts) do_add_app_inst(pes, opts)
elif opts.command == 'remove-app-inst': elif opts.command == 'remove-app-inst':
do_remove_app_inst(pes, opts) do_remove_app_inst(pes, opts)
elif opts.command == 'edit-mand-srv-list':
do_edit_mand_srv_list(pes, opts)
elif opts.command == 'tree': elif opts.command == 'tree':
do_tree(pes, opts) do_tree(pes, opts)

View File

@@ -1411,6 +1411,15 @@ class ProfileElementHeader(ProfileElement):
if profile_type: if profile_type:
self.decoded['profileType'] = profile_type self.decoded['profileType'] = profile_type
def mandatory_service_add(self, service_name):
self.decoded['eUICC-Mandatory-services'][service_name] = None
def mandatory_service_remove(self, service_name):
if service_name in self.decoded['eUICC-Mandatory-services'].keys():
del self.decoded['eUICC-Mandatory-services'][service_name]
else:
raise ValueError("service not in eUICC-Mandatory-services list, cannot remove")
class ProfileElementEnd(ProfileElement): class ProfileElementEnd(ProfileElement):
type = 'end' type = 'end'
def __init__(self, decoded: Optional[dict] = None, **kwargs): def __init__(self, decoded: Optional[dict] = None, **kwargs):