It's not obvious that something named __init__.py actually houses important code that isn't relevant to python packaging glue. Move the QEMUMachine and related error classes out into their own module. Adjust users to the new import location. Signed-off-by: John Snow <jsnow@redhat.com> Message-Id: <20190627212816.27298-2-jsnow@redhat.com> Signed-off-by: Eduardo Habkost <ehabkost@redhat.com>
123 lines
3.7 KiB
Python
Executable file
123 lines
3.7 KiB
Python
Executable file
#!/usr/bin/env python
|
|
#
|
|
# Render Qemu Block Graph
|
|
#
|
|
# Copyright (c) 2018 Virtuozzo International GmbH. All rights reserved.
|
|
#
|
|
# 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 2 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 <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import json
|
|
from graphviz import Digraph
|
|
|
|
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'python'))
|
|
from qemu.machine import MonitorResponseError
|
|
|
|
|
|
def perm(arr):
|
|
s = 'w' if 'write' in arr else '_'
|
|
s += 'r' if 'consistent-read' in arr else '_'
|
|
s += 'u' if 'write-unchanged' in arr else '_'
|
|
s += 'g' if 'graph-mod' in arr else '_'
|
|
s += 's' if 'resize' in arr else '_'
|
|
return s
|
|
|
|
|
|
def render_block_graph(qmp, filename, format='png'):
|
|
'''
|
|
Render graph in text (dot) representation into "@filename" and
|
|
representation in @format into "@filename.@format"
|
|
'''
|
|
|
|
bds_nodes = qmp.command('query-named-block-nodes')
|
|
bds_nodes = {n['node-name']: n for n in bds_nodes}
|
|
|
|
job_nodes = qmp.command('query-block-jobs')
|
|
job_nodes = {n['device']: n for n in job_nodes}
|
|
|
|
block_graph = qmp.command('x-debug-query-block-graph')
|
|
|
|
graph = Digraph(comment='Block Nodes Graph')
|
|
graph.format = format
|
|
graph.node('permission symbols:\l'
|
|
' w - Write\l'
|
|
' r - consistent-Read\l'
|
|
' u - write - Unchanged\l'
|
|
' g - Graph-mod\l'
|
|
' s - reSize\l'
|
|
'edge label scheme:\l'
|
|
' <child type>\l'
|
|
' <perm>\l'
|
|
' <shared_perm>\l', shape='none')
|
|
|
|
for n in block_graph['nodes']:
|
|
if n['type'] == 'block-driver':
|
|
info = bds_nodes[n['name']]
|
|
label = n['name'] + ' [' + info['drv'] + ']'
|
|
if info['drv'] == 'file':
|
|
label += '\n' + os.path.basename(info['file'])
|
|
shape = 'ellipse'
|
|
elif n['type'] == 'block-job':
|
|
info = job_nodes[n['name']]
|
|
label = info['type'] + ' job (' + n['name'] + ')'
|
|
shape = 'box'
|
|
else:
|
|
assert n['type'] == 'block-backend'
|
|
label = n['name'] if n['name'] else 'unnamed blk'
|
|
shape = 'box'
|
|
|
|
graph.node(str(n['id']), label, shape=shape)
|
|
|
|
for e in block_graph['edges']:
|
|
label = '%s\l%s\l%s\l' % (e['name'], perm(e['perm']),
|
|
perm(e['shared-perm']))
|
|
graph.edge(str(e['parent']), str(e['child']), label=label)
|
|
|
|
graph.render(filename)
|
|
|
|
|
|
class LibvirtGuest():
|
|
def __init__(self, name):
|
|
self.name = name
|
|
|
|
def command(self, cmd):
|
|
# only supports qmp commands without parameters
|
|
m = {'execute': cmd}
|
|
ar = ['virsh', 'qemu-monitor-command', self.name, json.dumps(m)]
|
|
|
|
reply = json.loads(subprocess.check_output(ar))
|
|
|
|
if 'error' in reply:
|
|
raise MonitorResponseError(reply)
|
|
|
|
return reply['return']
|
|
|
|
|
|
if __name__ == '__main__':
|
|
obj = sys.argv[1]
|
|
out = sys.argv[2]
|
|
|
|
if os.path.exists(obj):
|
|
# assume unix socket
|
|
qmp = QEMUMonitorProtocol(obj)
|
|
qmp.connect()
|
|
else:
|
|
# assume libvirt guest name
|
|
qmp = LibvirtGuest(obj)
|
|
|
|
render_block_graph(qmp, out)
|