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