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