1272343Sngie/* $NetBSD: t_getgroups.c,v 1.1 2011/07/07 06:57:53 jruoho Exp $ */ 2272343Sngie 3272343Sngie/*- 4272343Sngie * Copyright (c) 2011 The NetBSD Foundation, Inc. 5272343Sngie * All rights reserved. 6272343Sngie * 7272343Sngie * This code is derived from software contributed to The NetBSD Foundation 8272343Sngie * by Jukka Ruohonen. 9272343Sngie * 10272343Sngie * Redistribution and use in source and binary forms, with or without 11272343Sngie * modification, are permitted provided that the following conditions 12272343Sngie * are met: 13272343Sngie * 1. Redistributions of source code must retain the above copyright 14272343Sngie * notice, this list of conditions and the following disclaimer. 15272343Sngie * 2. Redistributions in binary form must reproduce the above copyright 16272343Sngie * notice, this list of conditions and the following disclaimer in the 17272343Sngie * documentation and/or other materials provided with the distribution. 18272343Sngie * 19272343Sngie * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20272343Sngie * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21272343Sngie * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22272343Sngie * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23272343Sngie * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24272343Sngie * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25272343Sngie * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26272343Sngie * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27272343Sngie * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28272343Sngie * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29272343Sngie * POSSIBILITY OF SUCH DAMAGE. 30272343Sngie */ 31272343Sngie#include <sys/cdefs.h> 32272343Sngie__RCSID("$NetBSD: t_getgroups.c,v 1.1 2011/07/07 06:57:53 jruoho Exp $"); 33272343Sngie 34272343Sngie#include <sys/wait.h> 35272343Sngie 36272343Sngie#include <atf-c.h> 37272343Sngie#include <errno.h> 38272343Sngie#include <limits.h> 39272343Sngie#include <stdlib.h> 40272343Sngie#include <string.h> 41272343Sngie#include <unistd.h> 42272343Sngie 43272343SngieATF_TC(getgroups_err); 44272343SngieATF_TC_HEAD(getgroups_err, tc) 45272343Sngie{ 46272343Sngie atf_tc_set_md_var(tc, "descr", "Test errors in getgroups(2)"); 47272343Sngie} 48272343Sngie 49272343SngieATF_TC_BODY(getgroups_err, tc) 50272343Sngie{ 51272343Sngie gid_t gidset[NGROUPS_MAX]; 52272343Sngie 53272343Sngie errno = 0; 54272343Sngie 55272343Sngie ATF_REQUIRE(getgroups(10, (gid_t *)-1) == -1); 56272343Sngie ATF_REQUIRE(errno == EFAULT); 57272343Sngie 58272343Sngie errno = 0; 59272343Sngie 60274626Sngie#ifdef __FreeBSD__ 61273517Sngie atf_tc_expect_fail("Reported as kern/189941"); 62273517Sngie#endif 63272343Sngie ATF_REQUIRE(getgroups(-1, gidset) == -1); 64272343Sngie ATF_REQUIRE(errno == EINVAL); 65272343Sngie} 66272343Sngie 67272343SngieATF_TC(getgroups_getgid); 68272343SngieATF_TC_HEAD(getgroups_getgid, tc) 69272343Sngie{ 70272343Sngie atf_tc_set_md_var(tc, "descr", "Test getgid(2) from getgroups(2)"); 71272343Sngie} 72272343Sngie 73272343SngieATF_TC_BODY(getgroups_getgid, tc) 74272343Sngie{ 75272343Sngie gid_t gidset[NGROUPS_MAX]; 76272343Sngie gid_t gid = getgid(); 77272343Sngie int i, n; 78272343Sngie 79272343Sngie /* 80272343Sngie * Check that getgid(2) is found from 81272343Sngie * the GIDs returned by getgroups(2). 82272343Sngie */ 83272343Sngie n = getgroups(NGROUPS_MAX, gidset); 84272343Sngie 85272343Sngie for (i = 0; i < n; i++) { 86272343Sngie 87272343Sngie if (gidset[i] == gid) 88272343Sngie return; 89272343Sngie } 90272343Sngie 91272343Sngie atf_tc_fail("getgid(2) not found from getgroups(2)"); 92272343Sngie} 93272343Sngie 94272343SngieATF_TC(getgroups_setgid); 95272343SngieATF_TC_HEAD(getgroups_setgid, tc) 96272343Sngie{ 97272343Sngie atf_tc_set_md_var(tc, "descr", "Test setgid(2) from getgroups(2)"); 98272343Sngie atf_tc_set_md_var(tc, "require.user", "root"); 99272343Sngie} 100272343Sngie 101272343SngieATF_TC_BODY(getgroups_setgid, tc) 102272343Sngie{ 103272343Sngie gid_t gidset[NGROUPS_MAX]; 104272343Sngie int i, n, rv, sta; 105272343Sngie pid_t pid; 106272343Sngie 107272343Sngie /* 108272343Sngie * Check that we can setgid(2) 109272343Sngie * to the returned group IDs. 110272343Sngie */ 111272343Sngie n = getgroups(NGROUPS_MAX, gidset); 112272343Sngie ATF_REQUIRE(n >= 0); 113272343Sngie 114272343Sngie for (i = 0; i < n; i++) { 115272343Sngie 116272343Sngie pid = fork(); 117272343Sngie ATF_REQUIRE(pid >= 0); 118272343Sngie 119272343Sngie if (pid == 0) { 120272343Sngie 121272343Sngie rv = setgid(gidset[i]); 122272343Sngie 123272343Sngie if (rv != 0) 124272343Sngie _exit(EXIT_FAILURE); 125272343Sngie 126272343Sngie _exit(EXIT_SUCCESS); 127272343Sngie } 128272343Sngie 129272343Sngie (void)wait(&sta); 130272343Sngie 131272343Sngie if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS) 132272343Sngie atf_tc_fail("getgroups(2) is inconsistent"); 133272343Sngie } 134272343Sngie} 135272343Sngie 136272343SngieATF_TC(getgroups_zero); 137272343SngieATF_TC_HEAD(getgroups_zero, tc) 138272343Sngie{ 139272343Sngie atf_tc_set_md_var(tc, "descr", "Test getgroups(2) with zero param"); 140272343Sngie} 141272343Sngie 142272343SngieATF_TC_BODY(getgroups_zero, tc) 143272343Sngie{ 144272343Sngie const gid_t val = 123456789; 145272343Sngie gid_t gidset[NGROUPS_MAX]; 146272343Sngie size_t i; 147272343Sngie 148272343Sngie /* 149272343Sngie * If the first parameter is zero, the number 150272343Sngie * of groups should be returned but the supplied 151272343Sngie * buffer should remain intact. 152272343Sngie */ 153272343Sngie for (i = 0; i < __arraycount(gidset); i++) 154272343Sngie gidset[i] = val; 155272343Sngie 156272343Sngie ATF_REQUIRE(getgroups(0, gidset) >= 0); 157272343Sngie 158272343Sngie for (i = 0; i < __arraycount(gidset); i++) { 159272343Sngie 160272343Sngie if (gidset[i] != val) 161272343Sngie atf_tc_fail("getgroups(2) modified the buffer"); 162272343Sngie } 163272343Sngie} 164272343Sngie 165272343SngieATF_TP_ADD_TCS(tp) 166272343Sngie{ 167272343Sngie 168272343Sngie ATF_TP_ADD_TC(tp, getgroups_err); 169272343Sngie ATF_TP_ADD_TC(tp, getgroups_getgid); 170272343Sngie ATF_TP_ADD_TC(tp, getgroups_setgid); 171272343Sngie ATF_TP_ADD_TC(tp, getgroups_zero); 172272343Sngie 173272343Sngie return atf_no_error(); 174272343Sngie} 175