tests/tst-pam_set_item.c

   1 /*
   2  * Redistribution and use in source and binary forms, with or without
   3  * modification, are permitted provided that the following conditions
   4  * are met:
   5  * 1. Redistributions of source code must retain the above copyright
   6  *    notice, and the entire permission notice in its entirety,
   7  *    including the disclaimer of warranties.
   8  * 2. Redistributions in binary form must reproduce the above copyright
   9  *    notice, this list of conditions and the following disclaimer in the
  10  *    documentation and/or other materials provided with the distribution.
  11  * 3. The name of the author may not be used to endorse or promote
  12  *    products derived from this software without specific prior
  13  *    written permission.
  14  *
  15  * ALTERNATIVELY, this product may be distributed under the terms of
  16  * the GNU Public License, in which case the provisions of the GPL are
  17  * required INSTEAD OF the above restrictions.  (This clause is
  18  * necessary due to a potential bad interaction between the GPL and
  19  * the restrictions contained in a BSD-style copyright.)
  20  *
  21  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED
  22  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  23  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  24  * DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
  25  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  26  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  27  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  29  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  31  * OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 #ifdef HAVE_CONFIG_H
  35 #include <config.h>
  36 #endif
  37
  38 #include <stdio.h>
  39 #include <unistd.h>
  40 #include <string.h>
  41
  42 #include <security/pam_appl.h>
  43
  44 struct mapping {
  45   int type;
  46   const char *string;
  47   int expected;
  48   const char *new_value;
  49 };
  50
  51 struct mapping items[] = {
  52   {PAM_SERVICE, "PAM_SERVICE", PAM_SUCCESS, "logout"},
  53   {PAM_USER, "PAM_USER", PAM_SUCCESS, "noroot"},
  54   {PAM_TTY, "PAM_TTY", PAM_SUCCESS, "TTyX"},
  55   {PAM_RHOST, "PAM_RHOST", PAM_SUCCESS, "remote"},
  56   {PAM_AUTHTOK, "PAM_AUTHTOK", PAM_BAD_ITEM, "none"},
  57   {PAM_OLDAUTHTOK, "PAM_OLDAUTHTOK", PAM_BAD_ITEM, "none"},
  58   {PAM_RUSER, "PAM_RUSER", PAM_SUCCESS, "noroot"},
  59   {PAM_USER_PROMPT, "PAM_USER_PROMPT", PAM_SUCCESS, "your name: "},
  60   {PAM_FAIL_DELAY, "PAM_FAIL_DELAY", PAM_SUCCESS, "4000"},
  61   {PAM_AUTHTOK_TYPE, "PAM_AUTHTOK_TYPE", PAM_SUCCESS, "U**X"}
  62 };
  63
  64 int
  65 main (void)
  66 {
  67   const char *service = "dummy";
  68   const char *user = "root";
  69   struct pam_conv conv;
  70   pam_handle_t *pamh;
  71   int retval, num, i;
  72
  73   /* 1: Call with NULL as pam handle */
  74   retval = pam_set_item (NULL, PAM_SERVICE, "dummy");
  75   if (retval == PAM_SUCCESS)
  76     {
  77       fprintf (stderr, "pam_set_item (NULL, ...) returned PAM_SUCCESS\n");
  78       return 1;
  79     }
  80
  81   /* setup pam handle */
  82   retval = pam_start (service, user, &conv, &pamh);
  83   if (retval != PAM_SUCCESS)
  84     {
  85       fprintf (stderr, "pam_start (%s, %s, &conv, &pamh) returned %d\n",
  86                service, user, retval);
  87       return 1;
  88     }
  89
  90   /* 2: check for bad item  */
  91   retval = pam_set_item (pamh, -1, "dummy");
  92   if (retval != PAM_BAD_ITEM)
  93     {
  94       fprintf (stderr,
  95                "pam_set_item returned %d when expecting PAM_BAD_ITEM\n",
  96                retval);
  97       return 1;
  98     }
  99
 100   /* 3: try to set PAM_CONV to NULL  */
 101   retval = pam_set_item (pamh, PAM_CONV, NULL);
 102   if (retval != PAM_PERM_DENIED)
 103     {
 104       fprintf (stderr,
 105                "pam_set_item (pamh, PAM_CONV, NULL) returned %d\n",
 106                retval);
 107       return 1;
 108     }
 109
 110   /* 4: try to replace all items */
 111   num = sizeof(items) / sizeof(struct mapping);
 112
 113   for (i = 0; i < num; i++)
 114     {
 115       retval = pam_set_item (pamh, items[i].type, items[i].new_value);
 116
 117       if (retval != items[i].expected)
 118         {
 119           fprintf (stderr,
 120                    "pam_set_item failed to set value for %s. Returned %d\n",
 121                    items[i].string, retval);
 122           return 1;
 123         }
 124       else if (items[i].expected == PAM_SUCCESS)
 125         {
 126           const void *value;
 127
 128           retval = pam_get_item (pamh, items[i].type, &value);
 129           if (retval != PAM_SUCCESS)
 130             {
 131               fprintf (stderr,
 132                        "pam_get_item was not able to fetch changed value: %d\n",
 133                        retval);
 134               return 1;
 135             }
 136           if (strcmp (items[i].new_value, value) != 0)
 137             {
 138               fprintf (stderr,
 139                        "pam_get_item got wrong value:\n"
 140                        "expected: %s\n"
 141                        "got: %s\n", items[i].new_value, (const char *)value);
 142               return 1;
 143             }
 144         }
 145     }
 146
 147   pam_end (pamh, 0);
 148
 149   return 0;
 150 }