#!/usr/bin/env python

#

# Copyright (C) 2015, 2016 Nippon Telegraph and Telephone Corporation.

#

# 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.

# from __future__ import print_function

import os

import sys

import yaml

import time

import shutil

import copy

import netaddr

import datetime

from argparse import ArgumentParser, REMAINDER

from itertools import chain, islice

from requests.exceptions import ConnectionError

from pyroute2 import IPRoute

from socket import AF_INET

from nsenter import Namespace

from base import *

from exabgp import ExaBGP, ExaBGP_MRTParse

from gobgp import GoBGP, GoBGPTarget

from bird import BIRD, BIRDTarget

from quagga import Quagga, QuaggaTarget

from frr import FRRouting, FRRoutingTarget

from mirage import MIRAGE, MIRAGETarget

from mirage_st import MIRAGE_ST, MIRAGESTTarget

from tester import ExaBGPTester

from mrt_tester import GoBGPMRTTester, ExaBGPMrtTester

from monitor import Monitor

from settings import dckr

from Queue import Queue

from mako.template import Template

from packaging import version

from docker.types import IPAMConfig, IPAMPool

from throughput import Throughput, ThroughputTarget

def gen_mako_macro():

'''

def rm_line():

print '\x1b[1A\x1b[2K\x1b[1D\x1b[1A'

def gc_thresh3():

return 200

def doctor(args):

print '/proc/sys/net/ipv4/neigh/default/gc_thresh3 ... {0}'.format(gc_thresh3())

def prepare(args):

FRRouting.build_image(args.force, checkout='stable/3.0', nocache=args.no_cache)

def clean(args):

dckr.remove_container(ctn_name, force=True)

def update(args):

FRRouting.build_image(True, checkout=args.checkout, nocache=args.no_cache)

def two_peer_test(args):

return

def multitest(args):

print 'total time: {0}, total CPU: {1:>4.2f}, max MEM: {2}'.format(end - start, cpu, mem_human(max_mem))

def bench(args):

backup = copy.deepcopy(args)

two_peer_test(args)

multitest(backup)

def gen_conf(args):

return gen_mako_macro() + yaml.dump(conf, default_flow_style=False)

def config(args):

f.write(conf)

if __name__ == '__main__':

parser = ArgumentParser(description='BGP performance measuring tool')

parser.add_argument('-b', '--bench-name', default='bgperf')

parser.add_argument('-d', '--dir', default='/tmp')

s = parser.add_subparsers()

parser_doctor = s.add_parser('doctor', help='check env')

parser_doctor.set_defaults(func=doctor)

parser_prepare = s.add_parser('prepare', help='prepare env')

parser_prepare.add_argument('-f', '--force', action='store_true', help='build even if the container already exists')

parser_prepare.add_argument('-n', '--no-cache', action='store_true')

parser_prepare.set_defaults(func=prepare)

parser_update = s.add_parser('update', help='rebuild bgp docker images')

parser_update.add_argument('image', choices=['exabgp', 'exabgp_mrtparse', 'gobgp', 'bird', 'quagga', 'frr', 'mirage', 'mirage_st', 'all', 'throughput'])

parser_update.add_argument('-c', '--checkout', default='HEAD')

parser_update.add_argument('-n', '--no-cache', action='store_true')

parser_update.set_defaults(func=update)

parser_clean = s.add_parser('clean', help='clean bgperf docker images')

parser_clean.set_defaults(func=clean)

def add_gen_conf_args(parser):

help='monitor\' router ID; default: same as --monitor-local-address')

parser_two_peer = s.add_parser('2peer', help='two_peer benchmark')

parser_two_peer.add_argument('-t', '--target', choices=['gobgp', 'bird', 'quagga', 'frr', 'mirage', 'mirage_st'],

default='gobgp')

parser_two_peer.add_argument('--docker-network-name',

help='Docker network name; this is the name given by \'docker network ls\'')

parser_two_peer.add_argument('-r', '--repeat', action='store_true', help='use existing tester/monitor container')

parser_two_peer.add_argument('-f', '--file', metavar='CONFIG_FILE')

parser_two_peer.add_argument('-g', '--cooling', default=5, type=int)

parser_two_peer.add_argument('-o', '--output', metavar='STAT_FILE')

parser_two_peer.add_argument('-v', '--verbose', default=False)

add_gen_conf_args(parser_two_peer)

parser_two_peer.set_defaults(func=two_peer_test)

parser_multi_test = s.add_parser('multi_test', help='run multitest benchmarks')

parser_multi_test.add_argument('-t', '--target', choices=['gobgp', 'bird', 'quagga', 'frr', 'mirage', 'mirage_st'],

default='gobgp')

parser_multi_test.add_argument('-i', '--image', help='specify custom docker image')

parser_multi_test.add_argument('--docker-network-name',

help='Docker network name; this is the name given by \'docker network ls\'')

parser_multi_test.add_argument('--bridge-name', help='Linux bridge name of the '

'interface corresponding to the Docker network; '

'use this argument only if bgperf can\'t '

'determine the Linux bridge name starting from '

'the Docker network name in case of tests of '

'remote targets.')

parser_multi_test.add_argument('-r', '--repeat', action='store_true', help='use existing tester/monitor container')

parser_multi_test.add_argument('-f', '--file', metavar='CONFIG_FILE')

parser_multi_test.add_argument('-g', '--cooling', default=5, type=int)

parser_multi_test.add_argument('-o', '--output', metavar='STAT_FILE')

parser_multi_test.add_argument('-v', '--verbose', default=False)

add_gen_conf_args(parser_multi_test)

parser_multi_test.set_defaults(func=multitest)

parser_bench = s.add_parser('bench', help='run benchmarks')

parser_bench.add_argument('-t', '--target', choices=['gobgp', 'bird', 'quagga', 'frr', 'mirage', 'mirage_st'], default='gobgp')

parser_bench.add_argument('-i', '--image', help='specify custom docker image')

parser_bench.add_argument('--docker-network-name', help='Docker network name; this is the name given by \'docker network ls\'')

parser_bench.add_argument('--bridge-name', help='Linux bridge name of the '

'interface corresponding to the Docker network; '

'use this argument only if bgperf can\'t '

'determine the Linux bridge name starting from '

'the Docker network name in case of tests of '

'remote targets.')

parser_bench.add_argument('-r', '--repeat', action='store_true', help='use existing tester/monitor container')

parser_bench.add_argument('-f', '--file', metavar='CONFIG_FILE')

parser_bench.add_argument('-g', '--cooling', default=5, type=int)

parser_bench.add_argument('-o', '--output', metavar='STAT_FILE')

parser_bench.add_argument('-v', '--verbose', default=False)

add_gen_conf_args(parser_bench)

parser_bench.set_defaults(func=bench)

parser_config = s.add_parser('config', help='generate config')

parser_config.add_argument('-o', '--output', default='bgperf.yml', type=str)

add_gen_conf_args(parser_config)

parser_config.set_defaults(func=config)

args = parser.parse_args()

args.func(args)