granicus.if.org Git - linux-pam/blob - libpam/pam_data.c

   1 /* pam_data.c */
   2
   3 /*
   4  * $Id$
   5  */
   6
   7 #include <stdlib.h>
   8 #include <string.h>
   9
  10 #include "pam_private.h"
  11
  12 static struct pam_data *_pam_locate_data(const pam_handle_t *pamh,
  13                                          const char *name)
  14 {
  15     struct pam_data *data;
  16
  17     D(("called"));
  18
  19     IF_NO_PAMH("_pam_locate_data", pamh, NULL);
  20
  21     data = pamh->data;
  22
  23     while (data) {
  24         if (!strcmp(data->name, name)) {
  25             return data;
  26         }
  27         data = data->next;
  28     }
  29
  30     return NULL;
  31 }
  32
  33 int pam_set_data(
  34     pam_handle_t *pamh,
  35     const char *module_data_name,
  36     void *data,
  37     void (*cleanup)(pam_handle_t *pamh, void *data, int error_status))
  38 {
  39     struct pam_data *data_entry;
  40
  41     D(("called"));
  42
  43     IF_NO_PAMH("pam_set_data", pamh, PAM_SYSTEM_ERR);
  44
  45     if (__PAM_FROM_APP(pamh)) {
  46         D(("called from application!?"));
  47         return PAM_SYSTEM_ERR;
  48     }
  49
  50     /* first check if there is some data already. If so clean it up */
  51
  52     if ((data_entry = _pam_locate_data(pamh, module_data_name))) {
  53         if (data_entry->cleanup) {
  54             data_entry->cleanup(pamh, data_entry->data,
  55                                 PAM_DATA_REPLACE | PAM_SUCCESS );
  56         }
  57     } else if ((data_entry = malloc(sizeof(*data_entry)))) {
  58         char *tname;
  59
  60         if ((tname = _pam_strdup(module_data_name)) == NULL) {
  61             _pam_system_log(LOG_CRIT, "pam_set_data: no memory for data name");
  62             _pam_drop(data_entry);
  63             return PAM_BUF_ERR;
  64         }
  65         data_entry->next = pamh->data;
  66         pamh->data = data_entry;
  67         data_entry->name = tname;
  68     } else {
  69         _pam_system_log(LOG_CRIT, "pam_set_data: cannot allocate data entry");
  70         return PAM_BUF_ERR;
  71     }
  72
  73     data_entry->data = data;           /* note this could be NULL */
  74     data_entry->cleanup = cleanup;
  75
  76     return PAM_SUCCESS;
  77 }
  78
  79 int pam_get_data(
  80     const pam_handle_t *pamh,
  81     const char *module_data_name,
  82     const void **datap)
  83 {
  84     struct pam_data *data;
  85
  86     D(("called"));
  87
  88     IF_NO_PAMH("pam_get_data", pamh, PAM_SYSTEM_ERR);
  89
  90     if (__PAM_FROM_APP(pamh)) {
  91         D(("called from application!?"));
  92         return PAM_SYSTEM_ERR;
  93     }
  94
  95     data = _pam_locate_data(pamh, module_data_name);
  96     if (data) {
  97         *datap = data->data;
  98         return PAM_SUCCESS;
  99     }
 100
 101     return PAM_NO_MODULE_DATA;
 102 }
 103
 104 void _pam_free_data(pam_handle_t *pamh, int status)
 105 {
 106     struct pam_data *last;
 107     struct pam_data *data;
 108
 109     D(("called"));
 110
 111     IF_NO_PAMH("_pam_free_data", pamh, /* no return value for void fn */);
 112     data = pamh->data;
 113
 114     while (data) {
 115         last = data;
 116         data = data->next;
 117         if (last->cleanup) {
 118             last->cleanup(pamh, last->data, status);
 119         }
 120         _pam_drop(last->name);
 121         _pam_drop(last);
 122     }
 123 }