123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152 |
- /*
- * This file is part of the UCB release of Plan 9. It is subject to the license
- * terms in the LICENSE file found in the top-level directory of this
- * distribution and at http://akaros.cs.berkeley.edu/files/Plan9License. No
- * part of the UCB release of Plan 9, including this file, may be copied,
- * modified, propagated, or distributed except according to the terms contained
- * in the LICENSE file.
- */
- /* Portions of this file are Copyright (C) 2015-2018 Giacomo Tesio <giacomo@tesio.it>
- * See /doc/license/gpl-2.0.txt for details about the licensing.
- */
- #include <u.h>
- #include <lib9.h>
- #include <bio.h>
- #include <chartypes.h>
- #include <ndb.h>
- #include "dns.h"
- #include "ip.h"
- static int domount;
- static char *mtpt, *dns, *srv;
- static int
- setup(int argc, char **argv)
- {
- int fd;
- if(argc == 1){
- domount = 0;
- mtpt = argv[0];
- }
- fd = sys_open(dns, ORDWR);
- if(fd < 0){
- if(domount == 0)
- sysfatal("can't open %s: %r", dns);
- fd = sys_open(srv, ORDWR);
- if(fd < 0)
- sysfatal("can't open %s: %r", srv);
- if(sys_mount(fd, -1, mtpt, MBEFORE, "", '9') < 0)
- sysfatal("can't sys_mount(%s, %s): %r", srv, mtpt);
- fd = sys_open(dns, ORDWR);
- if(fd < 0)
- sysfatal("can't open %s: %r", dns);
- }
- return fd;
- }
- static void
- querydns(int fd, char *line, int n)
- {
- char buf[1024];
- sys_seek(fd, 0, 0);
- if(jehanne_write(fd, line, n) != n) {
- print("!%r\n");
- return;
- }
- sys_seek(fd, 0, 0);
- buf[0] = '\0';
- while((n = jehanne_read(fd, buf, sizeof(buf))) > 0){
- buf[n] = '\0';
- print("%s\n", buf);
- }
- }
- static void
- query(int fd)
- {
- int n, len;
- char *lp, *p, *np;
- 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;
- }
- /* inverse queries may need to be permuted */
- if(n > 4 && strcmp(" ptr", &line[n-4]) == 0 &&
- cistrstr(line, ".arpa") == nil){
- /* TODO: reversing v6 addrs is harder */
- 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"); /* TODO: ip6.arpa for v6 */
- strcpy(line, buf);
- n = strlen(line);
- }
- querydns(fd, line, n);
- }
- Bterm(&in);
- }
- void
- main(int argc, char *argv[])
- {
- mtpt = "/net";
- dns = "/net/dns";
- srv = "/srv/dns";
- domount = 1;
- ARGBEGIN {
- case 'x':
- mtpt = "/net.alt";
- dns = "/net.alt/dns";
- srv = "/srv/dns_net.alt";
- break;
- default:
- fprint(2, "usage: %s [-x] [dns-mount-point]\n", argv0);
- exits("usage");
- } ARGEND;
- query(setup(argc, argv));
- exits(0);
- }
|