1/*
2 * Copyright (c) 1999 - 2001 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 the Institute nor the names of its contributors
18 *    may be used to endorse or promote products derived from this software
19 *    without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#include "krb5_locl.h"
35
36static krb5_error_code
37copy_hostname(krb5_context context,
38	      const char *orig_hostname,
39	      char **new_hostname)
40{
41    *new_hostname = strdup (orig_hostname);
42    if (*new_hostname == NULL) {
43	krb5_set_error_message(context, ENOMEM,
44			       N_("malloc: out of memory", ""));
45	return ENOMEM;
46    }
47    strlwr (*new_hostname);
48    return 0;
49}
50
51/**
52 * krb5_expand_hostname() tries to make orig_hostname into a more
53 * canonical one in the newly allocated space returned in
54 * new_hostname.
55
56 * @param context a Keberos context
57 * @param orig_hostname hostname to canonicalise.
58 * @param new_hostname output hostname, caller must free hostname with
59 *        krb5_xfree().
60 *
61 * @return Return an error code or 0, see krb5_get_error_message().
62 *
63 * @ingroup krb5_support
64 */
65
66KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
67krb5_expand_hostname (krb5_context context,
68		      const char *orig_hostname,
69		      char **new_hostname)
70{
71    struct addrinfo *ai, *a, hints;
72    int error;
73
74    if ((context->flags & KRB5_CTX_F_DNS_CANONICALIZE_HOSTNAME) == 0)
75	return copy_hostname (context, orig_hostname, new_hostname);
76
77    memset (&hints, 0, sizeof(hints));
78    hints.ai_flags = AI_CANONNAME;
79
80    error = getaddrinfo (orig_hostname, NULL, &hints, &ai);
81    if (error)
82	return copy_hostname (context, orig_hostname, new_hostname);
83    for (a = ai; a != NULL; a = a->ai_next) {
84	if (a->ai_canonname != NULL) {
85	    *new_hostname = strdup (a->ai_canonname);
86	    freeaddrinfo (ai);
87	    if (*new_hostname == NULL) {
88		krb5_set_error_message(context, ENOMEM,
89				       N_("malloc: out of memory", ""));
90		return ENOMEM;
91	    } else {
92		return 0;
93	    }
94	}
95    }
96    freeaddrinfo (ai);
97    return copy_hostname (context, orig_hostname, new_hostname);
98}
99
100/*
101 * handle the case of the hostname being unresolvable and thus identical
102 */
103
104static krb5_error_code
105vanilla_hostname (krb5_context context,
106		  const char *orig_hostname,
107		  char **new_hostname,
108		  char ***realms)
109{
110    krb5_error_code ret;
111
112    ret = copy_hostname (context, orig_hostname, new_hostname);
113    if (ret)
114	return ret;
115    strlwr (*new_hostname);
116
117    ret = krb5_get_host_realm (context, *new_hostname, realms);
118    if (ret) {
119	free (*new_hostname);
120	return ret;
121    }
122    return 0;
123}
124
125/**
126 * krb5_expand_hostname_realms() expands orig_hostname to a name we
127 * believe to be a hostname in newly allocated space in new_hostname
128 * and return the realms new_hostname is believed to belong to in
129 * realms.
130 *
131 * @param context a Keberos context
132 * @param orig_hostname hostname to canonicalise.
133 * @param new_hostname output hostname, caller must free hostname with
134 *        krb5_xfree().
135 * @param realms output possible realms, is an array that is terminated
136 *        with NULL. Caller must free with krb5_free_host_realm().
137 *
138 * @return Return an error code or 0, see krb5_get_error_message().
139 *
140 * @ingroup krb5_support
141 */
142
143KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
144krb5_expand_hostname_realms (krb5_context context,
145			     const char *orig_hostname,
146			     char **new_hostname,
147			     char ***realms)
148{
149    struct addrinfo *ai, *a, hints;
150    int error;
151    krb5_error_code ret = 0;
152
153    if ((context->flags & KRB5_CTX_F_DNS_CANONICALIZE_HOSTNAME) == 0)
154	return vanilla_hostname (context, orig_hostname, new_hostname,
155				 realms);
156
157    memset (&hints, 0, sizeof(hints));
158    hints.ai_flags = AI_CANONNAME;
159
160    error = getaddrinfo (orig_hostname, NULL, &hints, &ai);
161    if (error)
162	return vanilla_hostname (context, orig_hostname, new_hostname,
163				 realms);
164
165    for (a = ai; a != NULL; a = a->ai_next) {
166	if (a->ai_canonname != NULL) {
167	    ret = copy_hostname (context, a->ai_canonname, new_hostname);
168	    if (ret) {
169		freeaddrinfo (ai);
170		return ret;
171	    }
172	    strlwr (*new_hostname);
173	    ret = krb5_get_host_realm (context, *new_hostname, realms);
174	    if (ret == 0) {
175		freeaddrinfo (ai);
176		return 0;
177	    }
178	    free (*new_hostname);
179	}
180    }
181    freeaddrinfo(ai);
182    return vanilla_hostname (context, orig_hostname, new_hostname, realms);
183}
184