lustre/utils/gss/lgss_krb5_utils.c

   1 /*
   2  * Modifications for Lustre
   3  *
   4  * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
   5  *
   6  * Author: Eric Mei <ericm@clusterfs.com>
   7  */
   8
   9 /*
  10  *  Adapted in part from MIT Kerberos 5-1.2.1 slave/kprop.c and from
  11  *  http://docs.sun.com/?p=/doc/816-1331/6m7oo9sms&a=view
  12  *
  13  *  Copyright (c) 2002-2004 The Regents of the University of Michigan.
  14  *  All rights reserved.
  15  *
  16  *  Andy Adamson <andros@umich.edu>
  17  *  J. Bruce Fields <bfields@umich.edu>
  18  *  Marius Aamodt Eriksen <marius@umich.edu>
  19  *  Kevin Coffman <kwc@umich.edu>
  20  */
  21
  22 /*
  23  * slave/kprop.c
  24  *
  25  * Copyright 1990,1991 by the Massachusetts Institute of Technology.
  26  * All Rights Reserved.
  27  *
  28  * Export of this software from the United States of America may
  29  *   require a specific license from the United States Government.
  30  *   It is the responsibility of any person or organization contemplating
  31  *   export to obtain such a license before exporting.
  32  *
  33  * WITHIN THAT CONSTRAINT, permission to use, copy, modify, and
  34  * distribute this software and its documentation for any purpose and
  35  * without fee is hereby granted, provided that the above copyright
  36  * notice appear in all copies and that both that copyright notice and
  37  * this permission notice appear in supporting documentation, and that
  38  * the name of M.I.T. not be used in advertising or publicity pertaining
  39  * to distribution of the software without specific, written prior
  40  * permission.  Furthermore if you modify this software you must label
  41  * your software as modified software and not distribute it in such a
  42  * fashion that it might be confused with the original M.I.T. software.
  43  * M.I.T. makes no representations about the suitability of
  44  * this software for any purpose.  It is provided "as is" without express
  45  * or implied warranty.
  46  */
  47
  48 /*
  49  * Copyright 1994 by OpenVision Technologies, Inc.
  50  *
  51  * Permission to use, copy, modify, distribute, and sell this software
  52  * and its documentation for any purpose is hereby granted without fee,
  53  * provided that the above copyright notice appears in all copies and
  54  * that both that copyright notice and this permission notice appear in
  55  * supporting documentation, and that the name of OpenVision not be used
  56  * in advertising or publicity pertaining to distribution of the software
  57  * without specific, written prior permission. OpenVision makes no
  58  * representations about the suitability of this software for any
  59  * purpose.  It is provided "as is" without express or implied warranty.
  60  *
  61  * OPENVISION DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE,
  62  * INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO
  63  * EVENT SHALL OPENVISION BE LIABLE FOR ANY SPECIAL, INDIRECT OR
  64  * CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF
  65  * USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
  66  * OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
  67  * PERFORMANCE OF THIS SOFTWARE.
  68  */
  69 /*
  70   krb5_util.c
  71
  72   Copyright (c) 2004 The Regents of the University of Michigan.
  73   All rights reserved.
  74
  75   Redistribution and use in source and binary forms, with or without
  76   modification, are permitted provided that the following conditions
  77   are met:
  78
  79   1. Redistributions of source code must retain the above copyright
  80      notice, this list of conditions and the following disclaimer.
  81   2. Redistributions in binary form must reproduce the above copyright
  82      notice, this list of conditions and the following disclaimer in the
  83      documentation and/or other materials provided with the distribution.
  84   3. Neither the name of the University nor the names of its
  85      contributors may be used to endorse or promote products derived
  86      from this software without specific prior written permission.
  87
  88   THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED
  89   WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  90   MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  91   DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  92   FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  93   CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  94   SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
  95   BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  96   LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  97   NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  98   SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  99
 100 */
 101
 102 #ifndef _GNU_SOURCE
 103 #define _GNU_SOURCE
 104 #endif
 105 #include "config.h"
 106 #include <sys/param.h>
 107 //#include <rpc/rpc.h>
 108 #include <sys/types.h>
 109 #include <sys/stat.h>
 110 #include <sys/utsname.h>
 111 #include <sys/socket.h>
 112 #include <arpa/inet.h>
 113
 114 #include <unistd.h>
 115 #include <stdio.h>
 116 #include <stdlib.h>
 117 #include <string.h>
 118 #ifdef HAVE_NETDB_H
 119 # include <netdb.h>
 120 #endif
 121 #include <dirent.h>
 122 #include <fcntl.h>
 123 #include <errno.h>
 124 #include <time.h>
 125 #include <gssapi/gssapi.h>
 126 #ifdef USE_PRIVATE_KRB5_FUNCTIONS
 127 #include <gssapi/gssapi_krb5.h>
 128 #endif
 129 #include <krb5.h>
 130
 131 #include "lsupport.h"
 132 #include "lgss_utils.h"
 133 #include "lgss_krb5_utils.h"
 134
 135 static void lgss_krb5_mutex_lock(void)
 136 {
 137         if (lgss_mutex_lock(LGSS_MUTEX_KRB5)) {
 138                 logmsg(LL_ERR, "can't lock process, abort!\n");
 139                 exit(-1);
 140         }
 141 }
 142
 143 static void lgss_krb5_mutex_unlock(void)
 144 {
 145         if (lgss_mutex_unlock(LGSS_MUTEX_KRB5)) {
 146                 logmsg(LL_WARN, "can't unlock process, other processes "
 147                        "might need to wait long time\n");
 148         }
 149 }
 150
 151 /*
 152  * NOTE
 153  *  - currently we only support "normal" cache types: "FILE" and "MEMORY".
 154  */
 155
 156 #define krb5_err_msg(code)      error_message(code)
 157
 158 const char *krb5_cc_type_mem    = "MEMORY:";
 159 const char *krb5_cc_type_file   = "FILE:";
 160 const char *krb5_cred_root_suffix  = "lustre_root";
 161 const char *krb5_cred_mds_suffix   = "lustre_mds";
 162 const char *krb5_cred_oss_suffix   = "lustre_oss";
 163
 164 char    *krb5_this_realm        = NULL;
 165 char    *krb5_keytab_file       = "/etc/krb5.keytab";
 166 char    *krb5_cc_type           = "FILE:";
 167 char    *krb5_cc_dir            = "/tmp";
 168 char    *krb5_cred_prefix       = "krb5cc_";
 169
 170 struct lgss_krb5_cred {
 171         char            kc_ccname[128];
 172         int             kc_remove;        /* remove cache upon release */
 173 };
 174
 175 static
 176 int lgss_krb5_set_ccache_name(const char *ccname)
 177 {
 178 #ifdef USE_GSS_KRB5_CCACHE_NAME
 179         unsigned int    maj_stat, min_stat;
 180
 181         maj_stat = gss_krb5_ccache_name(&min_stat, ccname, NULL);
 182         if (maj_stat != GSS_S_COMPLETE) {
 183                 logmsg(LL_ERR, "failed to set ccache name\n");
 184                 return -1;
 185         }
 186 #else
 187         /*
 188          * Set the KRB5CCNAME environment variable to tell the krb5 code
 189          * which credentials cache to use.  (Instead of using the private
 190          * function above for which there is no generic gssapi equivalent)
 191          */
 192         if (setenv("KRB5CCNAME", ccname, 1)) {
 193                 logmsg(LL_ERR, "set env of krb5 ccname: %s\n",
 194                        strerror(errno));
 195                 return -1;
 196         }
 197 #endif
 198         logmsg(LL_DEBUG, "set cc: %s\n", ccname);
 199         return 0;
 200 }
 201
 202 static
 203 int lgss_krb5_get_local_realm(void)
 204 {
 205         krb5_context    context = NULL;
 206         krb5_error_code code;
 207         int             retval = -1;
 208
 209         if (krb5_this_realm != NULL)
 210                 return 0;
 211
 212         code = krb5_init_context(&context);
 213         if (code) {
 214                 logmsg(LL_ERR, "init ctx: %s\n", krb5_err_msg(code));
 215                 return -1;
 216         }
 217
 218         code = krb5_get_default_realm(context, &krb5_this_realm);
 219         if (code) {
 220                 logmsg(LL_ERR, "get default realm: %s\n", krb5_err_msg(code));
 221                 goto out;
 222         }
 223
 224         logmsg(LL_DEBUG, "Local realm: %s\n", krb5_this_realm);
 225         retval = 0;
 226 out:
 227         krb5_free_context(context);
 228         return retval;
 229 }
 230
 231 static
 232 int princ_is_local_realm(krb5_context ctx, krb5_principal princ)
 233 {
 234         return (lgss_krb5_strcasecmp(krb5_princ_realm(ctx, princ),
 235                                      krb5_this_realm) == 0);
 236 }
 237
 238 static
 239 int svc_princ_verify_host(krb5_context ctx,
 240                           krb5_principal princ,
 241                           uint64_t self_nid,
 242                           loglevel_t loglevel)
 243 {
 244         struct utsname utsbuf;
 245         struct hostent *host;
 246         const int max_namelen = 512;
 247         char namebuf[max_namelen];
 248         char *h_name;
 249
 250         if (krb5_princ_component(ctx, princ, 1) == NULL) {
 251                 logmsg(loglevel, "service principal has no host part\n");
 252                 return -1;
 253         }
 254
 255         if (self_nid != 0) {
 256                 if (lnet_nid2hostname(self_nid, namebuf, max_namelen)) {
 257                         logmsg(loglevel,
 258                                "can't resolve hostname from nid %llx\n",
 259                                self_nid);
 260                         return -1;
 261                 }
 262                 h_name = namebuf;
 263         } else {
 264                 if (uname(&utsbuf)) {
 265                         logmsg(loglevel, "get UTS name: %s\n", strerror(errno));
 266                         return -1;
 267                 }
 268
 269                 host = gethostbyname(utsbuf.nodename);
 270                 if (host == NULL) {
 271                         logmsg(loglevel, "failed to get local hostname\n");
 272                         return -1;
 273                 }
 274                 h_name = host->h_name;
 275         }
 276
 277         if (lgss_krb5_strcasecmp(krb5_princ_component(ctx, princ, 1),
 278                                  h_name)) {
 279                 logmsg(loglevel, "service principal: hostname %.*s "
 280                        "doesn't match localhost %s\n",
 281                        krb5_princ_component(ctx, princ, 1)->length,
 282                        krb5_princ_component(ctx, princ, 1)->data,
 283                        h_name);
 284                 return -1;
 285         }
 286
 287         return 0;
 288 }
 289
 290 static
 291 int lkrb5_cc_check_tgt_princ(krb5_context ctx,
 292                              krb5_ccache ccache,
 293                              krb5_principal princ,
 294                              unsigned int flag,
 295                              uint64_t self_nid)
 296 {
 297         const char     *princ_name;
 298
 299         logmsg(LL_DEBUG, "principal: realm %.*s, type %d, size %d, name %.*s\n",
 300                krb5_princ_realm(ctx, princ)->length,
 301                krb5_princ_realm(ctx, princ)->data,
 302                krb5_princ_type(ctx, princ),
 303                krb5_princ_size(ctx, princ),
 304                krb5_princ_name(ctx, princ)->length,
 305                krb5_princ_name(ctx, princ)->data);
 306
 307         /* check type */
 308         if (krb5_princ_type(ctx, princ) != KRB5_NT_PRINCIPAL) {
 309                 logmsg(LL_WARN, "principal type %d is not I want\n",
 310                        krb5_princ_type(ctx, princ));
 311                 return -1;
 312         }
 313
 314         /* check local realm */
 315         if (!princ_is_local_realm(ctx, princ)) {
 316                 logmsg(LL_WARN, "principal realm %.*s not local: %s\n",
 317                        krb5_princ_realm(ctx, princ)->length,
 318                        krb5_princ_realm(ctx, princ)->data,
 319                        krb5_this_realm);
 320                 return -1;
 321         }
 322
 323         /* check principal name */
 324         switch (flag) {
 325         case LGSS_ROOT_CRED_ROOT:
 326                 princ_name = LGSS_USR_ROOT_STR;
 327                 break;
 328         case LGSS_ROOT_CRED_MDT:
 329                 princ_name = LGSS_SVC_MDS_STR;
 330                 break;
 331         case LGSS_ROOT_CRED_OST:
 332                 princ_name = LGSS_SVC_OSS_STR;
 333                 break;
 334         default:
 335                 lassert(0);
 336         }
 337
 338         if (lgss_krb5_strcmp(krb5_princ_name(ctx, princ), princ_name)) {
 339                 logmsg(LL_WARN, "%.*s: we expect %s instead\n",
 340                        krb5_princ_name(ctx, princ)->length,
 341                        krb5_princ_name(ctx, princ)->data,
 342                        princ_name);
 343                 return -1;
 344         }
 345
 346         /*
 347          * verify the hostname part of the principal, except we do allow
 348          * lustre_root without binding to a host.
 349          */
 350         if (krb5_princ_component(ctx, princ, 1) == NULL) {
 351                 if (flag != LGSS_ROOT_CRED_ROOT) {
 352                         logmsg(LL_WARN, "%.*s: missing hostname\n",
 353                                krb5_princ_name(ctx, princ)->length,
 354                                krb5_princ_name(ctx, princ)->data);
 355                         return -1;
 356                 }
 357         } else {
 358                 if (svc_princ_verify_host(ctx, princ, self_nid, LL_WARN)) {
 359                         logmsg(LL_DEBUG, "%.*s: doesn't belong to this node\n",
 360                                krb5_princ_name(ctx, princ)->length,
 361                                krb5_princ_name(ctx, princ)->data);
 362                         return -1;
 363                 }
 364         }
 365
 366         logmsg(LL_TRACE, "principal is OK\n");
 367         return 0;
 368 }
 369
 370 /**
 371  * compose the TGT cc name, according to the root flags.
 372  */
 373 static
 374 void get_root_tgt_ccname(char *ccname, int size, unsigned int flag)
 375 {
 376         const char *suffix;
 377
 378         switch (flag) {
 379         case LGSS_ROOT_CRED_ROOT:
 380                 suffix = krb5_cred_root_suffix;
 381                 break;
 382         case LGSS_ROOT_CRED_MDT:
 383                 suffix = krb5_cred_mds_suffix;
 384                 break;
 385         case LGSS_ROOT_CRED_OST:
 386                 suffix = krb5_cred_oss_suffix;
 387                 break;
 388         default:
 389                 lassert(0);
 390         }
 391
 392         snprintf(ccname, size, "%s%s/%s%s_%s",
 393                  krb5_cc_type, krb5_cc_dir, krb5_cred_prefix,
 394                  suffix, krb5_this_realm);
 395 }
 396
 397 static
 398 int lkrb5_check_root_tgt_cc_base(krb5_context ctx,
 399                                  krb5_ccache ccache,
 400                                  char *ccname,
 401                                  unsigned int flag,
 402                                  uint64_t self_nid)
 403 {
 404         krb5_ccache             tgt_ccache;
 405         krb5_creds              cred;
 406         krb5_principal          princ;
 407         krb5_cc_cursor          cursor;
 408         krb5_error_code         code;
 409         time_t                  now;
 410         int                     rc = -1, found = 0;
 411
 412         /* prepare parsing the cache file */
 413         code = krb5_cc_resolve(ctx, ccname, &tgt_ccache);
 414         if (code) {
 415                 logmsg(LL_ERR, "resolve krb5 cc %s: %s\n",
 416                        ccname, krb5_err_msg(code));
 417                 return -1;
 418         }
 419
 420         /* checks the principal */
 421         code = krb5_cc_get_principal(ctx, tgt_ccache, &princ);
 422         if (code) {
 423                 logmsg(LL_ERR, "get cc principal: %s\n", krb5_err_msg(code));
 424                 goto out_cc;
 425         }
 426
 427         if (lkrb5_cc_check_tgt_princ(ctx, tgt_ccache, princ, flag, self_nid))
 428                 goto out_princ;
 429
 430         /*
 431          * find a valid entry
 432          */
 433         code = krb5_cc_start_seq_get(ctx, tgt_ccache, &cursor);
 434         if (code) {
 435                 logmsg(LL_ERR, "start cc iteration: %s\n", krb5_err_msg(code));
 436                 goto out_princ;
 437         }
 438
 439         now = time(0);
 440         do {
 441                 krb5_timestamp  duration, delta;
 442
 443                 code = krb5_cc_next_cred(ctx, tgt_ccache, &cursor, &cred);
 444                 if (code != 0)
 445                         break;
 446
 447                 logmsg(LL_DEBUG, "cred: server realm %.*s, type %d, name %.*s; "
 448                        "time (%d-%d, renew till %d), valid %d\n",
 449                        krb5_princ_realm(ctx, cred.server)->length,
 450                        krb5_princ_realm(ctx, cred.server)->data,
 451                        krb5_princ_type(ctx, cred.server),
 452                        krb5_princ_name(ctx, cred.server)->length,
 453                        krb5_princ_name(ctx, cred.server)->data,
 454                        cred.times.starttime, cred.times.endtime,
 455                        cred.times.renew_till, cred.times.endtime - now);
 456
 457                 /* FIXME
 458                  * we found the princ type is always 0 (KRB5_NT_UNKNOWN), why???
 459                  */
 460
 461                 /* FIXME how about inter-realm TGT??? FIXME */
 462                 if (lgss_krb5_strcasecmp(krb5_princ_name(ctx, cred.server),
 463                                          "krbtgt"))
 464                         continue;
 465
 466                 if (lgss_krb5_strcasecmp(krb5_princ_realm(ctx, cred.server),
 467                                          krb5_this_realm))
 468                         continue;
 469
 470                 /* check validity of time */
 471                 delta = 60 * 30; /* half an hour */
 472                 duration = cred.times.endtime - cred.times.starttime;
 473                 if (duration / 4 < delta)
 474                         delta = duration / 4;
 475
 476                 if (cred.times.starttime <= now &&
 477                     cred.times.endtime >= now + delta) {
 478                         found = 1;
 479                         break;
 480                 }
 481         } while (1);
 482
 483         if (!found) {
 484                 logmsg(LL_DEBUG, "doesn't find good TGT cache\n");
 485                 goto out_seq;
 486         }
 487
 488         /* found a good cred, store it into @ccache */
 489         logmsg(LL_DEBUG, "found good TGT cache\n");
 490
 491         code = krb5_cc_initialize(ctx, ccache, princ);
 492         if (code) {
 493                 logmsg(LL_ERR, "init private cc: %s\n", krb5_err_msg(code));
 494                 goto out_seq;
 495         }
 496
 497         code = krb5_cc_store_cred(ctx, ccache, &cred);
 498         if (code) {
 499                 logmsg(LL_ERR, "store private cred: %s\n", krb5_err_msg(code));
 500                 goto out_seq;
 501         }
 502
 503         logmsg(LL_DEBUG, "store private ccache OK\n");
 504         rc = 0;
 505
 506 out_seq:
 507         krb5_cc_end_seq_get(ctx, tgt_ccache, &cursor);
 508 out_princ:
 509         krb5_free_principal(ctx, princ);
 510 out_cc:
 511         krb5_cc_close(ctx, tgt_ccache);
 512
 513         return rc;
 514 }
 515
 516 /**
 517  * find out whether current TGT cache is valid or not
 518  */
 519 static
 520 int lkrb5_check_root_tgt_cc(krb5_context ctx,
 521                             krb5_ccache ccache,
 522                             unsigned int root_flags,
 523                             uint64_t self_nid)
 524 {
 525         struct stat             statbuf;
 526         unsigned int            flag;
 527         char                    ccname[1024];
 528         char                   *ccfile;
 529         int                     i, rc;
 530
 531         for (i = 0; i < LGSS_ROOT_CRED_NR; i++) {
 532                 flag = 1 << i;
 533
 534                 if ((root_flags & flag) == 0)
 535                         continue;
 536
 537                 get_root_tgt_ccname(ccname, sizeof(ccname), flag);
 538                 logmsg(LL_DEBUG, "root krb5 TGT ccname: %s\n", ccname);
 539
 540                 /* currently we only support type "FILE", firstly make sure
 541                  * the cache file is there */
 542                 ccfile = ccname + strlen(krb5_cc_type);
 543                 if (stat(ccfile, &statbuf)) {
 544                         logmsg(LL_DEBUG, "krb5 cc %s: %s\n",
 545                                ccname, strerror(errno));
 546                         continue;
 547                 }
 548
 549                 rc = lkrb5_check_root_tgt_cc_base(ctx, ccache, ccname, flag,
 550                                                   self_nid);
 551                 if (rc == 0)
 552                         return 0;
 553         }
 554
 555         logmsg(LL_TRACE, "doesn't find a valid tgt cc\n");
 556         return -1;
 557 }
 558
 559 static
 560 int lkrb5_get_root_tgt_keytab(krb5_context ctx,
 561                               krb5_ccache ccache,
 562                               krb5_keytab kt,
 563                               krb5_principal princ,
 564                               const char *ccname)
 565 {
 566         krb5_get_init_creds_opt opts;
 567         krb5_creds              cred;
 568         krb5_ccache             tgt_ccache;
 569         krb5_error_code         code;
 570         int                     rc = -1;
 571
 572         krb5_get_init_creds_opt_init(&opts);
 573         krb5_get_init_creds_opt_set_address_list(&opts, NULL);
 574         /*
 575          * by default krb5 library obtain ticket with lifetime shorter
 576          * than the max value. we can change it here if we want. but
 577          * seems not necessary now.
 578          *
 579         krb5_get_init_creds_opt_set_tkt_life(&opts, very-long-time);
 580          *
 581          */
 582
 583         /*
 584          * obtain TGT and store into global ccache
 585          */
 586         code = krb5_get_init_creds_keytab(ctx, &cred, princ, kt,
 587                                           0, NULL, &opts);
 588         if (code) {
 589                 logmsg(LL_ERR, "failed to get root TGT for "
 590                        "principal %.*s: %s\n",
 591                        krb5_princ_name(ctx, princ)->length,
 592                        krb5_princ_name(ctx, princ)->data,
 593                        krb5_err_msg(code));
 594                 return -1;
 595         }
 596
 597         code = krb5_cc_resolve(ctx, ccname, &tgt_ccache);
 598         if (code) {
 599                 logmsg(LL_ERR, "resolve cc %s: %s\n",
 600                        ccname, krb5_err_msg(code));
 601                 goto out_cred;
 602         }
 603
 604         code = krb5_cc_initialize(ctx, tgt_ccache, princ);
 605         if (code) {
 606                 logmsg(LL_ERR, "initialize cc %s: %s\n",
 607                        ccname, krb5_err_msg(code));
 608                 goto out_cc;
 609         }
 610
 611         code = krb5_cc_store_cred(ctx, tgt_ccache, &cred);
 612         if (code) {
 613                 logmsg(LL_ERR, "store cred to cc %s: %s\n",
 614                        ccname, krb5_err_msg(code));
 615                 goto out_cc;
 616         }
 617
 618         logmsg(LL_INFO, "installed TGT of %.*s in cc %s\n",
 619                krb5_princ_name(ctx, princ)->length,
 620                krb5_princ_name(ctx, princ)->data,
 621                ccname);
 622
 623         /*
 624          * now store the cred into my own cc too
 625          */
 626         code = krb5_cc_initialize(ctx, ccache, princ);
 627         if (code) {
 628                 logmsg(LL_ERR, "init mem cc: %s\n", krb5_err_msg(code));
 629                 goto out_cc;
 630         }
 631
 632         code = krb5_cc_store_cred(ctx, ccache, &cred);
 633         if (code) {
 634                 logmsg(LL_ERR, "store mm cred: %s\n", krb5_err_msg(code));
 635                 goto out_cc;
 636         }
 637
 638         logmsg(LL_DEBUG, "stored TGT into mem cc OK\n");
 639         rc = 0;
 640 out_cc:
 641         krb5_cc_close(ctx, tgt_ccache);
 642 out_cred:
 643         krb5_free_cred_contents(ctx, &cred);
 644         return rc;
 645 }
 646
 647 /*
 648  * obtain a new root TGT
 649  */
 650 static
 651 int lkrb5_refresh_root_tgt_cc(krb5_context ctx,
 652                               krb5_ccache ccache,
 653                               unsigned int root_flags,
 654                               uint64_t self_nid)
 655 {
 656         krb5_keytab             kt;
 657         krb5_keytab_entry       kte;
 658         krb5_kt_cursor          cursor;
 659         krb5_principal          princ = NULL;
 660         krb5_error_code         code;
 661         char                    ccname[1024];
 662         unsigned int            flag = 0;
 663         int                     rc = -1;
 664
 665         /* prepare parsing the keytab file */
 666         code = krb5_kt_resolve(ctx, krb5_keytab_file, &kt);
 667         if (code) {
 668                 logmsg(LL_ERR, "resolve keytab %s: %s\n",
 669                        krb5_keytab_file, krb5_err_msg(code));
 670                 return -1;
 671         }
 672
 673         code = krb5_kt_start_seq_get(ctx, kt, &cursor);
 674         if (code) {
 675                 logmsg(LL_ERR, "start kt iteration: %s\n", krb5_err_msg(code));
 676                 goto out_kt;
 677         }
 678
 679         /* iterate keytab to find proper an entry */
 680         do {
 681                 krb5_data      *princname;
 682
 683                 code = krb5_kt_next_entry(ctx, kt, &kte, &cursor);
 684                 if (code != 0)
 685                         break;
 686
 687                 logmsg(LL_TRACE, "kt entry: realm %.*s, type %d, "
 688                        "size %d, name %.*s\n",
 689                        krb5_princ_realm(ctx, kte.principal)->length,
 690                        krb5_princ_realm(ctx, kte.principal)->data,
 691                        krb5_princ_type(ctx, kte.principal),
 692                        krb5_princ_size(ctx, kte.principal),
 693                        krb5_princ_name(ctx, kte.principal)->length,
 694                        krb5_princ_name(ctx, kte.principal)->data);
 695
 696                 if (!princ_is_local_realm(ctx, kte.principal))
 697                         continue;
 698
 699                 princname = krb5_princ_name(ctx, kte.principal);
 700
 701                 if ((root_flags & LGSS_ROOT_CRED_ROOT) != 0 &&
 702                     lgss_krb5_strcmp(princname, LGSS_USR_ROOT_STR) == 0) {
 703                         flag = LGSS_ROOT_CRED_ROOT;
 704                 } else if ((root_flags & LGSS_ROOT_CRED_MDT) != 0 &&
 705                            lgss_krb5_strcmp(princname, LGSS_SVC_MDS_STR) == 0) {
 706                         flag = LGSS_ROOT_CRED_MDT;
 707                 } else if ((root_flags & LGSS_ROOT_CRED_OST) != 0 &&
 708                            lgss_krb5_strcmp(princname, LGSS_SVC_OSS_STR) == 0) {
 709                         flag = LGSS_ROOT_CRED_OST;
 710                 } else {
 711                         logmsg(LL_TRACE, "not what we want, skip\n");
 712                         continue;
 713                 }
 714
 715                 if (krb5_princ_component(ctx, kte.principal, 1) == NULL) {
 716                         if (flag != LGSS_ROOT_CRED_ROOT) {
 717                                 logmsg(LL_TRACE, "no hostname, skip\n");
 718                                 continue;
 719                         }
 720                 } else {
 721                         if (svc_princ_verify_host(ctx, kte.principal, self_nid,
 722                                                   LL_TRACE)) {
 723                                 logmsg(LL_TRACE, "doesn't belong to this "
 724                                        "node, skip\n");
 725                                 continue;
 726                         }
 727                 }
 728
 729                 code = krb5_copy_principal(ctx, kte.principal, &princ);
 730                 if (code) {
 731                         logmsg(LL_ERR, "copy princ: %s\n", krb5_err_msg(code));
 732                         continue;
 733                 }
 734
 735                 lassert(princ != NULL);
 736                 break;
 737         } while (1);
 738
 739         krb5_kt_end_seq_get(ctx, kt, &cursor);
 740
 741         if (princ == NULL) {
 742                 logmsg(LL_ERR, "can't find proper keytab entry\n");
 743                 goto out_kt;
 744         }
 745
 746         /* obtain root TGT */
 747         get_root_tgt_ccname(ccname, sizeof(ccname), flag);
 748         rc = lkrb5_get_root_tgt_keytab(ctx, ccache, kt, princ, ccname);
 749
 750         krb5_free_principal(ctx, princ);
 751 out_kt:
 752         krb5_kt_close(ctx, kt);
 753         return rc;
 754 }
 755
 756 static
 757 int lkrb5_prepare_root_cred(struct lgss_cred *cred)
 758 {
 759         krb5_context            ctx;
 760         krb5_ccache             ccache;
 761         krb5_error_code         code;
 762         struct lgss_krb5_cred  *kcred;
 763         int                     rc = -1;
 764
 765         lassert(krb5_this_realm != NULL);
 766
 767         kcred = (struct lgss_krb5_cred *) cred->lc_mech_cred;
 768
 769         /* compose the memory cc name, since the only user is myself,
 770          * the name could be fixed */
 771         snprintf(kcred->kc_ccname, sizeof(kcred->kc_ccname),
 772                  "%s/self", krb5_cc_type_mem);
 773         logmsg(LL_TRACE, "private cc: %s\n", kcred->kc_ccname);
 774
 775         code = krb5_init_context(&ctx);
 776         if (code) {
 777                 logmsg(LL_ERR, "initialize krb5 context: %s\n",
 778                        krb5_err_msg(code));
 779                 return -1;
 780         }
 781
 782         code = krb5_cc_resolve(ctx, kcred->kc_ccname, &ccache);
 783         if (code) {
 784                 logmsg(LL_ERR, "resolve krb5 cc %s: %s\n",
 785                        kcred->kc_ccname, krb5_err_msg(code));
 786                 goto out_ctx;
 787         }
 788
 789         /*
 790          * search and/or obtain root TGT credential.
 791          * it touched global (on-disk) tgt cache, do it inside mutex locking
 792          */
 793         lgss_krb5_mutex_lock();
 794
 795         rc = lkrb5_check_root_tgt_cc(ctx, ccache, cred->lc_root_flags,
 796                                      cred->lc_self_nid);
 797         if (rc != 0)
 798                 rc = lkrb5_refresh_root_tgt_cc(ctx, ccache,
 799                                                cred->lc_root_flags,
 800                                                cred->lc_self_nid);
 801
 802         if (rc == 0)
 803                 rc = lgss_krb5_set_ccache_name(kcred->kc_ccname);
 804
 805         lgss_krb5_mutex_unlock();
 806
 807         krb5_cc_close(ctx, ccache);
 808 out_ctx:
 809         krb5_free_context(ctx);
 810
 811         logmsg(LL_DEBUG, "prepare root credentail %s\n", rc ? "failed" : "OK");
 812         return rc;
 813 }
 814
 815 static
 816 int lkrb5_prepare_user_cred(struct lgss_cred *cred)
 817 {
 818         struct lgss_krb5_cred   *kcred;
 819         int                      rc;
 820
 821         lassert(krb5_this_realm == NULL);
 822
 823         kcred = (struct lgss_krb5_cred *) cred->lc_mech_cred;
 824
 825         /*
 826          * here we just specified a fix ccname, instead of searching
 827          * entire cc dir. is this OK??
 828          */
 829         snprintf(kcred->kc_ccname, sizeof(kcred->kc_ccname),
 830                  "%s%s/%s%u",
 831                  krb5_cc_type, krb5_cc_dir, krb5_cred_prefix, cred->lc_uid);
 832         logmsg(LL_DEBUG, "using krb5 cache name: %s\n", kcred->kc_ccname);
 833
 834         rc = lgss_krb5_set_ccache_name(kcred->kc_ccname);
 835         if (rc)
 836                 logmsg(LL_ERR, "can't set krb5 ccache name: %s\n",
 837                        kcred->kc_ccname);
 838
 839         return rc;
 840 }
 841
 842 static
 843 int lgss_krb5_prepare_cred(struct lgss_cred *cred)
 844 {
 845         struct lgss_krb5_cred  *kcred;
 846         int                     rc;
 847
 848         kcred = malloc(sizeof(*kcred));
 849         if (kcred == NULL) {
 850                 logmsg(LL_ERR, "can't allocate krb5 cred\n");
 851                 return -1;
 852         }
 853
 854         kcred->kc_ccname[0] = '\0';
 855         kcred->kc_remove = 0;
 856         cred->lc_mech_cred = kcred;
 857
 858         if (cred->lc_root_flags != 0) {
 859                 if (lgss_krb5_get_local_realm())
 860                         return -1;
 861
 862                 rc = lkrb5_prepare_root_cred(cred);
 863         } else {
 864                 rc = lkrb5_prepare_user_cred(cred);
 865         }
 866
 867         return rc;
 868 }
 869
 870 static
 871 void lgss_krb5_release_cred(struct lgss_cred *cred)
 872 {
 873         struct lgss_krb5_cred   *kcred;
 874
 875         kcred = (struct lgss_krb5_cred *) cred->lc_mech_cred;
 876
 877         free(kcred);
 878         cred->lc_mech_cred = NULL;
 879 }
 880
 881 struct lgss_mech_type lgss_mech_krb5 =
 882 {
 883         .lmt_name               = "krb5",
 884         .lmt_mech_n             = LGSS_MECH_KRB5,
 885         .lmt_prepare_cred       = lgss_krb5_prepare_cred,
 886         .lmt_release_cred       = lgss_krb5_release_cred,
 887 };