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

[Zhou Zheng Sheng]

Reviewed-by: Zhou Zheng Sheng <zhshzhou at linux.vnet.ibm.com>

on 2014/03/11 14:08, shaohef at linux.vnet.ibm.com wrote:
> 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
> 
> 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>
> ---
>  Makefile.am                |  1 +
>  src/kimchi/kvmusertests.py | 64 ++++++++++++++++++++++++++++++++++++++++++++++
>  2 files changed, 65 insertions(+)
>  create mode 100644 src/kimchi/kvmusertests.py
> 
> diff --git a/Makefile.am b/Makefile.am
> index 8442771..2fdafb1 100644
> --- a/Makefile.am
> +++ b/Makefile.am
> @@ -50,6 +50,7 @@ PEP8_WHITELIST = \
>  	src/kimchi/featuretests.py \
>  	src/kimchi/iscsi.py \
>  	src/kimchi/isoinfo.py \
> +	src/kimchi/kvmusertests.py \
>  	src/kimchi/mockmodel.py \
>  	src/kimchi/model/*.py \
>  	src/kimchi/osinfo.py \
> diff --git a/src/kimchi/kvmusertests.py b/src/kimchi/kvmusertests.py
> new file mode 100644
> index 0000000..1914a94
> --- /dev/null
> +++ b/src/kimchi/kvmusertests.py
> @@ -0,0 +1,64 @@
> +# Project Kimchi
> +#
> +# Copyright IBM, Corp. 2013
> +#
> +# 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 psutil
> +import uuid
> +
> +
> +import libvirt
> +
> +
> +from kimchi.rollbackcontext import RollbackContext
> +
> +
> +class UserTests(object):
> +    SIMPLE_VM_XML = """
> +    <domain type='kvm'>
> +      <name>%s</name>
> +      <uuid>%s</uuid>
> +      <memory unit='KiB'>10240</memory>
> +      <os>
> +        <type arch='x86_64' machine='pc'>hvm</type>
> +        <boot dev='hd'/>
> +      </os>
> +    </domain>"""
> +
> +    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 = self.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()
> 


-- 
Thanks and best regards!

Zhou Zheng Sheng / 周征晟
E-mail: zhshzhou at linux.vnet.ibm.com
Telephone: 86-10-82454397