<h2>Edit File: nfsdclnts</h2><form method="post"><textarea name="content" rows="20" cols="80">#!/usr/bin/python3
# -*- python-mode -*-
&#039;&#039;&#039;
    Copyright (C) 2020
    Authors:    Achilles Gaikwad &lt;agaikwad@redhat.com&gt;
                Kenneth  D&#039;souza &lt;kdsouza@redhat.com&gt;

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see &lt;https://www.gnu.org/licenses/&gt;.
&#039;&#039;&#039;

import multiprocessing as mp
import os
import signal
import sys

try:
    import argparse
except ImportError:
    print(&#039;%s:  Failed to import argparse - make sure argparse is installed!&#039;
        % sys.argv[0])
    sys.exit(1)
try:
    import yaml
except ImportError:
    print(&#039;%s:  Failed to import yaml - make sure python3-pyyaml is installed!&#039;
        % sys.argv[0])
    sys.exit(1)

BBOLD = &#039;\033[1;30;47m&#039; #Bold black text with white background.
ENDC = &#039;\033[m&#039; #Rest to defaults

def init_worker():
    signal.signal(signal.SIGINT, signal.SIG_IGN)

# this function converts the info file to a dictionary format, sorta. 
def file_to_dict(path):
    client_info = {}
    try:
        with open(path) as f:
            for line in f:
                try:
                    (key, val) = line.split(&#039;:&#039;, 1)
                    client_info[key] = val.strip()
                    # FIXME: There has to be a better way of converting the info file to a dictionary.
                except ValueError as reason:
                    if verbose:
                        print(&#039;Exception occured, %s&#039; % reason)

        if len(client_info) == 0 and verbose:
            print(&quot;Provided %s file is not valid&quot; %path)
        return client_info

    except OSError as reason:
        if verbose:
            print(&#039;%s&#039; % reason)

# this function gets the paths from /proc/fs/nfsd/clients/
# returns a list of paths for each client which has nfs-share mounted.
def getpaths():
    path = []
    try:
        dirs = os.listdir(&#039;/proc/fs/nfsd/clients/&#039;)
    except OSError as reason:
        exit(&#039;%s&#039; % reason)
    if len(dirs) !=0:
	    for i in dirs:
                path.append(&#039;/proc/fs/nfsd/clients/&#039; + i + &#039;/states&#039;)
	    return (path)
    else:
        exit(&#039;Nothing to process&#039;)

# A single function to rule them all, in this function we gather all the data
# from already populated data_list and client_info.
def printer(data_list, argument):
    client_info_path = data_list.pop()
    client_info = file_to_dict(client_info_path)
    for i in data_list:
        for key in i:
            inode = i[key][&#039;superblock&#039;].split(&#039;:&#039;)[-1]
            # The ip address is quoted, so we dequote it.
            try:
                client_ip = client_info[&#039;address&#039;][1:-1]
            except:
                client_ip = &quot;N/A&quot;
            try:
                # if the nfs-server reboots while the nfs-client holds the files open,
                # the nfs-server would print the filename as &#039;/&#039;. For such instaces we
                # print the output as disconnected dentry instead of &#039;/&#039;.
                if(i[key][&#039;filename&#039;]==&#039;/&#039;):
                    fname = &#039;disconnected dentry&#039;
                else:
                    fname = i[key][&#039;filename&#039;].split(&#039;/&#039;)[-1]
            except KeyError:
                # for older kernels which do not have the fname patch in kernel, they
                # won&#039;t be able to see the fname field. Therefore post it as N/A.
                fname = &quot;N/A&quot;
            otype = i[key][&#039;type&#039;]
            try:
                access = i[key][&#039;access&#039;]
            except:
                access = &#039;&#039;
            try:
                deny = i[key][&#039;deny&#039;]
            except:
                deny = &#039;&#039;
            try:
                hostname = client_info[&#039;name&#039;].split()[-1].split(&#039;&quot;&#039;)[0]
                hostname =  hostname.split(&#039;.&#039;)[0]
                # if the hostname is too long, it messes up with the output being in columns,
                # therefore we truncate the hostname followed by two &#039;..&#039; as suffix.
                if len(hostname) &gt; 20:
                    hostname = hostname[0:20] + &#039;..&#039;
            except:
                hostname = &quot;N/A&quot;
            try:
                clientid = client_info[&#039;clientid&#039;]
            except:
                clientid = &quot;N/A&quot;
            try:
                minorversion = &quot;4.&quot; + client_info[&#039;minor version&#039;]
            except:
                minorversion = &quot;N/A&quot;

            otype = i[key][&#039;type&#039;]
            # since some fields do not have deny column, we drop those if -t is either
            # layout or lock.
            drop = [&#039;layout&#039;, &#039;lock&#039;]

            # Printing the output this way instead of a single string which is concatenated
            # this makes it better to quickly add more columns in future.
            if(otype == argument.type or  argument.type == &#039;all&#039;):
                print(&#039;%-13s&#039; %inode, end=&#039;| &#039;)
                print(&#039;%-7s&#039; %otype, end=&#039;| &#039;)
                if (argument.type not in drop):
                    print(&#039;%-7s&#039; %access, end=&#039;| &#039;)
                if (argument.type not in drop and argument.type !=&#039;deleg&#039;):
                    print(&#039;%-5s&#039; %deny, end=&#039;| &#039;)
                if (argument.hostname == True):
                    print(&#039;%-22s&#039; %hostname, end=&#039;| &#039;)
                else:
                   print(&#039;%-22s&#039; %client_ip, end=&#039;| &#039;)
                if (argument.clientinfo == True) :
                    print(&#039;%-20s&#039; %clientid, end=&#039;| &#039;)
                    print(&#039;%-5s&#039; %minorversion, end=&#039;| &#039;)
                print(fname)

def opener(path):
    try:
        with open(path, &#039;r&#039;) as nfsdata:
            try:
                data = yaml.load(nfsdata, Loader = yaml.BaseLoader)
                if data is not None:
                    clientinfo = path.rsplit(&#039;/&#039;, 1)[0] + &#039;/info&#039;
                    data.append(clientinfo)
                return data
            except:
                if verbose:
                    print(&quot;Exception occurred, Please make sure %s is a YAML file&quot; %path)

    except OSError as reason:
        if verbose:
            print(&#039;%s&#039; % reason)

def print_cols(argument):
    title_inode = &#039;Inode number&#039;
    title_otype = &#039;Type&#039;
    title_access = &#039;Access&#039;
    title_deny = &#039;Deny&#039;
    title_fname = &#039;Filename&#039;
    title_clientID = &#039;Client ID&#039;
    title_hostname = &#039;Hostname&#039;
    title_ip = &#039;ip address&#039;
    title_nfsvers = &#039;vers&#039;

    drop = [&#039;lock&#039;, &#039;layout&#039;]
    print(BBOLD, end=&#039;&#039;)
    print(&#039;%-13s&#039; %title_inode, end=&#039;| &#039;)
    print(&#039;%-7s&#039; %title_otype, end=&#039;| &#039;)
    if (argument.type not in drop):
        print(&#039;%-7s&#039; %title_access, end=&#039;| &#039;)
    if (argument.type not in drop and argument.type !=&#039;deleg&#039;):
        print(&#039;%-5s&#039; %title_deny, end=&#039;| &#039;)
    if (argument.hostname == True):
        print(&#039;%-22s&#039; %title_hostname, end=&#039;| &#039;)
    else:
        print(&#039;%-22s&#039; %title_ip, end=&#039;| &#039;)
    if (argument.clientinfo == True):
        print(&#039;%-20s&#039; %title_clientID, end=&#039;| &#039;)
        print(&#039;%-5s&#039; %title_nfsvers, end=&#039;| &#039;)
    print(title_fname, end=&#039;&#039;)
    print(ENDC)

def nfsd4_show():

    parser = argparse.ArgumentParser(description = &#039;Parse the nfsd states and clientinfo files.&#039;)
    parser.add_argument(&#039;-t&#039;, &#039;--type&#039;, metavar = &#039;type&#039;, type = str, choices = [&#039;open&#039;,
        &#039;deleg&#039;, &#039;lock&#039;, &#039;layout&#039;, &#039;all&#039;],
        default = &#039;all&#039;,
        help = &#039;Input the type that you want to be printed: open, lock, deleg, layout, all&#039;)
    parser.add_argument(&#039;--clientinfo&#039;, action = &#039;store_true&#039;,
        help = &#039;output clients information, --hostname is implied.&#039;)
    parser.add_argument(&#039;--hostname&#039;, action = &#039;store_true&#039;,
        help = &#039;print hostname of client instead of its ip address. Longer hostnames are truncated.&#039;)
    parser.add_argument(&#039;-v&#039;, &#039;--verbose&#039;, action = &#039;store_true&#039;,
        help = &#039;Verbose operation, show debug messages.&#039;)
    parser.add_argument(&#039;-f&#039;, &#039;--file&#039;, nargs=&#039;+&#039;, type = str, metavar=&#039;&#039;,
        help = &#039;pass client states file, provided that info file resides in the same directory.&#039;)
    parser.add_argument(&#039;-q&#039;, &#039;--quiet&#039;, action = &#039;store_true&#039;,
        help = &#039;don\&#039;t print the header information&#039;)

    args = parser.parse_args()

    global verbose
    verbose = False
    signal.signal(signal.SIGPIPE, signal.SIG_DFL)
    if args.verbose:
        verbose = True

    if args.file:
        paths = args.file
    else:
        paths = getpaths()

    p = mp.Pool(mp.cpu_count(), init_worker)
    try:
        result = p.map(opener, paths)
        ### Drop None entries from list
        final_result = list(filter(None, result))
        p.close()
        p.join()

        if len(final_result) !=0 and not args.quiet:
            print_cols(args)

        for item in final_result:
            printer(item, args)

    except KeyboardInterrupt:
        print(&#039;Caught KeyboardInterrupt, terminating workers&#039;)
        p.terminate()
        p.join()

if __name__ == &quot;__main__&quot;:
    nfsd4_show()
</textarea><br><input type="hidden" name="edit_file" value="nfsdclnts"><input type="submit" value="Save"></form>