diff options
Diffstat (limited to 'src/spdk/test/json_config/clear_config.py')
-rwxr-xr-x | src/spdk/test/json_config/clear_config.py | 215 |
1 files changed, 215 insertions, 0 deletions
diff --git a/src/spdk/test/json_config/clear_config.py b/src/spdk/test/json_config/clear_config.py new file mode 100755 index 000000000..bac1beebb --- /dev/null +++ b/src/spdk/test/json_config/clear_config.py @@ -0,0 +1,215 @@ +#!/usr/bin/env python3 + +import os +import sys +import argparse +import logging +sys.path.append(os.path.join(os.path.dirname(__file__), "../../scripts")) +import rpc # noqa +from rpc.client import print_dict, JSONRPCException # noqa + + +def get_bdev_name_key(bdev): + bdev_name_key = 'name' + if 'method' in bdev and bdev['method'] == 'bdev_split_create': + bdev_name_key = "base_bdev" + return bdev_name_key + + +def get_bdev_name(bdev): + bdev_name = None + if 'params' in bdev: + if 'name' in bdev['params']: + bdev_name = bdev['params']['name'] + elif 'base_name' in bdev['params']: + bdev_name = bdev['params']['base_name'] + elif 'base_bdev' in bdev['params']: + bdev_name = bdev['params']['base_bdev'] + if 'method' in bdev and bdev['method'] == 'bdev_error_create': + bdev_name = "EE_%s" % bdev_name + return bdev_name + + +def get_bdev_delete_method(bdev): + delete_method_map = {'bdev_malloc_create': "bdev_malloc_delete", + 'bdev_null_create': "bdev_null_delete", + 'bdev_rbd_create': "bdev_rbd_delete", + 'bdev_pmem_create': "bdev_pmem_delete", + 'bdev_aio_create': "bdev_aio_delete", + 'bdev_error_create': "bdev_error_delete", + 'construct_split_vbdev': "destruct_split_vbdev", + 'bdev_virtio_attach_controller': "remove_virtio_bdev", + 'bdev_crypto_create': "bdev_crypto_delete", + 'bdev_delay_create': "bdev_delay_delete", + 'bdev_passthru_create': "bdev_passthru_delete", + 'bdev_compress_create': 'bdev_compress_delete', + } + destroy_method = None + if 'method' in bdev: + construct_method = bdev['method'] + if construct_method in list(delete_method_map.keys()): + destroy_method = delete_method_map[construct_method] + + return destroy_method + + +def clear_bdev_subsystem(args, bdev_config): + rpc_bdevs = args.client.call("bdev_get_bdevs") + for bdev in bdev_config: + bdev_name_key = get_bdev_name_key(bdev) + bdev_name = get_bdev_name(bdev) + destroy_method = get_bdev_delete_method(bdev) + if destroy_method: + args.client.call(destroy_method, {bdev_name_key: bdev_name}) + + nvme_controllers = args.client.call("bdev_nvme_get_controllers") + for ctrlr in nvme_controllers: + args.client.call('bdev_nvme_detach_controller', {'name': ctrlr['name']}) + + ''' Disable and reset hotplug ''' + rpc.bdev.bdev_nvme_set_hotplug(args.client, False) + + +def get_nvmf_destroy_method(nvmf): + delete_method_map = {'nvmf_create_subsystem': "nvmf_delete_subsystem"} + try: + return delete_method_map[nvmf['method']] + except KeyError: + return None + + +def clear_nvmf_subsystem(args, nvmf_config): + for nvmf in nvmf_config: + destroy_method = get_nvmf_destroy_method(nvmf) + if destroy_method: + args.client.call(destroy_method, {'nqn': nvmf['params']['nqn']}) + + +def get_iscsi_destroy_method(iscsi): + delete_method_map = {'iscsi_create_portal_group': "iscsi_delete_portal_group", + 'iscsi_create_initiator_group': "iscsi_delete_initiator_group", + 'iscsi_create_target_node': "iscsi_delete_target_node", + 'iscsi_set_options': None + } + return delete_method_map[iscsi['method']] + + +def get_iscsi_name(iscsi): + if 'name' in iscsi['params']: + return iscsi['params']['name'] + else: + return iscsi['params']['tag'] + + +def get_iscsi_name_key(iscsi): + if iscsi['method'] == 'iscsi_create_target_node': + return "name" + else: + return 'tag' + + +def clear_iscsi_subsystem(args, iscsi_config): + for iscsi in iscsi_config: + destroy_method = get_iscsi_destroy_method(iscsi) + if destroy_method: + args.client.call(destroy_method, {get_iscsi_name_key(iscsi): get_iscsi_name(iscsi)}) + + +def get_nbd_destroy_method(nbd): + delete_method_map = {'nbd_start_disk': "nbd_stop_disk" + } + return delete_method_map[nbd['method']] + + +def clear_nbd_subsystem(args, nbd_config): + for nbd in nbd_config: + destroy_method = get_nbd_destroy_method(nbd) + if destroy_method: + args.client.call(destroy_method, {'nbd_device': nbd['params']['nbd_device']}) + + +def clear_net_framework_subsystem(args, net_framework_config): + pass + + +def clear_accel_subsystem(args, accel_config): + pass + + +def clear_interface_subsystem(args, interface_config): + pass + + +def clear_vhost_subsystem(args, vhost_config): + for vhost in reversed(vhost_config): + if 'method' in vhost: + method = vhost['method'] + if method in ['vhost_scsi_controller_add_target']: + args.client.call("vhost_scsi_controller_remove_target", + {"ctrlr": vhost['params']['ctrlr'], + "scsi_target_num": vhost['params']['scsi_target_num']}) + elif method in ['vhost_create_scsi_controller', 'vhost_create_blk_controller', + 'vhost_create_nvme_controller']: + args.client.call("vhost_delete_controller", {'ctrlr': vhost['params']['ctrlr']}) + + +def clear_vmd_subsystem(args, vmd_config): + pass + + +def clear_sock_subsystem(args, sock_config): + pass + + +def call_test_cmd(func): + def rpc_test_cmd(*args, **kwargs): + try: + func(*args, **kwargs) + except JSONRPCException as ex: + print((ex.message)) + exit(1) + return rpc_test_cmd + + +if __name__ == "__main__": + parser = argparse.ArgumentParser(description='Clear config command') + parser.add_argument('-s', dest='server_addr', default='/var/tmp/spdk.sock') + parser.add_argument('-p', dest='port', default=5260, type=int) + parser.add_argument('-t', dest='timeout', default=60.0, type=float) + parser.add_argument('-v', dest='verbose', action='store_const', const="INFO", + help='Set verbose mode to INFO', default="ERROR") + parser.add_argument('--verbose', dest='verbose', choices=['DEBUG', 'INFO', 'ERROR'], + help="""Set verbose level. """) + subparsers = parser.add_subparsers(help='RPC methods') + + @call_test_cmd + def clear_config(args): + for subsystem_item in reversed(args.client.call('framework_get_subsystems')): + args.subsystem = subsystem_item['subsystem'] + clear_subsystem(args) + + p = subparsers.add_parser('clear_config', help="""Clear configuration of all SPDK subsystems and targets using JSON RPC""") + p.set_defaults(func=clear_config) + + @call_test_cmd + def clear_subsystem(args): + config = args.client.call('framework_get_config', {"name": args.subsystem}) + if config is None: + return + if args.verbose: + print("Calling clear_%s_subsystem" % args.subsystem) + globals()["clear_%s_subsystem" % args.subsystem](args, config) + + p = subparsers.add_parser('clear_subsystem', help="""Clear configuration of SPDK subsystem using JSON RPC""") + p.add_argument('--subsystem', help="""Subsystem name""") + p.set_defaults(func=clear_subsystem) + + args = parser.parse_args() + + with rpc.client.JSONRPCClient(args.server_addr, args.port, args.timeout, log_level=getattr(logging, args.verbose.upper())) as client: + try: + args.client = client + args.func(args) + except JSONRPCException as ex: + print((ex.message)) + exit(1) |