Commit 38e1dd57 authored by Sean Christopherson's avatar Sean Christopherson
Browse files

cgroup: selftests: Add API to find root of specific controller 



Add an API in the cgroups library to find the root of a specific
controller.  KVM selftests will use the API to find the memory controller.

Search for the controller on both v1 and v2 mounts, as KVM selftests'
usage will be completely oblivious of v1 versus v2.

Signed-off-by: default avatarJames Houghton <jthoughton@google.com>
Link: https://lore.kernel.org/r/20250508184649.2576210-6-jthoughton@google.com


Signed-off-by: default avatarSean Christopherson <seanjc@google.com>
parent 2c754a84

tools/testing/selftests/cgroup/lib/cgroup_util.c

+27 −7
Original line number Diff line number Diff line
@@ -217,7 +217,8 @@ int cg_write_numeric(const char *cgroup, const char *control, long value) 
	return cg_write(cgroup, control, buf);

}



int cg_find_unified_root(char *root, size_t len, bool *nsdelegate)

static int cg_find_root(char *root, size_t len, const char *controller,

			bool *nsdelegate)

{

	char buf[10 * PAGE_SIZE];

	char *fs, *mount, *type, *options;
@@ -236,18 +237,37 @@ int cg_find_unified_root(char *root, size_t len, bool *nsdelegate) 
		options = strtok(NULL, delim);

		strtok(NULL, delim);

		strtok(NULL, delim);



		if (strcmp(type, "cgroup2") == 0) {

		if (strcmp(type, "cgroup") == 0) {

			if (!controller || !strstr(options, controller))

				continue;

		} else if (strcmp(type, "cgroup2") == 0) {

			if (controller &&

					cg_read_strstr(mount, "cgroup.controllers", controller))

				continue;

		} else {

			continue;

		}

		strncpy(root, mount, len);



		if (nsdelegate)

			*nsdelegate = !!strstr(options, "nsdelegate");

		return 0;

		}



	}



	return -1;

}



int cg_find_controller_root(char *root, size_t len, const char *controller)

{

	return cg_find_root(root, len, controller, NULL);

}



int cg_find_unified_root(char *root, size_t len, bool *nsdelegate)

{

	return cg_find_root(root, len, NULL, nsdelegate);

}



int cg_create(const char *cgroup)

{

	return mkdir(cgroup, 0755);

tools/testing/selftests/cgroup/lib/include/cgroup_util.h

+1 −0
Original line number Diff line number Diff line
@@ -24,6 +24,7 @@ static inline int values_close(long a, long b, int err) 
extern ssize_t read_text(const char *path, char *buf, size_t max_len);

extern ssize_t write_text(const char *path, char *buf, ssize_t len);



extern int cg_find_controller_root(char *root, size_t len, const char *controller);

extern int cg_find_unified_root(char *root, size_t len, bool *nsdelegate);

extern char *cg_name(const char *root, const char *name);

extern char *cg_name_indexed(const char *root, const char *name, int index);