Archipelago » qemu

#!/usr/bin/python

#

# QMP command line tool

#

# Copyright IBM, Corp. 2011

#

# Authors:

#  Anthony Liguori <aliguori@us.ibm.com>

#

# This work is licensed under the terms of the GNU GPLv2 or later.

# See the COPYING file in the top-level directory.

import sys, os

from qmp import QEMUMonitorProtocol

def print_response(rsp, prefix=[]):

    if type(rsp) == list:

        i = 0

        for item in rsp:

            if prefix == []:

                prefix = ['item']

            print_response(item, prefix[:-1] + ['%s[%d]' % (prefix[-1], i)])

            i += 1

    elif type(rsp) == dict:

        for key in rsp.keys():

            print_response(rsp[key], prefix + [key])

    else:

        if len(prefix):

            print '%s: %s' % ('.'.join(prefix), rsp)

        else:

            print '%s' % (rsp)

def main(args):

    path = None

    # Use QMP_PATH if it's set

    if os.environ.has_key('QMP_PATH'):

        path = os.environ['QMP_PATH']

    while len(args):

        arg = args[0]

        if arg.startswith('--'):

            arg = arg[2:]

            if arg.find('=') == -1:

                value = True

            else:

                arg, value = arg.split('=', 1)

            if arg in ['path']:

                if type(value) == str:

                    path = value

            elif arg in ['help']:

                os.execlp('man', 'man', 'qmp')

            else:

                print 'Unknown argument "%s"' % arg

            args = args[1:]

        else:

            break

    if not path:

        print "QMP path isn't set, use --path=qmp-monitor-address or set QMP_PATH"

        return 1

    if len(args):

        command, args = args[0], args[1:]

    else:

        print 'No command found'

        print 'Usage: "qmp [--path=qmp-monitor-address] qmp-cmd arguments"'

        return 1

    if command in ['help']:

        os.execlp('man', 'man', 'qmp')

    srv = QEMUMonitorProtocol(path)

    srv.connect()

    def do_command(srv, cmd, **kwds):

        rsp = srv.cmd(cmd, kwds)

        if rsp.has_key('error'):

            raise Exception(rsp['error']['desc'])

        return rsp['return']

    commands = map(lambda x: x['name'], do_command(srv, 'query-commands'))

    srv.close()

    if command not in commands:

        fullcmd = 'qmp-%s' % command

        try:

            os.environ['QMP_PATH'] = path

            os.execvp(fullcmd, [fullcmd] + args)

        except OSError, (errno, msg):

            if errno == 2:

                print 'Command "%s" not found.' % (fullcmd)

                return 1

            raise

        return 0

    srv = QEMUMonitorProtocol(path)

    srv.connect()

    arguments = {}

    for arg in args:

        if not arg.startswith('--'):

            print 'Unknown argument "%s"' % arg

            return 1

        arg = arg[2:]

        if arg.find('=') == -1:

            value = True

        else:

            arg, value = arg.split('=', 1)

        if arg in ['help']:

            os.execlp('man', 'man', 'qmp-%s' % command)

            return 1

        arguments[arg] = value

    rsp = do_command(srv, command, **arguments)

    print_response(rsp)

if __name__ == '__main__':

    sys.exit(main(sys.argv[1:]))