Annotation of embedaddon/mtr/asn.c, revision 1.1.1.1
1.1 misho 1: /*
2: mtr -- a network diagnostic tool
3: Copyright (C) 1997,1998 Matt Kimball
4:
5: This program is free software; you can redistribute it and/or modify
6: it under the terms of the GNU General Public License version 2 as
7: published by the Free Software Foundation.
8:
9: This program is distributed in the hope that it will be useful,
10: but WITHOUT ANY WARRANTY; without even the implied warranty of
11: MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12: GNU General Public License for more details.
13:
14: You should have received a copy of the GNU General Public License
15: along with this program; if not, write to the Free Software
16: Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17: */
18:
19: #include <unistd.h>
20: #include <stdio.h>
21: #include <stdlib.h>
22: #include <sys/types.h>
23:
24: #ifndef __APPLE__
25: #define BIND_8_COMPAT
26: #endif
27: #include <arpa/nameser.h>
28: #ifdef HAVE_ARPA_NAMESER_COMPAT_H
29: #include <arpa/nameser_compat.h>
30: #endif
31: #include <netdb.h>
32: #include <netinet/in.h>
33: #include <resolv.h>
34: #include <string.h>
35: #include <sys/socket.h>
36: #include <search.h>
37:
38: #include "config.h"
39: #include "mtr.h"
40: #include "asn.h"
41:
42: /*
43: #ifndef IIDEBUG
44: #define IIDEBUG
45: #include <syslog.h>
46: #endif
47: */
48:
49: #define IIHASH_HI 128
50: #define ITEMSMAX 15
51: #define ITEMSEP '|'
52: #define NAMELEN 127
53: #define UNKN "???"
54:
55: int ipinfo_no = -1;
56: int ipinfo_max = -1;
57: int iihash = 0;
58: char fmtinfo[32];
59: extern int af; /* address family of remote target */
60:
61: // items width: ASN, Route, Country, Registry, Allocated
62: int iiwidth[] = { 6, 19, 4, 8, 11}; // item len + space
63: int iiwidth_len = sizeof(iiwidth)/sizeof((iiwidth)[0]);
64:
65: typedef char* items_t[ITEMSMAX + 1];
66: items_t items_a; // without hash: items
67: char txtrec[NAMELEN + 1]; // without hash: txtrec
68: items_t* items = &items_a;
69:
70:
71: char *ipinfo_lookup(const char *domain) {
72: unsigned char answer[PACKETSZ], *pt;
73: char host[128];
74: char *txt;
75: int len, exp, size, txtlen, type;
76:
77:
78: if(res_init() < 0) {
79: fprintf(stderr,"@res_init failed\n");
80: return NULL;
81: }
82:
83: memset(answer, 0, PACKETSZ);
84: if((len = res_query(domain, C_IN, T_TXT, answer, PACKETSZ)) < 0) {
85: #ifdef IIDEBUG
86: if (iihash)
87: syslog(LOG_INFO, "Malloc-txt: %s", UNKN);
88: #endif
89: return (iihash)?strdup(UNKN):UNKN;
90: }
91:
92: pt = answer + sizeof(HEADER);
93:
94: if((exp = dn_expand(answer, answer + len, pt, host, sizeof(host))) < 0) {
95: printf("@dn_expand failed\n"); return NULL;
96: }
97:
98: pt += exp;
99:
100: GETSHORT(type, pt);
101: if(type != T_TXT) {
102: printf("@Broken DNS reply.\n"); return NULL;
103: }
104:
105: pt += INT16SZ; /* class */
106:
107: if((exp = dn_expand(answer, answer + len, pt, host, sizeof(host))) < 0) {
108: printf("@second dn_expand failed\n"); return NULL;
109: }
110:
111: pt += exp;
112: GETSHORT(type, pt);
113: if(type != T_TXT) {
114: printf("@Not a TXT record\n"); return NULL;
115: }
116:
117: pt += INT16SZ; /* class */
118: pt += INT32SZ; /* ttl */
119: GETSHORT(size, pt);
120: txtlen = *pt;
121:
122:
123: if(txtlen >= size || !txtlen) {
124: printf("@Broken TXT record (txtlen = %d, size = %d)\n", txtlen, size); return NULL;
125: }
126:
127: if (txtlen > NAMELEN)
128: txtlen = NAMELEN;
129:
130: if (iihash) {
131: if (!(txt = malloc(txtlen + 1)))
132: return NULL;
133: } else
134: txt = (char*)txtrec;
135:
136: pt++;
137: strncpy(txt, (char*) pt, txtlen);
138: txt[txtlen] = 0;
139:
140: #ifdef IIDEBUG
141: if (iihash)
142: syslog(LOG_INFO, "Malloc-txt(%p): %s", txt, txt);
143: #endif
144:
145: return txt;
146: }
147:
148: char* trimsep(char *s) {
149: int l;
150: char *p = s;
151: while (*p == ' ' || *p == ITEMSEP)
152: *p++ = '\0';
153: for (l = strlen(p)-1; p[l] == ' ' || p[l] == ITEMSEP; l--)
154: p[l] = '\0';
155: return p;
156: }
157:
158: // originX.asn.cymru.com txtrec: ASN | Route | Country | Registry | Allocated
159: char* split_txtrec(char *txtrec) {
160: if (!txtrec)
161: return NULL;
162: if (iihash) {
163: #ifdef IIDEBUG
164: syslog(LOG_INFO, "Malloc-tbl: %s", txtrec);
165: #endif
166: if (!(items = malloc(sizeof(*items)))) {
167: #ifdef IIDEBUG
168: syslog(LOG_INFO, "Free-txt(%p)", txtrec);
169: #endif
170: free(txtrec);
171: return NULL;
172: }
173: }
174:
175: char* prev = (*items)[0] = trimsep(txtrec);
176: char* next;
177: int i = 0, j;
178:
179: while ((next = strchr(prev, ITEMSEP)) && (i < ITEMSMAX)) {
180: *next++ = '\0';
181: (*items)[i++] = trimsep(prev);
182: (*items)[i] = prev = trimsep(next);
183: }
184: if (i < ITEMSMAX)
185: i++;
186: for (j = i; j <= ITEMSMAX; j++)
187: (*items)[j] = NULL;
188:
189: if (i > ipinfo_max)
190: ipinfo_max = i;
191: if (ipinfo_no >= i) {
192: if (ipinfo_no >= ipinfo_max)
193: ipinfo_no = 0;
194: return (*items)[0];
195: } else
196: return (*items)[ipinfo_no];
197: }
198:
199: #ifdef ENABLE_IPV6
200: // from dns.c:addr2ip6arpa()
201: void reverse_host6(struct in6_addr *addr, char *buff) {
202: int i;
203: char *b = buff;
204: for (i=(sizeof(*addr)/2-1); i>=0; i--, b+=4) // 64b portion
205: sprintf(b, "%x.%x.", addr->s6_addr[i] & 0xf, addr->s6_addr[i] >> 4);
206: buff[strlen(buff) - 1] = '\0';
207: }
208: #endif
209:
210: char *get_ipinfo(ip_t *addr) {
211: if (!addr)
212: return NULL;
213:
214: char key[NAMELEN];
215: char lookup_key[NAMELEN];
216:
217: if (af == AF_INET6) {
218: #ifdef ENABLE_IPV6
219: reverse_host6(addr, key);
220: if (snprintf(lookup_key, NAMELEN, "%s.origin6.asn.cymru.com", key) >= NAMELEN)
221: return NULL;
222: #else
223: return NULL;
224: #endif
225: } else {
226: unsigned char buff[4];
227: memcpy(buff, addr, 4);
228: if (snprintf(key, NAMELEN, "%d.%d.%d.%d", buff[3], buff[2], buff[1], buff[0]) >= NAMELEN)
229: return NULL;
230: if (snprintf(lookup_key, NAMELEN, "%s.origin.asn.cymru.com", key) >= NAMELEN)
231: return NULL;
232: }
233:
234: char *val = NULL;
235: ENTRY item;
236:
237: if (iihash) {
238: #ifdef IIDEBUG
239: syslog(LOG_INFO, ">> Search: %s", key);
240: #endif
241: item.key = key;;
242: ENTRY *found_item;
243: if ((found_item = hsearch(item, FIND))) {
244: if (!(val = (*((items_t*)found_item->data))[ipinfo_no]))
245: val = (*((items_t*)found_item->data))[0];
246: #ifdef IIDEBUG
247: syslog(LOG_INFO, "Found (hashed): %s", val);
248: #endif
249: }
250: }
251:
252: if (!val) {
253: #ifdef IIDEBUG
254: syslog(LOG_INFO, "Lookup: %s", key);
255: #endif
256: if ((val = split_txtrec(ipinfo_lookup(lookup_key)))) {
257: #ifdef IIDEBUG
258: syslog(LOG_INFO, "Looked up: %s", key);
259: #endif
260: if (iihash)
261: if ((item.key = strdup(key))) {
262: item.data = items;
263: hsearch(item, ENTER);
264: #ifdef IIDEBUG
265: syslog(LOG_INFO, "Insert into hash: %s", key);
266: #endif
267: }
268: }
269: }
270:
271: return val;
272: }
273:
274: int get_iiwidth(void) {
275: return (ipinfo_no < iiwidth_len) ? iiwidth[ipinfo_no] : iiwidth[ipinfo_no % iiwidth_len];
276: }
277:
278: char *fmt_ipinfo(ip_t *addr) {
279: char *ipinfo = get_ipinfo(addr);
280: char fmt[8];
281: snprintf(fmt, sizeof(fmt), "%s%%-%ds", ipinfo_no?"":"AS", get_iiwidth());
282: snprintf(fmtinfo, sizeof(fmtinfo), fmt, ipinfo?ipinfo:UNKN);
283: return fmtinfo;
284: }
285:
286: int is_printii(void) {
287: return ((ipinfo_no >= 0) && (ipinfo_no != ipinfo_max));
288: }
289:
290: void asn_open(void) {
291: if (ipinfo_no >= 0) {
292: #ifdef IIDEBUG
293: syslog(LOG_INFO, "hcreate(%d)", IIHASH_HI);
294: #endif
295: if (!(iihash = hcreate(IIHASH_HI)))
296: perror("ipinfo hash");
297: }
298: }
299:
300: void asn_close(void) {
301: if (iihash) {
302: #ifdef IIDEBUG
303: syslog(LOG_INFO, "hdestroy()");
304: #endif
305: hdestroy();
306: iihash = 0;
307: }
308: }
309:
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to asn.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / mtr