tst-pam_get_item.c 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
  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. #ifdef HAVE_CONFIG_H
  34. #include <config.h>
  35. #endif
  36. #include <stdio.h>
  37. #include <unistd.h>
  38. #include <security/pam_appl.h>
  39. #include "pam_inline.h"
  40. struct mapping {
  41. int type;
  42. const char *string;
  43. int expected;
  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. {PAM_AUTHTOK_TYPE, "PAM_AUTHTOK_TYPE", 0}
  57. };
  58. int
  59. main (void)
  60. {
  61. const char *service = "dummy";
  62. const char *user = "root";
  63. struct pam_conv conv = { NULL, NULL };
  64. pam_handle_t *pamh;
  65. int retval;
  66. unsigned int i;
  67. const void *value;
  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. /* setup pam handle */
  76. retval = pam_start (service, user, &conv, &pamh);
  77. if (retval != PAM_SUCCESS)
  78. {
  79. fprintf (stderr, "pam_start (%s, %s, &conv, &pamh) returned %d\n",
  80. service, user, retval);
  81. return 1;
  82. }
  83. /* 2: check for valid item types. Expected return value is
  84. PAM_SUCCESS, except it has to fail. */
  85. for (i = 0; i < PAM_ARRAY_SIZE(items); i++)
  86. {
  87. retval = pam_get_item (pamh, items[i].type, &value);
  88. if (retval != items[i].expected)
  89. {
  90. fprintf (stderr,
  91. "pam_get_item failed to get value for %s. Returned %d\n",
  92. items[i].string, retval);
  93. return 1;
  94. }
  95. }
  96. /* 3: check for bad item */
  97. retval = pam_get_item (pamh, -1, &value);
  98. if (retval != PAM_BAD_ITEM)
  99. {
  100. fprintf (stderr,
  101. "pam_get_item returned %d when expecting PAM_BAD_ITEM\n",
  102. retval);
  103. return 1;
  104. }
  105. /* 4: check for valid item types, but NULL as value address. */
  106. for (i = 0; i < PAM_ARRAY_SIZE(items); i++)
  107. {
  108. retval = pam_get_item (pamh, items[i].type, NULL);
  109. if (retval != PAM_PERM_DENIED)
  110. {
  111. fprintf (stderr,
  112. "pam_get_item returned %d to get value for %s\n",
  113. retval, items[i].string);
  114. return 1;
  115. }
  116. }
  117. pam_end (pamh, 0);
  118. return 0;
  119. }