xref: /spdk/scripts/rpc.py (revision cdb0726b95631d46eaf4f2e39ddb6533f150fd27)
1#!/usr/bin/env python3
2
3import logging
4import argparse
5import importlib
6import os
7import sys
8import shlex
9import json
10
11try:
12    from shlex import quote
13except ImportError:
14    from pipes import quote
15
16sys.path.append(os.path.dirname(__file__) + '/../python')
17
18import spdk.rpc as rpc  # noqa
19from spdk.rpc.client import print_dict, print_json, JSONRPCException  # noqa
20from spdk.rpc.helpers import deprecated_aliases  # noqa
21
22
23def print_array(a):
24    print(" ".join((quote(v) for v in a)))
25
26
27if __name__ == "__main__":
28    parser = argparse.ArgumentParser(
29        description='SPDK RPC command line interface', usage='%(prog)s [options]')
30    parser.add_argument('-s', dest='server_addr',
31                        help='RPC domain socket path or IP address', default='/var/tmp/spdk.sock')
32    parser.add_argument('-p', dest='port',
33                        help='RPC port number (if server_addr is IP address)',
34                        default=5260, type=int)
35    parser.add_argument('-t', dest='timeout',
36                        help='Timeout as a floating point number expressed in seconds waiting for response. Default: 60.0',
37                        default=60.0, type=float)
38    parser.add_argument('-r', dest='conn_retries',
39                        help='Retry connecting to the RPC server N times with 0.2s interval. Default: 0',
40                        default=0, type=int)
41    parser.add_argument('-v', dest='verbose', action='store_const', const="INFO",
42                        help='Set verbose mode to INFO', default="ERROR")
43    parser.add_argument('--verbose', dest='verbose', choices=['DEBUG', 'INFO', 'ERROR'],
44                        help="""Set verbose level. """)
45    parser.add_argument('--dry-run', dest='dry_run', action='store_true', help="Display request and exit")
46    parser.set_defaults(dry_run=False)
47    parser.add_argument('--server', dest='is_server', action='store_true',
48                        help="Start listening on stdin, parse each line as a regular rpc.py execution and create \
49                                a separate connection for each command. Each command's output ends with either \
50                                **STATUS=0 if the command succeeded or **STATUS=1 if it failed. --server is meant \
51                                to be used in conjunction with bash coproc, where stdin and stdout are connected to \
52                                pipes and can be used as a faster way to send RPC commands. If enabled, rpc.py \
53                                must be executed without any other parameters.")
54    parser.set_defaults(is_server=False)
55    parser.add_argument('--plugin', dest='rpc_plugin', help='Module name of plugin with additional RPC commands')
56    subparsers = parser.add_subparsers(help='RPC methods', dest='called_rpc_name', metavar='')
57
58    def framework_start_init(args):
59        rpc.framework_start_init(args.client)
60
61    p = subparsers.add_parser('framework_start_init', help='Start initialization of subsystems')
62    p.set_defaults(func=framework_start_init)
63
64    def framework_wait_init(args):
65        rpc.framework_wait_init(args.client)
66
67    p = subparsers.add_parser('framework_wait_init', help='Block until subsystems have been initialized')
68    p.set_defaults(func=framework_wait_init)
69
70    def rpc_get_methods(args):
71        print_dict(rpc.rpc_get_methods(args.client,
72                                       current=args.current,
73                                       include_aliases=args.include_aliases))
74
75    p = subparsers.add_parser('rpc_get_methods', help='Get list of supported RPC methods')
76    p.add_argument('-c', '--current', help='Get list of RPC methods only callable in the current state.', action='store_true')
77    p.add_argument('-i', '--include-aliases', help='include RPC aliases', action='store_true')
78    p.set_defaults(func=rpc_get_methods)
79
80    def spdk_get_version(args):
81        print_json(rpc.spdk_get_version(args.client))
82
83    p = subparsers.add_parser('spdk_get_version', help='Get SPDK version')
84    p.set_defaults(func=spdk_get_version)
85
86    def save_config(args):
87        rpc.save_config(args.client,
88                        sys.stdout,
89                        indent=args.indent)
90
91    p = subparsers.add_parser('save_config', help="""Write current (live) configuration of SPDK subsystems and targets to stdout.
92    """)
93    p.add_argument('-i', '--indent', help="""Indent level. Value less than 0 mean compact mode. Default indent level is 2.
94    """, type=int, default=2)
95    p.set_defaults(func=save_config)
96
97    def load_config(args):
98        rpc.load_config(args.client, args.json_conf,
99                        include_aliases=args.include_aliases)
100
101    p = subparsers.add_parser('load_config', help="""Configure SPDK subsystems and targets using JSON RPC.""")
102    p.add_argument('-i', '--include-aliases', help='include RPC aliases', action='store_true')
103    p.add_argument('-j', '--json-conf', help='Valid JSON configuration', default=sys.stdin)
104    p.set_defaults(func=load_config)
105
106    def save_subsystem_config(args):
107        rpc.save_subsystem_config(args.client,
108                                  sys.stdout,
109                                  indent=args.indent,
110                                  name=args.name)
111
112    p = subparsers.add_parser('save_subsystem_config', help="""Write current (live) configuration of SPDK subsystem to stdout.
113    """)
114    p.add_argument('-i', '--indent', help="""Indent level. Value less than 0 mean compact mode. Default indent level is 2.
115    """, type=int, default=2)
116    p.add_argument('-n', '--name', help='Name of subsystem', required=True)
117    p.set_defaults(func=save_subsystem_config)
118
119    def load_subsystem_config(args):
120        rpc.load_subsystem_config(args.client,
121                                  args.json_conf)
122
123    p = subparsers.add_parser('load_subsystem_config', help="""Configure SPDK subsystem using JSON RPC.""")
124    p.add_argument('-j', '--json-conf', help='Valid JSON configuration', default=sys.stdin)
125    p.set_defaults(func=load_subsystem_config)
126
127    # app
128    def spdk_kill_instance(args):
129        rpc.app.spdk_kill_instance(args.client,
130                                   sig_name=args.sig_name)
131
132    p = subparsers.add_parser('spdk_kill_instance', help='Send signal to instance')
133    p.add_argument('sig_name', help='signal will be sent to server.')
134    p.set_defaults(func=spdk_kill_instance)
135
136    def framework_monitor_context_switch(args):
137        enabled = None
138        if args.enable:
139            enabled = True
140        if args.disable:
141            enabled = False
142        print_dict(rpc.app.framework_monitor_context_switch(args.client,
143                                                            enabled=enabled))
144
145    p = subparsers.add_parser('framework_monitor_context_switch',
146                              help='Control whether the context switch monitor is enabled')
147    p.add_argument('-e', '--enable', action='store_true', help='Enable context switch monitoring')
148    p.add_argument('-d', '--disable', action='store_true', help='Disable context switch monitoring')
149    p.set_defaults(func=framework_monitor_context_switch)
150
151    def framework_get_reactors(args):
152        print_dict(rpc.app.framework_get_reactors(args.client))
153
154    p = subparsers.add_parser(
155        'framework_get_reactors', help='Display list of all reactors')
156    p.set_defaults(func=framework_get_reactors)
157
158    def framework_set_scheduler(args):
159        rpc.app.framework_set_scheduler(args.client,
160                                        name=args.name,
161                                        period=args.period,
162                                        load_limit=args.load_limit,
163                                        core_limit=args.core_limit,
164                                        core_busy=args.core_busy)
165
166    p = subparsers.add_parser(
167        'framework_set_scheduler', help='Select thread scheduler that will be activated and its period (experimental)')
168    p.add_argument('name', help="Name of a scheduler")
169    p.add_argument('-p', '--period', help="Scheduler period in microseconds", type=int)
170    p.add_argument('--load-limit', help="Scheduler load limit. Reserved for dynamic scheduler", type=int, required=False)
171    p.add_argument('--core-limit', help="Scheduler core limit. Reserved for dynamic scheduler", type=int, required=False)
172    p.add_argument('--core-busy', help="Scheduler core busy limit. Reserved for dynamic schedler", type=int, required=False)
173    p.set_defaults(func=framework_set_scheduler)
174
175    def framework_get_scheduler(args):
176        print_dict(rpc.app.framework_get_scheduler(args.client))
177
178    p = subparsers.add_parser(
179        'framework_get_scheduler', help='Display currently set scheduler and its properties.')
180    p.set_defaults(func=framework_get_scheduler)
181
182    # bdev
183    def bdev_set_options(args):
184        rpc.bdev.bdev_set_options(args.client,
185                                  bdev_io_pool_size=args.bdev_io_pool_size,
186                                  bdev_io_cache_size=args.bdev_io_cache_size,
187                                  bdev_auto_examine=args.bdev_auto_examine,
188                                  small_buf_pool_size=args.small_buf_pool_size,
189                                  large_buf_pool_size=args.large_buf_pool_size)
190
191    p = subparsers.add_parser('bdev_set_options',
192                              help="""Set options of bdev subsystem""")
193    p.add_argument('-p', '--bdev-io-pool-size', help='Number of bdev_io structures in shared buffer pool', type=int)
194    p.add_argument('-c', '--bdev-io-cache-size', help='Maximum number of bdev_io structures cached per thread', type=int)
195    p.add_argument('-s', '--small-buf-pool-size', help='Maximum number of small buf (i.e., 8KB) pool size', type=int)
196    p.add_argument('-l', '--large-buf-pool-size', help='Maximum number of large buf (i.e., 64KB) pool size', type=int)
197    group = p.add_mutually_exclusive_group()
198    group.add_argument('-e', '--enable-auto-examine', dest='bdev_auto_examine', help='Allow to auto examine', action='store_true')
199    group.add_argument('-d', '--disable-auto-examine', dest='bdev_auto_examine', help='Not allow to auto examine', action='store_false')
200    p.set_defaults(bdev_auto_examine=True)
201    p.set_defaults(func=bdev_set_options)
202
203    def bdev_examine(args):
204        rpc.bdev.bdev_examine(args.client,
205                              name=args.name)
206
207    p = subparsers.add_parser('bdev_examine',
208                              help="""examine a bdev if it exists, or will examine it after it is created""")
209    p.add_argument('-b', '--name', help='Name or alias of the bdev')
210    p.set_defaults(func=bdev_examine)
211
212    def bdev_wait_for_examine(args):
213        rpc.bdev.bdev_wait_for_examine(args.client)
214
215    p = subparsers.add_parser('bdev_wait_for_examine',
216                              help="""Report when all bdevs have been examined""")
217    p.set_defaults(func=bdev_wait_for_examine)
218
219    def bdev_compress_create(args):
220        print_json(rpc.bdev.bdev_compress_create(args.client,
221                                                 base_bdev_name=args.base_bdev_name,
222                                                 pm_path=args.pm_path,
223                                                 lb_size=args.lb_size))
224
225    p = subparsers.add_parser('bdev_compress_create', help='Add a compress vbdev')
226    p.add_argument('-b', '--base-bdev-name', help="Name of the base bdev")
227    p.add_argument('-p', '--pm-path', help="Path to persistent memory")
228    p.add_argument('-l', '--lb-size', help="Compressed vol logical block size (optional, if used must be 512 or 4096)", type=int)
229    p.set_defaults(func=bdev_compress_create)
230
231    def bdev_compress_delete(args):
232        rpc.bdev.bdev_compress_delete(args.client,
233                                      name=args.name)
234
235    p = subparsers.add_parser('bdev_compress_delete', help='Delete a compress disk')
236    p.add_argument('name', help='compress bdev name')
237    p.set_defaults(func=bdev_compress_delete)
238
239    def bdev_compress_set_pmd(args):
240        rpc.bdev.bdev_compress_set_pmd(args.client,
241                                       pmd=args.pmd)
242    p = subparsers.add_parser('bdev_compress_set_pmd', help='Set pmd option for a compress disk')
243    p.add_argument('-p', '--pmd', type=int, help='0 = auto-select, 1= QAT only, 2 = ISAL only, 3 = mlx5_pci only')
244    p.set_defaults(func=bdev_compress_set_pmd)
245
246    def bdev_compress_get_orphans(args):
247        print_dict(rpc.bdev.bdev_compress_get_orphans(args.client,
248                                                      name=args.name))
249    p = subparsers.add_parser(
250        'bdev_compress_get_orphans', help='Display list of orphaned compress bdevs.')
251    p.add_argument('-b', '--name', help="Name of a comp bdev. Example: COMP_Nvme0n1", required=False)
252    p.set_defaults(func=bdev_compress_get_orphans)
253
254    def bdev_crypto_create(args):
255        print_json(rpc.bdev.bdev_crypto_create(args.client,
256                                               base_bdev_name=args.base_bdev_name,
257                                               name=args.name,
258                                               crypto_pmd=args.crypto_pmd,
259                                               key=args.key,
260                                               cipher=args.cipher,
261                                               key2=args.key2))
262    p = subparsers.add_parser('bdev_crypto_create', help='Add a crypto vbdev')
263    p.add_argument('base_bdev_name', help="Name of the base bdev")
264    p.add_argument('name', help="Name of the crypto vbdev")
265    p.add_argument('crypto_pmd', help="Name of the crypto device driver")
266    p.add_argument('key', help="Key")
267    p.add_argument('-c', '--cipher', help="cipher to use, AES_CBC or AES_XTS (QAT only)")
268    p.add_argument('-k2', '--key2', help="2nd key for cipher AET_XTS", default=None)
269    p.set_defaults(func=bdev_crypto_create)
270
271    def bdev_crypto_delete(args):
272        rpc.bdev.bdev_crypto_delete(args.client,
273                                    name=args.name)
274
275    p = subparsers.add_parser('bdev_crypto_delete', help='Delete a crypto disk')
276    p.add_argument('name', help='crypto bdev name')
277    p.set_defaults(func=bdev_crypto_delete)
278
279    def bdev_ocf_create(args):
280        print_json(rpc.bdev.bdev_ocf_create(args.client,
281                                            name=args.name,
282                                            mode=args.mode,
283                                            cache_line_size=args.cache_line_size,
284                                            cache_bdev_name=args.cache_bdev_name,
285                                            core_bdev_name=args.core_bdev_name))
286    p = subparsers.add_parser('bdev_ocf_create', help='Add an OCF block device')
287    p.add_argument('name', help='Name of resulting OCF bdev')
288    p.add_argument('mode', help='OCF cache mode', choices=['wb', 'wt', 'pt', 'wa', 'wi', 'wo'])
289    p.add_argument(
290        '--cache-line-size',
291        help='OCF cache line size. The unit is KiB',
292        type=int,
293        choices=[4, 8, 16, 32, 64],
294        required=False
295    )
296    p.add_argument('cache_bdev_name', help='Name of underlying cache bdev')
297    p.add_argument('core_bdev_name', help='Name of underlying core bdev')
298    p.set_defaults(func=bdev_ocf_create)
299
300    def bdev_ocf_delete(args):
301        rpc.bdev.bdev_ocf_delete(args.client,
302                                 name=args.name)
303
304    p = subparsers.add_parser('bdev_ocf_delete', help='Delete an OCF block device')
305    p.add_argument('name', help='Name of OCF bdev')
306    p.set_defaults(func=bdev_ocf_delete)
307
308    def bdev_ocf_get_stats(args):
309        print_dict(rpc.bdev.bdev_ocf_get_stats(args.client,
310                                               name=args.name))
311    p = subparsers.add_parser('bdev_ocf_get_stats', help='Get statistics of chosen OCF block device')
312    p.add_argument('name', help='Name of OCF bdev')
313    p.set_defaults(func=bdev_ocf_get_stats)
314
315    def bdev_ocf_get_bdevs(args):
316        print_dict(rpc.bdev.bdev_ocf_get_bdevs(args.client,
317                                               name=args.name))
318    p = subparsers.add_parser('bdev_ocf_get_bdevs', help='Get list of OCF devices including unregistered ones')
319    p.add_argument('name', nargs='?', help='name of OCF vbdev or name of cache device or name of core device (optional)')
320    p.set_defaults(func=bdev_ocf_get_bdevs)
321
322    def bdev_ocf_set_cache_mode(args):
323        print_json(rpc.bdev.bdev_ocf_set_cache_mode(args.client,
324                                                    name=args.name,
325                                                    mode=args.mode))
326    p = subparsers.add_parser('bdev_ocf_set_cache_mode',
327                              help='Set cache mode of OCF block device')
328    p.add_argument('name', help='Name of OCF bdev')
329    p.add_argument('mode', help='OCF cache mode', choices=['wb', 'wt', 'pt', 'wa', 'wi', 'wo'])
330    p.set_defaults(func=bdev_ocf_set_cache_mode)
331
332    def bdev_ocf_set_seqcutoff(args):
333        rpc.bdev.bdev_ocf_set_seqcutoff(args.client,
334                                        name=args.name,
335                                        policy=args.policy,
336                                        threshold=args.threshold,
337                                        promotion_count=args.promotion_count)
338    p = subparsers.add_parser('bdev_ocf_set_seqcutoff',
339                              help='Set sequential cutoff parameters on all cores for the given OCF cache device')
340    p.add_argument('name', help='Name of OCF cache bdev')
341    p.add_argument('-t', '--threshold', type=int,
342                   help='Activation threshold [KiB]')
343    p.add_argument('-c', '--promotion-count', type=int,
344                   help='Promotion request count')
345    p.add_argument('-p', '--policy', choices=['always', 'full', 'never'], required=True,
346                   help='Sequential cutoff policy')
347    p.set_defaults(func=bdev_ocf_set_seqcutoff)
348
349    def bdev_malloc_create(args):
350        num_blocks = (args.total_size * 1024 * 1024) // args.block_size
351        print_json(rpc.bdev.bdev_malloc_create(args.client,
352                                               num_blocks=int(num_blocks),
353                                               block_size=args.block_size,
354                                               name=args.name,
355                                               uuid=args.uuid,
356                                               optimal_io_boundary=args.optimal_io_boundary))
357    p = subparsers.add_parser('bdev_malloc_create', help='Create a bdev with malloc backend')
358    p.add_argument('-b', '--name', help="Name of the bdev")
359    p.add_argument('-u', '--uuid', help="UUID of the bdev")
360    p.add_argument(
361        'total_size', help='Size of malloc bdev in MB (float > 0)', type=float)
362    p.add_argument('block_size', help='Block size for this bdev', type=int)
363    p.add_argument('-o', '--optimal-io-boundary', help="""Split on optimal IO boundary, in number of
364    blocks, default 0 (disabled)""", type=int)
365    p.set_defaults(func=bdev_malloc_create)
366
367    def bdev_malloc_delete(args):
368        rpc.bdev.bdev_malloc_delete(args.client,
369                                    name=args.name)
370
371    p = subparsers.add_parser('bdev_malloc_delete', help='Delete a malloc disk')
372    p.add_argument('name', help='malloc bdev name')
373    p.set_defaults(func=bdev_malloc_delete)
374
375    def bdev_null_create(args):
376        num_blocks = (args.total_size * 1024 * 1024) // args.block_size
377        if args.dif_type and not args.md_size:
378            print("ERROR: --md-size must be > 0 when --dif-type is > 0")
379            exit(1)
380        print_json(rpc.bdev.bdev_null_create(args.client,
381                                             num_blocks=num_blocks,
382                                             block_size=args.block_size,
383                                             name=args.name,
384                                             uuid=args.uuid,
385                                             md_size=args.md_size,
386                                             dif_type=args.dif_type,
387                                             dif_is_head_of_md=args.dif_is_head_of_md))
388
389    p = subparsers.add_parser('bdev_null_create', help='Add a bdev with null backend')
390    p.add_argument('name', help='Block device name')
391    p.add_argument('-u', '--uuid', help='UUID of the bdev')
392    p.add_argument('total_size', help='Size of null bdev in MB (int > 0). Includes only data blocks.', type=int)
393    p.add_argument('block_size', help='Block size for this bdev.'
394                                      'Should be a sum of block size and metadata size if --md-size is used.', type=int)
395    p.add_argument('-m', '--md-size', type=int,
396                   help='Metadata size for this bdev. Default=0.')
397    p.add_argument('-t', '--dif-type', type=int, choices=[0, 1, 2, 3],
398                   help='Protection information type. Parameter --md-size needs'
399                        'to be set along --dif-type. Default=0 - no protection.')
400    p.add_argument('-d', '--dif-is-head-of-md', action='store_true',
401                   help='Protection information is in the first 8 bytes of metadata. Default=false.')
402    p.set_defaults(func=bdev_null_create)
403
404    def bdev_null_delete(args):
405        rpc.bdev.bdev_null_delete(args.client,
406                                  name=args.name)
407
408    p = subparsers.add_parser('bdev_null_delete', help='Delete a null bdev')
409    p.add_argument('name', help='null bdev name')
410    p.set_defaults(func=bdev_null_delete)
411
412    def bdev_null_resize(args):
413        print_json(rpc.bdev.bdev_null_resize(args.client,
414                                             name=args.name,
415                                             new_size=int(args.new_size)))
416
417    p = subparsers.add_parser('bdev_null_resize',
418                              help='Resize a null bdev')
419    p.add_argument('name', help='null bdev name')
420    p.add_argument('new_size', help='new bdev size for resize operation. The unit is MiB')
421    p.set_defaults(func=bdev_null_resize)
422
423    def bdev_aio_create(args):
424        print_json(rpc.bdev.bdev_aio_create(args.client,
425                                            filename=args.filename,
426                                            name=args.name,
427                                            block_size=args.block_size))
428
429    p = subparsers.add_parser('bdev_aio_create', help='Add a bdev with aio backend')
430    p.add_argument('filename', help='Path to device or file (ex: /dev/sda)')
431    p.add_argument('name', help='Block device name')
432    p.add_argument('block_size', help='Block size for this bdev', type=int, nargs='?')
433    p.set_defaults(func=bdev_aio_create)
434
435    def bdev_aio_rescan(args):
436        print_json(rpc.bdev.bdev_aio_rescan(args.client,
437                                            name=args.name))
438
439    p = subparsers.add_parser('bdev_aio_rescan', help='Rescan a bdev size with aio backend')
440    p.add_argument('name', help='Block device name')
441    p.set_defaults(func=bdev_aio_rescan)
442
443    def bdev_aio_delete(args):
444        rpc.bdev.bdev_aio_delete(args.client,
445                                 name=args.name)
446
447    p = subparsers.add_parser('bdev_aio_delete', help='Delete an aio disk')
448    p.add_argument('name', help='aio bdev name')
449    p.set_defaults(func=bdev_aio_delete)
450
451    def bdev_uring_create(args):
452        print_json(rpc.bdev.bdev_uring_create(args.client,
453                                              filename=args.filename,
454                                              name=args.name,
455                                              block_size=args.block_size))
456
457    p = subparsers.add_parser('bdev_uring_create', help='Create a bdev with io_uring backend')
458    p.add_argument('filename', help='Path to device or file (ex: /dev/nvme0n1)')
459    p.add_argument('name', help='bdev name')
460    p.add_argument('block_size', help='Block size for this bdev', type=int, nargs='?')
461    p.set_defaults(func=bdev_uring_create)
462
463    def bdev_uring_delete(args):
464        rpc.bdev.bdev_uring_delete(args.client,
465                                   name=args.name)
466
467    p = subparsers.add_parser('bdev_uring_delete', help='Delete a uring bdev')
468    p.add_argument('name', help='uring bdev name')
469    p.set_defaults(func=bdev_uring_delete)
470
471    def bdev_xnvme_create(args):
472        print_json(rpc.bdev.bdev_xnvme_create(args.client,
473                                              filename=args.filename,
474                                              name=args.name,
475                                              io_mechanism=args.io_mechanism))
476
477    p = subparsers.add_parser('bdev_xnvme_create', help='Create a bdev with xNVMe backend')
478    p.add_argument('filename', help='Path to device or file (ex: /dev/nvme0n1)')
479    p.add_argument('name', help='name of xNVMe bdev to create')
480    p.add_argument('io_mechanism', help='IO mechanism to use (ex: libaio, io_uring, io_uring_cmd, etc.)')
481    p.set_defaults(func=bdev_xnvme_create)
482
483    def bdev_xnvme_delete(args):
484        rpc.bdev.bdev_xnvme_delete(args.client,
485                                   name=args.name)
486
487    p = subparsers.add_parser('bdev_xnvme_delete', help='Delete a xNVMe bdev')
488    p.add_argument('name', help='xNVMe bdev name')
489    p.set_defaults(func=bdev_xnvme_delete)
490
491    def bdev_nvme_set_options(args):
492        rpc.bdev.bdev_nvme_set_options(args.client,
493                                       action_on_timeout=args.action_on_timeout,
494                                       timeout_us=args.timeout_us,
495                                       timeout_admin_us=args.timeout_admin_us,
496                                       keep_alive_timeout_ms=args.keep_alive_timeout_ms,
497                                       retry_count=args.retry_count,
498                                       arbitration_burst=args.arbitration_burst,
499                                       low_priority_weight=args.low_priority_weight,
500                                       medium_priority_weight=args.medium_priority_weight,
501                                       high_priority_weight=args.high_priority_weight,
502                                       nvme_adminq_poll_period_us=args.nvme_adminq_poll_period_us,
503                                       nvme_ioq_poll_period_us=args.nvme_ioq_poll_period_us,
504                                       io_queue_requests=args.io_queue_requests,
505                                       delay_cmd_submit=args.delay_cmd_submit,
506                                       transport_retry_count=args.transport_retry_count,
507                                       bdev_retry_count=args.bdev_retry_count,
508                                       transport_ack_timeout=args.transport_ack_timeout,
509                                       ctrlr_loss_timeout_sec=args.ctrlr_loss_timeout_sec,
510                                       reconnect_delay_sec=args.reconnect_delay_sec,
511                                       fast_io_fail_timeout_sec=args.fast_io_fail_timeout_sec,
512                                       disable_auto_failback=args.disable_auto_failback)
513
514    p = subparsers.add_parser('bdev_nvme_set_options',
515                              help='Set options for the bdev nvme type. This is startup command.')
516    p.add_argument('-a', '--action-on-timeout',
517                   help="Action to take on command time out. Valid values are: none, reset, abort")
518    p.add_argument('-t', '--timeout-us',
519                   help="Timeout for each command, in microseconds. If 0, don't track timeouts.", type=int)
520    p.add_argument('--timeout-admin-us',
521                   help="Timeout for each admin command, in microseconds. If 0, treat same as io timeouts.", type=int)
522    p.add_argument('-k', '--keep-alive-timeout-ms',
523                   help="Keep alive timeout period in millisecond. If 0, disable keep-alive.", type=int)
524    p.add_argument('-n', '--retry-count',
525                   help='the number of attempts per I/O when an I/O fails. (deprecated, please use --transport-retry-count.)', type=int)
526    p.add_argument('--arbitration-burst',
527                   help='the value is expressed as a power of two', type=int)
528    p.add_argument('--low-priority-weight',
529                   help='the maximum number of commands that the controller may launch at one time from a low priority queue', type=int)
530    p.add_argument('--medium-priority-weight',
531                   help='the maximum number of commands that the controller may launch at one time from a medium priority queue', type=int)
532    p.add_argument('--high-priority-weight',
533                   help='the maximum number of commands that the controller may launch at one time from a high priority queue', type=int)
534    p.add_argument('-p', '--nvme-adminq-poll-period-us',
535                   help='How often the admin queue is polled for asynchronous events', type=int)
536    p.add_argument('-i', '--nvme-ioq-poll-period-us',
537                   help='How often to poll I/O queues for completions', type=int)
538    p.add_argument('-s', '--io-queue-requests',
539                   help='The number of requests allocated for each NVMe I/O queue. Default: 512', type=int)
540    p.add_argument('-d', '--disable-delay-cmd-submit',
541                   help='Disable delaying NVMe command submission, i.e. no batching of multiple commands',
542                   action='store_false', dest='delay_cmd_submit')
543    p.add_argument('-c', '--transport-retry-count',
544                   help='the number of attempts per I/O in the transport layer when an I/O fails.', type=int)
545    p.add_argument('-r', '--bdev-retry-count',
546                   help='the number of attempts per I/O in the bdev layer when an I/O fails. -1 means infinite retries.', type=int)
547    p.add_argument('-e', '--transport-ack-timeout',
548                   help="""Time to wait ack until packet retransmission for RDMA or until closes connection for TCP.
549                   Range 0-31 where 0 is driver-specific default value.""", type=int)
550    p.add_argument('-l', '--ctrlr-loss-timeout-sec',
551                   help="""Time to wait until ctrlr is reconnected before deleting ctrlr.
552                   -1 means infinite reconnect retries. 0 means no reconnect retry.
553                   If reconnect_delay_sec is zero, ctrlr_loss_timeout_sec has to be zero.
554                   If reconnect_delay_sec is non-zero, ctrlr_loss_timeout_sec has to be -1 or not less than
555                   reconnect_delay_sec.
556                   This can be overridden by bdev_nvme_attach_controller.""",
557                   type=int)
558    p.add_argument('-o', '--reconnect-delay-sec',
559                   help="""Time to delay a reconnect retry.
560                   If ctrlr_loss_timeout_sec is zero, reconnect_delay_sec has to be zero.
561                   If ctrlr_loss_timeout_sec is -1, reconnect_delay_sec has to be non-zero.
562                   If ctrlr_loss_timeout_sec is not -1 or zero, reconnect_delay_sec has to be non-zero and
563                   less than ctrlr_loss_timeout_sec.
564                   This can be overridden by bdev_nvme_attach_controller.""",
565                   type=int)
566    p.add_argument('-u', '--fast-io-fail-timeout-sec',
567                   help="""Time to wait until ctrlr is reconnected before failing I/O to ctrlr.
568                   0 means no such timeout.
569                   If fast_io_fail_timeout_sec is not zero, it has to be not less than reconnect_delay_sec and
570                   less than ctrlr_loss_timeout_sec if ctrlr_loss_timeout_sec is not -1.
571                   This can be overridden by bdev_nvme_attach_controller.""",
572                   type=int)
573    p.add_argument('-f', '--disable-auto-failback',
574                   help="""Disable automatic failback. bdev_nvme_set_preferred_path can be used to do manual failback.
575                   By default, immediately failback to the preferred I/O path if it restored.""",
576                   action='store_true')
577
578    p.set_defaults(func=bdev_nvme_set_options)
579
580    def bdev_nvme_set_hotplug(args):
581        rpc.bdev.bdev_nvme_set_hotplug(args.client, enable=args.enable, period_us=args.period_us)
582
583    p = subparsers.add_parser('bdev_nvme_set_hotplug', help='Set hotplug options for bdev nvme type.')
584    p.add_argument('-d', '--disable', dest='enable', default=False, action='store_false', help="Disable hotplug (default)")
585    p.add_argument('-e', '--enable', dest='enable', action='store_true', help="Enable hotplug")
586    p.add_argument('-r', '--period-us',
587                   help='How often the hotplug is processed for insert and remove events', type=int)
588    p.set_defaults(func=bdev_nvme_set_hotplug)
589
590    def bdev_nvme_attach_controller(args):
591        print_array(rpc.bdev.bdev_nvme_attach_controller(args.client,
592                                                         name=args.name,
593                                                         trtype=args.trtype,
594                                                         traddr=args.traddr,
595                                                         adrfam=args.adrfam,
596                                                         trsvcid=args.trsvcid,
597                                                         priority=args.priority,
598                                                         subnqn=args.subnqn,
599                                                         hostnqn=args.hostnqn,
600                                                         hostaddr=args.hostaddr,
601                                                         hostsvcid=args.hostsvcid,
602                                                         prchk_reftag=args.prchk_reftag,
603                                                         prchk_guard=args.prchk_guard,
604                                                         hdgst=args.hdgst,
605                                                         ddgst=args.ddgst,
606                                                         fabrics_timeout=args.fabrics_timeout,
607                                                         multipath=args.multipath,
608                                                         num_io_queues=args.num_io_queues,
609                                                         ctrlr_loss_timeout_sec=args.ctrlr_loss_timeout_sec,
610                                                         reconnect_delay_sec=args.reconnect_delay_sec,
611                                                         fast_io_fail_timeout_sec=args.fast_io_fail_timeout_sec))
612
613    p = subparsers.add_parser('bdev_nvme_attach_controller', help='Add bdevs with nvme backend')
614    p.add_argument('-b', '--name', help="Name of the NVMe controller, prefix for each bdev name", required=True)
615    p.add_argument('-t', '--trtype',
616                   help='NVMe-oF target trtype: e.g., rdma, pcie', required=True)
617    p.add_argument('-a', '--traddr',
618                   help='NVMe-oF target address: e.g., an ip address or BDF', required=True)
619    p.add_argument('-f', '--adrfam',
620                   help='NVMe-oF target adrfam: e.g., ipv4, ipv6, ib, fc, intra_host')
621    p.add_argument('-s', '--trsvcid',
622                   help='NVMe-oF target trsvcid: e.g., a port number')
623    p.add_argument('-p', '--priority',
624                   help='NVMe-oF connection priority: e.g., a priority number')
625    p.add_argument('-n', '--subnqn', help='NVMe-oF target subnqn')
626    p.add_argument('-q', '--hostnqn', help='NVMe-oF host subnqn')
627    p.add_argument('-i', '--hostaddr',
628                   help='NVMe-oF host address: e.g., an ip address')
629    p.add_argument('-c', '--hostsvcid',
630                   help='NVMe-oF host svcid: e.g., a port number')
631    p.add_argument('-r', '--prchk-reftag',
632                   help='Enable checking of PI reference tag for I/O processing.', action='store_true')
633    p.add_argument('-g', '--prchk-guard',
634                   help='Enable checking of PI guard for I/O processing.', action='store_true')
635    p.add_argument('-e', '--hdgst',
636                   help='Enable TCP header digest.', action='store_true')
637    p.add_argument('-d', '--ddgst',
638                   help='Enable TCP data digest.', action='store_true')
639    p.add_argument('--fabrics-timeout', type=int, help='Fabrics connect timeout in microseconds')
640    p.add_argument('-x', '--multipath', help='Set multipath behavior (disable, failover, multipath)')
641    p.add_argument('--num-io-queues', type=int, help='Set the number of IO queues to request during initialization.')
642    p.add_argument('-l', '--ctrlr-loss-timeout-sec',
643                   help="""Time to wait until ctrlr is reconnected before deleting ctrlr.
644                   -1 means infinite reconnect retries. 0 means no reconnect retry.
645                   If reconnect_delay_sec is zero, ctrlr_loss_timeout_sec has to be zero.
646                   If reconnect_delay_sec is non-zero, ctrlr_loss_timeout_sec has to be -1 or not less than
647                   reconnect_delay_sec.""",
648                   type=int)
649    p.add_argument('-o', '--reconnect-delay-sec',
650                   help="""Time to delay a reconnect retry.
651                   If ctrlr_loss_timeout_sec is zero, reconnect_delay_sec has to be zero.
652                   If ctrlr_loss_timeout_sec is -1, reconnect_delay_sec has to be non-zero.
653                   If ctrlr_loss_timeout_sec is not -1 or zero, reconnect_delay_sec has to be non-zero and
654                   less than ctrlr_loss_timeout_sec.""",
655                   type=int)
656    p.add_argument('-u', '--fast-io-fail-timeout-sec',
657                   help="""Time to wait until ctrlr is reconnected before failing I/O to ctrlr.
658                   0 means no such timeout.
659                   If fast_io_fail_timeout_sec is not zero, it has to be not less than reconnect_delay_sec and
660                   less than ctrlr_loss_timeout_sec if ctrlr_loss_timeout_sec is not -1.""",
661                   type=int)
662    p.set_defaults(func=bdev_nvme_attach_controller)
663
664    def bdev_nvme_get_controllers(args):
665        print_dict(rpc.nvme.bdev_nvme_get_controllers(args.client,
666                                                      name=args.name))
667
668    p = subparsers.add_parser(
669        'bdev_nvme_get_controllers', help='Display current NVMe controllers list or required NVMe controller')
670    p.add_argument('-n', '--name', help="Name of the NVMe controller. Example: Nvme0", required=False)
671    p.set_defaults(func=bdev_nvme_get_controllers)
672
673    def bdev_nvme_detach_controller(args):
674        rpc.bdev.bdev_nvme_detach_controller(args.client,
675                                             name=args.name,
676                                             trtype=args.trtype,
677                                             traddr=args.traddr,
678                                             adrfam=args.adrfam,
679                                             trsvcid=args.trsvcid,
680                                             subnqn=args.subnqn,
681                                             hostaddr=args.hostaddr,
682                                             hostsvcid=args.hostsvcid)
683
684    p = subparsers.add_parser('bdev_nvme_detach_controller',
685                              help='Detach an NVMe controller and delete any associated bdevs')
686    p.add_argument('name', help="Name of the controller")
687    p.add_argument('-t', '--trtype',
688                   help='NVMe-oF target trtype: e.g., rdma, pcie')
689    p.add_argument('-a', '--traddr',
690                   help='NVMe-oF target address: e.g., an ip address or BDF')
691    p.add_argument('-f', '--adrfam',
692                   help='NVMe-oF target adrfam: e.g., ipv4, ipv6, ib, fc, intra_host')
693    p.add_argument('-s', '--trsvcid',
694                   help='NVMe-oF target trsvcid: e.g., a port number')
695    p.add_argument('-n', '--subnqn', help='NVMe-oF target subnqn')
696    p.add_argument('-i', '--hostaddr',
697                   help='NVMe-oF host address: e.g., an ip address')
698    p.add_argument('-c', '--hostsvcid',
699                   help='NVMe-oF host svcid: e.g., a port number')
700    p.set_defaults(func=bdev_nvme_detach_controller)
701
702    def bdev_nvme_reset_controller(args):
703        rpc.bdev.bdev_nvme_reset_controller(args.client, name=args.name)
704
705    p = subparsers.add_parser('bdev_nvme_reset_controller',
706                              help='Reset an NVMe controller')
707    p.add_argument('name', help="Name of the NVMe controller")
708    p.set_defaults(func=bdev_nvme_reset_controller)
709
710    def bdev_nvme_start_discovery(args):
711        rpc.bdev.bdev_nvme_start_discovery(args.client,
712                                           name=args.name,
713                                           trtype=args.trtype,
714                                           traddr=args.traddr,
715                                           adrfam=args.adrfam,
716                                           trsvcid=args.trsvcid,
717                                           hostnqn=args.hostnqn,
718                                           wait_for_attach=args.wait_for_attach,
719                                           attach_timeout_ms=args.attach_timeout_ms,
720                                           ctrlr_loss_timeout_sec=args.ctrlr_loss_timeout_sec,
721                                           reconnect_delay_sec=args.reconnect_delay_sec,
722                                           fast_io_fail_timeout_sec=args.fast_io_fail_timeout_sec)
723
724    p = subparsers.add_parser('bdev_nvme_start_discovery', help='Start automatic discovery')
725    p.add_argument('-b', '--name', help="Name of the NVMe controller prefix for each bdev name", required=True)
726    p.add_argument('-t', '--trtype',
727                   help='NVMe-oF target trtype: e.g., rdma, pcie', required=True)
728    p.add_argument('-a', '--traddr',
729                   help='NVMe-oF target address: e.g., an ip address or BDF', required=True)
730    p.add_argument('-f', '--adrfam',
731                   help='NVMe-oF target adrfam: e.g., ipv4, ipv6, ib, fc, intra_host')
732    p.add_argument('-s', '--trsvcid',
733                   help='NVMe-oF target trsvcid: e.g., a port number')
734    p.add_argument('-q', '--hostnqn', help='NVMe-oF host subnqn')
735    p.add_argument('-w', '--wait-for-attach', action='store_true',
736                   help='Do not complete RPC until all discovered NVM subsystems are attached')
737    p.add_argument('-T', '--attach-timeout-ms', type=int, required=False,
738                   help="""Time to wait until the discovery and all discovered NVM subsystems
739                        are attached (default: 0, meaning wait indefinitely).  Automatically
740                        selects the --wait-for-attach option.""")
741    p.add_argument('-l', '--ctrlr-loss-timeout-sec',
742                   help="""Time to wait until ctrlr is reconnected before deleting ctrlr.
743                   -1 means infinite reconnect retries. 0 means no reconnect retry.
744                   If reconnect_delay_sec is zero, ctrlr_loss_timeout_sec has to be zero.
745                   If reconnect_delay_sec is non-zero, ctrlr_loss_timeout_sec has to be -1 or not less than
746                   reconnect_delay_sec.""",
747                   type=int)
748    p.add_argument('-o', '--reconnect-delay-sec',
749                   help="""Time to delay a reconnect retry.
750                   If ctrlr_loss_timeout_sec is zero, reconnect_delay_sec has to be zero.
751                   If ctrlr_loss_timeout_sec is -1, reconnect_delay_sec has to be non-zero.
752                   If ctrlr_loss_timeout_sec is not -1 or zero, reconnect_delay_sec has to be non-zero and
753                   less than ctrlr_loss_timeout_sec.""",
754                   type=int)
755    p.add_argument('-u', '--fast-io-fail-timeout-sec',
756                   help="""Time to wait until ctrlr is reconnected before failing I/O to ctrlr.
757                   0 means no such timeout.
758                   If fast_io_fail_timeout_sec is not zero, it has to be not less than reconnect_delay_sec and
759                   less than ctrlr_loss_timeout_sec if ctrlr_loss_timeout_sec is not -1.""",
760                   type=int)
761    p.set_defaults(func=bdev_nvme_start_discovery)
762
763    def bdev_nvme_stop_discovery(args):
764        rpc.bdev.bdev_nvme_stop_discovery(args.client, name=args.name)
765
766    p = subparsers.add_parser('bdev_nvme_stop_discovery', help='Stop automatic discovery')
767    p.add_argument('-b', '--name', help="Name of the service to stop", required=True)
768    p.set_defaults(func=bdev_nvme_stop_discovery)
769
770    def bdev_nvme_get_discovery_info(args):
771        print_dict(rpc.bdev.bdev_nvme_get_discovery_info(args.client))
772
773    p = subparsers.add_parser('bdev_nvme_get_discovery_info', help='Get information about the automatic discovery')
774    p.set_defaults(func=bdev_nvme_get_discovery_info)
775
776    def bdev_nvme_get_io_paths(args):
777        print_dict(rpc.bdev.bdev_nvme_get_io_paths(args.client, name=args.name))
778
779    p = subparsers.add_parser('bdev_nvme_get_io_paths', help='Display active I/O paths')
780    p.add_argument('-n', '--name', help="Name of the NVMe bdev", required=False)
781    p.set_defaults(func=bdev_nvme_get_io_paths)
782
783    def bdev_nvme_set_preferred_path(args):
784        rpc.bdev.bdev_nvme_set_preferred_path(args.client,
785                                              name=args.name,
786                                              cntlid=args.cntlid)
787
788    p = subparsers.add_parser('bdev_nvme_set_preferred_path',
789                              help="""Set the preferred I/O path for an NVMe bdev when in multipath mode""")
790    p.add_argument('-b', '--name', help='Name of the NVMe bdev', required=True)
791    p.add_argument('-c', '--cntlid', help='NVMe-oF controller ID', type=int, required=True)
792    p.set_defaults(func=bdev_nvme_set_preferred_path)
793
794    def bdev_nvme_set_multipath_policy(args):
795        rpc.bdev.bdev_nvme_set_multipath_policy(args.client,
796                                                name=args.name,
797                                                policy=args.policy)
798
799    p = subparsers.add_parser('bdev_nvme_set_multipath_policy',
800                              help="""Set multipath policy of the NVMe bdev""")
801    p.add_argument('-b', '--name', help='Name of the NVMe bdev', required=True)
802    p.add_argument('-p', '--policy', help='Multipath policy (active_passive or active_active)', required=True)
803    p.set_defaults(func=bdev_nvme_set_multipath_policy)
804
805    def bdev_nvme_cuse_register(args):
806        rpc.bdev.bdev_nvme_cuse_register(args.client,
807                                         name=args.name)
808
809    p = subparsers.add_parser('bdev_nvme_cuse_register',
810                              help='Register CUSE devices on NVMe controller')
811    p.add_argument('-n', '--name',
812                   help='Name of the NVMe controller. Example: Nvme0', required=True)
813    p.set_defaults(func=bdev_nvme_cuse_register)
814
815    def bdev_nvme_cuse_unregister(args):
816        rpc.bdev.bdev_nvme_cuse_unregister(args.client,
817                                           name=args.name)
818
819    p = subparsers.add_parser('bdev_nvme_cuse_unregister',
820                              help='Unregister CUSE devices on NVMe controller')
821    p.add_argument('-n', '--name',
822                   help='Name of the NVMe controller. Example: Nvme0', required=True)
823    p.set_defaults(func=bdev_nvme_cuse_unregister)
824
825    def bdev_zone_block_create(args):
826        print_json(rpc.bdev.bdev_zone_block_create(args.client,
827                                                   name=args.name,
828                                                   base_bdev=args.base_bdev,
829                                                   zone_capacity=args.zone_capacity,
830                                                   optimal_open_zones=args.optimal_open_zones))
831
832    p = subparsers.add_parser('bdev_zone_block_create',
833                              help='Create virtual zone namespace device with block device backend')
834    p.add_argument('-b', '--name', help="Name of the zone device", required=True)
835    p.add_argument('-n', '--base-bdev', help='Name of underlying, non-zoned bdev', required=True)
836    p.add_argument('-z', '--zone-capacity', help='Surfaced zone capacity in blocks', type=int, required=True)
837    p.add_argument('-o', '--optimal-open-zones', help='Number of zones required to reach optimal write speed', type=int, required=True)
838    p.set_defaults(func=bdev_zone_block_create)
839
840    def bdev_zone_block_delete(args):
841        rpc.bdev.bdev_zone_block_delete(args.client,
842                                        name=args.name)
843
844    p = subparsers.add_parser('bdev_zone_block_delete', help='Delete a virtual zone namespace device')
845    p.add_argument('name', help='Virtual zone bdev name')
846    p.set_defaults(func=bdev_zone_block_delete)
847
848    def bdev_rbd_register_cluster(args):
849        config_param = None
850        if args.config_param:
851            config_param = {}
852            for entry in args.config_param:
853                parts = entry.split('=', 1)
854                if len(parts) != 2:
855                    raise Exception('--config %s not in key=value form' % entry)
856                config_param[parts[0]] = parts[1]
857        print_json(rpc.bdev.bdev_rbd_register_cluster(args.client,
858                                                      name=args.name,
859                                                      user=args.user,
860                                                      config_param=config_param,
861                                                      config_file=args.config_file,
862                                                      key_file=args.key_file))
863
864    p = subparsers.add_parser('bdev_rbd_register_cluster',
865                              help='Add a Rados cluster with ceph rbd backend')
866    p.add_argument('name', help="Name of the Rados cluster only known to rbd bdev")
867    p.add_argument('--user', help="Ceph user name (i.e. admin, not client.admin)", required=False)
868    p.add_argument('--config-param', action='append', metavar='key=value',
869                   help="adds a key=value configuration option for rados_conf_set (default: rely on config file)")
870    p.add_argument('--config-file', help="The file path of the Rados configuration file", required=False)
871    p.add_argument('--key-file', help="The file path of the Rados keyring file", required=False)
872    p.set_defaults(func=bdev_rbd_register_cluster)
873
874    def bdev_rbd_unregister_cluster(args):
875        rpc.bdev.bdev_rbd_unregister_cluster(args.client, name=args.name)
876
877    p = subparsers.add_parser('bdev_rbd_unregister_cluster',
878                              help='Unregister a Rados cluster object')
879    p.add_argument('name', help='Name of the Rados Cluster only known to rbd bdev')
880    p.set_defaults(func=bdev_rbd_unregister_cluster)
881
882    def bdev_rbd_get_clusters_info(args):
883        print_json(rpc.bdev.bdev_rbd_get_clusters_info(args.client, name=args.name))
884
885    p = subparsers.add_parser('bdev_rbd_get_clusters_info',
886                              help='Display registered Rados Cluster names and related info')
887    p.add_argument('-b', '--name', help="Name of the registered Rados Cluster Name. Example: Cluster1", required=False)
888    p.set_defaults(func=bdev_rbd_get_clusters_info)
889
890    def bdev_rbd_create(args):
891        config = None
892        if args.config:
893            config = {}
894            for entry in args.config:
895                parts = entry.split('=', 1)
896                if len(parts) != 2:
897                    raise Exception('--config %s not in key=value form' % entry)
898                config[parts[0]] = parts[1]
899        print_json(rpc.bdev.bdev_rbd_create(args.client,
900                                            name=args.name,
901                                            user=args.user,
902                                            config=config,
903                                            pool_name=args.pool_name,
904                                            rbd_name=args.rbd_name,
905                                            block_size=args.block_size,
906                                            cluster_name=args.cluster_name,
907                                            uuid=args.uuid))
908
909    p = subparsers.add_parser('bdev_rbd_create', help='Add a bdev with ceph rbd backend')
910    p.add_argument('-b', '--name', help="Name of the bdev", required=False)
911    p.add_argument('--user', help="Ceph user name (i.e. admin, not client.admin)", required=False)
912    p.add_argument('--config', action='append', metavar='key=value',
913                   help="adds a key=value configuration option for rados_conf_set (default: rely on config file)")
914    p.add_argument('pool_name', help='rbd pool name')
915    p.add_argument('rbd_name', help='rbd image name')
916    p.add_argument('block_size', help='rbd block size', type=int)
917    p.add_argument('-c', '--cluster-name', help="cluster name to identify the Rados cluster", required=False)
918    p.add_argument('-u', '--uuid', help="UUID of the bdev")
919    p.set_defaults(func=bdev_rbd_create)
920
921    def bdev_rbd_delete(args):
922        rpc.bdev.bdev_rbd_delete(args.client,
923                                 name=args.name)
924
925    p = subparsers.add_parser('bdev_rbd_delete', help='Delete a rbd bdev')
926    p.add_argument('name', help='rbd bdev name')
927    p.set_defaults(func=bdev_rbd_delete)
928
929    def bdev_rbd_resize(args):
930        print_json(rpc.bdev.bdev_rbd_resize(args.client,
931                                            name=args.name,
932                                            new_size=int(args.new_size)))
933
934    p = subparsers.add_parser('bdev_rbd_resize',
935                              help='Resize a rbd bdev')
936    p.add_argument('name', help='rbd bdev name')
937    p.add_argument('new_size', help='new bdev size for resize operation. The unit is MiB')
938    p.set_defaults(func=bdev_rbd_resize)
939
940    def bdev_delay_create(args):
941        print_json(rpc.bdev.bdev_delay_create(args.client,
942                                              base_bdev_name=args.base_bdev_name,
943                                              name=args.name,
944                                              avg_read_latency=args.avg_read_latency,
945                                              p99_read_latency=args.nine_nine_read_latency,
946                                              avg_write_latency=args.avg_write_latency,
947                                              p99_write_latency=args.nine_nine_write_latency))
948
949    p = subparsers.add_parser('bdev_delay_create',
950                              help='Add a delay bdev on existing bdev')
951    p.add_argument('-b', '--base-bdev-name', help="Name of the existing bdev", required=True)
952    p.add_argument('-d', '--name', help="Name of the delay bdev", required=True)
953    p.add_argument('-r', '--avg-read-latency',
954                   help="Average latency to apply before completing read ops (in microseconds)", required=True, type=int)
955    p.add_argument('-t', '--nine-nine-read-latency',
956                   help="latency to apply to 1 in 100 read ops (in microseconds)", required=True, type=int)
957    p.add_argument('-w', '--avg-write-latency',
958                   help="Average latency to apply before completing write ops (in microseconds)", required=True, type=int)
959    p.add_argument('-n', '--nine-nine-write-latency',
960                   help="latency to apply to 1 in 100 write ops (in microseconds)", required=True, type=int)
961    p.set_defaults(func=bdev_delay_create)
962
963    def bdev_delay_delete(args):
964        rpc.bdev.bdev_delay_delete(args.client,
965                                   name=args.name)
966
967    p = subparsers.add_parser('bdev_delay_delete', help='Delete a delay bdev')
968    p.add_argument('name', help='delay bdev name')
969    p.set_defaults(func=bdev_delay_delete)
970
971    def bdev_delay_update_latency(args):
972        print_json(rpc.bdev.bdev_delay_update_latency(args.client,
973                                                      delay_bdev_name=args.delay_bdev_name,
974                                                      latency_type=args.latency_type,
975                                                      latency_us=args.latency_us))
976    p = subparsers.add_parser('bdev_delay_update_latency',
977                              help='Update one of the latency values for a given delay bdev')
978    p.add_argument('delay_bdev_name', help='The name of the given delay bdev')
979    p.add_argument('latency_type', help='one of: avg_read, avg_write, p99_read, p99_write. No other values accepted.')
980    p.add_argument('latency_us', help='new latency value in microseconds.', type=int)
981    p.set_defaults(func=bdev_delay_update_latency)
982
983    def bdev_error_create(args):
984        print_json(rpc.bdev.bdev_error_create(args.client,
985                                              base_name=args.base_name))
986
987    p = subparsers.add_parser('bdev_error_create', help='Add bdev with error injection backend')
988    p.add_argument('base_name', help='base bdev name')
989    p.set_defaults(func=bdev_error_create)
990
991    def bdev_error_delete(args):
992        rpc.bdev.bdev_error_delete(args.client,
993                                   name=args.name)
994
995    p = subparsers.add_parser('bdev_error_delete', help='Delete an error bdev')
996    p.add_argument('name', help='error bdev name')
997    p.set_defaults(func=bdev_error_delete)
998
999    def bdev_iscsi_create(args):
1000        print_json(rpc.bdev.bdev_iscsi_create(args.client,
1001                                              name=args.name,
1002                                              url=args.url,
1003                                              initiator_iqn=args.initiator_iqn))
1004
1005    p = subparsers.add_parser('bdev_iscsi_create',
1006                              help='Add bdev with iSCSI initiator backend')
1007    p.add_argument('-b', '--name', help="Name of the bdev", required=True)
1008    p.add_argument('-i', '--initiator-iqn', help="Initiator IQN", required=True)
1009    p.add_argument('--url', help="iSCSI Lun URL", required=True)
1010    p.set_defaults(func=bdev_iscsi_create)
1011
1012    def bdev_iscsi_delete(args):
1013        rpc.bdev.bdev_iscsi_delete(args.client,
1014                                   name=args.name)
1015
1016    p = subparsers.add_parser('bdev_iscsi_delete', help='Delete an iSCSI bdev')
1017    p.add_argument('name', help='iSCSI bdev name')
1018    p.set_defaults(func=bdev_iscsi_delete)
1019
1020    def bdev_pmem_create(args):
1021        print_json(rpc.bdev.bdev_pmem_create(args.client,
1022                                             pmem_file=args.pmem_file,
1023                                             name=args.name))
1024
1025    p = subparsers.add_parser('bdev_pmem_create', help='Add a bdev with pmem backend')
1026    p.add_argument('pmem_file', help='Path to pmemblk pool file')
1027    p.add_argument('-n', '--name', help='Block device name', required=True)
1028    p.set_defaults(func=bdev_pmem_create)
1029
1030    def bdev_pmem_delete(args):
1031        rpc.bdev.bdev_pmem_delete(args.client,
1032                                  name=args.name)
1033
1034    p = subparsers.add_parser('bdev_pmem_delete', help='Delete a pmem bdev')
1035    p.add_argument('name', help='pmem bdev name')
1036    p.set_defaults(func=bdev_pmem_delete)
1037
1038    def bdev_passthru_create(args):
1039        print_json(rpc.bdev.bdev_passthru_create(args.client,
1040                                                 base_bdev_name=args.base_bdev_name,
1041                                                 name=args.name))
1042
1043    p = subparsers.add_parser('bdev_passthru_create', help='Add a pass through bdev on existing bdev')
1044    p.add_argument('-b', '--base-bdev-name', help="Name of the existing bdev", required=True)
1045    p.add_argument('-p', '--name', help="Name of the pass through bdev", required=True)
1046    p.set_defaults(func=bdev_passthru_create)
1047
1048    def bdev_passthru_delete(args):
1049        rpc.bdev.bdev_passthru_delete(args.client,
1050                                      name=args.name)
1051
1052    p = subparsers.add_parser('bdev_passthru_delete', help='Delete a pass through bdev')
1053    p.add_argument('name', help='pass through bdev name')
1054    p.set_defaults(func=bdev_passthru_delete)
1055
1056    def bdev_get_bdevs(args):
1057        print_dict(rpc.bdev.bdev_get_bdevs(args.client,
1058                                           name=args.name, timeout=args.timeout_ms))
1059
1060    p = subparsers.add_parser('bdev_get_bdevs',
1061                              help='Display current blockdev list or required blockdev')
1062    p.add_argument('-b', '--name', help="Name of the Blockdev. Example: Nvme0n1", required=False)
1063    p.add_argument('-t', '--timeout-ms', help="""Time in ms to wait for the bdev to appear (only used
1064    with the -b|--name option). The default timeout is 0, meaning the RPC returns immediately
1065    whether the bdev exists or not.""",
1066                   type=int, required=False)
1067    p.set_defaults(func=bdev_get_bdevs)
1068
1069    def bdev_get_iostat(args):
1070        print_dict(rpc.bdev.bdev_get_iostat(args.client,
1071                                            name=args.name))
1072
1073    p = subparsers.add_parser('bdev_get_iostat',
1074                              help='Display current I/O statistics of all the blockdevs or required blockdev.')
1075    p.add_argument('-b', '--name', help="Name of the Blockdev. Example: Nvme0n1", required=False)
1076    p.set_defaults(func=bdev_get_iostat)
1077
1078    def bdev_enable_histogram(args):
1079        rpc.bdev.bdev_enable_histogram(args.client, name=args.name, enable=args.enable)
1080
1081    p = subparsers.add_parser('bdev_enable_histogram',
1082                              help='Enable or disable histogram for specified bdev')
1083    p.add_argument('-e', '--enable', default=True, dest='enable', action='store_true', help='Enable histograms on specified device')
1084    p.add_argument('-d', '--disable', dest='enable', action='store_false', help='Disable histograms on specified device')
1085    p.add_argument('name', help='bdev name')
1086    p.set_defaults(func=bdev_enable_histogram)
1087
1088    def bdev_get_histogram(args):
1089        print_dict(rpc.bdev.bdev_get_histogram(args.client, name=args.name))
1090
1091    p = subparsers.add_parser('bdev_get_histogram',
1092                              help='Get histogram for specified bdev')
1093    p.add_argument('name', help='bdev name')
1094    p.set_defaults(func=bdev_get_histogram)
1095
1096    def bdev_set_qd_sampling_period(args):
1097        rpc.bdev.bdev_set_qd_sampling_period(args.client,
1098                                             name=args.name,
1099                                             period=args.period)
1100
1101    p = subparsers.add_parser('bdev_set_qd_sampling_period',
1102                              help="Enable or disable tracking of a bdev's queue depth.")
1103    p.add_argument('name', help='Blockdev name. Example: Malloc0')
1104    p.add_argument('period', help='Period with which to poll the block device queue depth in microseconds.'
1105                   ' If set to 0, polling will be disabled.',
1106                   type=int)
1107    p.set_defaults(func=bdev_set_qd_sampling_period)
1108
1109    def bdev_set_qos_limit(args):
1110        rpc.bdev.bdev_set_qos_limit(args.client,
1111                                    name=args.name,
1112                                    rw_ios_per_sec=args.rw_ios_per_sec,
1113                                    rw_mbytes_per_sec=args.rw_mbytes_per_sec,
1114                                    r_mbytes_per_sec=args.r_mbytes_per_sec,
1115                                    w_mbytes_per_sec=args.w_mbytes_per_sec)
1116
1117    p = subparsers.add_parser('bdev_set_qos_limit',
1118                              help='Set QoS rate limit on a blockdev')
1119    p.add_argument('name', help='Blockdev name to set QoS. Example: Malloc0')
1120    p.add_argument('--rw-ios-per-sec',
1121                   help='R/W IOs per second limit (>=1000, example: 20000). 0 means unlimited.',
1122                   type=int, required=False)
1123    p.add_argument('--rw-mbytes-per-sec',
1124                   help="R/W megabytes per second limit (>=10, example: 100). 0 means unlimited.",
1125                   type=int, required=False)
1126    p.add_argument('--r-mbytes-per-sec',
1127                   help="Read megabytes per second limit (>=10, example: 100). 0 means unlimited.",
1128                   type=int, required=False)
1129    p.add_argument('--w-mbytes-per-sec',
1130                   help="Write megabytes per second limit (>=10, example: 100). 0 means unlimited.",
1131                   type=int, required=False)
1132    p.set_defaults(func=bdev_set_qos_limit)
1133
1134    def bdev_error_inject_error(args):
1135        rpc.bdev.bdev_error_inject_error(args.client,
1136                                         name=args.name,
1137                                         io_type=args.io_type,
1138                                         error_type=args.error_type,
1139                                         num=args.num)
1140
1141    p = subparsers.add_parser('bdev_error_inject_error', help='bdev inject error')
1142    p.add_argument('name', help="""the name of the error injection bdev""")
1143    p.add_argument('io_type', help="""io_type: 'clear' 'read' 'write' 'unmap' 'flush' 'all'""")
1144    p.add_argument('error_type', help="""error_type: 'failure' 'pending'""")
1145    p.add_argument(
1146        '-n', '--num', help='the number of commands you want to fail', type=int)
1147    p.set_defaults(func=bdev_error_inject_error)
1148
1149    def bdev_nvme_apply_firmware(args):
1150        print_dict(rpc.bdev.bdev_nvme_apply_firmware(args.client,
1151                                                     bdev_name=args.bdev_name,
1152                                                     filename=args.filename))
1153
1154    p = subparsers.add_parser('bdev_nvme_apply_firmware', help='Download and commit firmware to NVMe device')
1155    p.add_argument('filename', help='filename of the firmware to download')
1156    p.add_argument('bdev_name', help='name of the NVMe device')
1157    p.set_defaults(func=bdev_nvme_apply_firmware)
1158
1159    def bdev_nvme_get_transport_statistics(args):
1160        print_dict(rpc.bdev.bdev_nvme_get_transport_statistics(args.client))
1161
1162    p = subparsers.add_parser('bdev_nvme_get_transport_statistics',
1163                              help='Get bdev_nvme poll group transport statistics')
1164    p.set_defaults(func=bdev_nvme_get_transport_statistics)
1165
1166    def bdev_nvme_get_controller_health_info(args):
1167        print_dict(rpc.bdev.bdev_nvme_get_controller_health_info(args.client,
1168                                                                 name=args.name))
1169
1170    p = subparsers.add_parser('bdev_nvme_get_controller_health_info',
1171                              help='Display health log of the required NVMe bdev controller.')
1172    p.add_argument('-c', '--name', help="Name of the NVMe bdev controller. Example: Nvme0", required=True)
1173    p.set_defaults(func=bdev_nvme_get_controller_health_info)
1174
1175    # iSCSI
1176    def iscsi_set_options(args):
1177        rpc.iscsi.iscsi_set_options(
1178            args.client,
1179            auth_file=args.auth_file,
1180            node_base=args.node_base,
1181            nop_timeout=args.nop_timeout,
1182            nop_in_interval=args.nop_in_interval,
1183            disable_chap=args.disable_chap,
1184            require_chap=args.require_chap,
1185            mutual_chap=args.mutual_chap,
1186            chap_group=args.chap_group,
1187            max_sessions=args.max_sessions,
1188            max_queue_depth=args.max_queue_depth,
1189            max_connections_per_session=args.max_connections_per_session,
1190            default_time2wait=args.default_time2wait,
1191            default_time2retain=args.default_time2retain,
1192            first_burst_length=args.first_burst_length,
1193            immediate_data=args.immediate_data,
1194            error_recovery_level=args.error_recovery_level,
1195            allow_duplicated_isid=args.allow_duplicated_isid,
1196            max_large_datain_per_connection=args.max_large_datain_per_connection,
1197            max_r2t_per_connection=args.max_r2t_per_connection,
1198            pdu_pool_size=args.pdu_pool_size,
1199            immediate_data_pool_size=args.immediate_data_pool_size,
1200            data_out_pool_size=args.data_out_pool_size)
1201
1202    p = subparsers.add_parser('iscsi_set_options',
1203                              help="""Set options of iSCSI subsystem""")
1204    p.add_argument('-f', '--auth-file', help='Path to CHAP shared secret file')
1205    p.add_argument('-b', '--node-base', help='Prefix of the name of iSCSI target node')
1206    p.add_argument('-o', '--nop-timeout', help='Timeout in seconds to nop-in request to the initiator', type=int)
1207    p.add_argument('-n', '--nop-in-interval', help='Time interval in secs between nop-in requests by the target', type=int)
1208    p.add_argument('-d', '--disable-chap', help="""CHAP for discovery session should be disabled.
1209    *** Mutually exclusive with --require-chap""", action='store_true')
1210    p.add_argument('-r', '--require-chap', help="""CHAP for discovery session should be required.
1211    *** Mutually exclusive with --disable-chap""", action='store_true')
1212    p.add_argument('-m', '--mutual-chap', help='CHAP for discovery session should be mutual', action='store_true')
1213    p.add_argument('-g', '--chap-group', help="""Authentication group ID for discovery session.
1214    *** Authentication group must be precreated ***""", type=int)
1215    p.add_argument('-a', '--max-sessions', help='Maximum number of sessions in the host.', type=int)
1216    p.add_argument('-q', '--max-queue-depth', help='Max number of outstanding I/Os per queue.', type=int)
1217    p.add_argument('-c', '--max-connections-per-session', help='Negotiated parameter, MaxConnections.', type=int)
1218    p.add_argument('-w', '--default-time2wait', help='Negotiated parameter, DefaultTime2Wait.', type=int)
1219    p.add_argument('-v', '--default-time2retain', help='Negotiated parameter, DefaultTime2Retain.', type=int)
1220    p.add_argument('-s', '--first-burst-length', help='Negotiated parameter, FirstBurstLength.', type=int)
1221    p.add_argument('-i', '--immediate-data', help='Negotiated parameter, ImmediateData.', action='store_true')
1222    p.add_argument('-l', '--error-recovery-level', help='Negotiated parameter, ErrorRecoveryLevel', type=int)
1223    p.add_argument('-p', '--allow-duplicated-isid', help='Allow duplicated initiator session ID.', action='store_true')
1224    p.add_argument('-x', '--max-large-datain-per-connection', help='Max number of outstanding split read I/Os per connection', type=int)
1225    p.add_argument('-k', '--max-r2t-per-connection', help='Max number of outstanding R2Ts per connection', type=int)
1226    p.add_argument('-u', '--pdu-pool-size', help='Number of PDUs in the pool', type=int)
1227    p.add_argument('-j', '--immediate-data-pool-size', help='Number of immediate data buffers in the pool', type=int)
1228    p.add_argument('-z', '--data-out-pool-size', help='Number of data out buffers in the pool', type=int)
1229    p.set_defaults(func=iscsi_set_options)
1230
1231    def iscsi_set_discovery_auth(args):
1232        rpc.iscsi.iscsi_set_discovery_auth(
1233            args.client,
1234            disable_chap=args.disable_chap,
1235            require_chap=args.require_chap,
1236            mutual_chap=args.mutual_chap,
1237            chap_group=args.chap_group)
1238
1239    p = subparsers.add_parser('iscsi_set_discovery_auth',
1240                              help="""Set CHAP authentication for discovery session.""")
1241    p.add_argument('-d', '--disable-chap', help="""CHAP for discovery session should be disabled.
1242    *** Mutually exclusive with --require-chap""", action='store_true')
1243    p.add_argument('-r', '--require-chap', help="""CHAP for discovery session should be required.
1244    *** Mutually exclusive with --disable-chap""", action='store_true')
1245    p.add_argument('-m', '--mutual-chap', help='CHAP for discovery session should be mutual', action='store_true')
1246    p.add_argument('-g', '--chap-group', help="""Authentication group ID for discovery session.
1247    *** Authentication group must be precreated ***""", type=int)
1248    p.set_defaults(func=iscsi_set_discovery_auth)
1249
1250    def iscsi_create_auth_group(args):
1251        secrets = None
1252        if args.secrets:
1253            secrets = [dict(u.split(":") for u in a.split(" ")) for a in args.secrets.split(",")]
1254
1255        rpc.iscsi.iscsi_create_auth_group(args.client, tag=args.tag, secrets=secrets)
1256
1257    p = subparsers.add_parser('iscsi_create_auth_group',
1258                              help='Create authentication group for CHAP authentication.')
1259    p.add_argument('tag', help='Authentication group tag (unique, integer > 0).', type=int)
1260    p.add_argument('-c', '--secrets', help="""Comma-separated list of CHAP secrets
1261<user:user_name secret:chap_secret muser:mutual_user_name msecret:mutual_chap_secret> enclosed in quotes.
1262Format: 'user:u1 secret:s1 muser:mu1 msecret:ms1,user:u2 secret:s2 muser:mu2 msecret:ms2'""", required=False)
1263    p.set_defaults(func=iscsi_create_auth_group)
1264
1265    def iscsi_delete_auth_group(args):
1266        rpc.iscsi.iscsi_delete_auth_group(args.client, tag=args.tag)
1267
1268    p = subparsers.add_parser('iscsi_delete_auth_group',
1269                              help='Delete an authentication group.')
1270    p.add_argument('tag', help='Authentication group tag', type=int)
1271    p.set_defaults(func=iscsi_delete_auth_group)
1272
1273    def iscsi_auth_group_add_secret(args):
1274        rpc.iscsi.iscsi_auth_group_add_secret(
1275            args.client,
1276            tag=args.tag,
1277            user=args.user,
1278            secret=args.secret,
1279            muser=args.muser,
1280            msecret=args.msecret)
1281
1282    p = subparsers.add_parser('iscsi_auth_group_add_secret',
1283                              help='Add a secret to an authentication group.')
1284    p.add_argument('tag', help='Authentication group tag', type=int)
1285    p.add_argument('-u', '--user', help='User name for one-way CHAP authentication', required=True)
1286    p.add_argument('-s', '--secret', help='Secret for one-way CHAP authentication', required=True)
1287    p.add_argument('-m', '--muser', help='User name for mutual CHAP authentication')
1288    p.add_argument('-r', '--msecret', help='Secret for mutual CHAP authentication')
1289    p.set_defaults(func=iscsi_auth_group_add_secret)
1290
1291    def iscsi_auth_group_remove_secret(args):
1292        rpc.iscsi.iscsi_auth_group_remove_secret(args.client, tag=args.tag, user=args.user)
1293
1294    p = subparsers.add_parser('iscsi_auth_group_remove_secret',
1295                              help='Remove a secret from an authentication group.')
1296    p.add_argument('tag', help='Authentication group tag', type=int)
1297    p.add_argument('-u', '--user', help='User name for one-way CHAP authentication', required=True)
1298    p.set_defaults(func=iscsi_auth_group_remove_secret)
1299
1300    def iscsi_get_auth_groups(args):
1301        print_dict(rpc.iscsi.iscsi_get_auth_groups(args.client))
1302
1303    p = subparsers.add_parser('iscsi_get_auth_groups',
1304                              help='Display current authentication group configuration')
1305    p.set_defaults(func=iscsi_get_auth_groups)
1306
1307    def iscsi_get_portal_groups(args):
1308        print_dict(rpc.iscsi.iscsi_get_portal_groups(args.client))
1309
1310    p = subparsers.add_parser('iscsi_get_portal_groups', help='Display current portal group configuration')
1311    p.set_defaults(func=iscsi_get_portal_groups)
1312
1313    def iscsi_get_initiator_groups(args):
1314        print_dict(rpc.iscsi.iscsi_get_initiator_groups(args.client))
1315
1316    p = subparsers.add_parser('iscsi_get_initiator_groups',
1317                              help='Display current initiator group configuration')
1318    p.set_defaults(func=iscsi_get_initiator_groups)
1319
1320    def iscsi_get_target_nodes(args):
1321        print_dict(rpc.iscsi.iscsi_get_target_nodes(args.client))
1322
1323    p = subparsers.add_parser('iscsi_get_target_nodes', help='Display target nodes')
1324    p.set_defaults(func=iscsi_get_target_nodes)
1325
1326    def iscsi_create_target_node(args):
1327        luns = []
1328        for u in args.bdev_name_id_pairs.strip().split(" "):
1329            bdev_name, lun_id = u.split(":")
1330            luns.append({"bdev_name": bdev_name, "lun_id": int(lun_id)})
1331
1332        pg_ig_maps = []
1333        for u in args.pg_ig_mappings.strip().split(" "):
1334            pg, ig = u.split(":")
1335            pg_ig_maps.append({"pg_tag": int(pg), "ig_tag": int(ig)})
1336
1337        rpc.iscsi.iscsi_create_target_node(
1338            args.client,
1339            luns=luns,
1340            pg_ig_maps=pg_ig_maps,
1341            name=args.name,
1342            alias_name=args.alias_name,
1343            queue_depth=args.queue_depth,
1344            chap_group=args.chap_group,
1345            disable_chap=args.disable_chap,
1346            require_chap=args.require_chap,
1347            mutual_chap=args.mutual_chap,
1348            header_digest=args.header_digest,
1349            data_digest=args.data_digest)
1350
1351    p = subparsers.add_parser('iscsi_create_target_node', help='Add a target node')
1352    p.add_argument('name', help='Target node name (ASCII)')
1353    p.add_argument('alias_name', help='Target node alias name (ASCII)')
1354    p.add_argument('bdev_name_id_pairs', help="""Whitespace-separated list of <bdev name:LUN ID> pairs enclosed
1355    in quotes.  Format:  'bdev_name0:id0 bdev_name1:id1' etc
1356    Example: 'Malloc0:0 Malloc1:1 Malloc5:2'
1357    *** The bdevs must pre-exist ***
1358    *** LUN0 (id = 0) is required ***
1359    *** bdevs names cannot contain space or colon characters ***""")
1360    p.add_argument('pg_ig_mappings', help="""List of (Portal_Group_Tag:Initiator_Group_Tag) mappings
1361    Whitespace separated, quoted, mapping defined with colon
1362    separated list of "tags" (int > 0)
1363    Example: '1:1 2:2 2:1'
1364    *** The Portal/Initiator Groups must be precreated ***""")
1365    p.add_argument('queue_depth', help='Desired target queue depth', type=int)
1366    p.add_argument('-g', '--chap-group', help="""Authentication group ID for this target node.
1367    *** Authentication group must be precreated ***""", type=int)
1368    p.add_argument('-d', '--disable-chap', help="""CHAP authentication should be disabled for this target node.
1369    *** Mutually exclusive with --require-chap ***""", action='store_true')
1370    p.add_argument('-r', '--require-chap', help="""CHAP authentication should be required for this target node.
1371    *** Mutually exclusive with --disable-chap ***""", action='store_true')
1372    p.add_argument(
1373        '-m', '--mutual-chap', help='CHAP authentication should be mutual/bidirectional.', action='store_true')
1374    p.add_argument('-H', '--header-digest',
1375                   help='Header Digest should be required for this target node.', action='store_true')
1376    p.add_argument('-D', '--data-digest',
1377                   help='Data Digest should be required for this target node.', action='store_true')
1378    p.set_defaults(func=iscsi_create_target_node)
1379
1380    def iscsi_target_node_add_lun(args):
1381        rpc.iscsi.iscsi_target_node_add_lun(
1382            args.client,
1383            name=args.name,
1384            bdev_name=args.bdev_name,
1385            lun_id=args.lun_id)
1386
1387    p = subparsers.add_parser('iscsi_target_node_add_lun',
1388                              help='Add LUN to the target node')
1389    p.add_argument('name', help='Target node name (ASCII)')
1390    p.add_argument('bdev_name', help="""bdev name enclosed in quotes.
1391    *** bdev name cannot contain space or colon characters ***""")
1392    p.add_argument('-i', dest='lun_id', help="""LUN ID (integer >= 0)
1393    *** If LUN ID is omitted or -1, the lowest free one is assigned ***""", type=int, required=False)
1394    p.set_defaults(func=iscsi_target_node_add_lun)
1395
1396    def iscsi_target_node_set_auth(args):
1397        rpc.iscsi.iscsi_target_node_set_auth(
1398            args.client,
1399            name=args.name,
1400            chap_group=args.chap_group,
1401            disable_chap=args.disable_chap,
1402            require_chap=args.require_chap,
1403            mutual_chap=args.mutual_chap)
1404
1405    p = subparsers.add_parser('iscsi_target_node_set_auth',
1406                              help='Set CHAP authentication for the target node')
1407    p.add_argument('name', help='Target node name (ASCII)')
1408    p.add_argument('-g', '--chap-group', help="""Authentication group ID for this target node.
1409    *** Authentication group must be precreated ***""", type=int)
1410    p.add_argument('-d', '--disable-chap', help="""CHAP authentication should be disabled for this target node.
1411    *** Mutually exclusive with --require-chap ***""", action='store_true')
1412    p.add_argument('-r', '--require-chap', help="""CHAP authentication should be required for this target node.
1413    *** Mutually exclusive with --disable-chap ***""", action='store_true')
1414    p.add_argument('-m', '--mutual-chap', help='CHAP authentication should be mutual/bidirectional.',
1415                   action='store_true')
1416    p.set_defaults(func=iscsi_target_node_set_auth)
1417
1418    def iscsi_target_node_add_pg_ig_maps(args):
1419        pg_ig_maps = []
1420        for u in args.pg_ig_mappings.strip().split(" "):
1421            pg, ig = u.split(":")
1422            pg_ig_maps.append({"pg_tag": int(pg), "ig_tag": int(ig)})
1423        rpc.iscsi.iscsi_target_node_add_pg_ig_maps(
1424            args.client,
1425            pg_ig_maps=pg_ig_maps,
1426            name=args.name)
1427
1428    p = subparsers.add_parser('iscsi_target_node_add_pg_ig_maps',
1429                              help='Add PG-IG maps to the target node')
1430    p.add_argument('name', help='Target node name (ASCII)')
1431    p.add_argument('pg_ig_mappings', help="""List of (Portal_Group_Tag:Initiator_Group_Tag) mappings
1432    Whitespace separated, quoted, mapping defined with colon
1433    separated list of "tags" (int > 0)
1434    Example: '1:1 2:2 2:1'
1435    *** The Portal/Initiator Groups must be precreated ***""")
1436    p.set_defaults(func=iscsi_target_node_add_pg_ig_maps)
1437
1438    def iscsi_target_node_remove_pg_ig_maps(args):
1439        pg_ig_maps = []
1440        for u in args.pg_ig_mappings.strip().split(" "):
1441            pg, ig = u.split(":")
1442            pg_ig_maps.append({"pg_tag": int(pg), "ig_tag": int(ig)})
1443        rpc.iscsi.iscsi_target_node_remove_pg_ig_maps(
1444            args.client, pg_ig_maps=pg_ig_maps, name=args.name)
1445
1446    p = subparsers.add_parser('iscsi_target_node_remove_pg_ig_maps',
1447                              help='Delete PG-IG maps from the target node')
1448    p.add_argument('name', help='Target node name (ASCII)')
1449    p.add_argument('pg_ig_mappings', help="""List of (Portal_Group_Tag:Initiator_Group_Tag) mappings
1450    Whitespace separated, quoted, mapping defined with colon
1451    separated list of "tags" (int > 0)
1452    Example: '1:1 2:2 2:1'
1453    *** The Portal/Initiator Groups must be precreated ***""")
1454    p.set_defaults(func=iscsi_target_node_remove_pg_ig_maps)
1455
1456    def iscsi_target_node_set_redirect(args):
1457        rpc.iscsi.iscsi_target_node_set_redirect(
1458            args.client,
1459            name=args.name,
1460            pg_tag=args.pg_tag,
1461            redirect_host=args.redirect_host,
1462            redirect_port=args.redirect_port)
1463
1464    p = subparsers.add_parser('iscsi_target_node_set_redirect',
1465                              help="""Update redirect portal of the public portal group for the target node.
1466    Omit redirect host and port to clear previously set redirect settings.""")
1467    p.add_argument('name', help='Target node name (ASCII)')
1468    p.add_argument('pg_tag', help='Portal group tag (unique, integer > 0)', type=int)
1469    p.add_argument('-a', '--redirect-host', help='Numeric IP address for redirect portal', required=False)
1470    p.add_argument('-p', '--redirect-port', help='Numeric TCP port for redirect portal', required=False)
1471    p.set_defaults(func=iscsi_target_node_set_redirect)
1472
1473    def iscsi_target_node_request_logout(args):
1474        rpc.iscsi.iscsi_target_node_request_logout(
1475            args.client,
1476            name=args.name,
1477            pg_tag=args.pg_tag)
1478
1479    p = subparsers.add_parser('iscsi_target_node_request_logout',
1480                              help="""For the target node, request connections whose portal group tag
1481    match to logout, or request all connections if portal group tag is omitted.""")
1482    p.add_argument('name', help='Target node name (ASCII)')
1483    p.add_argument('-t', '--pg-tag', help='Portal group tag (unique, integer > 0)', type=int, required=False)
1484    p.set_defaults(func=iscsi_target_node_request_logout)
1485
1486    def iscsi_create_portal_group(args):
1487        portals = []
1488        for p in args.portal_list.strip().split(' '):
1489            ip, separator, port_cpumask = p.rpartition(':')
1490            split_port_cpumask = port_cpumask.split('@')
1491            if len(split_port_cpumask) == 1:
1492                port = port_cpumask
1493                portals.append({'host': ip, 'port': port})
1494            else:
1495                port = split_port_cpumask[0]
1496                cpumask = split_port_cpumask[1]
1497                portals.append({'host': ip, 'port': port})
1498                print("WARNING: Specifying a portal group with a CPU mask is no longer supported. Ignoring it.")
1499        rpc.iscsi.iscsi_create_portal_group(
1500            args.client,
1501            portals=portals,
1502            tag=args.tag,
1503            private=args.private,
1504            wait=args.wait)
1505
1506    p = subparsers.add_parser('iscsi_create_portal_group',
1507                              help='Add a portal group')
1508    p.add_argument(
1509        'tag', help='Portal group tag (unique, integer > 0)', type=int)
1510    p.add_argument('portal_list', help="""List of portals in host:port format, separated by whitespace
1511    Example: '192.168.100.100:3260 192.168.100.100:3261 192.168.100.100:3262""")
1512    p.add_argument('-p', '--private', help="""Public (false) or private (true) portal group.
1513    Private portal groups do not have their portals returned by a discovery session. A public
1514    portal group may optionally specify a redirect portal for non-discovery logins. This redirect
1515    portal must be from a private portal group.""", action='store_true')
1516    p.add_argument('-w', '--wait', help="""Do not listening on portals until it is started explicitly.
1517    One major iSCSI initiator may not retry login once it failed. Hence for such initiator, listening
1518    on portals should be allowed after all associated target nodes are created.""", action='store_true')
1519    p.set_defaults(func=iscsi_create_portal_group)
1520
1521    def iscsi_start_portal_group(args):
1522        rpc.iscsi.iscsi_start_portal_group(args.client, tag=args.tag)
1523
1524    p = subparsers.add_parser('iscsi_start_portal_group',
1525                              help='Start listening on portals if it is not started yet.')
1526    p.add_argument(
1527        'tag', help='Portal group tag (unique, integer > 0)', type=int)
1528    p.set_defaults(func=iscsi_start_portal_group)
1529
1530    def iscsi_create_initiator_group(args):
1531        initiators = []
1532        netmasks = []
1533        for i in args.initiator_list.strip().split(' '):
1534            initiators.append(i)
1535        for n in args.netmask_list.strip().split(' '):
1536            netmasks.append(n)
1537        rpc.iscsi.iscsi_create_initiator_group(
1538            args.client,
1539            tag=args.tag,
1540            initiators=initiators,
1541            netmasks=netmasks)
1542
1543    p = subparsers.add_parser('iscsi_create_initiator_group',
1544                              help='Add an initiator group')
1545    p.add_argument(
1546        'tag', help='Initiator group tag (unique, integer > 0)', type=int)
1547    p.add_argument('initiator_list', help="""Whitespace-separated list of initiator hostnames or IP addresses,
1548    enclosed in quotes.  Example: 'ANY' or 'iqn.2016-06.io.spdk:host1 iqn.2016-06.io.spdk:host2'""")
1549    p.add_argument('netmask_list', help="""Whitespace-separated list of initiator netmasks enclosed in quotes.
1550    Example: '255.255.0.0 255.248.0.0' etc""")
1551    p.set_defaults(func=iscsi_create_initiator_group)
1552
1553    def iscsi_initiator_group_add_initiators(args):
1554        initiators = None
1555        netmasks = None
1556        if args.initiator_list:
1557            initiators = []
1558            for i in args.initiator_list.strip().split(' '):
1559                initiators.append(i)
1560        if args.netmask_list:
1561            netmasks = []
1562            for n in args.netmask_list.strip().split(' '):
1563                netmasks.append(n)
1564        rpc.iscsi.iscsi_initiator_group_add_initiators(
1565            args.client,
1566            tag=args.tag,
1567            initiators=initiators,
1568            netmasks=netmasks)
1569
1570    p = subparsers.add_parser('iscsi_initiator_group_add_initiators',
1571                              help='Add initiators to an existing initiator group')
1572    p.add_argument(
1573        'tag', help='Initiator group tag (unique, integer > 0)', type=int)
1574    p.add_argument('-n', dest='initiator_list', help="""Whitespace-separated list of initiator hostnames or IP addresses,
1575    enclosed in quotes.  This parameter can be omitted.  Example: 'ANY' or
1576    'iqn.2016-06.io.spdk:host1 iqn.2016-06.io.spdk:host2'""", required=False)
1577    p.add_argument('-m', dest='netmask_list', help="""Whitespace-separated list of initiator netmasks enclosed in quotes.
1578    This parameter can be omitted.  Example: '255.255.0.0 255.248.0.0' etc""", required=False)
1579    p.set_defaults(func=iscsi_initiator_group_add_initiators)
1580
1581    def iscsi_initiator_group_remove_initiators(args):
1582        initiators = None
1583        netmasks = None
1584        if args.initiator_list:
1585            initiators = []
1586            for i in args.initiator_list.strip().split(' '):
1587                initiators.append(i)
1588        if args.netmask_list:
1589            netmasks = []
1590            for n in args.netmask_list.strip().split(' '):
1591                netmasks.append(n)
1592        rpc.iscsi.iscsi_initiator_group_remove_initiators(
1593            args.client,
1594            tag=args.tag,
1595            initiators=initiators,
1596            netmasks=netmasks)
1597
1598    p = subparsers.add_parser('iscsi_initiator_group_remove_initiators',
1599                              help='Delete initiators from an existing initiator group')
1600    p.add_argument(
1601        'tag', help='Initiator group tag (unique, integer > 0)', type=int)
1602    p.add_argument('-n', dest='initiator_list', help="""Whitespace-separated list of initiator hostnames or IP addresses,
1603    enclosed in quotes.  This parameter can be omitted.  Example: 'ANY' or
1604    'iqn.2016-06.io.spdk:host1 iqn.2016-06.io.spdk:host2'""", required=False)
1605    p.add_argument('-m', dest='netmask_list', help="""Whitespace-separated list of initiator netmasks enclosed in quotes.
1606    This parameter can be omitted.  Example: '255.255.0.0 255.248.0.0' etc""", required=False)
1607    p.set_defaults(func=iscsi_initiator_group_remove_initiators)
1608
1609    def iscsi_delete_target_node(args):
1610        rpc.iscsi.iscsi_delete_target_node(
1611            args.client, target_node_name=args.target_node_name)
1612
1613    p = subparsers.add_parser('iscsi_delete_target_node',
1614                              help='Delete a target node')
1615    p.add_argument('target_node_name',
1616                   help='Target node name to be deleted. Example: iqn.2016-06.io.spdk:disk1.')
1617    p.set_defaults(func=iscsi_delete_target_node)
1618
1619    def iscsi_delete_portal_group(args):
1620        rpc.iscsi.iscsi_delete_portal_group(args.client, tag=args.tag)
1621
1622    p = subparsers.add_parser('iscsi_delete_portal_group',
1623                              help='Delete a portal group')
1624    p.add_argument(
1625        'tag', help='Portal group tag (unique, integer > 0)', type=int)
1626    p.set_defaults(func=iscsi_delete_portal_group)
1627
1628    def iscsi_delete_initiator_group(args):
1629        rpc.iscsi.iscsi_delete_initiator_group(args.client, tag=args.tag)
1630
1631    p = subparsers.add_parser('iscsi_delete_initiator_group',
1632                              help='Delete an initiator group')
1633    p.add_argument(
1634        'tag', help='Initiator group tag (unique, integer > 0)', type=int)
1635    p.set_defaults(func=iscsi_delete_initiator_group)
1636
1637    def iscsi_portal_group_set_auth(args):
1638        rpc.iscsi.iscsi_portal_group_set_auth(
1639            args.client,
1640            tag=args.tag,
1641            chap_group=args.chap_group,
1642            disable_chap=args.disable_chap,
1643            require_chap=args.require_chap,
1644            mutual_chap=args.mutual_chap)
1645
1646    p = subparsers.add_parser('iscsi_portal_group_set_auth',
1647                              help='Set CHAP authentication for discovery sessions specific for the portal group')
1648    p.add_argument('tag', help='Portal group tag (unique, integer > 0)', type=int)
1649    p.add_argument('-g', '--chap-group', help="""Authentication group ID for this portal group.
1650    *** Authentication group must be precreated ***""", type=int)
1651    p.add_argument('-d', '--disable-chap', help="""CHAP authentication should be disabled for this portal group.
1652    *** Mutually exclusive with --require-chap ***""", action='store_true')
1653    p.add_argument('-r', '--require-chap', help="""CHAP authentication should be required for this portal group.
1654    *** Mutually exclusive with --disable-chap ***""", action='store_true')
1655    p.add_argument('-m', '--mutual-chap', help='CHAP authentication should be mutual/bidirectional.',
1656                   action='store_true')
1657    p.set_defaults(func=iscsi_portal_group_set_auth)
1658
1659    def iscsi_get_connections(args):
1660        print_dict(rpc.iscsi.iscsi_get_connections(args.client))
1661
1662    p = subparsers.add_parser('iscsi_get_connections',
1663                              help='Display iSCSI connections')
1664    p.set_defaults(func=iscsi_get_connections)
1665
1666    def iscsi_get_options(args):
1667        print_dict(rpc.iscsi.iscsi_get_options(args.client))
1668
1669    p = subparsers.add_parser('iscsi_get_options',
1670                              help='Display iSCSI global parameters')
1671    p.set_defaults(func=iscsi_get_options)
1672
1673    def scsi_get_devices(args):
1674        print_dict(rpc.iscsi.scsi_get_devices(args.client))
1675
1676    p = subparsers.add_parser('scsi_get_devices', help='Display SCSI devices')
1677    p.set_defaults(func=scsi_get_devices)
1678
1679    # trace
1680    def trace_enable_tpoint_group(args):
1681        rpc.trace.trace_enable_tpoint_group(args.client, name=args.name)
1682
1683    p = subparsers.add_parser('trace_enable_tpoint_group',
1684                              help='enable trace on a specific tpoint group')
1685    p.add_argument(
1686        'name', help="""trace group name we want to enable in tpoint_group_mask.
1687        (for example "bdev" for bdev trace group, "all" for all trace groups).""")
1688    p.set_defaults(func=trace_enable_tpoint_group)
1689
1690    def trace_disable_tpoint_group(args):
1691        rpc.trace.trace_disable_tpoint_group(args.client, name=args.name)
1692
1693    p = subparsers.add_parser('trace_disable_tpoint_group',
1694                              help='disable trace on a specific tpoint group')
1695    p.add_argument(
1696        'name', help="""trace group name we want to disable in tpoint_group_mask.
1697        (for example "bdev" for bdev trace group, "all" for all trace groups).""")
1698    p.set_defaults(func=trace_disable_tpoint_group)
1699
1700    def trace_set_tpoint_mask(args):
1701        rpc.trace.trace_set_tpoint_mask(args.client, name=args.name, tpoint_mask=args.tpoint_mask)
1702
1703    p = subparsers.add_parser('trace_set_tpoint_mask',
1704                              help='enable tracepoint mask on a specific tpoint group')
1705    p.add_argument(
1706        'name', help="""trace group name we want to enable in tpoint_group_mask.
1707        (for example "bdev" for bdev trace group)""")
1708    p.add_argument(
1709        'tpoint_mask', help="""tracepoints to be enabled inside a given trace group.
1710        (for example value of "0x3" will enable only the first two tpoints in this group)""",
1711        type=lambda m: int(m, 16))
1712    p.set_defaults(func=trace_set_tpoint_mask)
1713
1714    def trace_clear_tpoint_mask(args):
1715        rpc.trace.trace_clear_tpoint_mask(args.client, name=args.name, tpoint_mask=args.tpoint_mask)
1716
1717    p = subparsers.add_parser('trace_clear_tpoint_mask',
1718                              help='disable tracepoint mask on a specific tpoint group')
1719    p.add_argument(
1720        'name', help="""trace group name we want to disable in tpoint_group_mask.
1721        (for example "bdev" for bdev trace group)""")
1722    p.add_argument(
1723        'tpoint_mask', help="""tracepoints to be disabled inside a given trace group.
1724        (for example value of "0x3" will disable the first two tpoints in this group)""",
1725        type=lambda m: int(m, 16))
1726    p.set_defaults(func=trace_clear_tpoint_mask)
1727
1728    def trace_get_tpoint_group_mask(args):
1729        print_dict(rpc.trace.trace_get_tpoint_group_mask(args.client))
1730
1731    p = subparsers.add_parser('trace_get_tpoint_group_mask', help='get trace point group mask')
1732    p.set_defaults(func=trace_get_tpoint_group_mask)
1733
1734    # log
1735    def log_set_flag(args):
1736        rpc.log.log_set_flag(args.client, flag=args.flag)
1737
1738    p = subparsers.add_parser('log_set_flag', help='set log flag')
1739    p.add_argument(
1740        'flag', help='log flag we want to set. (for example "nvme").')
1741    p.set_defaults(func=log_set_flag)
1742
1743    def log_clear_flag(args):
1744        rpc.log.log_clear_flag(args.client, flag=args.flag)
1745
1746    p = subparsers.add_parser('log_clear_flag', help='clear log flag')
1747    p.add_argument(
1748        'flag', help='log flag we want to clear. (for example "nvme").')
1749    p.set_defaults(func=log_clear_flag)
1750
1751    def log_get_flags(args):
1752        print_dict(rpc.log.log_get_flags(args.client))
1753
1754    p = subparsers.add_parser('log_get_flags', help='get log flags')
1755    p.set_defaults(func=log_get_flags)
1756
1757    def log_set_level(args):
1758        rpc.log.log_set_level(args.client, level=args.level)
1759
1760    p = subparsers.add_parser('log_set_level', help='set log level')
1761    p.add_argument('level', help='log level we want to set. (for example "DEBUG").')
1762    p.set_defaults(func=log_set_level)
1763
1764    def log_get_level(args):
1765        print_dict(rpc.log.log_get_level(args.client))
1766
1767    p = subparsers.add_parser('log_get_level', help='get log level')
1768    p.set_defaults(func=log_get_level)
1769
1770    def log_set_print_level(args):
1771        rpc.log.log_set_print_level(args.client, level=args.level)
1772
1773    p = subparsers.add_parser('log_set_print_level', help='set log print level')
1774    p.add_argument('level', help='log print level we want to set. (for example "DEBUG").')
1775    p.set_defaults(func=log_set_print_level)
1776
1777    def log_get_print_level(args):
1778        print_dict(rpc.log.log_get_print_level(args.client))
1779
1780    p = subparsers.add_parser('log_get_print_level', help='get log print level')
1781    p.set_defaults(func=log_get_print_level)
1782
1783    # lvol
1784    def bdev_lvol_create_lvstore(args):
1785        print_json(rpc.lvol.bdev_lvol_create_lvstore(args.client,
1786                                                     bdev_name=args.bdev_name,
1787                                                     lvs_name=args.lvs_name,
1788                                                     cluster_sz=args.cluster_sz,
1789                                                     clear_method=args.clear_method,
1790                                                     num_md_pages_per_cluster_ratio=args.md_pages_per_cluster_ratio))
1791
1792    p = subparsers.add_parser('bdev_lvol_create_lvstore', help='Add logical volume store on base bdev')
1793    p.add_argument('bdev_name', help='base bdev name')
1794    p.add_argument('lvs_name', help='name for lvol store')
1795    p.add_argument('-c', '--cluster-sz', help='size of cluster (in bytes)', type=int, required=False)
1796    p.add_argument('--clear-method', help="""Change clear method for data region.
1797        Available: none, unmap, write_zeroes""", required=False)
1798    p.add_argument('-m', '--md-pages-per-cluster-ratio', help='reserved metadata pages for each cluster', type=int, required=False)
1799    p.set_defaults(func=bdev_lvol_create_lvstore)
1800
1801    def bdev_lvol_rename_lvstore(args):
1802        rpc.lvol.bdev_lvol_rename_lvstore(args.client,
1803                                          old_name=args.old_name,
1804                                          new_name=args.new_name)
1805
1806    p = subparsers.add_parser('bdev_lvol_rename_lvstore', help='Change logical volume store name')
1807    p.add_argument('old_name', help='old name')
1808    p.add_argument('new_name', help='new name')
1809    p.set_defaults(func=bdev_lvol_rename_lvstore)
1810
1811    def bdev_lvol_grow_lvstore(args):
1812        print_dict(rpc.lvol.bdev_lvol_grow_lvstore(args.client,
1813                                                   uuid=args.uuid,
1814                                                   lvs_name=args.lvs_name))
1815
1816    p = subparsers.add_parser('bdev_lvol_grow_lvstore',
1817                              help='Grow the lvstore size to the underlying bdev size')
1818    p.add_argument('-u', '--uuid', help='lvol store UUID', required=False)
1819    p.add_argument('-l', '--lvs-name', help='lvol store name', required=False)
1820    p.set_defaults(func=bdev_lvol_grow_lvstore)
1821
1822    def bdev_lvol_create(args):
1823        print_json(rpc.lvol.bdev_lvol_create(args.client,
1824                                             lvol_name=args.lvol_name,
1825                                             size=args.size * 1024 * 1024,
1826                                             thin_provision=args.thin_provision,
1827                                             clear_method=args.clear_method,
1828                                             uuid=args.uuid,
1829                                             lvs_name=args.lvs_name))
1830
1831    p = subparsers.add_parser('bdev_lvol_create', help='Add a bdev with an logical volume backend')
1832    p.add_argument('-u', '--uuid', help='lvol store UUID', required=False)
1833    p.add_argument('-l', '--lvs-name', help='lvol store name', required=False)
1834    p.add_argument('-t', '--thin-provision', action='store_true', help='create lvol bdev as thin provisioned')
1835    p.add_argument('-c', '--clear-method', help="""Change default data clusters clear method.
1836        Available: none, unmap, write_zeroes""", required=False)
1837    p.add_argument('lvol_name', help='name for this lvol')
1838    p.add_argument('size', help='size in MiB for this bdev', type=int)
1839    p.set_defaults(func=bdev_lvol_create)
1840
1841    def bdev_lvol_snapshot(args):
1842        print_json(rpc.lvol.bdev_lvol_snapshot(args.client,
1843                                               lvol_name=args.lvol_name,
1844                                               snapshot_name=args.snapshot_name))
1845
1846    p = subparsers.add_parser('bdev_lvol_snapshot', help='Create a snapshot of an lvol bdev')
1847    p.add_argument('lvol_name', help='lvol bdev name')
1848    p.add_argument('snapshot_name', help='lvol snapshot name')
1849    p.set_defaults(func=bdev_lvol_snapshot)
1850
1851    def bdev_lvol_clone(args):
1852        print_json(rpc.lvol.bdev_lvol_clone(args.client,
1853                                            snapshot_name=args.snapshot_name,
1854                                            clone_name=args.clone_name))
1855
1856    p = subparsers.add_parser('bdev_lvol_clone', help='Create a clone of an lvol snapshot')
1857    p.add_argument('snapshot_name', help='lvol snapshot name')
1858    p.add_argument('clone_name', help='lvol clone name')
1859    p.set_defaults(func=bdev_lvol_clone)
1860
1861    def bdev_lvol_rename(args):
1862        rpc.lvol.bdev_lvol_rename(args.client,
1863                                  old_name=args.old_name,
1864                                  new_name=args.new_name)
1865
1866    p = subparsers.add_parser('bdev_lvol_rename', help='Change lvol bdev name')
1867    p.add_argument('old_name', help='lvol bdev name')
1868    p.add_argument('new_name', help='new lvol name')
1869    p.set_defaults(func=bdev_lvol_rename)
1870
1871    def bdev_lvol_inflate(args):
1872        rpc.lvol.bdev_lvol_inflate(args.client,
1873                                   name=args.name)
1874
1875    p = subparsers.add_parser('bdev_lvol_inflate', help='Make thin provisioned lvol a thick provisioned lvol')
1876    p.add_argument('name', help='lvol bdev name')
1877    p.set_defaults(func=bdev_lvol_inflate)
1878
1879    def bdev_lvol_decouple_parent(args):
1880        rpc.lvol.bdev_lvol_decouple_parent(args.client,
1881                                           name=args.name)
1882
1883    p = subparsers.add_parser('bdev_lvol_decouple_parent', help='Decouple parent of lvol')
1884    p.add_argument('name', help='lvol bdev name')
1885    p.set_defaults(func=bdev_lvol_decouple_parent)
1886
1887    def bdev_lvol_resize(args):
1888        rpc.lvol.bdev_lvol_resize(args.client,
1889                                  name=args.name,
1890                                  size=args.size * 1024 * 1024)
1891
1892    p = subparsers.add_parser('bdev_lvol_resize', help='Resize existing lvol bdev')
1893    p.add_argument('name', help='lvol bdev name')
1894    p.add_argument('size', help='new size in MiB for this bdev', type=int)
1895    p.set_defaults(func=bdev_lvol_resize)
1896
1897    def bdev_lvol_set_read_only(args):
1898        rpc.lvol.bdev_lvol_set_read_only(args.client,
1899                                         name=args.name)
1900
1901    p = subparsers.add_parser('bdev_lvol_set_read_only', help='Mark lvol bdev as read only')
1902    p.add_argument('name', help='lvol bdev name')
1903    p.set_defaults(func=bdev_lvol_set_read_only)
1904
1905    def bdev_lvol_delete(args):
1906        rpc.lvol.bdev_lvol_delete(args.client,
1907                                  name=args.name)
1908
1909    p = subparsers.add_parser('bdev_lvol_delete', help='Destroy a logical volume')
1910    p.add_argument('name', help='lvol bdev name')
1911    p.set_defaults(func=bdev_lvol_delete)
1912
1913    def bdev_lvol_delete_lvstore(args):
1914        rpc.lvol.bdev_lvol_delete_lvstore(args.client,
1915                                          uuid=args.uuid,
1916                                          lvs_name=args.lvs_name)
1917
1918    p = subparsers.add_parser('bdev_lvol_delete_lvstore', help='Destroy an logical volume store')
1919    p.add_argument('-u', '--uuid', help='lvol store UUID', required=False)
1920    p.add_argument('-l', '--lvs-name', help='lvol store name', required=False)
1921    p.set_defaults(func=bdev_lvol_delete_lvstore)
1922
1923    def bdev_lvol_get_lvstores(args):
1924        print_dict(rpc.lvol.bdev_lvol_get_lvstores(args.client,
1925                                                   uuid=args.uuid,
1926                                                   lvs_name=args.lvs_name))
1927
1928    p = subparsers.add_parser('bdev_lvol_get_lvstores', help='Display current logical volume store list')
1929    p.add_argument('-u', '--uuid', help='lvol store UUID', required=False)
1930    p.add_argument('-l', '--lvs-name', help='lvol store name', required=False)
1931    p.set_defaults(func=bdev_lvol_get_lvstores)
1932
1933    def bdev_raid_get_bdevs(args):
1934        print_array(rpc.bdev.bdev_raid_get_bdevs(args.client,
1935                                                 category=args.category))
1936
1937    p = subparsers.add_parser('bdev_raid_get_bdevs',
1938                              help="""This is used to list all the raid bdev names based on the input category
1939    requested. Category should be one of 'all', 'online', 'configuring' or 'offline'. 'all' means all the raid bdevs whether
1940    they are online or configuring or offline. 'online' is the raid bdev which is registered with bdev layer. 'configuring'
1941    is the raid bdev which does not have full configuration discovered yet. 'offline' is the raid bdev which is not registered
1942    with bdev as of now and it has encountered any error or user has requested to offline the raid bdev""")
1943    p.add_argument('category', help='all or online or configuring or offline')
1944    p.set_defaults(func=bdev_raid_get_bdevs)
1945
1946    def bdev_raid_create(args):
1947        base_bdevs = []
1948        for u in args.base_bdevs.strip().split(" "):
1949            base_bdevs.append(u)
1950
1951        rpc.bdev.bdev_raid_create(args.client,
1952                                  name=args.name,
1953                                  strip_size_kb=args.strip_size_kb,
1954                                  raid_level=args.raid_level,
1955                                  base_bdevs=base_bdevs)
1956    p = subparsers.add_parser('bdev_raid_create', help='Create new raid bdev')
1957    p.add_argument('-n', '--name', help='raid bdev name', required=True)
1958    p.add_argument('-z', '--strip-size-kb', help='strip size in KB', type=int)
1959    p.add_argument('-r', '--raid-level', help='raid level, raid0 and a special level concat are supported', required=True)
1960    p.add_argument('-b', '--base-bdevs', help='base bdevs name, whitespace separated list in quotes', required=True)
1961    p.set_defaults(func=bdev_raid_create)
1962
1963    def bdev_raid_delete(args):
1964        rpc.bdev.bdev_raid_delete(args.client,
1965                                  name=args.name)
1966    p = subparsers.add_parser('bdev_raid_delete', help='Delete existing raid bdev')
1967    p.add_argument('name', help='raid bdev name')
1968    p.set_defaults(func=bdev_raid_delete)
1969
1970    # split
1971    def bdev_split_create(args):
1972        print_array(rpc.bdev.bdev_split_create(args.client,
1973                                               base_bdev=args.base_bdev,
1974                                               split_count=args.split_count,
1975                                               split_size_mb=args.split_size_mb))
1976
1977    p = subparsers.add_parser('bdev_split_create',
1978                              help="""Add given disk name to split config. If bdev with base_name
1979    name exist the split bdevs will be created right away, if not split bdevs will be created when base bdev became
1980    available (during examination process).""")
1981    p.add_argument('base_bdev', help='base bdev name')
1982    p.add_argument('-s', '--split-size-mb', help='size in MiB for each bdev', type=int)
1983    p.add_argument('split_count', help="""Optional - number of split bdevs to create. Total size * split_count must not
1984    exceed the base bdev size.""", type=int)
1985    p.set_defaults(func=bdev_split_create)
1986
1987    def bdev_split_delete(args):
1988        rpc.bdev.bdev_split_delete(args.client,
1989                                   base_bdev=args.base_bdev)
1990
1991    p = subparsers.add_parser('bdev_split_delete', help="""Delete split config with all created splits.""")
1992    p.add_argument('base_bdev', help='base bdev name')
1993    p.set_defaults(func=bdev_split_delete)
1994
1995    # ftl
1996    def bdev_ftl_create(args):
1997        print_dict(rpc.bdev.bdev_ftl_create(args.client,
1998                                            name=args.name,
1999                                            base_bdev=args.base_bdev,
2000                                            uuid=args.uuid,
2001                                            cache=args.cache,
2002                                            overprovisioning=args.overprovisioning,
2003                                            core_mask=args.core_mask))
2004
2005    p = subparsers.add_parser('bdev_ftl_create', help='Add FTL bdev')
2006    p.add_argument('-b', '--name', help="Name of the bdev", required=True)
2007    p.add_argument('-d', '--base-bdev', help='Name of bdev used as underlying device',
2008                   required=True)
2009    p.add_argument('-u', '--uuid', help='UUID of restored bdev (not applicable when creating new '
2010                   'instance): e.g. b286d19a-0059-4709-abcd-9f7732b1567d (optional)')
2011    p.add_argument('-c', '--cache', help='Name of the bdev to be used as a write buffer cache',
2012                   required=True)
2013    p.add_argument('--overprovisioning', help='Percentage of device used for relocation, not exposed'
2014                   ' to user (optional); default 20', type=int)
2015    p.add_argument('--core-mask', help='CPU core mask - which cores will be used for ftl core thread, '
2016                   'by default core thread will be set to the main application core (optional)')
2017    p.set_defaults(func=bdev_ftl_create)
2018
2019    def bdev_ftl_load(args):
2020        print_dict(rpc.bdev.bdev_ftl_load(args.client,
2021                                          name=args.name,
2022                                          base_bdev=args.base_bdev,
2023                                          uuid=args.uuid,
2024                                          cache=args.cache,
2025                                          overprovisioning=args.overprovisioning,
2026                                          core_mask=args.core_mask))
2027
2028    p = subparsers.add_parser('bdev_ftl_load', help='Load FTL bdev')
2029    p.add_argument('-b', '--name', help="Name of the bdev", required=True)
2030    p.add_argument('-d', '--base-bdev', help='Name of bdev used as underlying device',
2031                   required=True)
2032    p.add_argument('-u', '--uuid', help='UUID of restored bdev', required=True)
2033    p.add_argument('-c', '--cache', help='Name of the bdev to be used as a write buffer cache',
2034                   required=True)
2035    p.add_argument('--overprovisioning', help='Percentage of device used for relocation, not exposed'
2036                   ' to user (optional); default 20', type=int)
2037    p.add_argument('--core-mask', help='CPU core mask - which cores will be used for ftl core thread, '
2038                   'by default core thread will be set to the main application core (optional)')
2039    p.set_defaults(func=bdev_ftl_load)
2040
2041    def bdev_ftl_unload(args):
2042        print_dict(rpc.bdev.bdev_ftl_unload(args.client, name=args.name))
2043
2044    p = subparsers.add_parser('bdev_ftl_unload', help='Unload FTL bdev')
2045    p.add_argument('-b', '--name', help="Name of the bdev", required=True)
2046    p.set_defaults(func=bdev_ftl_unload)
2047
2048    def bdev_ftl_delete(args):
2049        print_dict(rpc.bdev.bdev_ftl_delete(args.client, name=args.name))
2050
2051    p = subparsers.add_parser('bdev_ftl_delete', help='Delete FTL bdev')
2052    p.add_argument('-b', '--name', help="Name of the bdev", required=True)
2053    p.set_defaults(func=bdev_ftl_delete)
2054
2055    # vmd
2056    def enable_vmd(args):
2057        print_dict(rpc.vmd.enable_vmd(args.client))
2058
2059    p = subparsers.add_parser('enable_vmd', help='Enable VMD enumeration')
2060    p.set_defaults(func=enable_vmd)
2061
2062    # nbd
2063    def nbd_start_disk(args):
2064        print(rpc.nbd.nbd_start_disk(args.client,
2065                                     bdev_name=args.bdev_name,
2066                                     nbd_device=args.nbd_device))
2067
2068    p = subparsers.add_parser('nbd_start_disk',
2069                              help='Export a bdev as an nbd disk')
2070    p.add_argument('bdev_name', help='Blockdev name to be exported. Example: Malloc0.')
2071    p.add_argument('nbd_device', help='Nbd device name to be assigned. Example: /dev/nbd0.', nargs='?')
2072    p.set_defaults(func=nbd_start_disk)
2073
2074    def nbd_stop_disk(args):
2075        rpc.nbd.nbd_stop_disk(args.client,
2076                              nbd_device=args.nbd_device)
2077
2078    p = subparsers.add_parser('nbd_stop_disk',
2079                              help='Stop an nbd disk')
2080    p.add_argument('nbd_device', help='Nbd device name to be stopped. Example: /dev/nbd0.')
2081    p.set_defaults(func=nbd_stop_disk)
2082
2083    def nbd_get_disks(args):
2084        print_dict(rpc.nbd.nbd_get_disks(args.client,
2085                                         nbd_device=args.nbd_device))
2086
2087    p = subparsers.add_parser('nbd_get_disks',
2088                              help='Display full or specified nbd device list')
2089    p.add_argument('-n', '--nbd-device', help="Path of the nbd device. Example: /dev/nbd0", required=False)
2090    p.set_defaults(func=nbd_get_disks)
2091
2092    # NVMe-oF
2093    def nvmf_set_max_subsystems(args):
2094        rpc.nvmf.nvmf_set_max_subsystems(args.client,
2095                                         max_subsystems=args.max_subsystems)
2096
2097    p = subparsers.add_parser('nvmf_set_max_subsystems',
2098                              help='Set the maximum number of NVMf target subsystems')
2099    p.add_argument('-x', '--max-subsystems', help='Max number of NVMf subsystems', type=int, required=True)
2100    p.set_defaults(func=nvmf_set_max_subsystems)
2101
2102    def nvmf_set_config(args):
2103        rpc.nvmf.nvmf_set_config(args.client,
2104                                 passthru_identify_ctrlr=args.passthru_identify_ctrlr,
2105                                 poll_groups_mask=args.poll_groups_mask,
2106                                 discovery_filter=args.discovery_filter)
2107
2108    p = subparsers.add_parser('nvmf_set_config', help='Set NVMf target config')
2109    p.add_argument('-i', '--passthru-identify-ctrlr', help="""Passthrough fields like serial number and model number
2110    when the controller has a single namespace that is an NVMe bdev""", action='store_true')
2111    p.add_argument('-m', '--poll-groups-mask', help='Set cpumask for NVMf poll groups (optional)', type=str)
2112    p.add_argument('-d', '--discovery-filter', help="""Set discovery filter (optional), possible values are: `match_any` (default) or
2113         comma separated values: `transport`, `address`, `svcid`""", type=str)
2114    p.set_defaults(func=nvmf_set_config)
2115
2116    def nvmf_create_transport(args):
2117        rpc.nvmf.nvmf_create_transport(**vars(args))
2118
2119    p = subparsers.add_parser('nvmf_create_transport', help='Create NVMf transport')
2120    p.add_argument('-t', '--trtype', help='Transport type (ex. RDMA)', type=str, required=True)
2121    p.add_argument('-g', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2122    p.add_argument('-q', '--max-queue-depth', help='Max number of outstanding I/O per queue', type=int)
2123    p.add_argument('-m', '--max-io-qpairs-per-ctrlr', help='Max number of IO qpairs per controller', type=int)
2124    p.add_argument('-c', '--in-capsule-data-size', help='Max number of in-capsule data size', type=int)
2125    p.add_argument('-i', '--max-io-size', help='Max I/O size (bytes)', type=int)
2126    p.add_argument('-u', '--io-unit-size', help='I/O unit size (bytes)', type=int)
2127    p.add_argument('-a', '--max-aq-depth', help='Max number of admin cmds per AQ', type=int)
2128    p.add_argument('-n', '--num-shared-buffers', help='The number of pooled data buffers available to the transport', type=int)
2129    p.add_argument('-b', '--buf-cache-size', help='The number of shared buffers to reserve for each poll group', type=int)
2130    p.add_argument('-z', '--zcopy', action='store_true', help='''Use zero-copy operations if the
2131    underlying bdev supports them''')
2132    p.add_argument('-d', '--num-cqe', help="""The number of CQ entires. Only used when no_srq=true.
2133    Relevant only for RDMA transport""", type=int)
2134    p.add_argument('-s', '--max-srq-depth', help='Max number of outstanding I/O per SRQ. Relevant only for RDMA transport', type=int)
2135    p.add_argument('-r', '--no-srq', action='store_true', help='Disable per-thread shared receive queue. Relevant only for RDMA transport')
2136    p.add_argument('-o', '--c2h-success', action='store_false', help='Disable C2H success optimization. Relevant only for TCP transport')
2137    p.add_argument('-f', '--dif-insert-or-strip', action='store_true', help='Enable DIF insert/strip. Relevant only for TCP transport')
2138    p.add_argument('-y', '--sock-priority', help='The sock priority of the tcp connection. Relevant only for TCP transport', type=int)
2139    p.add_argument('-l', '--acceptor-backlog', help='Pending connections allowed at one time. Relevant only for RDMA transport', type=int)
2140    p.add_argument('-x', '--abort-timeout-sec', help='Abort execution timeout value, in seconds', type=int)
2141    p.add_argument('-w', '--no-wr-batching', action='store_true', help='Disable work requests batching. Relevant only for RDMA transport')
2142    p.add_argument('-e', '--control-msg-num', help="""The number of control messages per poll group.
2143    Relevant only for TCP transport""", type=int)
2144    p.add_argument('-M', '--disable-mappable-bar0', action='store_true', help="""Disable mmap() of BAR0.
2145    Relevant only for VFIO-USER transport""")
2146    p.add_argument('-I', '--disable-adaptive-irq', action='store_true', help="""Disable adaptive interrupt feature.
2147    Relevant only for VFIO-USER transport""")
2148    p.add_argument('-S', '--disable-shadow-doorbells', action='store_true', help="""Disable shadow doorbell support.
2149    Relevant only for VFIO-USER transport""")
2150    p.add_argument('--acceptor-poll-rate', help='Polling interval of the acceptor for incoming connections (usec)', type=int)
2151    p.set_defaults(func=nvmf_create_transport)
2152
2153    def nvmf_get_transports(args):
2154        print_dict(rpc.nvmf.nvmf_get_transports(args.client, trtype=args.trtype, tgt_name=args.tgt_name))
2155
2156    p = subparsers.add_parser('nvmf_get_transports', help='Display nvmf transports or required transport')
2157    p.add_argument('--trtype', help='Transport type (optional)')
2158    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2159    p.set_defaults(func=nvmf_get_transports)
2160
2161    def nvmf_get_subsystems(args):
2162        print_dict(rpc.nvmf.nvmf_get_subsystems(args.client, nqn=args.nqn, tgt_name=args.tgt_name))
2163
2164    p = subparsers.add_parser('nvmf_get_subsystems', help='Display nvmf subsystems or required subsystem')
2165    p.add_argument('nqn', help='Subsystem NQN (optional)', nargs="?")
2166    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2167    p.set_defaults(func=nvmf_get_subsystems)
2168
2169    def nvmf_create_subsystem(args):
2170        rpc.nvmf.nvmf_create_subsystem(args.client,
2171                                       nqn=args.nqn,
2172                                       tgt_name=args.tgt_name,
2173                                       serial_number=args.serial_number,
2174                                       model_number=args.model_number,
2175                                       allow_any_host=args.allow_any_host,
2176                                       max_namespaces=args.max_namespaces,
2177                                       ana_reporting=args.ana_reporting,
2178                                       min_cntlid=args.min_cntlid,
2179                                       max_cntlid=args.max_cntlid)
2180
2181    p = subparsers.add_parser('nvmf_create_subsystem', help='Create an NVMe-oF subsystem')
2182    p.add_argument('nqn', help='Subsystem NQN (ASCII)')
2183    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2184    p.add_argument("-s", "--serial-number", help="""
2185    Format:  'sn' etc
2186    Example: 'SPDK00000000000001'""")
2187    p.add_argument("-d", "--model-number", help="""
2188    Format:  'mn' etc
2189    Example: 'SPDK Controller'""")
2190    p.add_argument("-a", "--allow-any-host", action='store_true', help="Allow any host to connect (don't enforce allowed host NQN list)")
2191    p.add_argument("-m", "--max-namespaces", help="Maximum number of namespaces allowed",
2192                   type=int)
2193    p.add_argument("-r", "--ana-reporting", action='store_true', help="Enable ANA reporting feature")
2194    p.add_argument("-i", "--min_cntlid", help="Minimum controller ID", type=int)
2195    p.add_argument("-I", "--max_cntlid", help="Maximum controller ID", type=int)
2196    p.set_defaults(func=nvmf_create_subsystem)
2197
2198    def nvmf_delete_subsystem(args):
2199        rpc.nvmf.nvmf_delete_subsystem(args.client,
2200                                       nqn=args.subsystem_nqn,
2201                                       tgt_name=args.tgt_name)
2202
2203    p = subparsers.add_parser('nvmf_delete_subsystem', help='Delete a nvmf subsystem')
2204    p.add_argument('subsystem_nqn',
2205                   help='subsystem nqn to be deleted. Example: nqn.2016-06.io.spdk:cnode1.')
2206    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2207    p.set_defaults(func=nvmf_delete_subsystem)
2208
2209    def nvmf_subsystem_add_listener(args):
2210        rpc.nvmf.nvmf_subsystem_add_listener(**vars(args))
2211
2212    p = subparsers.add_parser('nvmf_subsystem_add_listener', help='Add a listener to an NVMe-oF subsystem')
2213    p.add_argument('nqn', help='NVMe-oF subsystem NQN (\'discovery\' can be used as shortcut for discovery NQN)')
2214    p.add_argument('-t', '--trtype', help='NVMe-oF transport type: e.g., rdma', required=True)
2215    p.add_argument('-a', '--traddr', help='NVMe-oF transport address: e.g., an ip address', required=True)
2216    p.add_argument('-p', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2217    p.add_argument('-f', '--adrfam', help='NVMe-oF transport adrfam: e.g., ipv4, ipv6, ib, fc, intra_host')
2218    p.add_argument('-s', '--trsvcid', help='NVMe-oF transport service id: e.g., a port number (required for RDMA or TCP)')
2219    p.set_defaults(func=nvmf_subsystem_add_listener)
2220
2221    def nvmf_subsystem_remove_listener(args):
2222        rpc.nvmf.nvmf_subsystem_remove_listener(args.client,
2223                                                nqn=args.nqn,
2224                                                trtype=args.trtype,
2225                                                traddr=args.traddr,
2226                                                tgt_name=args.tgt_name,
2227                                                adrfam=args.adrfam,
2228                                                trsvcid=args.trsvcid)
2229
2230    p = subparsers.add_parser('nvmf_subsystem_remove_listener', help='Remove a listener from an NVMe-oF subsystem')
2231    p.add_argument('nqn', help='NVMe-oF subsystem NQN (\'discovery\' can be used as shortcut for discovery NQN)')
2232    p.add_argument('-t', '--trtype', help='NVMe-oF transport type: e.g., rdma', required=True)
2233    p.add_argument('-a', '--traddr', help='NVMe-oF transport address: e.g., an ip address', required=True)
2234    p.add_argument('-p', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2235    p.add_argument('-f', '--adrfam', help='NVMe-oF transport adrfam: e.g., ipv4, ipv6, ib, fc, intra_host')
2236    p.add_argument('-s', '--trsvcid', help='NVMe-oF transport service id: e.g., a port number (required for TCP and RDMA transport types)')
2237    p.set_defaults(func=nvmf_subsystem_remove_listener)
2238
2239    def nvmf_subsystem_listener_set_ana_state(args):
2240        rpc.nvmf.nvmf_subsystem_listener_set_ana_state(args.client,
2241                                                       nqn=args.nqn,
2242                                                       ana_state=args.ana_state,
2243                                                       trtype=args.trtype,
2244                                                       traddr=args.traddr,
2245                                                       tgt_name=args.tgt_name,
2246                                                       adrfam=args.adrfam,
2247                                                       trsvcid=args.trsvcid,
2248                                                       anagrpid=args.anagrpid)
2249
2250    p = subparsers.add_parser('nvmf_subsystem_listener_set_ana_state', help='Set ANA state of a listener for an NVMe-oF subsystem')
2251    p.add_argument('nqn', help='NVMe-oF subsystem NQN')
2252    p.add_argument('-n', '--ana-state', help='ANA state to set: optimized, non_optimized, or inaccessible', required=True)
2253    p.add_argument('-t', '--trtype', help='NVMe-oF transport type: e.g., rdma', required=True)
2254    p.add_argument('-a', '--traddr', help='NVMe-oF transport address: e.g., an ip address', required=True)
2255    p.add_argument('-p', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2256    p.add_argument('-f', '--adrfam', help='NVMe-oF transport adrfam: e.g., ipv4, ipv6, ib, fc, intra_host')
2257    p.add_argument('-s', '--trsvcid', help='NVMe-oF transport service id: e.g., a port number')
2258    p.add_argument('-g', '--anagrpid', help='ANA group ID (optional)', type=int)
2259    p.set_defaults(func=nvmf_subsystem_listener_set_ana_state)
2260
2261    def nvmf_subsystem_add_ns(args):
2262        rpc.nvmf.nvmf_subsystem_add_ns(args.client,
2263                                       nqn=args.nqn,
2264                                       bdev_name=args.bdev_name,
2265                                       tgt_name=args.tgt_name,
2266                                       ptpl_file=args.ptpl_file,
2267                                       nsid=args.nsid,
2268                                       nguid=args.nguid,
2269                                       eui64=args.eui64,
2270                                       uuid=args.uuid,
2271                                       anagrpid=args.anagrpid)
2272
2273    p = subparsers.add_parser('nvmf_subsystem_add_ns', help='Add a namespace to an NVMe-oF subsystem')
2274    p.add_argument('nqn', help='NVMe-oF subsystem NQN')
2275    p.add_argument('bdev_name', help='The name of the bdev that will back this namespace')
2276    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2277    p.add_argument('-p', '--ptpl-file', help='The persistent reservation storage location (optional)', type=str)
2278    p.add_argument('-n', '--nsid', help='The requested NSID (optional)', type=int)
2279    p.add_argument('-g', '--nguid', help='Namespace globally unique identifier (optional)')
2280    p.add_argument('-e', '--eui64', help='Namespace EUI-64 identifier (optional)')
2281    p.add_argument('-u', '--uuid', help='Namespace UUID (optional)')
2282    p.add_argument('-a', '--anagrpid', help='ANA group ID (optional)', type=int)
2283    p.set_defaults(func=nvmf_subsystem_add_ns)
2284
2285    def nvmf_subsystem_remove_ns(args):
2286        rpc.nvmf.nvmf_subsystem_remove_ns(args.client,
2287                                          nqn=args.nqn,
2288                                          nsid=args.nsid,
2289                                          tgt_name=args.tgt_name)
2290
2291    p = subparsers.add_parser('nvmf_subsystem_remove_ns', help='Remove a namespace to an NVMe-oF subsystem')
2292    p.add_argument('nqn', help='NVMe-oF subsystem NQN')
2293    p.add_argument('nsid', help='The requested NSID', type=int)
2294    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2295    p.set_defaults(func=nvmf_subsystem_remove_ns)
2296
2297    def nvmf_subsystem_add_host(args):
2298        rpc.nvmf.nvmf_subsystem_add_host(args.client,
2299                                         nqn=args.nqn,
2300                                         host=args.host,
2301                                         tgt_name=args.tgt_name)
2302
2303    p = subparsers.add_parser('nvmf_subsystem_add_host', help='Add a host to an NVMe-oF subsystem')
2304    p.add_argument('nqn', help='NVMe-oF subsystem NQN')
2305    p.add_argument('host', help='Host NQN to allow')
2306    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2307    p.set_defaults(func=nvmf_subsystem_add_host)
2308
2309    def nvmf_subsystem_remove_host(args):
2310        rpc.nvmf.nvmf_subsystem_remove_host(args.client,
2311                                            nqn=args.nqn,
2312                                            host=args.host,
2313                                            tgt_name=args.tgt_name)
2314
2315    p = subparsers.add_parser('nvmf_subsystem_remove_host', help='Remove a host from an NVMe-oF subsystem')
2316    p.add_argument('nqn', help='NVMe-oF subsystem NQN')
2317    p.add_argument('host', help='Host NQN to remove')
2318    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2319    p.set_defaults(func=nvmf_subsystem_remove_host)
2320
2321    def nvmf_subsystem_allow_any_host(args):
2322        rpc.nvmf.nvmf_subsystem_allow_any_host(args.client,
2323                                               nqn=args.nqn,
2324                                               disable=args.disable,
2325                                               tgt_name=args.tgt_name)
2326
2327    p = subparsers.add_parser('nvmf_subsystem_allow_any_host', help='Allow any host to connect to the subsystem')
2328    p.add_argument('nqn', help='NVMe-oF subsystem NQN')
2329    p.add_argument('-e', '--enable', action='store_true', help='Enable allowing any host')
2330    p.add_argument('-d', '--disable', action='store_true', help='Disable allowing any host')
2331    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2332    p.set_defaults(func=nvmf_subsystem_allow_any_host)
2333
2334    def nvmf_subsystem_get_controllers(args):
2335        print_dict(rpc.nvmf.nvmf_subsystem_get_controllers(args.client,
2336                                                           nqn=args.nqn,
2337                                                           tgt_name=args.tgt_name))
2338
2339    p = subparsers.add_parser('nvmf_subsystem_get_controllers',
2340                              help='Display controllers of an NVMe-oF subsystem.')
2341    p.add_argument('nqn', help='NVMe-oF subsystem NQN')
2342    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2343    p.set_defaults(func=nvmf_subsystem_get_controllers)
2344
2345    def nvmf_subsystem_get_qpairs(args):
2346        print_dict(rpc.nvmf.nvmf_subsystem_get_qpairs(args.client,
2347                                                      nqn=args.nqn,
2348                                                      tgt_name=args.tgt_name))
2349
2350    p = subparsers.add_parser('nvmf_subsystem_get_qpairs',
2351                              help='Display queue pairs of an NVMe-oF subsystem.')
2352    p.add_argument('nqn', help='NVMe-oF subsystem NQN')
2353    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2354    p.set_defaults(func=nvmf_subsystem_get_qpairs)
2355
2356    def nvmf_subsystem_get_listeners(args):
2357        print_dict(rpc.nvmf.nvmf_subsystem_get_listeners(args.client,
2358                                                         nqn=args.nqn,
2359                                                         tgt_name=args.tgt_name))
2360
2361    p = subparsers.add_parser('nvmf_subsystem_get_listeners',
2362                              help='Display listeners of an NVMe-oF subsystem.')
2363    p.add_argument('nqn', help='NVMe-oF subsystem NQN')
2364    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2365    p.set_defaults(func=nvmf_subsystem_get_listeners)
2366
2367    def nvmf_get_stats(args):
2368        print_dict(rpc.nvmf.nvmf_get_stats(args.client, tgt_name=args.tgt_name))
2369
2370    p = subparsers.add_parser(
2371        'nvmf_get_stats', help='Display current statistics for NVMf subsystem')
2372    p.add_argument('-t', '--tgt-name', help='The name of the parent NVMe-oF target (optional)', type=str)
2373    p.set_defaults(func=nvmf_get_stats)
2374
2375    def nvmf_set_crdt(args):
2376        print_dict(rpc.nvmf.nvmf_set_crdt(args.client, args.crdt1, args.crdt2, args.crdt3))
2377
2378    p = subparsers.add_parser(
2379        'nvmf_set_crdt',
2380        help="""Set the 3 crdt (Command Retry Delay Time) values for NVMf subsystem. All
2381        values are in units of 100 milliseconds (same as the NVM Express specification).""")
2382    p.add_argument('-t1', '--crdt1', help='Command Retry Delay Time 1, in units of 100 milliseconds', type=int)
2383    p.add_argument('-t2', '--crdt2', help='Command Retry Delay Time 2, in units of 100 milliseconds', type=int)
2384    p.add_argument('-t3', '--crdt3', help='Command Retry Delay Time 3, in units of 100 milliseconds', type=int)
2385    p.set_defaults(func=nvmf_set_crdt)
2386
2387    # pmem
2388    def bdev_pmem_create_pool(args):
2389        num_blocks = int((args.total_size * 1024 * 1024) / args.block_size)
2390        rpc.pmem.bdev_pmem_create_pool(args.client,
2391                                       pmem_file=args.pmem_file,
2392                                       num_blocks=num_blocks,
2393                                       block_size=args.block_size)
2394
2395    p = subparsers.add_parser('bdev_pmem_create_pool', help='Create pmem pool')
2396    p.add_argument('pmem_file', help='Path to pmemblk pool file')
2397    p.add_argument('total_size', help='Size of pmem bdev in MB (int > 0)', type=int)
2398    p.add_argument('block_size', help='Block size for this pmem pool', type=int)
2399    p.set_defaults(func=bdev_pmem_create_pool)
2400
2401    def bdev_pmem_get_pool_info(args):
2402        print_dict(rpc.pmem.bdev_pmem_get_pool_info(args.client,
2403                                                    pmem_file=args.pmem_file))
2404
2405    p = subparsers.add_parser('bdev_pmem_get_pool_info', help='Display pmem pool info and check consistency')
2406    p.add_argument('pmem_file', help='Path to pmemblk pool file')
2407    p.set_defaults(func=bdev_pmem_get_pool_info)
2408
2409    def bdev_pmem_delete_pool(args):
2410        rpc.pmem.bdev_pmem_delete_pool(args.client,
2411                                       pmem_file=args.pmem_file)
2412
2413    p = subparsers.add_parser('bdev_pmem_delete_pool', help='Delete pmem pool')
2414    p.add_argument('pmem_file', help='Path to pmemblk pool file')
2415    p.set_defaults(func=bdev_pmem_delete_pool)
2416
2417    # subsystem
2418    def framework_get_subsystems(args):
2419        print_dict(rpc.subsystem.framework_get_subsystems(args.client))
2420
2421    p = subparsers.add_parser('framework_get_subsystems',
2422                              help="""Print subsystems array in initialization order. Each subsystem
2423    entry contain (unsorted) array of subsystems it depends on.""")
2424    p.set_defaults(func=framework_get_subsystems)
2425
2426    def framework_get_config(args):
2427        print_dict(rpc.subsystem.framework_get_config(args.client, args.name))
2428
2429    p = subparsers.add_parser('framework_get_config', help="""Print subsystem configuration""")
2430    p.add_argument('name', help='Name of subsystem to query')
2431    p.set_defaults(func=framework_get_config)
2432
2433    # vhost
2434    def vhost_controller_set_coalescing(args):
2435        rpc.vhost.vhost_controller_set_coalescing(args.client,
2436                                                  ctrlr=args.ctrlr,
2437                                                  delay_base_us=args.delay_base_us,
2438                                                  iops_threshold=args.iops_threshold)
2439
2440    p = subparsers.add_parser('vhost_controller_set_coalescing', help='Set vhost controller coalescing')
2441    p.add_argument('ctrlr', help='controller name')
2442    p.add_argument('delay_base_us', help='Base delay time', type=int)
2443    p.add_argument('iops_threshold', help='IOPS threshold when coalescing is enabled', type=int)
2444    p.set_defaults(func=vhost_controller_set_coalescing)
2445
2446    def virtio_blk_create_transport(args):
2447        rpc.vhost.virtio_blk_create_transport(**vars(args))
2448
2449    p = subparsers.add_parser('virtio_blk_create_transport',
2450                              help='Create virtio blk transport')
2451    p.add_argument('name', help='transport name')
2452    p.set_defaults(func=virtio_blk_create_transport)
2453
2454    def vhost_create_scsi_controller(args):
2455        rpc.vhost.vhost_create_scsi_controller(args.client,
2456                                               ctrlr=args.ctrlr,
2457                                               cpumask=args.cpumask)
2458
2459    p = subparsers.add_parser('vhost_create_scsi_controller', help='Add new vhost controller')
2460    p.add_argument('ctrlr', help='controller name')
2461    p.add_argument('--cpumask', help='cpu mask for this controller')
2462    p.set_defaults(func=vhost_create_scsi_controller)
2463
2464    def vhost_scsi_controller_add_target(args):
2465        print_json(rpc.vhost.vhost_scsi_controller_add_target(args.client,
2466                                                              ctrlr=args.ctrlr,
2467                                                              scsi_target_num=args.scsi_target_num,
2468                                                              bdev_name=args.bdev_name))
2469
2470    p = subparsers.add_parser('vhost_scsi_controller_add_target', help='Add lun to vhost controller')
2471    p.add_argument('ctrlr', help='controller name where add lun')
2472    p.add_argument('scsi_target_num', help='scsi_target_num', type=int)
2473    p.add_argument('bdev_name', help='bdev name')
2474    p.set_defaults(func=vhost_scsi_controller_add_target)
2475
2476    def vhost_scsi_controller_remove_target(args):
2477        rpc.vhost.vhost_scsi_controller_remove_target(args.client,
2478                                                      ctrlr=args.ctrlr,
2479                                                      scsi_target_num=args.scsi_target_num)
2480
2481    p = subparsers.add_parser('vhost_scsi_controller_remove_target',
2482                              help='Remove target from vhost controller')
2483    p.add_argument('ctrlr', help='controller name to remove target from')
2484    p.add_argument('scsi_target_num', help='scsi_target_num', type=int)
2485    p.set_defaults(func=vhost_scsi_controller_remove_target)
2486
2487    def vhost_create_blk_controller(args):
2488        rpc.vhost.vhost_create_blk_controller(**vars(args))
2489
2490    p = subparsers.add_parser('vhost_create_blk_controller', help='Add a new vhost block controller')
2491    p.add_argument('ctrlr', help='controller name')
2492    p.add_argument('dev_name', help='device name')
2493    p.add_argument('--cpumask', help='cpu mask for this controller')
2494    p.add_argument('--transport', help='virtio blk transport name (default: vhost_user_blk)')
2495    p.add_argument("-r", "--readonly", action='store_true', help='Set controller as read-only')
2496    p.add_argument("-p", "--packed_ring", action='store_true', help='Set controller as packed ring supported')
2497    p.add_argument("-l", "--packed_ring_recovery", action='store_true', help='Enable packed ring live recovery')
2498    p.set_defaults(func=vhost_create_blk_controller)
2499
2500    def vhost_get_controllers(args):
2501        print_dict(rpc.vhost.vhost_get_controllers(args.client, args.name))
2502
2503    p = subparsers.add_parser('vhost_get_controllers', help='List all or specific vhost controller(s)')
2504    p.add_argument('-n', '--name', help="Name of vhost controller", required=False)
2505    p.set_defaults(func=vhost_get_controllers)
2506
2507    def vhost_delete_controller(args):
2508        rpc.vhost.vhost_delete_controller(args.client,
2509                                          ctrlr=args.ctrlr)
2510
2511    p = subparsers.add_parser('vhost_delete_controller', help='Delete a vhost controller')
2512    p.add_argument('ctrlr', help='controller name')
2513    p.set_defaults(func=vhost_delete_controller)
2514
2515    def bdev_virtio_attach_controller(args):
2516        print_array(rpc.vhost.bdev_virtio_attach_controller(args.client,
2517                                                            name=args.name,
2518                                                            trtype=args.trtype,
2519                                                            traddr=args.traddr,
2520                                                            dev_type=args.dev_type,
2521                                                            vq_count=args.vq_count,
2522                                                            vq_size=args.vq_size))
2523
2524    p = subparsers.add_parser('bdev_virtio_attach_controller',
2525                              help="""Attach virtio controller using provided
2526    transport type and device type. This will also create bdevs for any block devices connected to the
2527    controller (for example, SCSI devices for a virtio-scsi controller).
2528    Result is array of added bdevs.""")
2529    p.add_argument('name', help="Use this name as base for new created bdevs")
2530    p.add_argument('-t', '--trtype',
2531                   help='Virtio target transport type: pci or user', required=True)
2532    p.add_argument('-a', '--traddr',
2533                   help='Transport type specific target address: e.g. UNIX domain socket path or BDF', required=True)
2534    p.add_argument('-d', '--dev-type',
2535                   help='Device type: blk or scsi', required=True)
2536    p.add_argument('--vq-count', help='Number of virtual queues to be used.', type=int)
2537    p.add_argument('--vq-size', help='Size of each queue', type=int)
2538    p.set_defaults(func=bdev_virtio_attach_controller)
2539
2540    def bdev_virtio_scsi_get_devices(args):
2541        print_dict(rpc.vhost.bdev_virtio_scsi_get_devices(args.client))
2542
2543    p = subparsers.add_parser('bdev_virtio_scsi_get_devices', help='List all Virtio-SCSI devices.')
2544    p.set_defaults(func=bdev_virtio_scsi_get_devices)
2545
2546    def bdev_virtio_detach_controller(args):
2547        rpc.vhost.bdev_virtio_detach_controller(args.client,
2548                                                name=args.name)
2549
2550    p = subparsers.add_parser('bdev_virtio_detach_controller', help="""Remove a Virtio device
2551    This will delete all bdevs exposed by this device""")
2552    p.add_argument('name', help='Virtio device name. E.g. VirtioUser0')
2553    p.set_defaults(func=bdev_virtio_detach_controller)
2554
2555    def bdev_virtio_blk_set_hotplug(args):
2556        rpc.vhost.bdev_virtio_blk_set_hotplug(args.client, enable=args.enable, period_us=args.period_us)
2557
2558    p = subparsers.add_parser('bdev_virtio_blk_set_hotplug', help='Set hotplug options for bdev virtio_blk type.')
2559    p.add_argument('-d', '--disable', dest='enable', default=False, action='store_false', help="Disable hotplug (default)")
2560    p.add_argument('-e', '--enable', dest='enable', action='store_true', help="Enable hotplug")
2561    p.add_argument('-r', '--period-us',
2562                   help='How often the hotplug is processed for insert and remove events', type=int)
2563    p.set_defaults(func=bdev_virtio_blk_set_hotplug)
2564
2565    # accel_fw
2566    def accel_get_opc_assignments(args):
2567        print_dict(rpc.accel.accel_get_opc_assignments(args.client))
2568
2569    p = subparsers.add_parser('accel_get_opc_assignments', help='Get list of opcode name to engine assignments.')
2570    p.set_defaults(func=accel_get_opc_assignments)
2571
2572    def accel_get_engine_info(args):
2573        print_dict(rpc.accel.accel_get_engine_info(args.client))
2574
2575    p = subparsers.add_parser('accel_get_engine_info', help='Get list of valid engine names and their operations.')
2576    p.set_defaults(func=accel_get_engine_info)
2577
2578    def accel_assign_opc(args):
2579        rpc.accel.accel_assign_opc(args.client, opname=args.opname, engine=args.engine)
2580
2581    p = subparsers.add_parser('accel_assign_opc', help='Manually assign an operation to an engine.')
2582    p.add_argument('-o', '--opname', help='opname')
2583    p.add_argument('-e', '--engine', help='name of engine')
2584    p.set_defaults(func=accel_assign_opc)
2585
2586    # ioat
2587    def ioat_scan_accel_engine(args):
2588        rpc.ioat.ioat_scan_accel_engine(args.client)
2589
2590    p = subparsers.add_parser('ioat_scan_accel_engine', help='Enable IOAT accel engine offload.')
2591    p.set_defaults(func=ioat_scan_accel_engine)
2592
2593    # dsa
2594    def dsa_scan_accel_engine(args):
2595        rpc.dsa.dsa_scan_accel_engine(args.client, config_kernel_mode=args.config_kernel_mode)
2596
2597    p = subparsers.add_parser('dsa_scan_accel_engine',
2598                              help='Set config and enable dsa accel engine offload.')
2599    p.add_argument('-k', '--config-kernel-mode', help='Use Kernel mode dsa',
2600                   action='store_true', dest='config_kernel_mode')
2601    p.set_defaults(func=dsa_scan_accel_engine, config_kernel_mode=None)
2602
2603    # iaa
2604    def iaa_scan_accel_engine(args):
2605        rpc.iaa.iaa_scan_accel_engine(args.client)
2606
2607    p = subparsers.add_parser('iaa_scan_accel_engine',
2608                              help='Set config and enable iaa accel engine offload.')
2609    p.set_defaults(func=iaa_scan_accel_engine)
2610
2611    # opal
2612    def bdev_nvme_opal_init(args):
2613        rpc.nvme.bdev_nvme_opal_init(args.client,
2614                                     nvme_ctrlr_name=args.nvme_ctrlr_name,
2615                                     password=args.password)
2616
2617    p = subparsers.add_parser('bdev_nvme_opal_init', help='take ownership and activate')
2618    p.add_argument('-b', '--nvme-ctrlr-name', help='nvme ctrlr name')
2619    p.add_argument('-p', '--password', help='password for admin')
2620    p.set_defaults(func=bdev_nvme_opal_init)
2621
2622    def bdev_nvme_opal_revert(args):
2623        rpc.nvme.bdev_nvme_opal_revert(args.client,
2624                                       nvme_ctrlr_name=args.nvme_ctrlr_name,
2625                                       password=args.password)
2626    p = subparsers.add_parser('bdev_nvme_opal_revert', help='Revert to default factory settings')
2627    p.add_argument('-b', '--nvme-ctrlr-name', help='nvme ctrlr name')
2628    p.add_argument('-p', '--password', help='password')
2629    p.set_defaults(func=bdev_nvme_opal_revert)
2630
2631    def bdev_opal_create(args):
2632        print_json(rpc.bdev.bdev_opal_create(args.client,
2633                                             nvme_ctrlr_name=args.nvme_ctrlr_name,
2634                                             nsid=args.nsid,
2635                                             locking_range_id=args.locking_range_id,
2636                                             range_start=args.range_start,
2637                                             range_length=args.range_length,
2638                                             password=args.password))
2639
2640    p = subparsers.add_parser('bdev_opal_create', help="""Create opal bdev on specified NVMe controller""")
2641    p.add_argument('-b', '--nvme-ctrlr-name', help='nvme ctrlr name', required=True)
2642    p.add_argument('-n', '--nsid', help='namespace ID (only support nsid=1 for now)', type=int, required=True)
2643    p.add_argument('-i', '--locking-range-id', help='locking range id', type=int, required=True)
2644    p.add_argument('-s', '--range-start', help='locking range start LBA', type=int, required=True)
2645    p.add_argument('-l', '--range-length', help='locking range length (in blocks)', type=int, required=True)
2646    p.add_argument('-p', '--password', help='admin password', required=True)
2647    p.set_defaults(func=bdev_opal_create)
2648
2649    def bdev_opal_get_info(args):
2650        print_dict(rpc.bdev.bdev_opal_get_info(args.client,
2651                                               bdev_name=args.bdev_name,
2652                                               password=args.password))
2653
2654    p = subparsers.add_parser('bdev_opal_get_info', help='get opal locking range info for this bdev')
2655    p.add_argument('-b', '--bdev-name', help='opal bdev')
2656    p.add_argument('-p', '--password', help='password')
2657    p.set_defaults(func=bdev_opal_get_info)
2658
2659    def bdev_opal_delete(args):
2660        rpc.bdev.bdev_opal_delete(args.client,
2661                                  bdev_name=args.bdev_name,
2662                                  password=args.password)
2663
2664    p = subparsers.add_parser('bdev_opal_delete', help="""delete a virtual opal bdev""")
2665    p.add_argument('-b', '--bdev-name', help='opal virtual bdev', required=True)
2666    p.add_argument('-p', '--password', help='admin password', required=True)
2667    p.set_defaults(func=bdev_opal_delete)
2668
2669    def bdev_opal_new_user(args):
2670        rpc.bdev.bdev_opal_new_user(args.client,
2671                                    bdev_name=args.bdev_name,
2672                                    admin_password=args.admin_password,
2673                                    user_id=args.user_id,
2674                                    user_password=args.user_password)
2675
2676    p = subparsers.add_parser('bdev_opal_new_user', help="""Add a user to opal bdev who can set lock state for this bdev""")
2677    p.add_argument('-b', '--bdev-name', help='opal bdev', required=True)
2678    p.add_argument('-p', '--admin-password', help='admin password', required=True)
2679    p.add_argument('-i', '--user-id', help='ID for new user', type=int, required=True)
2680    p.add_argument('-u', '--user-password', help='password set for this user', required=True)
2681    p.set_defaults(func=bdev_opal_new_user)
2682
2683    def bdev_opal_set_lock_state(args):
2684        rpc.bdev.bdev_opal_set_lock_state(args.client,
2685                                          bdev_name=args.bdev_name,
2686                                          user_id=args.user_id,
2687                                          password=args.password,
2688                                          lock_state=args.lock_state)
2689
2690    p = subparsers.add_parser('bdev_opal_set_lock_state', help="""set lock state for an opal bdev""")
2691    p.add_argument('-b', '--bdev-name', help='opal bdev', required=True)
2692    p.add_argument('-i', '--user-id', help='ID of the user who want to set lock state, either admin or a user assigned to this bdev',
2693                   type=int, required=True)
2694    p.add_argument('-p', '--password', help='password of this user', required=True)
2695    p.add_argument('-l', '--lock-state', help='lock state to set, choose from {readwrite, readonly, rwlock}', required=True)
2696    p.set_defaults(func=bdev_opal_set_lock_state)
2697
2698    # bdev_nvme_send_cmd
2699    def bdev_nvme_send_cmd(args):
2700        print_dict(rpc.nvme.bdev_nvme_send_cmd(args.client,
2701                                               name=args.nvme_name,
2702                                               cmd_type=args.cmd_type,
2703                                               data_direction=args.data_direction,
2704                                               cmdbuf=args.cmdbuf,
2705                                               data=args.data,
2706                                               metadata=args.metadata,
2707                                               data_len=args.data_length,
2708                                               metadata_len=args.metadata_length,
2709                                               timeout_ms=args.timeout_ms))
2710
2711    p = subparsers.add_parser('bdev_nvme_send_cmd', help='NVMe passthrough cmd.')
2712    p.add_argument('-n', '--nvme-name', help="""Name of the operating NVMe controller""")
2713    p.add_argument('-t', '--cmd-type', help="""Type of nvme cmd. Valid values are: admin, io""")
2714    p.add_argument('-r', '--data-direction', help="""Direction of data transfer. Valid values are: c2h, h2c""")
2715    p.add_argument('-c', '--cmdbuf', help="""NVMe command encoded by base64 urlsafe""")
2716    p.add_argument('-d', '--data', help="""Data transferring to controller from host, encoded by base64 urlsafe""")
2717    p.add_argument('-m', '--metadata', help="""Metadata transferring to controller from host, encoded by base64 urlsafe""")
2718    p.add_argument('-D', '--data-length', help="""Data length required to transfer from controller to host""", type=int)
2719    p.add_argument('-M', '--metadata-length', help="""Metadata length required to transfer from controller to host""", type=int)
2720    p.add_argument('-T', '--timeout-ms',
2721                   help="""Command execution timeout value, in milliseconds,  if 0, don't track timeout""", type=int)
2722    p.set_defaults(func=bdev_nvme_send_cmd)
2723
2724    # Notifications
2725    def notify_get_types(args):
2726        print_dict(rpc.notify.notify_get_types(args.client))
2727
2728    p = subparsers.add_parser('notify_get_types', help='List available notifications that user can subscribe to.')
2729    p.set_defaults(func=notify_get_types)
2730
2731    def notify_get_notifications(args):
2732        ret = rpc.notify.notify_get_notifications(args.client,
2733                                                  id=args.id,
2734                                                  max=args.max)
2735        print_dict(ret)
2736
2737    p = subparsers.add_parser('notify_get_notifications', help='Get notifications')
2738    p.add_argument('-i', '--id', help="""First ID to start fetching from""", type=int)
2739    p.add_argument('-n', '--max', help="""Maximum number of notifications to return in response""", type=int)
2740    p.set_defaults(func=notify_get_notifications)
2741
2742    def thread_get_stats(args):
2743        print_dict(rpc.app.thread_get_stats(args.client))
2744
2745    p = subparsers.add_parser(
2746        'thread_get_stats', help='Display current statistics of all the threads')
2747    p.set_defaults(func=thread_get_stats)
2748
2749    def thread_set_cpumask(args):
2750        ret = rpc.app.thread_set_cpumask(args.client,
2751                                         id=args.id,
2752                                         cpumask=args.cpumask)
2753    p = subparsers.add_parser('thread_set_cpumask',
2754                              help="""set the cpumask of the thread whose ID matches to the
2755    specified value. The thread may be migrated to one of the specified CPUs.""")
2756    p.add_argument('-i', '--id', type=int, help='thread ID')
2757    p.add_argument('-m', '--cpumask', help='cpumask for this thread')
2758    p.set_defaults(func=thread_set_cpumask)
2759
2760    def log_enable_timestamps(args):
2761        ret = rpc.app.log_enable_timestamps(args.client,
2762                                            enabled=args.enabled)
2763    p = subparsers.add_parser('log_enable_timestamps',
2764                              help='Enable or disable timestamps.')
2765    p.add_argument('-d', '--disable', dest='enabled', default=False, action='store_false', help="Disable timestamps")
2766    p.add_argument('-e', '--enable', dest='enabled', action='store_true', help="Enable timestamps")
2767    p.set_defaults(func=log_enable_timestamps)
2768
2769    def thread_get_pollers(args):
2770        print_dict(rpc.app.thread_get_pollers(args.client))
2771
2772    p = subparsers.add_parser(
2773        'thread_get_pollers', help='Display current pollers of all the threads')
2774    p.set_defaults(func=thread_get_pollers)
2775
2776    def thread_get_io_channels(args):
2777        print_dict(rpc.app.thread_get_io_channels(args.client))
2778
2779    p = subparsers.add_parser(
2780        'thread_get_io_channels', help='Display current IO channels of all the threads')
2781    p.set_defaults(func=thread_get_io_channels)
2782
2783    def env_dpdk_get_mem_stats(args):
2784        print_dict(rpc.env_dpdk.env_dpdk_get_mem_stats(args.client))
2785
2786    p = subparsers.add_parser(
2787        'env_dpdk_get_mem_stats', help='write the dpdk memory stats to a file.')
2788    p.set_defaults(func=env_dpdk_get_mem_stats)
2789
2790    # blobfs
2791    def blobfs_detect(args):
2792        print(rpc.blobfs.blobfs_detect(args.client,
2793                                       bdev_name=args.bdev_name))
2794
2795    p = subparsers.add_parser('blobfs_detect', help='Detect whether a blobfs exists on bdev')
2796    p.add_argument('bdev_name', help='Blockdev name to detect blobfs. Example: Malloc0.')
2797    p.set_defaults(func=blobfs_detect)
2798
2799    def blobfs_create(args):
2800        print(rpc.blobfs.blobfs_create(args.client,
2801                                       bdev_name=args.bdev_name,
2802                                       cluster_sz=args.cluster_sz))
2803
2804    p = subparsers.add_parser('blobfs_create', help='Build a blobfs on bdev')
2805    p.add_argument('bdev_name', help='Blockdev name to build blobfs. Example: Malloc0.')
2806    p.add_argument('-c', '--cluster-sz',
2807                   help="""Size of cluster in bytes (Optional). Must be multiple of 4KB page size. Default and minimal value is 1M.""")
2808    p.set_defaults(func=blobfs_create)
2809
2810    def blobfs_mount(args):
2811        print(rpc.blobfs.blobfs_mount(args.client,
2812                                      bdev_name=args.bdev_name,
2813                                      mountpoint=args.mountpoint))
2814
2815    p = subparsers.add_parser('blobfs_mount', help='Mount a blobfs on bdev to host path by FUSE')
2816    p.add_argument('bdev_name', help='Blockdev name where the blobfs is. Example: Malloc0.')
2817    p.add_argument('mountpoint', help='Mountpoint path in host to mount blobfs. Example: /mnt/.')
2818    p.set_defaults(func=blobfs_mount)
2819
2820    def blobfs_set_cache_size(args):
2821        print(rpc.blobfs.blobfs_set_cache_size(args.client,
2822                                               size_in_mb=args.size_in_mb))
2823
2824    p = subparsers.add_parser('blobfs_set_cache_size', help='Set cache size for blobfs')
2825    p.add_argument('size_in_mb', help='Cache size for blobfs in megabytes.', type=int)
2826    p.set_defaults(func=blobfs_set_cache_size)
2827
2828    # sock
2829    def sock_impl_get_options(args):
2830        print_json(rpc.sock.sock_impl_get_options(args.client,
2831                                                  impl_name=args.impl))
2832
2833    p = subparsers.add_parser('sock_impl_get_options', help="""Get options of socket layer implementation""")
2834    p.add_argument('-i', '--impl', help='Socket implementation name, e.g. posix', required=True)
2835    p.set_defaults(func=sock_impl_get_options)
2836
2837    def sock_impl_set_options(args):
2838        rpc.sock.sock_impl_set_options(args.client,
2839                                       impl_name=args.impl,
2840                                       recv_buf_size=args.recv_buf_size,
2841                                       send_buf_size=args.send_buf_size,
2842                                       enable_recv_pipe=args.enable_recv_pipe,
2843                                       enable_quickack=args.enable_quickack,
2844                                       enable_placement_id=args.enable_placement_id,
2845                                       enable_zerocopy_send_server=args.enable_zerocopy_send_server,
2846                                       enable_zerocopy_send_client=args.enable_zerocopy_send_client,
2847                                       zerocopy_threshold=args.zerocopy_threshold,
2848                                       tls_version=args.tls_version,
2849                                       enable_ktls=args.enable_ktls,
2850                                       psk_key=args.psk_key,
2851                                       psk_identity=args.psk_identity)
2852
2853    p = subparsers.add_parser('sock_impl_set_options', help="""Set options of socket layer implementation""")
2854    p.add_argument('-i', '--impl', help='Socket implementation name, e.g. posix', required=True)
2855    p.add_argument('-r', '--recv-buf-size', help='Size of receive buffer on socket in bytes', type=int)
2856    p.add_argument('-s', '--send-buf-size', help='Size of send buffer on socket in bytes', type=int)
2857    p.add_argument('-p', '--enable-placement-id', help='Option for placement-id. 0:disable,1:incoming_napi,2:incoming_cpu', type=int)
2858    p.add_argument('--enable-recv-pipe', help='Enable receive pipe',
2859                   action='store_true', dest='enable_recv_pipe')
2860    p.add_argument('--disable-recv-pipe', help='Disable receive pipe',
2861                   action='store_false', dest='enable_recv_pipe')
2862    p.add_argument('--enable-quickack', help='Enable quick ACK',
2863                   action='store_true', dest='enable_quickack')
2864    p.add_argument('--disable-quickack', help='Disable quick ACK',
2865                   action='store_false', dest='enable_quickack')
2866    p.add_argument('--enable-zerocopy-send-server', help='Enable zerocopy on send for server sockets',
2867                   action='store_true', dest='enable_zerocopy_send_server')
2868    p.add_argument('--disable-zerocopy-send-server', help='Disable zerocopy on send for server sockets',
2869                   action='store_false', dest='enable_zerocopy_send_server')
2870    p.add_argument('--enable-zerocopy-send-client', help='Enable zerocopy on send for client sockets',
2871                   action='store_true', dest='enable_zerocopy_send_client')
2872    p.add_argument('--disable-zerocopy-send-client', help='Disable zerocopy on send for client sockets',
2873                   action='store_false', dest='enable_zerocopy_send_client')
2874    p.add_argument('--zerocopy-threshold', help='Set zerocopy_threshold in bytes', type=int)
2875    p.add_argument('--tls-version', help='TLS protocol version', type=int)
2876    p.add_argument('--enable-ktls', help='Enable Kernel TLS',
2877                   action='store_true', dest='enable_ktls')
2878    p.add_argument('--disable-ktls', help='Disable Kernel TLS',
2879                   action='store_false', dest='enable_ktls')
2880    p.add_argument('--psk-key', help='Set default PSK KEY', dest='psk_key')
2881    p.add_argument('--psk-identity', help='Set default PSK ID', dest='psk_identity')
2882    p.set_defaults(func=sock_impl_set_options, enable_recv_pipe=None, enable_quickack=None,
2883                   enable_placement_id=None, enable_zerocopy_send_server=None, enable_zerocopy_send_client=None,
2884                   zerocopy_threshold=None, tls_version=None, enable_ktls=None, psk_key=None, psk_identity=None)
2885
2886    def sock_set_default_impl(args):
2887        print_json(rpc.sock.sock_set_default_impl(args.client,
2888                                                  impl_name=args.impl))
2889
2890    p = subparsers.add_parser('sock_set_default_impl', help="""Set the default sock implementation""")
2891    p.add_argument('-i', '--impl', help='Socket implementation name, e.g. posix', required=True)
2892    p.set_defaults(func=sock_set_default_impl)
2893
2894    def framework_get_pci_devices(args):
2895        def splitbuf(buf, step):
2896            return [buf[i:i+step] for i in range(0, len(buf), step)]
2897
2898        devices = rpc.subsystem.framework_get_pci_devices(args.client)
2899        if not args.format_lspci:
2900            print_json(devices)
2901        else:
2902            for devid, dev in enumerate(devices):
2903                print('{} device #{}'.format(dev['address'], devid))
2904                for lineid, line in enumerate(splitbuf(dev['config_space'], 32)):
2905                    print('{:02x}: {}'.format(lineid * 16, ' '.join(splitbuf(line.lower(), 2))))
2906                print()
2907
2908    p = subparsers.add_parser('framework_get_pci_devices', help='''Get a list of attached PCI devices''')
2909    p.add_argument('--format-lspci', help='Format the output in a way to be consumed by lspci -F',
2910                   action='store_true')
2911    p.set_defaults(func=framework_get_pci_devices)
2912
2913    # bdev_nvme_add_error_injection
2914    def bdev_nvme_add_error_injection(args):
2915        print_dict(rpc.nvme.bdev_nvme_add_error_injection(args.client,
2916                                                          name=args.nvme_name,
2917                                                          cmd_type=args.cmd_type,
2918                                                          opc=args.opc,
2919                                                          do_not_submit=args.do_not_submit,
2920                                                          timeout_in_us=args.timeout_in_us,
2921                                                          err_count=args.err_count,
2922                                                          sct=args.sct,
2923                                                          sc=args.sc))
2924    p = subparsers.add_parser('bdev_nvme_add_error_injection',
2925                              help='Add a NVMe command error injection.')
2926    p.add_argument('-n', '--nvme-name', help="""Name of the operating NVMe controller""", required=True)
2927    p.add_argument('-t', '--cmd-type', help="""Type of NVMe command. Valid values are: admin, io""", required=True)
2928    p.add_argument('-o', '--opc', help="""Opcode of the NVMe command.""", required=True, type=int)
2929    p.add_argument('-s', '--do-not-submit',
2930                   help="""Set to true if request should not be submitted to the controller""",
2931                   dest="do_not_submit", action='store_true')
2932    p.add_argument('-w', '--timeout-in-us', help="""Wait specified microseconds when do_not_submit is true""", type=int)
2933    p.add_argument('-e', '--err-count', help="""Number of matching NVMe commands to inject errors""", type=int)
2934    p.add_argument('-u', '--sct', help="""Status code type""", type=int)
2935    p.add_argument('-c', '--sc', help="""Status code""", type=int)
2936    p.set_defaults(func=bdev_nvme_add_error_injection)
2937
2938    # bdev_nvme_remove_error_injection
2939    def bdev_nvme_remove_error_injection(args):
2940        print_dict(rpc.nvme.bdev_nvme_remove_error_injection(args.client,
2941                                                             name=args.nvme_name,
2942                                                             cmd_type=args.cmd_type,
2943                                                             opc=args.opc))
2944    p = subparsers.add_parser('bdev_nvme_remove_error_injection',
2945                              help='Removes a NVMe command error injection.')
2946    p.add_argument('-n', '--nvme-name', help="""Name of the operating NVMe controller""", required=True)
2947    p.add_argument('-t', '--cmd-type', help="""Type of nvme cmd. Valid values are: admin, io""", required=True)
2948    p.add_argument('-o', '--opc', help="""Opcode of the nvme cmd.""", required=True, type=int)
2949    p.set_defaults(func=bdev_nvme_remove_error_injection)
2950
2951    def check_called_name(name):
2952        if name in deprecated_aliases:
2953            print("{} is deprecated, use {} instead.".format(name, deprecated_aliases[name]), file=sys.stderr)
2954
2955    class dry_run_client:
2956        def call(self, method, params=None):
2957            print("Request:\n" + json.dumps({"method": method, "params": params}, indent=2))
2958
2959    def null_print(arg):
2960        pass
2961
2962    def call_rpc_func(args):
2963        args.func(args)
2964        check_called_name(args.called_rpc_name)
2965
2966    def execute_script(parser, client, fd):
2967        executed_rpc = ""
2968        for rpc_call in map(str.rstrip, fd):
2969            if not rpc_call.strip():
2970                continue
2971            executed_rpc = "\n".join([executed_rpc, rpc_call])
2972            rpc_args = shlex.split(rpc_call)
2973            if rpc_args[0][0] == '#':
2974                # Ignore lines starting with # - treat them as comments
2975                continue
2976            args = parser.parse_args(rpc_args)
2977            args.client = client
2978            try:
2979                call_rpc_func(args)
2980            except JSONRPCException as ex:
2981                print("Exception:")
2982                print(executed_rpc.strip() + " <<<")
2983                print(ex.message)
2984                exit(1)
2985
2986    def load_plugin(args):
2987        # Create temporary parser, pull out the plugin parameter, load the module, and then run the real argument parser
2988        plugin_parser = argparse.ArgumentParser(add_help=False)
2989        plugin_parser.add_argument('--plugin', dest='rpc_plugin', help='Module name of plugin with additional RPC commands')
2990
2991        rpc_module = plugin_parser.parse_known_args()[0].rpc_plugin
2992        if args is not None:
2993            rpc_module = plugin_parser.parse_known_args(args)[0].rpc_plugin
2994
2995        if rpc_module is not None:
2996            try:
2997                rpc_plugin = importlib.import_module(rpc_module)
2998                try:
2999                    rpc_plugin.spdk_rpc_plugin_initialize(subparsers)
3000                except AttributeError:
3001                    print("Module %s does not contain 'spdk_rpc_plugin_initialize' function" % rpc_module)
3002            except ModuleNotFoundError:
3003                print("Module %s not found" % rpc_module)
3004
3005    def replace_arg_underscores(args):
3006        # All option names are defined with dashes only - for example: --tgt-name
3007        # But if user used underscores, convert them to dashes (--tgt_name => --tgt-name)
3008        # SPDK was inconsistent previously and had some options with underscores, so
3009        # doing this conversion ensures backward compatibility with older scripts.
3010        for i in range(len(args)):
3011            arg = args[i]
3012            if arg.startswith('--') and "_" in arg:
3013                args[i] = arg.replace('_', '-')
3014
3015    load_plugin(None)
3016
3017    replace_arg_underscores(sys.argv)
3018
3019    args = parser.parse_args()
3020
3021    if sys.stdin.isatty() and not hasattr(args, 'func'):
3022        # No arguments and no data piped through stdin
3023        parser.print_help()
3024        exit(1)
3025    if args.is_server:
3026        for input in sys.stdin:
3027            cmd = shlex.split(input)
3028            replace_arg_underscores(cmd)
3029            try:
3030                load_plugin(cmd)
3031                tmp_args = parser.parse_args(cmd)
3032            except SystemExit as ex:
3033                print("**STATUS=1", flush=True)
3034                continue
3035
3036            try:
3037                tmp_args.client = rpc.client.JSONRPCClient(
3038                    tmp_args.server_addr, tmp_args.port, tmp_args.timeout,
3039                    log_level=getattr(logging, tmp_args.verbose.upper()), conn_retries=tmp_args.conn_retries)
3040                call_rpc_func(tmp_args)
3041                print("**STATUS=0", flush=True)
3042            except JSONRPCException as ex:
3043                print(ex.message)
3044                print("**STATUS=1", flush=True)
3045        exit(0)
3046    elif args.dry_run:
3047        args.client = dry_run_client()
3048        print_dict = null_print
3049        print_json = null_print
3050        print_array = null_print
3051    else:
3052        try:
3053            args.client = rpc.client.JSONRPCClient(args.server_addr, args.port, args.timeout,
3054                                                   log_level=getattr(logging, args.verbose.upper()),
3055                                                   conn_retries=args.conn_retries)
3056        except JSONRPCException as ex:
3057            print(ex.message)
3058            exit(1)
3059
3060    if hasattr(args, 'func'):
3061        try:
3062            call_rpc_func(args)
3063        except JSONRPCException as ex:
3064            print(ex.message)
3065            exit(1)
3066    else:
3067        execute_script(parser, args.client, sys.stdin)
3068