| // SPDX-License-Identifier: GPL-2.0-or-later |
| /* |
| * Copyright (c) 2021 Linux Test Project |
| */ |
| |
| #include <sys/types.h> |
| #include <grp.h> |
| #include <errno.h> |
| |
| #define TST_NO_DEFAULT_MAIN |
| #include "tst_test.h" |
| #include "tst_uid.h" |
| |
| #define MAX_GID 32767 |
| |
| gid_t tst_get_free_gid_(const char *file, const int lineno, gid_t skip) |
| { |
| gid_t ret; |
| |
| errno = 0; |
| |
| for (ret = 1; ret < MAX_GID; ret++) { |
| if (ret == skip || getgrgid(ret)) |
| continue; |
| |
| if (errno == 0 || errno == ENOENT || errno == ESRCH) { |
| tst_res_(file, lineno, TINFO | TERRNO, |
| "Found unused GID %d", (int)ret); |
| return ret; |
| } |
| |
| tst_brk_(file, lineno, TBROK|TERRNO, "Group ID lookup failed"); |
| return (gid_t)-1; |
| } |
| |
| tst_brk_(file, lineno, TBROK, "No free group ID found"); |
| return (gid_t)-1; |
| } |
| |
| void tst_get_uids(uid_t *buf, unsigned int start, unsigned int count) |
| { |
| unsigned int i, j; |
| uid_t id; |
| |
| for (i = start, id = 1; i < count; id++) { |
| for (j = 0; j < start; j++) { |
| if (buf[j] == id) |
| break; |
| } |
| |
| if (j >= start) |
| buf[i++] = id; |
| } |
| } |
| |
| void tst_get_gids(gid_t *buf, unsigned int start, unsigned int count) |
| { |
| unsigned int i, j; |
| gid_t id; |
| |
| for (i = start, id = 1; i < count; id++) { |
| for (j = 0; j < start; j++) { |
| if (buf[j] == id) |
| break; |
| } |
| |
| if (j >= start) |
| buf[i++] = id; |
| } |
| } |
| |
| int tst_check_resuid_(const char *file, const int lineno, const char *callstr, |
| uid_t exp_ruid, uid_t exp_euid, uid_t exp_suid) |
| { |
| uid_t ruid, euid, suid; |
| |
| SAFE_GETRESUID(&ruid, &euid, &suid); |
| |
| if (ruid == exp_ruid && euid == exp_euid && suid == exp_suid) |
| return 1; |
| |
| if (callstr) { |
| tst_res_(file, lineno, TFAIL, "Unexpected process UID after %s", |
| callstr); |
| } else { |
| tst_res_(file, lineno, TFAIL, "Unexpected process UID"); |
| } |
| |
| tst_res_(file, lineno, TINFO, "Got: ruid = %d, euid = %d, suid = %d", |
| (int)ruid, (int)euid, (int)suid); |
| tst_res_(file, lineno, TINFO, |
| "Expected: ruid = %d, euid = %d, suid = %d", |
| (int)exp_ruid, (int)exp_euid, (int)exp_suid); |
| return 0; |
| } |
| |
| int tst_check_resgid_(const char *file, const int lineno, const char *callstr, |
| gid_t exp_rgid, gid_t exp_egid, gid_t exp_sgid) |
| { |
| gid_t rgid, egid, sgid; |
| |
| SAFE_GETRESGID(&rgid, &egid, &sgid); |
| |
| if (rgid == exp_rgid && egid == exp_egid && sgid == exp_sgid) |
| return 1; |
| |
| if (callstr) { |
| tst_res_(file, lineno, TFAIL, "Unexpected process GID after %s", |
| callstr); |
| } else { |
| tst_res_(file, lineno, TFAIL, "Unexpected process GID"); |
| } |
| |
| tst_res_(file, lineno, TINFO, "Got: rgid = %d, egid = %d, sgid = %d", |
| (int)rgid, (int)egid, (int)sgid); |
| tst_res_(file, lineno, TINFO, |
| "Expected: rgid = %d, egid = %d, sgid = %d", |
| (int)exp_rgid, (int)exp_egid, (int)exp_sgid); |
| return 0; |
| } |