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
37RCSID("$Id: test_hostname.c 15965 2005-08-23 20:18:55Z lha $");
38
39static int debug_flag	= 0;
40static int version_flag = 0;
41static int help_flag	= 0;
42
43static int
44expand_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
76static int
77test_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
97static 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
106static void
107usage (int ret)
108{
109    arg_printusage (args, sizeof(args)/sizeof(*args), NULL, "hostname ...");
110    exit (ret);
111}
112
113
114int
115main(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