684750ab4f
Use a shorter name. We are going to move in iotests from qmp() to
command() where possible. But command() is longer than qmp() and don't
look better. Let's rename.
You can simply grep for '\.command(' and for 'def command(' to check
that everything is updated (command() in tests/docker/docker.py is
unrelated).
Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@yandex-team.ru>
Reviewed-by: Daniel P. Berrangé <berrange@redhat.com>
Reviewed-by: Eric Blake <eblake@redhat.com>
Reviewed-by: Cédric Le Goater <clg@kaod.org>
Reviewed-by: Juan Quintela <quintela@redhat.com>
Message-id: 20231006154125.1068348-6-vsementsov@yandex-team.ru
[vsementsov: also update three occurrences in
tests/avocado/machine_aspeed.py and keep r-b]
Signed-off-by: John Snow <jsnow@redhat.com>
123 lines
3.7 KiB
Python
Executable File
123 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# 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.qmp import QMPError
|
|
from qemu.qmp.legacy import QEMUMonitorProtocol
|
|
|
|
|
|
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 += '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.cmd('query-named-block-nodes')
|
|
bds_nodes = {n['node-name']: n for n in bds_nodes}
|
|
|
|
job_nodes = qmp.cmd('query-block-jobs')
|
|
job_nodes = {n['device']: n for n in job_nodes}
|
|
|
|
block_graph = qmp.cmd('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 cmd(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 QMPError(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)
|