run_demo.py

import argparse
import subprocess
import time
def setup_veth(veth0,veth1):
  subprocess.Popen("bash veth_setup.sh %s %s"%(veth0,veth1),shell=True).wait()

BMV2_PATH="/home/whr/behavioral-model/"
SWITCH_PATH=BMV2_PATH+"targets/simple_switch/simple_switch "
CLI_PATH=BMV2_PATH+"targets/simple_switch/sswitch_CLI "

parser = argparse.ArgumentParser()
parser.add_argument("topo_file",help="topo file")
parser.add_argument("-p","--p4-file",help="p4 file path")
parser.add_argument("-t","--thrift-port",help="thrift port of s0, default is 9000",default=9000)
parser.add_argument("-j","--json-file",help="json file name, default is t.json",default="t.json")
parser.add_argument("-c","--console-log",help="print log of simple switch", action='store_true')
parser.add_argument("-m","--command-file",help="use sx-commands.txt as init entry", action='store_true')
args = parser.parse_args()
args.thrift_port=int(args.thrift_port)
if args.p4_file:
  subprocess.Popen("p4c-bm2-ss --std p4-14 -o %s %s"%(args.json_file,args.p4_file),shell=True).wait()

with open(args.topo_file,"r") as f:
  cmds=f.readlines()

tors=cmds[:2][0][:-1].split(":")[-1]
core=cmds[:2][1][:-1].split(":")[-1]
tors=[_[1:] for _ in tors.split(",")]
core=[_[1:] for _ in core.split(",")]
sws=tors+core
print(sws)
links=dict(zip(sws,[{} for _ in sws]))
topo=cmds[2:]
for s in sws:
  setup_veth("s%s-int"%(s,),"s%s-mgr"%(s,))
for s in tors:
  setup_veth("s%s-trf"%(s,),"s%s-tin"%(s,))
for t in topo:
  s1=t[:-1].split("-")[0]
  s2=t[:-1].split("-")[1]
  s1n=s1.split(":")[0][1:]
  s1i=s1.split(":")[1]
  s2n=s2.split(":")[0][1:]
  s2i=s2.split(":")[1]
  setup_veth("s%s-eth%s"%(s1n,s1i),"s%s-eth%s"%(s2n,s2i))
  links[s1n][s1i]="s%s-eth%s"%(s1n,s1i)
  links[s2n][s2i]="s%s-eth%s"%(s2n,s2i)
switch_args_list=[]
for i in sws:
  switch_args="%s --thrift-port %d --device-id %s "%(args.json_file,args.thrift_port+int(i),i)
  if i in tors:
    switch_args+="-i 10@s%s-tin "%(i,)
  for j,p in links[i].items():
    switch_args+="-i %s@%s "%(j,p)
  switch_args+="-i 11@s%s-mgr "%(i,)
  if args.console_log:
    switch_args+="--log-console "
  print(switch_args)
  switch_args_list.append(switch_args)

for i in range (10):
  subprocess.Popen(SWITCH_PATH+switch_args_list[i],shell=True)

time.sleep(4)
if args.command_file:
  for i in sws:
    subprocess.Popen(CLI_PATH+"%s %d < s%s-commands.txt"%(args.json_file,args.thrift_port+int(i),i),shell=True).wait()