1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
#include <u.h>
#include <libc.h>
#include <bio.h>
#include <ctype.h>
#include <ndb.h>
#include "dns.h"
#include "ip.h"
static void
querydns(int fd, char *line, int n)
{
char buf[8192+1];
seek(fd, 0, 0);
if(write(fd, line, n) != n) {
print("!%r\n");
return;
}
seek(fd, 0, 0);
while((n = read(fd, buf, sizeof(buf)-1)) > 0){
buf[n++] = '\n';
write(1, buf, n);
}
}
/*
* convert address into a reverse lookup address
*/
static void
mkptrname(char *ip, char *rip, int rlen)
{
uchar a[IPaddrlen];
char *p, *e;
int i;
if(cistrstr(ip, "in-addr.arpa") || cistrstr(ip, "ip6.arpa") || parseip(a, ip) == -1)
snprint(rip, rlen, "%s", ip);
else if(isv4(a))
snprint(rip, rlen, "%ud.%ud.%ud.%ud.in-addr.arpa",
a[15], a[14], a[13], a[12]);
else{
p = rip;
e = rip + rlen;
for(i = 15; i >= 0; i--){
p = seprint(p, e, "%ux.", a[i]&0xf);
p = seprint(p, e, "%ux.", a[i]>>4);
}
seprint(p, e, "ip6.arpa");
}
}
static void
query(int fd)
{
int n;
char *lp;
char buf[1024], line[1024];
Biobuf in;
Binit(&in, 0, OREAD);
for(print("> "); lp = Brdline(&in, '\n'); print("> ")){
n = Blinelen(&in) -1;
while(isspace(lp[n]))
lp[n--] = 0;
n++;
while(isspace(*lp)){
lp++;
n--;
}
if(!*lp)
continue;
strcpy(line, lp);
/* default to an "ip" request if alpha, "ptr" if numeric */
if(strchr(line, ' ') == nil)
if(strcmp(ipattr(line), "ip") == 0) {
strcat(line, " ptr");
n += 4;
} else {
strcat(line, " ip");
n += 3;
}
if(n > 4 && strcmp(" ptr", &line[n-4]) == 0){
line[n-4] = 0;
mkptrname(line, buf, sizeof buf);
n = snprint(line, sizeof line, "%s ptr", buf);
}
querydns(fd, line, n);
}
Bterm(&in);
}
void
main(int argc, char *argv[])
{
char *dns = "/net/dns";
int fd;
ARGBEGIN {
case 'x':
dns = "/net.alt/dns";
break;
default:
fprint(2, "usage: %s [-x] [/net/dns]\n", argv0);
exits("usage");
} ARGEND;
if(argc > 0)
dns = argv[0];
fd = open(dns, ORDWR);
if(fd < 0)
sysfatal("can't open %s: %r", dns);
query(fd);
exits(0);
}
|