1 /* 2 * Copyright (c) 2003 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 #include "krb5_locl.h" 34 #include <getarg.h> 35 #include <err.h> 36 37 static void 38 test_alname(krb5_context context, krb5_const_realm realm, 39 const char *user, const char *inst, 40 const char *localuser, int ok) 41 { 42 krb5_principal p; 43 char localname[1024]; 44 krb5_error_code ret; 45 char *princ; 46 47 ret = krb5_make_principal(context, &p, realm, user, inst, NULL); 48 if (ret) 49 krb5_err(context, 1, ret, "krb5_build_principal"); 50 51 ret = krb5_unparse_name(context, p, &princ); 52 if (ret) 53 krb5_err(context, 1, ret, "krb5_unparse_name"); 54 55 ret = krb5_aname_to_localname(context, p, sizeof(localname), localname); 56 krb5_free_principal(context, p); 57 free(princ); 58 if (ret) { 59 if (!ok) 60 return; 61 krb5_err(context, 1, ret, "krb5_aname_to_localname: %s -> %s", 62 princ, localuser); 63 } 64 65 if (strcmp(localname, localuser) != 0) { 66 if (ok) 67 errx(1, "compared failed %s != %s (should have succeded)", 68 localname, localuser); 69 } else { 70 if (!ok) 71 errx(1, "compared failed %s == %s (should have failed)", 72 localname, localuser); 73 } 74 75 } 76 77 static int version_flag = 0; 78 static int help_flag = 0; 79 80 static struct getargs args[] = { 81 {"version", 0, arg_flag, &version_flag, 82 "print version", NULL }, 83 {"help", 0, arg_flag, &help_flag, 84 NULL, NULL } 85 }; 86 87 static void 88 usage (int ret) 89 { 90 arg_printusage (args, 91 sizeof(args)/sizeof(*args), 92 NULL, 93 ""); 94 exit (ret); 95 } 96 97 int 98 main(int argc, char **argv) 99 { 100 krb5_context context; 101 krb5_error_code ret; 102 krb5_realm realm; 103 int optidx = 0; 104 char *user; 105 106 setprogname(argv[0]); 107 108 if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx)) 109 usage(1); 110 111 if (help_flag) 112 usage (0); 113 114 if(version_flag){ 115 print_version(NULL); 116 exit(0); 117 } 118 119 argc -= optidx; 120 argv += optidx; 121 122 if (argc != 1) 123 errx(1, "first argument should be a local user that in root .k5login"); 124 125 user = argv[0]; 126 127 ret = krb5_init_context(&context); 128 if (ret) 129 errx (1, "krb5_init_context failed: %d", ret); 130 131 ret = krb5_get_default_realm(context, &realm); 132 if (ret) 133 krb5_err(context, 1, ret, "krb5_get_default_realm"); 134 135 test_alname(context, realm, user, NULL, user, 1); 136 test_alname(context, realm, user, "root", "root", 1); 137 138 test_alname(context, "FOO.BAR.BAZ.KAKA", user, NULL, user, 0); 139 test_alname(context, "FOO.BAR.BAZ.KAKA", user, "root", "root", 0); 140 141 test_alname(context, realm, user, NULL, 142 "not-same-as-user", 0); 143 test_alname(context, realm, user, "root", 144 "not-same-as-user", 0); 145 146 test_alname(context, "FOO.BAR.BAZ.KAKA", user, NULL, 147 "not-same-as-user", 0); 148 test_alname(context, "FOO.BAR.BAZ.KAKA", user, "root", 149 "not-same-as-user", 0); 150 151 krb5_free_context(context); 152 153 return 0; 154 } 155