diff options
| -rw-r--r-- | tools/config/quagga-rsconfig.go | 103 | ||||
| -rwxr-xr-x | tools/config/root-server-docker.sh | 67 |
2 files changed, 170 insertions, 0 deletions
diff --git a/tools/config/quagga-rsconfig.go b/tools/config/quagga-rsconfig.go new file mode 100644 index 00000000..cf3d211c --- /dev/null +++ b/tools/config/quagga-rsconfig.go @@ -0,0 +1,103 @@ +package main + +import ( + "bytes" + "fmt" + "github.com/BurntSushi/toml" + "github.com/jessevdk/go-flags" + "github.com/osrg/gobgp/config" + "io/ioutil" + "log" + "net" + "os" + "path/filepath" +) + +type QuaggaConfig struct { + id int + config *config.NeighborType + gobgpConfig *config.GlobalType + serverIP net.IP +} + +func NewQuaggaConfig(id int, gConfig *config.GlobalType, myConfig *config.NeighborType, server net.IP) *QuaggaConfig { + return &QuaggaConfig{ + id: id, + config: myConfig, + gobgpConfig: gConfig, + serverIP: server, + } +} + +func (qt *QuaggaConfig) Config() *bytes.Buffer { + buf := bytes.NewBuffer(nil) + + buf.WriteString("hostname bgpd\n") + buf.WriteString("password zebra\n") + buf.WriteString(fmt.Sprintf("router bgp %d\n", qt.config.PeerAs)) + buf.WriteString(fmt.Sprintf("bgp router-id 192.168.0.%d\n", qt.id)) + buf.WriteString(fmt.Sprintf("network 192.168.%d.0/24\n", qt.id)) + buf.WriteString(fmt.Sprintf("neighbor %s remote-as %d\n", qt.serverIP, qt.gobgpConfig.As)) + buf.WriteString(fmt.Sprintf("neighbor %s password %s\n", qt.serverIP, qt.config.AuthPassword)) + buf.WriteString("log file /var/log/quagga/bgpd.log") + return buf +} + +func create_config_files(nr int, outputDir string) { + quaggaConfigList := make([]*QuaggaConfig, 0) + + gobgpConf := config.BgpType{ + Global: config.GlobalType{ + As: 64512, + RouterId: net.ParseIP("192.168.255.1"), + }, + } + + for i := 1; i < nr+1; i++ { + c := config.NeighborType{ + PeerAs: 64513 + uint32(i), + NeighborAddress: net.ParseIP(fmt.Sprintf("10.0.0.%d", i)), + AuthPassword: fmt.Sprintf("hoge%d", i), + } + gobgpConf.NeighborList = append(gobgpConf.NeighborList, c) + q := NewQuaggaConfig(i, &gobgpConf.Global, &c, net.ParseIP("10.0.255.1")) + quaggaConfigList = append(quaggaConfigList, q) + os.Mkdir(fmt.Sprintf("%s/q%d", outputDir, i), 0755) + err := ioutil.WriteFile(fmt.Sprintf("%s/q%d/bgpd.conf", outputDir, i), q.Config().Bytes(), 0644) + if err != nil { + log.Fatal(err) + } + } + + var buffer bytes.Buffer + encoder := toml.NewEncoder(&buffer) + encoder.Encode(gobgpConf) + + err := ioutil.WriteFile(fmt.Sprintf("%s/gobgpd.conf", outputDir), buffer.Bytes(), 0644) + if err != nil { + log.Fatal(err) + } +} + +func main() { + var opts struct { + ClientNumber int `short:"n" long:"client-number" description:"specfying the number of clients" default:"8"` + OutputDir string `short:"c" long:"output" description:"specifing the output directory"` + } + parser := flags.NewParser(&opts, flags.Default) + + _, err := parser.Parse() + if err != nil { + os.Exit(1) + } + + if opts.OutputDir == "" { + opts.OutputDir, _ = filepath.Abs(".") + } else { + if _, err := os.Stat(opts.OutputDir); os.IsNotExist(err) { + os.Mkdir(opts.OutputDir, 0755) + } + } + + create_config_files(opts.ClientNumber, opts.OutputDir) +} diff --git a/tools/config/root-server-docker.sh b/tools/config/root-server-docker.sh new file mode 100755 index 00000000..f2af3d7d --- /dev/null +++ b/tools/config/root-server-docker.sh @@ -0,0 +1,67 @@ +#!/bin/sh + +NR_PEERS=8 +BRIDGE_NAME=br0 +CONFIG_DIR=`pwd` + +run_quagga() { + local docker_name=q$1 + docker run --privileged=true -v $CONFIG_DIR/$docker_name:/etc/quagga --name $docker_name -id osrg/quagga + sudo pipework $BRIDGE_NAME $docker_name 10.0.0.$1/16 +} + +stop_quagga() { + local docker_name=q$1 + docker rm -f $docker_name +} + +delete_bridge() { + local name=$1 + local sysfs_name=/sys/class/net/$name + if [ -e $sysfs_name ]; then + sudo ifconfig $name down + sudo brctl delbr $name + fi +} + +while getopts c: OPT +do + case $OPT in + c) CONFIG_DIR="$OPTARG" + ;; + *) echo "Unknown option" + exit 1 + ;; + esac +done + +shift $((OPTIND - 1)) + +case "$1" in + start) + i=1 + while [ $i -le $NR_PEERS ] + do + run_quagga $i + i=$(( i+1 )) + done + sudo ip addr add 10.0.255.1/16 dev $BRIDGE_NAME + ;; + + stop) + i=1 + while [ $i -le $NR_PEERS ] + do + stop_quagga $i + i=$(( i+1 )) + done + delete_bridge $BRIDGE_NAME + ;; + *) + echo $1 + echo "Usage: root-server-docker {start|stop}" + exit 2 + ;; +esac + + |