#dnsperf

#!/usr/bin/python # # This script outputs: # - Infoblox-formatted CSV with the number of A records specified # - dnsperf-formatted TSV to query all of the A records in the CSV file # # Requirements: # - Python (only tested with version 3.4) # - Infoblox temporary/permanent license # - Infoblox model that accommodates the number of A records being #   queried and clients querying them # - Text file in the same directory as the script that contains words #   desired in the A records separated by newlines. I used this one: #   http://www-01.sil.org/linguistics/wordlists/english/ # - dnsperf (optional) - http://nominum.com/measurement-tools/ # # Sample dnsperf command to query records: # "dnsperf -s <dns_IP> -d </path_to_TSV> -c <num_clients> -Q #  <num_queries_per_sec> -v > <output_file>" # # IMPORTANT: explicitly allow DNS queries from end devices in # 'Grid > Grid Manager > Edit > Queries' in the Grid interface # # IMPORTANT: once CSV is imported to Infoblox, make sure to associate a # name server with the lab authoritative zone

import sys import random import os import errno

def main():

   # Check if user entered one argument    if len(sys.argv) != 2:        print("Usage: infoblox_arecord_gen <number of records to generate>")        sys.exit(1)    else:        numRecsToGen = sys.argv[1]

   # Check if user specified an integer for number of records to generate    try:        numRecsToGen = int(numRecsToGen)    except ValueError:        print("Error: number of records to generate must be an integer")        sys.exit(1)

   # Generate a list of random hostnames and a list of random IPs    randHostnameList = genRandHostnameSet(numRecsToGen)    randIPList = genRandIPList(numRecsToGen)

   # Generate Infoblox CSV and dnsperf data file        genInfobloxCSV(randHostnameList,randIPList)    genDnsperfFile(randHostnameList)

def genRandHostnameSet(numHostnamesToGen):

   # Use wordList to generate random hostnames on the .lab domain and store them    # in a set to ensure they are all unique, and then convert the set to a list    # on return

   with open('./words.txt') as f:        wordList = f.read().splitlines()

   randHostnameSet = set()

   while len(randHostnameSet) < numHostnamesToGen:

       # Generate random words until they consist of only alphabetical characters        while True:            tempWord1 = wordList[random.randint(0, len(wordList)-1)]            if tempWord1.isalpha():                break        while True:            tempWord2 = wordList[random.randint(0, len(wordList)-1)]            if tempWord2.isalpha():                break

       randHostnameSet.add(tempWord1 + "-" + tempWord2 + ".lab")

   return list(randHostnameSet)

def genRandIPList(numIPsToGen):

   # Generate random IPs and store them in a set to ensure they are all unique,    # and then convert the set to a list on return

   randIPSet = set()

   while len(randIPSet) < numIPsToGen:        randIPSet.add(str(random.randint(0, 255)) + "." + str(random.randint(0, 255)) + "." + str(random.randint(0, 255)) + "." + str(random.randint(0, 255)))

   return list(randIPSet)

def genInfobloxCSV(randHostnameList,randIPList):

   infobloxCSVRowList = []

   # Insert forward zone for .lab    infobloxCSVRowList.append("header-authzone,fqdn*,zone_format*")    infobloxCSVRowList.append("authzone,lab,FORWARD")

   # Begin inserting A records    infobloxCSVRowList.append("header-arecord,address*,fqdn*")

   counter = 0    while counter < len(randHostnameList):        infobloxCSVRowList.append("arecord," + randIPList[counter] + "," + randHostnameList[counter])        counter += 1

   # Delete the file if it already exists and then write infobloxCSVRowList to a CSV file    filename = "Infoblox_" + str(len(randHostnameList)) + "_Records.csv"

   try:        os.remove(filename)    except OSError:        pass

   file = open(filename, "w+")

   for item in infobloxCSVRowList:        file.write(item + "\n")

   file.close()

def genDnsperfFile(randHostnameList):

   # Delete the file if it already exists and then write randHostnameList to a TSV file    filename = "dnsperf_" + str(len(randHostnameList)) + "_Records.tsv"

   try:        os.remove(filename)    except OSError:        pass

   file = open(filename, "w+")

   for item in randHostnameList:        file.write(item + " A\t\n")