lib/gssapi/test_kcred.c

   1 /*
   2  * Copyright (c) 2003-2004 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 KTH nor the names of its contributors may be
  18  *    used to endorse or promote products derived from this software without
  19  *    specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY KTH AND ITS CONTRIBUTORS ``AS IS'' AND ANY
  22  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  24  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL KTH OR ITS CONTRIBUTORS BE
  25  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  26  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  27  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
  28  * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  29  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  30  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  31  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 #ifdef HAVE_CONFIG_H
  35 #include <config.h>
  36 #endif
  37
  38 #include <roken.h>
  39 #include <stdio.h>
  40 #include <stdlib.h>
  41 #include <string.h>
  42 #include <stdarg.h>
  43 #include <gssapi.h>
  44 #include <gssapi_krb5.h>
  45 #include <gssapi_spnego.h>
  46 #include <krb5.h>
  47 #include <err.h>
  48 #include <getarg.h>
  49
  50 static int version_flag = 0;
  51 static int help_flag    = 0;
  52
  53 static void
  54 copy_import(void)
  55 {
  56     gss_cred_id_t cred1, cred2;
  57     OM_uint32 maj_stat, min_stat;
  58     gss_name_t name1, name2;
  59     OM_uint32 lifetime1, lifetime2;
  60     gss_cred_usage_t usage1, usage2;
  61     gss_OID_set mechs1, mechs2;
  62     krb5_ccache id;
  63     krb5_error_code ret;
  64     krb5_context context;
  65     int equal;
  66
  67     maj_stat = gss_acquire_cred(&min_stat, GSS_C_NO_NAME, GSS_C_INDEFINITE,
  68                                 GSS_C_NO_OID_SET, GSS_C_INITIATE,
  69                                 &cred1, NULL, NULL);
  70     if (maj_stat != GSS_S_COMPLETE)
  71         errx(1, "gss_acquire_cred");
  72
  73     maj_stat = gss_inquire_cred(&min_stat, cred1, &name1, &lifetime1,
  74                                 &usage1, &mechs1);
  75     if (maj_stat != GSS_S_COMPLETE)
  76         errx(1, "gss_inquire_cred");
  77
  78     ret = krb5_init_context(&context);
  79     if (ret)
  80         errx(1, "krb5_init_context");
  81
  82     ret = krb5_cc_new_unique(context, krb5_cc_type_memory, NULL, &id);
  83     if (ret)
  84         krb5_err(context, 1, ret, "krb5_cc_new_unique");
  85
  86     maj_stat = gss_krb5_copy_ccache(&min_stat, cred1, id);
  87     if (maj_stat != GSS_S_COMPLETE)
  88         errx(1, "gss_krb5_copy_ccache");
  89
  90     maj_stat = gss_krb5_import_cred(&min_stat, id, NULL, NULL, &cred2);
  91     if (maj_stat != GSS_S_COMPLETE)
  92         errx(1, "gss_krb5_import_cred");
  93
  94     maj_stat = gss_inquire_cred(&min_stat, cred2, &name2, &lifetime2,
  95                                 &usage2, &mechs2);
  96     if (maj_stat != GSS_S_COMPLETE)
  97         errx(1, "gss_inquire_cred 2");
  98
  99     maj_stat = gss_compare_name(&min_stat, name1, name2, &equal);
 100     if (maj_stat != GSS_S_COMPLETE)
 101         errx(1, "gss_compare_name");
 102     if (!equal)
 103         errx(1, "names not equal");
 104
 105     /*
 106      * This check is racy!  It tends to fail when run with valgrind.
 107      *
 108      * make check-valgrind sets TESTS_ENVIRONMENT in the environment...
 109      */
 110     if (getenv("TESTS_ENVIRONMENT") == NULL && lifetime1 != lifetime2)
 111         errx(1, "lifetime not equal %lu != %lu",
 112              (unsigned long)lifetime1, (unsigned long)lifetime2);
 113     if (lifetime1 != lifetime2)
 114         warnx("lifetime not equal %lu != %lu",
 115               (unsigned long)lifetime1, (unsigned long)lifetime2);
 116
 117     if (usage1 != usage2) {
 118         /* as long any of them is both are everything it ok */
 119         if (usage1 != GSS_C_BOTH && usage2 != GSS_C_BOTH)
 120             errx(1, "usages disjoined");
 121     }
 122
 123     gss_release_name(&min_stat, &name2);
 124     gss_release_oid_set(&min_stat, &mechs2);
 125
 126     maj_stat = gss_inquire_cred(&min_stat, cred2, &name2, &lifetime2,
 127                                 &usage2, &mechs2);
 128     if (maj_stat != GSS_S_COMPLETE)
 129         errx(1, "gss_inquire_cred");
 130
 131     maj_stat = gss_compare_name(&min_stat, name1, name2, &equal);
 132     if (maj_stat != GSS_S_COMPLETE)
 133         errx(1, "gss_compare_name");
 134     if (!equal)
 135         errx(1, "names not equal");
 136
 137     /* This check is racy! */
 138     if (getenv("TESTS_ENVIRONMENT") == NULL && lifetime1 != lifetime2)
 139         errx(1, "lifetime not equal %lu != %lu",
 140              (unsigned long)lifetime1, (unsigned long)lifetime2);
 141     if (lifetime1 != lifetime2)
 142         warnx("lifetime not equal %lu != %lu",
 143               (unsigned long)lifetime1, (unsigned long)lifetime2);
 144
 145     gss_release_cred(&min_stat, &cred1);
 146     gss_release_cred(&min_stat, &cred2);
 147
 148     gss_release_name(&min_stat, &name1);
 149     gss_release_name(&min_stat, &name2);
 150
 151 #if 0
 152     compare(mechs1, mechs2);
 153 #endif
 154
 155     gss_release_oid_set(&min_stat, &mechs1);
 156     gss_release_oid_set(&min_stat, &mechs2);
 157
 158     krb5_cc_destroy(context, id);
 159     krb5_free_context(context);
 160 }
 161
 162 static struct getargs args[] = {
 163     {"version", 0,      arg_flag,       &version_flag, "print version", NULL },
 164     {"help",    0,      arg_flag,       &help_flag,  NULL, NULL }
 165 };
 166
 167 static void
 168 usage (int ret)
 169 {
 170     arg_printusage (args, sizeof(args)/sizeof(*args),
 171                     NULL, "");
 172     exit (ret);
 173 }
 174
 175 int
 176 main(int argc, char **argv)
 177 {
 178     int optidx = 0;
 179
 180     setprogname(argv[0]);
 181     if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
 182         usage(1);
 183
 184     if (help_flag)
 185         usage (0);
 186
 187     if(version_flag){
 188         print_version(NULL);
 189         exit(0);
 190     }
 191
 192     argc -= optidx;
 193     argv += optidx;
 194
 195     copy_import();
 196
 197     return 0;
 198 }