source4/heimdal/lib/gssapi/test_names.c

   1 /*
   2  * Copyright (c) 2006 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 <err.h>
  47 #include <getarg.h>
  48
  49 static void
  50 gss_print_errors (int min_stat)
  51 {
  52     OM_uint32 new_stat;
  53     OM_uint32 msg_ctx = 0;
  54     gss_buffer_desc status_string;
  55     OM_uint32 ret;
  56
  57     do {
  58         ret = gss_display_status (&new_stat,
  59                                   min_stat,
  60                                   GSS_C_MECH_CODE,
  61                                   GSS_C_NO_OID,
  62                                   &msg_ctx,
  63                                   &status_string);
  64         if (!GSS_ERROR(ret)) {
  65             fprintf (stderr, "%.*s\n", (int)status_string.length,
  66                                         (char *)status_string.value);
  67             gss_release_buffer (&new_stat, &status_string);
  68         }
  69     } while (!GSS_ERROR(ret) && msg_ctx != 0);
  70 }
  71
  72 static void
  73 gss_err(int exitval, int status, const char *fmt, ...)
  74 {
  75     va_list args;
  76
  77     va_start(args, fmt);
  78     vwarnx (fmt, args);
  79     gss_print_errors (status);
  80     va_end(args);
  81     exit (exitval);
  82 }
  83
  84 static int version_flag = 0;
  85 static int help_flag    = 0;
  86 static int anon_flag    = 0;
  87
  88 static struct getargs args[] = {
  89     {"version", 0,      arg_flag,       &version_flag, "print version", NULL },
  90     {"anonymous", 0,    arg_flag,       &anon_flag, "test anonymous names", NULL },
  91     {"help",    0,      arg_flag,       &help_flag,  NULL, NULL }
  92 };
  93
  94 static void
  95 usage (int ret)
  96 {
  97     arg_printusage (args, sizeof(args)/sizeof(*args),
  98                     NULL, "service@host");
  99     exit (ret);
 100 }
 101
 102
 103 int
 104 main(int argc, char **argv)
 105 {
 106     gss_buffer_desc name_buffer;
 107     OM_uint32 maj_stat, min_stat;
 108     gss_name_t name, MNname, MNname2;
 109     int optidx = 0;
 110     char *str;
 111     int len, equal;
 112     gss_OID mech_oid;
 113
 114     setprogname(argv[0]);
 115     if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
 116         usage(1);
 117
 118     if (help_flag)
 119         usage (0);
 120
 121     if(version_flag){
 122         print_version(NULL);
 123         exit(0);
 124     }
 125
 126     argc -= optidx;
 127     argv += optidx;
 128
 129     gsskrb5_set_default_realm("MIT.EDU");
 130
 131     /*
 132      * test import/export
 133      */
 134
 135     str = NULL;
 136     len = asprintf(&str, anon_flag ?
 137         "WELLKNOWN@ANONYMOUS" : "ftp@freeze-arrow.mit.edu");
 138     if (len < 0 || str == NULL)
 139         errx(1, "asprintf");
 140
 141     name_buffer.value = str;
 142     name_buffer.length = len;
 143
 144     maj_stat = gss_import_name(&min_stat, &name_buffer,
 145                                GSS_C_NT_HOSTBASED_SERVICE,
 146                                &name);
 147     if (maj_stat != GSS_S_COMPLETE)
 148         gss_err(1, min_stat, "import name error");
 149     free(str);
 150
 151     if (anon_flag)
 152         mech_oid = GSS_SANON_X25519_MECHANISM;
 153     else
 154         mech_oid = GSS_KRB5_MECHANISM;
 155
 156     maj_stat = gss_canonicalize_name (&min_stat,
 157                                       name,
 158                                       mech_oid,
 159                                       &MNname);
 160     if (maj_stat != GSS_S_COMPLETE)
 161         gss_err(1, min_stat, "canonicalize name error");
 162
 163     maj_stat = gss_export_name(&min_stat,
 164                                MNname,
 165                                &name_buffer);
 166     if (maj_stat != GSS_S_COMPLETE)
 167         gss_err(1, min_stat, "export name error (KRB5)");
 168
 169     /*
 170      * Import the exported name and compare
 171      */
 172
 173     maj_stat = gss_import_name(&min_stat, &name_buffer,
 174                                GSS_C_NT_EXPORT_NAME,
 175                                &MNname2);
 176     if (maj_stat != GSS_S_COMPLETE)
 177         gss_err(1, min_stat, "import name error (exported KRB5 name)");
 178
 179
 180     maj_stat = gss_compare_name(&min_stat, MNname, MNname2, &equal);
 181     if (maj_stat != GSS_S_COMPLETE)
 182         errx(1, "gss_compare_name");
 183     if (equal == anon_flag)
 184         errx(1, "names %s equal", anon_flag ? "incorrectly" : "not");
 185
 186     gss_release_name(&min_stat, &MNname2);
 187     gss_release_buffer(&min_stat, &name_buffer);
 188     gss_release_name(&min_stat, &MNname);
 189     gss_release_name(&min_stat, &name);
 190
 191     /*
 192      * Import oid less name and compare to mech name.
 193      * Dovecot SASL lib does this.
 194      */
 195
 196     str = NULL;
 197     len = asprintf(&str, "lha");
 198     if (len < 0 || str == NULL)
 199         errx(1, "asprintf");
 200
 201     name_buffer.value = str;
 202     name_buffer.length = len;
 203
 204     maj_stat = gss_import_name(&min_stat, &name_buffer,
 205                                GSS_C_NO_OID,
 206                                &name);
 207     if (maj_stat != GSS_S_COMPLETE)
 208         gss_err(1, min_stat, "import (no oid) name error");
 209
 210     maj_stat = gss_import_name(&min_stat, &name_buffer,
 211                                GSS_KRB5_NT_USER_NAME,
 212                                &MNname);
 213     if (maj_stat != GSS_S_COMPLETE)
 214         gss_err(1, min_stat, "import (krb5 mn) name error");
 215
 216     free(str);
 217
 218     maj_stat = gss_compare_name(&min_stat, name, MNname, &equal);
 219     if (maj_stat != GSS_S_COMPLETE)
 220         errx(1, "gss_compare_name");
 221     if (!equal)
 222         errx(1, "names not equal");
 223
 224     gss_release_name(&min_stat, &MNname);
 225     gss_release_name(&min_stat, &name);
 226
 227 #if 0
 228     maj_stat = gss_canonicalize_name (&min_stat,
 229                                       name,
 230                                       GSS_SPNEGO_MECHANISM,
 231                                       &MNname);
 232     if (maj_stat != GSS_S_COMPLETE)
 233         gss_err(1, min_stat, "canonicalize name error");
 234
 235
 236     maj_stat = gss_export_name(&maj_stat,
 237                                MNname,
 238                                &name_buffer);
 239     if (maj_stat != GSS_S_COMPLETE)
 240         gss_err(1, min_stat, "export name error (SPNEGO)");
 241
 242     gss_release_name(&min_stat, &MNname);
 243     gss_release_buffer(&min_stat, &name_buffer);
 244 #endif
 245
 246     if (anon_flag) {
 247         /* check anonymous name canonicalizes to well known name */
 248         gss_OID name_type;
 249
 250         name_buffer.length = 0;
 251         name_buffer.value = NULL;
 252
 253         maj_stat = gss_import_name(&min_stat, &name_buffer,
 254                                    GSS_C_NT_ANONYMOUS, &name);
 255         if (maj_stat != GSS_S_COMPLETE)
 256             gss_err(1, min_stat, "import (anon) name error");
 257
 258         maj_stat = gss_canonicalize_name(&min_stat, name,
 259                                          GSS_SANON_X25519_MECHANISM,
 260                                          &MNname);
 261         if (maj_stat != GSS_S_COMPLETE)
 262             gss_err(1, min_stat, "canonicalize (anon) name error");
 263
 264         maj_stat = gss_display_name(&min_stat, MNname,
 265                                     &name_buffer, &name_type);
 266         if (maj_stat != GSS_S_COMPLETE)
 267             gss_err(1, min_stat, "display_name (anon) name error");
 268
 269         if (!gss_oid_equal(name_type, GSS_C_NT_ANONYMOUS))
 270             gss_err(1, 0, "display name type not anonymous");
 271         if (memcmp(name_buffer.value, "WELLKNOWN/ANONYMOUS@WELLKNOWN:ANONYMOUS",
 272                    sizeof("WELLKNOWN/ANONYMOUS@WELLKNOWN:ANONYMOUS") - 1) != 0)
 273             gss_err(1, 0, "display name string not well known anonymous name");
 274
 275         gss_release_name(&min_stat, &MNname);
 276         gss_release_name(&min_stat, &name);
 277         gss_release_buffer(&min_stat, &name_buffer);
 278     }
 279
 280     return 0;
 281 }