102 lines
3.2 KiB
Python
Executable file
102 lines
3.2 KiB
Python
Executable file
#!/usr/bin/python3
|
|
# Copyright (c) 2021, Dell Inc. or its subsidiaries. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
# See the LICENSE file for details.
|
|
#
|
|
# This file is part of NVMe STorage Appliance Services (nvme-stas).
|
|
#
|
|
# Authors: Martin Belanger <Martin.Belanger@dell.com>
|
|
#
|
|
''' STorage Appliance Connector Control Utility
|
|
'''
|
|
import sys
|
|
import json
|
|
import pprint
|
|
from argparse import ArgumentParser
|
|
import dasbus.error
|
|
from dasbus.connection import SystemMessageBus
|
|
from staslib import defs
|
|
|
|
|
|
def tron(args): # pylint: disable=unused-argument
|
|
'''@brief Trace ON'''
|
|
bus = SystemMessageBus()
|
|
iface = bus.get_proxy(defs.STACD_DBUS_NAME, defs.STACD_DBUS_PATH)
|
|
iface.tron = True # pylint: disable=assigning-non-slot
|
|
print(f'tron = {iface.tron}') # Read value back from stacd and print
|
|
|
|
|
|
def troff(args): # pylint: disable=unused-argument
|
|
'''@brief Trace OFF'''
|
|
bus = SystemMessageBus()
|
|
iface = bus.get_proxy(defs.STACD_DBUS_NAME, defs.STACD_DBUS_PATH)
|
|
iface.tron = False # pylint: disable=assigning-non-slot
|
|
print(f'tron = {iface.tron}') # Read value back from stacd and print
|
|
|
|
|
|
def _extract_cid(ctrl):
|
|
return (
|
|
ctrl['transport'],
|
|
ctrl['traddr'],
|
|
ctrl['trsvcid'],
|
|
ctrl['subsysnqn'],
|
|
ctrl['host-traddr'],
|
|
ctrl['host-iface'],
|
|
ctrl['host-nqn'],
|
|
)
|
|
|
|
|
|
def status(args): # pylint: disable=unused-argument
|
|
'''@brief retrieve stacd's status information'''
|
|
bus = SystemMessageBus()
|
|
iface = bus.get_proxy(defs.STACD_DBUS_NAME, defs.STACD_DBUS_PATH)
|
|
info = json.loads(iface.process_info())
|
|
info['controllers'] = iface.list_controllers(True)
|
|
for controller in info['controllers']:
|
|
transport, traddr, trsvcid, subsysnqn, host_traddr, host_iface, host_nqn = _extract_cid(controller)
|
|
controller.update(
|
|
json.loads(iface.controller_info(transport, traddr, trsvcid, subsysnqn, host_traddr, host_iface, host_nqn))
|
|
)
|
|
|
|
print(pprint.pformat(info, width=120))
|
|
|
|
|
|
def ls(args):
|
|
'''@brief list the I/O controller's that stacd is
|
|
connected (or trying to connect) to.
|
|
'''
|
|
bus = SystemMessageBus()
|
|
iface = bus.get_proxy(defs.STACD_DBUS_NAME, defs.STACD_DBUS_PATH)
|
|
info = iface.list_controllers(args.detailed)
|
|
print(pprint.pformat(info, width=120))
|
|
|
|
|
|
PARSER = ArgumentParser(description='STorage Appliance Connector (STAC)')
|
|
PARSER.add_argument('-v', '--version', action='store_true', help='Print version, then exit', default=False)
|
|
|
|
SUBPARSER = PARSER.add_subparsers(title='Commands')
|
|
|
|
PRSR = SUBPARSER.add_parser('tron', help='Trace ON')
|
|
PRSR.set_defaults(func=tron)
|
|
|
|
PRSR = SUBPARSER.add_parser('troff', help='Trace OFF')
|
|
PRSR.set_defaults(func=troff)
|
|
|
|
PRSR = SUBPARSER.add_parser('status', help='Show runtime status information about stacd')
|
|
PRSR.set_defaults(func=status)
|
|
|
|
PRSR = SUBPARSER.add_parser('ls', help='List I/O controllers')
|
|
PRSR.add_argument(
|
|
'-d', '--detailed', action='store_true', help='Print detailed info (default: "%(default)s")', default=False
|
|
)
|
|
PRSR.set_defaults(func=ls)
|
|
|
|
ARGS = PARSER.parse_args()
|
|
if ARGS.version:
|
|
print(f'nvme-stas {defs.VERSION}')
|
|
sys.exit(0)
|
|
|
|
try:
|
|
ARGS.func(ARGS)
|
|
except dasbus.error.DBusError:
|
|
sys.exit('Unable to communicate with stacd over D-Bus. Is stacd running?')
|