1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
|
#include "config.h"
#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <cmocka.h>
#include <sys/types.h>
#include <unistd.h>
#include <errno.h>
#include <pwd.h>
static void test_uwrap_seteuid(void **state)
{
int rc;
uid_t u;
#ifdef HAVE_GETRESUID
uid_t cp_ruid, cp_euid, cp_suid;
#endif
(void) state; /* unused */
#ifdef HAVE_GETRESUID
cp_ruid = cp_euid = cp_suid = -1;
rc = getresuid(&cp_ruid, &cp_euid, &cp_suid);
assert_return_code(rc, errno);
assert_int_equal(cp_ruid, 0);
assert_int_equal(cp_euid, 0);
assert_int_equal(cp_suid, 0);
#endif
u = geteuid();
assert_int_equal(u, 0x0);
#ifndef BSD
/* BSD sets 0xFFFFFF as the UID number in this case */
rc = seteuid(-1);
assert_int_equal(rc, -1);
assert_int_equal(errno, EINVAL);
#endif
u = geteuid();
assert_int_equal(u, 0x0);
rc = seteuid(0x4444);
assert_return_code(rc, errno);
#ifdef HAVE_GETRESUID
cp_ruid = cp_euid = cp_suid = -1;
rc = getresuid(&cp_ruid, &cp_euid, &cp_suid);
assert_return_code(rc, errno);
assert_int_equal(cp_ruid, 0);
assert_int_equal(cp_euid, 0x4444);
assert_int_equal(cp_suid, 0);
#endif
u = geteuid();
assert_int_equal(u, 0x4444);
/*
* The euid needs to be 0 in order to change to an
* unknown value (here 0x5555).
*/
rc = seteuid(0x5555);
assert_int_equal(rc, -1);
assert_int_equal(errno, EPERM);
#ifdef HAVE_GETRESUID
cp_ruid = cp_euid = cp_suid = -1;
rc = getresuid(&cp_ruid, &cp_euid, &cp_suid);
assert_return_code(rc, errno);
assert_int_equal(cp_ruid, 0);
assert_int_equal(cp_euid, 0x4444);
assert_int_equal(cp_suid, 0);
#endif
u = geteuid();
assert_int_equal(u, 0x4444);
rc = seteuid(0);
assert_int_equal(rc, 0);
#ifdef HAVE_GETRESUID
cp_ruid = cp_euid = cp_suid = -1;
rc = getresuid(&cp_ruid, &cp_euid, &cp_suid);
assert_return_code(rc, errno);
assert_int_equal(cp_ruid, 0);
assert_int_equal(cp_euid, 0);
assert_int_equal(cp_suid, 0);
#endif
u = geteuid();
assert_int_equal(u, 0);
rc = seteuid(0x5555);
assert_return_code(rc, errno);
#ifdef HAVE_GETRESUID
cp_ruid = cp_euid = cp_suid = -1;
rc = getresuid(&cp_ruid, &cp_euid, &cp_suid);
assert_return_code(rc, errno);
assert_int_equal(cp_ruid, 0);
assert_int_equal(cp_euid, 0x5555);
assert_int_equal(cp_suid, 0);
#endif
u = geteuid();
assert_int_equal(u, 0x5555);
}
int main(void) {
int rc;
const struct CMUnitTest uwrap_tests[] = {
cmocka_unit_test(test_uwrap_seteuid),
};
rc = cmocka_run_group_tests(uwrap_tests, NULL, NULL);
return rc;
}
|