src/lib/krb5/krb/preauth2.c

   1 /* -*- mode: c; c-basic-offset: 4; indent-tabs-mode: nil -*- */
   2 /*
   3  * Copyright 1995, 2003, 2008, 2012 by the Massachusetts Institute of Technology.  All
   4  * Rights Reserved.
   5  *
   6  * Export of this software from the United States of America may
   7  *   require a specific license from the United States Government.
   8  *   It is the responsibility of any person or organization contemplating
   9  *   export to obtain such a license before exporting.
  10  *
  11  * WITHIN THAT CONSTRAINT, permission to use, copy, modify, and
  12  * distribute this software and its documentation for any purpose and
  13  * without fee is hereby granted, provided that the above copyright
  14  * notice appear in all copies and that both that copyright notice and
  15  * this permission notice appear in supporting documentation, and that
  16  * the name of M.I.T. not be used in advertising or publicity pertaining
  17  * to distribution of the software without specific, written prior
  18  * permission.  Furthermore if you modify this software you must label
  19  * your software as modified software and not distribute it in such a
  20  * fashion that it might be confused with the original M.I.T. software.
  21  * M.I.T. makes no representations about the suitability of
  22  * this software for any purpose.  It is provided "as is" without express
  23  * or implied warranty.
  24  *
  25  */
  26
  27 /*
  28  * This file contains routines for establishing, verifying, and any other
  29  * necessary functions, for utilizing the pre-authentication field of the
  30  * kerberos kdc request, with various hardware/software verification devices.
  31  */
  32
  33 #include "k5-int.h"
  34 #include "osconf.h"
  35 #include <krb5/preauth_plugin.h>
  36 #include "int-proto.h"
  37 #include "fast.h"
  38
  39 #if !defined(_WIN32)
  40 #include <unistd.h>
  41 #endif
  42
  43 /* This structure lets us keep track of all of the modules which are loaded,
  44  * turning the list of modules and their lists of implemented preauth types
  45  * into a single list which we can walk easily. */
  46 struct krb5_preauth_context_st {
  47     int n_modules;
  48     struct krb5_preauth_context_module_st {
  49         /* Which of the possibly more than one preauth types which the
  50          * module supports we're using at this point in the list. */
  51         krb5_preauthtype pa_type;
  52         /* Encryption types which the client claims to support -- we
  53          * copy them directly into the krb5_kdc_req structure during
  54          * krb5_preauth_prepare_request(). */
  55         krb5_enctype *enctypes;
  56         /* The plugin's module data and a function to clear it. */
  57         krb5_clpreauth_moddata moddata;
  58         krb5_clpreauth_fini_fn client_fini;
  59         /* The module's table, and some of its members, copied here for
  60          * convenience when we populated the list. */
  61         const char *name;
  62         int flags, use_count;
  63         krb5_clpreauth_process_fn client_process;
  64         krb5_clpreauth_tryagain_fn client_tryagain;
  65         krb5_clpreauth_supply_gic_opts_fn client_supply_gic_opts;
  66         krb5_clpreauth_request_init_fn client_req_init;
  67         krb5_clpreauth_request_fini_fn client_req_fini;
  68         /* The per-request context which the client_req_init() function
  69          * might allocate, which we'll need to clean up later by
  70          * calling the client_req_fini() function. */
  71         krb5_clpreauth_modreq modreq;
  72         /* A pointer to the request_context pointer.  All modules within
  73          * a plugin will point at the request_context of the first
  74          * module within the plugin. */
  75         krb5_clpreauth_modreq *modreq_p;
  76     } *modules;
  77 };
  78
  79 typedef krb5_error_code (*pa_function)(krb5_context,
  80                                        krb5_kdc_req *request,
  81                                        krb5_pa_data *in_padata,
  82                                        krb5_pa_data **out_padata,
  83                                        krb5_data *salt, krb5_data *s2kparams,
  84                                        krb5_enctype *etype,
  85                                        krb5_keyblock *as_key,
  86                                        krb5_prompter_fct prompter_fct,
  87                                        void *prompter_data,
  88                                        krb5_gic_get_as_key_fct gak_fct,
  89                                        void *gak_data);
  90
  91 typedef struct _pa_types_t {
  92     krb5_preauthtype type;
  93     pa_function fct;
  94     int flags;
  95 } pa_types_t;
  96
  97 /* Create the per-krb5_context context. This means loading the modules
  98  * if we haven't done that yet (applications which never obtain initial
  99  * credentials should never hit this routine), breaking up the module's
 100  * list of support pa_types so that we can iterate over the modules more
 101  * easily, and copying over the relevant parts of the module's table. */
 102 void KRB5_CALLCONV
 103 krb5_init_preauth_context(krb5_context kcontext)
 104 {
 105     int n_tables, n_modules, i, count;
 106     krb5_plugin_initvt_fn *plugins = NULL, *pl;
 107     struct krb5_clpreauth_vtable_st *vtables = NULL, *vt;
 108     struct krb5_preauth_context_module_st *mod;
 109     krb5_preauth_context *context = NULL;
 110     krb5_clpreauth_moddata moddata;
 111     krb5_preauthtype pa_type, *pat;
 112     krb5_boolean first;
 113     krb5_clpreauth_modreq *rcpp;
 114
 115     /* Only do this once for each krb5_context */
 116     if (kcontext->preauth_context != NULL)
 117         return;
 118
 119     /* Auto-register built-in modules. */
 120     k5_plugin_register_dyn(kcontext, PLUGIN_INTERFACE_CLPREAUTH, "pkinit",
 121                            "preauth");
 122     k5_plugin_register(kcontext, PLUGIN_INTERFACE_CLPREAUTH,
 123                        "encrypted_challenge",
 124                        clpreauth_encrypted_challenge_initvt);
 125     k5_plugin_register(kcontext, PLUGIN_INTERFACE_CLPREAUTH,
 126                        "encrypted_timestamp",
 127                        clpreauth_encrypted_timestamp_initvt);
 128     k5_plugin_register(kcontext, PLUGIN_INTERFACE_CLPREAUTH, "sam2",
 129                        clpreauth_sam2_initvt);
 130
 131     /* Get all available clpreauth vtables. */
 132     if (k5_plugin_load_all(kcontext, PLUGIN_INTERFACE_CLPREAUTH, &plugins))
 133         return;
 134     for (count = 0; plugins[count] != NULL; count++);
 135     vtables = calloc(count, sizeof(*vtables));
 136     if (vtables == NULL)
 137         goto cleanup;
 138     for (pl = plugins, n_tables = 0; *pl != NULL; pl++) {
 139         if ((*pl)(kcontext, 1, 1, (krb5_plugin_vtable)&vtables[n_tables]) == 0)
 140             n_tables++;
 141     }
 142
 143     /* Count how many modules we ended up loading, and how many preauth
 144      * types we may claim to support as a result. */
 145     n_modules = 0;
 146     for (i = 0; i < n_tables; i++) {
 147         for (count = 0; vtables[i].pa_type_list[count] > 0; count++);
 148         n_modules += count;
 149     }
 150
 151     /* Allocate the space we need. */
 152     context = malloc(sizeof(*context));
 153     if (context == NULL)
 154         goto cleanup;
 155     context->modules = calloc(n_modules, sizeof(*context->modules));
 156     if (context->modules == NULL)
 157         goto cleanup;
 158
 159     /* fill in the structure */
 160     n_modules = 0;
 161     for (i = 0; i < n_tables; i++) {
 162         vt = &vtables[i];
 163         if ((vt->pa_type_list != NULL) && (vt->process != NULL)) {
 164             moddata = NULL;
 165             if (vt->init != NULL && vt->init(kcontext, &moddata) != 0) {
 166 #ifdef DEBUG
 167                 fprintf(stderr, "init err, skipping module \"%s\"\n",
 168                         vt->name);
 169 #endif
 170                 continue;
 171             }
 172
 173             rcpp = NULL;
 174             for (pat = vt->pa_type_list, first = TRUE; *pat > 0;
 175                  pat++, first = FALSE) {
 176                 pa_type = *pat;
 177                 mod = &context->modules[n_modules];
 178                 mod->pa_type = pa_type;
 179                 mod->enctypes = vt->enctype_list;
 180                 mod->moddata = moddata;
 181                 /* Only call client_fini once per plugin */
 182                 if (first)
 183                     mod->client_fini = vt->fini;
 184                 else
 185                     mod->client_fini = NULL;
 186                 mod->name = vt->name;
 187                 mod->flags = (*vt->flags)(kcontext, pa_type);
 188                 mod->use_count = 0;
 189                 mod->client_process = vt->process;
 190                 mod->client_tryagain = vt->tryagain;
 191                 mod->client_supply_gic_opts = first ? vt->gic_opts : NULL;
 192                 mod->modreq = NULL;
 193                 /*
 194                  * Only call request_init and request_fini once per plugin.
 195                  * Only the first module within each plugin will ever
 196                  * have request_context filled in.  Every module within
 197                  * the plugin will have its request_context_pp pointing
 198                  * to that entry's request_context.  That way all the
 199                  * modules within the plugin share the same request_context
 200                  */
 201                 if (first) {
 202                     mod->client_req_init = vt->request_init;
 203                     mod->client_req_fini = vt->request_fini;
 204                     rcpp = &mod->modreq;
 205                 } else {
 206                     mod->client_req_init = NULL;
 207                     mod->client_req_fini = NULL;
 208                 }
 209                 mod->modreq_p = rcpp;
 210 #ifdef DEBUG
 211                 fprintf(stderr, "init module \"%s\", pa_type %d, flag %d\n",
 212                         mod->name, mod->pa_type, mod->flags);
 213 #endif
 214                 n_modules++;
 215             }
 216         }
 217     }
 218     context->n_modules = n_modules;
 219
 220     /* Place the constructed preauth context into the krb5 context. */
 221     kcontext->preauth_context = context;
 222     context = NULL;
 223
 224 cleanup:
 225     if (context)
 226         free(context->modules);
 227     free(context);
 228     k5_plugin_free_modules(kcontext, plugins);
 229     free(vtables);
 230 }
 231
 232 /* Zero the use counts for the modules herein.  Usually used before we
 233  * start processing any data from the server, at which point every module
 234  * will again be able to take a crack at whatever the server sent. */
 235 void KRB5_CALLCONV
 236 krb5_clear_preauth_context_use_counts(krb5_context context)
 237 {
 238     int i;
 239     if (context->preauth_context != NULL) {
 240         for (i = 0; i < context->preauth_context->n_modules; i++) {
 241             context->preauth_context->modules[i].use_count = 0;
 242         }
 243     }
 244 }
 245
 246
 247 /* Free the per-krb5_context preauth_context. This means clearing any
 248  * plugin-specific context which may have been created, and then
 249  * freeing the context itself. */
 250 void KRB5_CALLCONV
 251 krb5_free_preauth_context(krb5_context context)
 252 {
 253     int i;
 254     struct krb5_preauth_context_module_st *mod;
 255
 256     if (context == NULL || context->preauth_context == NULL)
 257         return;
 258     for (i = 0; i < context->preauth_context->n_modules; i++) {
 259         mod = &context->preauth_context->modules[i];
 260         if (mod->client_fini != NULL)
 261             mod->client_fini(context, mod->moddata);
 262         zap(mod, sizeof(*mod));
 263     }
 264     free(context->preauth_context->modules);
 265     free(context->preauth_context);
 266     context->preauth_context = NULL;
 267 }
 268
 269 /* Initialize the per-AS-REQ context. This means calling the client_req_init
 270  * function to give the plugin a chance to allocate a per-request context. */
 271 void KRB5_CALLCONV
 272 krb5_preauth_request_context_init(krb5_context context)
 273 {
 274     int i;
 275     struct krb5_preauth_context_module_st *mod;
 276
 277     if (context->preauth_context == NULL)
 278         krb5_init_preauth_context(context);
 279     if (context->preauth_context == NULL)
 280         return;
 281     for (i = 0; i < context->preauth_context->n_modules; i++) {
 282         context->preauth_context->modules[i].use_count = 0;
 283         mod = &context->preauth_context->modules[i];
 284         if (mod->client_req_init != NULL)
 285             mod->client_req_init(context, mod->moddata, mod->modreq_p);
 286     }
 287 }
 288
 289 /* Free the per-AS-REQ context. This means clearing any request-specific
 290  * context which the plugin may have created. */
 291 void KRB5_CALLCONV
 292 krb5_preauth_request_context_fini(krb5_context context)
 293 {
 294     int i;
 295     struct krb5_preauth_context_module_st *mod;
 296
 297     if (context->preauth_context == NULL)
 298         return;
 299     for (i = 0; i < context->preauth_context->n_modules; i++) {
 300         mod = &context->preauth_context->modules[i];
 301         if (mod->modreq != NULL) {
 302             if (mod->client_req_fini != NULL)
 303                 mod->client_req_fini(context, mod->moddata, mod->modreq);
 304             mod->modreq = NULL;
 305         }
 306     }
 307 }
 308
 309 /* Add the named encryption type to the existing list of ktypes. */
 310 static void
 311 grow_ktypes(krb5_enctype **out_ktypes, int *out_nktypes, krb5_enctype ktype)
 312 {
 313     int i;
 314     krb5_enctype *ktypes;
 315     for (i = 0; i < *out_nktypes; i++) {
 316         if ((*out_ktypes)[i] == ktype)
 317             return;
 318     }
 319     ktypes = malloc((*out_nktypes + 2) * sizeof(ktype));
 320     if (ktypes) {
 321         for (i = 0; i < *out_nktypes; i++)
 322             ktypes[i] = (*out_ktypes)[i];
 323         ktypes[i++] = ktype;
 324         ktypes[i] = 0;
 325         free(*out_ktypes);
 326         *out_ktypes = ktypes;
 327         *out_nktypes = i;
 328     }
 329 }
 330
 331 /*
 332  * Add the given list of pa_data items to the existing list of items.
 333  * Factored out here to make reading the do_preauth logic easier to read.
 334  */
 335 static int
 336 grow_pa_list(krb5_pa_data ***out_pa_list, int *out_pa_list_size,
 337              krb5_pa_data **addition, int num_addition)
 338 {
 339     krb5_pa_data **pa_list;
 340     int i, j;
 341
 342     if (out_pa_list == NULL || addition == NULL) {
 343         return EINVAL;
 344     }
 345
 346     if (*out_pa_list == NULL) {
 347         /* Allocate room for the new additions and a NULL terminator. */
 348         pa_list = malloc((num_addition + 1) * sizeof(krb5_pa_data *));
 349         if (pa_list == NULL)
 350             return ENOMEM;
 351         for (i = 0; i < num_addition; i++)
 352             pa_list[i] = addition[i];
 353         pa_list[i] = NULL;
 354         *out_pa_list = pa_list;
 355         *out_pa_list_size = num_addition;
 356     } else {
 357         /*
 358          * Allocate room for the existing entries plus
 359          * the new additions and a NULL terminator.
 360          */
 361         pa_list = malloc((*out_pa_list_size + num_addition + 1)
 362                          * sizeof(krb5_pa_data *));
 363         if (pa_list == NULL)
 364             return ENOMEM;
 365         for (i = 0; i < *out_pa_list_size; i++)
 366             pa_list[i] = (*out_pa_list)[i];
 367         for (j = 0; j < num_addition;)
 368             pa_list[i++] = addition[j++];
 369         pa_list[i] = NULL;
 370         free(*out_pa_list);
 371         *out_pa_list = pa_list;
 372         *out_pa_list_size = i;
 373     }
 374     return 0;
 375 }
 376
 377 static krb5_enctype
 378 get_etype(krb5_context context, krb5_clpreauth_rock rock)
 379 {
 380     return *rock->etype;
 381 }
 382
 383 static krb5_keyblock *
 384 fast_armor(krb5_context context, krb5_clpreauth_rock rock)
 385 {
 386     return rock->fast_state->armor_key;
 387 }
 388
 389 static krb5_error_code
 390 get_as_key(krb5_context context, krb5_clpreauth_rock rock,
 391            krb5_keyblock **keyblock)
 392 {
 393     krb5_error_code ret;
 394     krb5_data *salt;
 395
 396     if (rock->as_key->length == 0) {
 397         salt = (*rock->default_salt) ? NULL : rock->salt;
 398         ret = (*rock->gak_fct)(context, rock->client, *rock->etype,
 399                                rock->prompter, rock->prompter_data, salt,
 400                                rock->s2kparams, rock->as_key, *rock->gak_data);
 401         if (ret)
 402             return ret;
 403     }
 404     *keyblock = rock->as_key;
 405     return 0;
 406 }
 407
 408 static krb5_error_code
 409 set_as_key(krb5_context context, krb5_clpreauth_rock rock,
 410            const krb5_keyblock *keyblock)
 411 {
 412     krb5_free_keyblock_contents(context, rock->as_key);
 413     return krb5_copy_keyblock_contents(context, keyblock, rock->as_key);
 414 }
 415
 416 static krb5_error_code
 417 get_preauth_time(krb5_context context, krb5_clpreauth_rock rock,
 418                  krb5_boolean allow_unauth_time, krb5_timestamp *time_out,
 419                  krb5_int32 *usec_out)
 420 {
 421     if (rock->pa_offset_state != NO_OFFSET &&
 422         (allow_unauth_time || rock->pa_offset_state == AUTH_OFFSET) &&
 423         (context->library_options & KRB5_LIBOPT_SYNC_KDCTIME)) {
 424         /* Use the offset we got from the preauth-required error. */
 425         return k5_time_with_offset(rock->pa_offset, rock->pa_offset_usec,
 426                                    time_out, usec_out);
 427
 428     } else {
 429         /* Use the time offset from the context, or no offset. */
 430         return krb5_us_timeofday(context, time_out, usec_out);
 431     }
 432 }
 433
 434 static struct krb5_clpreauth_callbacks_st callbacks = {
 435     2,
 436     get_etype,
 437     fast_armor,
 438     get_as_key,
 439     set_as_key,
 440     get_preauth_time
 441 };
 442
 443 /* Tweak the request body, for now adding any enctypes which the module claims
 444  * to add support for to the list, but in the future perhaps doing more
 445  * involved things. */
 446 void KRB5_CALLCONV
 447 krb5_preauth_prepare_request(krb5_context kcontext,
 448                              krb5_gic_opt_ext *opte,
 449                              krb5_kdc_req *request)
 450 {
 451     int i, j;
 452
 453     if (kcontext->preauth_context == NULL) {
 454         return;
 455     }
 456     /* Add the module-specific enctype list to the request, but only if
 457      * it's something we can safely modify. */
 458     if (!(opte && (opte->flags & KRB5_GET_INIT_CREDS_OPT_ETYPE_LIST))) {
 459         for (i = 0; i < kcontext->preauth_context->n_modules; i++) {
 460             if (kcontext->preauth_context->modules[i].enctypes == NULL)
 461                 continue;
 462             for (j = 0; kcontext->preauth_context->modules[i].enctypes[j] != 0; j++) {
 463                 grow_ktypes(&request->ktype, &request->nktypes,
 464                             kcontext->preauth_context->modules[i].enctypes[j]);
 465             }
 466         }
 467     }
 468 }
 469
 470 /* Find the first module which provides for the named preauth type which also
 471  * hasn't had a chance to run yet (INFO modules don't count, because as a rule
 472  * they don't generate preauth data), and run it. */
 473 static krb5_error_code
 474 run_preauth_plugins(krb5_context kcontext,
 475                     int module_required_flags,
 476                     krb5_kdc_req *request,
 477                     krb5_data *encoded_request_body,
 478                     krb5_data *encoded_previous_request,
 479                     krb5_pa_data *in_padata,
 480                     krb5_prompter_fct prompter,
 481                     void *prompter_data,
 482                     krb5_clpreauth_rock preauth_rock,
 483                     krb5_pa_data ***out_pa_list,
 484                     int *out_pa_list_size,
 485                     int *module_ret,
 486                     int *module_flags,
 487                     krb5_gic_opt_ext *opte)
 488 {
 489     int i;
 490     krb5_pa_data **out_pa_data;
 491     krb5_error_code ret;
 492     struct krb5_preauth_context_module_st *module;
 493
 494     if (kcontext->preauth_context == NULL) {
 495         return ENOENT;
 496     }
 497     /* iterate over all loaded modules */
 498     for (i = 0; i < kcontext->preauth_context->n_modules; i++) {
 499         module = &kcontext->preauth_context->modules[i];
 500         /* skip over those which don't match the preauth type */
 501         if (module->pa_type != in_padata->pa_type)
 502             continue;
 503         /* skip over those which don't match the flags (INFO vs REAL, mainly) */
 504         if ((module->flags & module_required_flags) == 0)
 505             continue;
 506         /* if it's a REAL module, try to call it only once per library call */
 507         if (module_required_flags & PA_REAL) {
 508             if (module->use_count > 0) {
 509                 TRACE_PREAUTH_SKIP(kcontext, module->name, module->pa_type);
 510                 continue;
 511             }
 512             module->use_count++;
 513         }
 514         /* run the module's callback function */
 515         out_pa_data = NULL;
 516 #ifdef DEBUG
 517         fprintf(stderr, "using module \"%s\" (%d), flags = %d\n",
 518                 module->name, module->pa_type, module->flags);
 519 #endif
 520         ret = module->client_process(kcontext, module->moddata,
 521                                      *module->modreq_p,
 522                                      (krb5_get_init_creds_opt *)opte,
 523                                      &callbacks, preauth_rock,
 524                                      request, encoded_request_body,
 525                                      encoded_previous_request, in_padata,
 526                                      prompter, prompter_data, &out_pa_data);
 527         TRACE_PREAUTH_PROCESS(kcontext, module->name, module->pa_type,
 528                               module->flags, ret);
 529         /* Make note of the module's flags and status. */
 530         *module_flags = module->flags;
 531         *module_ret = ret;
 532         /* Save the new preauth data item. */
 533         if (out_pa_data != NULL) {
 534             int j;
 535             for (j = 0; out_pa_data[j] != NULL; j++);
 536             ret = grow_pa_list(out_pa_list, out_pa_list_size, out_pa_data, j);
 537             free(out_pa_data);
 538             if (ret != 0)
 539                 return ret;
 540         }
 541         break;
 542     }
 543     if (i >= kcontext->preauth_context->n_modules) {
 544         return ENOENT;
 545     }
 546     return 0;
 547 }
 548
 549 static inline krb5_data
 550 padata2data(krb5_pa_data p)
 551 {
 552     krb5_data d;
 553     d.magic = KV5M_DATA;
 554     d.length = p.length;
 555     d.data = (char *) p.contents;
 556     return d;
 557 }
 558
 559 /* Set etype info parameters in rock based on padata. */
 560 static krb5_error_code
 561 get_etype_info(krb5_context context, krb5_pa_data **padata,
 562                krb5_kdc_req *request, krb5_clpreauth_rock rock)
 563 {
 564     krb5_error_code ret = 0;
 565     krb5_pa_data *pa;
 566     krb5_data d;
 567     krb5_etype_info etype_info = NULL, e;
 568     krb5_etype_info_entry *entry;
 569     krb5_boolean valid_found;
 570     const char *p;
 571     int i;
 572
 573     /* Find an etype-info2 or etype-info element in padata. */
 574     pa = krb5int_find_pa_data(context, padata, KRB5_PADATA_ETYPE_INFO2);
 575     if (pa != NULL) {
 576         d = padata2data(*pa);
 577         ret = decode_krb5_etype_info2(&d, &etype_info);
 578     } else {
 579         pa = krb5int_find_pa_data(context, padata, KRB5_PADATA_ETYPE_INFO2);
 580         if (pa != NULL) {
 581             d = padata2data(*pa);
 582             ret = decode_krb5_etype_info2(&d, &etype_info);
 583         }
 584     }
 585
 586     if (etype_info != NULL) {
 587         /* Search entries in order of the requests's enctype preference. */
 588         entry = NULL;
 589         valid_found = FALSE;
 590         for (i = 0; i < request->nktypes && entry == NULL; i++) {
 591             for (e = etype_info; *e != NULL && entry == NULL; e++) {
 592                 if ((*e)->etype == request->ktype[i])
 593                     entry = *e;
 594                 if (krb5_c_valid_enctype((*e)->etype))
 595                     valid_found = TRUE;
 596             }
 597         }
 598         if (entry == NULL) {
 599             ret = (valid_found) ? KRB5_CONFIG_ETYPE_NOSUPP :
 600                 KRB5_PROG_ETYPE_NOSUPP;
 601             goto cleanup;
 602         }
 603
 604         /* Set rock fields based on the entry we selected. */
 605         *rock->etype = entry->etype;
 606         krb5_free_data_contents(context, rock->salt);
 607         if (entry->length != KRB5_ETYPE_NO_SALT) {
 608             *rock->salt = make_data(entry->salt, entry->length);
 609             entry->salt = NULL;
 610             *rock->default_salt = FALSE;
 611         } else {
 612             *rock->salt = empty_data();
 613             *rock->default_salt = TRUE;
 614         }
 615         krb5_free_data_contents(context, rock->s2kparams);
 616         *rock->s2kparams = entry->s2kparams;
 617         entry->s2kparams = empty_data();
 618         TRACE_PREAUTH_ETYPE_INFO(context, *rock->etype, rock->salt,
 619                                  rock->s2kparams);
 620     } else {
 621         /* Look for a pw-salt or afs3-salt element. */
 622         pa = krb5int_find_pa_data(context, padata, KRB5_PADATA_PW_SALT);
 623         if (pa == NULL)
 624             pa = krb5int_find_pa_data(context, padata, KRB5_PADATA_AFS3_SALT);
 625         if (pa != NULL) {
 626             /* Set rock->salt based on the element we found. */
 627             krb5_free_data_contents(context, rock->salt);
 628             d = padata2data(*pa);
 629             ret = krb5int_copy_data_contents(context, &d, rock->salt);
 630             if (ret)
 631                 goto cleanup;
 632             if (pa->pa_type == KRB5_PADATA_AFS3_SALT) {
 633                 /* Work around a (possible) old Heimdal KDC foible. */
 634                 p = memchr(rock->salt->data, '@', rock->salt->length);
 635                 if (p != NULL)
 636                     rock->salt->length = p - rock->salt->data;
 637                 /* Tolerate extra null in MIT KDC afs3-salt value. */
 638                 if (rock->salt->length > 0 &&
 639                     rock->salt->data[rock->salt->length - 1] == '\0')
 640                     rock->salt->length--;
 641                 /* Set an s2kparams value to indicate AFS string-to-key. */
 642                 krb5_free_data_contents(context, rock->s2kparams);
 643                 ret = alloc_data(rock->s2kparams, 1);
 644                 if (ret)
 645                     goto cleanup;
 646                 rock->s2kparams->data[0] = '\1';
 647             }
 648             *rock->default_salt = FALSE;
 649             TRACE_PREAUTH_SALT(context, rock->salt, pa->pa_type);
 650         }
 651     }
 652
 653 cleanup:
 654     krb5_free_etype_info(context, etype_info);
 655     return ret;
 656 }
 657
 658 static krb5_error_code
 659 pa_fx_cookie(krb5_context context, krb5_kdc_req *request,
 660              krb5_pa_data *in_padata, krb5_pa_data **out_padata,
 661              krb5_data *salt, krb5_data *s2kparams, krb5_enctype *etype,
 662              krb5_keyblock *as_key, krb5_prompter_fct prompter,
 663              void *prompter_data, krb5_gic_get_as_key_fct gak_fct,
 664              void *gak_data)
 665 {
 666     krb5_pa_data *pa = calloc(1, sizeof(krb5_pa_data));
 667     krb5_octet *contents;
 668
 669     TRACE_PREAUTH_COOKIE(context, in_padata->length, in_padata->contents);
 670     if (pa == NULL)
 671         return ENOMEM;
 672     contents = malloc(in_padata->length);
 673     if (contents == NULL) {
 674         free(pa);
 675         return ENOMEM;
 676     }
 677     *pa = *in_padata;
 678     pa->contents = contents;
 679     memcpy(contents, in_padata->contents, pa->length);
 680     *out_padata = pa;
 681     return 0;
 682 }
 683
 684 static krb5_error_code
 685 pa_s4u_x509_user(krb5_context context, krb5_kdc_req *request,
 686                  krb5_pa_data *in_padata, krb5_pa_data **out_padata,
 687                  krb5_data *salt, krb5_data *s2kparams, krb5_enctype *etype,
 688                  krb5_keyblock *as_key, krb5_prompter_fct prompter,
 689                  void *prompter_data, krb5_gic_get_as_key_fct gak_fct,
 690                  void *gak_data)
 691 {
 692     krb5_s4u_userid *userid = (krb5_s4u_userid *)gak_data; /* XXX private contract */
 693     krb5_pa_data *s4u_padata;
 694     krb5_error_code code;
 695     krb5_principal client;
 696
 697     *out_padata = NULL;
 698
 699     if (userid == NULL)
 700         return EINVAL;
 701
 702     code = krb5_copy_principal(context, request->client, &client);
 703     if (code != 0)
 704         return code;
 705
 706     if (userid->user != NULL)
 707         krb5_free_principal(context, userid->user);
 708     userid->user = client;
 709
 710     if (userid->subject_cert.length != 0) {
 711         s4u_padata = malloc(sizeof(*s4u_padata));
 712         if (s4u_padata == NULL)
 713             return ENOMEM;
 714
 715         s4u_padata->magic = KV5M_PA_DATA;
 716         s4u_padata->pa_type = KRB5_PADATA_S4U_X509_USER;
 717         s4u_padata->contents = malloc(userid->subject_cert.length);
 718         if (s4u_padata->contents == NULL) {
 719             free(s4u_padata);
 720             return ENOMEM;
 721         }
 722         memcpy(s4u_padata->contents, userid->subject_cert.data, userid->subject_cert.length);
 723         s4u_padata->length = userid->subject_cert.length;
 724
 725         *out_padata = s4u_padata;
 726     }
 727
 728     return 0;
 729 }
 730
 731 /* FIXME - order significant? */
 732 static const pa_types_t pa_types[] = {
 733     {
 734         KRB5_PADATA_FX_COOKIE,
 735         pa_fx_cookie,
 736         PA_INFO,
 737     },
 738     {
 739         KRB5_PADATA_S4U_X509_USER,
 740         pa_s4u_x509_user,
 741         PA_INFO,
 742     },
 743     {
 744         -1,
 745         NULL,
 746         0,
 747     },
 748 };
 749
 750 /*
 751  * If one of the modules can adjust its AS_REQ data using the contents of the
 752  * err_reply, return 0.  If it's the sort of correction which requires that we
 753  * ask the user another question, we let the calling application deal with it.
 754  */
 755 krb5_error_code KRB5_CALLCONV
 756 krb5_do_preauth_tryagain(krb5_context kcontext,
 757                          krb5_kdc_req *request,
 758                          krb5_data *encoded_request_body,
 759                          krb5_data *encoded_previous_request,
 760                          krb5_pa_data **padata,
 761                          krb5_pa_data ***return_padata,
 762                          krb5_error *err_reply,
 763                          krb5_pa_data **err_padata,
 764                          krb5_prompter_fct prompter, void *prompter_data,
 765                          krb5_clpreauth_rock preauth_rock,
 766                          krb5_gic_opt_ext *opte)
 767 {
 768     krb5_error_code ret;
 769     krb5_pa_data **out_padata;
 770     krb5_preauth_context *context;
 771     struct krb5_preauth_context_module_st *module;
 772     int i, j;
 773     int out_pa_list_size = 0;
 774
 775     ret = KRB5KRB_ERR_GENERIC;
 776     if (kcontext->preauth_context == NULL) {
 777         return KRB5KRB_ERR_GENERIC;
 778     }
 779     context = kcontext->preauth_context;
 780     if (context == NULL) {
 781         return KRB5KRB_ERR_GENERIC;
 782     }
 783
 784     TRACE_PREAUTH_TRYAGAIN_INPUT(kcontext, padata);
 785
 786     for (i = 0; padata[i] != NULL && padata[i]->pa_type != 0; i++) {
 787         out_padata = NULL;
 788         for (j = 0; j < context->n_modules; j++) {
 789             module = &context->modules[j];
 790             if (module->pa_type != padata[i]->pa_type) {
 791                 continue;
 792             }
 793             if (module->client_tryagain == NULL) {
 794                 continue;
 795             }
 796             if ((*module->client_tryagain)(kcontext, module->moddata,
 797                                            *module->modreq_p,
 798                                            (krb5_get_init_creds_opt *)opte,
 799                                            &callbacks, preauth_rock,
 800                                            request,
 801                                            encoded_request_body,
 802                                            encoded_previous_request,
 803                                            padata[i]->pa_type,
 804                                            err_reply, err_padata,
 805                                            prompter, prompter_data,
 806                                            &out_padata) == 0) {
 807                 if (out_padata != NULL) {
 808                     int k;
 809                     for (k = 0; out_padata[k] != NULL; k++);
 810                     grow_pa_list(return_padata, &out_pa_list_size,
 811                                  out_padata, k);
 812                     free(out_padata);
 813                     TRACE_PREAUTH_TRYAGAIN_OUTPUT(kcontext, *return_padata);
 814                     return 0;
 815                 }
 816             }
 817         }
 818     }
 819     return ret;
 820 }
 821
 822 krb5_error_code KRB5_CALLCONV
 823 krb5_do_preauth(krb5_context context, krb5_kdc_req *request,
 824                 krb5_data *encoded_request_body,
 825                 krb5_data *encoded_previous_request,
 826                 krb5_pa_data **in_padata, krb5_pa_data ***out_padata,
 827                 krb5_prompter_fct prompter, void *prompter_data,
 828                 krb5_clpreauth_rock rock, krb5_gic_opt_ext *opte,
 829                 krb5_boolean *got_real_out)
 830 {
 831     unsigned int h;
 832     int i, j, out_pa_list_size;
 833     krb5_pa_data *out_pa = NULL, **out_pa_list = NULL;
 834     krb5_error_code ret;
 835     static const int paorder[] = { PA_INFO, PA_REAL };
 836     int realdone;
 837
 838     *got_real_out = FALSE;
 839
 840     if (in_padata == NULL) {
 841         *out_padata = NULL;
 842         return(0);
 843     }
 844
 845     TRACE_PREAUTH_INPUT(context, in_padata);
 846
 847     /* Scan the padata list and process etype-info or salt elements. */
 848     ret = get_etype_info(context, in_padata, request, rock);
 849     if (ret)
 850         return ret;
 851
 852     out_pa_list = NULL;
 853     out_pa_list_size = 0;
 854
 855     /* first do all the informational preauths, then the first real one */
 856
 857     for (h=0; h<(sizeof(paorder)/sizeof(paorder[0])); h++) {
 858         realdone = 0;
 859         for (i=0; in_padata[i] && !realdone; i++) {
 860             /* Try the internally-provided preauth type list. */
 861             if (!realdone) for (j=0; pa_types[j].type >= 0; j++) {
 862                     if ((in_padata[i]->pa_type == pa_types[j].type) &&
 863                         (pa_types[j].flags & paorder[h])) {
 864 #ifdef DEBUG
 865                         fprintf (stderr, "calling internal function for pa_type "
 866                                  "%d, flag %d\n", pa_types[j].type, paorder[h]);
 867 #endif
 868                         out_pa = NULL;
 869
 870                         ret = pa_types[j].fct(context, request, in_padata[i],
 871                                               &out_pa, rock->salt,
 872                                               rock->s2kparams, rock->etype,
 873                                               rock->as_key, prompter,
 874                                               prompter_data, *rock->gak_fct,
 875                                               *rock->gak_data);
 876                         if (ret) {
 877                             if (paorder[h] == PA_INFO) {
 878                                 TRACE_PREAUTH_INFO_FAIL(context,
 879                                                         in_padata[i]->pa_type,
 880                                                         ret);
 881                                 ret = 0;
 882                                 continue; /* PA_INFO type failed, ignore */
 883                             }
 884
 885                             goto cleanup;
 886                         }
 887
 888                         ret = grow_pa_list(&out_pa_list, &out_pa_list_size,
 889                                            &out_pa, 1);
 890                         if (ret != 0) {
 891                             goto cleanup;
 892                         }
 893                         if (paorder[h] == PA_REAL)
 894                             realdone = 1;
 895                     }
 896                 }
 897
 898             /* Try to use plugins now. */
 899             if (!realdone) {
 900                 krb5_init_preauth_context(context);
 901                 if (context->preauth_context != NULL) {
 902                     int module_ret = 0, module_flags;
 903 #ifdef DEBUG
 904                     fprintf (stderr, "trying modules for pa_type %d, flag %d\n",
 905                              in_padata[i]->pa_type, paorder[h]);
 906 #endif
 907                     ret = run_preauth_plugins(context,
 908                                               paorder[h],
 909                                               request,
 910                                               encoded_request_body,
 911                                               encoded_previous_request,
 912                                               in_padata[i],
 913                                               prompter,
 914                                               prompter_data,
 915                                               rock,
 916                                               &out_pa_list,
 917                                               &out_pa_list_size,
 918                                               &module_ret,
 919                                               &module_flags,
 920                                               opte);
 921                     if (ret == 0) {
 922                         if (module_ret == 0) {
 923                             if (paorder[h] == PA_REAL) {
 924                                 realdone = 1;
 925                             }
 926                         }
 927                     }
 928                 }
 929             }
 930         }
 931     }
 932
 933     TRACE_PREAUTH_OUTPUT(context, out_pa_list);
 934     *out_padata = out_pa_list;
 935
 936     *got_real_out = realdone;
 937     return(0);
 938 cleanup:
 939     if (out_pa_list) {
 940         out_pa_list[out_pa_list_size++] = NULL;
 941         krb5_free_pa_data(context, out_pa_list);
 942     }
 943     return (ret);
 944 }
 945
 946 /*
 947  * Give all the preauth plugins a look at the preauth option which
 948  * has just been set
 949  */
 950 krb5_error_code
 951 krb5_preauth_supply_preauth_data(krb5_context context, krb5_gic_opt_ext *opte,
 952                                  const char *attr, const char *value)
 953 {
 954     krb5_error_code retval = 0;
 955     int i;
 956     struct krb5_preauth_context_module_st *mod;
 957     const char *emsg = NULL;
 958
 959     if (context->preauth_context == NULL)
 960         krb5_init_preauth_context(context);
 961     if (context->preauth_context == NULL) {
 962         retval = EINVAL;
 963         krb5_set_error_message(context, retval,
 964                                _("Unable to initialize preauth context"));
 965         return retval;
 966     }
 967
 968     /*
 969      * Go down the list of preauth modules, and supply them with the
 970      * attribute/value pair.
 971      */
 972     for (i = 0; i < context->preauth_context->n_modules; i++) {
 973         mod = &context->preauth_context->modules[i];
 974         if (mod->client_supply_gic_opts == NULL)
 975             continue;
 976         retval = mod->client_supply_gic_opts(context, mod->moddata,
 977                                              (krb5_get_init_creds_opt *)opte,
 978                                              attr, value);
 979         if (retval) {
 980             emsg = krb5_get_error_message(context, retval);
 981             krb5_set_error_message(context, retval, _("Preauth plugin %s: %s"),
 982                                    mod->name, emsg);
 983             krb5_free_error_message(context, emsg);
 984             break;
 985         }
 986     }
 987     return retval;
 988 }