14192
|
1 /**
|
|
2 * @file dnssrv.c
|
|
3 *
|
|
4 * gaim
|
|
5 *
|
|
6 * Copyright (C) 2005 Thomas Butter <butter@uni-mannheim.de>
|
|
7 *
|
|
8 * This program is free software; you can redistribute it and/or modify
|
|
9 * it under the terms of the GNU General Public License as published by
|
|
10 * the Free Software Foundation; either version 2 of the License, or
|
|
11 * (at your option) any later version.
|
|
12 *
|
|
13 * This program is distributed in the hope that it will be useful,
|
|
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
16 * GNU General Public License for more details.
|
|
17 *
|
|
18 * You should have received a copy of the GNU General Public License
|
|
19 * along with this program; if not, write to the Free Software
|
|
20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
21 */
|
|
22
|
|
23 #include "internal.h"
|
|
24
|
|
25 #ifndef _WIN32
|
|
26 #include <resolv.h>
|
|
27 #include <arpa/nameser.h>
|
|
28 #ifdef HAVE_ARPA_NAMESER_COMPAT_H
|
|
29 #include <arpa/nameser_compat.h>
|
|
30 #endif
|
|
31 #ifndef T_SRV
|
|
32 #define T_SRV 33
|
|
33 #endif
|
|
34 #else
|
|
35 #include <windns.h>
|
|
36 /* Missing from the mingw headers */
|
|
37 #ifndef DNS_TYPE_SRV
|
|
38 # define DNS_TYPE_SRV 33
|
|
39 #endif
|
|
40 #endif
|
|
41
|
|
42 #include "dnssrv.h"
|
|
43 #include "eventloop.h"
|
|
44 #include "debug.h"
|
|
45
|
|
46 #ifndef _WIN32
|
|
47 typedef union {
|
|
48 HEADER hdr;
|
|
49 u_char buf[1024];
|
|
50 } queryans;
|
|
51 #else
|
|
52 static DNS_STATUS WINAPI (*MyDnsQuery_UTF8) (
|
|
53 PCSTR lpstrName, WORD wType, DWORD fOptions,
|
|
54 PIP4_ARRAY aipServers, PDNS_RECORD* ppQueryResultsSet,
|
|
55 PVOID* pReserved) = NULL;
|
|
56 static void WINAPI (*MyDnsRecordListFree) (PDNS_RECORD pRecordList,
|
|
57 DNS_FREE_TYPE FreeType) = NULL;
|
|
58 #endif
|
|
59
|
14308
|
60 struct _GaimSrvQueryData {
|
|
61 GaimSrvCallback cb;
|
14192
|
62 gpointer extradata;
|
|
63 #ifndef _WIN32
|
|
64 guint handle;
|
|
65 #else
|
14308
|
66 GThread *resolver;
|
14192
|
67 char *query;
|
14308
|
68 char *error_message;
|
14192
|
69 GSList *results;
|
|
70 #endif
|
|
71 };
|
|
72
|
14308
|
73 static gint
|
|
74 responsecompare(gconstpointer ar, gconstpointer br)
|
|
75 {
|
14192
|
76 GaimSrvResponse *a = (GaimSrvResponse*)ar;
|
|
77 GaimSrvResponse *b = (GaimSrvResponse*)br;
|
|
78
|
|
79 if(a->pref == b->pref) {
|
|
80 if(a->weight == b->weight)
|
|
81 return 0;
|
|
82 if(a->weight < b->weight)
|
|
83 return -1;
|
|
84 return 1;
|
|
85 }
|
|
86 if(a->pref < b->pref)
|
|
87 return -1;
|
|
88 return 1;
|
|
89 }
|
14308
|
90
|
14192
|
91 #ifndef _WIN32
|
14308
|
92
|
|
93 static void
|
|
94 resolve(int in, int out)
|
|
95 {
|
14192
|
96 GList *ret = NULL;
|
|
97 GaimSrvResponse *srvres;
|
|
98 queryans answer;
|
|
99 int size;
|
|
100 int qdcount;
|
|
101 int ancount;
|
|
102 guchar *end;
|
|
103 guchar *cp;
|
|
104 gchar name[256];
|
|
105 guint16 type, dlen, pref, weight, port;
|
|
106 gchar query[256];
|
|
107
|
14308
|
108 if (read(in, query, 256) <= 0)
|
14192
|
109 _exit(0);
|
14308
|
110
|
14192
|
111 size = res_query( query, C_IN, T_SRV, (u_char*)&answer, sizeof( answer));
|
|
112
|
|
113 qdcount = ntohs(answer.hdr.qdcount);
|
|
114 ancount = ntohs(answer.hdr.ancount);
|
|
115
|
|
116 cp = (guchar*)&answer + sizeof(HEADER);
|
|
117 end = (guchar*)&answer + size;
|
|
118
|
|
119 /* skip over unwanted stuff */
|
|
120 while (qdcount-- > 0 && cp < end) {
|
|
121 size = dn_expand( (unsigned char*)&answer, end, cp, name, 256);
|
|
122 if(size < 0) goto end;
|
|
123 cp += size + QFIXEDSZ;
|
|
124 }
|
|
125
|
|
126 while (ancount-- > 0 && cp < end) {
|
|
127 size = dn_expand((unsigned char*)&answer, end, cp, name, 256);
|
|
128 if(size < 0)
|
|
129 goto end;
|
|
130
|
|
131 cp += size;
|
|
132
|
|
133 GETSHORT(type,cp);
|
|
134
|
|
135 /* skip ttl and class since we already know it */
|
|
136 cp += 6;
|
|
137
|
|
138 GETSHORT(dlen,cp);
|
|
139
|
|
140 if (type == T_SRV) {
|
|
141 GETSHORT(pref,cp);
|
|
142
|
|
143 GETSHORT(weight,cp);
|
|
144
|
|
145 GETSHORT(port,cp);
|
|
146
|
|
147 size = dn_expand( (unsigned char*)&answer, end, cp, name, 256);
|
|
148 if(size < 0 )
|
|
149 goto end;
|
|
150
|
|
151 cp += size;
|
|
152
|
|
153 srvres = g_new0(GaimSrvResponse, 1);
|
|
154 strcpy(srvres->hostname, name);
|
|
155 srvres->pref = pref;
|
|
156 srvres->port = port;
|
|
157 srvres->weight = weight;
|
|
158
|
|
159 ret = g_list_insert_sorted(ret, srvres, responsecompare);
|
|
160 } else {
|
|
161 cp += dlen;
|
|
162 }
|
|
163 }
|
14308
|
164
|
|
165 end:
|
|
166 size = g_list_length(ret);
|
14192
|
167 write(out, &size, sizeof(int));
|
14308
|
168 while (ret != NULL)
|
|
169 {
|
|
170 write(out, ret->data, sizeof(GaimSrvResponse));
|
|
171 g_free(ret->data);
|
|
172 ret = g_list_remove(ret, ret->data);
|
14192
|
173 }
|
|
174
|
|
175 _exit(0);
|
|
176 }
|
|
177
|
14308
|
178 static void
|
|
179 resolved(gpointer data, gint source, GaimInputCondition cond)
|
|
180 {
|
14192
|
181 int size;
|
14308
|
182 GaimSrvQueryData *query_data = (GaimSrvQueryData*)data;
|
14192
|
183 GaimSrvResponse *res;
|
|
184 GaimSrvResponse *tmp;
|
|
185 int i;
|
14308
|
186 GaimSrvCallback cb = query_data->cb;
|
14192
|
187
|
|
188 read(source, &size, sizeof(int));
|
14308
|
189 gaim_debug_info("dnssrv","found %d SRV entries\n", size);
|
14192
|
190 tmp = res = g_new0(GaimSrvResponse, size);
|
14308
|
191 for (i = 0; i < size; i++) {
|
14192
|
192 read(source, tmp++, sizeof(GaimSrvResponse));
|
|
193 }
|
14308
|
194
|
|
195 cb(res, size, query_data->extradata);
|
|
196
|
|
197 gaim_srv_cancel(query_data);
|
14192
|
198 }
|
|
199
|
|
200 #else /* _WIN32 */
|
|
201
|
|
202 /** The Jabber Server code was inspiration for parts of this. */
|
|
203
|
14308
|
204 static gboolean
|
|
205 res_main_thread_cb(gpointer data)
|
|
206 {
|
14192
|
207 GaimSrvResponse *srvres = NULL;
|
|
208 int size = 0;
|
14308
|
209 GaimSrvQueryData *query_data = data;
|
14192
|
210
|
14308
|
211 if (query_data->error_message != NULL) {
|
|
212 gaim_debug_error("dnssrv", query_data->error_message);
|
14192
|
213 } else {
|
|
214 GaimSrvResponse *srvres_tmp;
|
14308
|
215 GSList *lst = query_data->results;
|
14192
|
216
|
14308
|
217 size = g_slist_length(query_data->results);
|
14192
|
218
|
|
219 srvres_tmp = srvres = g_new0(GaimSrvResponse, size);
|
|
220 while (lst) {
|
|
221 memcpy(srvres_tmp++, lst->data, sizeof(GaimSrvResponse));
|
|
222 g_free(lst->data);
|
|
223 lst = g_slist_remove(lst, lst->data);
|
|
224 }
|
|
225
|
14308
|
226 query_data->results = lst;
|
14192
|
227 }
|
|
228
|
14308
|
229 gaim_debug_info("dnssrv", "found %d SRV entries\n", size);
|
14192
|
230
|
14308
|
231 query_data->cb(srvres, size, query_data->extradata);
|
14192
|
232
|
14308
|
233 gaim_srv_cancel(query_data);
|
14192
|
234
|
|
235 return FALSE;
|
|
236 }
|
|
237
|
14308
|
238 static gpointer
|
|
239 res_thread(gpointer data)
|
|
240 {
|
14192
|
241 PDNS_RECORD dr = NULL;
|
|
242 int type = DNS_TYPE_SRV;
|
|
243 DNS_STATUS ds;
|
14308
|
244 GaimSrvQueryData *query_data = data;
|
14192
|
245
|
14308
|
246 ds = MyDnsQuery_UTF8(query_data->query, type, DNS_QUERY_STANDARD, NULL, &dr, NULL);
|
14192
|
247 if (ds != ERROR_SUCCESS) {
|
|
248 gchar *msg = g_win32_error_message(ds);
|
14308
|
249 query_data->error_message = g_strdup_printf("Couldn't look up SRV record. %s (%lu).\n", msg, ds);
|
14192
|
250 g_free(msg);
|
|
251 } else {
|
|
252 PDNS_RECORD dr_tmp;
|
|
253 GSList *lst = NULL;
|
|
254 DNS_SRV_DATA *srv_data;
|
|
255 GaimSrvResponse *srvres;
|
|
256
|
|
257 for (dr_tmp = dr; dr_tmp != NULL; dr_tmp = dr_tmp->pNext) {
|
|
258 /* Discard any incorrect entries. I'm not sure if this is necessary */
|
14308
|
259 if (dr_tmp->wType != type || strcmp(dr_tmp->pName, query_data->query) != 0) {
|
14192
|
260 continue;
|
|
261 }
|
|
262
|
|
263 srv_data = &dr_tmp->Data.SRV;
|
|
264 srvres = g_new0(GaimSrvResponse, 1);
|
|
265 strncpy(srvres->hostname, srv_data->pNameTarget, 255);
|
|
266 srvres->hostname[255] = '\0';
|
|
267 srvres->pref = srv_data->wPriority;
|
|
268 srvres->port = srv_data->wPort;
|
|
269 srvres->weight = srv_data->wWeight;
|
|
270
|
|
271 lst = g_slist_insert_sorted(lst, srvres, responsecompare);
|
|
272 }
|
|
273
|
|
274 MyDnsRecordListFree(dr, DnsFreeRecordList);
|
14308
|
275 query_data->results = lst;
|
14192
|
276 }
|
|
277
|
|
278 /* back to main thread */
|
14308
|
279 g_idle_add(res_main_thread_cb, query_data);
|
14192
|
280
|
|
281 g_thread_exit(NULL);
|
|
282 return NULL;
|
|
283 }
|
|
284
|
|
285 #endif
|
|
286
|
14308
|
287 GaimSrvQueryData *
|
|
288 gaim_srv_resolve(const char *protocol, const char *transport, const char *domain, GaimSrvCallback cb, gpointer extradata)
|
|
289 {
|
|
290 char *query;
|
|
291 GaimSrvQueryData *query_data;
|
14192
|
292 #ifndef _WIN32
|
|
293 int in[2], out[2];
|
|
294 int pid;
|
14308
|
295 #else
|
|
296 GError* err = NULL;
|
|
297 static gboolean initialized = FALSE;
|
|
298 #endif
|
|
299
|
|
300 #ifndef _WIN32
|
|
301 query = g_strdup_printf("_%s._%s.%s", protocol, transport, domain);
|
|
302 gaim_debug_info("dnssrv","querying SRV record for %s\n", query);
|
|
303
|
14192
|
304 if(pipe(in) || pipe(out)) {
|
14308
|
305 gaim_debug_error("dnssrv", "Could not create pipe\n");
|
14192
|
306 g_free(query);
|
|
307 cb(NULL, 0, extradata);
|
14308
|
308 return NULL;
|
14192
|
309 }
|
|
310
|
|
311 pid = fork();
|
14308
|
312 if (pid == -1) {
|
|
313 gaim_debug_error("dnssrv", "Could not create process!\n");
|
14192
|
314 cb(NULL, 0, extradata);
|
|
315 g_free(query);
|
14308
|
316 return NULL;
|
14192
|
317 }
|
14308
|
318
|
14192
|
319 /* Child */
|
14308
|
320 if (pid == 0)
|
|
321 {
|
14192
|
322 close(out[0]);
|
|
323 close(in[1]);
|
|
324 resolve(in[0], out[1]);
|
|
325 }
|
|
326
|
|
327 close(out[1]);
|
|
328 close(in[0]);
|
|
329
|
14308
|
330 if (write(in[1], query, strlen(query)+1) < 0)
|
|
331 gaim_debug_error("dnssrv", "Could not write to SRV resolver\n");
|
|
332
|
|
333 query_data = g_new0(GaimSrvQueryData, 1);
|
|
334 query_data->cb = cb;
|
|
335 query_data->extradata = extradata;
|
|
336 query_data->handle = gaim_input_add(out[0], GAIM_INPUT_READ, resolved, query_data);
|
14192
|
337
|
|
338 g_free(query);
|
14308
|
339
|
|
340 return query_data;
|
14192
|
341 #else
|
|
342 if (!initialized) {
|
|
343 MyDnsQuery_UTF8 = (void*) wgaim_find_and_loadproc("dnsapi.dll", "DnsQuery_UTF8");
|
|
344 MyDnsRecordListFree = (void*) wgaim_find_and_loadproc(
|
|
345 "dnsapi.dll", "DnsRecordListFree");
|
|
346 initialized = TRUE;
|
|
347 }
|
|
348
|
|
349 if (!MyDnsQuery_UTF8 || !MyDnsRecordListFree) {
|
14308
|
350 gaim_debug_error("dnssrv", "System missing DNS API (Requires W2K+)\n");
|
14192
|
351 g_free(query);
|
|
352 cb(NULL, 0, extradata);
|
14308
|
353 return NULL;
|
14192
|
354 }
|
|
355
|
14308
|
356 query_data = g_new0(GaimSrvQueryData, 1);
|
|
357 query_data->cb = cb;
|
|
358 query_data->query = query;
|
|
359 query_data->extradata = extradata;
|
14192
|
360
|
14308
|
361 query_data->resolver = g_thread_create(res_thread, query_data, FALSE, &err);
|
|
362 if (query_data->resolver == NULL)
|
|
363 {
|
|
364 query_data->error_message = g_strdup_printf("SRV thread create failure: %s\n", err ? err->message : "");
|
14192
|
365 g_error_free(err);
|
14308
|
366 res_main_thread_cb(query_data);
|
|
367 return NULL;
|
14192
|
368 }
|
14308
|
369
|
|
370 return query_data;
|
14192
|
371 #endif
|
|
372 }
|
|
373
|
14308
|
374 void
|
|
375 gaim_srv_cancel(GaimSrvQueryData *query_data)
|
|
376 {
|
|
377 #ifndef _WIN32
|
|
378 if (query_data->handle > 0)
|
|
379 gaim_input_remove(query_data->handle);
|
|
380 #else
|
|
381 if (query_data->resolver != NULL)
|
|
382 {
|
|
383 /*
|
|
384 * It's not really possible to kill a thread. So instead we
|
|
385 * just set the callback to NULL and let the DNS lookup
|
|
386 * finish.
|
|
387 */
|
|
388 query_data->callback = NULL;
|
|
389 return;
|
|
390 }
|
|
391 g_free(query_data->query);
|
|
392 g_free(query_data->error_message);
|
|
393 #endif
|
|
394 g_free(query_data);
|
|
395 }
|