fix some formatting and add some comments (issue #5403)
[asterisk/asterisk.git] / 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 /*
22  *
23  * ENUM Support for Asterisk
24  *
25  */
26
27 #include <sys/types.h>
28 #include <netinet/in.h>
29 #include <arpa/nameser.h>
30 #if __APPLE_CC__ >= 1495
31 #include <arpa/nameser_compat.h>
32 #endif
33 #include <resolv.h>
34 #include <stdio.h>
35 #include <string.h>
36 #include <unistd.h>
37
38 #include "asterisk.h"
39
40 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
41
42 #include "asterisk/channel.h"
43 #include "asterisk/logger.h"
44 #include "asterisk/srv.h"
45 #include "asterisk/dns.h"
46 #include "asterisk/options.h"
47 #include "asterisk/utils.h"
48
49 #ifdef __APPLE__
50 #undef T_SRV
51 #define T_SRV 33
52 #endif
53
54 struct srv {
55         unsigned short priority;
56         unsigned short weight;
57         unsigned short portnum;
58 } __attribute__ ((__packed__));
59
60 static int parse_srv(char *host, int hostlen, int *portno, char *answer, int len, char *msg)
61 {
62         int res = 0;
63         struct srv *srv = (struct srv *)answer;
64         char repl[256] = "";
65
66         if (len < sizeof(struct srv)) {
67                 printf("Length too short\n");
68                 return -1;
69         }
70         answer += sizeof(struct srv);
71         len -= sizeof(struct srv);
72
73         if ((res = dn_expand((unsigned char *)msg, (unsigned char *)answer + len, (unsigned char *)answer, repl, sizeof(repl) - 1)) < 0) {
74                 ast_log(LOG_WARNING, "Failed to expand hostname\n");
75                 return -1;
76         }
77         if (res && strcmp(repl, ".")) {
78                 if (option_verbose > 3)
79                         ast_verbose( VERBOSE_PREFIX_3 "parse_srv: SRV mapped to host %s, port %d\n", repl, ntohs(srv->portnum));
80                 if (host) {
81                         ast_copy_string(host, repl, hostlen);
82                         host[hostlen-1] = '\0';
83                 }
84                 if (portno)
85                         *portno = ntohs(srv->portnum);
86                 return 0;
87         }
88         return -1;
89 }
90
91 struct srv_context {
92         char *host;
93         int hostlen;
94         int *port;
95 };
96
97 static int srv_callback(void *context, char *answer, int len, char *fullanswer)
98 {
99         struct srv_context *c = (struct srv_context *)context;
100
101         if (parse_srv(c->host, c->hostlen, c->port, answer, len, fullanswer)) {
102                 ast_log(LOG_WARNING, "Failed to parse srv\n");
103                 return -1;
104         }
105
106         if (!ast_strlen_zero(c->host))
107                 return 1;
108
109         return 0;
110 }
111
112 int ast_get_srv(struct ast_channel *chan, char *host, int hostlen, int *port, const char *service)
113 {
114         struct srv_context context;
115         int ret;
116
117         context.host = host;
118         context.hostlen = hostlen;
119         context.port = port;
120
121         if (chan && ast_autoservice_start(chan) < 0)
122                 return -1;
123
124         ret = ast_search_dns(&context, service, C_IN, T_SRV, srv_callback);
125
126         if (chan)
127                 ret |= ast_autoservice_stop(chan);
128
129         if (ret <= 0) {
130                 host[0] = '\0';
131                 *port = -1;
132                 return ret;
133         }
134         return ret;
135 }