def populate_children(self, deep=False) :
"""populates all of the children of the linecard. Children are the interfaces.
If deep is set to true, it will also try to populate the children of the children.
INPUT: [boolean]
RETURNS: None
"""
interfaces = Interface.get(self._session, self)
for interface in interfaces :
self.add_child(interface)
if deep :
for child in self._children :
child.populate_children(deep=True)
return None
def populate_children(self, deep=False) :
"""Populates all of the children of the linecard. Children are the interfaces.
If deep is set to true, it will also try to populate the children of the children.
:param deep: boolean that when true will cause the entire sub-tree to be populated\
when false, only the immediate children are populated
:returns: None
"""
interfaces = Interface.get(self._session, self)
for interface in interfaces :
self.add_child(interface)
if deep :
for child in self._children :
child.populate_children(deep=True)
return None
def populate_children(self, deep=False):
"""Populates all of the children of the linecard. Children are the interfaces.
If deep is set to true, it will also try to populate the children of the children.
:param deep: boolean that when true will cause the entire sub-tree to be populated\
when false, only the immediate children are populated
:returns: None
"""
interfaces = Interface.get(self._session, self)
for interface in interfaces:
self.add_child(interface)
if deep:
for child in self._children:
child.populate_children(deep=True)
return None
def main():
"""
Main execution routine
:return: None
"""
# Create a tenant
tenant = Tenant('Solar_Meter')
# Create a Context and Allow_All
context = Context('Solar_Meter', tenant)
context.set_allow_all()
#Create App Profiles
app_sme = AppProfile('Solar_Meter_East', tenant)
app_smw = AppProfile('Solar_Meter_West', tenant)
app_corp = AppProfile('Corporate_Network', tenant)
# Create BridgeDomains
bd_east = BridgeDomain('Bridge_Domain_East', tenant)
bd_east.add_context(context)
bd_east.set_arp_flood('no')
bd_east.set_unicast_route('yes')
subnet_east = Subnet('10.20.12.3', bd_east)
subnet_east.set_addr('10.20.12.3/24')
subnet_east.set_scope("public")
bd_west = BridgeDomain('Bridge_Domain_West', tenant)
bd_west.add_context(context)
bd_west.set_arp_flood('no')
bd_west.set_unicast_route('yes')
subnet_west = Subnet('10.20.13.3', bd_west)
subnet_west.set_addr('10.20.13.3/24')
subnet_west.set_scope("public")
bd_corp = BridgeDomain('Corporate_Bridge_Domain', tenant)
bd_corp.add_context(context)
bd_corp.set_arp_flood('no')
bd_corp.set_unicast_route('yes')
subnet_corp = Subnet('10.20.14.3', bd_corp)
subnet_corp.set_addr('10.20.14.3/24')
subnet_corp.set_scope("public")
# Create an EPGs and assign to App Profiles
epg_db_east = EPG('Database_EPG_East', app_sme)
epg_db_east.add_bd(bd_east)
epg_mw_east = EPG('Middleware_EPG_East', app_sme)
epg_mw_east.add_bd(bd_east)
epg_web_east = EPG('Webserver_EPG_East', app_sme)
epg_web_east.add_bd(bd_east)
epg_db_west = EPG('Database_EPG_West', app_smw)
epg_db_west.add_bd(bd_west)
epg_mw_west = EPG('Middleware_EPG_West', app_smw)
epg_mw_west.add_bd(bd_west)
epg_web_west = EPG('Webserver_EPG_West', app_smw)
epg_web_west.add_bd(bd_west)
epg_finance_corp = EPG('Corporate_Finance_EPG', app_corp)
epg_finance_corp.add_bd(bd_corp)
epg_hr_corp = EPG('Corporate_HR_EPG', app_corp)
epg_hr_corp.add_bd(bd_corp)
epg_sales_corp = EPG('Corporate_Sales_EPG', app_corp)
epg_sales_corp.add_bd(bd_corp)
epg_engineering_corp = EPG('Corporate_Engineering_EPG', app_corp)
epg_engineering_corp.add_bd(bd_corp)
epg_marketing_corp = EPG('Corporate_Marketing_EPG', app_corp)
epg_marketing_corp.add_bd(bd_corp)
# Attach the EPGs to interfaces using VLAN as the encap
# First - Create the physical interface objects
if_101_61 = Interface('eth', '1', '101', '1', '61')
if_101_62 = Interface('eth', '1', '101', '1', '62')
if_101_63 = Interface('eth', '1', '101', '1', '63')
if_101_64 = Interface('eth', '1', '101', '1', '64')
if_101_65 = Interface('eth', '1', '101', '1', '65')
if_101_66 = Interface('eth', '1', '101', '1', '66')
if_101_67 = Interface('eth', '1', '101', '1', '67')
if_101_68 = Interface('eth', '1', '101', '1', '68')
if_102_61 = Interface('eth', '1', '102', '1', '61')
if_102_62 = Interface('eth', '1', '102', '1', '62')
if_102_63 = Interface('eth', '1', '102', '1', '63')
if_102_64 = Interface('eth', '1', '102', '1', '64')
if_102_65 = Interface('eth', '1', '102', '1', '65')
if_102_66 = Interface('eth', '1', '102', '1', '66')
if_102_67 = Interface('eth', '1', '102', '1', '67')
if_102_68 = Interface('eth', '1', '102', '1', '68')
if_101_71 = Interface('eth', '1', '101', '1', '71')
if_101_72 = Interface('eth', '1', '101', '1', '72')
if_101_73 = Interface('eth', '1', '101', '1', '73')
if_101_74 = Interface('eth', '1', '101', '1', '74')
if_101_75 = Interface('eth', '1', '101', '1', '75')
if_102_71 = Interface('eth', '1', '102', '1', '71')
if_102_72 = Interface('eth', '1', '102', '1', '72')
if_102_73 = Interface('eth', '1', '102', '1', '73')
if_102_74 = Interface('eth', '1', '102', '1', '74')
if_102_75 = Interface('eth', '1', '102', '1', '75')
# Second - Create a VLAN interfaces
vlan161_db = L2Interface('vlan161_db', 'vlan', '161')
vlan261_db = L2Interface('vlan261_db', 'vlan', '261')
vlan162_web = L2Interface('vlan162_web', 'vlan', '162')
vlan262_web = L2Interface('vlan262_web', 'vlan', '262')
vlan163_mw = L2Interface('vlan163_web', 'vlan', '163')
vlan263_mw = L2Interface('vlan263_web', 'vlan', '263')
vlan64_corp = L2Interface('vlan64_corp', 'vlan', '64')
vlan65_corp = L2Interface('vlan65_corp', 'vlan', '65')
vlan66_corp = L2Interface('vlan66_corp', 'vlan', '66')
vlan67_corp = L2Interface('vlan67_corp', 'vlan', '67')
vlan68_corp = L2Interface('vlan68_corp', 'vlan', '68')
# Third - Attach the VLANs to the physical interfaces
vlan161_db.attach(if_101_61)
vlan161_db.attach(if_101_62)
vlan161_db.attach(if_101_63)
vlan261_db.attach(if_102_61)
vlan261_db.attach(if_102_62)
vlan261_db.attach(if_102_63)
vlan162_web.attach(if_101_64)
vlan162_web.attach(if_101_65)
vlan162_web.attach(if_101_66)
vlan262_web.attach(if_102_64)
vlan262_web.attach(if_102_65)
vlan262_web.attach(if_102_66)
vlan163_mw.attach(if_101_67)
vlan163_mw.attach(if_101_68)
vlan263_mw.attach(if_102_67)
vlan263_mw.attach(if_102_68)
vlan64_corp.attach(if_101_71)
vlan64_corp.attach(if_102_71)
vlan65_corp.attach(if_101_72)
vlan65_corp.attach(if_102_72)
vlan66_corp.attach(if_101_73)
vlan66_corp.attach(if_102_73)
vlan67_corp.attach(if_101_74)
vlan67_corp.attach(if_102_74)
vlan68_corp.attach(if_101_75)
vlan68_corp.attach(if_102_75)
# Forth - Attach the EPGs to the VLAN interfaces
epg_db_east.attach(vlan161_db)
epg_db_west.attach(vlan261_db)
epg_web_east.attach(vlan162_web)
epg_web_west.attach(vlan262_web)
epg_mw_east.attach(vlan163_mw)
epg_mw_west.attach(vlan263_mw)
epg_finance_corp.attach(vlan64_corp)
epg_hr_corp.attach(vlan65_corp)
epg_sales_corp.attach(vlan66_corp)
epg_engineering_corp.attach(vlan67_corp)
epg_marketing_corp.attach(vlan68_corp)
# Create the Endpoints
# 3x DB East & 3x DB West
db_east_ep1 = Endpoint(name='East_DB_Server_1', parent=epg_db_east)
db_east_ep1.mac = '00:11:11:11:11:11'
db_east_ep1.ip = '10.20.12.4'
db_east_ep2 = Endpoint(name='East_DB_Server_2', parent=epg_db_east)
db_east_ep2.mac = '00:11:11:11:12:11'
db_east_ep2.ip = '10.20.12.5'
db_east_ep3 = Endpoint(name='East_DB_Server_3', parent=epg_db_east)
db_east_ep3.mac = '00:11:11:11:13:11'
db_east_ep3.ip = '10.20.12.6'
db_west_ep1 = Endpoint(name='West_DB_Server_1', parent=epg_db_west)
db_west_ep1.mac = '00:11:11:12:11:11'
db_west_ep1.ip = '10.20.13.4'
db_west_ep2 = Endpoint(name='West_DB_Server_2', parent=epg_db_west)
db_west_ep2.mac = '00:11:11:12:12:11'
db_west_ep2.ip = '10.20.13.5'
db_west_ep3 = Endpoint(name='West_DB_Server_3', parent=epg_db_west)
db_west_ep3.mac = '00:11:11:12:13:11'
db_west_ep3.ip = '10.20.13.6'
# Assign it to the L2Interface
db_east_ep1.attach(vlan161_db)
db_east_ep2.attach(vlan161_db)
db_east_ep3.attach(vlan161_db)
db_west_ep1.attach(vlan261_db)
db_west_ep2.attach(vlan261_db)
db_west_ep3.attach(vlan261_db)
# Create the Contract between Database and Middleware
contract_db2mw = Contract('contract_db2mw', tenant)
icmp_entry = FilterEntry('icmpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='unspecified',
dToPort='unspecified',
etherT='ip',
prot='icmp',
sFromPort='unspecified',
sToPort='unspecified',
tcpRules='unspecified',
parent=contract_db2mw)
arp_entry = FilterEntry('arpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='unspecified',
dToPort='unspecified',
etherT='arp',
prot='unspecified',
sFromPort='unspecified',
sToPort='unspecified',
tcpRules='unspecified',
parent=contract_db2mw)
tcp_entry = FilterEntry('tcpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='1433',
dToPort='1433',
etherT='ip',
prot='tcp',
sFromPort='1433',
sToPort='1433',
tcpRules='unspecified',
parent=contract_db2mw)
udp_entry = FilterEntry('udpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='1433',
dToPort='1433',
etherT='ip',
prot='udp',
sFromPort='1433',
sToPort='1433',
tcpRules='unspecified',
parent=contract_db2mw)
# Provide and consume the Contract
epg_db_east.provide(contract_db2mw)
epg_db_west.provide(contract_db2mw)
epg_mw_east.consume(contract_db2mw)
epg_mw_west.consume(contract_db2mw)
# Create the Contract between Web and Middleware
contract_web2mw = Contract('contract_web2mw', tenant)
icmp_entry = FilterEntry('icmpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='unspecified',
dToPort='unspecified',
etherT='ip',
prot='icmp',
sFromPort='unspecified',
sToPort='unspecified',
tcpRules='unspecified',
parent=contract_web2mw)
arp_entry = FilterEntry('arpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='unspecified',
dToPort='unspecified',
etherT='arp',
prot='unspecified',
sFromPort='unspecified',
sToPort='unspecified',
tcpRules='unspecified',
parent=contract_web2mw)
tcp_entry = FilterEntry('tcpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='443',
dToPort='443',
etherT='ip',
prot='tcp',
sFromPort='443',
sToPort='443',
tcpRules='unspecified',
parent=contract_web2mw)
udp_entry = FilterEntry('udpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='443',
dToPort='443',
etherT='ip',
prot='udp',
sFromPort='443',
sToPort='443',
tcpRules='unspecified',
parent=contract_web2mw)
# Provide and consume the Contract
epg_web_east.provide(contract_web2mw)
epg_web_west.provide(contract_web2mw)
epg_mw_east.consume(contract_web2mw)
epg_mw_west.consume(contract_web2mw)
# Create the Contract for Corporate users
contract_corp = Contract('contract_corp', tenant)
icmp_entry = FilterEntry('icmpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='unspecified',
dToPort='unspecified',
etherT='ip',
prot='icmp',
sFromPort='unspecified',
sToPort='unspecified',
tcpRules='unspecified',
parent=contract_corp)
arp_entry = FilterEntry('arpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='unspecified',
dToPort='unspecified',
etherT='arp',
prot='unspecified',
sFromPort='unspecified',
sToPort='unspecified',
tcpRules='unspecified',
parent=contract_corp)
tcp_entry = FilterEntry('tcpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='443',
dToPort='443',
etherT='ip',
prot='tcp',
sFromPort='443',
sToPort='443',
tcpRules='unspecified',
parent=contract_corp)
udp_entry = FilterEntry('udpentry',
applyToFrag='no',
arpOpc='unspecified',
dFromPort='443',
dToPort='443',
etherT='ip',
prot='udp',
sFromPort='443',
sToPort='443',
tcpRules='unspecified',
parent=contract_corp)
# Provide and consume the Contract
epg_engineering_corp.provide(contract_corp)
epg_finance_corp.provide(contract_corp)
epg_hr_corp.provide(contract_corp)
epg_sales_corp.provide(contract_corp)
epg_marketing_corp.provide(contract_corp)
epg_web_east.consume(contract_corp)
epg_web_west.consume(contract_corp)
epg_mw_east.consume(contract_corp)
epg_mw_west.consume(contract_corp)
epg_db_east.consume(contract_corp)
epg_db_west.consume(contract_corp)
# Dump the necessary configuration
print('URL: ' + str(tenant.get_url()))
print('JSON: ' + str(tenant.get_json()))
send_to_apic(tenant)
def create_random_tenant(self, interfaces=[]):
max_string_length = int(
self._config.get('GlobalDefaults', 'MaximumStringLength'))
# Create the Tenant object
tenant_prefix = 'acitoolkitrandomized-'
tenant_name_len = random_number(1,
max_string_length - len(tenant_prefix))
tenant_name = tenant_prefix + random_string(tenant_name_len)
tenant = Tenant(tenant_name)
# Create some number of BridgeDomains
bridge_domains = []
maximum_bds = int(self._config.get('BridgeDomains', 'Maximum'))
if maximum_bds > self._global_limits.max_bds:
maximum_bds = self._global_limits.max_bds
for i in range(
0,
random_number(
0,
random_number(
int(self._config.get('BridgeDomains', 'Minimum')),
maximum_bds))):
self._global_limits.max_bds -= 1
bd = BridgeDomain(
random_string(random_number(1, max_string_length)), tenant)
# Randomly choose settings for the BridgeDomain
if self._config.get('BridgeDomains',
'AllowFloodUnkMacUcast').lower() == 'true':
bd.set_unknown_mac_unicast(random.choice(['proxy', 'flood']))
if self._config.get(
'BridgeDomains',
'AllowOptimizedFloodUnknownMcast').lower() == 'true':
bd.set_unknown_multicast(random.choice(['flood', 'opt-flood']))
if self._config.get('BridgeDomains',
'AllowArpFlood').lower() == 'true':
bd.set_arp_flood(random.choice(['yes', 'no']))
if self._config.get('BridgeDomains',
'AllowDisableUnicastRoute').lower() == 'true':
bd.set_unicast_route(random.choice(['yes', 'no']))
if self._config.get(
'BridgeDomains',
'AllowNonDefaultMultiDstPkt').lower() == 'true':
bd.set_multidestination(
random.choice(['drop', 'bd-flood', 'encap-flood']))
bridge_domains.append(bd)
# Create some number of Contexts
contexts = []
max_contexts = int(self._config.get('Contexts', 'Maximum'))
if max_contexts > self._global_limits.max_contexts:
max_contexts = self._global_limits.max_contexts
if max_contexts > int(self._config.get('Contexts',
'MaximumPerTenant')):
max_contexts = int(self._config.get('Contexts',
'MaximumPerTenant'))
for i in range(
0,
random_number(
0,
random_number(int(self._config.get('Contexts', 'Minimum')),
max_contexts))):
context = Context(
random_string(random_number(1, max_string_length)), tenant)
self._global_limits.max_contexts -= 1
if self._config.get('Contexts',
'AllowUnenforced').lower() == 'true':
context.set_allow_all(random.choice([True, False]))
contexts.append(context)
# Randomly associate BridgeDomains with the Contexts (or use default)
for bd in bridge_domains:
if random.choice([True, True, False]) and len(contexts):
bd.add_context(random.choice(contexts))
# Create some number of Application Profiles
apps = []
for i in range(
0,
random_number(
0,
random_number(
int(self._config.get('ApplicationProfiles',
'Minimum')),
int(self._config.get('ApplicationProfiles',
'Maximum'))))):
app = AppProfile(
random_string(random_number(1, max_string_length)), tenant)
apps.append(app)
# Create some number of EPGs and place in AppProfiles
epgs = []
max_epgs = int(self._config.get('EPGs', 'Maximum'))
if max_epgs > self._global_limits.max_epgs:
max_epgs = self._global_limits.max_epgs
if len(apps):
for i in range(
0,
random_number(
0,
random_number(int(self._config.get('EPGs', 'Minimum')),
max_epgs))):
epg = EPG(random_string(random_number(1, max_string_length)),
random.choice(apps))
self._global_limits.max_epgs -= 1
epgs.append(epg)
# Randomly associate the EPGs to BridgeDomains
bd_epg_count = [0] * len(bridge_domains)
for epg in epgs:
if random_number(0, 9) == 1 or len(
bridge_domains) == 0: # 1 in 10 chance for no bridgedomain
continue
keep_trying = 100
while keep_trying:
bd_choice = random_number(0, len(bridge_domains) - 1)
if bd_epg_count[bd_choice] <= int(
self._config.get('BridgeDomains', 'MaximumEPGs')):
epg.add_bd(bridge_domains[bd_choice])
bd_epg_count[bd_choice] += 1
break
else:
keep_trying -= 1
# Randomly assign the EPGs to the interfaces provided
interface_objs = {}
for interface in interfaces:
# Create the Interface objects
interface_objs[interface] = Interface.create_from_name(interface)
for epg in epgs:
# Pick an interface
interface_choice = random.choice(interfaces)
# Pick a VLAN
vlan_choice = 0
keep_trying = 100
while vlan_choice in self._interfaces[interface_choice]:
vlan_choice = random_number(
int(self._config.get('VLANs', 'Minimum')),
int(self._config.get('VLANs', 'Maximum')))
keep_trying -= 1
if not keep_trying:
continue
# Create the VLAN interface
vlan_intf = L2Interface(
'vlan%s-on-%s' %
(str(vlan_choice),
interface_objs[interface_choice].name.replace(' ', '')),
'vlan', str(vlan_choice))
# Attach the VLAN interface to the Interface object
vlan_intf.attach(interface_objs[interface_choice])
# Attach the EPG to the VLAN interface
epg.attach(vlan_intf)
# Create some filters
filters = []
max_filters = int(self._config.get('Filters', 'Maximum'))
if max_filters > self._global_limits.max_filters:
max_filters = self._global_limits.max_filters
for i in range(
0,
random_number(
0,
random_number(int(self._config.get('Filters', 'Minimum')),
max_filters))):
filter = Filter(random_string(random_number(1, max_string_length)),
tenant)
self._global_limits.max_filters -= 1
filters.append(filter)
# Create some filter entries
filter_entries = []
max_filter_entries = int(self._config.get('FilterEntries', 'Maximum'))
if max_filter_entries > self._global_limits.max_filter_entries:
max_filter_entries = self._global_limits.max_filter_entries
ip_protocols = {
'icmp': '1',
'igmp': '2',
'tcp': '6',
'egp': '8',
'igp': '9',
'udp': '17',
'icmpv6': '58',
'eigrp': '88',
'ospfigp': '89',
'pim': '103',
'l2tp': '115'
}
if len(filters):
for i in range(
0,
random_number(
0,
random_number(
int(self._config.get('FilterEntries', 'Minimum')),
max_filter_entries))):
applyToFrag = '0'
arpOpc = '0'
dFromPort = '0'
dToPort = '0'
prot = '0'
sFromPort = '0'
sToPort = '0'
tcpRules = '0'
stateful = '0'
icmpv4T = 'not-given'
icmpv6T = 'not-given'
if random_chance(20): # 20% chance of ARP
arpOpc = random.choice(['req', 'reply'])
etherT = 'arp'
elif random_chance(
25): # 20% of remaining 80% is non-IP (16% of total)
ethertype_choices = [
'trill', 'mpls_ucast', 'mac_security', 'fcoe'
]
# if not filter.has_wildcard_entry():
# ethertype_choices += ['0']
etherT = random.choice(ethertype_choices)
else: # remaining is IP
applyToFrag = random.choice(['0', '1'])
etherT = 'ip'
if random_chance(
20
): # Choose more obscure protocols 20% of the time
prot = ip_protocols[random.choice([
'igmp', 'egp', 'igp', 'eigrp', 'ospfigp', 'pim',
'l2tp'
])]
else:
prot = ip_protocols[random.choice(
['icmp', 'tcp', 'udp', 'icmpv6'])]
if prot == ip_protocols['icmp']:
icmpv4T = random.choice([
'echo-rep', 'dst-unreach', 'src-quench',
'echo', 'time-exceeded', 'unspecified',
'not-given'
])
elif prot == ip_protocols['icmpv6']:
icmpv6T = random.choice([
'unspecified', 'dst-unreach', 'time-exceeded',
'echo-req', 'echo-rep', 'nbr-solicit',
'nbr-advert', 'redirect', 'not-given'
])
else:
# Remainder is TCP or UDP
dFromPort, dToPort = random_range(0, 65535)
sFromPort, sToPort = random_range(0, 65535)
if dFromPort != '0' or dToPort != '0' or sFromPort != '0' or sToPort != '0':
applyToFrag = '0'
if prot == ip_protocols['tcp']:
# Randomly choose whether to specify tcpRules
if random_chance(30):
# TODO: should actually take odds from the config file
# Choose a random number of the possible tcpRules but
# if est is chosen, then it must be the only tcpRule. Otherwise, APIC rejects it
tcp_rule_choices = []
tcp_rule_possibilities = [
'est', 'syn', 'ack', 'fin', 'rst'
]
tcp_choice = random.choice(
tcp_rule_possibilities)
tcp_rule_choices.append(tcp_choice)
while tcp_choice != 'est':
tcp_choice = random.choice(
tcp_rule_possibilities)
if tcp_choice != 'est' and tcp_choice not in tcp_rule_choices:
tcp_rule_choices.append(tcp_choice)
tcpRules = ''
for tcp_choice in tcp_rule_choices:
tcpRules += str(tcp_choice) + ','
parent = random.choice(filters)
if not parent.has_entry(applyToFrag, arpOpc, dFromPort,
dToPort, etherT, prot, sFromPort,
sToPort, tcpRules, stateful, icmpv4T,
icmpv6T):
filter_entry = FilterEntry(name=random_string(
random_number(1, max_string_length)),
parent=random.choice(filters),
applyToFrag=applyToFrag,
arpOpc=arpOpc,
dFromPort=dFromPort,
dToPort=dToPort,
etherT=etherT,
prot=prot,
sFromPort=sFromPort,
sToPort=sToPort,
tcpRules=tcpRules,
stateful=stateful,
icmpv4T=icmpv4T,
icmpv6T=icmpv6T)
self._global_limits.max_filter_entries -= 1
# Create some Contracts
contracts = []
max_contracts = int(self._config.get('Contracts', 'Maximum'))
if max_contracts > self._global_limits.max_contracts:
max_contracts = self._global_limits.max_contracts
for i in range(
0,
random_number(
0,
random_number(
int(self._config.get('Contracts', 'Minimum')),
max_contracts))):
contract = Contract(
random_string(random_number(1, max_string_length)), tenant)
self._global_limits.max_contracts -= 1
contracts.append(contract)
# Create some ContractSubjects
if len(contracts):
contract_subjects = []
for i in range(
0,
random_number(
0,
random_number(
int(self._config.get('ContractSubjects',
'Minimum')),
int(self._config.get('ContractSubjects',
'Maximum'))))):
contract_subject = ContractSubject(
random_string(random_number(1, max_string_length)),
random.choice(contracts))
contract_subjects.append(contract_subject)
# Randomly assign Filters to the ContractSubjects
for filter in filters:
if len(contracts) and len(contract_subjects):
already_picked = []
# Pick an arbitrary number of Subjects
for i in range(0, random_number(1, len(contract_subjects))):
pick = random_number(0, len(contract_subjects) - 1)
# Only choose each subject at most once
if pick not in already_picked:
contract_subjects[pick].add_filter(filter)
already_picked.append(pick)
# Randomly provide and consume the Contracts from the EPGs
for action, max_num_epgs in [
('provide',
int(self._config.get('Contracts', 'MaximumProvidingEPGs'))),
('consume',
int(self._config.get('Contracts', 'MaximumConsumingEPGs')))
]:
contract_count = [0] * len(contracts)
for epg in epgs:
already_picked = []
for i in range(0, random_number(0, len(contracts))):
keep_trying = 20 # Try 20 times to randomly pick a contract
while keep_trying:
pick = random_number(0, len(contracts) - 1)
if pick not in already_picked and contract_count[
pick] < max_num_epgs:
getattr(epg, action)(contracts[pick])
already_picked.append(pick)
contract_count[pick] += 1
keep_trying = 0
else:
keep_trying -= 1
return tenant
def create_random_tenant(self, interfaces=[]):
max_string_length = int(
self._config.get('GlobalDefaults', 'MaximumStringLength'))
# Create the Tenant object
tenant_prefix = self._config.get('GlobalDefaults', 'TenantPrefix')
tenant_name_len = random_number(1,
max_string_length - len(tenant_prefix))
tenant_name = tenant_prefix + random_string(tenant_name_len)
tenant = Tenant(tenant_name)
# Create some number of BridgeDomains
bridge_domains = []
maximum_bds = int(self._config.get('BridgeDomains', 'Maximum'))
if maximum_bds > self._global_limits.max_bds:
maximum_bds = self._global_limits.max_bds
for i in range(
0,
random_number(
0,
random_number(
int(self._config.get('BridgeDomains', 'Minimum')),
maximum_bds))):
self._global_limits.max_bds -= 1
bd = BridgeDomain(
random_string(random_number(1, max_string_length)), tenant)
# Randomly choose settings for the BridgeDomain
if self._config.get('BridgeDomains',
'AllowFloodUnkMacUcast').lower() == 'true':
bd.set_unknown_mac_unicast(
random.choice(
ast.literal_eval(
self._config.get('BridgeDomainSettings',
'UnknownMacUnicast'))))
if self._config.get(
'BridgeDomains',
'AllowOptimizedFloodUnknownMcast').lower() == 'true':
bd.set_unknown_multicast(
random.choice(
ast.literal_eval(
self._config.get('BridgeDomainSettings',
'UnknownMulticast'))))
if self._config.get('BridgeDomains',
'AllowArpFlood').lower() == 'true':
bd.set_arp_flood(
random.choice(
ast.literal_eval(
self._config.get('BridgeDomainSettings',
'ArpFlood'))))
if self._config.get('BridgeDomains',
'AllowDisableUnicastRoute').lower() == 'true':
bd.set_unicast_route(
random.choice(
ast.literal_eval(
self._config.get('BridgeDomainSettings',
'UnicastRoute'))))
if self._config.get(
'BridgeDomains',
'AllowNonDefaultMultiDstPkt').lower() == 'true':
bd.set_multidestination(
random.choice(
ast.literal_eval(
self._config.get('BridgeDomainSettings',
'Multidestination'))))
bridge_domains.append(bd)
# Create some number of Contexts
contexts = []
max_contexts = int(self._config.get('Contexts', 'Maximum'))
if max_contexts > self._global_limits.max_contexts:
max_contexts = self._global_limits.max_contexts
if max_contexts > int(self._config.get('Contexts',
'MaximumPerTenant')):
max_contexts = int(self._config.get('Contexts',
'MaximumPerTenant'))
for i in range(
0,
random_number(
0,
random_number(int(self._config.get('Contexts', 'Minimum')),
max_contexts))):
context = Context(
random_string(random_number(1, max_string_length)), tenant)
self._global_limits.max_contexts -= 1
if self._config.get('Contexts',
'AllowUnenforced').lower() == 'true':
context.set_allow_all(random.choice([True, False]))
contexts.append(context)
# Randomly associate BridgeDomains with the Contexts (or use default)
for bd in bridge_domains:
if random.choice([True, True, False]) and len(contexts):
bd.add_context(random.choice(contexts))
# Create some number of Application Profiles
apps = []
for i in range(
0,
random_number(
0,
random_number(
int(self._config.get('ApplicationProfiles',
'Minimum')),
int(self._config.get('ApplicationProfiles',
'Maximum'))))):
app = AppProfile(
random_string(random_number(1, max_string_length)), tenant)
apps.append(app)
# Create some number of EPGs and place in AppProfiles
epgs = []
max_epgs = int(self._config.get('EPGs', 'Maximum'))
if max_epgs > self._global_limits.max_epgs:
max_epgs = self._global_limits.max_epgs
if len(apps):
for i in range(
0,
random_number(
0,
random_number(int(self._config.get('EPGs', 'Minimum')),
max_epgs))):
epg = EPG(random_string(random_number(1, max_string_length)),
random.choice(apps))
self._global_limits.max_epgs -= 1
epgs.append(epg)
# Randomly associate the EPGs to BridgeDomains
bd_epg_count = [0] * len(bridge_domains)
for epg in epgs:
if random_number(0, 9) == 1 or len(
bridge_domains) == 0: # 1 in 10 chance for no bridgedomain
continue
keep_trying = 100
while keep_trying:
bd_choice = random_number(0, len(bridge_domains) - 1)
if bd_epg_count[bd_choice] <= int(
self._config.get('BridgeDomains', 'MaximumEPGs')):
epg.add_bd(bridge_domains[bd_choice])
bd_epg_count[bd_choice] += 1
break
else:
keep_trying -= 1
# Randomly assign the EPGs to the interfaces provided
interface_objs = {}
for interface in interfaces:
# Create the Interface objects
interface_objs[interface] = Interface.create_from_name(interface)
for epg in epgs:
# Pick an interface
interface_choice = random.choice(interfaces)
# Pick a VLAN
vlan_choice = 0
keep_trying = 100
while vlan_choice in self._interfaces[interface_choice]:
vlan_choice = random_number(
int(self._config.get('VLANs', 'Minimum')),
int(self._config.get('VLANs', 'Maximum')))
keep_trying -= 1
if not keep_trying:
continue
# Create the VLAN interface
vlan_intf = L2Interface(
'vlan%s-on-%s' %
(str(vlan_choice),
interface_objs[interface_choice].name.replace(' ', '')),
'vlan', str(vlan_choice))
# Attach the VLAN interface to the Interface object
vlan_intf.attach(interface_objs[interface_choice])
# Attach the EPG to the VLAN interface
epg.attach(vlan_intf)
# Create some filters
filters = []
max_filters = int(self._config.get('Filters', 'Maximum'))
if max_filters > self._global_limits.max_filters:
max_filters = self._global_limits.max_filters
for i in range(
0,
random_number(
0,
random_number(int(self._config.get('Filters', 'Minimum')),
max_filters))):
filter = Filter(random_string(random_number(1, max_string_length)),
tenant)
self._global_limits.max_filters -= 1
filters.append(filter)
# Create some filter entries
filter_entries = []
max_filter_entries = int(self._config.get('FilterEntries', 'Maximum'))
if max_filter_entries > self._global_limits.max_filter_entries:
max_filter_entries = self._global_limits.max_filter_entries
if len(filters):
for i in range(
0,
random_number(
0,
random_number(
int(self._config.get('FilterEntries', 'Minimum')),
max_filter_entries))):
applyToFrag = '0'
arpOpc = '0'
dFromPort = '0'
dToPort = '0'
prot = '0'
sFromPort = '0'
sToPort = '0'
tcpRules = '0'
stateful = '0'
icmpv4T = 'not-given'
icmpv6T = 'not-given'
if random_chance(20): # 20% chance of ARP
arpOpc = random.choice(
ast.literal_eval(
self._config.get('FilterEntryOptions', 'ARPCode')))
etherT = 'arp'
elif random_chance(
25): # 20% of remaining 80% is non-IP (16% of total)
ethertype_choices = ast.literal_eval(
self._config.get('Ethertypes', 'Choice16PC'))
# if not filter.has_wildcard_entry():
# ethertype_choices += ['0']
etherT = random.choice(ethertype_choices)
else: # remaining is IP
applyToFrag = random.choice(
ast.literal_eval(
self._config.get('FilterEntryOptions',
'Fragmentation')))
etherT = 'ip'
if random_chance(
20
): # Choose more obscure protocols 20% of the time
prot = ConfigRandomizer._ip_protocols[random.choice(
ast.literal_eval(
self._config.get('IPProtocols',
'Choice20PC')))]
else:
prot = ConfigRandomizer._ip_protocols[random.choice(
ast.literal_eval(
self._config.get('IPProtocols',
'Choice80PC')))]
if prot == ConfigRandomizer._ip_protocols['icmp']:
icmpv4T = random.choice(
ast.literal_eval(
self._config.get('FilterEntryOptions',
'ICMP4Types')))
if icmpv4T != 'not-given':
# APIC will complain if both icmpv4T is specified and applyToFrag is set
applyToFrag = '0'
elif prot == ConfigRandomizer._ip_protocols['icmpv6']:
icmpv6T = random.choice(
ast.literal_eval(
self._config.get('FilterEntryOptions',
'ICMP6Types')))
if icmpv6T != 'not-given':
# APIC will complain if both icmpv6T is specified and applyToFrag is set
applyToFrag = '0'
else:
# Remainder is TCP or UDP
dFromPort, dToPort = random_range(
int(
self._config.get('FilterEntryOptions',
'PortRangeMin')),
int(
self._config.get('FilterEntryOptions',
'PortRangeMax')))
sFromPort, sToPort = random_range(
int(
self._config.get('FilterEntryOptions',
'PortRangeMin')),
int(
self._config.get('FilterEntryOptions',
'PortRangeMax')))
if dFromPort != '0' or dToPort != '0' or sFromPort != '0' or sToPort != '0':
applyToFrag = '0'
if prot == ConfigRandomizer._ip_protocols['tcp']:
# Randomly choose whether to specify tcpRules
if random_chance(30):
# TODO: should actually take odds from the config file
# Choose a random number of the possible tcpRules but
# if est is chosen, then it must be the only tcpRule. Otherwise, APIC rejects it
tcp_rule_choices = []
tcp_rule_possibilities = ast.literal_eval(
self._config.get(
'FilterEntryOptions', 'TCPRules'))
tcp_choice = random.choice(
tcp_rule_possibilities)
tcp_rule_choices.append(tcp_choice)
while tcp_choice != 'est':
tcp_choice = random.choice(
tcp_rule_possibilities)
if tcp_choice != 'est' and tcp_choice not in tcp_rule_choices:
tcp_rule_choices.append(tcp_choice)
tcpRules = ''
for tcp_choice in tcp_rule_choices:
tcpRules += str(tcp_choice) + ','
parent = random.choice(filters)
if not parent.has_entry(applyToFrag, arpOpc, dFromPort,
dToPort, etherT, prot, sFromPort,
sToPort, tcpRules, stateful, icmpv4T,
icmpv6T):
filter_entry = FilterEntry(name=random_string(
random_number(1, max_string_length)),
parent=parent,
applyToFrag=applyToFrag,
arpOpc=arpOpc,
dFromPort=dFromPort,
dToPort=dToPort,
etherT=etherT,
prot=prot,
sFromPort=sFromPort,
sToPort=sToPort,
tcpRules=tcpRules,
stateful=stateful,
icmpv4T=icmpv4T,
icmpv6T=icmpv6T)
# for l2tp traffic type we also need to udp filter with src and dst ports 1701
if etherT == 'ip' and prot == ConfigRandomizer._ip_protocols[
'l2tp']:
filter_entry = FilterEntry(
name=random_string(
random_number(1, max_string_length)),
parent=parent,
applyToFrag='0',
arpOpc=arpOpc,
dFromPort='1701',
dToPort='1701',
etherT=etherT,
prot=ConfigRandomizer._ip_protocols['udp'],
sFromPort='1701',
sToPort='1701',
tcpRules=tcpRules,
stateful=stateful,
icmpv4T=icmpv4T,
icmpv6T=icmpv6T)
self._global_limits.max_filter_entries -= 1
# Create some Contracts
contracts = []
max_contracts = int(self._config.get('Contracts', 'Maximum'))
if max_contracts > self._global_limits.max_contracts:
max_contracts = self._global_limits.max_contracts
for i in range(
0,
random_number(
0,
random_number(
int(self._config.get('Contracts', 'Minimum')),
max_contracts))):
contract = Contract(
random_string(random_number(1, max_string_length)), tenant)
self._global_limits.max_contracts -= 1
contracts.append(contract)
# Create some ContractSubjects
contract_subjects = []
if len(contracts):
for i in range(
0,
random_number(
0,
random_number(
int(self._config.get('ContractSubjects',
'Minimum')),
int(self._config.get('ContractSubjects',
'Maximum'))))):
contract_subject = ContractSubject(
random_string(random_number(1, max_string_length)),
random.choice(contracts))
contract_subjects.append(contract_subject)
# Randomly assign Filters to the ContractSubjects
for filter in filters:
if len(contracts) and len(contract_subjects):
already_picked = []
# Pick an arbitrary number of Subjects
for i in range(0, random_number(1, len(contract_subjects))):
pick = random_number(0, len(contract_subjects) - 1)
# Only choose each subject at most once
if pick not in already_picked:
contract_subjects[pick].add_filter(filter)
already_picked.append(pick)
# Randomly provide and consume the Contracts from the EPGs
for action, max_num_epgs in [
('provide',
int(self._config.get('Contracts', 'MaximumProvidingEPGs'))),
('consume',
int(self._config.get('Contracts', 'MaximumConsumingEPGs')))
]:
contract_count = [0] * len(contracts)
for epg in epgs:
already_picked = []
for i in range(0, random_number(0, len(contracts))):
keep_trying = 20 # Try 20 times to randomly pick a contract
while keep_trying:
pick = random_number(0, len(contracts) - 1)
if pick not in already_picked and contract_count[
pick] < max_num_epgs:
getattr(epg, action)(contracts[pick])
already_picked.append(pick)
contract_count[pick] += 1
keep_trying = 0
else:
keep_trying -= 1
# Create some Taboos
taboos = []
for i in range(
0,
random_number(
0,
random_number(int(self._config.get('Taboos', 'Minimum')),
int(self._config.get('Taboos',
'Maximum'))))):
taboo = Taboo(random_string(random_number(1, max_string_length)),
tenant)
taboos.append(taboo)
# Create some Taboo ContractSubjects
taboo_contract_subjects = []
if len(taboos):
for i in range(
0,
random_number(
1,
random_number(
int(
self._config.get('TabooContractSubjects',
'Minimum')),
int(
self._config.get('TabooContractSubjects',
'Maximum'))))):
taboo_contract_subject = ContractSubject(
random_string(random_number(1, max_string_length)),
random.choice(taboos))
taboo_contract_subjects.append(taboo_contract_subject)
# Randomly assign Filters to TabooContractSubjects
for filter in filters:
if len(taboo_contract_subjects):
already_picked = []
# Pick an arbitrary number of Subjects
for i in range(0, random_number(1,
len(taboo_contract_subjects))):
pick = random_number(0, len(taboo_contract_subjects) - 1)
# Only choose each subject at most once
if pick not in already_picked:
taboo_contract_subjects[pick].add_filter(filter)
already_picked.append(pick)
# Randomly protect epgs with taboos
for epg in epgs:
if random.choice([True, False, True]) and len(taboos):
epg.protect(taboos[random_number(0, len(taboos) - 1)])
return tenant
sToPort='5010',
tcpRules='unspecified',
parent=contract)
# Provide and consume the Contract
first_epg.provide(contract)
second_epg.consume(contract)
# Create the networking stuff and put both EPGs in the same BridgeDomain
vrf = Context('vrf-1', tenant)
bd = BridgeDomain('bd-1', tenant)
bd.add_context(vrf)
first_epg.add_bd(bd)
second_epg.add_bd(bd)
# Create the physical interface objects representing the physical ethernet ports
first_intf = Interface('eth', '1', '101', '1', '17')
second_intf = Interface('eth', '1', '102', '1', '17')
# Create a VLAN interface and attach to each physical interface
first_vlan_intf = L2Interface('vlan5-on-eth1-101-1-17', 'vlan', '5')
first_vlan_intf.attach(first_intf)
second_vlan_intf = L2Interface('vlan5-on-eth1-102-1-17', 'vlan', '5')
second_vlan_intf.attach(second_intf)
# Attach the EPGs to the VLAN interfaces
first_epg.attach(first_vlan_intf)
second_epg.attach(second_vlan_intf)
# Push the tenant configuration to the APIC
resp = session.push_to_apic(tenant.get_url(),
def create_random_tenant(self, interfaces=[]):
max_string_length = int(self._config.get('GlobalDefaults', 'MaximumStringLength'))
# Create the Tenant object
tenant_prefix = 'acitoolkitrandomized-'
tenant_name_len = random_number(1, max_string_length - len(tenant_prefix))
tenant_name = tenant_prefix + random_string(tenant_name_len)
tenant = Tenant(tenant_name)
# Create some number of BridgeDomains
bridge_domains = []
maximum_bds = int(self._config.get('BridgeDomains', 'Maximum'))
if maximum_bds > self._global_limits.max_bds:
maximum_bds = self._global_limits.max_bds
for i in range(0, random_number(0, random_number(int(self._config.get('BridgeDomains', 'Minimum')),
maximum_bds))):
self._global_limits.max_bds -= 1
bd = BridgeDomain(random_string(random_number(1, max_string_length)), tenant)
# Randomly choose settings for the BridgeDomain
if self._config.get('BridgeDomains', 'AllowFloodUnkMacUcast').lower() == 'true':
bd.set_unknown_mac_unicast(random.choice(['proxy', 'flood']))
if self._config.get('BridgeDomains', 'AllowOptimizedFloodUnknownMcast').lower() == 'true':
bd.set_unknown_multicast(random.choice(['flood', 'opt-flood']))
if self._config.get('BridgeDomains', 'AllowArpFlood').lower() == 'true':
bd.set_arp_flood(random.choice(['yes', 'no']))
if self._config.get('BridgeDomains', 'AllowDisableUnicastRoute').lower() == 'true':
bd.set_unicast_route(random.choice(['yes', 'no']))
if self._config.get('BridgeDomains', 'AllowNonDefaultMultiDstPkt').lower() == 'true':
bd.set_multidestination(random.choice(['drop', 'bd-flood', 'encap-flood']))
bridge_domains.append(bd)
# Create some number of Contexts
contexts = []
max_contexts = int(self._config.get('Contexts', 'Maximum'))
if max_contexts > self._global_limits.max_contexts:
max_contexts = self._global_limits.max_contexts
if max_contexts > int(self._config.get('Contexts', 'MaximumPerTenant')):
max_contexts = int(self._config.get('Contexts', 'MaximumPerTenant'))
for i in range(0, random_number(0, random_number(int(self._config.get('Contexts', 'Minimum')),
max_contexts))):
context = Context(random_string(random_number(1, max_string_length)), tenant)
self._global_limits.max_contexts -= 1
if self._config.get('Contexts', 'AllowUnenforced').lower() == 'true':
context.set_allow_all(random.choice([True, False]))
contexts.append(context)
# Randomly associate BridgeDomains with the Contexts (or use default)
for bd in bridge_domains:
if random.choice([True, True, False]) and len(contexts):
bd.add_context(random.choice(contexts))
# Create some number of Application Profiles
apps = []
for i in range(0, random_number(0, random_number(int(self._config.get('ApplicationProfiles', 'Minimum')),
int(self._config.get('ApplicationProfiles', 'Maximum'))))):
app = AppProfile(random_string(random_number(1, max_string_length)), tenant)
apps.append(app)
# Create some number of EPGs and place in AppProfiles
epgs = []
max_epgs = int(self._config.get('EPGs', 'Maximum'))
if max_epgs > self._global_limits.max_epgs:
max_epgs = self._global_limits.max_epgs
if len(apps):
for i in range(0, random_number(0, random_number(int(self._config.get('EPGs', 'Minimum')),
max_epgs))):
epg = EPG(random_string(random_number(1, max_string_length)), random.choice(apps))
self._global_limits.max_epgs -= 1
epgs.append(epg)
# Randomly associate the EPGs to BridgeDomains
bd_epg_count = [0] * len(bridge_domains)
for epg in epgs:
if random_number(0, 9) == 1 or len(bridge_domains) == 0: # 1 in 10 chance for no bridgedomain
continue
keep_trying = 100
while keep_trying:
bd_choice = random_number(0, len(bridge_domains) - 1)
if bd_epg_count[bd_choice] <= int(self._config.get('BridgeDomains', 'MaximumEPGs')):
epg.add_bd(bridge_domains[bd_choice])
bd_epg_count[bd_choice] += 1
break
else:
keep_trying -= 1
# Randomly assign the EPGs to the interfaces provided
interface_objs = {}
for interface in interfaces:
# Create the Interface objects
interface_objs[interface] = Interface.create_from_name(interface)
for epg in epgs:
# Pick an interface
interface_choice = random.choice(interfaces)
# Pick a VLAN
vlan_choice = 0
keep_trying = 100
while vlan_choice in self._interfaces[interface_choice]:
vlan_choice = random_number(int(self._config.get('VLANs', 'Minimum')),
int(self._config.get('VLANs', 'Maximum')))
keep_trying -= 1
if not keep_trying:
continue
# Create the VLAN interface
vlan_intf = L2Interface('vlan%s-on-%s' % (str(vlan_choice),
interface_objs[interface_choice].name.replace(' ', '')),
'vlan',
str(vlan_choice))
# Attach the VLAN interface to the Interface object
vlan_intf.attach(interface_objs[interface_choice])
# Attach the EPG to the VLAN interface
epg.attach(vlan_intf)
# Create some filters
filters = []
max_filters = int(self._config.get('Filters', 'Maximum'))
if max_filters > self._global_limits.max_filters:
max_filters = self._global_limits.max_filters
for i in range(0, random_number(0, random_number(int(self._config.get('Filters', 'Minimum')),
max_filters))):
filter = Filter(random_string(random_number(1, max_string_length)), tenant)
self._global_limits.max_filters -= 1
filters.append(filter)
# Create some filter entries
filter_entries = []
max_filter_entries = int(self._config.get('FilterEntries', 'Maximum'))
if max_filter_entries > self._global_limits.max_filter_entries:
max_filter_entries = self._global_limits.max_filter_entries
ip_protocols = {
'icmp': '1',
'igmp': '2',
'tcp': '6',
'egp': '8',
'igp': '9',
'udp': '17',
'icmpv6': '58',
'eigrp': '88',
'ospfigp': '89',
'pim': '103',
'l2tp': '115'
}
if len(filters):
for i in range(0, random_number(0, random_number(int(self._config.get('FilterEntries', 'Minimum')),
max_filter_entries))):
applyToFrag = '0'
arpOpc = '0'
dFromPort = '0'
dToPort = '0'
prot = '0'
sFromPort = '0'
sToPort = '0'
tcpRules = '0'
stateful = '0'
icmpv4T = 'not-given'
icmpv6T = 'not-given'
if random_chance(20): # 20% chance of ARP
arpOpc = random.choice(['req', 'reply'])
etherT = 'arp'
elif random_chance(25): # 20% of remaining 80% is non-IP (16% of total)
ethertype_choices = ['trill', 'mpls_ucast', 'mac_security', 'fcoe']
# if not filter.has_wildcard_entry():
# ethertype_choices += ['0']
etherT = random.choice(ethertype_choices)
else: # remaining is IP
applyToFrag = random.choice(['0', '1'])
etherT = 'ip'
if random_chance(20): # Choose more obscure protocols 20% of the time
prot = ip_protocols[random.choice(['igmp', 'egp', 'igp', 'eigrp', 'ospfigp', 'pim', 'l2tp'])]
else:
prot = ip_protocols[random.choice(['icmp', 'tcp', 'udp', 'icmpv6'])]
if prot == ip_protocols['icmp']:
icmpv4T = random.choice(['echo-rep', 'dst-unreach', 'src-quench', 'echo',
'time-exceeded', 'unspecified', 'not-given'])
elif prot == ip_protocols['icmpv6']:
icmpv6T = random.choice(['unspecified', 'dst-unreach', 'time-exceeded',
'echo-req', 'echo-rep', 'nbr-solicit', 'nbr-advert',
'redirect', 'not-given'])
else:
# Remainder is TCP or UDP
dFromPort, dToPort = random_range(0, 65535)
sFromPort, sToPort = random_range(0, 65535)
if dFromPort != '0' or dToPort != '0' or sFromPort != '0' or sToPort != '0':
applyToFrag = '0'
if prot == ip_protocols['tcp']:
# Randomly choose whether to specify tcpRules
if random_chance(30):
# TODO: should actually take odds from the config file
# Choose a random number of the possible tcpRules but
# if est is chosen, then it must be the only tcpRule. Otherwise, APIC rejects it
tcp_rule_choices = []
tcp_rule_possibilities = ['est', 'syn', 'ack', 'fin', 'rst']
tcp_choice = random.choice(tcp_rule_possibilities)
tcp_rule_choices.append(tcp_choice)
while tcp_choice != 'est':
tcp_choice = random.choice(tcp_rule_possibilities)
if tcp_choice != 'est' and tcp_choice not in tcp_rule_choices:
tcp_rule_choices.append(tcp_choice)
tcpRules = ''
for tcp_choice in tcp_rule_choices:
tcpRules += str(tcp_choice) + ','
parent = random.choice(filters)
if not parent.has_entry(applyToFrag, arpOpc, dFromPort, dToPort, etherT, prot, sFromPort, sToPort,
tcpRules, stateful, icmpv4T, icmpv6T):
filter_entry = FilterEntry(name=random_string(random_number(1, max_string_length)),
parent=random.choice(filters),
applyToFrag=applyToFrag,
arpOpc=arpOpc,
dFromPort=dFromPort,
dToPort=dToPort,
etherT=etherT,
prot=prot,
sFromPort=sFromPort,
sToPort=sToPort,
tcpRules=tcpRules,
stateful=stateful,
icmpv4T=icmpv4T,
icmpv6T=icmpv6T)
self._global_limits.max_filter_entries -= 1
# Create some Contracts
contracts = []
max_contracts = int(self._config.get('Contracts', 'Maximum'))
if max_contracts > self._global_limits.max_contracts:
max_contracts = self._global_limits.max_contracts
for i in range(0, random_number(0, random_number(int(self._config.get('Contracts', 'Minimum')),
max_contracts))):
contract = Contract(random_string(random_number(1, max_string_length)), tenant)
self._global_limits.max_contracts -= 1
contracts.append(contract)
# Create some ContractSubjects
if len(contracts):
contract_subjects = []
for i in range(0, random_number(0, random_number(int(self._config.get('ContractSubjects', 'Minimum')),
int(self._config.get('ContractSubjects', 'Maximum'))))):
contract_subject = ContractSubject(random_string(random_number(1, max_string_length)), random.choice(contracts))
contract_subjects.append(contract_subject)
# Randomly assign Filters to the ContractSubjects
for filter in filters:
if len(contracts) and len(contract_subjects):
already_picked = []
# Pick an arbitrary number of Subjects
for i in range(0, random_number(1, len(contract_subjects))):
pick = random_number(0, len(contract_subjects) - 1)
# Only choose each subject at most once
if pick not in already_picked:
contract_subjects[pick].add_filter(filter)
already_picked.append(pick)
# Randomly provide and consume the Contracts from the EPGs
for action, max_num_epgs in [('provide', int(self._config.get('Contracts', 'MaximumProvidingEPGs'))),
('consume', int(self._config.get('Contracts', 'MaximumConsumingEPGs')))]:
contract_count = [0] * len(contracts)
for epg in epgs:
already_picked = []
for i in range(0, random_number(0, len(contracts))):
keep_trying = 20 # Try 20 times to randomly pick a contract
while keep_trying:
pick = random_number(0, len(contracts) - 1)
if pick not in already_picked and contract_count[pick] < max_num_epgs:
getattr(epg, action)(contracts[pick])
already_picked.append(pick)
contract_count[pick] += 1
keep_trying = 0
else:
keep_trying -= 1
return tenant
def create_random_tenant(self, interfaces=[]):
max_string_length = int(self._config.get('GlobalDefaults', 'MaximumStringLength'))
# Create the Tenant object
tenant_prefix = self._config.get('GlobalDefaults', 'TenantPrefix')
tenant_name_len = random_number(1, max_string_length - len(tenant_prefix))
tenant_name = tenant_prefix + random_string(tenant_name_len)
tenant = Tenant(tenant_name)
# Create some number of BridgeDomains
bridge_domains = []
maximum_bds = int(self._config.get('BridgeDomains', 'Maximum'))
if maximum_bds > self._global_limits.max_bds:
maximum_bds = self._global_limits.max_bds
for i in range(0, random_number(0, random_number(int(self._config.get('BridgeDomains', 'Minimum')),
maximum_bds))):
self._global_limits.max_bds -= 1
bd = BridgeDomain(random_string(random_number(1, max_string_length)), tenant)
# Randomly choose settings for the BridgeDomain
if self._config.get('BridgeDomains', 'AllowFloodUnkMacUcast').lower() == 'true':
bd.set_unknown_mac_unicast(random.choice(ast.literal_eval(self._config.get('BridgeDomainSettings', 'UnknownMacUnicast'))))
if self._config.get('BridgeDomains', 'AllowOptimizedFloodUnknownMcast').lower() == 'true':
bd.set_unknown_multicast(random.choice(ast.literal_eval(self._config.get('BridgeDomainSettings', 'UnknownMulticast'))))
if self._config.get('BridgeDomains', 'AllowArpFlood').lower() == 'true':
bd.set_arp_flood(random.choice(ast.literal_eval(self._config.get('BridgeDomainSettings', 'ArpFlood'))))
if self._config.get('BridgeDomains', 'AllowDisableUnicastRoute').lower() == 'true':
bd.set_unicast_route(random.choice(ast.literal_eval(self._config.get('BridgeDomainSettings', 'UnicastRoute'))))
if self._config.get('BridgeDomains', 'AllowNonDefaultMultiDstPkt').lower() == 'true':
bd.set_multidestination(random.choice(ast.literal_eval(self._config.get('BridgeDomainSettings', 'Multidestination'))))
bridge_domains.append(bd)
# Create some number of Contexts
contexts = []
max_contexts = int(self._config.get('Contexts', 'Maximum'))
if max_contexts > self._global_limits.max_contexts:
max_contexts = self._global_limits.max_contexts
if max_contexts > int(self._config.get('Contexts', 'MaximumPerTenant')):
max_contexts = int(self._config.get('Contexts', 'MaximumPerTenant'))
for i in range(0, random_number(0, random_number(int(self._config.get('Contexts', 'Minimum')),
max_contexts))):
context = Context(random_string(random_number(1, max_string_length)), tenant)
self._global_limits.max_contexts -= 1
if self._config.get('Contexts', 'AllowUnenforced').lower() == 'true':
context.set_allow_all(random.choice([True, False]))
contexts.append(context)
# Randomly associate BridgeDomains with the Contexts (or use default)
for bd in bridge_domains:
if random.choice([True, True, False]) and len(contexts):
bd.add_context(random.choice(contexts))
# Create some number of Application Profiles
apps = []
for i in range(0, random_number(0, random_number(int(self._config.get('ApplicationProfiles', 'Minimum')),
int(self._config.get('ApplicationProfiles', 'Maximum'))))):
app = AppProfile(random_string(random_number(1, max_string_length)), tenant)
apps.append(app)
# Create some number of EPGs and place in AppProfiles
epgs = []
max_epgs = int(self._config.get('EPGs', 'Maximum'))
if max_epgs > self._global_limits.max_epgs:
max_epgs = self._global_limits.max_epgs
if len(apps):
for i in range(0, random_number(0, random_number(int(self._config.get('EPGs', 'Minimum')),
max_epgs))):
epg = EPG(random_string(random_number(1, max_string_length)), random.choice(apps))
self._global_limits.max_epgs -= 1
epgs.append(epg)
# Randomly associate the EPGs to BridgeDomains
bd_epg_count = [0] * len(bridge_domains)
for epg in epgs:
if random_number(0, 9) == 1 or len(bridge_domains) == 0: # 1 in 10 chance for no bridgedomain
continue
keep_trying = 100
while keep_trying:
bd_choice = random_number(0, len(bridge_domains) - 1)
if bd_epg_count[bd_choice] <= int(self._config.get('BridgeDomains', 'MaximumEPGs')):
epg.add_bd(bridge_domains[bd_choice])
bd_epg_count[bd_choice] += 1
break
else:
keep_trying -= 1
# Randomly assign the EPGs to the interfaces provided
interface_objs = {}
for interface in interfaces:
# Create the Interface objects
interface_objs[interface] = Interface.create_from_name(interface)
for epg in epgs:
# Pick an interface
interface_choice = random.choice(interfaces)
# Pick a VLAN
vlan_choice = 0
keep_trying = 100
while vlan_choice in self._interfaces[interface_choice]:
vlan_choice = random_number(int(self._config.get('VLANs', 'Minimum')),
int(self._config.get('VLANs', 'Maximum')))
keep_trying -= 1
if not keep_trying:
continue
# Create the VLAN interface
vlan_intf = L2Interface('vlan%s-on-%s' % (str(vlan_choice),
interface_objs[interface_choice].name.replace(' ', '')),
'vlan',
str(vlan_choice))
# Attach the VLAN interface to the Interface object
vlan_intf.attach(interface_objs[interface_choice])
# Attach the EPG to the VLAN interface
epg.attach(vlan_intf)
# Create some filters
filters = []
max_filters = int(self._config.get('Filters', 'Maximum'))
if max_filters > self._global_limits.max_filters:
max_filters = self._global_limits.max_filters
for i in range(0, random_number(0, random_number(int(self._config.get('Filters', 'Minimum')),
max_filters))):
filter = Filter(random_string(random_number(1, max_string_length)), tenant)
self._global_limits.max_filters -= 1
filters.append(filter)
# Create some filter entries
filter_entries = []
max_filter_entries = int(self._config.get('FilterEntries', 'Maximum'))
if max_filter_entries > self._global_limits.max_filter_entries:
max_filter_entries = self._global_limits.max_filter_entries
if len(filters):
for i in range(0, random_number(0, random_number(int(self._config.get('FilterEntries', 'Minimum')),
max_filter_entries))):
applyToFrag = '0'
arpOpc = '0'
dFromPort = '0'
dToPort = '0'
prot = '0'
sFromPort = '0'
sToPort = '0'
tcpRules = '0'
stateful = '0'
icmpv4T = 'not-given'
icmpv6T = 'not-given'
if random_chance(20): # 20% chance of ARP
arpOpc = random.choice(ast.literal_eval(self._config.get('FilterEntryOptions', 'ARPCode')))
etherT = 'arp'
elif random_chance(25): # 20% of remaining 80% is non-IP (16% of total)
ethertype_choices = ast.literal_eval(self._config.get('Ethertypes', 'Choice16PC'))
# if not filter.has_wildcard_entry():
# ethertype_choices += ['0']
etherT = random.choice(ethertype_choices)
else: # remaining is IP
applyToFrag = random.choice(ast.literal_eval(self._config.get('FilterEntryOptions', 'Fragmentation')))
etherT = 'ip'
if random_chance(20): # Choose more obscure protocols 20% of the time
prot = ConfigRandomizer._ip_protocols[random.choice(ast.literal_eval(self._config.get('IPProtocols', 'Choice20PC')))]
else:
prot = ConfigRandomizer._ip_protocols[random.choice(ast.literal_eval(self._config.get('IPProtocols', 'Choice80PC')))]
if prot == ConfigRandomizer._ip_protocols['icmp']:
icmpv4T = random.choice(ast.literal_eval(self._config.get('FilterEntryOptions', 'ICMP4Types')))
elif prot == ConfigRandomizer._ip_protocols['icmpv6']:
icmpv6T = random.choice(ast.literal_eval(self._config.get('FilterEntryOptions', 'ICMP6Types')))
else:
# Remainder is TCP or UDP
dFromPort, dToPort = random_range(int(self._config.get('FilterEntryOptions', 'PortRangeMin')), int(self._config.get('FilterEntryOptions', 'PortRangeMax')))
sFromPort, sToPort = random_range(int(self._config.get('FilterEntryOptions', 'PortRangeMin')), int(self._config.get('FilterEntryOptions', 'PortRangeMax')))
if dFromPort != '0' or dToPort != '0' or sFromPort != '0' or sToPort != '0':
applyToFrag = '0'
if prot == ConfigRandomizer._ip_protocols['tcp']:
# Randomly choose whether to specify tcpRules
if random_chance(30):
# TODO: should actually take odds from the config file
# Choose a random number of the possible tcpRules but
# if est is chosen, then it must be the only tcpRule. Otherwise, APIC rejects it
tcp_rule_choices = []
tcp_rule_possibilities = ast.literal_eval(self._config.get('FilterEntryOptions', 'TCPRules'))
tcp_choice = random.choice(tcp_rule_possibilities)
tcp_rule_choices.append(tcp_choice)
while tcp_choice != 'est':
tcp_choice = random.choice(tcp_rule_possibilities)
if tcp_choice != 'est' and tcp_choice not in tcp_rule_choices:
tcp_rule_choices.append(tcp_choice)
tcpRules = ''
for tcp_choice in tcp_rule_choices:
tcpRules += str(tcp_choice) + ','
parent = random.choice(filters)
if not parent.has_entry(applyToFrag, arpOpc, dFromPort, dToPort, etherT, prot, sFromPort, sToPort,
tcpRules, stateful, icmpv4T, icmpv6T):
filter_entry = FilterEntry(name=random_string(random_number(1, max_string_length)),
parent=parent,
applyToFrag=applyToFrag,
arpOpc=arpOpc,
dFromPort=dFromPort,
dToPort=dToPort,
etherT=etherT,
prot=prot,
sFromPort=sFromPort,
sToPort=sToPort,
tcpRules=tcpRules,
stateful=stateful,
icmpv4T=icmpv4T,
icmpv6T=icmpv6T)
#for l2tp traffic type we also need to udp filter with src and dst ports 1701
if etherT == 'ip' and prot == ConfigRandomizer._ip_protocols['l2tp']:
filter_entry = FilterEntry(name=random_string(random_number(1, max_string_length)),
parent=parent,
applyToFrag='0',
arpOpc=arpOpc,
dFromPort='1701',
dToPort='1701',
etherT=etherT,
prot=ConfigRandomizer._ip_protocols['udp'],
sFromPort='1701',
sToPort='1701',
tcpRules=tcpRules,
stateful=stateful,
icmpv4T=icmpv4T,
icmpv6T=icmpv6T)
self._global_limits.max_filter_entries -= 1
# Create some Contracts
contracts = []
max_contracts = int(self._config.get('Contracts', 'Maximum'))
if max_contracts > self._global_limits.max_contracts:
max_contracts = self._global_limits.max_contracts
for i in range(0, random_number(0, random_number(int(self._config.get('Contracts', 'Minimum')),
max_contracts))):
contract = Contract(random_string(random_number(1, max_string_length)), tenant)
self._global_limits.max_contracts -= 1
contracts.append(contract)
# Create some ContractSubjects
contract_subjects = []
if len(contracts):
for i in range(0, random_number(0, random_number(int(self._config.get('ContractSubjects', 'Minimum')),
int(self._config.get('ContractSubjects', 'Maximum'))))):
contract_subject = ContractSubject(random_string(random_number(1, max_string_length)), random.choice(contracts))
contract_subjects.append(contract_subject)
# Randomly assign Filters to the ContractSubjects
for filter in filters:
if len(contracts) and len(contract_subjects):
already_picked = []
# Pick an arbitrary number of Subjects
for i in range(0, random_number(1, len(contract_subjects))):
pick = random_number(0, len(contract_subjects) - 1)
# Only choose each subject at most once
if pick not in already_picked:
contract_subjects[pick].add_filter(filter)
already_picked.append(pick)
# Randomly provide and consume the Contracts from the EPGs
for action, max_num_epgs in [('provide', int(self._config.get('Contracts', 'MaximumProvidingEPGs'))),
('consume', int(self._config.get('Contracts', 'MaximumConsumingEPGs')))]:
contract_count = [0] * len(contracts)
for epg in epgs:
already_picked = []
for i in range(0, random_number(0, len(contracts))):
keep_trying = 20 # Try 20 times to randomly pick a contract
while keep_trying:
pick = random_number(0, len(contracts) - 1)
if pick not in already_picked and contract_count[pick] < max_num_epgs:
getattr(epg, action)(contracts[pick])
already_picked.append(pick)
contract_count[pick] += 1
keep_trying = 0
else:
keep_trying -= 1
# Create some Taboos
taboos = []
for i in range(0, random_number(0, random_number(int(self._config.get('Taboos', 'Minimum')),
int(self._config.get('Taboos', 'Maximum'))))):
taboo = Taboo(random_string(random_number(1, max_string_length)), tenant)
taboos.append(taboo)
# Create some Taboo ContractSubjects
taboo_contract_subjects = []
if len(taboos):
for i in range(0, random_number(1, random_number(int(self._config.get('TabooContractSubjects', 'Minimum')),
int(self._config.get('TabooContractSubjects', 'Maximum'))))):
taboo_contract_subject = ContractSubject(random_string(random_number(1, max_string_length)), random.choice(taboos))
taboo_contract_subjects.append(taboo_contract_subject)
# Randomly assign Filters to TabooContractSubjects
for filter in filters:
if len(taboo_contract_subjects):
already_picked = []
# Pick an arbitrary number of Subjects
for i in range(0, random_number(1, len(taboo_contract_subjects))):
pick = random_number(0, len(taboo_contract_subjects) - 1)
# Only choose each subject at most once
if pick not in already_picked:
taboo_contract_subjects[pick].add_filter(filter)
already_picked.append(pick)
# Randomly protect epgs with taboos
for epg in epgs:
if random.choice([True, False, True]) and len(taboos):
epg.protect(taboos[random_number(0, len(taboos)-1)])
return tenant