]>
git.proxmox.com Git - mirror_qemu.git/blob - scripts/qmp/qmp-shell
3 # Low-level QEMU shell on top of QMP.
5 # Copyright (C) 2009, 2010 Red Hat Inc.
8 # Luiz Capitulino <lcapitulino@redhat.com>
10 # This work is licensed under the terms of the GNU GPL, version 2. See
11 # the COPYING file in the top-level directory.
17 # # qemu [...] -qmp unix:./qmp-sock,server
21 # $ qmp-shell ./qmp-sock
23 # Commands have the following format:
25 # < command-name > [ arg-name1=arg1 ] ... [ arg-nameN=argN ]
29 # (QEMU) device_add driver=e1000 id=net1
39 class QMPCompleter(list):
40 def complete(self
, text
, state
):
42 if cmd
.startswith(text
):
48 class QMPShellError(Exception):
51 class QMPShellBadPort(QMPShellError
):
54 # TODO: QMPShell's interface is a bit ugly (eg. _fill_completion() and
55 # _execute_cmd()). Let's design a better one.
56 class QMPShell(qmp
.QEMUMonitorProtocol
):
57 def __init__(self
, address
, pp
=None):
58 qmp
.QEMUMonitorProtocol
.__init
__(self
, self
.__get
_address
(address
))
60 self
._completer
= None
63 def __get_address(self
, arg
):
65 Figure out if the argument is in the port:host form, if it's not it's
74 return ( addr
[0], port
)
78 def _fill_completion(self
):
79 for cmd
in self
.cmd('query-commands')['return']:
80 self
._completer
.append(cmd
['name'])
82 def __completer_setup(self
):
83 self
._completer
= QMPCompleter()
84 self
._fill
_completion
()
85 readline
.set_completer(self
._completer
.complete
)
86 readline
.parse_and_bind("tab: complete")
87 # XXX: default delimiters conflict with some command names (eg. query-),
88 # clearing everything as it doesn't seem to matter
89 readline
.set_completer_delims('')
91 def __build_cmd(self
, cmdline
):
93 Build a QMP input object from a user provided command-line in the
96 < command-name > [ arg-name1=arg1 ] ... [ arg-nameN=argN ]
98 cmdargs
= cmdline
.split()
99 qmpcmd
= { 'execute': cmdargs
[0], 'arguments': {} }
100 for arg
in cmdargs
[1:]:
104 opt
[1] = '='.join(opt
[1:])
109 elif opt
[1] == 'false':
111 elif opt
[1].startswith('{'):
112 value
= json
.loads(opt
[1])
115 qmpcmd
['arguments'][opt
[0]] = value
118 def _execute_cmd(self
, cmdline
):
120 qmpcmd
= self
.__build
_cmd
(cmdline
)
122 print 'command format: <command-name> ',
123 print '[arg-name1=arg1] ... [arg-nameN=argN]'
125 resp
= self
.cmd_obj(qmpcmd
)
130 if self
._pp
is not None:
131 self
._pp
.pprint(resp
)
137 self
._greeting
= qmp
.QEMUMonitorProtocol
.connect(self
)
138 self
.__completer
_setup
()
140 def show_banner(self
, msg
='Welcome to the QMP low-level shell!'):
142 version
= self
._greeting
['QMP']['version']['qemu']
143 print 'Connected to QEMU %d.%d.%d\n' % (version
['major'],version
['minor'],version
['micro'])
145 def read_exec_command(self
, prompt
):
147 Read and execute a command.
149 @return True if execution was ok, return False if disconnected.
152 cmdline
= raw_input(prompt
)
157 for ev
in self
.get_events():
162 return self
._execute
_cmd
(cmdline
)
164 class HMPShell(QMPShell
):
165 def __init__(self
, address
):
166 QMPShell
.__init
__(self
, address
)
169 def __cmd_completion(self
):
170 for cmd
in self
.__cmd
_passthrough
('help')['return'].split('\r\n'):
171 if cmd
and cmd
[0] != '[' and cmd
[0] != '\t':
172 name
= cmd
.split()[0] # drop help text
175 if name
.find('|') != -1:
176 # Command in the form 'foobar|f' or 'f|foobar', take the
178 opt
= name
.split('|')
183 self
._completer
.append(name
)
184 self
._completer
.append('help ' + name
) # help completion
186 def __info_completion(self
):
187 for cmd
in self
.__cmd
_passthrough
('info')['return'].split('\r\n'):
189 self
._completer
.append('info ' + cmd
.split()[1])
191 def __other_completion(self
):
193 self
._completer
.append('help info')
195 def _fill_completion(self
):
196 self
.__cmd
_completion
()
197 self
.__info
_completion
()
198 self
.__other
_completion
()
200 def __cmd_passthrough(self
, cmdline
, cpu_index
= 0):
201 return self
.cmd_obj({ 'execute': 'human-monitor-command', 'arguments':
202 { 'command-line': cmdline
,
203 'cpu-index': cpu_index
} })
205 def _execute_cmd(self
, cmdline
):
206 if cmdline
.split()[0] == "cpu":
207 # trap the cpu command, it requires special setting
209 idx
= int(cmdline
.split()[1])
210 if not 'return' in self
.__cmd
_passthrough
('info version', idx
):
211 print 'bad CPU index'
213 self
.__cpu
_index
= idx
215 print 'cpu command takes an integer argument'
217 resp
= self
.__cmd
_passthrough
(cmdline
, self
.__cpu
_index
)
221 assert 'return' in resp
or 'error' in resp
224 if len(resp
['return']) > 0:
225 print resp
['return'],
228 print '%s: %s' % (resp
['error']['class'], resp
['error']['desc'])
231 def show_banner(self
):
232 QMPShell
.show_banner(self
, msg
='Welcome to the HMP shell!')
235 sys
.stderr
.write('ERROR: %s\n' % msg
)
238 def fail_cmdline(option
=None):
240 sys
.stderr
.write('ERROR: bad command-line option \'%s\'\n' % option
)
241 sys
.stderr
.write('qemu-shell [ -p ] [ -H ] < UNIX socket path> | < TCP address:port >\n')
251 for arg
in sys
.argv
[1:]:
259 pp
= pprint
.PrettyPrinter(indent
=4)
266 qemu
= QMPShell(arg
, pp
)
271 except QMPShellBadPort
:
272 die('bad port number in command-line')
276 except qmp
.QMPConnectError
:
277 die('Didn\'t get QMP greeting message')
278 except qmp
.QMPCapabilitiesError
:
279 die('Could not negotiate capabilities')
281 die('Could not connect to %s' % addr
)
284 while qemu
.read_exec_command('(QEMU) '):
288 if __name__
== '__main__':