From mboxrd@z Thu Jan 1 00:00:00 1970 Received: from eggs.gnu.org ([2001:4830:134:3::10]:41867) by lists.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1WBiuK-0004dP-JG for qemu-devel@nongnu.org; Fri, 07 Feb 2014 05:43:45 -0500 Received: from Debian-exim by eggs.gnu.org with spam-scanned (Exim 4.71) (envelope-from ) id 1WBiuE-00067o-TY for qemu-devel@nongnu.org; Fri, 07 Feb 2014 05:43:40 -0500 Received: from mx1.redhat.com ([209.132.183.28]:41969) by eggs.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1WBiuE-00067k-MC for qemu-devel@nongnu.org; Fri, 07 Feb 2014 05:43:34 -0500 From: Fam Zheng Date: Fri, 7 Feb 2014 18:43:30 +0800 Message-Id: <1391769813-15437-3-git-send-email-famz@redhat.com> In-Reply-To: <1391769813-15437-1-git-send-email-famz@redhat.com> References: <1391769813-15437-1-git-send-email-famz@redhat.com> Subject: [Qemu-devel] [PATCH v2 2/5] qtest: Add scripts/qtest/qtest.py List-Id: List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , To: qemu-devel@nongnu.org Cc: Kevin Wolf , =?UTF-8?q?Beno=C3=AEt=20Canet?= , Stefan Hajnoczi , pbonzini@redhat.com This removes the dummy scripts/qtest and adds scripts/qtest/qtest.py as a python library for qtest protocol. This is a skeleton with a basic "cmd" method to execute a command, reading and parsing of qtest output will be added later on demand. Signed-off-by: Fam Zheng --- scripts/qtest | 5 ---- scripts/qtest/qtest.py | 78 ++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 78 insertions(+), 5 deletions(-) delete mode 100755 scripts/qtest create mode 100644 scripts/qtest/qtest.py diff --git a/scripts/qtest b/scripts/qtest deleted file mode 100755 index 4ef6c1c..0000000 --- a/scripts/qtest +++ /dev/null @@ -1,5 +0,0 @@ -#!/bin/sh - -export QTEST_QEMU_BINARY=$1 -shift -"$@" diff --git a/scripts/qtest/qtest.py b/scripts/qtest/qtest.py new file mode 100644 index 0000000..a5daf50 --- /dev/null +++ b/scripts/qtest/qtest.py @@ -0,0 +1,78 @@ +# QEMU qtest library +# +# Copyright (C) 2014 Red Hat Inc. +# +# Authors: +# Fam Zheng +# +# This work is licensed under the terms of the GNU GPL, version 2. See +# the COPYING file in the top-level directory. +# +# Based on qmp.py. +# + +import errno +import socket + +class QEMUQtestProtocol: + def __init__(self, address, server=False): + """ + Create a QEMUQtestProtocol object. + + @param address: QEMU address, can be either a unix socket path (string) + or a tuple in the form ( address, port ) for a TCP + connection + @param server: server mode listens on the socket (bool) + @raise socket.error on socket connection errors + @note No connection is established, this is done by the connect() or + accept() methods + """ + self.__address = address + self.__sock = self.__get_sock() + if server: + self.__sock.bind(self.__address) + self.__sock.listen(1) + + def __get_sock(self): + if isinstance(self.__address, tuple): + family = socket.AF_INET + else: + family = socket.AF_UNIX + return socket.socket(family, socket.SOCK_STREAM) + + error = socket.error + + def connect(self): + """ + Connect to the qtest socket. + + @raise socket.error on socket connection errors + """ + self.__sock.connect(self.__address) + self.__sockfile = self.__sock.makefile() + + def accept(self): + """ + Await connection from QEMU. + + @raise socket.error on socket connection errors + """ + self.__sock, _ = self.__sock.accept() + self.__sockfile = self.__sock.makefile() + + def cmd(self, qtest_cmd): + """ + Send a qtest command on the wire. + + @param qtest_cmd: qtest command to be sent as a Python dict + """ + self.__sock.sendall(qtest_cmd + "\n") + + def close(self): + self.__sock.close() + self.__sockfile.close() + + timeout = socket.timeout + + def settimeout(self, timeout): + self.__sock.settimeout(timeout) -- 1.8.5.4