98 lines
1.7 KiB
C
Executable file
98 lines
1.7 KiB
C
Executable file
#include <u.h>
|
|
#include <libc.h>
|
|
#include <bio.h>
|
|
#include <ndb.h>
|
|
#include <thread.h>
|
|
#include <9pclient.h>
|
|
#include "dns.h"
|
|
#include "ip.h"
|
|
|
|
void
|
|
usage(void)
|
|
{
|
|
fprint(2, "usage: dnsquery [-x service]\n");
|
|
threadexitsall("usage");
|
|
}
|
|
|
|
void
|
|
threadmain(int argc, char *argv[])
|
|
{
|
|
CFid *fd;
|
|
int n, len;
|
|
Biobuf in;
|
|
char line[1024], *lp, *p, *np, *dns;
|
|
char buf[1024];
|
|
|
|
dns = "dns";
|
|
ARGBEGIN{
|
|
case 'x':
|
|
dns = EARGF(usage());
|
|
break;
|
|
default:
|
|
usage();
|
|
}ARGEND;
|
|
|
|
if(argc)
|
|
usage();
|
|
|
|
fd = nsopen(dns, nil, "dns", ORDWR);
|
|
if(fd == nil)
|
|
sysfatal("open %s!dns: %r", dns);
|
|
|
|
Binit(&in, 0, OREAD);
|
|
for(print("> "); lp = Brdline(&in, '\n'); print("> ")){
|
|
n = Blinelen(&in)-1;
|
|
strncpy(line, lp, n);
|
|
line[n] = 0;
|
|
if (n<=1)
|
|
continue;
|
|
/* default to an "ip" request if alpha, "ptr" if numeric */
|
|
if (strchr(line, ' ')==0) {
|
|
if(strcmp(ipattr(line), "ip") == 0) {
|
|
strcat(line, " ptr");
|
|
n += 4;
|
|
} else {
|
|
strcat(line, " ip");
|
|
n += 3;
|
|
}
|
|
}
|
|
|
|
/* inverse queries may need to be permuted */
|
|
if(n > 4 && strcmp("ptr", &line[n-3]) == 0
|
|
&& strstr(line, "IN-ADDR") == 0 && strstr(line, "in-addr") == 0){
|
|
for(p = line; *p; p++)
|
|
if(*p == ' '){
|
|
*p = '.';
|
|
break;
|
|
}
|
|
np = buf;
|
|
len = 0;
|
|
while(p >= line){
|
|
len++;
|
|
p--;
|
|
if(*p == '.'){
|
|
memmove(np, p+1, len);
|
|
np += len;
|
|
len = 0;
|
|
}
|
|
}
|
|
memmove(np, p+1, len);
|
|
np += len;
|
|
strcpy(np, "in-addr.arpa ptr");
|
|
strcpy(line, buf);
|
|
n = strlen(line);
|
|
}
|
|
|
|
fsseek(fd, 0, 0);
|
|
if(fswrite(fd, line, n) < 0) {
|
|
print("!%r\n");
|
|
continue;
|
|
}
|
|
fsseek(fd, 0, 0);
|
|
while((n = fsread(fd, buf, sizeof(buf))) > 0){
|
|
buf[n] = 0;
|
|
print("%s\n", buf);
|
|
}
|
|
}
|
|
threadexitsall(0);
|
|
}
|