granicus.if.org Git - linux-pam/blob - tests/tst-pam_get_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 #include <stdio.h>
  35 #include <unistd.h>
  36
  37 #include <security/pam_appl.h>
  38
  39 struct mapping {
  40   int type;
  41   const char *string;
  42   int expected;
  43 };
  44
  45 struct mapping items[] = {
  46   {PAM_SERVICE, "PAM_SERVICE", PAM_SUCCESS},
  47   {PAM_USER, "PAM_USER", 0},
  48   {PAM_TTY, "PAM_TTY", 0},
  49   {PAM_RHOST, "PAM_RHOST", 0},
  50   {PAM_CONV, "PAM_CONV", 0},
  51   {PAM_AUTHTOK, "PAM_AUTHTOK", PAM_BAD_ITEM},
  52   {PAM_OLDAUTHTOK, "PAM_OLDAUTHTOK", PAM_BAD_ITEM},
  53   {PAM_RUSER, "PAM_RUSER", 0},
  54   {PAM_USER_PROMPT, "PAM_USER_PROMPT", 0},
  55   {PAM_FAIL_DELAY, "PAM_FAIL_DELAY", 0}
  56 };
  57
  58 int
  59 main (void)
  60 {
  61   const char *service = "dummy";
  62   const char *user = "root";
  63   struct pam_conv conv;
  64   pam_handle_t *pamh;
  65   int retval, num, i;
  66   const void *value;
  67
  68   /* 1: Call with NULL as pam handle */
  69   retval = pam_get_item (NULL, PAM_SERVICE, &value);
  70   if (retval == PAM_SUCCESS)
  71     {
  72       fprintf (stderr, "pam_get_item (NULL, 0) returned PAM_SUCCESS\n");
  73       return 1;
  74     }
  75
  76   /* setup pam handle */
  77   retval = pam_start (service, user, &conv, &pamh);
  78   if (retval != PAM_SUCCESS)
  79     {
  80       fprintf (stderr, "pam_start (%s, %s, &conv, &pamh) returned %d\n",
  81                service, user, retval);
  82       return 1;
  83     }
  84
  85   /* 2: check for valid item types. Expected return value is
  86      PAM_SUCCESS, except it has to fail. */
  87   num = sizeof(items) / sizeof(struct mapping);
  88
  89   for (i = 0; i < num; i++)
  90     {
  91       retval = pam_get_item (pamh, items[i].type, &value);
  92
  93       if (retval != items[i].expected)
  94         {
  95           fprintf (stderr,
  96                    "pam_get_item failed to get value for %s. Returned %d\n",
  97                    items[i].string, retval);
  98           return 1;
  99         }
 100     }
 101
 102   /* 3: check for bad item  */
 103   retval = pam_get_item (pamh, -1, &value);
 104   if (retval != PAM_BAD_ITEM)
 105     {
 106       fprintf (stderr,
 107                "pam_get_item returned %d when expecting PAM_BAD_ITEM\n",
 108                retval);
 109       return 1;
 110     }
 111
 112   /* 4: check for valid item types, but NULL as value address. */
 113   for (i = 0; i < num; i++)
 114     {
 115       retval = pam_get_item (pamh, items[i].type, NULL);
 116
 117       if (retval != PAM_PERM_DENIED)
 118         {
 119           fprintf (stderr,
 120                    "pam_get_item returned %d to get value for %s\n",
 121                    retval, items[i].string);
 122           return 1;
 123         }
 124     }
 125
 126   pam_end (pamh, 0);
 127
 128   return 0;
 129 }