lib: extract igt_drop_root

Useful in other tests.

Signed-off-by: Daniel Vetter <daniel.vetter@ffwll.ch>

diff --git a/lib/drmtest.c b/lib/drmtest.c
index 7275b7f..246f2f7 100644 (file)
--- a/lib/drmtest.c
+++ b/lib/drmtest.c
@@ -1659,3 +1659,14 @@ void igt_system_suspend_autoresume(void)
        ret = system("rtcwake -s 30 -m mem");
        igt_assert(ret == 0);
 }
+
+void igt_drop_root(void)
+{
+       igt_assert(getuid() == 0);
+
+       igt_assert(setgid(2) == 0);
+       igt_assert(setuid(2) == 0);
+
+       igt_assert(getgid() == 2);
+       igt_assert(getuid() == 2);
+}

diff --git a/lib/drmtest.h b/lib/drmtest.h
index a8de2b4..0240a05 100644 (file)
--- a/lib/drmtest.h
+++ b/lib/drmtest.h
@@ -357,4 +357,7 @@ void igt_enable_prefault(void);
 /* suspend and auto-resume system */
 void igt_system_suspend_autoresume(void);
 
+/* dropping priviledges */
+void igt_drop_root(void);
+
 #endif /* DRMTEST_H */

diff --git a/tests/gem_reset_stats.c b/tests/gem_reset_stats.c
index a98e2ad..034a298 100644 (file)
--- a/tests/gem_reset_stats.c
+++ b/tests/gem_reset_stats.c
@@ -874,17 +874,6 @@ static void test_close_pending_fork(const bool reverse)
        close(fd);
 }
 
-static void drop_root(void)
-{
-       igt_assert(getuid() == 0);
-
-       igt_assert(setgid(2) == 0);
-       igt_assert(setuid(2) == 0);
-
-       igt_assert(getgid() == 2);
-       igt_assert(getuid() == 2);
-}
-
 static void test_reset_count(const bool create_ctx)
 {
        int fd, h, ctx;
@@ -912,7 +901,7 @@ static void test_reset_count(const bool create_ctx)
        igt_assert(c2 == (c1 + 1));
 
        igt_fork(child, 1) {
-               drop_root();
+               igt_drop_root();
 
                c2 = get_reset_count(fd, ctx);
 
@@ -988,7 +977,7 @@ static void _test_param(const int fd, const int ctx)
        igt_fork(child, 1) {
                check_params(fd, ctx, root);
 
-               drop_root();
+               igt_drop_root();
 
                check_params(fd, ctx, user);
        }