forked from rpms/libvirt
235 lines
7.0 KiB
Diff
235 lines
7.0 KiB
Diff
From a756729403915485dbd3cbdf4f252806318c227c Mon Sep 17 00:00:00 2001
|
|
Message-Id: <a756729403915485dbd3cbdf4f252806318c227c@dist-git>
|
|
From: Pavel Hrdina <phrdina@redhat.com>
|
|
Date: Mon, 1 Jul 2019 17:06:37 +0200
|
|
Subject: [PATCH] vircgroup: extract virCgroupV1AddTask
|
|
MIME-Version: 1.0
|
|
Content-Type: text/plain; charset=UTF-8
|
|
Content-Transfer-Encoding: 8bit
|
|
|
|
Reviewed-by: Fabiano Fidêncio <fidencio@redhat.com>
|
|
Reviewed-by: Ján Tomko <jtomko@redhat.com>
|
|
Signed-off-by: Pavel Hrdina <phrdina@redhat.com>
|
|
(cherry picked from commit 064024e70afd6ad08993444b3839eb3ba67426f2)
|
|
|
|
Resolves: https://bugzilla.redhat.com/show_bug.cgi?id=1689297
|
|
|
|
Signed-off-by: Pavel Hrdina <phrdina@redhat.com>
|
|
Message-Id: <558f1a90ce9b0fc5a9688c5a44bf85dd523c37e5.1561993100.git.phrdina@redhat.com>
|
|
Reviewed-by: Ján Tomko <jtomko@redhat.com>
|
|
---
|
|
src/util/vircgroup.c | 56 ++++---------------------------------
|
|
src/util/vircgroupbackend.h | 20 +++++++++++++
|
|
src/util/vircgrouppriv.h | 5 ++++
|
|
src/util/vircgroupv1.c | 31 ++++++++++++++++++++
|
|
4 files changed, 62 insertions(+), 50 deletions(-)
|
|
|
|
diff --git a/src/util/vircgroup.c b/src/util/vircgroup.c
|
|
index 33d34d6eda..a5076cb554 100644
|
|
--- a/src/util/vircgroup.c
|
|
+++ b/src/util/vircgroup.c
|
|
@@ -544,7 +544,7 @@ virCgroupSetValueU64(virCgroupPtr group,
|
|
}
|
|
|
|
|
|
-static int
|
|
+int
|
|
virCgroupSetValueI64(virCgroupPtr group,
|
|
int controller,
|
|
const char *key,
|
|
@@ -672,50 +672,6 @@ virCgroupNew(pid_t pid,
|
|
}
|
|
|
|
|
|
-typedef enum {
|
|
- /* Adds a whole process with all threads to specific cgroup except
|
|
- * to systemd named controller. */
|
|
- VIR_CGROUP_TASK_PROCESS = 1 << 0,
|
|
-
|
|
- /* Same as VIR_CGROUP_TASK_PROCESS but it also adds the task to systemd
|
|
- * named controller. */
|
|
- VIR_CGROUP_TASK_SYSTEMD = 1 << 1,
|
|
-
|
|
- /* Moves only specific thread into cgroup except to systemd
|
|
- * named controller. */
|
|
- VIR_CGROUP_TASK_THREAD = 1 << 2,
|
|
-} virCgroupTaskFlags;
|
|
-
|
|
-
|
|
-static int
|
|
-virCgroupAddTaskInternal(virCgroupPtr group,
|
|
- pid_t pid,
|
|
- unsigned int flags)
|
|
-{
|
|
- int ret = -1;
|
|
- size_t i;
|
|
-
|
|
- for (i = 0; i < VIR_CGROUP_CONTROLLER_LAST; i++) {
|
|
- /* Skip over controllers not mounted */
|
|
- if (!group->controllers[i].mountPoint)
|
|
- continue;
|
|
-
|
|
- /* We must never add tasks in systemd's hierarchy
|
|
- * unless we're intentionally trying to move a
|
|
- * task into a systemd machine scope */
|
|
- if (i == VIR_CGROUP_CONTROLLER_SYSTEMD &&
|
|
- !(flags & VIR_CGROUP_TASK_SYSTEMD))
|
|
- continue;
|
|
-
|
|
- if (virCgroupSetValueI64(group, i, "tasks", pid) < 0)
|
|
- goto cleanup;
|
|
- }
|
|
-
|
|
- ret = 0;
|
|
- cleanup:
|
|
- return ret;
|
|
-}
|
|
-
|
|
/**
|
|
* virCgroupAddProcess:
|
|
*
|
|
@@ -730,7 +686,7 @@ virCgroupAddTaskInternal(virCgroupPtr group,
|
|
int
|
|
virCgroupAddProcess(virCgroupPtr group, pid_t pid)
|
|
{
|
|
- return virCgroupAddTaskInternal(group, pid, VIR_CGROUP_TASK_PROCESS);
|
|
+ return group->backend->addTask(group, pid, VIR_CGROUP_TASK_PROCESS);
|
|
}
|
|
|
|
/**
|
|
@@ -747,9 +703,9 @@ virCgroupAddProcess(virCgroupPtr group, pid_t pid)
|
|
int
|
|
virCgroupAddMachineProcess(virCgroupPtr group, pid_t pid)
|
|
{
|
|
- return virCgroupAddTaskInternal(group, pid,
|
|
- VIR_CGROUP_TASK_PROCESS |
|
|
- VIR_CGROUP_TASK_SYSTEMD);
|
|
+ return group->backend->addTask(group, pid,
|
|
+ VIR_CGROUP_TASK_PROCESS |
|
|
+ VIR_CGROUP_TASK_SYSTEMD);
|
|
}
|
|
|
|
/**
|
|
@@ -767,7 +723,7 @@ int
|
|
virCgroupAddThread(virCgroupPtr group,
|
|
pid_t pid)
|
|
{
|
|
- return virCgroupAddTaskInternal(group, pid, VIR_CGROUP_TASK_THREAD);
|
|
+ return group->backend->addTask(group, pid, VIR_CGROUP_TASK_THREAD);
|
|
}
|
|
|
|
|
|
diff --git a/src/util/vircgroupbackend.h b/src/util/vircgroupbackend.h
|
|
index 1f28c51c49..bff1f8938e 100644
|
|
--- a/src/util/vircgroupbackend.h
|
|
+++ b/src/util/vircgroupbackend.h
|
|
@@ -35,6 +35,20 @@ typedef enum {
|
|
*/
|
|
} virCgroupBackendFlags;
|
|
|
|
+typedef enum {
|
|
+ /* Adds a whole process with all threads to specific cgroup except
|
|
+ * to systemd named controller. */
|
|
+ VIR_CGROUP_TASK_PROCESS = 1 << 0,
|
|
+
|
|
+ /* Same as VIR_CGROUP_TASK_PROCESS but it also adds the task to systemd
|
|
+ * named controller. */
|
|
+ VIR_CGROUP_TASK_SYSTEMD = 1 << 1,
|
|
+
|
|
+ /* Moves only specific thread into cgroup except to systemd
|
|
+ * named controller. */
|
|
+ VIR_CGROUP_TASK_THREAD = 1 << 2,
|
|
+} virCgroupBackendTaskFlags;
|
|
+
|
|
typedef enum {
|
|
VIR_CGROUP_BACKEND_TYPE_V1 = 0,
|
|
VIR_CGROUP_BACKEND_TYPE_LAST,
|
|
@@ -103,6 +117,11 @@ typedef int
|
|
typedef int
|
|
(*virCgroupRemoveCB)(virCgroupPtr group);
|
|
|
|
+typedef int
|
|
+(*virCgroupAddTaskCB)(virCgroupPtr group,
|
|
+ pid_t pid,
|
|
+ unsigned int flags);
|
|
+
|
|
struct _virCgroupBackend {
|
|
virCgroupBackendType type;
|
|
|
|
@@ -121,6 +140,7 @@ struct _virCgroupBackend {
|
|
virCgroupPathOfControllerCB pathOfController;
|
|
virCgroupMakeGroupCB makeGroup;
|
|
virCgroupRemoveCB remove;
|
|
+ virCgroupAddTaskCB addTask;
|
|
};
|
|
typedef struct _virCgroupBackend virCgroupBackend;
|
|
typedef virCgroupBackend *virCgroupBackendPtr;
|
|
diff --git a/src/util/vircgrouppriv.h b/src/util/vircgrouppriv.h
|
|
index a760b9bcfd..01714370be 100644
|
|
--- a/src/util/vircgrouppriv.h
|
|
+++ b/src/util/vircgrouppriv.h
|
|
@@ -73,6 +73,11 @@ int virCgroupGetValueU64(virCgroupPtr group,
|
|
const char *key,
|
|
unsigned long long int *value);
|
|
|
|
+int virCgroupSetValueI64(virCgroupPtr group,
|
|
+ int controller,
|
|
+ const char *key,
|
|
+ long long int value);
|
|
+
|
|
int virCgroupPartitionEscape(char **path);
|
|
|
|
int virCgroupNewPartition(const char *path,
|
|
diff --git a/src/util/vircgroupv1.c b/src/util/vircgroupv1.c
|
|
index 4d4b51094a..fda2c43eef 100644
|
|
--- a/src/util/vircgroupv1.c
|
|
+++ b/src/util/vircgroupv1.c
|
|
@@ -706,6 +706,36 @@ virCgroupV1Remove(virCgroupPtr group)
|
|
}
|
|
|
|
|
|
+static int
|
|
+virCgroupV1AddTask(virCgroupPtr group,
|
|
+ pid_t pid,
|
|
+ unsigned int flags)
|
|
+{
|
|
+ int ret = -1;
|
|
+ size_t i;
|
|
+
|
|
+ for (i = 0; i < VIR_CGROUP_CONTROLLER_LAST; i++) {
|
|
+ /* Skip over controllers not mounted */
|
|
+ if (!group->controllers[i].mountPoint)
|
|
+ continue;
|
|
+
|
|
+ /* We must never add tasks in systemd's hierarchy
|
|
+ * unless we're intentionally trying to move a
|
|
+ * task into a systemd machine scope */
|
|
+ if (i == VIR_CGROUP_CONTROLLER_SYSTEMD &&
|
|
+ !(flags & VIR_CGROUP_TASK_SYSTEMD))
|
|
+ continue;
|
|
+
|
|
+ if (virCgroupSetValueI64(group, i, "tasks", pid) < 0)
|
|
+ goto cleanup;
|
|
+ }
|
|
+
|
|
+ ret = 0;
|
|
+ cleanup:
|
|
+ return ret;
|
|
+}
|
|
+
|
|
+
|
|
virCgroupBackend virCgroupV1Backend = {
|
|
.type = VIR_CGROUP_BACKEND_TYPE_V1,
|
|
|
|
@@ -723,6 +753,7 @@ virCgroupBackend virCgroupV1Backend = {
|
|
.pathOfController = virCgroupV1PathOfController,
|
|
.makeGroup = virCgroupV1MakeGroup,
|
|
.remove = virCgroupV1Remove,
|
|
+ .addTask = virCgroupV1AddTask,
|
|
};
|
|
|
|
|
|
--
|
|
2.22.0
|
|
|