| 1 | #!/usr/bin/python |
| 2 | # ldd - DNS implementation in Python |
| 3 | # Copyright (C) 2006 Fredrik Tolf <fredrik@dolda2000.com> |
| 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 as published by |
| 7 | # the Free Software Foundation; either version 2 of the License, or |
| 8 | # (at your option) any later version. |
| 9 | # |
| 10 | # This program is distributed in the hope that it will be useful, |
| 11 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 12 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 13 | # GNU General Public License for more details. |
| 14 | # |
| 15 | # You should have received a copy of the GNU General Public License |
| 16 | # along with this program; if not, write to the Free Software |
| 17 | # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA |
| 18 | |
| 19 | |
| 20 | import sys |
| 21 | from getopt import getopt |
| 22 | import socket |
| 23 | |
| 24 | from ldd import dbzone, rec, dn |
| 25 | |
| 26 | if len(sys.argv) < 3: |
| 27 | print "usage: dnsdbtool [-d] DBDIR DBNAME [COMMAND [ARGS...]]" |
| 28 | print " dnsdbtool -c DBDIR DBNAME" |
| 29 | sys.exit(1) |
| 30 | |
| 31 | debug = False |
| 32 | currrset = None |
| 33 | curname = None |
| 34 | opts, args = getopt(sys.argv[1:], "cd") |
| 35 | for o, a in opts: |
| 36 | if o == "-c": |
| 37 | dbzone.dnsdb.create(args[0], args[1]) |
| 38 | sys.exit(0) |
| 39 | if o == "-d": |
| 40 | debug = True |
| 41 | |
| 42 | db = dbzone.dnsdb(args[0], args[1]) |
| 43 | |
| 44 | args = args[2:] |
| 45 | |
| 46 | class error(Exception): |
| 47 | def __init__(self, text): |
| 48 | self.text = text |
| 49 | |
| 50 | def __str__(self): |
| 51 | return self.text |
| 52 | |
| 53 | def rrusage(name, syn): |
| 54 | u = "" |
| 55 | for e in syn: |
| 56 | u += " " + e[1].upper() |
| 57 | return "usage: " + name + u |
| 58 | |
| 59 | def mkrr(line): |
| 60 | rtype = rec.rtypebyname(line[0]) |
| 61 | if rtype is None: |
| 62 | raise error("no such RR type " + line[0]) |
| 63 | rtype = rec.rtypebyid(rtype) |
| 64 | syn = rtype[2] |
| 65 | if(len(line) != len(syn) + 1): |
| 66 | raise error(rrusage(rtype[1], syn)) |
| 67 | return rec.rrdata(*line) |
| 68 | |
| 69 | def tokenize(line): |
| 70 | tokens = [] |
| 71 | state = 0 |
| 72 | for c in line + " ": |
| 73 | if state == 0: |
| 74 | if not c.isspace(): |
| 75 | ctok = "" |
| 76 | state = 1 |
| 77 | if state == 1: |
| 78 | if c.isspace(): |
| 79 | tokens += [ctok] |
| 80 | state = 0 |
| 81 | elif c == '"': |
| 82 | state = 2 |
| 83 | elif c == "\\": |
| 84 | state = 3 |
| 85 | else: |
| 86 | ctok += c |
| 87 | elif state == 2: |
| 88 | if c == '"': |
| 89 | state = 1 |
| 90 | elif c == "\\": |
| 91 | state = 4 |
| 92 | else: |
| 93 | ctok += c |
| 94 | elif state == 3: |
| 95 | ctok += c |
| 96 | state = 1 |
| 97 | elif state == 4: |
| 98 | ctok += c |
| 99 | state = 2 |
| 100 | return tokens |
| 101 | |
| 102 | def assertargs(cmd, line, num, usage): |
| 103 | if(len(line) < num): |
| 104 | raise error("usage: " + cmd + " " + usage) |
| 105 | |
| 106 | def runcommand(line): |
| 107 | global currrset, curname |
| 108 | cmd = line[0] |
| 109 | line = line[1:] |
| 110 | if cmd == "addnm": |
| 111 | assertargs(cmd, line, 3, "[-f FLAG] NAME TTL RR RRPARAM...") |
| 112 | opts, line = getopt(line, "f:") |
| 113 | opts = dict(opts) |
| 114 | head = rec.rrhead(dn.fromstring(line[0]), line[2]) |
| 115 | ttl = int(line[1]) |
| 116 | data = mkrr(line[2:]) |
| 117 | rr = rec.rr(head, ttl, data) |
| 118 | if "-f" in opts: rr.setflags(opts["-f"].split(",")) |
| 119 | db.addrr(head.name, rr) |
| 120 | elif cmd == "rmname": |
| 121 | assertargs(cmd, line, 1, "NAME") |
| 122 | name = dn.fromstring(line[0]) |
| 123 | db.rmname(name) |
| 124 | elif cmd == "rmrt": |
| 125 | assertargs(cmd, line, 2, "NAME RRTYPE") |
| 126 | name = dn.fromstring(line[0]) |
| 127 | db.rmrtype(name, line[1]) |
| 128 | elif cmd == "lsnames": |
| 129 | for name in db.listnames(): |
| 130 | print str(name) |
| 131 | elif cmd == "lsrr": |
| 132 | assertargs(cmd, line, 1, "NAME") |
| 133 | name = dn.fromstring(line[0]) |
| 134 | rrset = db.lookup(name) |
| 135 | # dbzone.rootify(rrset, dn.fromstring("dolda2000.com.")) |
| 136 | if rrset is None: |
| 137 | raise error("no such name in database") |
| 138 | for rr in rrset: |
| 139 | print str(rr) |
| 140 | elif cmd == "load": |
| 141 | assertargs(cmd, line, 1, "NAME") |
| 142 | name = dn.fromstring(line[0]) |
| 143 | currrset = db.lookup(name) |
| 144 | curname = name |
| 145 | if currrset is None: |
| 146 | currrset = [] |
| 147 | elif cmd == "add": |
| 148 | assertargs(cmd, line, 2, "[-f FLAG] TTL RR RRPARAM...") |
| 149 | opts, line = getopt(line, "f:") |
| 150 | opts = dict(opts) |
| 151 | head = rec.rrhead(curname, line[1]) |
| 152 | ttl = int(line[0]) |
| 153 | data = mkrr(line[1:]) |
| 154 | rr = rec.rr(head, ttl, data) |
| 155 | if "-f" in opts: rr.setflags(opts["-f"].split(",")) |
| 156 | currrset += [rr] |
| 157 | elif cmd == "ls": |
| 158 | if currrset is None: |
| 159 | raise error("no RRset loaded") |
| 160 | for i, rr in enumerate(currrset): |
| 161 | print str(i) + ": " + str(rr) |
| 162 | elif cmd == "rm": |
| 163 | assertargs(cmd, line, 1, "ID") |
| 164 | rrid = int(line[0]) |
| 165 | currrset[rrid:rrid + 1] = [] |
| 166 | elif cmd == "chttl": |
| 167 | assertargs(cmd, line, 2, "ID NEWTTL") |
| 168 | rrid = int(line[0]) |
| 169 | ttl = int(line[1]) |
| 170 | currrset[rrid].head.ttl = ttl |
| 171 | elif cmd == "chdt": |
| 172 | assertargs(cmd, line, 3, "ID NAME DATA") |
| 173 | rrid = int(line[0]) |
| 174 | currrset[rrid].data[line[1]] = line[2] |
| 175 | elif cmd == "sf": |
| 176 | assertargs(cmd, line, 2, "ID FLAGS...") |
| 177 | rrid = int(line[0]) |
| 178 | currrset[rrid].setflags(line[1:]) |
| 179 | elif cmd == "cf": |
| 180 | assertargs(cmd, line, 2, "ID FLAGS...") |
| 181 | rrid = int(line[0]) |
| 182 | currrset[rrid].clrflags(line[1:]) |
| 183 | elif cmd == "store": |
| 184 | if len(line) > 0: |
| 185 | name = line[0] |
| 186 | else: |
| 187 | name = curname |
| 188 | db.set(name, currrset) |
| 189 | elif cmd == "?" or cmd == "help": |
| 190 | print "Available commands:" |
| 191 | print "addnm, rmname, rmrt, lsnames, lsrr, load, add, ls, rm," |
| 192 | print "chttl, chdt, sf, cf, store" |
| 193 | else: |
| 194 | print "no such command: " + cmd |
| 195 | |
| 196 | if len(args) == 0: |
| 197 | while True: |
| 198 | if sys.stdin.isatty(): |
| 199 | sys.stderr.write("> ") |
| 200 | line = sys.stdin.readline() |
| 201 | if line == "": |
| 202 | if sys.stdin.isatty(): print |
| 203 | break |
| 204 | try: |
| 205 | tokens = tokenize(line) |
| 206 | if len(tokens) > 0: runcommand(tokens) |
| 207 | except error: |
| 208 | sys.stderr.write(str(sys.exc_info()[1]) + "\n") |
| 209 | else: |
| 210 | try: |
| 211 | runcommand(args) |
| 212 | except SystemExit: |
| 213 | raise |
| 214 | except: |
| 215 | if debug: |
| 216 | raise |
| 217 | else: |
| 218 | sys.stderr.write(str(sys.exc_info()[1]) + "\n") |