1 /* $NetBSD: test_hostname.c,v 1.1.1.1 2011/04/13 18:15:38 elric Exp $ */
2
3 /*
4 * Copyright (c) 2005 Kungliga Tekniska Högskolan
5 * (Royal Institute of Technology, Stockholm, Sweden).
6 * All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 *
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 *
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * 3. Neither the name of KTH nor the names of its contributors may be
20 * used to endorse or promote products derived from this software without
21 * specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY KTH AND ITS CONTRIBUTORS ``AS IS'' AND ANY
24 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
26 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL KTH OR ITS CONTRIBUTORS BE
27 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
28 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
29 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
30 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
31 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
32 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
33 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */
34
35 #include "krb5_locl.h"
36 #include <err.h>
37 #include <krb5/getarg.h>
38
39 static int debug_flag = 0;
40 static int version_flag = 0;
41 static int help_flag = 0;
42
43 static int
expand_hostname(krb5_context context,const char * host)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
test_expand_hostname(krb5_context context)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
usage(int ret)107 usage (int ret)
108 {
109 arg_printusage (args, sizeof(args)/sizeof(*args), NULL, "hostname ...");
110 exit (ret);
111 }
112
113
114 int
main(int argc,char ** argv)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