lib/kadm5/get_princs_s.c

   1 /*
   2  * Copyright (c) 1997, 1998, 1999 Kungliga Tekniska Högskolan
   3  * (Royal Institute of Technology, Stockholm, Sweden).
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  *
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  *
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  *
  17  * 3. Neither the name of the Institute nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 #include "kadm5_locl.h"
  35
  36 RCSID("$Id$");
  37
  38 struct foreach_data {
  39     const char *exp;
  40     char *exp2;
  41     char **princs;
  42     size_t nalloced;
  43     size_t count;
  44 };
  45
  46 static krb5_error_code
  47 add_princ(krb5_context context, struct foreach_data *d, char *princ)
  48 {
  49
  50     if (d->count == INT_MAX)
  51         return ERANGE;
  52     if (d->nalloced == d->count) {
  53         size_t n = d->nalloced + (d->nalloced >> 1) + 128; /* No O(N^2) pls */
  54         char **tmp;
  55
  56         if (SIZE_MAX / sizeof(*tmp) <= n)
  57             return ERANGE;
  58         if ((tmp = realloc(d->princs, n * sizeof(*tmp))) == NULL)
  59             return krb5_enomem(context);
  60         d->princs = tmp;
  61         d->nalloced = n;
  62     }
  63     d->princs[d->count++] = princ;
  64     return 0;
  65 }
  66
  67 static krb5_error_code
  68 foreach(krb5_context context, HDB *db, hdb_entry *ent, void *data)
  69 {
  70     struct foreach_data *d = data;
  71     char *princ;
  72     krb5_error_code ret;
  73     ret = krb5_unparse_name(context, ent->principal, &princ);
  74     if(ret)
  75         return ret;
  76     if(d->exp){
  77         if(fnmatch(d->exp, princ, 0) == 0 || fnmatch(d->exp2, princ, 0) == 0)
  78             ret = add_princ(context, d, princ);
  79         else
  80             free(princ);
  81     }else{
  82         ret = add_princ(context, d, princ);
  83     }
  84     if(ret)
  85         free(princ);
  86     return ret;
  87 }
  88
  89 kadm5_ret_t
  90 kadm5_s_get_principals(void *server_handle,
  91                        const char *expression,
  92                        char ***princs,
  93                        int *count)
  94 {
  95     struct foreach_data d;
  96     kadm5_server_context *context = server_handle;
  97     kadm5_ret_t ret = 0;
  98
  99     if (!context->keep_open) {
 100         ret = context->db->hdb_open(context->context, context->db, O_RDONLY, 0);
 101         if (ret) {
 102             krb5_warn(context->context, ret, "opening database");
 103             return ret;
 104         }
 105     }
 106     d.exp = expression;
 107     d.exp2 = NULL;
 108     if (expression) {
 109         krb5_realm r;
 110         int aret;
 111
 112         ret = krb5_get_default_realm(context->context, &r);
 113         if (ret == 0) {
 114             aret = asprintf(&d.exp2, "%s@%s", expression, r);
 115             free(r);
 116             if (aret == -1 || d.exp2 == NULL)
 117                 ret = krb5_enomem(context->context);
 118         }
 119     }
 120     d.princs = NULL;
 121     d.nalloced = 0;
 122     d.count = 0;
 123     if (ret == 0)
 124         ret = hdb_foreach(context->context, context->db, HDB_F_ADMIN_DATA,
 125                           foreach, &d);
 126
 127     if (ret == 0)
 128         ret = add_princ(context->context, &d, NULL);
 129     if (d.count >= INT_MAX)
 130         *count = INT_MAX;
 131     else
 132         *count = d.count - 1;
 133     if (ret == 0)
 134         *princs = d.princs;
 135     else
 136         kadm5_free_name_list(context, d.princs, count);
 137     free(d.exp2);
 138     if (!context->keep_open)
 139         context->db->hdb_close(context->context, context->db);
 140     return _kadm5_error_code(ret);
 141 }
 142
 143 struct foreach_online_data {
 144     const char *exp;
 145     char *exp2;
 146     int (*cb)(void *, const char *);
 147     void *cbdata;
 148 };
 149
 150 static krb5_error_code
 151 foreach_online(krb5_context context, HDB *db, hdb_entry *ent, void *data)
 152 {
 153     struct foreach_online_data *d = data;
 154     krb5_error_code ret;
 155     char *princ = NULL;
 156
 157     ret = krb5_unparse_name(context, ent->principal, &princ);
 158     if (ret == 0) {
 159         if (!d->exp ||
 160             fnmatch(d->exp, princ, 0) == 0 || fnmatch(d->exp2, princ, 0) == 0)
 161             ret = d->cb(d->cbdata, princ);
 162         free(princ);
 163     }
 164     return ret;
 165 }
 166
 167 kadm5_ret_t
 168 kadm5_s_iter_principals(void *server_handle,
 169                         const char *expression,
 170                         int (*cb)(void *, const char *),
 171                         void *cbdata)
 172 {
 173     struct foreach_online_data d;
 174     kadm5_server_context *context = server_handle;
 175     kadm5_ret_t ret = 0;
 176
 177     if (!context->keep_open) {
 178         ret = context->db->hdb_open(context->context, context->db, O_RDONLY, 0);
 179         if (ret) {
 180             krb5_warn(context->context, ret, "opening database");
 181             return ret;
 182         }
 183     }
 184     d.exp = expression;
 185     d.exp2 = NULL;
 186     d.cb = cb;
 187     d.cbdata = cbdata;
 188     if (expression) {
 189         krb5_realm r;
 190         int aret;
 191
 192         ret = krb5_get_default_realm(context->context, &r);
 193         if (ret == 0) {
 194             aret = asprintf(&d.exp2, "%s@%s", expression, r);
 195             free(r);
 196             if (aret == -1 || d.exp2 == NULL)
 197                 ret = krb5_enomem(context->context);
 198         }
 199     }
 200     if (ret == 0)
 201         ret = hdb_foreach(context->context, context->db, HDB_F_ADMIN_DATA,
 202                           foreach_online, &d);
 203     free(d.exp2);
 204     if (!context->keep_open)
 205         context->db->hdb_close(context->context, context->db);
 206     return _kadm5_error_code(ret);
 207 }