lib/gssapi/krb5/test_cred.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 #include "gssapi_locl.h"
  35 #include <err.h>
  36 #include <getarg.h>
  37
  38 RCSID("$Id$");
  39
  40 static void
  41 acquire_release_loop(gss_name_t name, int counter, gss_cred_usage_t usage)
  42 {
  43     OM_uint32 maj_stat, min_stat;
  44     gss_cred_id_t cred;
  45     int i;
  46
  47     for (i = 0; i < counter; i++) {
  48         maj_stat = gss_acquire_cred(&min_stat, name,
  49                                     GSS_C_INDEFINITE,
  50                                     GSS_C_NO_OID_SET,
  51                                     usage,
  52                                     &cred,
  53                                     NULL,
  54                                     NULL);
  55         if (maj_stat != GSS_S_COMPLETE)
  56             errx(1, "aquire %d %d != GSS_S_COMPLETE", i, (int)maj_stat);
  57
  58         maj_stat = gss_release_cred(&min_stat, &cred);
  59         if (maj_stat != GSS_S_COMPLETE)
  60             errx(1, "release %d %d != GSS_S_COMPLETE", i, (int)maj_stat);
  61     }
  62 }
  63
  64
  65 static void
  66 acquire_add_release_add(gss_name_t name, gss_cred_usage_t usage)
  67 {
  68     OM_uint32 maj_stat, min_stat;
  69     gss_cred_id_t cred, cred2, cred3;
  70
  71     maj_stat = gss_acquire_cred(&min_stat, name,
  72                                 GSS_C_INDEFINITE,
  73                                 GSS_C_NO_OID_SET,
  74                                 usage,
  75                                 &cred,
  76                                 NULL,
  77                                 NULL);
  78     if (maj_stat != GSS_S_COMPLETE)
  79         errx(1, "aquire %d != GSS_S_COMPLETE", (int)maj_stat);
  80
  81     maj_stat = gss_add_cred(&min_stat,
  82                             cred,
  83                             GSS_C_NO_NAME,
  84                             GSS_KRB5_MECHANISM,
  85                             usage,
  86                             GSS_C_INDEFINITE,
  87                             GSS_C_INDEFINITE,
  88                             &cred2,
  89                             NULL,
  90                             NULL,
  91                             NULL);
  92
  93     if (maj_stat != GSS_S_COMPLETE)
  94         errx(1, "add_cred %d != GSS_S_COMPLETE", (int)maj_stat);
  95
  96     maj_stat = gss_release_cred(&min_stat, &cred);
  97     if (maj_stat != GSS_S_COMPLETE)
  98         errx(1, "release %d != GSS_S_COMPLETE", (int)maj_stat);
  99
 100     maj_stat = gss_add_cred(&min_stat,
 101                             cred2,
 102                             GSS_C_NO_NAME,
 103                             GSS_KRB5_MECHANISM,
 104                             GSS_C_BOTH,
 105                             GSS_C_INDEFINITE,
 106                             GSS_C_INDEFINITE,
 107                             &cred3,
 108                             NULL,
 109                             NULL,
 110                             NULL);
 111
 112     maj_stat = gss_release_cred(&min_stat, &cred2);
 113     if (maj_stat != GSS_S_COMPLETE)
 114         errx(1, "release 2 %d != GSS_S_COMPLETE", (int)maj_stat);
 115
 116     maj_stat = gss_release_cred(&min_stat, &cred3);
 117     if (maj_stat != GSS_S_COMPLETE)
 118         errx(1, "release 2 %d != GSS_S_COMPLETE", (int)maj_stat);
 119 }
 120
 121 static int version_flag = 0;
 122 static int help_flag    = 0;
 123
 124 static struct getargs args[] = {
 125     {"version", 0,      arg_flag,       &version_flag, "print version", NULL },
 126     {"help",    0,      arg_flag,       &help_flag,  NULL, NULL }
 127 };
 128
 129 static void
 130 usage (int ret)
 131 {
 132     arg_printusage (args, sizeof(args)/sizeof(*args),
 133                     NULL, "service@host");
 134     exit (ret);
 135 }
 136
 137
 138 int
 139 main(int argc, char **argv)
 140 {
 141     struct gss_buffer_desc_struct name_buffer;
 142     OM_uint32 maj_stat, min_stat;
 143     gss_name_t name;
 144     int optind = 0;
 145
 146     setprogname(argv[0]);
 147     if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optind))
 148         usage(1);
 149
 150     if (help_flag)
 151         usage (0);
 152
 153     if(version_flag){
 154         print_version(NULL);
 155         exit(0);
 156     }
 157
 158     argc -= optind;
 159     argv += optind;
 160
 161     if (argc < 1)
 162         errx(1, "argc < 1");
 163
 164     name_buffer.value = argv[0];
 165     name_buffer.length = strlen(argv[0]);
 166
 167     maj_stat = gss_import_name(&min_stat, &name_buffer,
 168                                GSS_C_NT_HOSTBASED_SERVICE,
 169                                &name);
 170     if (maj_stat != GSS_S_COMPLETE)
 171         errx(1, "import name error");
 172
 173     acquire_release_loop(name, 100, GSS_C_ACCEPT);
 174     acquire_release_loop(name, 100, GSS_C_INITIATE);
 175     acquire_release_loop(name, 100, GSS_C_BOTH);
 176
 177     acquire_add_release_add(name, GSS_C_ACCEPT);
 178     acquire_add_release_add(name, GSS_C_INITIATE);
 179     acquire_add_release_add(name, GSS_C_BOTH);
 180
 181     gss_release_name(&min_stat, &name);
 182
 183     return 0;
 184 }