1  /* Test for processing of invalid passwd entries.  [BZ #18724]
       2     Copyright (C) 2015-2023 Free Software Foundation, Inc.
       3     This file is part of the GNU C Library.
       4  
       5     The GNU C Library is free software; you can redistribute it and/or
       6     modify it under the terms of the GNU Lesser General Public
       7     License as published by the Free Software Foundation; either
       8     version 2.1 of the License, or (at your option) any later version.
       9  
      10     The GNU C Library is distributed in the hope that it will be useful,
      11     but WITHOUT ANY WARRANTY; without even the implied warranty of
      12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      13     Lesser General Public License for more details.
      14  
      15     You should have received a copy of the GNU Lesser General Public
      16     License along with the GNU C Library; if not, see
      17     <https://www.gnu.org/licenses/>.  */
      18  
      19  #include <errno.h>
      20  #include <pwd.h>
      21  #include <stdbool.h>
      22  #include <stdio.h>
      23  #include <string.h>
      24  #include <stdlib.h>
      25  
      26  static bool errors;
      27  
      28  static void
      29  check (struct passwd p, const char *expected)
      30  {
      31    char *buf;
      32    size_t buf_size;
      33    FILE *f = open_memstream (&buf, &buf_size);
      34  
      35    if (f == NULL)
      36      {
      37        printf ("open_memstream: %m\n");
      38        errors = true;
      39        return;
      40      }
      41  
      42    int ret = putpwent (&p, f);
      43  
      44    if (expected == NULL)
      45      {
      46        if (ret == -1)
      47  	{
      48  	  if (errno != EINVAL)
      49  	    {
      50  	      printf ("putpwent: unexpected error code: %m\n");
      51  	      errors = true;
      52  	    }
      53  	}
      54        else
      55  	{
      56  	  printf ("putpwent: unexpected success (\"%s\")\n", p.pw_name);
      57  	  errors = true;
      58  	}
      59      }
      60    else
      61      {
      62        /* Expect success.  */
      63        size_t expected_length = strlen (expected);
      64        if (ret == 0)
      65  	{
      66  	  long written = ftell (f);
      67  
      68  	  if (written <= 0 || fflush (f) < 0)
      69  	    {
      70  	      printf ("stream error: %m\n");
      71  	      errors = true;
      72  	    }
      73  	  else if (buf[written - 1] != '\n')
      74  	    {
      75  	      printf ("FAILED: \"%s\" without newline\n", expected);
      76  	      errors = true;
      77  	    }
      78  	  else if (strncmp (buf, expected, written - 1) != 0
      79  		   || written - 1 != expected_length)
      80  	    {
      81  	      printf ("FAILED: \"%s\" (%ld), expected \"%s\" (%zu)\n",
      82  		      buf, written - 1, expected, expected_length);
      83  	      errors = true;
      84  	    }
      85  	}
      86        else
      87  	{
      88  	  printf ("FAILED: putpwent (expected \"%s\"): %m\n", expected);
      89  	  errors = true;
      90  	}
      91      }
      92  
      93    fclose (f);
      94    free (buf);
      95  }
      96  
      97  static int
      98  do_test (void)
      99  {
     100    check ((struct passwd) {
     101        .pw_name = (char *) "root",
     102      },
     103      "root::0:0:::");
     104    check ((struct passwd) {
     105        .pw_name = (char *) "root",
     106        .pw_passwd = (char *) "password",
     107      },
     108      "root:password:0:0:::");
     109    check ((struct passwd) {
     110        .pw_name = (char *) "root",
     111        .pw_passwd = (char *) "password",
     112        .pw_uid = 12,
     113        .pw_gid = 34,
     114        .pw_gecos = (char *) "gecos",
     115        .pw_dir = (char *) "home",
     116        .pw_shell = (char *) "shell",
     117      },
     118      "root:password:12:34:gecos:home:shell");
     119    check ((struct passwd) {
     120        .pw_name = (char *) "root",
     121        .pw_passwd = (char *) "password",
     122        .pw_uid = 12,
     123        .pw_gid = 34,
     124        .pw_gecos = (char *) ":ge\n:cos\n",
     125        .pw_dir = (char *) "home",
     126        .pw_shell = (char *) "shell",
     127      },
     128      "root:password:12:34: ge  cos :home:shell");
     129  
     130    /* Bad values.  */
     131    {
     132      static const char *const bad_strings[] = {
     133        ":",
     134        "\n",
     135        ":bad",
     136        "\nbad",
     137        "b:ad",
     138        "b\nad",
     139        "bad:",
     140        "bad\n",
     141        "b:a\nd",
     142        NULL
     143      };
     144      for (const char *const *bad = bad_strings; *bad != NULL; ++bad)
     145        {
     146  	check ((struct passwd) {
     147  	    .pw_name = (char *) *bad,
     148  	  }, NULL);
     149  	check ((struct passwd) {
     150  	    .pw_name = (char *) "root",
     151  	    .pw_passwd = (char *) *bad,
     152  	  }, NULL);
     153  	check ((struct passwd) {
     154  	    .pw_name = (char *) "root",
     155  	    .pw_dir = (char *) *bad,
     156  	  }, NULL);
     157  	check ((struct passwd) {
     158  	    .pw_name = (char *) "root",
     159  	    .pw_shell = (char *) *bad,
     160  	  }, NULL);
     161        }
     162    }
     163  
     164    return errors > 0;
     165  }
     166  
     167  #define TEST_FUNCTION do_test ()
     168  #include "../test-skeleton.c"