xref: /freebsd-src/crypto/heimdal/kdc/connect.c (revision 6a068746777241722b2b32c5d0bc443a2a64d80b)
1b528cefcSMark Murray /*
2*ae771770SStanislav Sedov  * Copyright (c) 1997-2005 Kungliga Tekniska Högskolan
3b528cefcSMark Murray  * (Royal Institute of Technology, Stockholm, Sweden).
4b528cefcSMark Murray  * All rights reserved.
5b528cefcSMark Murray  *
6b528cefcSMark Murray  * Redistribution and use in source and binary forms, with or without
7b528cefcSMark Murray  * modification, are permitted provided that the following conditions
8b528cefcSMark Murray  * are met:
9b528cefcSMark Murray  *
10b528cefcSMark Murray  * 1. Redistributions of source code must retain the above copyright
11b528cefcSMark Murray  *    notice, this list of conditions and the following disclaimer.
12b528cefcSMark Murray  *
13b528cefcSMark Murray  * 2. Redistributions in binary form must reproduce the above copyright
14b528cefcSMark Murray  *    notice, this list of conditions and the following disclaimer in the
15b528cefcSMark Murray  *    documentation and/or other materials provided with the distribution.
16b528cefcSMark Murray  *
17b528cefcSMark Murray  * 3. Neither the name of the Institute nor the names of its contributors
18b528cefcSMark Murray  *    may be used to endorse or promote products derived from this software
19b528cefcSMark Murray  *    without specific prior written permission.
20b528cefcSMark Murray  *
21b528cefcSMark Murray  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22b528cefcSMark Murray  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23b528cefcSMark Murray  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24b528cefcSMark Murray  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25b528cefcSMark Murray  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26b528cefcSMark Murray  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27b528cefcSMark Murray  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28b528cefcSMark Murray  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29b528cefcSMark Murray  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30b528cefcSMark Murray  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31b528cefcSMark Murray  * SUCH DAMAGE.
32b528cefcSMark Murray  */
33b528cefcSMark Murray 
34b528cefcSMark Murray #include "kdc_locl.h"
35b528cefcSMark Murray 
36c19800e8SDoug Rabson /* Should we enable the HTTP hack? */
37c19800e8SDoug Rabson int enable_http = -1;
38c19800e8SDoug Rabson 
39c19800e8SDoug Rabson /* Log over requests to the KDC */
40c19800e8SDoug Rabson const char *request_log;
41c19800e8SDoug Rabson 
42c19800e8SDoug Rabson /* A string describing on what ports to listen */
43c19800e8SDoug Rabson const char *port_str;
44c19800e8SDoug Rabson 
45c19800e8SDoug Rabson krb5_addresses explicit_addresses;
46c19800e8SDoug Rabson 
47*ae771770SStanislav Sedov size_t max_request_udp;
48*ae771770SStanislav Sedov size_t max_request_tcp;
49283d988cSMark Murray 
50283d988cSMark Murray /*
51283d988cSMark Murray  * a tuple describing on what to listen
52283d988cSMark Murray  */
53b528cefcSMark Murray 
54b528cefcSMark Murray struct port_desc{
55b528cefcSMark Murray     int family;
56b528cefcSMark Murray     int type;
57b528cefcSMark Murray     int port;
58b528cefcSMark Murray };
59b528cefcSMark Murray 
60283d988cSMark Murray /* the current ones */
61283d988cSMark Murray 
62b528cefcSMark Murray static struct port_desc *ports;
63*ae771770SStanislav Sedov static size_t num_ports;
64b528cefcSMark Murray 
65283d988cSMark Murray /*
66283d988cSMark Murray  * add `family, port, protocol' to the list with duplicate suppresion.
67283d988cSMark Murray  */
68283d988cSMark Murray 
69b528cefcSMark Murray static void
add_port(krb5_context context,int family,int port,const char * protocol)70c19800e8SDoug Rabson add_port(krb5_context context,
71c19800e8SDoug Rabson 	 int family, int port, const char *protocol)
72b528cefcSMark Murray {
73b528cefcSMark Murray     int type;
74*ae771770SStanislav Sedov     size_t i;
75b528cefcSMark Murray 
76b528cefcSMark Murray     if(strcmp(protocol, "udp") == 0)
77b528cefcSMark Murray 	type = SOCK_DGRAM;
78b528cefcSMark Murray     else if(strcmp(protocol, "tcp") == 0)
79b528cefcSMark Murray 	type = SOCK_STREAM;
80b528cefcSMark Murray     else
81b528cefcSMark Murray 	return;
82b528cefcSMark Murray     for(i = 0; i < num_ports; i++){
83b528cefcSMark Murray 	if(ports[i].type == type
84b528cefcSMark Murray 	   && ports[i].port == port
85b528cefcSMark Murray 	   && ports[i].family == family)
86b528cefcSMark Murray 	    return;
87b528cefcSMark Murray     }
88b528cefcSMark Murray     ports = realloc(ports, (num_ports + 1) * sizeof(*ports));
89283d988cSMark Murray     if (ports == NULL)
90283d988cSMark Murray 	krb5_err (context, 1, errno, "realloc");
91b528cefcSMark Murray     ports[num_ports].family = family;
92b528cefcSMark Murray     ports[num_ports].type   = type;
93b528cefcSMark Murray     ports[num_ports].port   = port;
94b528cefcSMark Murray     num_ports++;
95b528cefcSMark Murray }
96b528cefcSMark Murray 
97283d988cSMark Murray /*
98283d988cSMark Murray  * add a triple but with service -> port lookup
99283d988cSMark Murray  * (this prints warnings for stuff that does not exist)
100283d988cSMark Murray  */
101283d988cSMark Murray 
102b528cefcSMark Murray static void
add_port_service(krb5_context context,int family,const char * service,int port,const char * protocol)103c19800e8SDoug Rabson add_port_service(krb5_context context,
104c19800e8SDoug Rabson 		 int family, const char *service, int port,
105b528cefcSMark Murray 		 const char *protocol)
106b528cefcSMark Murray {
107b528cefcSMark Murray     port = krb5_getportbyname (context, service, protocol, port);
108c19800e8SDoug Rabson     add_port (context, family, port, protocol);
109b528cefcSMark Murray }
110b528cefcSMark Murray 
111283d988cSMark Murray /*
112283d988cSMark Murray  * add the port with service -> port lookup or string -> number
113283d988cSMark Murray  * (no warning is printed)
114283d988cSMark Murray  */
115283d988cSMark Murray 
116b528cefcSMark Murray static void
add_port_string(krb5_context context,int family,const char * str,const char * protocol)117c19800e8SDoug Rabson add_port_string (krb5_context context,
118c19800e8SDoug Rabson 		 int family, const char *str, const char *protocol)
119b528cefcSMark Murray {
120b528cefcSMark Murray     struct servent *sp;
121b528cefcSMark Murray     int port;
122b528cefcSMark Murray 
123c19800e8SDoug Rabson     sp = roken_getservbyname (str, protocol);
124b528cefcSMark Murray     if (sp != NULL) {
125b528cefcSMark Murray 	port = sp->s_port;
126b528cefcSMark Murray     } else {
127b528cefcSMark Murray 	char *end;
128b528cefcSMark Murray 
129c19800e8SDoug Rabson 	port = htons(strtol(str, &end, 0));
130c19800e8SDoug Rabson 	if (end == str)
131b528cefcSMark Murray 	    return;
132b528cefcSMark Murray     }
133c19800e8SDoug Rabson     add_port (context, family, port, protocol);
134b528cefcSMark Murray }
135b528cefcSMark Murray 
136283d988cSMark Murray /*
137283d988cSMark Murray  * add the standard collection of ports for `family'
138283d988cSMark Murray  */
139283d988cSMark Murray 
140b528cefcSMark Murray static void
add_standard_ports(krb5_context context,krb5_kdc_configuration * config,int family)141c19800e8SDoug Rabson add_standard_ports (krb5_context context,
142c19800e8SDoug Rabson 		    krb5_kdc_configuration *config,
143c19800e8SDoug Rabson 		    int family)
144b528cefcSMark Murray {
145c19800e8SDoug Rabson     add_port_service(context, family, "kerberos", 88, "udp");
146c19800e8SDoug Rabson     add_port_service(context, family, "kerberos", 88, "tcp");
147c19800e8SDoug Rabson     add_port_service(context, family, "kerberos-sec", 88, "udp");
148c19800e8SDoug Rabson     add_port_service(context, family, "kerberos-sec", 88, "tcp");
149b528cefcSMark Murray     if(enable_http)
150c19800e8SDoug Rabson 	add_port_service(context, family, "http", 80, "tcp");
151c19800e8SDoug Rabson     if(config->enable_kx509) {
152c19800e8SDoug Rabson 	add_port_service(context, family, "kca_service", 9878, "udp");
153c19800e8SDoug Rabson 	add_port_service(context, family, "kca_service", 9878, "tcp");
154c19800e8SDoug Rabson     }
155c19800e8SDoug Rabson 
156b528cefcSMark Murray }
157b528cefcSMark Murray 
158283d988cSMark Murray /*
159283d988cSMark Murray  * parse the set of space-delimited ports in `str' and add them.
160283d988cSMark Murray  * "+" => all the standard ones
161283d988cSMark Murray  * otherwise it's port|service[/protocol]
162283d988cSMark Murray  */
163283d988cSMark Murray 
164b528cefcSMark Murray static void
parse_ports(krb5_context context,krb5_kdc_configuration * config,const char * str)165c19800e8SDoug Rabson parse_ports(krb5_context context,
166c19800e8SDoug Rabson 	    krb5_kdc_configuration *config,
167c19800e8SDoug Rabson 	    const char *str)
168b528cefcSMark Murray {
169b528cefcSMark Murray     char *pos = NULL;
170b528cefcSMark Murray     char *p;
171b528cefcSMark Murray     char *str_copy = strdup (str);
172b528cefcSMark Murray 
173b528cefcSMark Murray     p = strtok_r(str_copy, " \t", &pos);
174b528cefcSMark Murray     while(p != NULL) {
175b528cefcSMark Murray 	if(strcmp(p, "+") == 0) {
176b528cefcSMark Murray #ifdef HAVE_IPV6
177c19800e8SDoug Rabson 	    add_standard_ports(context, config, AF_INET6);
178b528cefcSMark Murray #endif
179c19800e8SDoug Rabson 	    add_standard_ports(context, config, AF_INET);
180b528cefcSMark Murray 	} else {
181b528cefcSMark Murray 	    char *q = strchr(p, '/');
182b528cefcSMark Murray 	    if(q){
183b528cefcSMark Murray 		*q++ = 0;
184b528cefcSMark Murray #ifdef HAVE_IPV6
185c19800e8SDoug Rabson 		add_port_string(context, AF_INET6, p, q);
186b528cefcSMark Murray #endif
187c19800e8SDoug Rabson 		add_port_string(context, AF_INET, p, q);
188b528cefcSMark Murray 	    }else {
189b528cefcSMark Murray #ifdef HAVE_IPV6
190c19800e8SDoug Rabson 		add_port_string(context, AF_INET6, p, "udp");
191c19800e8SDoug Rabson 		add_port_string(context, AF_INET6, p, "tcp");
192b528cefcSMark Murray #endif
193c19800e8SDoug Rabson 		add_port_string(context, AF_INET, p, "udp");
194c19800e8SDoug Rabson 		add_port_string(context, AF_INET, p, "tcp");
195b528cefcSMark Murray 	    }
196b528cefcSMark Murray 	}
197b528cefcSMark Murray 
198b528cefcSMark Murray 	p = strtok_r(NULL, " \t", &pos);
199b528cefcSMark Murray     }
200b528cefcSMark Murray     free (str_copy);
201b528cefcSMark Murray }
202b528cefcSMark Murray 
203283d988cSMark Murray /*
204283d988cSMark Murray  * every socket we listen on
205283d988cSMark Murray  */
206283d988cSMark Murray 
207b528cefcSMark Murray struct descr {
208*ae771770SStanislav Sedov     krb5_socket_t s;
209b528cefcSMark Murray     int type;
210c19800e8SDoug Rabson     int port;
211b528cefcSMark Murray     unsigned char *buf;
212b528cefcSMark Murray     size_t size;
213b528cefcSMark Murray     size_t len;
214b528cefcSMark Murray     time_t timeout;
215d61f1c79SMark Murray     struct sockaddr_storage __ss;
216d61f1c79SMark Murray     struct sockaddr *sa;
2175e9cd1aeSAssar Westerlund     socklen_t sock_len;
218d61f1c79SMark Murray     char addr_string[128];
219b528cefcSMark Murray };
220b528cefcSMark Murray 
2215e9cd1aeSAssar Westerlund static void
init_descr(struct descr * d)2225e9cd1aeSAssar Westerlund init_descr(struct descr *d)
2235e9cd1aeSAssar Westerlund {
2245e9cd1aeSAssar Westerlund     memset(d, 0, sizeof(*d));
2255e9cd1aeSAssar Westerlund     d->sa = (struct sockaddr *)&d->__ss;
226*ae771770SStanislav Sedov     d->s = rk_INVALID_SOCKET;
2275e9cd1aeSAssar Westerlund }
2285e9cd1aeSAssar Westerlund 
2295e9cd1aeSAssar Westerlund /*
230bbd80c28SJacques Vidrine  * re-initialize all `n' ->sa in `d'.
2315e9cd1aeSAssar Westerlund  */
2325e9cd1aeSAssar Westerlund 
2335e9cd1aeSAssar Westerlund static void
reinit_descrs(struct descr * d,int n)2345e9cd1aeSAssar Westerlund reinit_descrs (struct descr *d, int n)
2355e9cd1aeSAssar Westerlund {
2365e9cd1aeSAssar Westerlund     int i;
2375e9cd1aeSAssar Westerlund 
2385e9cd1aeSAssar Westerlund     for (i = 0; i < n; ++i)
2395e9cd1aeSAssar Westerlund 	d[i].sa = (struct sockaddr *)&d[i].__ss;
2405e9cd1aeSAssar Westerlund }
2415e9cd1aeSAssar Westerlund 
242b528cefcSMark Murray /*
243b528cefcSMark Murray  * Create the socket (family, type, port) in `d'
244b528cefcSMark Murray  */
245b528cefcSMark Murray 
246b528cefcSMark Murray static void
init_socket(krb5_context context,krb5_kdc_configuration * config,struct descr * d,krb5_address * a,int family,int type,int port)247c19800e8SDoug Rabson init_socket(krb5_context context,
248c19800e8SDoug Rabson 	    krb5_kdc_configuration *config,
249c19800e8SDoug Rabson 	    struct descr *d, krb5_address *a, int family, int type, int port)
250b528cefcSMark Murray {
251b528cefcSMark Murray     krb5_error_code ret;
252b528cefcSMark Murray     struct sockaddr_storage __ss;
253b528cefcSMark Murray     struct sockaddr *sa = (struct sockaddr *)&__ss;
254c19800e8SDoug Rabson     krb5_socklen_t sa_size = sizeof(__ss);
255b528cefcSMark Murray 
2565e9cd1aeSAssar Westerlund     init_descr (d);
257b528cefcSMark Murray 
258adb0ddaeSAssar Westerlund     ret = krb5_addr2sockaddr (context, a, sa, &sa_size, port);
259b528cefcSMark Murray     if (ret) {
260283d988cSMark Murray 	krb5_warn(context, ret, "krb5_addr2sockaddr");
261*ae771770SStanislav Sedov 	rk_closesocket(d->s);
262*ae771770SStanislav Sedov 	d->s = rk_INVALID_SOCKET;
263b528cefcSMark Murray 	return;
264b528cefcSMark Murray     }
265b528cefcSMark Murray 
266b528cefcSMark Murray     if (sa->sa_family != family)
267b528cefcSMark Murray 	return;
268b528cefcSMark Murray 
269b528cefcSMark Murray     d->s = socket(family, type, 0);
270*ae771770SStanislav Sedov     if(rk_IS_BAD_SOCKET(d->s)){
271b528cefcSMark Murray 	krb5_warn(context, errno, "socket(%d, %d, 0)", family, type);
272*ae771770SStanislav Sedov 	d->s = rk_INVALID_SOCKET;
273b528cefcSMark Murray 	return;
274b528cefcSMark Murray     }
275b528cefcSMark Murray #if defined(HAVE_SETSOCKOPT) && defined(SOL_SOCKET) && defined(SO_REUSEADDR)
276b528cefcSMark Murray     {
277b528cefcSMark Murray 	int one = 1;
278b528cefcSMark Murray 	setsockopt(d->s, SOL_SOCKET, SO_REUSEADDR, (void *)&one, sizeof(one));
279b528cefcSMark Murray     }
280b528cefcSMark Murray #endif
281b528cefcSMark Murray     d->type = type;
282c19800e8SDoug Rabson     d->port = port;
283b528cefcSMark Murray 
284*ae771770SStanislav Sedov     if(rk_IS_SOCKET_ERROR(bind(d->s, sa, sa_size))){
285283d988cSMark Murray 	char a_str[256];
286283d988cSMark Murray 	size_t len;
287283d988cSMark Murray 
288283d988cSMark Murray 	krb5_print_address (a, a_str, sizeof(a_str), &len);
289283d988cSMark Murray 	krb5_warn(context, errno, "bind %s/%d", a_str, ntohs(port));
290*ae771770SStanislav Sedov 	rk_closesocket(d->s);
291*ae771770SStanislav Sedov 	d->s = rk_INVALID_SOCKET;
292b528cefcSMark Murray 	return;
293b528cefcSMark Murray     }
294*ae771770SStanislav Sedov     if(type == SOCK_STREAM && rk_IS_SOCKET_ERROR(listen(d->s, SOMAXCONN))){
295283d988cSMark Murray 	char a_str[256];
296283d988cSMark Murray 	size_t len;
297283d988cSMark Murray 
298283d988cSMark Murray 	krb5_print_address (a, a_str, sizeof(a_str), &len);
299283d988cSMark Murray 	krb5_warn(context, errno, "listen %s/%d", a_str, ntohs(port));
300*ae771770SStanislav Sedov 	rk_closesocket(d->s);
301*ae771770SStanislav Sedov 	d->s = rk_INVALID_SOCKET;
302b528cefcSMark Murray 	return;
303b528cefcSMark Murray     }
304b528cefcSMark Murray }
305b528cefcSMark Murray 
306b528cefcSMark Murray /*
307b528cefcSMark Murray  * Allocate descriptors for all the sockets that we should listen on
308b528cefcSMark Murray  * and return the number of them.
309b528cefcSMark Murray  */
310b528cefcSMark Murray 
311b528cefcSMark Murray static int
init_sockets(krb5_context context,krb5_kdc_configuration * config,struct descr ** desc)312c19800e8SDoug Rabson init_sockets(krb5_context context,
313c19800e8SDoug Rabson 	     krb5_kdc_configuration *config,
314c19800e8SDoug Rabson 	     struct descr **desc)
315b528cefcSMark Murray {
316b528cefcSMark Murray     krb5_error_code ret;
317*ae771770SStanislav Sedov     size_t i, j;
318b528cefcSMark Murray     struct descr *d;
319b528cefcSMark Murray     int num = 0;
320b528cefcSMark Murray     krb5_addresses addresses;
321b528cefcSMark Murray 
322283d988cSMark Murray     if (explicit_addresses.len) {
323283d988cSMark Murray 	addresses = explicit_addresses;
324283d988cSMark Murray     } else {
325b528cefcSMark Murray 	ret = krb5_get_all_server_addrs (context, &addresses);
326b528cefcSMark Murray 	if (ret)
327b528cefcSMark Murray 	    krb5_err (context, 1, ret, "krb5_get_all_server_addrs");
328283d988cSMark Murray     }
329c19800e8SDoug Rabson     parse_ports(context, config, port_str);
330b528cefcSMark Murray     d = malloc(addresses.len * num_ports * sizeof(*d));
331b528cefcSMark Murray     if (d == NULL)
3325e9cd1aeSAssar Westerlund 	krb5_errx(context, 1, "malloc(%lu) failed",
3335e9cd1aeSAssar Westerlund 		  (unsigned long)num_ports * sizeof(*d));
334b528cefcSMark Murray 
335b528cefcSMark Murray     for (i = 0; i < num_ports; i++){
336b528cefcSMark Murray 	for (j = 0; j < addresses.len; ++j) {
337c19800e8SDoug Rabson 	    init_socket(context, config, &d[num], &addresses.val[j],
338b528cefcSMark Murray 			ports[i].family, ports[i].type, ports[i].port);
339*ae771770SStanislav Sedov 	    if(d[num].s != rk_INVALID_SOCKET){
340b528cefcSMark Murray 		char a_str[80];
341b528cefcSMark Murray 		size_t len;
342b528cefcSMark Murray 
343b528cefcSMark Murray 		krb5_print_address (&addresses.val[j], a_str,
344b528cefcSMark Murray 				    sizeof(a_str), &len);
345b528cefcSMark Murray 
346c19800e8SDoug Rabson 		kdc_log(context, config, 5, "listening on %s port %u/%s",
347b528cefcSMark Murray 			a_str,
348b528cefcSMark Murray 			ntohs(ports[i].port),
349b528cefcSMark Murray 			(ports[i].type == SOCK_STREAM) ? "tcp" : "udp");
350b528cefcSMark Murray 		/* XXX */
351b528cefcSMark Murray 		num++;
352b528cefcSMark Murray 	    }
353b528cefcSMark Murray 	}
354b528cefcSMark Murray     }
355b528cefcSMark Murray     krb5_free_addresses (context, &addresses);
356b528cefcSMark Murray     d = realloc(d, num * sizeof(*d));
357b528cefcSMark Murray     if (d == NULL && num != 0)
3585e9cd1aeSAssar Westerlund 	krb5_errx(context, 1, "realloc(%lu) failed",
3595e9cd1aeSAssar Westerlund 		  (unsigned long)num * sizeof(*d));
3605e9cd1aeSAssar Westerlund     reinit_descrs (d, num);
361b528cefcSMark Murray     *desc = d;
362b528cefcSMark Murray     return num;
363b528cefcSMark Murray }
364b528cefcSMark Murray 
365283d988cSMark Murray /*
366c19800e8SDoug Rabson  *
367283d988cSMark Murray  */
368b528cefcSMark Murray 
369c19800e8SDoug Rabson static const char *
descr_type(struct descr * d)370c19800e8SDoug Rabson descr_type(struct descr *d)
371b528cefcSMark Murray {
372c19800e8SDoug Rabson     if (d->type == SOCK_DGRAM)
373c19800e8SDoug Rabson 	return "udp";
374c19800e8SDoug Rabson     else if (d->type == SOCK_STREAM)
375c19800e8SDoug Rabson 	return "tcp";
376c19800e8SDoug Rabson     return "unknown";
377b528cefcSMark Murray }
378b528cefcSMark Murray 
379b528cefcSMark Murray static void
addr_to_string(krb5_context context,struct sockaddr * addr,size_t addr_len,char * str,size_t len)380c19800e8SDoug Rabson addr_to_string(krb5_context context,
381c19800e8SDoug Rabson 	       struct sockaddr *addr, size_t addr_len, char *str, size_t len)
382b528cefcSMark Murray {
383b528cefcSMark Murray     krb5_address a;
3848373020dSJacques Vidrine     if(krb5_sockaddr2address(context, addr, &a) == 0) {
385b528cefcSMark Murray 	if(krb5_print_address(&a, str, len, &len) == 0) {
386b528cefcSMark Murray 	    krb5_free_address(context, &a);
387b528cefcSMark Murray 	    return;
388b528cefcSMark Murray 	}
389b528cefcSMark Murray 	krb5_free_address(context, &a);
3908373020dSJacques Vidrine     }
391b528cefcSMark Murray     snprintf(str, len, "<family=%d>", addr->sa_family);
392b528cefcSMark Murray }
393b528cefcSMark Murray 
394d61f1c79SMark Murray /*
395c19800e8SDoug Rabson  *
396c19800e8SDoug Rabson  */
397c19800e8SDoug Rabson 
398c19800e8SDoug Rabson static void
send_reply(krb5_context context,krb5_kdc_configuration * config,krb5_boolean prependlength,struct descr * d,krb5_data * reply)399c19800e8SDoug Rabson send_reply(krb5_context context,
400c19800e8SDoug Rabson 	   krb5_kdc_configuration *config,
401c19800e8SDoug Rabson 	   krb5_boolean prependlength,
402c19800e8SDoug Rabson 	   struct descr *d,
403c19800e8SDoug Rabson 	   krb5_data *reply)
404c19800e8SDoug Rabson {
405c19800e8SDoug Rabson     kdc_log(context, config, 5,
406c19800e8SDoug Rabson 	    "sending %lu bytes to %s", (unsigned long)reply->length,
407c19800e8SDoug Rabson 	    d->addr_string);
408c19800e8SDoug Rabson     if(prependlength){
409c19800e8SDoug Rabson 	unsigned char l[4];
410c19800e8SDoug Rabson 	l[0] = (reply->length >> 24) & 0xff;
411c19800e8SDoug Rabson 	l[1] = (reply->length >> 16) & 0xff;
412c19800e8SDoug Rabson 	l[2] = (reply->length >> 8) & 0xff;
413c19800e8SDoug Rabson 	l[3] = reply->length & 0xff;
414*ae771770SStanislav Sedov 	if(rk_IS_SOCKET_ERROR(sendto(d->s, l, sizeof(l), 0, d->sa, d->sock_len))) {
415c19800e8SDoug Rabson 	    kdc_log (context, config,
416*ae771770SStanislav Sedov 		     0, "sendto(%s): %s", d->addr_string,
417*ae771770SStanislav Sedov 		     strerror(rk_SOCK_ERRNO));
418c19800e8SDoug Rabson 	    return;
419c19800e8SDoug Rabson 	}
420c19800e8SDoug Rabson     }
421*ae771770SStanislav Sedov     if(rk_IS_SOCKET_ERROR(sendto(d->s, reply->data, reply->length, 0, d->sa, d->sock_len))) {
422*ae771770SStanislav Sedov 	kdc_log (context, config, 0, "sendto(%s): %s", d->addr_string,
423*ae771770SStanislav Sedov 		 strerror(rk_SOCK_ERRNO));
424c19800e8SDoug Rabson 	return;
425c19800e8SDoug Rabson     }
426c19800e8SDoug Rabson }
427c19800e8SDoug Rabson 
428c19800e8SDoug Rabson /*
429d61f1c79SMark Murray  * Handle the request in `buf, len' to socket `d'
430d61f1c79SMark Murray  */
431d61f1c79SMark Murray 
432b528cefcSMark Murray static void
do_request(krb5_context context,krb5_kdc_configuration * config,void * buf,size_t len,krb5_boolean prependlength,struct descr * d)433c19800e8SDoug Rabson do_request(krb5_context context,
434c19800e8SDoug Rabson 	   krb5_kdc_configuration *config,
435c19800e8SDoug Rabson 	   void *buf, size_t len, krb5_boolean prependlength,
436d61f1c79SMark Murray 	   struct descr *d)
437b528cefcSMark Murray {
438b528cefcSMark Murray     krb5_error_code ret;
439b528cefcSMark Murray     krb5_data reply;
440c19800e8SDoug Rabson     int datagram_reply = (d->type == SOCK_DGRAM);
441b528cefcSMark Murray 
442c19800e8SDoug Rabson     krb5_kdc_update_time(NULL);
443c19800e8SDoug Rabson 
444c19800e8SDoug Rabson     krb5_data_zero(&reply);
445c19800e8SDoug Rabson     ret = krb5_kdc_process_request(context, config,
446c19800e8SDoug Rabson 				   buf, len, &reply, &prependlength,
447c19800e8SDoug Rabson 				   d->addr_string, d->sa,
448c19800e8SDoug Rabson 				   datagram_reply);
449c19800e8SDoug Rabson     if(request_log)
450c19800e8SDoug Rabson 	krb5_kdc_save_request(context, request_log, buf, len, &reply, d->sa);
451b528cefcSMark Murray     if(reply.length){
452c19800e8SDoug Rabson 	send_reply(context, config, prependlength, d, &reply);
453b528cefcSMark Murray 	krb5_data_free(&reply);
454b528cefcSMark Murray     }
455b528cefcSMark Murray     if(ret)
456c19800e8SDoug Rabson 	kdc_log(context, config, 0,
457c19800e8SDoug Rabson 		"Failed processing %lu byte request from %s",
458d61f1c79SMark Murray 		(unsigned long)len, d->addr_string);
459b528cefcSMark Murray }
460b528cefcSMark Murray 
461d61f1c79SMark Murray /*
462d61f1c79SMark Murray  * Handle incoming data to the UDP socket in `d'
463d61f1c79SMark Murray  */
464d61f1c79SMark Murray 
465b528cefcSMark Murray static void
handle_udp(krb5_context context,krb5_kdc_configuration * config,struct descr * d)466c19800e8SDoug Rabson handle_udp(krb5_context context,
467c19800e8SDoug Rabson 	   krb5_kdc_configuration *config,
468c19800e8SDoug Rabson 	   struct descr *d)
469b528cefcSMark Murray {
470b528cefcSMark Murray     unsigned char *buf;
471*ae771770SStanislav Sedov     ssize_t n;
472b528cefcSMark Murray 
473*ae771770SStanislav Sedov     buf = malloc(max_request_udp);
474b528cefcSMark Murray     if(buf == NULL){
475*ae771770SStanislav Sedov 	kdc_log(context, config, 0, "Failed to allocate %lu bytes", (unsigned long)max_request_udp);
476b528cefcSMark Murray 	return;
477b528cefcSMark Murray     }
478b528cefcSMark Murray 
479d61f1c79SMark Murray     d->sock_len = sizeof(d->__ss);
480*ae771770SStanislav Sedov     n = recvfrom(d->s, buf, max_request_udp, 0, d->sa, &d->sock_len);
481*ae771770SStanislav Sedov     if(rk_IS_SOCKET_ERROR(n))
482*ae771770SStanislav Sedov 	krb5_warn(context, rk_SOCK_ERRNO, "recvfrom");
483d61f1c79SMark Murray     else {
484c19800e8SDoug Rabson 	addr_to_string (context, d->sa, d->sock_len,
485d61f1c79SMark Murray 			d->addr_string, sizeof(d->addr_string));
486*ae771770SStanislav Sedov 	if ((size_t)n == max_request_udp) {
487*ae771770SStanislav Sedov 	    krb5_data data;
488*ae771770SStanislav Sedov 	    krb5_warn(context, errno,
489*ae771770SStanislav Sedov 		      "recvfrom: truncated packet from %s, asking for TCP",
490*ae771770SStanislav Sedov 		      d->addr_string);
491*ae771770SStanislav Sedov 	    krb5_mk_error(context,
492*ae771770SStanislav Sedov 			  KRB5KRB_ERR_RESPONSE_TOO_BIG,
493*ae771770SStanislav Sedov 			  NULL,
494*ae771770SStanislav Sedov 			  NULL,
495*ae771770SStanislav Sedov 			  NULL,
496*ae771770SStanislav Sedov 			  NULL,
497*ae771770SStanislav Sedov 			  NULL,
498*ae771770SStanislav Sedov 			  NULL,
499*ae771770SStanislav Sedov 			  &data);
500*ae771770SStanislav Sedov 	    send_reply(context, config, FALSE, d, &data);
501*ae771770SStanislav Sedov 	    krb5_data_free(&data);
502*ae771770SStanislav Sedov 	} else {
503c19800e8SDoug Rabson 	    do_request(context, config, buf, n, FALSE, d);
504b528cefcSMark Murray 	}
505*ae771770SStanislav Sedov     }
506b528cefcSMark Murray     free (buf);
507b528cefcSMark Murray }
508b528cefcSMark Murray 
509b528cefcSMark Murray static void
clear_descr(struct descr * d)510b528cefcSMark Murray clear_descr(struct descr *d)
511b528cefcSMark Murray {
512b528cefcSMark Murray     if(d->buf)
513b528cefcSMark Murray 	memset(d->buf, 0, d->size);
514b528cefcSMark Murray     d->len = 0;
515*ae771770SStanislav Sedov     if(d->s != rk_INVALID_SOCKET)
516*ae771770SStanislav Sedov 	rk_closesocket(d->s);
517*ae771770SStanislav Sedov     d->s = rk_INVALID_SOCKET;
518b528cefcSMark Murray }
519b528cefcSMark Murray 
520b528cefcSMark Murray 
521b528cefcSMark Murray /* remove HTTP %-quoting from buf */
522b528cefcSMark Murray static int
de_http(char * buf)523b528cefcSMark Murray de_http(char *buf)
524b528cefcSMark Murray {
525c19800e8SDoug Rabson     unsigned char *p, *q;
526c19800e8SDoug Rabson     for(p = q = (unsigned char *)buf; *p; p++, q++) {
5275bda878eSJacques Vidrine 	if(*p == '%' && isxdigit(p[1]) && isxdigit(p[2])) {
528b528cefcSMark Murray 	    unsigned int x;
529c19800e8SDoug Rabson 	    if(sscanf((char *)p + 1, "%2x", &x) != 1)
530b528cefcSMark Murray 		return -1;
531b528cefcSMark Murray 	    *q = x;
532b528cefcSMark Murray 	    p += 2;
533b528cefcSMark Murray 	} else
534b528cefcSMark Murray 	    *q = *p;
535b528cefcSMark Murray     }
536b528cefcSMark Murray     *q = '\0';
537b528cefcSMark Murray     return 0;
538b528cefcSMark Murray }
539b528cefcSMark Murray 
540b528cefcSMark Murray #define TCP_TIMEOUT 4
541b528cefcSMark Murray 
542b528cefcSMark Murray /*
5435e9cd1aeSAssar Westerlund  * accept a new TCP connection on `d[parent]' and store it in `d[child]'
544b528cefcSMark Murray  */
545b528cefcSMark Murray 
546b528cefcSMark Murray static void
add_new_tcp(krb5_context context,krb5_kdc_configuration * config,struct descr * d,int parent,int child)547c19800e8SDoug Rabson add_new_tcp (krb5_context context,
548c19800e8SDoug Rabson 	     krb5_kdc_configuration *config,
549c19800e8SDoug Rabson 	     struct descr *d, int parent, int child)
550b528cefcSMark Murray {
551*ae771770SStanislav Sedov     krb5_socket_t s;
552b528cefcSMark Murray 
5535e9cd1aeSAssar Westerlund     if (child == -1)
5545e9cd1aeSAssar Westerlund 	return;
5555e9cd1aeSAssar Westerlund 
5565e9cd1aeSAssar Westerlund     d[child].sock_len = sizeof(d[child].__ss);
5575e9cd1aeSAssar Westerlund     s = accept(d[parent].s, d[child].sa, &d[child].sock_len);
558*ae771770SStanislav Sedov     if(rk_IS_BAD_SOCKET(s)) {
559*ae771770SStanislav Sedov 	krb5_warn(context, rk_SOCK_ERRNO, "accept");
560b528cefcSMark Murray 	return;
561b528cefcSMark Murray     }
5625e9cd1aeSAssar Westerlund 
563*ae771770SStanislav Sedov #ifdef FD_SETSIZE
5645e9cd1aeSAssar Westerlund     if (s >= FD_SETSIZE) {
5655e9cd1aeSAssar Westerlund 	krb5_warnx(context, "socket FD too large");
566*ae771770SStanislav Sedov 	rk_closesocket (s);
567b528cefcSMark Murray 	return;
568b528cefcSMark Murray     }
569*ae771770SStanislav Sedov #endif
570b528cefcSMark Murray 
5715e9cd1aeSAssar Westerlund     d[child].s = s;
5725e9cd1aeSAssar Westerlund     d[child].timeout = time(NULL) + TCP_TIMEOUT;
5735e9cd1aeSAssar Westerlund     d[child].type = SOCK_STREAM;
574c19800e8SDoug Rabson     addr_to_string (context,
575c19800e8SDoug Rabson 		    d[child].sa, d[child].sock_len,
5765e9cd1aeSAssar Westerlund 		    d[child].addr_string, sizeof(d[child].addr_string));
577b528cefcSMark Murray }
578b528cefcSMark Murray 
579b528cefcSMark Murray /*
580b528cefcSMark Murray  * Grow `d' to handle at least `n'.
581b528cefcSMark Murray  * Return != 0 if fails
582b528cefcSMark Murray  */
583b528cefcSMark Murray 
584b528cefcSMark Murray static int
grow_descr(krb5_context context,krb5_kdc_configuration * config,struct descr * d,size_t n)585c19800e8SDoug Rabson grow_descr (krb5_context context,
586c19800e8SDoug Rabson 	    krb5_kdc_configuration *config,
587c19800e8SDoug Rabson 	    struct descr *d, size_t n)
588b528cefcSMark Murray {
589b528cefcSMark Murray     if (d->size - d->len < n) {
590b528cefcSMark Murray 	unsigned char *tmp;
5911c43270aSJacques Vidrine 	size_t grow;
592b528cefcSMark Murray 
5931c43270aSJacques Vidrine 	grow = max(1024, d->len + n);
594*ae771770SStanislav Sedov 	if (d->size + grow > max_request_tcp) {
595c19800e8SDoug Rabson 	    kdc_log(context, config, 0, "Request exceeds max request size (%lu bytes).",
5961c43270aSJacques Vidrine 		    (unsigned long)d->size + grow);
597b528cefcSMark Murray 	    clear_descr(d);
598b528cefcSMark Murray 	    return -1;
599b528cefcSMark Murray 	}
6001c43270aSJacques Vidrine 	tmp = realloc (d->buf, d->size + grow);
601b528cefcSMark Murray 	if (tmp == NULL) {
602c19800e8SDoug Rabson 	    kdc_log(context, config, 0, "Failed to re-allocate %lu bytes.",
6031c43270aSJacques Vidrine 		    (unsigned long)d->size + grow);
604b528cefcSMark Murray 	    clear_descr(d);
605b528cefcSMark Murray 	    return -1;
606b528cefcSMark Murray 	}
6071c43270aSJacques Vidrine 	d->size += grow;
608b528cefcSMark Murray 	d->buf = tmp;
609b528cefcSMark Murray     }
610b528cefcSMark Murray     return 0;
611b528cefcSMark Murray }
612b528cefcSMark Murray 
613b528cefcSMark Murray /*
614b528cefcSMark Murray  * Try to handle the TCP data at `d->buf, d->len'.
615b528cefcSMark Murray  * Return -1 if failed, 0 if succesful, and 1 if data is complete.
616b528cefcSMark Murray  */
617b528cefcSMark Murray 
618b528cefcSMark Murray static int
handle_vanilla_tcp(krb5_context context,krb5_kdc_configuration * config,struct descr * d)619c19800e8SDoug Rabson handle_vanilla_tcp (krb5_context context,
620c19800e8SDoug Rabson 		    krb5_kdc_configuration *config,
621c19800e8SDoug Rabson 		    struct descr *d)
622b528cefcSMark Murray {
623b528cefcSMark Murray     krb5_storage *sp;
624c19800e8SDoug Rabson     uint32_t len;
625b528cefcSMark Murray 
626b528cefcSMark Murray     sp = krb5_storage_from_mem(d->buf, d->len);
627b528cefcSMark Murray     if (sp == NULL) {
628c19800e8SDoug Rabson 	kdc_log (context, config, 0, "krb5_storage_from_mem failed");
629b528cefcSMark Murray 	return -1;
630b528cefcSMark Murray     }
631c19800e8SDoug Rabson     krb5_ret_uint32(sp, &len);
632b528cefcSMark Murray     krb5_storage_free(sp);
633b528cefcSMark Murray     if(d->len - 4 >= len) {
6345e9cd1aeSAssar Westerlund 	memmove(d->buf, d->buf + 4, d->len - 4);
635c19800e8SDoug Rabson 	d->len -= 4;
636b528cefcSMark Murray 	return 1;
637b528cefcSMark Murray     }
638b528cefcSMark Murray     return 0;
639b528cefcSMark Murray }
640b528cefcSMark Murray 
641b528cefcSMark Murray /*
642b528cefcSMark Murray  * Try to handle the TCP/HTTP data at `d->buf, d->len'.
643b528cefcSMark Murray  * Return -1 if failed, 0 if succesful, and 1 if data is complete.
644b528cefcSMark Murray  */
645b528cefcSMark Murray 
646b528cefcSMark Murray static int
handle_http_tcp(krb5_context context,krb5_kdc_configuration * config,struct descr * d)647c19800e8SDoug Rabson handle_http_tcp (krb5_context context,
648c19800e8SDoug Rabson 		 krb5_kdc_configuration *config,
649c19800e8SDoug Rabson 		 struct descr *d)
650b528cefcSMark Murray {
651b528cefcSMark Murray     char *s, *p, *t;
652b528cefcSMark Murray     void *data;
653b528cefcSMark Murray     char *proto;
654b528cefcSMark Murray     int len;
655b528cefcSMark Murray 
656b528cefcSMark Murray     s = (char *)d->buf;
657b528cefcSMark Murray 
658*ae771770SStanislav Sedov     /* If its a multi line query, truncate off the first line */
659b528cefcSMark Murray     p = strstr(s, "\r\n");
660*ae771770SStanislav Sedov     if (p)
661b528cefcSMark Murray 	*p = 0;
662b528cefcSMark Murray 
663b528cefcSMark Murray     p = NULL;
664b528cefcSMark Murray     t = strtok_r(s, " \t", &p);
665b528cefcSMark Murray     if (t == NULL) {
666*ae771770SStanislav Sedov 	kdc_log(context, config, 0,
667*ae771770SStanislav Sedov 		"Missing HTTP operand (GET) request from %s", d->addr_string);
668b528cefcSMark Murray 	return -1;
669b528cefcSMark Murray     }
670*ae771770SStanislav Sedov 
671b528cefcSMark Murray     t = strtok_r(NULL, " \t", &p);
672b528cefcSMark Murray     if(t == NULL) {
673*ae771770SStanislav Sedov 	kdc_log(context, config, 0,
674*ae771770SStanislav Sedov 		"Missing HTTP GET data in request from %s", d->addr_string);
675b528cefcSMark Murray 	return -1;
676b528cefcSMark Murray     }
677*ae771770SStanislav Sedov 
678b528cefcSMark Murray     data = malloc(strlen(t));
679b528cefcSMark Murray     if (data == NULL) {
680c19800e8SDoug Rabson 	kdc_log(context, config, 0, "Failed to allocate %lu bytes",
681adb0ddaeSAssar Westerlund 		(unsigned long)strlen(t));
682b528cefcSMark Murray 	return -1;
683b528cefcSMark Murray     }
684b528cefcSMark Murray     if(*t == '/')
685b528cefcSMark Murray 	t++;
686b528cefcSMark Murray     if(de_http(t) != 0) {
687c19800e8SDoug Rabson 	kdc_log(context, config, 0, "Malformed HTTP request from %s", d->addr_string);
688c19800e8SDoug Rabson 	kdc_log(context, config, 5, "HTTP request: %s", t);
689b528cefcSMark Murray 	free(data);
690b528cefcSMark Murray 	return -1;
691b528cefcSMark Murray     }
692b528cefcSMark Murray     proto = strtok_r(NULL, " \t", &p);
693b528cefcSMark Murray     if (proto == NULL) {
694c19800e8SDoug Rabson 	kdc_log(context, config, 0, "Malformed HTTP request from %s", d->addr_string);
695b528cefcSMark Murray 	free(data);
696b528cefcSMark Murray 	return -1;
697b528cefcSMark Murray     }
698b528cefcSMark Murray     len = base64_decode(t, data);
699b528cefcSMark Murray     if(len <= 0){
700b528cefcSMark Murray 	const char *msg =
701b528cefcSMark Murray 	    " 404 Not found\r\n"
702b528cefcSMark Murray 	    "Server: Heimdal/" VERSION "\r\n"
7038373020dSJacques Vidrine 	    "Cache-Control: no-cache\r\n"
7048373020dSJacques Vidrine 	    "Pragma: no-cache\r\n"
705b528cefcSMark Murray 	    "Content-type: text/html\r\n"
706b528cefcSMark Murray 	    "Content-transfer-encoding: 8bit\r\n\r\n"
707b528cefcSMark Murray 	    "<TITLE>404 Not found</TITLE>\r\n"
708b528cefcSMark Murray 	    "<H1>404 Not found</H1>\r\n"
709b528cefcSMark Murray 	    "That page doesn't exist, maybe you are looking for "
710c19800e8SDoug Rabson 	    "<A HREF=\"http://www.h5l.org/\">Heimdal</A>?\r\n";
711c19800e8SDoug Rabson 	kdc_log(context, config, 0, "HTTP request from %s is non KDC request", d->addr_string);
712c19800e8SDoug Rabson 	kdc_log(context, config, 5, "HTTP request: %s", t);
713b528cefcSMark Murray 	free(data);
714*ae771770SStanislav Sedov 	if (rk_IS_SOCKET_ERROR(send(d->s, proto, strlen(proto), 0))) {
715c19800e8SDoug Rabson 	    kdc_log(context, config, 0, "HTTP write failed: %s: %s",
716*ae771770SStanislav Sedov 		    d->addr_string, strerror(rk_SOCK_ERRNO));
717c19800e8SDoug Rabson 	    return -1;
718c19800e8SDoug Rabson 	}
719*ae771770SStanislav Sedov 	if (rk_IS_SOCKET_ERROR(send(d->s, msg, strlen(msg), 0))) {
720c19800e8SDoug Rabson 	    kdc_log(context, config, 0, "HTTP write failed: %s: %s",
721*ae771770SStanislav Sedov 		    d->addr_string, strerror(rk_SOCK_ERRNO));
722c19800e8SDoug Rabson 	    return -1;
723c19800e8SDoug Rabson 	}
724b528cefcSMark Murray 	return -1;
725b528cefcSMark Murray     }
726b528cefcSMark Murray     {
727b528cefcSMark Murray 	const char *msg =
728b528cefcSMark Murray 	    " 200 OK\r\n"
729b528cefcSMark Murray 	    "Server: Heimdal/" VERSION "\r\n"
7308373020dSJacques Vidrine 	    "Cache-Control: no-cache\r\n"
7318373020dSJacques Vidrine 	    "Pragma: no-cache\r\n"
732b528cefcSMark Murray 	    "Content-type: application/octet-stream\r\n"
733b528cefcSMark Murray 	    "Content-transfer-encoding: binary\r\n\r\n";
734*ae771770SStanislav Sedov 	if (rk_IS_SOCKET_ERROR(send(d->s, proto, strlen(proto), 0))) {
735*ae771770SStanislav Sedov 	    free(data);
736c19800e8SDoug Rabson 	    kdc_log(context, config, 0, "HTTP write failed: %s: %s",
737*ae771770SStanislav Sedov 		    d->addr_string, strerror(rk_SOCK_ERRNO));
738c19800e8SDoug Rabson 	    return -1;
739c19800e8SDoug Rabson 	}
740*ae771770SStanislav Sedov 	if (rk_IS_SOCKET_ERROR(send(d->s, msg, strlen(msg), 0))) {
741*ae771770SStanislav Sedov 	    free(data);
742c19800e8SDoug Rabson 	    kdc_log(context, config, 0, "HTTP write failed: %s: %s",
743*ae771770SStanislav Sedov 		    d->addr_string, strerror(rk_SOCK_ERRNO));
744c19800e8SDoug Rabson 	    return -1;
745c19800e8SDoug Rabson 	}
746b528cefcSMark Murray     }
747*ae771770SStanislav Sedov     if ((size_t)len > d->len)
748*ae771770SStanislav Sedov         len = d->len;
749b528cefcSMark Murray     memcpy(d->buf, data, len);
750b528cefcSMark Murray     d->len = len;
751b528cefcSMark Murray     free(data);
752b528cefcSMark Murray     return 1;
753b528cefcSMark Murray }
754b528cefcSMark Murray 
755b528cefcSMark Murray /*
756b528cefcSMark Murray  * Handle incoming data to the TCP socket in `d[index]'
757b528cefcSMark Murray  */
758b528cefcSMark Murray 
759b528cefcSMark Murray static void
handle_tcp(krb5_context context,krb5_kdc_configuration * config,struct descr * d,int idx,int min_free)760c19800e8SDoug Rabson handle_tcp(krb5_context context,
761c19800e8SDoug Rabson 	   krb5_kdc_configuration *config,
762c19800e8SDoug Rabson 	   struct descr *d, int idx, int min_free)
763b528cefcSMark Murray {
764b528cefcSMark Murray     unsigned char buf[1024];
765b528cefcSMark Murray     int n;
766b528cefcSMark Murray     int ret = 0;
767b528cefcSMark Murray 
768c19800e8SDoug Rabson     if (d[idx].timeout == 0) {
769c19800e8SDoug Rabson 	add_new_tcp (context, config, d, idx, min_free);
770b528cefcSMark Murray 	return;
771b528cefcSMark Murray     }
772b528cefcSMark Murray 
773c19800e8SDoug Rabson     n = recvfrom(d[idx].s, buf, sizeof(buf), 0, NULL, NULL);
774*ae771770SStanislav Sedov     if(rk_IS_SOCKET_ERROR(n)){
775*ae771770SStanislav Sedov 	krb5_warn(context, rk_SOCK_ERRNO, "recvfrom failed from %s to %s/%d",
776c19800e8SDoug Rabson 		  d[idx].addr_string, descr_type(d + idx),
777c19800e8SDoug Rabson 		  ntohs(d[idx].port));
778b528cefcSMark Murray 	return;
7791c43270aSJacques Vidrine     } else if (n == 0) {
7808d4ba808SJacques Vidrine 	krb5_warnx(context, "connection closed before end of data after %lu "
781c19800e8SDoug Rabson 		   "bytes from %s to %s/%d", (unsigned long)d[idx].len,
782c19800e8SDoug Rabson 		   d[idx].addr_string, descr_type(d + idx),
783c19800e8SDoug Rabson 		   ntohs(d[idx].port));
784c19800e8SDoug Rabson 	clear_descr (d + idx);
7851c43270aSJacques Vidrine 	return;
786b528cefcSMark Murray     }
787c19800e8SDoug Rabson     if (grow_descr (context, config, &d[idx], n))
788b528cefcSMark Murray 	return;
789c19800e8SDoug Rabson     memcpy(d[idx].buf + d[idx].len, buf, n);
790c19800e8SDoug Rabson     d[idx].len += n;
791c19800e8SDoug Rabson     if(d[idx].len > 4 && d[idx].buf[0] == 0) {
792c19800e8SDoug Rabson 	ret = handle_vanilla_tcp (context, config, &d[idx]);
793b528cefcSMark Murray     } else if(enable_http &&
794c19800e8SDoug Rabson 	      d[idx].len >= 4 &&
795c19800e8SDoug Rabson 	      strncmp((char *)d[idx].buf, "GET ", 4) == 0 &&
796c19800e8SDoug Rabson 	      strncmp((char *)d[idx].buf + d[idx].len - 4,
797b528cefcSMark Murray 		      "\r\n\r\n", 4) == 0) {
798*ae771770SStanislav Sedov 
799*ae771770SStanislav Sedov         /* remove the trailing \r\n\r\n so the string is NUL terminated */
800*ae771770SStanislav Sedov         d[idx].buf[d[idx].len - 4] = '\0';
801*ae771770SStanislav Sedov 
802c19800e8SDoug Rabson 	ret = handle_http_tcp (context, config, &d[idx]);
803b528cefcSMark Murray 	if (ret < 0)
804c19800e8SDoug Rabson 	    clear_descr (d + idx);
805c19800e8SDoug Rabson     } else if (d[idx].len > 4) {
806c19800e8SDoug Rabson 	kdc_log (context, config,
807c19800e8SDoug Rabson 		 0, "TCP data of strange type from %s to %s/%d",
808c19800e8SDoug Rabson 		 d[idx].addr_string, descr_type(d + idx),
809c19800e8SDoug Rabson 		 ntohs(d[idx].port));
810c19800e8SDoug Rabson 	if (d[idx].buf[0] & 0x80) {
811c19800e8SDoug Rabson 	    krb5_data reply;
812c19800e8SDoug Rabson 
813c19800e8SDoug Rabson 	    kdc_log (context, config, 0, "TCP extension not supported");
814c19800e8SDoug Rabson 
815c19800e8SDoug Rabson 	    ret = krb5_mk_error(context,
816c19800e8SDoug Rabson 				KRB5KRB_ERR_FIELD_TOOLONG,
817c19800e8SDoug Rabson 				NULL,
818c19800e8SDoug Rabson 				NULL,
819c19800e8SDoug Rabson 				NULL,
820c19800e8SDoug Rabson 				NULL,
821c19800e8SDoug Rabson 				NULL,
822c19800e8SDoug Rabson 				NULL,
823c19800e8SDoug Rabson 				&reply);
824c19800e8SDoug Rabson 	    if (ret == 0) {
825c19800e8SDoug Rabson 		send_reply(context, config, TRUE, d + idx, &reply);
826c19800e8SDoug Rabson 		krb5_data_free(&reply);
827c19800e8SDoug Rabson 	    }
828c19800e8SDoug Rabson 	}
829c19800e8SDoug Rabson 	clear_descr(d + idx);
830b528cefcSMark Murray 	return;
831b528cefcSMark Murray     }
832b528cefcSMark Murray     if (ret < 0)
833b528cefcSMark Murray 	return;
834b528cefcSMark Murray     else if (ret == 1) {
835c19800e8SDoug Rabson 	do_request(context, config,
836c19800e8SDoug Rabson 		   d[idx].buf, d[idx].len, TRUE, &d[idx]);
837c19800e8SDoug Rabson 	clear_descr(d + idx);
838b528cefcSMark Murray     }
839b528cefcSMark Murray }
840b528cefcSMark Murray 
841b528cefcSMark Murray void
loop(krb5_context context,krb5_kdc_configuration * config)842c19800e8SDoug Rabson loop(krb5_context context,
843c19800e8SDoug Rabson      krb5_kdc_configuration *config)
844b528cefcSMark Murray {
845b528cefcSMark Murray     struct descr *d;
846*ae771770SStanislav Sedov     unsigned int ndescr;
847b528cefcSMark Murray 
848c19800e8SDoug Rabson     ndescr = init_sockets(context, config, &d);
849b528cefcSMark Murray     if(ndescr <= 0)
850b528cefcSMark Murray 	krb5_errx(context, 1, "No sockets!");
851c19800e8SDoug Rabson     kdc_log(context, config, 0, "KDC started");
852b528cefcSMark Murray     while(exit_flag == 0){
853b528cefcSMark Murray 	struct timeval tmout;
854b528cefcSMark Murray 	fd_set fds;
855b528cefcSMark Murray 	int min_free = -1;
856b528cefcSMark Murray 	int max_fd = 0;
857*ae771770SStanislav Sedov 	size_t i;
8585e9cd1aeSAssar Westerlund 
859b528cefcSMark Murray 	FD_ZERO(&fds);
860b528cefcSMark Murray 	for(i = 0; i < ndescr; i++) {
861*ae771770SStanislav Sedov 	    if(!rk_IS_BAD_SOCKET(d[i].s)){
862b528cefcSMark Murray 		if(d[i].type == SOCK_STREAM &&
863b528cefcSMark Murray 		   d[i].timeout && d[i].timeout < time(NULL)) {
864c19800e8SDoug Rabson 		    kdc_log(context, config, 1,
865c19800e8SDoug Rabson 			    "TCP-connection from %s expired after %lu bytes",
866adb0ddaeSAssar Westerlund 			    d[i].addr_string, (unsigned long)d[i].len);
867b528cefcSMark Murray 		    clear_descr(&d[i]);
868b528cefcSMark Murray 		    continue;
869b528cefcSMark Murray 		}
870*ae771770SStanislav Sedov #ifndef NO_LIMIT_FD_SETSIZE
871b528cefcSMark Murray 		if(max_fd < d[i].s)
872b528cefcSMark Murray 		    max_fd = d[i].s;
873*ae771770SStanislav Sedov #ifdef FD_SETSIZE
8745e9cd1aeSAssar Westerlund 		if (max_fd >= FD_SETSIZE)
8755e9cd1aeSAssar Westerlund 		    krb5_errx(context, 1, "fd too large");
876*ae771770SStanislav Sedov #endif
877*ae771770SStanislav Sedov #endif
878b528cefcSMark Murray 		FD_SET(d[i].s, &fds);
879*ae771770SStanislav Sedov 	    } else if(min_free < 0 || i < (size_t)min_free)
880b528cefcSMark Murray 		min_free = i;
881b528cefcSMark Murray 	}
882b528cefcSMark Murray 	if(min_free == -1){
883b528cefcSMark Murray 	    struct descr *tmp;
884b528cefcSMark Murray 	    tmp = realloc(d, (ndescr + 4) * sizeof(*d));
885b528cefcSMark Murray 	    if(tmp == NULL)
886b528cefcSMark Murray 		krb5_warnx(context, "No memory");
887b528cefcSMark Murray 	    else {
888b528cefcSMark Murray 		d = tmp;
8895e9cd1aeSAssar Westerlund 		reinit_descrs (d, ndescr);
890b528cefcSMark Murray 		memset(d + ndescr, 0, 4 * sizeof(*d));
891b528cefcSMark Murray 		for(i = ndescr; i < ndescr + 4; i++)
8925e9cd1aeSAssar Westerlund 		    init_descr (&d[i]);
893b528cefcSMark Murray 		min_free = ndescr;
894b528cefcSMark Murray 		ndescr += 4;
895b528cefcSMark Murray 	    }
896b528cefcSMark Murray 	}
897b528cefcSMark Murray 
898b528cefcSMark Murray 	tmout.tv_sec = TCP_TIMEOUT;
899b528cefcSMark Murray 	tmout.tv_usec = 0;
900b528cefcSMark Murray 	switch(select(max_fd + 1, &fds, 0, 0, &tmout)){
901b528cefcSMark Murray 	case 0:
902b528cefcSMark Murray 	    break;
903b528cefcSMark Murray 	case -1:
904283d988cSMark Murray 	    if (errno != EINTR)
905*ae771770SStanislav Sedov 		krb5_warn(context, rk_SOCK_ERRNO, "select");
906b528cefcSMark Murray 	    break;
907b528cefcSMark Murray 	default:
908b528cefcSMark Murray 	    for(i = 0; i < ndescr; i++)
909*ae771770SStanislav Sedov 		if(!rk_IS_BAD_SOCKET(d[i].s) && FD_ISSET(d[i].s, &fds)) {
910b528cefcSMark Murray 		    if(d[i].type == SOCK_DGRAM)
911c19800e8SDoug Rabson 			handle_udp(context, config, &d[i]);
912b528cefcSMark Murray 		    else if(d[i].type == SOCK_STREAM)
913c19800e8SDoug Rabson 			handle_tcp(context, config, d, i, min_free);
914b528cefcSMark Murray 		}
915b528cefcSMark Murray 	}
916b528cefcSMark Murray     }
917*ae771770SStanislav Sedov     if (0);
918*ae771770SStanislav Sedov #ifdef SIGXCPU
919*ae771770SStanislav Sedov     else if(exit_flag == SIGXCPU)
920c19800e8SDoug Rabson 	kdc_log(context, config, 0, "CPU time limit exceeded");
921*ae771770SStanislav Sedov #endif
922c19800e8SDoug Rabson     else if(exit_flag == SIGINT || exit_flag == SIGTERM)
923c19800e8SDoug Rabson 	kdc_log(context, config, 0, "Terminated");
924c19800e8SDoug Rabson     else
925c19800e8SDoug Rabson 	kdc_log(context, config, 0, "Unexpected exit reason: %d", exit_flag);
926b528cefcSMark Murray     free (d);
927b528cefcSMark Murray }
928