diff --git a/python/generator.py b/python/generator.py index 672b38f7255133ee6f09dcb4d61a9895f29298e3..97434ed37d798a701508d08f8dfe93e5e860064b 100755 --- a/python/generator.py +++ b/python/generator.py @@ -345,6 +345,7 @@ skip_impl = ( 'virDomainGetVcpus', 'virDomainPinVcpu', 'virDomainPinVcpuFlags', + 'virDomainGetVcpuPinInfo', 'virSecretGetValue', 'virSecretSetValue', 'virSecretGetUUID', diff --git a/python/libvirt-override-api.xml b/python/libvirt-override-api.xml index 4d9f54b556ee6b235af7962c531e3e1e7e4929b2..a8929b12d02cdb24c6a99ff977adb956ba8f4fdc 100644 --- a/python/libvirt-override-api.xml +++ b/python/libvirt-override-api.xml @@ -182,6 +182,12 @@ + + Query the CPU affinity setting of all virtual CPUs of domain + + + + Change the scheduler parameters diff --git a/python/libvirt-override.c b/python/libvirt-override.c index 678840e0edd812cd0f2909372bec2a52fbd96c04..9d1dac2e20c6196e743bcaf64f01996527eea62b 100644 --- a/python/libvirt-override.c +++ b/python/libvirt-override.c @@ -784,6 +784,74 @@ libvirt_virDomainPinVcpuFlags(PyObject *self ATTRIBUTE_UNUSED, return VIR_PY_INT_SUCCESS; } +static PyObject * +libvirt_virDomainGetVcpuPinInfo(PyObject *self ATTRIBUTE_UNUSED, + PyObject *args) { + virDomainPtr domain; + PyObject *pyobj_domain, *pycpumaps = NULL; + virNodeInfo nodeinfo; + virDomainInfo dominfo; + unsigned char *cpumaps; + int cpumaplen, vcpu, pcpu; + unsigned int flags; + int i_retval; + + if (!PyArg_ParseTuple(args, (char *)"Oi:virDomainGetVcpuPinInfo", + &pyobj_domain, &flags)) + return(NULL); + domain = (virDomainPtr) PyvirDomain_Get(pyobj_domain); + + LIBVIRT_BEGIN_ALLOW_THREADS; + i_retval = virNodeGetInfo(virDomainGetConnect(domain), &nodeinfo); + LIBVIRT_END_ALLOW_THREADS; + if (i_retval < 0) + return VIR_PY_NONE; + + LIBVIRT_BEGIN_ALLOW_THREADS; + i_retval = virDomainGetInfo(domain, &dominfo); + LIBVIRT_END_ALLOW_THREADS; + if (i_retval < 0) + return VIR_PY_NONE; + + cpumaplen = VIR_CPU_MAPLEN(VIR_NODEINFO_MAXCPUS(nodeinfo)); + if ((cpumaps = malloc(dominfo.nrVirtCpu * cpumaplen)) == NULL) + goto cleanup; + memset(cpumaps, 0, dominfo.nrVirtCpu * cpumaplen); + + LIBVIRT_BEGIN_ALLOW_THREADS; + i_retval = virDomainGetVcpuPinInfo(domain, dominfo.nrVirtCpu, + cpumaps, cpumaplen, flags); + LIBVIRT_END_ALLOW_THREADS; + if (i_retval < 0) + goto cleanup; + + if ((pycpumaps = PyList_New(dominfo.nrVirtCpu)) == NULL) + goto cleanup; + + for (vcpu = 0; vcpu < dominfo.nrVirtCpu; vcpu++) { + PyObject *mapinfo = PyTuple_New(VIR_NODEINFO_MAXCPUS(nodeinfo)); + if (mapinfo == NULL) + goto cleanup; + + for (pcpu = 0; pcpu < VIR_NODEINFO_MAXCPUS(nodeinfo); pcpu++) { + PyTuple_SetItem(mapinfo, pcpu, + PyBool_FromLong(VIR_CPU_USABLE(cpumaps, cpumaplen, vcpu, pcpu))); + } + PyList_SetItem(pycpumaps, vcpu, mapinfo); + } + + free(cpumaps); + + return pycpumaps; + +cleanup: + free(cpumaps); + + if (pycpumaps) { Py_DECREF(pycpumaps);} + + return VIR_PY_NONE; +} + /************************************************************************ * * * Global error handler at the Python level * @@ -4157,6 +4225,7 @@ static PyMethodDef libvirtMethods[] = { {(char *) "virDomainGetVcpus", libvirt_virDomainGetVcpus, METH_VARARGS, NULL}, {(char *) "virDomainPinVcpu", libvirt_virDomainPinVcpu, METH_VARARGS, NULL}, {(char *) "virDomainPinVcpuFlags", libvirt_virDomainPinVcpuFlags, METH_VARARGS, NULL}, + {(char *) "virDomainGetVcpuPinInfo", libvirt_virDomainGetVcpuPinInfo, METH_VARARGS, NULL}, {(char *) "virConnectListStoragePools", libvirt_virConnectListStoragePools, METH_VARARGS, NULL}, {(char *) "virConnectListDefinedStoragePools", libvirt_virConnectListDefinedStoragePools, METH_VARARGS, NULL}, {(char *) "virStoragePoolGetAutostart", libvirt_virStoragePoolGetAutostart, METH_VARARGS, NULL},