Mercurial > pidgin
view libpurple/dnssrv.c @ 27930:ffbcfc22a7af
NEWSed
author | maiku@pidgin.im |
---|---|
date | Mon, 17 Aug 2009 22:27:31 +0000 |
parents | 7fbf964c6c6c |
children | bbb9e0ea67fe |
line wrap: on
line source
/** * @file dnssrv.c */ /* purple * * Copyright (C) 2005 Thomas Butter <butter@uni-mannheim.de> * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111-1301 USA */ #include "internal.h" #include "util.h" #ifndef _WIN32 #include <arpa/nameser.h> #include <resolv.h> #ifdef HAVE_ARPA_NAMESER_COMPAT_H #include <arpa/nameser_compat.h> #endif #ifndef T_SRV #define T_SRV 33 #endif #ifndef T_TXT #define T_TXT 16 #endif #else /* WIN32 */ #include <windns.h> /* Missing from the mingw headers */ #ifndef DNS_TYPE_SRV # define DNS_TYPE_SRV 33 #endif #ifndef DNS_TYPE_TXT # define DNS_TYPE_TXT 16 #endif #endif #include "debug.h" #include "dnssrv.h" #include "eventloop.h" #include "network.h" #ifndef _WIN32 typedef union { HEADER hdr; u_char buf[1024]; } queryans; #else static DNS_STATUS (WINAPI *MyDnsQuery_UTF8) ( PCSTR lpstrName, WORD wType, DWORD fOptions, PIP4_ARRAY aipServers, PDNS_RECORD* ppQueryResultsSet, PVOID* pReserved) = NULL; static void (WINAPI *MyDnsRecordListFree) (PDNS_RECORD pRecordList, DNS_FREE_TYPE FreeType) = NULL; #endif struct _PurpleTxtResponse { char *content; }; struct _PurpleSrvQueryData { union { PurpleSrvCallback srv; PurpleTxtCallback txt; } cb; gpointer extradata; guint handle; int type; #ifdef _WIN32 GThread *resolver; char *query; char *error_message; GList *results; #else int fd_in, fd_out; pid_t pid; #endif }; typedef struct _PurpleSrvInternalQuery { int type; char query[256]; } PurpleSrvInternalQuery; typedef struct _PurpleSrvResponseContainer { PurpleSrvResponse *response; int sum; } PurpleSrvResponseContainer; /** * Sort by priority, then by weight. Strictly numerically--no * randomness. Technically we only need to sort by pref and then * make sure any records with weight 0 are at the beginning of * their group, but it's just as easy to sort by weight. */ static gint responsecompare(gconstpointer ar, gconstpointer br) { PurpleSrvResponse *a = (PurpleSrvResponse*)ar; PurpleSrvResponse *b = (PurpleSrvResponse*)br; if(a->pref == b->pref) { if(a->weight == b->weight) return 0; if(a->weight < b->weight) return -1; return 1; } if(a->pref < b->pref) return -1; return 1; } /** * Iterate over a list of PurpleSrvResponseContainer making the sum * the running total of the sums. Select a random integer in the range * (1, sum+1), then find the first element greater than or equal to the * number selected. From RFC 2782. * * @param list The list of PurpleSrvResponseContainer. This function * removes a node from this list and returns the new list. * @param container_ptr The PurpleSrvResponseContainer that was chosen * will be returned here. */ static GList * select_random_response(GList *list, PurpleSrvResponseContainer **container_ptr) { GList *cur; size_t runningtotal; int r; runningtotal = 0; cur = list; while (cur) { PurpleSrvResponseContainer *container = cur->data; runningtotal += container->response->weight; container->sum = runningtotal; cur = cur->next; } /* * If the running total is greater than 0, pick a number between * 1 and the runningtotal inclusive. (This is not precisely what * the RFC algorithm describes, but we wish to deal with integers * and avoid floats. This is functionally equivalent.) * If running total is 0, then choose r = 0. */ r = runningtotal ? g_random_int_range(1, runningtotal + 1) : 0; cur = list; while (r > ((PurpleSrvResponseContainer *)cur->data)->sum) { cur = cur->next; } /* Set the return parameter and remove cur from the list */ *container_ptr = cur->data; return g_list_delete_link(list, cur); } /** * Reorder a GList of PurpleSrvResponses that have the same priority * (aka "pref"). */ static void srv_reorder(GList *list, int num) { int i; GList *cur, *container_list = NULL; PurpleSrvResponseContainer *container; if (num < 2) /* Nothing to sort */ return; /* First build a list of container structs */ for (i = 0, cur = list; i < num; i++, cur = cur->next) { container = g_new(PurpleSrvResponseContainer, 1); container->response = cur->data; container_list = g_list_prepend(container_list, container); } container_list = g_list_reverse(container_list); /* * Re-order the list that was passed in as a parameter. We leave * the list nodes in place, but replace their data pointers. */ cur = list; while (container_list) { container_list = select_random_response(container_list, &container); cur->data = container->response; g_free(container); cur = cur->next; } } /** * Sorts a GList of PurpleSrvResponses according to the * algorithm described in RFC 2782. * * @param response GList of PurpleSrvResponse's * @param The original list, resorted */ static GList * purple_srv_sort(GList *list) { int pref, count; GList *cur, *start; if (!list || !list->next) { /* Nothing to sort */ return list; } list = g_list_sort(list, responsecompare); start = cur = list; count = 1; while (cur) { PurpleSrvResponse *next_response; pref = ((PurpleSrvResponse *)cur->data)->pref; next_response = cur->next ? cur->next->data : NULL; if (!next_response || next_response->pref != pref) { /* * The 'count' records starting at 'start' all have the same * priority. Sort them by weight. */ srv_reorder(start, count); start = cur->next; count = 0; } count++; cur = cur->next; } return list; } static gboolean dns_str_is_ascii(const char *name) { guchar *c; for (c = (guchar *)name; c && *c; ++c) { if (*c > 0x7f) return FALSE; } return TRUE; } #ifndef _WIN32 G_GNUC_NORETURN static void resolve(int in, int out) { GList *ret = NULL; PurpleSrvResponse *srvres; PurpleTxtResponse *txtres; queryans answer; int size, qdcount, ancount; guchar *end, *cp; gchar name[256]; guint16 type, dlen, pref, weight, port; PurpleSrvInternalQuery query; #ifdef HAVE_SIGNAL_H purple_restore_default_signal_handlers(); #endif if (read(in, &query, sizeof(query)) <= 0) { close(out); close(in); _exit(0); } size = res_query( query.query, C_IN, query.type, (u_char*)&answer, sizeof( answer)); if (size == -1) { write(out, &(query.type), sizeof(query.type)); write(out, &size, sizeof(int)); close(out); close(in); _exit(0); } qdcount = ntohs(answer.hdr.qdcount); ancount = ntohs(answer.hdr.ancount); cp = (guchar*)&answer + sizeof(HEADER); end = (guchar*)&answer + size; /* skip over unwanted stuff */ while (qdcount-- > 0 && cp < end) { size = dn_expand( (unsigned char*)&answer, end, cp, name, 256); if(size < 0) goto end; cp += size + QFIXEDSZ; } while (ancount-- > 0 && cp < end) { size = dn_expand((unsigned char*)&answer, end, cp, name, 256); if(size < 0) goto end; cp += size; GETSHORT(type,cp); /* skip ttl and class since we already know it */ cp += 6; GETSHORT(dlen,cp); if (query.type == T_SRV) { GETSHORT(pref,cp); GETSHORT(weight,cp); GETSHORT(port,cp); size = dn_expand( (unsigned char*)&answer, end, cp, name, 256); if(size < 0 ) goto end; cp += size; srvres = g_new0(PurpleSrvResponse, 1); strcpy(srvres->hostname, name); srvres->pref = pref; srvres->port = port; srvres->weight = weight; ret = g_list_prepend(ret, srvres); } else if (query.type == T_TXT) { txtres = g_new0(PurpleTxtResponse, 1); txtres->content = g_strndup((gchar*)(++cp), dlen-1); ret = g_list_append(ret, txtres); cp += dlen - 1; } else { cp += dlen; } } end: size = g_list_length(ret); if (query.type == T_SRV) ret = purple_srv_sort(ret); /* TODO: Check return value */ write(out, &(query.type), sizeof(query.type)); write(out, &size, sizeof(size)); while (ret != NULL) { /* TODO: Check return value */ if (query.type == T_SRV) write(out, ret->data, sizeof(PurpleSrvResponse)); if (query.type == T_TXT) write(out, ret->data, sizeof(PurpleTxtResponse)); g_free(ret->data); ret = g_list_remove(ret, ret->data); } close(out); close(in); _exit(0); } static void resolved(gpointer data, gint source, PurpleInputCondition cond) { int size; int type; PurpleSrvQueryData *query_data = (PurpleSrvQueryData*)data; int i; int status; if (read(source, &type, sizeof(type)) == sizeof(type)) { if (read(source, &size, sizeof(size)) == sizeof(size)) { if (size == -1 || size == 0) { if (size == -1) { purple_debug_warning("dnssrv", "res_query returned an error\n"); /* Re-read resolv.conf and friends in case DNS servers have changed */ res_init(); } else purple_debug_info("dnssrv", "Found 0 entries, errno is %i\n", errno); if (type == T_SRV) { PurpleSrvCallback cb = query_data->cb.srv; cb(NULL, 0, query_data->extradata); } else if (type == T_TXT) { PurpleTxtCallback cb = query_data->cb.txt; cb(NULL, query_data->extradata); } else { purple_debug_error("dnssrv", "type unknown of DNS result entry; errno is %i\n", errno); } } else if (size) { if (type == T_SRV) { PurpleSrvResponse *res; PurpleSrvResponse *tmp; PurpleSrvCallback cb = query_data->cb.srv; ssize_t red; purple_debug_info("dnssrv","found %d SRV entries\n", size); tmp = res = g_new0(PurpleSrvResponse, size); for (i = 0; i < size; i++) { red = read(source, tmp++, sizeof(PurpleSrvResponse)); if (red != sizeof(PurpleSrvResponse)) { purple_debug_error("dnssrv","unable to read srv " "response: %s\n", g_strerror(errno)); size = 0; g_free(res); res = NULL; } } cb(res, size, query_data->extradata); } else if (type == T_TXT) { GList *responses = NULL; PurpleTxtResponse *res; PurpleTxtCallback cb = query_data->cb.txt; ssize_t red; purple_debug_info("dnssrv","found %d TXT entries\n", size); res = g_new0(PurpleTxtResponse, 1); for (i = 0; i < size; i++) { red = read(source, res, sizeof(PurpleTxtResponse)); if (red != sizeof(PurpleTxtResponse)) { purple_debug_error("dnssrv","unable to read txt " "response: %s\n", g_strerror(errno)); size = 0; g_free(res); g_list_foreach(responses, (GFunc)purple_txt_response_destroy, NULL); g_list_free(responses); responses = NULL; break; } } cb(responses, query_data->extradata); } else { purple_debug_error("dnssrv", "type unknown of DNS result entry; errno is %i\n", errno); } } } } waitpid(query_data->pid, &status, 0); purple_srv_cancel(query_data); } #else /* _WIN32 */ /** The Jabber Server code was inspiration for parts of this. */ static gboolean res_main_thread_cb(gpointer data) { PurpleSrvResponse *srvres = NULL; PurpleSrvQueryData *query_data = data; if(query_data->error_message != NULL) purple_debug_error("dnssrv", query_data->error_message); else { if (query_data->type == DNS_TYPE_SRV) { PurpleSrvResponse *srvres_tmp = NULL; GList *lst = query_data->results; int size = g_list_length(lst); if(query_data->cb.srv && size > 0) srvres_tmp = srvres = g_new0(PurpleSrvResponse, size); while (lst) { PurpleSrvResponse *lstdata = lst->data; lst = g_list_delete_link(lst, lst); if(query_data->cb.srv) memcpy(srvres_tmp++, lstdata, sizeof(PurpleSrvResponse)); g_free(lstdata); } query_data->results = NULL; purple_debug_info("dnssrv", "found %d SRV entries\n", size); if(query_data->cb.srv) query_data->cb.srv(srvres, size, query_data->extradata); } else if (query_data->type == DNS_TYPE_TXT) { GList *lst = query_data->results; purple_debug_info("dnssrv", "found %d TXT entries\n", g_list_length(lst)); if (query_data->cb.txt) { query_data->results = NULL; query_data->cb.txt(lst, query_data->extradata); } } else { purple_debug_error("dnssrv", "unknown query type"); } } query_data->resolver = NULL; query_data->handle = 0; purple_srv_cancel(query_data); return FALSE; } static gpointer res_thread(gpointer data) { PDNS_RECORD dr = NULL; int type; DNS_STATUS ds; PurpleSrvQueryData *query_data = data; type = query_data->type; ds = MyDnsQuery_UTF8(query_data->query, type, DNS_QUERY_STANDARD, NULL, &dr, NULL); if (ds != ERROR_SUCCESS) { gchar *msg = g_win32_error_message(ds); if (type == DNS_TYPE_SRV) { query_data->error_message = g_strdup_printf("Couldn't look up SRV record. %s (%lu).\n", msg, ds); } else if (type == DNS_TYPE_TXT) { query_data->error_message = g_strdup_printf("Couldn't look up TXT record. %s (%lu).\n", msg, ds); } g_free(msg); } else { if (type == DNS_TYPE_SRV) { PDNS_RECORD dr_tmp; GList *lst = NULL; DNS_SRV_DATA *srv_data; PurpleSrvResponse *srvres; for (dr_tmp = dr; dr_tmp != NULL; dr_tmp = dr_tmp->pNext) { /* Discard any incorrect entries. I'm not sure if this is necessary */ if (dr_tmp->wType != type || strcmp(dr_tmp->pName, query_data->query) != 0) { continue; } srv_data = &dr_tmp->Data.SRV; srvres = g_new0(PurpleSrvResponse, 1); strncpy(srvres->hostname, srv_data->pNameTarget, 255); srvres->hostname[255] = '\0'; srvres->pref = srv_data->wPriority; srvres->port = srv_data->wPort; srvres->weight = srv_data->wWeight; lst = g_list_prepend(lst, srvres); } MyDnsRecordListFree(dr, DnsFreeRecordList); query_data->results = purple_srv_sort(lst); } else if (type == DNS_TYPE_TXT) { PDNS_RECORD dr_tmp; GList *lst = NULL; DNS_TXT_DATA *txt_data; PurpleTxtResponse *txtres; for (dr_tmp = dr; dr_tmp != NULL; dr_tmp = dr_tmp->pNext) { GString *s; int i; /* Discard any incorrect entries. I'm not sure if this is necessary */ if (dr_tmp->wType != type || strcmp(dr_tmp->pName, query_data->query) != 0) { continue; } txt_data = &dr_tmp->Data.TXT; txtres = g_new0(PurpleTxtResponse, 1); s = g_string_new(""); for (i = 0; i < txt_data->dwStringCount; ++i) s = g_string_append(s, txt_data->pStringArray[i]); txtres->content = g_string_free(s, FALSE); lst = g_list_append(lst, txtres); } MyDnsRecordListFree(dr, DnsFreeRecordList); query_data->results = lst; } else { } } /* back to main thread */ /* Note: this should *not* be attached to query_data->handle - it will cause leakage */ purple_timeout_add(0, res_main_thread_cb, query_data); g_thread_exit(NULL); return NULL; } #endif PurpleSrvQueryData * purple_srv_resolve(const char *protocol, const char *transport, const char *domain, PurpleSrvCallback cb, gpointer extradata) { char *query; char *hostname; PurpleSrvQueryData *query_data; #ifndef _WIN32 PurpleSrvInternalQuery internal_query; int in[2], out[2]; int pid; #else GError* err = NULL; static gboolean initialized = FALSE; #endif if (!protocol || !*protocol || !transport || !*transport || !domain || !*domain) { purple_debug_error("dnssrv", "Wrong arguments\n"); cb(NULL, 0, extradata); g_return_val_if_reached(NULL); } #ifdef USE_IDN if (!dns_str_is_ascii(domain)) { int ret = purple_network_convert_idn_to_ascii(domain, &hostname); if (ret != 0) { purple_debug_error("dnssrv", "IDNA ToASCII failed\n"); cb(NULL, 0, extradata); return NULL; } } else /* Fallthru is intentional */ #endif hostname = g_strdup(domain); query = g_strdup_printf("_%s._%s.%s", protocol, transport, hostname); purple_debug_info("dnssrv","querying SRV record for %s: %s\n", domain, query); g_free(hostname); #ifndef _WIN32 if(pipe(in) || pipe(out)) { purple_debug_error("dnssrv", "Could not create pipe\n"); g_free(query); cb(NULL, 0, extradata); return NULL; } pid = fork(); if (pid == -1) { purple_debug_error("dnssrv", "Could not create process!\n"); cb(NULL, 0, extradata); g_free(query); return NULL; } /* Child */ if (pid == 0) { g_free(query); close(out[0]); close(in[1]); resolve(in[0], out[1]); /* resolve() does not return */ } close(out[1]); close(in[0]); internal_query.type = T_SRV; strncpy(internal_query.query, query, 255); if (write(in[1], &internal_query, sizeof(internal_query)) < 0) purple_debug_error("dnssrv", "Could not write to SRV resolver\n"); query_data = g_new0(PurpleSrvQueryData, 1); query_data->type = T_SRV; query_data->cb.srv = cb; query_data->extradata = extradata; query_data->pid = pid; query_data->fd_out = out[0]; query_data->fd_in = in[1]; query_data->handle = purple_input_add(out[0], PURPLE_INPUT_READ, resolved, query_data); g_free(query); return query_data; #else if (!initialized) { MyDnsQuery_UTF8 = (void*) wpurple_find_and_loadproc("dnsapi.dll", "DnsQuery_UTF8"); MyDnsRecordListFree = (void*) wpurple_find_and_loadproc( "dnsapi.dll", "DnsRecordListFree"); initialized = TRUE; } query_data = g_new0(PurpleSrvQueryData, 1); query_data->type = DNS_TYPE_SRV; query_data->cb.srv = cb; query_data->query = query; query_data->extradata = extradata; if (!MyDnsQuery_UTF8 || !MyDnsRecordListFree) query_data->error_message = g_strdup("System missing DNS API (Requires W2K+)\n"); else { query_data->resolver = g_thread_create(res_thread, query_data, FALSE, &err); if (query_data->resolver == NULL) { query_data->error_message = g_strdup_printf("SRV thread create failure: %s\n", (err && err->message) ? err->message : ""); g_error_free(err); } } /* The query isn't going to happen, so finish the SRV lookup now. * Asynchronously call the callback since stuff may not expect * the callback to be called before this returns */ if (query_data->error_message != NULL) query_data->handle = purple_timeout_add(0, res_main_thread_cb, query_data); return query_data; #endif } PurpleSrvQueryData *purple_txt_resolve(const char *owner, const char *domain, PurpleTxtCallback cb, gpointer extradata) { char *query; char *hostname; PurpleSrvQueryData *query_data; #ifndef _WIN32 PurpleSrvInternalQuery internal_query; int in[2], out[2]; int pid; #else GError* err = NULL; static gboolean initialized = FALSE; #endif #ifdef USE_IDN if (!dns_str_is_ascii(domain)) { int ret = purple_network_convert_idn_to_ascii(domain, &hostname); if (ret != 0) { purple_debug_error("dnssrv", "IDNA ToASCII failed\n"); cb(NULL, extradata); return NULL; } } else /* fallthru is intentional */ #endif hostname = g_strdup(domain); query = g_strdup_printf("%s.%s", owner, hostname); purple_debug_info("dnssrv","querying TXT record for %s: %s\n", domain, query); g_free(hostname); #ifndef _WIN32 if(pipe(in) || pipe(out)) { purple_debug_error("dnssrv", "Could not create pipe\n"); g_free(query); cb(NULL, extradata); return NULL; } pid = fork(); if (pid == -1) { purple_debug_error("dnssrv", "Could not create process!\n"); cb(NULL, extradata); g_free(query); return NULL; } /* Child */ if (pid == 0) { g_free(query); close(out[0]); close(in[1]); resolve(in[0], out[1]); /* resolve() does not return */ } close(out[1]); close(in[0]); internal_query.type = T_TXT; strncpy(internal_query.query, query, 255); if (write(in[1], &internal_query, sizeof(internal_query)) < 0) purple_debug_error("dnssrv", "Could not write to TXT resolver\n"); query_data = g_new0(PurpleSrvQueryData, 1); query_data->type = T_TXT; query_data->cb.txt = cb; query_data->extradata = extradata; query_data->pid = pid; query_data->fd_out = out[0]; query_data->fd_in = in[1]; query_data->handle = purple_input_add(out[0], PURPLE_INPUT_READ, resolved, query_data); g_free(query); return query_data; #else if (!initialized) { MyDnsQuery_UTF8 = (void*) wpurple_find_and_loadproc("dnsapi.dll", "DnsQuery_UTF8"); MyDnsRecordListFree = (void*) wpurple_find_and_loadproc( "dnsapi.dll", "DnsRecordListFree"); initialized = TRUE; } query_data = g_new0(PurpleSrvQueryData, 1); query_data->type = DNS_TYPE_TXT; query_data->cb.txt = cb; query_data->query = query; query_data->extradata = extradata; if (!MyDnsQuery_UTF8 || !MyDnsRecordListFree) query_data->error_message = g_strdup("System missing DNS API (Requires W2K+)\n"); else { query_data->resolver = g_thread_create(res_thread, query_data, FALSE, &err); if (query_data->resolver == NULL) { query_data->error_message = g_strdup_printf("TXT thread create failure: %s\n", (err && err->message) ? err->message : ""); g_error_free(err); } } /* The query isn't going to happen, so finish the TXT lookup now. * Asynchronously call the callback since stuff may not expect * the callback to be called before this returns */ if (query_data->error_message != NULL) query_data->handle = purple_timeout_add(0, res_main_thread_cb, query_data); return query_data; #endif } void purple_srv_cancel(PurpleSrvQueryData *query_data) { if (query_data->handle > 0) purple_input_remove(query_data->handle); #ifdef _WIN32 if (query_data->resolver != NULL) { /* * It's not really possible to kill a thread. So instead we * just set the callback to NULL and let the DNS lookup * finish. */ query_data->cb.srv = NULL; return; } g_free(query_data->query); g_free(query_data->error_message); #else close(query_data->fd_out); close(query_data->fd_in); #endif g_free(query_data); } void purple_txt_cancel(PurpleSrvQueryData *query_data) { purple_srv_cancel(query_data); } const gchar * purple_txt_response_get_content(PurpleTxtResponse *resp) { g_return_val_if_fail(resp != NULL, NULL); return resp->content; } void purple_txt_response_destroy(PurpleTxtResponse *resp) { g_return_if_fail(resp != NULL); g_free(resp->content); g_free(resp); }