[Kimchi-devel] [PATCH V4] qemu user tests: probe the username of qemu process started by libvirt

[shaohef at linux.vnet.ibm.com]

From: ShaoHe Feng <shaohef at linux.vnet.ibm.com>

We want to fix the searchable permission for qemu user.
But we should find the username of qemu process firstly.

searchable permission is a known problem.
We have discussed several times on IRC.

Royce reports the qemu username is different on different distros

Adam, Aline, Royce and I think we can probe qemu username with the
similar method of qemu iso stream support.

Zhou Zheng Sheng gives a better way to find the qemu process ID

Signed-off-by: ShaoHe Feng <shaohef at linux.vnet.ibm.com>
Signed-off-by: Zhou Zheng Sheng <zhshzhou at linux.vnet.ibm.com>
Signed-off-by: Royce Lv <lvroyce at linux.vnet.ibm.com>
Signed-off-by: Aline Manera <alinefm at linux.vnet.ibm.com>
---
 Makefile.am                |  3 ++-
 src/kimchi/kvmusertests.py | 66 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 68 insertions(+), 1 deletion(-)
 create mode 100644 src/kimchi/kvmusertests.py

diff --git a/Makefile.am b/Makefile.am
index 1fb3502..bbb1273 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -45,12 +45,13 @@ PEP8_WHITELIST = \
 	src/kimchi/auth.py \
 	src/kimchi/cachebust.py \
 	src/kimchi/config.py.in \
+	src/kimchi/control/*.py \
 	src/kimchi/disks.py \
 	src/kimchi/featuretests.py \
+	src/kimchi/kvmusertests.py \
 	src/kimchi/rollbackcontext.py \
 	src/kimchi/root.py \
 	src/kimchi/server.py \
-	src/kimchi/control/*.py \
 	tests/test_mockmodel.py \
 	tests/test_model.py \
 	tests/test_plugin.py \
diff --git a/src/kimchi/kvmusertests.py b/src/kimchi/kvmusertests.py
new file mode 100644
index 0000000..1e55390
--- /dev/null
+++ b/src/kimchi/kvmusertests.py
@@ -0,0 +1,66 @@
+#
+# Project Kimchi
+#
+# Copyright IBM, Corp. 2013
+#
+# Authors:
+#  ShaoHe Feng <shaohef at linux.vnet.ibm.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library 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
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+
+import libvirt
+import uuid
+import psutil
+
+from kimchi.rollbackcontext import RollbackContext
+
+
+SIMPLE_VM_XML = """
+<domain type='kvm'>
+  <name>%s</name>
+  <uuid>%s</uuid>
+  <memory unit='KiB'>32768</memory>
+  <os>
+    <type arch='x86_64' machine='pc'>hvm</type>
+    <boot dev='hd'/>
+  </os>
+</domain>"""
+
+
+class UserTests(object):
+    def __init__(self):
+        self.vm_uuid = uuid.uuid3(uuid.NAMESPACE_DNS, 'vm-test.kimchi.org')
+        self.vm_name = "kimchi_test_%s" % self.vm_uuid
+
+    def probe_user(self):
+        xml = SIMPLE_VM_XML % (self.vm_name, self.vm_uuid)
+        user = None
+        with RollbackContext() as rollback:
+            conn = libvirt.open('qemu:///system')
+            rollback.prependDefer(conn.close)
+            dom = conn.defineXML(xml)
+            rollback.prependDefer(dom.undefine)
+            dom.create()
+            rollback.prependDefer(dom.destroy)
+            with open('/var/run/libvirt/qemu/%s.pid' % self.vm_name) as f:
+                pidStr = f.read()
+            p = psutil.Process(int(pidStr))
+            user = p.username
+        return user
+
+
+if __name__ == '__main__':
+    ut = UserTests()
+    print ut.probe_user()
-- 
1.8.4.2