Do a massive conversion for using the ast_verb() macro
[asterisk/asterisk.git] / main / srv.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 1999 - 2005, Digium, Inc.
5  *
6  * Mark Spencer <markster@digium.com>
7  *
8  * Funding provided by nic.at
9  *
10  * See http://www.asterisk.org for more information about
11  * the Asterisk project. Please do not directly contact
12  * any of the maintainers of this project for assistance;
13  * the project provides a web site, mailing lists and IRC
14  * channels for your use.
15  *
16  * This program is free software, distributed under the terms of
17  * the GNU General Public License Version 2. See the LICENSE file
18  * at the top of the source tree.
19  */
20
21 /*! \file
22  *
23  * \brief DNS SRV Record Lookup Support for Asterisk
24  *
25  * \author Mark Spencer <markster@digium.com>
26  * 
27  * \arg See also \ref AstENUM
28  *
29  * \note Funding provided by nic.at
30  */
31
32 #include "asterisk.h"
33
34 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
35
36 #include <sys/types.h>
37 #include <netinet/in.h>
38 #include <arpa/nameser.h>
39 #ifdef __APPLE__
40 #if __APPLE_CC__ >= 1495
41 #include <arpa/nameser_compat.h>
42 #endif
43 #endif
44 #include <resolv.h>
45 #include <stdio.h>
46 #include <string.h>
47 #include <unistd.h>
48
49 #include "asterisk/channel.h"
50 #include "asterisk/logger.h"
51 #include "asterisk/srv.h"
52 #include "asterisk/dns.h"
53 #include "asterisk/options.h"
54 #include "asterisk/utils.h"
55
56 #ifdef __APPLE__
57 #undef T_SRV
58 #define T_SRV 33
59 #endif
60
61 struct srv {
62         unsigned short priority;
63         unsigned short weight;
64         unsigned short portnum;
65 } __attribute__ ((__packed__));
66
67 static int parse_srv(char *host, int hostlen, int *portno, unsigned char *answer, int len, unsigned char *msg)
68 {
69         int res = 0;
70         struct srv *srv = (struct srv *)answer;
71         char repl[256] = "";
72
73         if (len < sizeof(struct srv)) {
74                 printf("Length too short\n");
75                 return -1;
76         }
77         answer += sizeof(struct srv);
78         len -= sizeof(struct srv);
79
80         if ((res = dn_expand(msg, answer + len, answer, repl, sizeof(repl) - 1)) < 0) {
81                 ast_log(LOG_WARNING, "Failed to expand hostname\n");
82                 return -1;
83         }
84         if (res && strcmp(repl, ".")) {
85                 ast_verb(3, "parse_srv: SRV mapped to host %s, port %d\n", repl, ntohs(srv->portnum));
86                 if (host) {
87                         ast_copy_string(host, repl, hostlen);
88                         host[hostlen-1] = '\0';
89                 }
90                 if (portno)
91                         *portno = ntohs(srv->portnum);
92                 return 0;
93         }
94         return -1;
95 }
96
97 struct srv_context {
98         char *host;
99         int hostlen;
100         int *port;
101 };
102
103 static int srv_callback(void *context, unsigned char *answer, int len, unsigned char *fullanswer)
104 {
105         struct srv_context *c = (struct srv_context *)context;
106
107         if (parse_srv(c->host, c->hostlen, c->port, answer, len, fullanswer)) {
108                 ast_log(LOG_WARNING, "Failed to parse srv\n");
109                 return -1;
110         }
111
112         if (!ast_strlen_zero(c->host))
113                 return 1;
114
115         return 0;
116 }
117
118 int ast_get_srv(struct ast_channel *chan, char *host, int hostlen, int *port, const char *service)
119 {
120         struct srv_context context;
121         int ret;
122
123         context.host = host;
124         context.hostlen = hostlen;
125         context.port = port;
126
127         if (chan && ast_autoservice_start(chan) < 0)
128                 return -1;
129
130         ret = ast_search_dns(&context, service, C_IN, T_SRV, srv_callback);
131
132         if (chan)
133                 ret |= ast_autoservice_stop(chan);
134
135         if (ret <= 0) {
136                 host[0] = '\0';
137                 *port = -1;
138                 return ret;
139         }
140         return ret;
141 }