2 * Copyright (c) 1997 - 2002 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
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.
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.
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
34 #include "krb5_locl.h"
35 #include "send_to_kdc_plugin.h"
38 krb5_send_to_kdc_func func;
43 * send the data in `req' on the socket `fd' (which is datagram iff udp)
44 * waiting `tmout' for a reply and returning the reply in `rep'.
45 * iff limit read up to this many bytes
46 * returns 0 and data in `rep' if succesful, otherwise -1
50 recv_loop (krb5_socket_t fd,
57 struct timeval timeout;
61 #ifndef NO_LIMIT_FD_SETSIZE
62 if (fd >= FD_SETSIZE) {
71 timeout.tv_sec = tmout;
73 ret = select (fd + 1, &fdset, NULL, NULL, &timeout);
78 } else if (ret == 0) {
83 if (rk_SOCK_IOCTL (fd, FIONREAD, &nbytes) < 0) {
91 nbytes = min((size_t)nbytes, limit - rep->length);
93 tmp = realloc (rep->data, rep->length + nbytes);
99 ret = recv (fd, (char*)tmp + rep->length, nbytes, 0);
101 krb5_data_free (rep);
106 } while(!udp && (limit == 0 || rep->length < limit));
111 * Send kerberos requests and receive a reply on a udp or any other kind
112 * of a datagram socket. See `recv_loop'.
116 send_and_recv_udp(krb5_socket_t fd,
118 const krb5_data *req,
121 if (send (fd, req->data, req->length, 0) < 0)
124 return recv_loop(fd, tmout, 1, 0, rep);
128 * `send_and_recv' for a TCP (or any other stream) socket.
129 * Since there are no record limits on a stream socket the protocol here
130 * is to prepend the request with 4 bytes of its length and the reply
131 * is similarly encoded.
135 send_and_recv_tcp(krb5_socket_t fd,
137 const krb5_data *req,
140 unsigned char len[4];
141 unsigned long rep_len;
144 _krb5_put_int(len, req->length, 4);
145 if(net_write (fd, len, sizeof(len)) < 0)
147 if(net_write (fd, req->data, req->length) < 0)
149 if (recv_loop (fd, tmout, 0, 4, &len_data) < 0)
151 if (len_data.length != 4) {
152 krb5_data_free (&len_data);
155 _krb5_get_int(len_data.data, &rep_len, 4);
156 krb5_data_free (&len_data);
157 if (recv_loop (fd, tmout, 0, rep_len, rep) < 0)
159 if(rep->length != rep_len) {
160 krb5_data_free (rep);
167 _krb5_send_and_recv_tcp(krb5_socket_t fd,
169 const krb5_data *req,
172 return send_and_recv_tcp(fd, tmout, req, rep);
176 * `send_and_recv' tailored for the HTTP protocol.
180 send_and_recv_http(krb5_socket_t fd,
183 const krb5_data *req,
186 char *request = NULL;
189 int len = base64_encode(req->data, req->length, &str);
193 ret = asprintf(&request, "GET %s%s HTTP/1.0\r\n\r\n", prefix, str);
195 if (ret < 0 || request == NULL)
197 ret = net_write (fd, request, strlen(request));
201 ret = recv_loop(fd, tmout, 0, 0, rep);
205 unsigned long rep_len;
208 s = realloc(rep->data, rep->length + 1);
210 krb5_data_free (rep);
214 p = strstr(s, "\r\n\r\n");
222 rep->length -= p - s;
223 if(rep->length < 4) { /* remove length */
229 _krb5_get_int(p, &rep_len, 4);
230 if (rep_len != rep->length) {
235 memmove(rep->data, p + 4, rep->length);
241 init_port(const char *s, int fallback)
246 sscanf (s, "%d", &tmp);
253 * Return 0 if succesful, otherwise 1
257 send_via_proxy (krb5_context context,
258 const krb5_krbhst_info *hi,
259 const krb5_data *send_data,
262 char *proxy2 = strdup(context->http_proxy);
263 char *proxy = proxy2;
266 struct addrinfo hints;
267 struct addrinfo *ai, *a;
269 krb5_socket_t s = rk_INVALID_SOCKET;
270 char portstr[NI_MAXSERV];
274 if (strncmp (proxy, "http://", 7) == 0)
277 colon = strchr(proxy, ':');
280 memset (&hints, 0, sizeof(hints));
281 hints.ai_family = PF_UNSPEC;
282 hints.ai_socktype = SOCK_STREAM;
283 snprintf (portstr, sizeof(portstr), "%d",
284 ntohs(init_port (colon, htons(80))));
285 ret = getaddrinfo (proxy, portstr, &hints, &ai);
288 return krb5_eai_to_heim_errno(ret, errno);
290 for (a = ai; a != NULL; a = a->ai_next) {
291 s = socket (a->ai_family, a->ai_socktype | SOCK_CLOEXEC, a->ai_protocol);
295 if (connect (s, a->ai_addr, a->ai_addrlen) < 0) {
307 ret = asprintf(&prefix, "http://%s/", hi->hostname);
308 if(ret < 0 || prefix == NULL) {
312 ret = send_and_recv_http(s, context->kdc_timeout,
313 prefix, send_data, receive);
316 if(ret == 0 && receive->length != 0)
321 static krb5_error_code
322 send_via_plugin(krb5_context context,
323 krb5_krbhst_info *hi,
325 const krb5_data *send_data,
328 struct krb5_plugin *list = NULL, *e;
331 ret = _krb5_plugin_find(context, PLUGIN_TYPE_DATA, KRB5_PLUGIN_SEND_TO_KDC, &list);
332 if(ret != 0 || list == NULL)
333 return KRB5_PLUGIN_NO_HANDLE;
335 for (e = list; e != NULL; e = _krb5_plugin_get_next(e)) {
336 krb5plugin_send_to_kdc_ftable *service;
339 service = _krb5_plugin_get_symbol(e);
340 if (service->minor_version != 0)
343 (*service->init)(context, &ctx);
344 ret = (*service->send_to_kdc)(context, ctx, hi,
345 timeout, send_data, receive);
346 (*service->fini)(ctx);
349 if (ret != KRB5_PLUGIN_NO_HANDLE) {
350 krb5_set_error_message(context, ret,
351 N_("Plugin send_to_kdc failed to "
352 "lookup with error: %d", ""), ret);
356 _krb5_plugin_free(list);
357 return KRB5_PLUGIN_NO_HANDLE;
362 * Send the data `send' to one host from `handle` and get back the reply
366 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
367 krb5_sendto (krb5_context context,
368 const krb5_data *send_data,
369 krb5_krbhst_handle handle,
376 krb5_data_zero(receive);
378 for (i = 0; i < context->max_retries; ++i) {
379 krb5_krbhst_info *hi;
381 while (krb5_krbhst_next(context, handle, &hi) == 0) {
382 struct addrinfo *ai, *a;
384 _krb5_debug(context, 2,
385 "trying to communicate with host %s in realm %s",
386 hi->hostname, _krb5_krbhst_get_realm(handle));
388 if (context->send_to_kdc) {
389 struct send_to_kdc *s = context->send_to_kdc;
391 ret = (*s->func)(context, s->data, hi,
392 context->kdc_timeout, send_data, receive);
393 if (ret == 0 && receive->length != 0)
398 ret = send_via_plugin(context, hi, context->kdc_timeout,
400 if (ret == 0 && receive->length != 0)
402 else if (ret != KRB5_PLUGIN_NO_HANDLE)
405 if(hi->proto == KRB5_KRBHST_HTTP && context->http_proxy) {
406 if (send_via_proxy (context, hi, send_data, receive) == 0) {
413 ret = krb5_krbhst_get_addrinfo(context, hi, &ai);
417 for (a = ai; a != NULL; a = a->ai_next) {
418 fd = socket (a->ai_family, a->ai_socktype | SOCK_CLOEXEC, a->ai_protocol);
419 if (rk_IS_BAD_SOCKET(fd))
422 if (connect (fd, a->ai_addr, a->ai_addrlen) < 0) {
427 case KRB5_KRBHST_HTTP :
428 ret = send_and_recv_http(fd, context->kdc_timeout,
429 "", send_data, receive);
431 case KRB5_KRBHST_TCP :
432 ret = send_and_recv_tcp (fd, context->kdc_timeout,
435 case KRB5_KRBHST_UDP :
436 ret = send_and_recv_udp (fd, context->kdc_timeout,
441 if(ret == 0 && receive->length != 0)
445 krb5_krbhst_reset(context, handle);
447 krb5_clear_error_message (context);
448 ret = KRB5_KDC_UNREACH;
450 _krb5_debug(context, 2,
451 "result of trying to talk to realm %s = %d",
452 _krb5_krbhst_get_realm(handle), ret);
456 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
457 krb5_sendto_kdc(krb5_context context,
458 const krb5_data *send_data,
459 const krb5_realm *realm,
462 return krb5_sendto_kdc_flags(context, send_data, realm, receive, 0);
465 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
466 krb5_sendto_kdc_flags(krb5_context context,
467 const krb5_data *send_data,
468 const krb5_realm *realm,
475 ret = krb5_sendto_ctx_alloc(context, &ctx);
478 krb5_sendto_ctx_add_flags(ctx, flags);
479 krb5_sendto_ctx_set_func(ctx, _krb5_kdc_retry, NULL);
481 ret = krb5_sendto_context(context, ctx, send_data, *realm, receive);
482 krb5_sendto_ctx_free(context, ctx);
486 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
487 krb5_set_send_to_kdc_func(krb5_context context,
488 krb5_send_to_kdc_func func,
491 free(context->send_to_kdc);
493 context->send_to_kdc = NULL;
497 context->send_to_kdc = malloc(sizeof(*context->send_to_kdc));
498 if (context->send_to_kdc == NULL) {
499 krb5_set_error_message(context, ENOMEM,
500 N_("malloc: out of memory", ""));
504 context->send_to_kdc->func = func;
505 context->send_to_kdc->data = data;
509 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
510 _krb5_copy_send_to_kdc_func(krb5_context context, krb5_context to)
512 if (context->send_to_kdc)
513 return krb5_set_send_to_kdc_func(to,
514 context->send_to_kdc->func,
515 context->send_to_kdc->data);
517 return krb5_set_send_to_kdc_func(to, NULL, NULL);
522 struct krb5_sendto_ctx_data {
525 krb5_sendto_ctx_func func;
529 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
530 krb5_sendto_ctx_alloc(krb5_context context, krb5_sendto_ctx *ctx)
532 *ctx = calloc(1, sizeof(**ctx));
534 krb5_set_error_message(context, ENOMEM,
535 N_("malloc: out of memory", ""));
541 KRB5_LIB_FUNCTION void KRB5_LIB_CALL
542 krb5_sendto_ctx_add_flags(krb5_sendto_ctx ctx, int flags)
547 KRB5_LIB_FUNCTION int KRB5_LIB_CALL
548 krb5_sendto_ctx_get_flags(krb5_sendto_ctx ctx)
553 KRB5_LIB_FUNCTION void KRB5_LIB_CALL
554 krb5_sendto_ctx_set_type(krb5_sendto_ctx ctx, int type)
560 KRB5_LIB_FUNCTION void KRB5_LIB_CALL
561 krb5_sendto_ctx_set_func(krb5_sendto_ctx ctx,
562 krb5_sendto_ctx_func func,
569 KRB5_LIB_FUNCTION void KRB5_LIB_CALL
570 krb5_sendto_ctx_free(krb5_context context, krb5_sendto_ctx ctx)
572 memset(ctx, 0, sizeof(*ctx));
576 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
577 krb5_sendto_context(krb5_context context,
579 const krb5_data *send_data,
580 const krb5_realm realm,
584 krb5_krbhst_handle handle = NULL;
585 int type, freectx = 0;
588 krb5_data_zero(receive);
592 ret = krb5_sendto_ctx_alloc(context, &ctx);
599 if ((ctx->flags & KRB5_KRBHST_FLAGS_MASTER) || context->use_admin_kdc)
600 type = KRB5_KRBHST_ADMIN;
602 type = KRB5_KRBHST_KDC;
605 if ((int)send_data->length > context->large_msg_size)
606 ctx->flags |= KRB5_KRBHST_FLAGS_LARGE_MSG;
608 /* loop until we get back a appropriate response */
611 action = KRB5_SENDTO_DONE;
613 krb5_data_free(receive);
615 if (handle == NULL) {
616 ret = krb5_krbhst_init_flags(context, realm, type,
617 ctx->flags, &handle);
620 krb5_sendto_ctx_free(context, ctx);
625 ret = krb5_sendto(context, send_data, handle, receive);
629 ret = (*ctx->func)(context, ctx, ctx->data, receive, &action);
633 if (action != KRB5_SENDTO_CONTINUE) {
634 krb5_krbhst_free(context, handle);
637 } while (action != KRB5_SENDTO_DONE);
639 krb5_krbhst_free(context, handle);
640 if (ret == KRB5_KDC_UNREACH)
641 krb5_set_error_message(context, ret,
642 N_("unable to reach any KDC in realm %s", ""),
645 krb5_data_free(receive);
647 krb5_sendto_ctx_free(context, ctx);
651 krb5_error_code KRB5_CALLCONV
652 _krb5_kdc_retry(krb5_context context, krb5_sendto_ctx ctx, void *data,
653 const krb5_data *reply, int *action)
658 if(krb5_rd_error(context, reply, &error))
661 ret = krb5_error_from_rd_error(context, &error, NULL);
662 krb5_free_error_contents(context, &error);
665 case KRB5KRB_ERR_RESPONSE_TOO_BIG: {
666 if (krb5_sendto_ctx_get_flags(ctx) & KRB5_KRBHST_FLAGS_LARGE_MSG)
668 krb5_sendto_ctx_add_flags(ctx, KRB5_KRBHST_FLAGS_LARGE_MSG);
669 *action = KRB5_SENDTO_RESTART;
672 case KRB5KDC_ERR_SVC_UNAVAILABLE:
673 *action = KRB5_SENDTO_CONTINUE;