tests/lib/libc/sys/t_getlogin.c

   1 /* $NetBSD: t_getlogin.c,v 1.1 2011/07/07 06:57:53 jruoho Exp $ */
   2
   3 /*-
   4  * Copyright (c) 2011 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Jukka Ruohonen.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31 #include <sys/cdefs.h>
  32 __RCSID("$NetBSD: t_getlogin.c,v 1.1 2011/07/07 06:57:53 jruoho Exp $");
  33
  34 #include <sys/param.h>
  35 #include <sys/wait.h>
  36
  37 #include <atf-c.h>
  38 #include <errno.h>
  39 #include <stdlib.h>
  40 #include <string.h>
  41 #include <unistd.h>
  42
  43 ATF_TC(getlogin_r_err);
  44 ATF_TC_HEAD(getlogin_r_err, tc)
  45 {
  46         atf_tc_set_md_var(tc, "descr", "Test errors from getlogin_r(2)");
  47 }
  48
  49 ATF_TC_BODY(getlogin_r_err, tc)
  50 {
  51         char small[0];
  52
  53         ATF_REQUIRE(getlogin_r(small, sizeof(small)) == ERANGE);
  54 }
  55
  56 ATF_TC(getlogin_same);
  57 ATF_TC_HEAD(getlogin_same, tc)
  58 {
  59         atf_tc_set_md_var(tc, "descr", "getlogin(2) vs. getlogin_r(2)");
  60 }
  61
  62 ATF_TC_BODY(getlogin_same, tc)
  63 {
  64         char buf[MAXLOGNAME];
  65         char *str;
  66
  67         str = getlogin();
  68
  69         if (str == NULL)
  70                 return;
  71
  72         ATF_REQUIRE(getlogin_r(buf, sizeof(buf)) == 0);
  73
  74         if (strcmp(str, buf) != 0)
  75                 atf_tc_fail("getlogin(2) and getlogin_r(2) differ");
  76 }
  77
  78 ATF_TC(setlogin_basic);
  79 ATF_TC_HEAD(setlogin_basic, tc)
  80 {
  81         atf_tc_set_md_var(tc, "descr", "Test that setlogin(2) works");
  82         atf_tc_set_md_var(tc, "require.user", "root");
  83 }
  84
  85 ATF_TC_BODY(setlogin_basic, tc)
  86 {
  87         char *name;
  88         pid_t pid;
  89         int sta;
  90
  91         pid = fork();
  92         ATF_REQUIRE(pid >= 0);
  93
  94         if (pid == 0) {
  95
  96                 (void)setsid();
  97
  98                 if (setlogin("foobar") != 0)
  99                         _exit(EXIT_FAILURE);
 100
 101                 name = getlogin();
 102
 103                 if (name == NULL)
 104                         _exit(EXIT_FAILURE);
 105
 106                 if (strcmp(name, "foobar") != 0)
 107                         _exit(EXIT_FAILURE);
 108
 109                 _exit(EXIT_SUCCESS);
 110         }
 111
 112         (void)wait(&sta);
 113
 114         if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS)
 115                 atf_tc_fail("setlogin(2) failed to set login name");
 116 }
 117
 118 ATF_TC(setlogin_err);
 119 ATF_TC_HEAD(setlogin_err, tc)
 120 {
 121         atf_tc_set_md_var(tc, "descr", "Test errors from setlogin(2)");
 122         atf_tc_set_md_var(tc, "require.user", "root");
 123 }
 124
 125 ATF_TC_BODY(setlogin_err, tc)
 126 {
 127         char buf[MAXLOGNAME + 1];
 128         char *name;
 129         pid_t pid;
 130         int sta;
 131
 132         pid = fork();
 133         ATF_REQUIRE(pid >= 0);
 134
 135         (void)memset(buf, 'x', sizeof(buf));
 136
 137         if (pid == 0) {
 138
 139                 (void)setsid();
 140
 141                 errno = 0;
 142
 143                 if (setlogin(buf) != -1)
 144                         _exit(EINVAL);
 145
 146                 if (errno != EINVAL)
 147                         _exit(EINVAL);
 148
 149                 errno = 0;
 150
 151                 if (setlogin((void *)-1) != -1)
 152                         _exit(EFAULT);
 153
 154                 if (errno != EFAULT)
 155                         _exit(EFAULT);
 156
 157                 name = getlogin();
 158
 159                 if (name == NULL)
 160                         _exit(EXIT_FAILURE);
 161
 162                 if (strcmp(name, "foobar") == 0)
 163                         _exit(EXIT_FAILURE);
 164
 165                 _exit(EXIT_SUCCESS);
 166         }
 167
 168         (void)wait(&sta);
 169
 170         if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS) {
 171
 172                 if (WEXITSTATUS(sta) == EFAULT)
 173                         atf_tc_fail("expected EFAULT, but the call succeeded");
 174
 175                 if (WEXITSTATUS(sta) == EINVAL)
 176                         atf_tc_fail("expected EINVAL, but the call succeeded");
 177
 178                 atf_tc_fail("setlogin(2) failed, but login name was set");
 179         }
 180 }
 181
 182 ATF_TC(setlogin_perm);
 183 ATF_TC_HEAD(setlogin_perm, tc)
 184 {
 185         atf_tc_set_md_var(tc, "descr", "Test setlogin(2) as normal user");
 186         atf_tc_set_md_var(tc, "require.user", "unprivileged");
 187 }
 188
 189 ATF_TC_BODY(setlogin_perm, tc)
 190 {
 191         char *name;
 192         pid_t pid;
 193         int sta;
 194
 195         pid = fork();
 196         ATF_REQUIRE(pid >= 0);
 197
 198         if (pid == 0) {
 199
 200                 (void)setsid();
 201
 202                 errno = 0;
 203
 204                 if (setlogin("foobar") != -1)
 205                         _exit(EXIT_FAILURE);
 206
 207                 if (errno != EPERM)
 208                         _exit(EXIT_FAILURE);
 209
 210                 name = getlogin();
 211
 212                 if (name == NULL)
 213                         _exit(EXIT_FAILURE);
 214
 215                 if (strcmp(name, "foobar") == 0)
 216                         _exit(EXIT_FAILURE);
 217
 218                 _exit(EXIT_SUCCESS);
 219         }
 220
 221         (void)wait(&sta);
 222
 223         if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS)
 224                 atf_tc_fail("login name was set as an unprivileged user");
 225 }
 226
 227 ATF_TP_ADD_TCS(tp)
 228 {
 229
 230         ATF_TP_ADD_TC(tp, getlogin_r_err);
 231         ATF_TP_ADD_TC(tp, getlogin_same);
 232         ATF_TP_ADD_TC(tp, setlogin_basic);
 233         ATF_TP_ADD_TC(tp, setlogin_err);
 234         ATF_TP_ADD_TC(tp, setlogin_perm);
 235
 236         return atf_no_error();
 237 }