123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120 |
- #include <u.h>
- #include <libc.h>
- #include <bio.h>
- #include <ctype.h>
- #include <ndb.h>
- #include "dns.h"
- #include "ip.h"
- void
- main(int argc, char *argv[])
- {
- int fd, n, len, domount;
- Biobuf in;
- char line[1024], *lp, *p, *np, *mtpt, *srv, *dns;
- char buf[1024];
- dns = "/net/dns";
- mtpt = "/net";
- srv = "/srv/dns";
- domount = 1;
- ARGBEGIN {
- case 'x':
- dns = "/net.alt/dns";
- mtpt = "/net.alt";
- srv = "/srv/dns_net.alt";
- break;
- default:
- fprint(2, "usage: %s -x [dns-mount-point]\n", argv0);
- exits("usage");
- } ARGEND;
- if(argc == 1){
- domount = 0;
- mtpt = argv[0];
- }
- fd = open(dns, ORDWR);
- if(fd < 0){
- if(domount == 0){
- fprint(2, "can't open %s: %r\n", mtpt);
- exits(0);
- }
- fd = open(srv, ORDWR);
- if(fd < 0){
- print("can't open %s: %r\n", srv);
- exits(0);
- }
- if(mount(fd, -1, mtpt, MBEFORE, "") < 0){
- print("can't mount(%s, %s): %r\n", srv, mtpt);
- exits(0);
- }
- fd = open(mtpt, ORDWR);
- if(fd < 0){
- print("can't open %s: %r\n", mtpt);
- exits(0);
- }
- }
- 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, ' ')==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);
- }
- seek(fd, 0, 0);
- if(write(fd, line, n) < 0) {
- print("!%r\n");
- continue;
- }
- seek(fd, 0, 0);
- while((n = read(fd, buf, sizeof(buf))) > 0){
- buf[n] = 0;
- print("%s\n", buf);
- }
- }
- exits(0);
- }
|