google3/third_party/grte/v5_src/glibc-2.27/pwd/tst-putpwent.c

/* Test for processing of invalid passwd entries.  [BZ #18724]
   Copyright (C) 2015-2018 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 <pwd.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

static bool errors;

static void
check (struct passwd p, 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 = putpwent (&p, f);

  if (expected == NULL)
    {
      if (ret == -1)
	{
	  if (errno != EINVAL)
	    {
	      printf ("putpwent: unexpected error code: %m\n");
	      errors = true;
	    }
	}
      else
	{
	  printf ("putpwent: unexpected success (\"%s\")\n", p.pw_name);
	  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: putpwent (expected \"%s\"): %m\n", expected);
	  errors = true;
	}
    }

  fclose (f);
  free (buf);
}

static int
do_test (void)
{
  check ((struct passwd) {
      .pw_name = (char *) "root",
    },
    "root::0:0:::");
  check ((struct passwd) {
      .pw_name = (char *) "root",
      .pw_passwd = (char *) "password",
    },
    "root:password:0:0:::");
  check ((struct passwd) {
      .pw_name = (char *) "root",
      .pw_passwd = (char *) "password",
      .pw_uid = 12,
      .pw_gid = 34,
      .pw_gecos = (char *) "gecos",
      .pw_dir = (char *) "home",
      .pw_shell = (char *) "shell",
    },
    "root:password:12:34:gecos:home:shell");
  check ((struct passwd) {
      .pw_name = (char *) "root",
      .pw_passwd = (char *) "password",
      .pw_uid = 12,
      .pw_gid = 34,
      .pw_gecos = (char *) ":ge\n:cos\n",
      .pw_dir = (char *) "home",
      .pw_shell = (char *) "shell",
    },
    "root:password:12:34: ge  cos :home:shell");

  /* Bad values.  */
  {
    static const char *const bad_strings[] = {
      ":",
      "\n",
      ":bad",
      "\nbad",
      "b:ad",
      "b\nad",
      "bad:",
      "bad\n",
      "b:a\nd",
      NULL
    };
    for (const char *const *bad = bad_strings; *bad != NULL; ++bad)
      {
	check ((struct passwd) {
	    .pw_name = (char *) *bad,
	  }, NULL);
	check ((struct passwd) {
	    .pw_name = (char *) "root",
	    .pw_passwd = (char *) *bad,
	  }, NULL);
	check ((struct passwd) {
	    .pw_name = (char *) "root",
	    .pw_dir = (char *) *bad,
	  }, NULL);
	check ((struct passwd) {
	    .pw_name = (char *) "root",
	    .pw_shell = (char *) *bad,
	  }, NULL);
      }
  }

  return errors > 0;
}

#define TEST_FUNCTION do_test ()
#include "../test-skeleton.c"