header ordering fixes for FreeBSD (pending a global merge into asterisk.h) (bug ...
[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 #if __APPLE_CC__ >= 1495
18 #include <arpa/nameser_compat.h>
19 #endif
20 #include <resolv.h>
21 #include <stdio.h>
22 #include <string.h>
23 #include <unistd.h>
24
25 #include "asterisk.h"
26
27 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
28
29 #include "asterisk/channel.h"
30 #include "asterisk/logger.h"
31 #include "asterisk/srv.h"
32 #include "asterisk/dns.h"
33 #include "asterisk/options.h"
34 #include "asterisk/utils.h"
35
36 #ifdef __APPLE__
37 #undef T_SRV
38 #define T_SRV 33
39 #endif
40
41 struct srv {
42         unsigned short priority;
43         unsigned short weight;
44         unsigned short portnum;
45 } __attribute__ ((__packed__));
46
47 static int parse_srv(unsigned char *host, int hostlen, int *portno, unsigned char *answer, int len, unsigned char *msg)
48 {
49         int res = 0;
50         struct srv *srv = (struct srv *)answer;
51         char repl[256] = "";
52
53         if (len < sizeof(struct srv)) {
54                 printf("Length too short\n");
55                 return -1;
56         }
57         answer += sizeof(struct srv);
58         len -= sizeof(struct srv);
59
60         if ((res = dn_expand(msg,answer + len,answer, repl, sizeof(repl) - 1)) < 0) {
61                 ast_log(LOG_WARNING, "Failed to expand hostname\n");
62                 return -1;
63         }
64         if (res && strcmp(repl, ".")) {
65                 if (option_verbose > 3)
66                         ast_verbose( VERBOSE_PREFIX_3 "parse_srv: SRV mapped to host %s, port %d\n", repl, ntohs(srv->portnum));
67                 if (host) {
68                         strncpy(host, repl, hostlen - 1);
69                         host[hostlen-1] = '\0';
70                 }
71                 if (portno)
72                         *portno = ntohs(srv->portnum);
73                 return(0);
74         }
75         return(-1);
76 }
77
78 struct srv_context {
79         char *host;
80         int hostlen;
81         int *port;
82 };
83
84 static int srv_callback(void *context, u_char *answer, int len, u_char *fullanswer)
85 {
86         struct srv_context *c = (struct srv_context *)context;
87
88         if (parse_srv(c->host, c->hostlen, c->port, answer, len, fullanswer)) {
89                 ast_log(LOG_WARNING, "Failed to parse srv\n");
90                 return -1;
91         }
92
93         if (!ast_strlen_zero(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                 host[0] = '\0';
118                 *port = -1;
119                 return ret;
120         }
121         return ret;
122 }