1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
|
#!/bin/python3
# -*- python-mode -*-
'''
Copyright (C) 2020
Authors: Achilles Gaikwad <agaikwad@redhat.com>
Kenneth D'souza <kdsouza@redhat.com>
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 <https://www.gnu.org/licenses/>.
'''
import multiprocessing as mp
import os
import signal
import sys
try:
import argparse
except ImportError:
print('%s: Failed to import argparse - make sure argparse is installed!'
% sys.argv[0])
sys.exit(1)
try:
import yaml
except ImportError:
print('%s: Failed to import yaml - make sure python3-pyyaml is installed!'
% sys.argv[0])
sys.exit(1)
BBOLD = '\033[1;30;47m' #Bold black text with white background.
ENDC = '\033[m' #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 = {}
with open(path) as f:
for line in f:
try:
(key, val) = line.split(':')
client_info[key] = val
# FIXME: There has to be a better way of converting the info file to a dictionary.
except ValueError:
try:
(key, val) = line.split()
client_info[key] = val
except:
pass
return client_info
# 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('/proc/fs/nfsd/clients/')
except OSError as reason:
exit('%s' % reason)
if len(dirs) !=0:
for i in dirs:
path.append('/proc/fs/nfsd/clients/' + i + '/states')
return (path)
else:
exit('Nothing to process')
# 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]['superblock'].split(':')[-1]
# get the ip address from client_info as 'address:' note the extra
# ':' as a suffix to address. If there is a better way to convert
# the file to dictionary, please change the following value too.
client_ip = client_info['address:']
# The ip address is quoted, so we dequote it.
client_ip = client_ip[1:-1]
try:
# if the nfs-server reboots while the nfs-client holds the files open,
# the nfs-server would print the filename as '/'. For such instaces we
# print the output as disconnected dentry instead of '/'.
if(i[key]['filename']=='/'):
fname = 'disconnected dentry'
else:
fname = i[key]['filename'].split('/')[-1]
except KeyError:
# for older kernels which do not have the fname patch in kernel, they
# won't be able to see the fname field. Therefore post it as N/A.
fname = "N/A"
otype = i[key]['type']
try:
access = i[key]['access']
except:
access = ''
try:
deny = i[key]['deny']
except:
deny = ''
hostname = client_info['name'].split()[-1].split('"')[0]
hostname = hostname.split('.')[0]
otype = i[key]['type']
# if the hostname is too long, it messes up with the output being in columns,
# therefore we truncate the hostname followed by two '..' as suffix.
if len(hostname) > 20:
hostname = hostname[0:20] + '..'
clientid = client_info['clientid'].strip()
minorversion = client_info['minor version'].rstrip().rsplit()[0]
# since some fields do not have deny column, we drop those if -t is either
# layout or lock.
drop = ['layout', 'lock']
# 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 == 'all'):
print('%-13s' %inode, end='| ')
print('%-7s' %otype, end='| ')
if (argument.type not in drop):
print('%-7s' %access, end='| ')
if (argument.type not in drop and argument.type !='deleg'):
print('%-5s' %deny, end='| ')
if (argument.hostname == True):
print('%-22s' %hostname, end='| ')
else:
print('%-22s' %client_ip, end='| ')
if (argument.clientinfo == True) :
print('%-20s' %clientid, end='| ')
print('4.%-3s' %minorversion, end='| ')
print(fname)
def opener(path):
try:
with open(path, 'r') as nfsdata:
data = yaml.load(nfsdata, Loader = yaml.BaseLoader)
if data is not None:
clientinfo = path.rsplit('/', 1)[0] + '/info'
data.append(clientinfo)
return data
except OSError as reason:
print('%s' % reason)
def print_cols(argument):
title_inode = 'Inode number'
title_otype = 'Type'
title_access = 'Access'
title_deny = 'Deny'
title_fname = 'Filename'
title_clientID = 'Client ID'
title_hostname = 'Hostname'
title_ip = 'ip address'
title_nfsvers = 'vers'
drop = ['lock', 'layout']
print(BBOLD, end='')
print('%-13s' %title_inode, end='| ')
print('%-7s' %title_otype, end='| ')
if (argument.type not in drop):
print('%-7s' %title_access, end='| ')
if (argument.type not in drop and argument.type !='deleg'):
print('%-5s' %title_deny, end='| ')
if (argument.hostname == True):
print('%-22s' %title_hostname, end='| ')
else:
print('%-22s' %title_ip, end='| ')
if (argument.clientinfo == True):
print('%-20s' %title_clientID, end='| ')
print('%-5s' %title_nfsvers, end='| ')
print(title_fname, end='')
print(ENDC)
def nfsd4_show():
parser = argparse.ArgumentParser(description = 'Parse the nfsd states and clientinfo files.')
parser.add_argument('-t', '--type', metavar = 'type', type = str, choices = ['open',
'deleg', 'lock', 'layout', 'all'],
default = 'all',
help = 'Input the type that you want to be printed: open, lock, deleg, layout, all')
parser.add_argument('--clientinfo', action = 'store_true',
help = 'output clients information, --hostname is implied.')
parser.add_argument('--hostname', action = 'store_true',
help = 'print hostname of client instead of its ip address. Longer hostnames are truncated.')
parser.add_argument('-q', '--quiet', action = 'store_true',
help = 'don\'t print the header information')
args = parser.parse_args()
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('Caught KeyboardInterrupt, terminating workers')
p.terminate()
p.join()
if __name__ == "__main__":
nfsd4_show()
|
