<h2>Edit File: rpcctl</h2><form method="post"><textarea name="content" rows="20" cols="80">#!/usr/bin/python3
import argparse
import collections
import errno
import os
import pathlib
import socket
import sys

with open(&quot;/proc/mounts&quot;, &#039;r&#039;) as f:
    mount = [ line.split()[1] for line in f if &quot;sysfs&quot; in line ]
    if len(mount) == 0:
        print(&quot;ERROR: sysfs is not mounted&quot;)
        sys.exit(1)

sunrpc = pathlib.Path(mount[0]) / &quot;kernel&quot; / &quot;sunrpc&quot;
if not sunrpc.is_dir():
    print(&quot;ERROR: sysfs does not have sunrpc directory&quot;)
    sys.exit(1)

def read_addr_file(path):
    try:
        with open(path, &#039;r&#039;) as f:
            return f.readline().strip()
    except:
        return &quot;(enoent)&quot;

def write_addr_file(path, newaddr):
    with open(path, &#039;w&#039;) as f:
        f.write(newaddr)
    return read_addr_file(path)

def read_info_file(path):
    res = collections.defaultdict(int)
    try:
        with open(path) as info:
            lines = [ l.split(&quot;=&quot;, 1) for l in info if &quot;=&quot; in l ]
            res.update({ key:int(val.strip()) for (key, val) in lines })
    finally:
        return res


class Xprt:
    def __init__(self, path):
        self.path = path
        self.name = path.stem.rsplit(&quot;-&quot;, 1)[0]
        self.type = path.stem.split(&quot;-&quot;)[2]
        self.info = read_info_file(path / &quot;xprt_info&quot;)
        self.dstaddr = read_addr_file(path / &quot;dstaddr&quot;)
        self.srcaddr = read_addr_file(path / &quot;srcaddr&quot;)
        self.read_state()

    def __lt__(self, rhs):
        return self.name &lt; rhs.name

    def _xprt(self):
        main = &quot;, main&quot; if self.info.get(&quot;main_xprt&quot;) else &quot;&quot;
        return f&quot;{self.name}: {self.type}, {self.dstaddr}, &quot; \
               f&quot;port {self.info[&#039;dst_port&#039;]}, state &lt;{self.state}&gt;{main}&quot;

    def _src_reqs(self):
        return f&quot;	Source: {self.srcaddr}, port {self.info[&#039;src_port&#039;]}, &quot; \
               f&quot;Requests: {self.info[&#039;num_reqs&#039;]}&quot;

    def _cong_slots(self):
        return f&quot;	Congestion: cur {self.info[&#039;cur_cong&#039;]}, win {self.info[&#039;cong_win&#039;]}, &quot; \
               f&quot;Slots: min {self.info[&#039;min_num_slots&#039;]}, max {self.info[&#039;max_num_slots&#039;]}&quot;

    def _queues(self):
        return f&quot;	Queues: binding {self.info[&#039;binding_q_len&#039;]}, &quot; \
               f&quot;sending {self.info[&#039;sending_q_len&#039;]}, pending {self.info[&#039;pending_q_len&#039;]}, &quot; \
               f&quot;backlog {self.info[&#039;backlog_q_len&#039;]}, tasks {self.info[&#039;tasks_queuelen&#039;]}&quot;

    def __str__(self):
        if not self.path.exists():
            return f&quot;{self.name}: has been removed&quot;
        return &quot;\n&quot;.join([self._xprt(), self._src_reqs(),
                          self._cong_slots(), self._queues() ])

    def read_state(self):
        if self.path.exists():
            with open(self.path / &quot;xprt_state&quot;) as f:
                self.state = &#039;,&#039;.join(f.readline().split()[1:])

    def small_str(self):
        main = &quot; [main]&quot; if self.info.get(&quot;main_xprt&quot;) else &quot;&quot;
        return f&quot;{self.name}: {self.type}, {self.dstaddr}{main}&quot;

    def set_dstaddr(self, newaddr):
        self.dstaddr = write_addr_file(self.path / &quot;dstaddr&quot;, newaddr)

    def set_state(self, state):
        if self.info.get(&quot;main_xprt&quot;):
            raise Exception(f&quot;Main xprts cannot be set {state}&quot;)
        with open(self.path / &quot;xprt_state&quot;, &#039;w&#039;) as f:
            f.write(state)
        self.read_state()

    def remove(self):
        if self.info.get(&quot;main_xprt&quot;):
            raise Exception(&quot;Main xprts cannot be removed&quot;)
        self.set_state(&quot;offline&quot;)
        self.set_state(&quot;remove&quot;)

    def add_command(subparser):
        parser = subparser.add_parser(&quot;xprt&quot;, help=&quot;Commands for individual xprts&quot;)
        parser.set_defaults(func=Xprt.show, xprt=None)
        subparser = parser.add_subparsers()

        remove = subparser.add_parser(&quot;remove&quot;, help=&quot;Remove an xprt&quot;)
        remove.add_argument(&quot;xprt&quot;, metavar=&quot;XPRT&quot;, nargs=1,
                            help=&quot;Name of the xprt to remove&quot;)
        remove.set_defaults(func=Xprt.set_property, property=&quot;remove&quot;)

        show = subparser.add_parser(&quot;show&quot;, help=&quot;Show xprts&quot;)
        show.add_argument(&quot;xprt&quot;, metavar=&quot;XPRT&quot;, nargs=&#039;?&#039;,
                          help=&quot;Name of a specific xprt to show&quot;)
        show.set_defaults(func=Xprt.show)

        set = subparser.add_parser(&quot;set&quot;, help=&quot;Change an xprt property&quot;)
        set.add_argument(&quot;xprt&quot;, metavar=&quot;XPRT&quot;, nargs=1,
                         help=&quot;Name of a specific xprt to modify&quot;)
        subparser = set.add_subparsers(required=True)
        online = subparser.add_parser(&quot;online&quot;, help=&quot;Set an xprt online&quot;)
        online.set_defaults(func=Xprt.set_property, property=&quot;online&quot;)
        offline = subparser.add_parser(&quot;offline&quot;, help=&quot;Set an xprt offline&quot;)
        offline.set_defaults(func=Xprt.set_property, property=&quot;offline&quot;)
        dstaddr = subparser.add_parser(&quot;dstaddr&quot;, help=&quot;Change an xprt&#039;s dstaddr&quot;)
        dstaddr.add_argument(&quot;newaddr&quot;, metavar=&quot;NEWADDR&quot;, nargs=1,
                             help=&quot;The new address for the xprt&quot;)
        dstaddr.set_defaults(func=Xprt.set_property, property=&quot;dstaddr&quot;)

    def get_by_name(name):
        glob = f&quot;**/{name}-*&quot; if name else &quot;**/xprt-*&quot;
        res = [ Xprt(x) for x in (sunrpc / &quot;xprt-switches&quot;).glob(glob) ]
        if name and len(res) == 0:
            raise FileNotFoundError(errno.ENOENT, os.strerror(errno.ENOENT),
                                    f&quot;{sunrpc / &#039;xprt-switches&#039; / glob}&quot;)
        return sorted(res)

    def show(args):
        for xprt in Xprt.get_by_name(args.xprt):
            print(xprt)

    def set_property(args):
        for xprt in Xprt.get_by_name(args.xprt[0]):
            if args.property == &quot;dstaddr&quot;:
                xprt.set_dstaddr(socket.gethostbyname(args.newaddr[0]))
            elif args.property == &quot;remove&quot;:
                xprt.remove()
            else:
                xprt.set_state(args.property)
        print(xprt)


class XprtSwitch:
    def __init__(self, path, sep=&quot;:&quot;):
        self.path = path
        self.name = path.stem
        self.info = read_info_file(path / &quot;xprt_switch_info&quot;)
        self.xprts = sorted([ Xprt(p) for p in self.path.iterdir() if p.is_dir() ])
        self.sep = sep

    def __lt__(self, rhs):
        return self.name &lt; rhs.name

    def __str__(self):
        switch =  f&quot;{self.name}{self.sep} &quot; \
                  f&quot;xprts {self.info[&#039;num_xprts&#039;]}, &quot; \
                  f&quot;active {self.info[&#039;num_active&#039;]}, &quot; \
                  f&quot;queue {self.info[&#039;queue_len&#039;]}&quot;
        xprts = [ f&quot;	{x.small_str()}&quot; for x in self.xprts ]
        return &quot;\n&quot;.join([ switch ] + xprts)

    def add_command(subparser):
        parser = subparser.add_parser(&quot;switch&quot;, help=&quot;Commands for xprt switches&quot;)
        parser.set_defaults(func=XprtSwitch.show, switch=None)
        subparser = parser.add_subparsers()

        show = subparser.add_parser(&quot;show&quot;, help=&quot;Show xprt switches&quot;)
        show.add_argument(&quot;switch&quot;, metavar=&quot;SWITCH&quot;, nargs=&#039;?&#039;,
                          help=&quot;Name of a specific switch to show&quot;)
        show.set_defaults(func=XprtSwitch.show)

        set = subparser.add_parser(&quot;set&quot;, help=&quot;Change an xprt switch property&quot;)
        set.add_argument(&quot;switch&quot;, metavar=&quot;SWITCH&quot;, nargs=1,
                         help=&quot;Name of a specific xprt switch to modify&quot;)
        subparser = set.add_subparsers(required=True)
        dstaddr = subparser.add_parser(&quot;dstaddr&quot;, help=&quot;Change an xprt switch&#039;s dstaddr&quot;)
        dstaddr.add_argument(&quot;newaddr&quot;, metavar=&quot;NEWADDR&quot;, nargs=1,
                             help=&quot;The new address for the xprt switch&quot;)
        dstaddr.set_defaults(func=XprtSwitch.set_property, property=&quot;dstaddr&quot;)

    def get_by_name(name):
        xprt_switches = sunrpc / &quot;xprt-switches&quot;
        if name:
            return [ XprtSwitch(xprt_switches / name) ]
        return [ XprtSwitch(f) for f in sorted(xprt_switches.iterdir()) ]

    def show(args):
        for switch in XprtSwitch.get_by_name(args.switch):
            print(switch)

    def set_property(args):
        for switch in XprtSwitch.get_by_name(args.switch[0]):
            resolved = socket.gethostbyname(args.newaddr[0])
            for xprt in switch.xprts:
                xprt.set_dstaddr(resolved)
        print(switch)


class RpcClient:
    def __init__(self, path):
        self.path = path
        self.name = path.stem
        self.switch = XprtSwitch(path / (path / &quot;switch&quot;).readlink(), sep=&quot;,&quot;)

    def __lt__(self, rhs):
        return self.name &lt; rhs.name

    def __str__(self):
        return f&quot;{self.name}: {self.switch}&quot;

    def add_command(subparser):
        parser = subparser.add_parser(&quot;client&quot;, help=&quot;Commands for rpc clients&quot;)
        parser.set_defaults(func=RpcClient.show, client=None)
        subparser = parser.add_subparsers()

        show = subparser.add_parser(&quot;show&quot;, help=&quot;Show rpc clients&quot;)
        show.add_argument(&quot;client&quot;, metavar=&quot;CLIENT&quot;, nargs=&#039;?&#039;,
                          help=&quot;Name of a specific rpc client to show&quot;)
        parser.set_defaults(func=RpcClient.show)

    def get_by_name(name):
        rpc_clients = sunrpc / &quot;rpc-clients&quot;
        if name:
            return [ RpcClient(rpc_clients / name) ]
        return [ RpcClient(f) for f in sorted(rpc_clients.iterdir()) ]

    def show(args):
        for client in RpcClient.get_by_name(args.client):
            print(client)


parser = argparse.ArgumentParser()

def show_small_help(args):
    parser.print_usage()
    print(&quot;sunrpc dir:&quot;, sunrpc)
parser.set_defaults(func=show_small_help)

subparser = parser.add_subparsers(title=&quot;commands&quot;)
RpcClient.add_command(subparser)
XprtSwitch.add_command(subparser)
Xprt.add_command(subparser)

args = parser.parse_args()
try:
    args.func(args)
except Exception as e:
    print(str(e))
    sys.exit(1)
</textarea><br><input type="hidden" name="edit_file" value="rpcctl"><input type="submit" value="Save"></form>