mirror of
https://github.com/samba-team/samba.git
synced 2024-12-24 21:34:56 +03:00
7057abcfcd
Scripts to generate representative network traffic and replay this to a samba instance. For load testing, performance profiling and capacity planning. traffic_learner process a file generated by traffic_summary and generate a model that can be used by traffic_replay to generate samba network traffic. traffic_replay Replay a summary file generated by traffic_summary, or use a model created by traffic_learner to generate network traffic. Signed-off-by: Gary Lockyer <gary@catalyst.net.nz> Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz> Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz> Reviewed-by: Garming Sam <garming@catalyst.net.nz> Pair-programmed-with: Garming Sam <garming@catalyst.net.nz> Pair-programmed-with: Douglas Bagnall <douglas.bagnall@catalyst.net.nz> Pair-Programmed-With: Andrew Bartlett <abartlet@samba.org> Pair-Programmed-With: Tim Beale <timbeale@catalyst.net.nz>
64 lines
2.1 KiB
Python
Executable File
64 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# Generate a traffic model from a traffic summary file
|
|
#
|
|
# Copyright (C) Catalyst IT Ltd. 2017
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
import sys
|
|
import argparse
|
|
|
|
sys.path.insert(0, "bin/python")
|
|
from samba.emulate import traffic
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description=__doc__,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter)
|
|
parser.add_argument('-o', '--out', type=argparse.FileType('w'),
|
|
help="write model here")
|
|
parser.add_argument('--dns-mode', choices=['inline', 'count'],
|
|
help='how to deal with DNS', default='count')
|
|
parser.add_argument('SUMMARY_FILE', nargs='*', type=argparse.FileType('r'),
|
|
default=[sys.stdin],
|
|
help="read from this file (default STDIN)")
|
|
args = parser.parse_args()
|
|
|
|
if not args.out:
|
|
print >> sys.stdout, "No output file was specified to write the model to."
|
|
print >> sys.stdout, "Please specify a filename using the --out option."
|
|
return
|
|
|
|
if args.SUMMARY_FILE is sys.stdin:
|
|
print >> sys.stderr, "reading from STDIN..."
|
|
|
|
(conversations,
|
|
interval,
|
|
duration,
|
|
dns_counts) = traffic.ingest_summaries(args.SUMMARY_FILE,
|
|
dns_mode=args.dns_mode)
|
|
|
|
model = traffic.TrafficModel()
|
|
print >> sys.stderr, "learning model"
|
|
if args.dns_mode == 'count':
|
|
model.learn(conversations, dns_counts)
|
|
else:
|
|
model.learn(conversations)
|
|
|
|
model.save(args.out)
|
|
|
|
|
|
main()
|