1 files changed, 113 insertions, 0 deletions
diff --git a/contrib/queryperf/utils/gen-data-queryperf.py b/contrib/queryperf/utils/gen-data-queryperf.py
new file mode 100644
index 0000000..52074b7
--- /dev/null
+++ b/contrib/queryperf/utils/gen-data-queryperf.py
@@ -0,0 +1,113 @@
+#!/usr/bin/python
+
+#
+# $Id: gen-data-queryperf.py,v 1.2 2008/06/13 18:17:08 jinmei Exp $
+#
+# Contributed by Stephane Bortzmeyer <bortzmeyer@nic.fr>
+#
+# "A small tool which may be useful with contrib/queryperf. This script
+#  can generate files of queries, both with random names (to test the
+#  behaviour with NXdomain) and with domains from a real zone file."
+#
+
+import sys
+import getopt
+import random
+import re
+
+ldh = []
+# Letters
+for i in range(97, 122):
+    ldh.append(chr(i))
+# Digits
+for i in range(48, 57):
+    ldh.append(chr(i))
+# Hyphen
+ldh.append('-')
+
+maxsize=10
+tld='org'
+num=4
+percent_random = 0.3
+gen = None
+zone_file = None
+domains = {}
+domain_ns = r'^([a-z0-9-\.]+)((\s+\d+)?(\s+IN)?|(\s+IN)(\s+\d+)?)\s+NS'
+domain_ns_re = re.compile(domain_ns, re.IGNORECASE)
+
+def remove_tld(label, tld):
+    if label.endswith('.' + tld + '.'):
+        return label[0:-(1+ len(tld) + 1)]
+    else:
+        return label
+
+def gen_random_label():
+    label = ""
+    for i in range(gen.randint(1, maxsize)):
+        label = label + gen.choice(ldh)
+    return label
+
+def make_domain(label):
+    return "www." + label + "." + tld + "     A"
+
+def usage():
+    sys.stdout.write("Usage: " + sys.argv[0] + " [-n number] " + \
+                     "[-p percent-random] [-t TLD]\n")
+    sys.stdout.write("       [-m MAXSIZE] [-f zone-file]\n")
+    
+try:
+    optlist, args = getopt.getopt(sys.argv[1:], "hp:f:n:t:m:",
+                                  ["help", "percentrandom=", "zonefile=",
+                                   "number=", "tld=",
+                                   "maxsize="])
+    for option, value in optlist:
+        if option == "--help" or option == "-h":
+            usage()
+            sys.exit(0)
+        elif option == "--number" or option == "-n":
+            num = int(value)
+        elif option == "--maxsize" or option == "-m":
+            maxsize = int(value)
+        elif option == "--percentrandom" or option == "-p":
+            percent_random = float(value)
+        elif option == "--tld" or option == "-t":
+            tld = str(value)
+        elif option == "--zonefile" or option == "-f":
+            zone_file = str(value)
+        else:
+            error("Unknown option " + option)
+except getopt.error, reason:
+    sys.stderr.write(sys.argv[0] + ": " + str(reason) + "\n")
+    usage()
+    sys.exit(1)
+    
+if len(args) <> 0:
+    usage()
+    sys.exit(1)
+
+gen = random.Random()
+if zone_file:
+    file = open(zone_file)
+    line = file.readline()
+    while line:
+        domain_line = domain_ns_re.match(line)
+        if domain_line:
+            print domain_line.group(1)
+            domain = remove_tld(domain_line.group(1), tld)
+            domains[domain] = 1
+        line = file.readline()
+    file.close()
+if zone_file:
+    domains = domains.keys()
+    if len(domains) == 0:
+        sys.stderr.write("No domains found in '%s'\n" % zone_file)
+        sys.exit(1)
+for i in range(num):
+    if zone_file:
+        if gen.random() < percent_random:
+            sys.stdout.write(make_domain(gen_random_label()))
+        else:
+            sys.stdout.write(make_domain(gen.choice(domains)))
+    else:
+        sys.stdout.write(make_domain(gen_random_label()))
+    sys.stdout.write("\n")