1 /* 2 * Copyright (c) 2005 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 <err.h> 35 #include <getarg.h> 36 37 RCSID("$Id: test_hostname.c 15965 2005-08-23 20:18:55Z lha $"); 38 39 static int debug_flag = 0; 40 static int version_flag = 0; 41 static int help_flag = 0; 42 43 static int 44 expand_hostname(krb5_context context, const char *host) 45 { 46 krb5_error_code ret; 47 char *h, **r; 48 49 ret = krb5_expand_hostname(context, host, &h); 50 if (ret) 51 krb5_err(context, 1, ret, "krb5_expand_hostname(%s)", host); 52 53 free(h); 54 55 if (debug_flag) 56 printf("hostname: %s -> %s\n", host, h); 57 58 ret = krb5_expand_hostname_realms(context, host, &h, &r); 59 if (ret) 60 krb5_err(context, 1, ret, "krb5_expand_hostname_realms(%s)", host); 61 62 if (debug_flag) { 63 int j; 64 65 printf("hostname: %s -> %s\n", host, h); 66 for (j = 0; r[j]; j++) { 67 printf("\trealm: %s\n", r[j]); 68 } 69 } 70 free(h); 71 krb5_free_host_realm(context, r); 72 73 return 0; 74 } 75 76 static int 77 test_expand_hostname(krb5_context context) 78 { 79 int i, errors = 0; 80 81 struct t { 82 krb5_error_code ret; 83 const char *orig_hostname; 84 const char *new_hostname; 85 } tests[] = { 86 { 0, "pstn1.su.se", "pstn1.su.se" }, 87 { 0, "pstnproxy.su.se", "pstnproxy.su.se" }, 88 }; 89 90 for (i = 0; i < sizeof(tests)/sizeof(tests[0]); i++) { 91 errors += expand_hostname(context, tests[i].orig_hostname); 92 } 93 94 return errors; 95 } 96 97 static struct getargs args[] = { 98 {"debug", 'd', arg_flag, &debug_flag, 99 "turn on debuggin", NULL }, 100 {"version", 0, arg_flag, &version_flag, 101 "print version", NULL }, 102 {"help", 0, arg_flag, &help_flag, 103 NULL, NULL } 104 }; 105 106 static void 107 usage (int ret) 108 { 109 arg_printusage (args, sizeof(args)/sizeof(*args), NULL, "hostname ..."); 110 exit (ret); 111 } 112 113 114 int 115 main(int argc, char **argv) 116 { 117 krb5_context context; 118 krb5_error_code ret; 119 int optidx = 0, errors = 0; 120 121 setprogname(argv[0]); 122 123 if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx)) 124 usage(1); 125 126 if (help_flag) 127 usage (0); 128 129 if(version_flag){ 130 print_version(NULL); 131 exit(0); 132 } 133 134 argc -= optidx; 135 argv += optidx; 136 137 ret = krb5_init_context(&context); 138 if (ret) 139 errx (1, "krb5_init_context failed: %d", ret); 140 141 if (argc > 0) { 142 while (argc-- > 0) 143 errors += expand_hostname(context, *argv++); 144 return errors; 145 } 146 147 errors += test_expand_hostname(context); 148 149 krb5_free_context(context); 150 151 return errors; 152 } 153