granicus.if.org Git - linux-pam/blob - xtests/tst-pam_limits1.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 /*
  35   test case:
  36
  37   Check the following line in limits.conf:
  38   *               soft    nice     19
  39   *               hard    nice     -20
  40
  41   getrlimit should return soft=1 and hard=40.
  42 */
  43
  44 #ifdef HAVE_CONFIG_H
  45 #include <config.h>
  46 #endif
  47
  48 #include <stdio.h>
  49 #include <stdlib.h>
  50 #include <string.h>
  51 #include <sys/time.h>
  52 #include <sys/resource.h>
  53
  54 #include <security/pam_appl.h>
  55
  56 /* A conversation function which uses an internally-stored value for
  57    the responses. */
  58 static int
  59 fake_conv (int num_msg, const struct pam_message **msgm UNUSED,
  60            struct pam_response **response, void *appdata_ptr UNUSED)
  61 {
  62   struct pam_response *reply;
  63   int count;
  64
  65   /* Sanity test. */
  66   if (num_msg <= 0)
  67     return PAM_CONV_ERR;
  68
  69   /* Allocate memory for the responses. */
  70   reply = calloc (num_msg, sizeof (struct pam_response));
  71   if (reply == NULL)
  72     return PAM_CONV_ERR;
  73
  74   /* Each prompt elicits the same response. */
  75   for (count = 0; count < num_msg; ++count)
  76     {
  77       reply[count].resp_retcode = 0;
  78       reply[count].resp = strdup ("!!");
  79     }
  80
  81   /* Set the pointers in the response structure and return. */
  82   *response = reply;
  83   return PAM_SUCCESS;
  84 }
  85
  86 static struct pam_conv conv = {
  87     fake_conv,
  88     NULL
  89 };
  90
  91 int
  92 main(int argc, char *argv[])
  93 {
  94   pam_handle_t *pamh = NULL;
  95   const char *user="tstpamlimits";
  96   int retval;
  97   int debug = 0;
  98
  99   if (argc > 1 && strcmp (argv[1], "-d") == 0)
 100     debug = 1;
 101
 102 #ifdef RLIMIT_NICE
 103   retval = pam_start("tst-pam_limits1", user, &conv, &pamh);
 104   if (retval != PAM_SUCCESS)
 105     {
 106       if (debug)
 107         fprintf (stderr, "pam_limits1: pam_start returned %d\n", retval);
 108       return 1;
 109     }
 110
 111   retval = pam_set_item (pamh, PAM_TTY, "/dev/tty1");
 112   if (retval != PAM_SUCCESS)
 113     {
 114       if (debug)
 115         fprintf (stderr,
 116                  "pam_limits1: pam_set_item(PAM_TTY) returned %d\n",
 117                  retval);
 118       return 1;
 119     }
 120
 121   retval = pam_open_session (pamh, 0);
 122   if (retval != PAM_SUCCESS)
 123     {
 124       if (debug)
 125         fprintf (stderr, "pam_limits1: pam_open_session returned %d\n",
 126                  retval);
 127       return 1;
 128     }
 129
 130   struct rlimit rlim;
 131
 132   getrlimit (RLIMIT_NICE, &rlim);
 133
 134   if (rlim.rlim_cur != 1 && rlim.rlim_max != 40)
 135     {
 136       if (debug)
 137         fprintf (stderr, "pam_limits1: getrlimit failed, soft=%u, hard=%u\n",
 138                  (unsigned int) rlim.rlim_cur, (unsigned int) rlim.rlim_max);
 139       return 1;
 140     }
 141
 142   retval = pam_end (pamh,retval);
 143   if (retval != PAM_SUCCESS)
 144     {
 145       if (debug)
 146         fprintf (stderr, "pam_limits1: pam_end returned %d\n", retval);
 147       return 1;
 148     }
 149   return 0;
 150 #else
 151   if (debug)
 152     fprintf (stderr, "pam_limits1: RLIMIT_NICE does not exist)\n");
 153
 154   return 77;
 155 #endif
 156 }