test_bgp_config.py

xorp源码hg

set local-ip 127.0.0.1
set as 4

commit
"""

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True
    
def conf_redist_static(builddir, create_static = True):
    """
    Redistribute static into BGP
    """

    if create_static:
        create_static_command = \
"""
create protocols static
edit protocols static

top
"""
    else:
        create_static_command = ''

    # Configure the xorpsh
    xorpsh_commands = \
"""
configure

%s

create policy
edit policy

create policy-statement static
edit policy-statement static

create term 1
edit term 1

create from
edit from
set protocol static

top

edit protocols bgp
set export static

commit
""" % (create_static_command)

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_redist_static_incomplete(builddir):
    """
    Redistribute static into BGP and set the origin to INCOMPLETE
    """

    conf_redist_static(builddir)

    # Configure the xorpsh
    xorpsh_commands = \
"""
configure

edit policy policy-statement static term 1

create then
edit then
set origin 2

commit
"""

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_redist_static_no_export(builddir):
    """
    Redistribute static into BGP and set NO_EXPORT
    """

    conf_redist_static(builddir)

    # Configure the xorpsh
    xorpsh_commands = \
"""
configure

edit policy policy-statement static term 1

create then
edit then
set community NO_EXPORT

commit
"""

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_redist_static_med(builddir):
    """
    Redistribute static into BGP and set the MED to 42
    """

    conf_redist_static(builddir)

    # Configure the xorpsh
    xorpsh_commands = \
"""
configure

edit policy policy-statement static term 1

create then
edit then
set med 42
up
up

create term 2
edit term 2
create to
edit to
set as-path ^$
up

create then
edit then
set med-remove true

commit
"""

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_preference_TR1(builddir):
    """
    Configure TR1 to have a higher preference than TR2
    """

    # Configure the xorpsh
    xorpsh_commands = \
    """
configure

create policy policy-statement preference term 1
edit policy policy-statement preference term 1

create from
edit from
set nexthop4 127.0.0.2..127.0.0.2
up

create then
edit then
set localpref 200
top

edit protocols bgp
set import preference

commit
    """

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_import_med_change(builddir):
    """
    Set the med of all incoming packets to 42
    """

    # Configure the xorpsh
    xorpsh_commands = \
    """
configure

create policy policy-statement preference term 1
edit policy policy-statement preference term 1

create then
edit then
set med 42
top

edit protocols bgp
set import preference

commit
    """

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_export_med_change(builddir):
    """
    Set the med of all outgoing packets to 42
    """

    # Configure the xorpsh
    xorpsh_commands = \
    """
configure

create policy policy-statement preference term 1
edit policy policy-statement preference term 1

create then
edit then
set med 42
top

edit protocols bgp
set export preference

commit
    """

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_import_origin_change(builddir):
    """
    Set the origin of all incoming packets to incomplete
    """

    # Configure the xorpsh
    xorpsh_commands = \
    """
configure

create policy policy-statement preference term 1
edit policy policy-statement preference term 1

create then
edit then
set origin 2
top

edit protocols bgp
set import preference

commit
    """

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_export_origin_change(builddir):
    """
    Set the origin of all outgoing packets to incomplete
    """

    # Configure the xorpsh
    xorpsh_commands = \
    """
configure

create policy policy-statement preference term 1
edit policy policy-statement preference term 1

create then
edit then
set origin 2
top

edit protocols bgp
set export preference

commit
    """

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_damping(builddir):
    """
    Configure damping
    """

    # Configure the xorpsh
    xorpsh_commands = \
"""
configure

edit protocols bgp

create damping
edit damping
set suppress 2000
set reuse 800
set half-life 3
set max-suppress 5

commit
"""

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True


def conf_aggregate_brief(builddir):

    if not conf_aggregate(builddir, "true"):
        return False

    return True


def conf_aggregate_asset(builddir):

    if not conf_aggregate(builddir, "false"):
        return False

    return True


def conf_aggregate(builddir, brief_mode):
    """
    Configure aggregate 
    """

    # Configure the xorpsh
    xorpsh_commands = \
"""
configure

create policy
edit policy

create policy-statement aggregate
edit policy-statement aggregate

create term 1
edit term 1

create from
edit from

set network4 <= 192.0.0.0/8

up
create then
edit then
set aggregate-prefix-len 8
set aggregate-brief-mode %s

top
edit policy
create policy-statement drop-component
edit policy-statement drop-component

create term 1
edit term 1

create to
edit to

set was-aggregated true

up
create then
edit then

set reject

top
edit protocol bgp
set import aggregate
set export drop-component

commit
""" % (brief_mode)

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_create_protocol_static(builddir):
    """
    Create the static protocol
    """

    # Configure the xorpsh
    xorpsh_commands = \
"""
configure

create protocols static

commit
"""

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True


def conf_add_static_route4(builddir, net, next_hop = "127.0.0.1"):
    """
    Add a static route
    """

    # Configure the xorpsh
    xorpsh_commands = \
"""
configure

edit protocols static
create route %s
edit route %s
set next-hop %s

commit
""" % (net, net, next_hop)

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_delete_static_route4(builddir, net, next_hop = "127.0.0.1"):
    """
    Delete a static route
    """

    # Configure the xorpsh
    xorpsh_commands = \
"""
configure

delete protocols static route %s

commit
""" % (net)

    if not xorpsh(builddir, xorpsh_commands):
        return False

    return True

def conf_multiprotocol(builddir):
    """
    Configure multiprotocol
    """

    return True

def show_bgp_routes(builddir):
    """
    Return the output of the show bgp routes command.
    """

    xorpsh_commands = \
"""
show bgp routes
"""

    result, output = xorpsh(builddir, xorpsh_commands)

    if not result:
        return False, output

    return True, output
    
    
# Local Variables:
# mode: python
# py-indent-offset: 4
# End: