121 lines
2.6 KiB
C
121 lines
2.6 KiB
C
// 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;
|
|
}
|