Add missing enum files
[asterisk/asterisk.git] / enum.c
1 /*
2  * ENUM support
3  *
4  *
5  * ENUM Support for Asterisk
6  *
7  * Copyright (C) 2003 Digium
8  *
9  * Distributed under the terms of the GNU GPL
10  *
11  */
12
13 #include <string.h>
14 #include <fcntl.h>
15 #include <unistd.h>
16 #include <stdlib.h>
17 #include <resolv.h>
18 #include <errno.h>
19
20 #include <asterisk/logger.h>
21 #include <asterisk/options.h>
22 #include <asterisk/enum.h>
23 #include <asterisk/channel.h>
24
25 #define MAX_SIZE 4096
26
27 #define TOPLEV "e164.arpa."
28
29
30 static int skip_name(unsigned char *s, int len)
31 {
32         /* Shamelessly take from SER */
33         int x = 0;
34         while(x < len) {
35                 if (!*s) {
36                         s++;
37                         x++;
38                         break;
39                 }
40                 if (((*s) & 0xc0) == 0xc0) {
41                         s += 2;
42                         x += 2;
43                         break;
44                 }
45                 x += *s + 1;
46                 s += *s + 1;
47         }
48         if (x >= len)
49                 return -1;
50         return x;
51 }
52
53 struct dn_answer {
54         unsigned short rtype;
55         unsigned short class;
56         unsigned int ttl;
57         unsigned short size;
58 } __attribute__ ((__packed__));
59
60 struct naptr {
61         unsigned short order;
62         unsigned short pref;
63 } __attribute__ ((__packed__));
64
65 static int parse_ie(unsigned char *data, int maxdatalen, unsigned char *src, int srclen)
66 {
67         int len, olen;
68         len = olen = (int)src[0];
69         src++;
70         srclen--;
71         if (len > srclen) {
72                 ast_log(LOG_WARNING, "Want %d, got %d\n", len, srclen);
73                 return -1;
74         }
75         if (len > maxdatalen)
76                 len = maxdatalen;
77         memcpy(data, src, len);
78         return olen + 1;
79 }
80
81 static int parse_naptr(unsigned char *dst, int dstsize, char *tech, int techsize, unsigned char *answer, int len)
82 {
83         unsigned char *oanswer = answer;
84         unsigned char flags[80] = "";
85         unsigned char services[80] = "";
86         unsigned char regexp[80] = "";
87         unsigned char repl[80] = "";
88         int res;
89         
90         if (len < sizeof(struct naptr)) {
91                 printf("Length too short\n");
92                 return -1;
93         }
94         answer += sizeof(struct naptr);
95         len -= sizeof(struct naptr);
96         if ((res = parse_ie(flags, sizeof(flags) - 1, answer, len)) < 0) {
97                 ast_log(LOG_WARNING, "Failed to get flags\n");
98                 return -1; 
99         } else { answer += res; len -= res; }
100         if ((res = parse_ie(services, sizeof(services) - 1, answer, len)) < 0) {
101                 ast_log(LOG_WARNING, "Failed to get services\n");
102                 return -1; 
103         } else { answer += res; len -= res; }
104         if ((res = parse_ie(regexp, sizeof(regexp) - 1, answer, len)) < 0)
105                 return -1; else { answer += res; len -= res; }
106         if ((res = dn_expand(oanswer,answer + len,answer, repl, sizeof(repl) - 1)) < 0) {
107                 ast_log(LOG_WARNING, "Failed to expand hostname\n");
108                 return -1;
109         } 
110 #if 0
111         printf("Flags: %s\n", flags);
112         printf("Services: %s\n", services);
113         printf("Regexp: %s\n", regexp);
114         printf("Repl: %s\n", repl);
115 #endif
116         if (!strncmp(regexp, "!^.*$!", 6)) {
117                 if (!strncmp(services, "E2U+voice:", 10)) {
118                         if (regexp[strlen(regexp) - 1] == '!')
119                                 regexp[strlen(regexp) - 1] = '\0';
120 #if 0
121                         printf("Technology: %s\n", services + 10);
122                         printf("Destination: %s\n", regexp + 6);
123 #endif
124                         strncpy(dst, regexp + 6, dstsize);
125                         strncpy(tech, services + 10, techsize);
126                 }
127         } else
128                 ast_log(LOG_WARNING, "Non-total substitution not yet supported\n");
129         return 0;
130 }
131
132 static int parse_answer(unsigned char *dst, int dstlen, unsigned char *tech, int techlen, unsigned char *answer, int len)
133 {
134         /*
135          * This function is influenced by "ser" the SIP router.
136          */
137         int x;
138         int res;
139         HEADER *h;
140         struct dn_answer *ans;
141         dst[0] = '\0';
142         tech[0] = '\0';
143 #if 0
144         for (x=0;x<len;x++) {
145                 if ((answer[x] < 32) || (answer[x] > 127)) {
146                         if (lastlit)
147                                 printf("\"");
148                         printf(" 0x%02x", answer[x]);
149                         lastlit = 0;
150                 } else {
151                         if (!lastlit) 
152                                 printf(" \"");
153                         printf("%c", answer[x]);
154                         lastlit = 1;
155                 }
156         }
157         printf("\n");
158 #endif  
159         h = (HEADER *)answer;
160         /* Skip over DNS header */
161         answer += sizeof(HEADER);
162         len -= sizeof(HEADER);
163 #if 0
164         printf("Query count: %d\n", ntohs(h->qdcount));
165 #endif
166         for (x=0;x<ntohs(h->qdcount);x++) {
167                 if ((res = skip_name(answer, len)) < 0) {
168                         ast_log(LOG_WARNING, "Couldn't skip over name\n");
169                         return -1;
170                 }
171                 answer += res;
172                 len -= res;
173                 answer += 4;    /* Skip QCODE / QCLASS */
174                 len -= 4;
175                 if (len < 0) {
176                         ast_log(LOG_WARNING, "Strange query size\n");
177                         return -1;
178                 }
179         }
180 #if 0
181         printf("Length remaining: %d\n", len);
182         printf("Answer count: %d\n", ntohs(h->ancount));
183         printf("Looking for %d/%d\n", C_IN, T_NAPTR);
184 #endif
185         for (x=0;x<ntohs(h->ancount);x++) {
186                 if ((res = skip_name(answer, len) < 0)) {
187                         ast_log(LOG_WARNING, "Failed to skip name :(\n");
188                         return -1;
189                 }
190                 answer += res;
191                 len -= res;
192                 /* XXX Why am I adding 2 here? XXX */
193                 answer += 2;
194                 len -= 2;
195                 ans = (struct dn_answer *)answer;
196                 answer += sizeof(struct dn_answer);
197                 len -= sizeof(struct dn_answer);
198                 if (len < 0)
199                         return -1;
200 #if 0
201                 printf("Type: %d, class: %d, ttl: %d, length: %d\n", ntohs(ans->rtype), ntohs(ans->class),
202                         ntohl(ans->ttl), ntohs(ans->size));
203 #endif                  
204                 len -= ntohs(ans->size);
205                 if (len < 0) {
206                         ast_log(LOG_WARNING, "Length exceeds frame\n");
207                         return -1;
208                 }
209                 if ((ntohs(ans->class) == C_IN) && (ntohs(ans->rtype) == T_NAPTR)) {
210                         if (parse_naptr(dst, dstlen, tech, techlen, answer, ntohs(ans->size)))
211                                 ast_log(LOG_WARNING, "Failed to parse naptr :(\n");
212                         if (strlen(dst))
213                                 return 0;
214                 }
215                 answer += ntohs(ans->size);
216         }
217         return 0;
218 }
219
220 int ast_get_enum(struct ast_channel *chan, const char *number, char *dst, int dstlen, char *tech, int techlen)
221 {
222         unsigned char answer[MAX_SIZE];
223         char tmp[259 + strlen(TOPLEV)];
224         int pos = strlen(number) - 1;
225         int newpos=0;
226         int res;
227         int ret = -1;
228         struct __res_state enumstate;
229         res_ninit(&enumstate);  
230         if (chan && ast_autoservice_start(chan) < 0)
231                 return -1;
232         
233         if (pos > 128)
234                 pos = 128;
235         while(pos >= 0) {
236                 tmp[newpos++] = number[pos--];
237                 tmp[newpos++] = '.';
238         }
239         strcpy(tmp + newpos, TOPLEV);
240 #if 0
241         printf("Looking for '%s'\n", tmp);
242 #endif  
243         res = res_nsearch(&enumstate, tmp, C_IN, T_NAPTR, answer, sizeof(answer));
244         if (res > 0) {
245                 if ((res = parse_answer(dst, dstlen, tech, techlen, answer, res))) {
246                         ast_log(LOG_WARNING, "Parse error returned %d\n", res);
247                         ret = 0;
248                 } else {
249                         ast_log(LOG_DEBUG, "Found technology '%s', destination '%s'\n", tech, dst);
250                         ret = 1;
251                 }
252         } else {
253                 ast_log(LOG_DEBUG, "No such number found: %s (%s)\n", tmp, strerror(errno));
254                 ret = 0;
255         }
256         if (chan)
257                 ret |= ast_autoservice_stop(chan);
258         res_nclose(&enumstate);
259         return ret;
260 }