[LTP] [PATCH v3 1/2] lib: Add .ulimit

Wei Gao via ltp
2024-01-09T07:00:15Z

Fixs: #530
Signed-off-by: Wei Gao <wegao@suse.com>
---
doc/C-Test-API.asciidoc | 18 ++++++++++++++++++
include/tst_test.h | 16 ++++++++++++++++
lib/tst_test.c | 32 ++++++++++++++++++++++++++++++++
3 files changed, 66 insertions(+)

diff --git a/doc/C-Test-API.asciidoc b/doc/C-Test-API.asciidoc
index db16be36e..c42a9754c 100644
--- a/doc/C-Test-API.asciidoc
+++ b/doc/C-Test-API.asciidoc
@@ -2426,6 +2426,24 @@ Test can be skipped on various conditions: on enabled SecureBoot
('.skip_in_secureboot = 1'), lockdown ('.skip_in_lockdown = 1') or in 32-bit
compat mode ('.skip_in_compat = 1').

+1.43 Set resource limits
+~~~~~~~~~~~~~~~~~~~~~~~~
+
+'.ulimit' allows to set resource limits on particular resource. NOTE: It sets 'rlim_cur'
+only if it's higher than 'rlim_cur'.
+
+[source,c]
+-------------------------------------------------------------------------------
+#include "tst_test.h"
+
+static struct tst_test test = {
+ ...
+ .ulimit = (const struct tst_ulimit_val[]) {
+ {RLIMIT_STACK, RLIM_INFINITY},
+ {}
+ },
+};
+
2. Common problems
------------------

diff --git a/include/tst_test.h b/include/tst_test.h
index 0c3171e5b..374a8615c 100644
--- a/include/tst_test.h
+++ b/include/tst_test.h
@@ -15,6 +15,7 @@
#include <limits.h>
#include <string.h>
#include <errno.h>
+#include <sys/resource.h>

#include "tst_common.h"
#include "tst_res_flags.h"
@@ -149,6 +150,11 @@ extern unsigned int tst_variant;

#define TST_UNLIMITED_RUNTIME (-1)

+struct tst_ulimit_val {
+ int resource;
+ rlim_t rlim_cur;
+};
+
struct tst_test {
/* number of tests available in test() function */
unsigned int tcnt;
@@ -307,6 +313,11 @@ struct tst_test {
*/
const struct tst_path_val *save_restore;

+ /*
+ * {} terminated array of ulimit resource type and value.
+ */
+ const struct tst_ulimit_val *ulimit;
+
/*
* NULL terminated array of kernel config options required for the
* test.
@@ -393,6 +404,11 @@ int tst_validate_children_(const char *file, const int lineno,
#define tst_validate_children(child_count) \
tst_validate_children_(__FILE__, __LINE__, (child_count))

+/*
+ * Set system resource limits
+ */
+void tst_set_ulimit_(const char *file, const int lineno, const struct tst_ulimit_val *conf);
+
#ifndef TST_NO_DEFAULT_MAIN

static struct tst_test test;
diff --git a/lib/tst_test.c b/lib/tst_test.c
index bcf2c4555..f5037330a 100644
--- a/lib/tst_test.c
+++ b/lib/tst_test.c
@@ -1147,6 +1147,29 @@ static void do_cgroup_requires(void)
tst_cg_init();
}

+#define tst_set_ulimit(conf) \
+ tst_set_ulimit_(__FILE__, __LINE__, (conf))
+
+/*
+ * Set resource limits.
+ */
+void tst_set_ulimit_(const char *file, const int lineno, const struct tst_ulimit_val *conf)
+{
+ struct rlimit rlim;
+
+ safe_getrlimit(file, lineno, conf->resource, &rlim);
+
+ rlim.rlim_cur = conf->rlim_cur;
+
+ if (conf->rlim_cur > rlim.rlim_max)
+ rlim.rlim_max = conf->rlim_cur;
+
+ tst_res_(file, lineno, TINFO, "Set ulimit resource: %d rlim_cur: %lu rlim_max: %lu",
+ conf->resource, rlim.rlim_cur, rlim.rlim_max);
+
+ safe_setrlimit(file, lineno, conf->resource, &rlim);
+}
+
static void do_setup(int argc, char *argv[])
{
char *tdebug_env = getenv("LTP_ENABLE_DEBUG");
@@ -1252,6 +1275,15 @@ static void do_setup(int argc, char *argv[])
}
}

+ if (tst_test->ulimit) {
+ const struct tst_ulimit_val *pvl = tst_test->ulimit;
+
+ while (pvl->resource) {
+ tst_set_ulimit(pvl);
+ pvl++;
+ }
+ }
+
if (tst_test->mntpoint)
SAFE_MKDIR(tst_test->mntpoint, 0777);