1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
#!/usr/bin/env python
# Copyright 2009 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Simple DNS server comparison benchmarking tool.
Designed to assist system administrators in selection and prioritization.
"""
__author__ = 'tstromberg@google.com (Thomas Stromberg)'
import datetime
import math
import sys
import base_ui
import conn_quality
import nameserver_list
class NameBenchCli(base_ui.BaseUI):
"""A command-line implementation of the namebench workflow."""
def __init__(self, options, supplied_ns, global_ns, regional_ns, version=None):
self.options = options
self.supplied_ns = supplied_ns
self.global_ns = global_ns
self.include_internal = True
self.regional_ns = regional_ns
self.version = version
self.last_msg = (None, None, None, None)
self.last_msg_count_posted = 0
self.preferred = []
self.secondary = []
super(NameBenchCli, self).__init__()
def UpdateStatus(self, msg, count=None, total=None, error=False, debug=False):
"""Status updates for the command-line. A lot of voodoo here."""
if self.last_msg == (msg, count, total, error):
return None
if debug:
return None
if error:
print
print '* ERROR: %s' % msg
sys.exit(2)
elif not total:
self.last_msg_count_posted = 0
sys.stdout.write('- %s\n' % msg)
elif self.last_msg[0] != msg:
self.last_msg_count_posted = 0
sys.stdout.write('- %s: %s/%s' % (msg, count, total))
last_count = 0
else:
last_count = self.last_msg[1]
if total:
if count and (count - last_count > 0):
# Write a few dots to catch up to where we should be.
catch_up = int(math.ceil((count - last_count) / 2.0))
sys.stdout.write('.' * catch_up)
if count == total:
sys.stdout.write('%s/%s\n' % (count, total))
elif total > 25 and count and (count - self.last_msg_count_posted > (total * 0.20)):
sys.stdout.write(str(count))
self.last_msg_count_posted = count
sys.stdout.flush()
self.last_msg = (msg, count, total, error)
def PrepareNameServers(self):
super(NameBenchCli, self).PrepareNameServers()
print ''
print 'Final list of nameservers considered:'
print '-' * 78
for n in self.nameservers.SortByFastest():
print '%-15.15s %-18.18s %-4.0fms | %s' % (n.ip, n.name, n.check_average,
n.warnings_string)
print ''
def RunAndOpenReports(self):
self.RunBenchmark()
print "\n%s\n" % self.reporter.CreateReport(format='ascii')
self.CreateReports()
if self.options.open_webbrowser:
self.DisplayHtmlReport()
def Execute(self):
"""Called by namebench.py to start the show."""
print('namebench %s - %s (%s) on %s' %
(self.version, self.options.input_source or 'best source',
self.options.select_mode, datetime.datetime.now()))
print ('threads=%s/%s queries=%s runs=%s timeout=%s health_timeout=%s servers=%s' %
(self.options.health_thread_count, self.options.benchmark_thread_count,
self.options.query_count,
self.options.run_count, self.options.timeout,
self.options.health_timeout, self.options.num_servers))
print '-' * 78
if self.options.only:
if not self.supplied_ns:
print 'If you use --only, you must provide nameservers to use.'
sys.exit(1)
self.include_internal = False
try:
self.LoadDataSources()
self.PrepareTestRecords()
print ''
self.PrepareNameServers()
self.PrepareBenchmark()
self.RunAndOpenReports()
except (nameserver_list.OutgoingUdpInterception,
nameserver_list.TooFewNameservers,
conn_quality.OfflineConnection):
(exc_type, exception) = sys.exc_info()[0:2]
self.UpdateStatus("%s - %s" % (exc_type, exception), error=True)
|
