update DNS resolver
[asterisk/asterisk.git] / srv.c
1 /*
2  * ENUM Support for Asterisk
3  *
4  * Copyright (C) 2003 Digium
5  *
6  * Written by Mark Spencer <markster@digium.com>
7  *
8  * Funding provided by nic.at
9  *
10  * Distributed under the terms of the GNU GPL
11  *
12  */
13
14 #include <sys/types.h>
15 #include <netinet/in.h>
16 #include <arpa/nameser.h>
17 #include <resolv.h>
18 #include <stdio.h>
19 #include <string.h>
20 #include <unistd.h>
21
22 #include <asterisk/channel.h>
23 #include <asterisk/logger.h>
24 #include <asterisk/srv.h>
25 #include <asterisk/dns.h>
26 #include <asterisk/options.h>
27
28 struct srv {
29         unsigned short priority;
30         unsigned short weight;
31         unsigned short portnum;
32 } __attribute__ ((__packed__));
33
34 static int parse_ie(unsigned char *data, int maxdatalen, unsigned char *src, int srclen)
35 {
36         int len, olen;
37         len = olen = (int)src[0];
38         src++;
39         srclen--;
40         if (len > srclen) {
41                 ast_log(LOG_WARNING, "Want %d, got %d\n", len, srclen);
42                 return -1;
43         }
44         if (len > maxdatalen)
45                 len = maxdatalen;
46         memcpy(data, src, len);
47         return olen + 1;
48 }
49
50 static int parse_srv(unsigned char *host, int hostlen, int *portno, unsigned char *answer, int len, unsigned char *msg)
51 {
52         int res = 0;
53         struct srv *srv = (struct srv *)answer;
54         char repl[256] = "";
55
56         if (len < sizeof(struct srv)) {
57                 printf("Length too short\n");
58                 return -1;
59         }
60         answer += sizeof(struct srv);
61         len -= sizeof(struct srv);
62
63         if ((res = dn_expand(msg,answer + len,answer, repl, sizeof(repl) - 1)) < 0) {
64                 ast_log(LOG_WARNING, "Failed to expand hostname\n");
65                 return -1;
66         }
67         if (res && strcmp(repl, ".")) {
68                 ast_verbose( VERBOSE_PREFIX_3 "parse_srv: SRV mapped to host %s, port %d\n", repl, ntohs(srv->portnum));
69                 if (host) {
70                         strncpy(host, repl, hostlen - 2);
71                         host[hostlen-1] = '\0';
72                 }
73                 if (portno)
74                         *portno = ntohs(srv->portnum);
75                 return(0);
76         }
77         return(-1);
78 }
79
80 struct srv_context {
81         char *host;
82         int hostlen;
83         int *port;
84 };
85
86 static int srv_callback(void *context, u_char *answer, int len, u_char *fullanswer)
87 {
88         struct srv_context *c = (struct srv_context *)context;
89
90         if (parse_srv(c->host, c->hostlen, c->port, answer, len, fullanswer))
91                 ast_log(LOG_WARNING, "Failed to parse srv\n");
92
93         if (strlen(c->host))
94                 return 1;
95
96     return 0;
97 }
98
99 int ast_get_srv(struct ast_channel *chan, char *host, int hostlen, int *port, const char *service)
100 {
101         struct srv_context context;
102         int ret;
103
104         context.host = host;
105         context.hostlen = hostlen;
106         context.port = port;
107
108         if (chan && ast_autoservice_start(chan) < 0)
109                 return -1;
110
111         ret = ast_search_dns(&context, service, C_IN, T_SRV, srv_callback);
112
113         if (chan)
114                 ret |= ast_autoservice_stop(chan);
115
116         if (ret <= 0)
117                 return ret;
118         strcpy(host, "");
119     *port = -1;
120         return ret;
121 }