glibc/gshadow/tst-putsgent.c

169 lines
3.8 KiB
C

/* Test for processing of invalid gshadow entries. [BZ #18724]
Copyright (C) 2015-2017 Free Software Foundation, Inc.
This file is part of the GNU C Library.
The GNU C Library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
The GNU C Library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with the GNU C Library; if not, see
<http://www.gnu.org/licenses/>. */
#include <errno.h>
#include <gshadow.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
static bool errors;
static void
check (struct sgrp e, const char *expected)
{
char *buf;
size_t buf_size;
FILE *f = open_memstream (&buf, &buf_size);
if (f == NULL)
{
printf ("open_memstream: %m\n");
errors = true;
return;
}
int ret = putsgent (&e, f);
if (expected == NULL)
{
if (ret == -1)
{
if (errno != EINVAL)
{
printf ("putsgent: unexpected error code: %m\n");
errors = true;
}
}
else
{
printf ("putsgent: unexpected success (\"%s\")\n", e.sg_namp);
errors = true;
}
}
else
{
/* Expect success. */
size_t expected_length = strlen (expected);
if (ret == 0)
{
long written = ftell (f);
if (written <= 0 || fflush (f) < 0)
{
printf ("stream error: %m\n");
errors = true;
}
else if (buf[written - 1] != '\n')
{
printf ("FAILED: \"%s\" without newline\n", expected);
errors = true;
}
else if (strncmp (buf, expected, written - 1) != 0
|| written - 1 != expected_length)
{
printf ("FAILED: \"%s\" (%ld), expected \"%s\" (%zu)\n",
buf, written - 1, expected, expected_length);
errors = true;
}
}
else
{
printf ("FAILED: putsgent (expected \"%s\"): %m\n", expected);
errors = true;
}
}
fclose (f);
free (buf);
}
static int
do_test (void)
{
check ((struct sgrp) {
.sg_namp = (char *) "root",
},
"root:::");
check ((struct sgrp) {
.sg_namp = (char *) "root",
.sg_passwd = (char *) "password",
},
"root:password::");
check ((struct sgrp) {
.sg_namp = (char *) "root",
.sg_passwd = (char *) "password",
.sg_adm = (char *[]) {(char *) "adm1", (char *) "adm2", NULL},
.sg_mem = (char *[]) {(char *) "mem1", (char *) "mem2", NULL},
},
"root:password:adm1,adm2:mem1,mem2");
/* Bad values. */
{
static const char *const bad_strings[] = {
":",
"\n",
":bad",
"\nbad",
"b:ad",
"b\nad",
"bad:",
"bad\n",
"b:a\nd",
",",
"\n,",
":,",
",bad",
"b,ad",
"bad,",
NULL
};
for (const char *const *bad = bad_strings; *bad != NULL; ++bad)
{
char *members[]
= {(char *) "first", (char *) *bad, (char *) "last", NULL};
if (strpbrk (*bad, ":\n") != NULL)
{
check ((struct sgrp) {
.sg_namp = (char *) *bad,
}, NULL);
check ((struct sgrp) {
.sg_namp = (char *) "root",
.sg_passwd = (char *) *bad,
}, NULL);
}
check ((struct sgrp) {
.sg_namp = (char *) "root",
.sg_passwd = (char *) "password",
.sg_adm = members
}, NULL);
check ((struct sgrp) {
.sg_namp = (char *) "root",
.sg_passwd = (char *) "password",
.sg_mem = members
}, NULL);
}
}
return errors;
}
#define TEST_FUNCTION do_test ()
#include "../test-skeleton.c"