#include "krb5_locl.h"
#include <resolve.h>
#include "locate_plugin.h"
static void append_host_hostinfo(struct krb5_krbhst_data *, struct krb5_krbhst_info *);
struct krb5_krbhst_data {
struct heim_base_uniq base;
HEIMDAL_MUTEX mutex;
char *realm;
unsigned int flags;
int def_port;
int port;
#define KD_CONFIG 0x1
#define KD_SRV_UDP 0x2
#define KD_SRV_TCP 0x4
#define KD_SRV_HTTP 0x8
#define KD_SRV_KKDCP 0x10
#define KD_FALLBACK 0x20
#define KD_CONFIG_EXISTS 0x40
#define KD_LARGE_MSG 0x80
#define KD_PLUGIN 0x100
#define KD_HOSTNAMES 0x200
krb5_error_code (*get_next)(krb5_context, struct krb5_krbhst_data *,
krb5_krbhst_info**);
char *hostname;
unsigned int fallback_count;
struct krb5_krbhst_info *hosts, **index, **end;
krb5_context context;
heim_queue_t process_queue;
heim_queue_t queue;
void (*callback)(void *, krb5_krbhst_info *);
void *userctx;
};
static int
string_to_proto(const char *string)
{
if(strcasecmp(string, "udp") == 0)
return KRB5_KRBHST_UDP;
else if(strcasecmp(string, "tcp") == 0)
return KRB5_KRBHST_TCP;
else if(strcasecmp(string, "http") == 0)
return KRB5_KRBHST_HTTP;
else if(strcasecmp(string, "kkdcp") == 0)
return KRB5_KRBHST_KKDCP;
return -1;
}
static void
query_release(void *ctx)
{
struct _krb5_srv_query_ctx *query = (struct _krb5_srv_query_ctx *)ctx;
free(query->domain);
heim_release(query->handle);
if (query->sema)
dispatch_release((dispatch_semaphore_t)query->sema);
#ifdef __APPLE__
if (query->array)
free(query->array);
#endif
}
#ifdef __APPLE__
#include <dns_sd.h>
static int
compare_srv(const void *a, const void *b)
{
const struct srv_reply *aa = a, *bb = b;
if(aa->priority == bb->priority)
return aa->weight - bb->weight;
return aa->priority - aa->priority;
}
void
_krb5_state_srv_sort(struct _krb5_srv_query_ctx *query)
{
size_t n, m, o, prio_marker;
uint32_t rnd, sum = 0;
if (query->len < 2)
return;
qsort(query->array, query->len, sizeof(query->array[0]), compare_srv);
prio_marker = 0;
for (n = 1; n < query->len; n++) {
if (query->array[prio_marker].priority != query->array[n].priority) {
for (m = prio_marker; m < n && sum != 0; m++) {
int32_t count = -1;
rnd = rk_random() % sum;
for (o = prio_marker; o < n; o++) {
if (query->array[o].weight < 0)
continue;
if (rnd <= query->array[o].weight) {
sum -= query->array[o].weight;
query->array[o].weight = count--;
break;
}
rnd -= query->array[o].weight;
}
if (o >= n)
_krb5_debugx(query->context, 2,
"o too large: sum %d", (int)sum);
}
sum = 0;
prio_marker = n;
} else {
sum += query->array[n].weight;
}
}
qsort(query->array, query->len, sizeof(query->array[0]), compare_srv);
}
static void
state_append_hosts(struct _krb5_srv_query_ctx *query)
{
size_t n;
for (n = 0; n < query->len; n++)
append_host_hostinfo(query->handle, query->array[n].hi);
dispatch_semaphore_signal((dispatch_semaphore_t)query->sema);
heim_release(query);
}
static void
QueryReplyCallback(DNSServiceRef sdRef,
DNSServiceFlags flags,
uint32_t ifIndex,
DNSServiceErrorType errorCode,
const char *fullname,
uint16_t rrtype,
uint16_t rrclass,
uint16_t rdlen,
const void *rdata,
uint32_t ttl,
void *context)
{
const uint8_t *end_rd = ((uint8_t *)rdata) + rdlen, *rd = rdata;
struct _krb5_srv_query_ctx *query = context;
uint16_t priority, weight, port;
struct srv_reply *tmp;
krb5_krbhst_info *hi;
int status;
if (errorCode != kDNSServiceErr_NoError) {
flags = 0;
goto end;
}
if (rrtype != kDNSServiceType_SRV)
goto end;
if (rdlen < 7)
goto end;
priority = (rd[0] << 8) | rd[1];
weight = (rd[2] << 8) | rd[3];
port = (rd[4] << 8) | rd[5];
if (rdlen + sizeof(*hi) < rdlen)
goto end;
hi = calloc(1, sizeof(*hi) + rdlen);
if (hi == NULL)
goto end;
status = dn_expand(rdata, end_rd, rd + 6, hi->hostname, rdlen);
if(status < 0 || (size_t)status + 6 > rdlen) {
free(hi);
goto end;
}
hi->proto = query->proto_num;
hi->def_port = 0;
if (query->port != 0)
hi->port = query->port;
else
hi->port = port;
hi->path = rk_UNCONST(query->path);
tmp = realloc(query->array, (query->len + 1) * sizeof(query->array[0]));
if (tmp == NULL) {
free(hi);
goto end;
}
query->array = tmp;
query->array[query->len].hi = hi;
query->array[query->len].priority = priority;
query->array[query->len].weight = weight;
query->len++;
end:
if ((flags & kDNSServiceFlagsMoreComing) == 0) {
_krb5_state_srv_sort(query);
state_append_hosts(query);
}
}
static krb5_error_code
srv_find_realm(krb5_context context, struct krb5_krbhst_data *handle,
heim_queue_t dnsQueue,
struct _krb5_srv_query_ctx *query)
{
DNSServiceRef client = NULL;
DNSServiceErrorType error;
krb5_error_code ret;
error = DNSServiceQueryRecord(&client,
kDNSServiceFlagsTimeout | kDNSServiceFlagsReturnIntermediates,
0,
query->domain,
kDNSServiceType_SRV,
kDNSServiceClass_IN,
QueryReplyCallback,
query);
if (!error) {
heim_retain(query);
DNSServiceSetDispatchQueue(client, (dispatch_queue_t)dnsQueue);
if (dispatch_semaphore_wait((dispatch_semaphore_t)query->sema, dispatch_time(DISPATCH_TIME_NOW, 10ull * NSEC_PER_SEC))) {
_krb5_debugx(context, 2,
"searching DNS %s for domain timed out",
query->domain);
ret = KRB5_KDC_UNREACH;
} else
ret = 0;
DNSServiceRefDeallocate (client);
} else {
_krb5_debugx(context, 2,
"searching DNS for domain %s failed: %d",
query->domain, error);
ret = KRB5_KDC_UNREACH;
}
heim_release(query);
return ret;
}
#else
static void
srv_query_domain(void *ctx)
{
struct _krb5_srv_query_ctx *query = ctx;
struct rk_dns_reply *r;
struct rk_resource_record *rr;
r = rk_dns_lookup(query->domain, "SRV");
if(r == NULL)
goto out;
rk_dns_srv_order(r);
for(rr = r->head; rr; rr = rr->next) {
if(rr->type == rk_ns_t_srv) {
krb5_krbhst_info *hi;
size_t len = strlen(rr->u.srv->target);
hi = calloc(1, sizeof(*hi) + len);
if(hi == NULL)
goto out;
hi->proto = query->proto_num;
hi->def_port = query->def_port;
if (handle->port != 0)
hi->port = handle->port;
else
hi->port = rr->u.srv->port;
hi->path = query->path;
strlcpy(hi->hostname, rr->u.srv->target, len + 1);
append_host_hostinfo(kd, hi);
}
}
out:
if (r)
rk_dns_free_data(r);
heim_sema_signal(queue->sema);
heim_release(queue->handle);
}
static krb5_error_code
srv_find_realm(krb5_context context, struct krb5_krbhst_data *handle,
heim_queue_t queue,
struct _krb5_srv_query_ctx *query)
{
heim_async_f(queue, query, srv_query_domain);
heim_sema_wait(query->sema, 10);
heim_release(query);
}
#endif
static krb5_boolean
krbhst_empty(struct krb5_krbhst_data *kd)
{
krb5_boolean empty;
HEIMDAL_MUTEX_lock(&kd->mutex);
empty = (kd->index == &kd->hosts);
HEIMDAL_MUTEX_unlock(&kd->mutex);
return empty;
}
static int
krbhst_get_default_proto(struct krb5_krbhst_data *kd)
{
if (kd->flags & KD_LARGE_MSG)
return KRB5_KRBHST_TCP;
return KRB5_KRBHST_UDP;
}
static int
krbhst_get_default_port(struct krb5_krbhst_data *kd)
{
return kd->def_port;
}
const char *
_krb5_krbhst_get_realm(krb5_krbhst_handle handle)
{
return handle->realm;
}
static struct krb5_krbhst_info*
parse_hostspec(krb5_context context, struct krb5_krbhst_data *kd,
const char *spec, int def_port, int port)
{
const char *p = spec, *q, *portstr = NULL;
struct krb5_krbhst_info *hi;
size_t end_hostname;
hi = calloc(1, sizeof(*hi) + strlen(spec));
if(hi == NULL)
return NULL;
hi->proto = krbhst_get_default_proto(kd);
if(strncmp(p, "http://", 7) == 0){
hi->proto = KRB5_KRBHST_HTTP;
p += 7;
} else if(strncmp(p, "http/", 5) == 0) {
hi->proto = KRB5_KRBHST_HTTP;
p += 5;
def_port = ntohs(krb5_getportbyname (context, "http", "tcp", 80));
} else if(strncmp(p, "kkdcp://", 8) == 0) {
hi->proto = KRB5_KRBHST_KKDCP;
p += 8;
def_port = ntohs(krb5_getportbyname (context, "https", "tcp", 443));
}else if(strncmp(p, "tcp/", 4) == 0){
hi->proto = KRB5_KRBHST_TCP;
p += 4;
} else if(strncmp(p, "udp/", 4) == 0) {
hi->proto = KRB5_KRBHST_UDP;
p += 4;
}
if (p[0] == '[' && (q = strchr(p, ']')) != NULL) {
memcpy(hi->hostname, &p[1], q - p - 1);
hi->hostname[q - p - 1] = '\0';
p = q + 1;
if (p[0] == ':')
portstr = ++p;
p = strchr(p, '/');
} else if ((end_hostname = strcspn(p, ":/")) != 0) {
memcpy(hi->hostname, p, end_hostname);
hi->hostname[end_hostname] = '\0';
if (p[end_hostname] == ':') {
portstr = p + end_hostname + 1;
p = strchr(p, '/');
} else {
p = p + end_hostname + 1;
}
} else {
memcpy(hi->hostname, p, strlen(p) + 1);
}
if (p && p[0] == '/')
hi->path = (char *)(p + 1);
strlwr(hi->hostname);
hi->port = hi->def_port = def_port;
if(portstr != NULL && portstr[0]) {
char *end;
hi->port = strtol(portstr, &end, 0);
if(end == portstr) {
free(hi);
return NULL;
}
}
if (port)
hi->port = port;
return hi;
}
void
_krb5_free_krbhst_info(krb5_krbhst_info *hi)
{
if (hi->ai != NULL)
freeaddrinfo(hi->ai);
free(hi);
}
krb5_error_code
_krb5_krbhost_info_move(krb5_context context,
krb5_krbhst_info *from,
krb5_krbhst_info **to)
{
size_t hostnamelen = strlen(from->hostname);
*to = calloc(1, sizeof(**to) + hostnamelen);
if(*to == NULL) {
krb5_set_error_message(context, ENOMEM,
N_("malloc: out of memory", ""));
return ENOMEM;
}
(*to)->proto = from->proto;
(*to)->port = from->port;
(*to)->def_port = from->def_port;
(*to)->ai = from->ai;
from->ai = NULL;
(*to)->next = NULL;
memcpy((*to)->hostname, from->hostname, hostnamelen + 1);
return 0;
}
static void
append_host_hostinfo(struct krb5_krbhst_data *kd, struct krb5_krbhst_info *host)
{
struct krb5_krbhst_info *h;
HEIMDAL_MUTEX_lock(&kd->mutex);
for(h = kd->hosts; h && host; h = h->next) {
if(h->proto == host->proto &&
h->port == host->port &&
strcasecmp(h->hostname, host->hostname) == 0)
{
_krb5_free_krbhst_info(host);
host = NULL;
}
}
if (host) {
*kd->end = host;
kd->end = &host->next;
}
HEIMDAL_MUTEX_unlock(&kd->mutex);
}
static krb5_error_code
append_host_string(krb5_context context, struct krb5_krbhst_data *kd,
const char *host, int def_port, int port)
{
struct krb5_krbhst_info *hi;
hi = parse_hostspec(context, kd, host, def_port, port);
if(hi == NULL)
return ENOMEM;
append_host_hostinfo(kd, hi);
return 0;
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_krbhst_format_string(krb5_context context, const krb5_krbhst_info *host,
char *hostname, size_t hostlen)
{
const char *proto = "";
char portstr[7] = "";
if(host->proto == KRB5_KRBHST_TCP)
proto = "tcp/";
else if(host->proto == KRB5_KRBHST_HTTP)
proto = "http://";
if(host->port != host->def_port)
snprintf(portstr, sizeof(portstr), ":%d", host->port);
snprintf(hostname, hostlen, "%s%s%s", proto, host->hostname, portstr);
return 0;
}
static void
make_hints(struct addrinfo *hints, int proto)
{
memset(hints, 0, sizeof(*hints));
hints->ai_family = AF_UNSPEC;
switch(proto) {
case KRB5_KRBHST_UDP :
hints->ai_socktype = SOCK_DGRAM;
break;
case KRB5_KRBHST_KKDCP :
case KRB5_KRBHST_HTTP :
case KRB5_KRBHST_TCP :
hints->ai_socktype = SOCK_STREAM;
break;
}
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_krbhst_get_addrinfo(krb5_context context, krb5_krbhst_info *host,
struct addrinfo **ai)
{
int ret = 0;
if (host->ai == NULL) {
struct addrinfo hints;
char portstr[NI_MAXSERV];
char *hostname = host->hostname;
snprintf (portstr, sizeof(portstr), "%d", host->port);
make_hints(&hints, host->proto);
hints.ai_flags |= AI_NUMERICHOST | AI_NUMERICSERV;
ret = getaddrinfo(host->hostname, portstr, &hints, &host->ai);
if (ret == 0)
goto out;
hints.ai_flags &= ~(AI_NUMERICHOST);
if (strchr(hostname, '.') && hostname[strlen(hostname) - 1] != '.') {
ret = asprintf(&hostname, "%s.", host->hostname);
if (ret < 0 || hostname == NULL)
return ENOMEM;
}
ret = getaddrinfo(hostname, portstr, &hints, &host->ai);
if (hostname != host->hostname)
free(hostname);
if (ret) {
ret = krb5_eai_to_heim_errno(ret, errno);
goto out;
}
}
out:
*ai = host->ai;
return ret;
}
static krb5_boolean
get_next(struct krb5_krbhst_data *kd, krb5_krbhst_info **host)
{
struct krb5_krbhst_info *hi;
HEIMDAL_MUTEX_lock(&kd->mutex);
hi = *kd->index;
if(hi != NULL) {
*host = hi;
kd->index = &(*kd->index)->next;
}
HEIMDAL_MUTEX_unlock(&kd->mutex);
return hi ? TRUE : FALSE;
}
static void
srv_get_dns_queue(void *ctx)
{
heim_queue_t *dnsQueue = ctx;
*dnsQueue = heim_queue_create("com.apple.kerberos.dns", NULL);
}
static krb5_error_code
srv_get_hosts(krb5_context context, struct krb5_krbhst_data *kd,
const char *proto, const char *service)
{
struct _krb5_srv_query_ctx *query = NULL;
static heim_queue_t dnsQueue;
static heim_base_once_t once;
int proto_num, def_port;
const char *path = NULL;
char *domain = NULL;
if (krb5_realm_is_lkdc(kd->realm))
return 0;
heim_base_once_f(&once, &dnsQueue, srv_get_dns_queue);
proto_num = string_to_proto(proto);
if(proto_num < 0) {
_krb5_debugx(context, 1, N_("unknown protocol `%s' to lookup", ""), proto);
return 0;
}
if(proto_num == KRB5_KRBHST_HTTP) {
def_port = ntohs(krb5_getportbyname (context, "http", "tcp", 80));
} else if(proto_num == KRB5_KRBHST_KKDCP) {
def_port = ntohs(krb5_getportbyname (context, "https", "tcp", 443));
path = "kkdcp";
} else if(kd->port) {
def_port = kd->port;
} else {
def_port = ntohs(krb5_getportbyname (context, service, proto, 88));
}
asprintf(&domain, "_%s._%s.%s.", service, proto, kd->realm);
if (domain == NULL) {
return krb5_enomem(context);
}
query = heim_uniq_alloc(sizeof(*query), "heim-query-ctx", query_release);
if (query == NULL) {
free(domain);
return krb5_enomem(context);
}
query->context = context;
query->sema = heim_sema_create(0);
query->domain = domain;
if (query->sema == NULL) {
heim_release(query);
return krb5_enomem(context);
}
query->handle = heim_retain(kd);
query->def_port = def_port;
query->proto_num = proto_num;
query->path = path;
#ifdef __APPLE__
query->array = NULL;
query->len = 0;
#endif
srv_find_realm(context, kd, dnsQueue, query);
return 0;
}
static void
config_get_hosts(krb5_context context, struct krb5_krbhst_data *kd,
const char *conf_string)
{
int i;
char **hostlist;
hostlist = krb5_config_get_strings(context, NULL,
"realms", kd->realm, conf_string, NULL);
_krb5_debugx(context, 2, "configuration file for realm %s%s found",
kd->realm, hostlist ? "" : " not");
if(hostlist == NULL)
return;
kd->flags |= KD_CONFIG_EXISTS;
for(i = 0; hostlist && hostlist[i] != NULL; i++)
append_host_string(context, kd, hostlist[i], kd->def_port, kd->port);
krb5_config_free_strings(hostlist);
}
static krb5_error_code
fallback_get_hosts(krb5_context context, struct krb5_krbhst_data *kd,
const char *serv_string, int port, int proto)
{
char *host = NULL;
int ret;
struct addrinfo *ai;
struct addrinfo hints;
char portstr[NI_MAXSERV];
ret = krb5_config_get_bool_default(context, NULL, KRB5_FALLBACK_DEFAULT,
"libdefaults", "use_fallback", NULL);
if (!ret) {
kd->flags |= KD_FALLBACK;
return 0;
}
_krb5_debugx(context, 2, "fallback lookup %d for realm %s (service %s)",
kd->fallback_count, kd->realm, serv_string);
if(kd->fallback_count >= 5 || krb5_realm_is_lkdc(kd->realm)) {
kd->flags |= KD_FALLBACK;
return 0;
}
if(kd->fallback_count == 0)
ret = asprintf(&host, "%s.%s.", serv_string, kd->realm);
else
ret = asprintf(&host, "%s-%d.%s.",
serv_string, kd->fallback_count, kd->realm);
if (ret < 0 || host == NULL)
return ENOMEM;
make_hints(&hints, proto);
snprintf(portstr, sizeof(portstr), "%d", port);
ret = getaddrinfo(host, portstr, &hints, &ai);
if (ret) {
free(host);
kd->flags |= KD_FALLBACK;
} else {
struct krb5_krbhst_info *hi;
size_t hostlen = strlen(host);
hi = calloc(1, sizeof(*hi) + hostlen);
if(hi == NULL) {
free(host);
return ENOMEM;
}
hi->proto = proto;
hi->port = hi->def_port = port;
hi->ai = ai;
memmove(hi->hostname, host, hostlen);
hi->hostname[hostlen] = '\0';
free(host);
append_host_hostinfo(kd, hi);
kd->fallback_count++;
}
return 0;
}
static krb5_error_code
add_plugin_host(struct krb5_krbhst_data *kd,
const char *host,
const char *port,
int portnum,
int proto)
{
struct krb5_krbhst_info *hi;
struct addrinfo hints, *ai;
size_t hostlen;
int ret;
make_hints(&hints, proto);
ret = getaddrinfo(host, port, &hints, &ai);
if (ret)
return 0;
hostlen = strlen(host);
hi = calloc(1, sizeof(*hi) + hostlen);
if(hi == NULL)
return ENOMEM;
hi->proto = proto;
hi->port = hi->def_port = portnum;
hi->ai = ai;
memmove(hi->hostname, host, hostlen);
hi->hostname[hostlen] = '\0';
append_host_hostinfo(kd, hi);
return 0;
}
static krb5_error_code
add_locate(void *ctx, int type, struct sockaddr *addr)
{
struct krb5_krbhst_data *kd = ctx;
char host[NI_MAXHOST], port[NI_MAXSERV];
socklen_t socklen;
krb5_error_code ret;
int proto, portnum;
socklen = socket_sockaddr_size(addr);
portnum = socket_get_port(addr);
ret = getnameinfo(addr, socklen, host, sizeof(host), port, sizeof(port),
NI_NUMERICHOST|NI_NUMERICSERV);
if (ret != 0)
return 0;
if (kd->port)
snprintf(port, sizeof(port), "%d", kd->port);
else if (atoi(port) == 0)
snprintf(port, sizeof(port), "%d", krbhst_get_default_port(kd));
proto = krbhst_get_default_proto(kd);
ret = add_plugin_host(kd, host, port, portnum, proto);
if (ret)
return ret;
if (proto == KRB5_KRBHST_UDP) {
ret = add_plugin_host(kd, host, port, portnum, KRB5_KRBHST_TCP);
if (ret)
return ret;
}
return 0;
}
struct plctx {
enum locate_service_type type;
struct krb5_krbhst_data *kd;
unsigned long flags;
};
static krb5_error_code
plcallback(krb5_context context,
const void *plug, void *plugctx, void *userctx)
{
const krb5plugin_service_locate_ftable *locate = plug;
struct plctx *plctx = userctx;
if (locate->minor_version >= KRB5_PLUGIN_LOCATE_VERSION_2)
return locate->lookup(plugctx, plctx->flags, plctx->type, plctx->kd->realm, 0, 0, add_locate, plctx->kd);
if (plctx->flags & KRB5_PLF_ALLOW_HOMEDIR)
return locate->old_lookup(plugctx, plctx->type, plctx->kd->realm, 0, 0, add_locate, plctx->kd);
return KRB5_PLUGIN_NO_HANDLE;
}
static void
plugin_get_hosts(krb5_context context,
struct krb5_krbhst_data *kd,
enum locate_service_type type)
{
struct plctx ctx = { type, kd, 0 };
if (krb5_homedir_access(context))
ctx.flags |= KRB5_PLF_ALLOW_HOMEDIR;
krb5_plugin_run_f(context, "krb5", KRB5_PLUGIN_LOCATE,
KRB5_PLUGIN_LOCATE_VERSION_0,
0, &ctx, plcallback);
}
static void
hostnames_get_hosts(krb5_context context,
struct krb5_krbhst_data *kd,
const char *type)
{
kd->flags |= KD_HOSTNAMES;
if (kd->hostname)
append_host_string(context, kd, kd->hostname, kd->def_port, kd->port);
}
static krb5_error_code
kdc_get_next(krb5_context context,
struct krb5_krbhst_data *kd,
krb5_krbhst_info **host)
{
krb5_error_code ret;
if ((kd->flags & KD_HOSTNAMES) == 0) {
hostnames_get_hosts(context, kd, "kdc");
if(get_next(kd, host))
return 0;
}
if ((kd->flags & KD_PLUGIN) == 0) {
plugin_get_hosts(context, kd, locate_service_kdc);
kd->flags |= KD_PLUGIN;
if(get_next(kd, host))
return 0;
}
if((kd->flags & KD_CONFIG) == 0) {
config_get_hosts(context, kd, "kdc");
kd->flags |= KD_CONFIG;
if(get_next(kd, host))
return 0;
}
if (kd->flags & KD_CONFIG_EXISTS) {
_krb5_debugx(context, 1,
"Configuration exists for realm %s, wont go to DNS",
kd->realm);
return KRB5_KDC_UNREACH;
}
if(context->srv_lookup) {
if((kd->flags & KD_SRV_UDP) == 0 && (kd->flags & KD_LARGE_MSG) == 0) {
srv_get_hosts(context, kd, "udp", "kerberos");
kd->flags |= KD_SRV_UDP;
if(get_next(kd, host))
return 0;
}
if((kd->flags & KD_SRV_TCP) == 0) {
srv_get_hosts(context, kd, "tcp", "kerberos");
kd->flags |= KD_SRV_TCP;
if(get_next(kd, host))
return 0;
}
if((kd->flags & KD_SRV_HTTP) == 0) {
srv_get_hosts(context, kd, "http", "kerberos");
kd->flags |= KD_SRV_HTTP;
if(get_next(kd, host))
return 0;
}
if((kd->flags & KD_SRV_KKDCP) == 0) {
srv_get_hosts(context, kd, "kkdcp", "kerberos");
kd->flags |= KD_SRV_KKDCP;
if(get_next(kd, host))
return 0;
}
}
while((kd->flags & KD_FALLBACK) == 0) {
ret = fallback_get_hosts(context, kd, "kerberos",
kd->def_port,
krbhst_get_default_proto(kd));
if(ret)
return ret;
if(get_next(kd, host))
return 0;
}
_krb5_debugx(context, 0, "No KDC entries found for %s", kd->realm);
return KRB5_KDC_UNREACH;
}
static krb5_error_code
admin_get_next(krb5_context context,
struct krb5_krbhst_data *kd,
krb5_krbhst_info **host)
{
krb5_error_code ret;
if ((kd->flags & KD_PLUGIN) == 0) {
plugin_get_hosts(context, kd, locate_service_kadmin);
kd->flags |= KD_PLUGIN;
if(get_next(kd, host))
return 0;
}
if((kd->flags & KD_CONFIG) == 0) {
config_get_hosts(context, kd, "admin_server");
kd->flags |= KD_CONFIG;
if(get_next(kd, host))
return 0;
}
if (kd->flags & KD_CONFIG_EXISTS) {
_krb5_debugx(context, 1,
"Configuration exists for realm %s, wont go to DNS",
kd->realm);
return KRB5_KDC_UNREACH;
}
if(context->srv_lookup) {
if((kd->flags & KD_SRV_TCP) == 0) {
srv_get_hosts(context, kd, "tcp", "kerberos-adm");
kd->flags |= KD_SRV_TCP;
if(get_next(kd, host))
return 0;
}
}
if (krbhst_empty(kd)
&& (kd->flags & KD_FALLBACK) == 0) {
ret = fallback_get_hosts(context, kd, "kerberos",
kd->def_port,
krbhst_get_default_proto(kd));
if(ret)
return ret;
kd->flags |= KD_FALLBACK;
if(get_next(kd, host))
return 0;
}
_krb5_debugx(context, 0, "No admin entries found for realm %s", kd->realm);
return KRB5_KDC_UNREACH;
}
static krb5_error_code
kpasswd_get_next(krb5_context context,
struct krb5_krbhst_data *kd,
krb5_krbhst_info **host)
{
krb5_error_code ret;
if ((kd->flags & KD_PLUGIN) == 0) {
plugin_get_hosts(context, kd, locate_service_kpasswd);
kd->flags |= KD_PLUGIN;
if(get_next(kd, host))
return 0;
}
if((kd->flags & KD_CONFIG) == 0) {
config_get_hosts(context, kd, "kpasswd_server");
kd->flags |= KD_CONFIG;
if(get_next(kd, host))
return 0;
}
if (kd->flags & KD_CONFIG_EXISTS) {
_krb5_debugx(context, 1,
"Configuration exists for realm %s, wont go to DNS",
kd->realm);
return KRB5_KDC_UNREACH;
}
if(context->srv_lookup) {
if((kd->flags & KD_SRV_UDP) == 0) {
srv_get_hosts(context, kd, "udp", "kpasswd");
kd->flags |= KD_SRV_UDP;
if(get_next(kd, host))
return 0;
}
if((kd->flags & KD_SRV_TCP) == 0) {
srv_get_hosts(context, kd, "tcp", "kpasswd");
kd->flags |= KD_SRV_TCP;
if(get_next(kd, host))
return 0;
}
}
if (krbhst_empty(kd)) {
kd->flags = 0;
kd->port = kd->def_port;
kd->get_next = admin_get_next;
ret = (*kd->get_next)(context, kd, host);
if (ret == 0)
(*host)->proto = krbhst_get_default_proto(kd);
return ret;
}
_krb5_debugx(context, 0, "No kpasswd entries found for realm %s", kd->realm);
return KRB5_KDC_UNREACH;
}
static void
krbhost_dealloc(void *ptr)
{
struct krb5_krbhst_data *handle = (struct krb5_krbhst_data *)ptr;
krb5_krbhst_info *h, *next;
for (h = handle->hosts; h != NULL; h = next) {
next = h->next;
_krb5_free_krbhst_info(h);
}
if (handle->hostname)
free(handle->hostname);
HEIMDAL_MUTEX_destroy(&handle->mutex);
free(handle->realm);
}
static struct krb5_krbhst_data*
common_init(krb5_context context,
const char *service,
const char *realm,
int flags)
{
struct krb5_krbhst_data *kd;
if ((kd = heim_uniq_alloc(sizeof(*kd), "krbhst-context", krbhost_dealloc)) == NULL)
return NULL;
if((kd->realm = strdup(realm)) == NULL) {
heim_release(kd);
return NULL;
}
_krb5_debugx(context, 2, "Trying to find service %s for realm %s flags %x",
service, realm, flags);
if (!strchr(realm, '.'))
kd->flags |= KD_CONFIG_EXISTS;
if (flags & KRB5_KRBHST_FLAGS_LARGE_MSG)
kd->flags |= KD_LARGE_MSG;
kd->end = kd->index = &kd->hosts;
HEIMDAL_MUTEX_init(&kd->mutex);
return kd;
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_krbhst_init(krb5_context context,
const char *realm,
unsigned int type,
krb5_krbhst_handle *handle)
{
return krb5_krbhst_init_flags(context, realm, type, 0, handle);
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_krbhst_init_flags(krb5_context context,
const char *realm,
unsigned int type,
int flags,
krb5_krbhst_handle *handle)
{
struct krb5_krbhst_data *kd;
krb5_error_code (*next)(krb5_context, struct krb5_krbhst_data *,
krb5_krbhst_info **);
int def_port;
const char *service;
*handle = NULL;
switch(type) {
case KRB5_KRBHST_KDC:
next = kdc_get_next;
def_port = ntohs(krb5_getportbyname (context, "kerberos", "udp", 88));
service = "kdc";
break;
case KRB5_KRBHST_ADMIN:
next = admin_get_next;
def_port = ntohs(krb5_getportbyname (context, "kerberos-adm",
"tcp", 749));
service = "admin";
break;
case KRB5_KRBHST_CHANGEPW:
next = kpasswd_get_next;
def_port = ntohs(krb5_getportbyname (context, "kpasswd", "udp",
KPASSWD_PORT));
service = "change_password";
break;
default:
krb5_set_error_message(context, ENOTTY,
N_("unknown krbhst type (%u)", ""), type);
return ENOTTY;
}
if((kd = common_init(context, service, realm, flags)) == NULL)
return ENOMEM;
kd->get_next = next;
kd->def_port = def_port;
*handle = kd;
return 0;
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_krbhst_next(krb5_context context,
krb5_krbhst_handle handle,
krb5_krbhst_info **host)
{
if(get_next(handle, host))
return 0;
return (*handle->get_next)(context, handle, host);
}
static void
krbhst_callback(void *ctx)
{
krb5_krbhst_info *host = ctx;
krb5_krbhst_handle handle = host->__private;
if (handle->callback)
handle->callback(handle->userctx, host);
}
static void
krbhst_callback_done(void *ctx)
{
krb5_krbhst_handle handle = ctx;
void (*callback)(void *, krb5_krbhst_info *) = handle->callback;
void *userctx = handle->userctx;
heim_release(handle->queue);
handle->queue = NULL;
handle->callback = NULL;
handle->userctx = NULL;
if (callback)
callback(userctx, NULL);
heim_release(handle);
}
static void
krbhst_callback_cancel(void *ctx)
{
krb5_krbhst_handle handle = ctx;
handle->callback(handle->userctx, NULL);
handle->callback = NULL;
}
static void
process_loop(void *ctx)
{
krb5_krbhst_handle handle = ctx;
krb5_krbhst_info *hi;
while(krb5_krbhst_next(handle->context, handle, &hi)) {
hi->__private = heim_retain(handle);
heim_async_f(handle->queue, hi, krbhst_callback);
}
heim_async_f(handle->queue, handle, krbhst_callback_done);
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
_krb5_krbhst_async(krb5_context context,
krb5_krbhst_handle handle,
heim_queue_t queue,
void *userctx,
void (*callback)(void *userctx, krb5_krbhst_info *host))
{
heim_assert(handle->queue == NULL, "krbhst have a outstanding request already");
handle->queue = heim_retain(queue);
handle->context = context;
if (handle->process_queue == NULL)
handle->process_queue = heim_queue_create("krbhst", NULL);
handle->userctx = userctx;
handle->callback = callback;
heim_async_f(handle->process_queue, heim_retain(handle), process_loop);
return 0;
}
KRB5_LIB_FUNCTION void KRB5_LIB_CALL
_krb5_krbhst_cancel(krb5_context context,
krb5_krbhst_handle handle)
{
heim_assert(handle->process_queue != NULL, "cancel non async krbhst");
heim_assert(handle->callback != NULL, "cancel on already canceled handle");
heim_async_f(handle->queue, handle, krbhst_callback_cancel);
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_krbhst_next_as_string(krb5_context context,
krb5_krbhst_handle handle,
char *hostname,
size_t hostlen)
{
krb5_error_code ret;
krb5_krbhst_info *host;
ret = krb5_krbhst_next(context, handle, &host);
if(ret)
return ret;
return krb5_krbhst_format_string(context, host, hostname, hostlen);
}
krb5_error_code KRB5_LIB_FUNCTION
krb5_krbhst_set_hostname(krb5_context context,
krb5_krbhst_handle handle,
const char *hostname)
{
if (handle->hostname)
free(handle->hostname);
handle->hostname = strdup(hostname);
if (handle->hostname == NULL)
return ENOMEM;
return 0;
}
KRB5_LIB_FUNCTION void KRB5_LIB_CALL
krb5_krbhst_reset(krb5_context context, krb5_krbhst_handle handle)
{
HEIMDAL_MUTEX_lock(&handle->mutex);
handle->index = &handle->hosts;
HEIMDAL_MUTEX_unlock(&handle->mutex);
}
KRB5_LIB_FUNCTION void KRB5_LIB_CALL
krb5_krbhst_free(krb5_context context, krb5_krbhst_handle handle)
{
heim_release(handle);
}
#ifndef HEIMDAL_SMALLER
static krb5_error_code
gethostlist(krb5_context context, const char *realm,
unsigned int type, char ***hostlist)
{
krb5_error_code ret;
int nhost = 0;
krb5_krbhst_handle handle;
char host[MAXHOSTNAMELEN];
krb5_krbhst_info *hostinfo;
ret = krb5_krbhst_init(context, realm, type, &handle);
if (ret)
return ret;
while(krb5_krbhst_next(context, handle, &hostinfo) == 0)
nhost++;
if(nhost == 0) {
krb5_set_error_message(context, KRB5_KDC_UNREACH,
N_("No KDC found for realm %s", ""), realm);
return KRB5_KDC_UNREACH;
}
*hostlist = calloc(nhost + 1, sizeof(**hostlist));
if(*hostlist == NULL) {
krb5_krbhst_free(context, handle);
return ENOMEM;
}
krb5_krbhst_reset(context, handle);
nhost = 0;
while(krb5_krbhst_next_as_string(context, handle,
host, sizeof(host)) == 0) {
if(((*hostlist)[nhost++] = strdup(host)) == NULL) {
krb5_free_krbhst(context, *hostlist);
krb5_krbhst_free(context, handle);
return ENOMEM;
}
}
(*hostlist)[nhost] = NULL;
krb5_krbhst_free(context, handle);
return 0;
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_get_krb_admin_hst (krb5_context context,
const krb5_realm *realm,
char ***hostlist)
{
return gethostlist(context, *realm, KRB5_KRBHST_ADMIN, hostlist);
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_get_krb_changepw_hst (krb5_context context,
const krb5_realm *realm,
char ***hostlist)
{
return gethostlist(context, *realm, KRB5_KRBHST_CHANGEPW, hostlist);
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_get_krb524hst (krb5_context context,
const krb5_realm *realm,
char ***hostlist)
{
return gethostlist(context, *realm, KRB5_KRBHST_KRB524, hostlist);
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_get_krbhst (krb5_context context,
const krb5_realm *realm,
char ***hostlist)
{
return gethostlist(context, *realm, KRB5_KRBHST_KDC, hostlist);
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
krb5_free_krbhst (krb5_context context,
char **hostlist)
{
char **p;
for (p = hostlist; *p; ++p)
free (*p);
free (hostlist);
return 0;
}
#endif