[libvirt] [PATCH 02/12] Introduce the function virCgroupMoveTask
Hu Tao
hutao at cn.fujitsu.com
Thu Jul 26 10:08:04 UTC 2012
On Wed, Jul 25, 2012 at 01:21:44PM +0800, tangchen wrote:
> From: Wen Congyang <wency at cn.fujitsu.com>
>
> Introduce a new API to move tasks of one controller from a cgroup to another cgroup
>
> Signed-off-by: Wen Congyang <wency at cn.fujitsu.com>
> Signed-off-by: Tang Chen <tangchen at cn.fujitsu.com>
> Signed-off-by: Hu Tao <hutao at cn.fujitsu.com>
> ---
> src/libvirt_private.syms | 2 +
> src/util/cgroup.c | 111 ++++++++++++++++++++++++++++++++++++++++++++++
> src/util/cgroup.h | 8 ++++
> 3 files changed, 121 insertions(+)
>
> diff --git a/src/libvirt_private.syms b/src/libvirt_private.syms
> index d2c2844..738c70f 100644
> --- a/src/libvirt_private.syms
> +++ b/src/libvirt_private.syms
> @@ -60,6 +60,7 @@ virCapabilitiesSetMacPrefix;
>
> # cgroup.h
> virCgroupAddTask;
> +virCgroupAddTaskController;
> virCgroupAllowDevice;
> virCgroupAllowDeviceMajor;
> virCgroupAllowDevicePath;
> @@ -91,6 +92,7 @@ virCgroupKill;
> virCgroupKillPainfully;
> virCgroupKillRecursive;
> virCgroupMounted;
> +virCgroupMoveTask;
> virCgroupPathOfController;
> virCgroupRemove;
> virCgroupSetBlkioDeviceWeight;
> diff --git a/src/util/cgroup.c b/src/util/cgroup.c
> index 1ac8278..d94f07b 100644
> --- a/src/util/cgroup.c
> +++ b/src/util/cgroup.c
> @@ -791,6 +791,117 @@ int virCgroupAddTask(virCgroupPtr group, pid_t pid)
> return rc;
> }
>
> +/**
> + * virCgroupAddTaskController:
> + *
> + * @group: The cgroup to add a task to
> + * @pid: The pid of the task to add
> + * @controller: The cgroup controller to be operated on
> + *
> + * Returns: 0 on success or -errno on failure
> + */
> +int virCgroupAddTaskController(virCgroupPtr group, pid_t pid, int controller)
> +{
> + if (controller < VIR_CGROUP_CONTROLLER_CPU ||
> + controller > VIR_CGROUP_CONTROLLER_BLKIO)
> + return -EINVAL;
> +
> + if (!group->controllers[controller].mountPoint)
> + return -EINVAL;
> +
> + return virCgroupSetValueU64(group, controller, "tasks",
> + (unsigned long long)pid);
> +}
> +
> +
> +static int virCgroupAddTaskStrController(virCgroupPtr group,
> + const char *pidstr,
> + int controller)
> +{
> + char *str = NULL, *cur = NULL, *next = NULL;
> + unsigned long long pid = 0;
> + int rc = 0;
> +
> + if (virAsprintf(&str, "%s", pidstr) < 0)
> + return -1;
> +
> + cur = str;
> + while ((next = strchr(cur, '\n')) != NULL) {
> + *next = '\0';
> + rc = virStrToLong_ull(cur, NULL, 10, &pid);
> + if (rc != 0)
At least gives a warning message here because some pids may have added
successfully, some failed.
> + goto cleanup;
> + cur = next + 1;
> +
> + rc = virCgroupAddTaskController(group, pid, controller);
> + if (rc != 0)
Some as above.
> + goto cleanup;
> + }
> + if (cur != '\0') {
> + rc = virStrToLong_ull(cur, NULL, 10, &pid);
> + if (rc != 0)
Some as above.
> + goto cleanup;
> + rc = virCgroupAddTaskController(group, pid, controller);
> + if (rc != 0)
Some as above.
> + goto cleanup;
> + }
> +
> +cleanup:
> + VIR_FREE(str);
> + return rc;
> +}
> +
> +/**
> + * virCgroupMoveTask:
> + *
> + * @src_group: The source cgroup where all tasks are removed from
> + * @dest_group: The destination where all tasks are added to
> + * @controller: The cgroup controller to be operated on
> + *
> + * Returns: 0 on success or -errno on failure
> + */
> +int virCgroupMoveTask(virCgroupPtr src_group, virCgroupPtr dest_group,
> + int controller)
> +{
> + int rc = 0, err = 0;
> + char *content = NULL;
> +
> + if (controller < VIR_CGROUP_CONTROLLER_CPU ||
> + controller > VIR_CGROUP_CONTROLLER_BLKIO)
> + return -EINVAL;
> +
> + if (!src_group->controllers[controller].mountPoint ||
> + !dest_group->controllers[controller].mountPoint) {
> + VIR_WARN("no vm cgroup in controller %d", controller);
> + return 0;
> + }
> +
> + rc = virCgroupGetValueStr(src_group, controller, "tasks", &content);
> + if (rc != 0)
> + return rc;
> +
> + rc = virCgroupAddTaskStrController(dest_group, content, controller);
> + if (rc != 0)
> + goto cleanup;
> +
> + VIR_FREE(content);
> +
> + return 0;
> +
> +cleanup:
> + /*
> + * We don't need to recover dest_cgroup because cgroup will make sure
> + * that one task only resides in one cgroup of the same controller.
> + */
> + err = virCgroupAddTaskStrController(src_group, content, controller);
> + if (err != 0)
> + VIR_ERROR(_("Cannot recover cgroup %s from %s"),
> + src_group->controllers[controller].mountPoint,
> + dest_group->controllers[controller].mountPoint);
> + VIR_FREE(content);
> +
> + return rc;
> +}
>
> /**
> * virCgroupForDriver:
> diff --git a/src/util/cgroup.h b/src/util/cgroup.h
> index 315ebd6..f14c167 100644
> --- a/src/util/cgroup.h
> +++ b/src/util/cgroup.h
> @@ -58,6 +58,14 @@ int virCgroupPathOfController(virCgroupPtr group,
>
> int virCgroupAddTask(virCgroupPtr group, pid_t pid);
>
> +int virCgroupAddTaskController(virCgroupPtr group,
> + pid_t pid,
> + int controller);
> +
> +int virCgroupMoveTask(virCgroupPtr src_group,
> + virCgroupPtr dest_group,
> + int controller);
> +
> int virCgroupSetBlkioWeight(virCgroupPtr group, unsigned int weight);
> int virCgroupGetBlkioWeight(virCgroupPtr group, unsigned int *weight);
>
> --
> 1.7.10.2
>
> --
> libvir-list mailing list
> libvir-list at redhat.com
> https://www.redhat.com/mailman/listinfo/libvir-list
--
Thanks,
Hu Tao
More information about the libvir-list
mailing list