1/*
2 * Copyright (c) 2006 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#include "locate_plugin.h"
36
37static krb5_error_code
38resolve_init(krb5_context context, void **ctx)
39{
40    *ctx = NULL;
41    return 0;
42}
43
44static void
45resolve_fini(void *ctx)
46{
47}
48
49static krb5_error_code
50resolve_lookup(void *ctx,
51	       unsigned long flags,
52	       enum locate_service_type service,
53	       const char *realm,
54	       int domain,
55	       int type,
56	       int (*add)(void *,int,struct sockaddr *),
57	       void *addctx)
58{
59    struct sockaddr_in s;
60
61    memset(&s, 0, sizeof(s));
62
63#ifdef HAVE_STRUCT_SOCKADDR_SA_LEN
64    s.sin_len = sizeof(s);
65#endif
66    s.sin_family = AF_INET;
67    s.sin_port = htons(88);
68    s.sin_addr.s_addr = htonl(0x7f000002);
69
70    if (strcmp(realm, "NOTHERE.H5L.SE") == 0)
71	(*add)(addctx, type, (struct sockaddr *)&s);
72
73    return 0;
74}
75
76static krb5_error_code
77resolve_lookup_old(void *ctx,
78		   unsigned long flags,
79		   enum locate_service_type service,
80		   const char *realm,
81		   int domain,
82		   int type,
83		   int (*add)(void *,int,struct sockaddr *),
84		   void *addctx)
85{
86    return resolve_lookup(ctx, KRB5_PLF_ALLOW_HOMEDIR, service,
87			  realm, domain, type, add, addctx);
88}
89
90krb5plugin_service_locate_ftable resolve = {
91    KRB5_PLUGIN_LOCATE_VERSION_2,
92    resolve_init,
93    resolve_fini,
94    resolve_lookup_old,
95    resolve_lookup
96
97};
98
99
100int
101main(int argc, char **argv)
102{
103    krb5_error_code ret;
104    krb5_context context;
105    krb5_krbhst_handle handle;
106    char host[MAXHOSTNAMELEN];
107    int found = 0;
108
109    setprogname(argv[0]);
110
111    ret = krb5_init_context(&context);
112    if (ret)
113	errx(1, "krb5_init_contex");
114
115    ret = krb5_plugin_register(context, PLUGIN_TYPE_DATA,
116			       KRB5_PLUGIN_LOCATE, &resolve);
117    if (ret)
118	krb5_err(context, 1, ret, "krb5_plugin_register");
119
120
121    ret = krb5_krbhst_init_flags(context,
122				 "NOTHERE.H5L.SE",
123				 KRB5_KRBHST_KDC,
124				 0,
125				 &handle);
126    if (ret)
127	krb5_err(context, 1, ret, "krb5_krbhst_init_flags");
128
129
130    while(krb5_krbhst_next_as_string(context, handle, host, sizeof(host)) == 0){
131	found++;
132 	if (strcmp(host, "127.0.0.2") != 0)
133	    krb5_errx(context, 1, "wrong address: %s", host);
134    }
135    if (!found)
136	krb5_errx(context, 1, "failed to find host");
137
138    krb5_krbhst_free(context, handle);
139
140    krb5_free_context(context);
141    return 0;
142}
143