hs-test: add tls proxy test
[vpp.git] / extras / hs-test / vppinstance.go
index 4092d35..e909b85 100644 (file)
@@ -2,11 +2,16 @@ package main
 
 import (
        "fmt"
-       "github.com/edwarnicke/exechelper"
+       "os"
        "os/exec"
+       "os/signal"
+       "strconv"
        "strings"
+       "syscall"
        "time"
 
+       "github.com/edwarnicke/exechelper"
+
        "go.fd.io/govpp"
        "go.fd.io/govpp/api"
        "go.fd.io/govpp/binapi/af_packet"
@@ -51,6 +56,9 @@ plugins {
   plugin af_packet_plugin.so { enable }
   plugin hs_apps_plugin.so { enable }
   plugin http_plugin.so { enable }
+  plugin http_static_plugin.so { enable }
+  plugin prom_plugin.so { enable }
+  plugin tlsopenssl_plugin.so { enable }
 }
 
 logging {
@@ -68,34 +76,35 @@ const (
 
 type VppInstance struct {
        container        *Container
-       additionalConfig Stanza
+       additionalConfig []Stanza
        connection       *core.Connection
        apiChannel       api.Channel
+       cpus             []int
 }
 
-func (vpp *VppInstance) Suite() *HstSuite {
+func (vpp *VppInstance) getSuite() *HstSuite {
        return vpp.container.suite
 }
 
 func (vpp *VppInstance) getCliSocket() string {
-       return fmt.Sprintf("%s%s", vpp.container.GetContainerWorkDir(), defaultCliSocketFilePath)
+       return fmt.Sprintf("%s%s", vpp.container.getContainerWorkDir(), defaultCliSocketFilePath)
 }
 
 func (vpp *VppInstance) getRunDir() string {
-       return vpp.container.GetContainerWorkDir() + "/var/run/vpp"
+       return vpp.container.getContainerWorkDir() + "/var/run/vpp"
 }
 
 func (vpp *VppInstance) getLogDir() string {
-       return vpp.container.GetContainerWorkDir() + "/var/log/vpp"
+       return vpp.container.getContainerWorkDir() + "/var/log/vpp"
 }
 
 func (vpp *VppInstance) getEtcDir() string {
-       return vpp.container.GetContainerWorkDir() + "/etc/vpp"
+       return vpp.container.getContainerWorkDir() + "/etc/vpp"
 }
 
 func (vpp *VppInstance) start() error {
        // Create folders
-       containerWorkDir := vpp.container.GetContainerWorkDir()
+       containerWorkDir := vpp.container.getContainerWorkDir()
 
        vpp.container.exec("mkdir --mode=0700 -p " + vpp.getRunDir())
        vpp.container.exec("mkdir --mode=0700 -p " + vpp.getLogDir())
@@ -109,15 +118,41 @@ func (vpp *VppInstance) start() error {
                defaultApiSocketFilePath,
                defaultLogFilePath,
        )
-       configContent += vpp.additionalConfig.ToString()
+       configContent += vpp.generateCpuConfig()
+       for _, c := range vpp.additionalConfig {
+               configContent += c.toString()
+       }
        startupFileName := vpp.getEtcDir() + "/startup.conf"
        vpp.container.createFile(startupFileName, configContent)
 
-       // Start VPP
-       vpp.container.execServer("su -c \"vpp -c " + startupFileName + " &> /proc/1/fd/1\"")
+       // create wrapper script for vppctl with proper CLI socket path
+       cliContent := "#!/usr/bin/bash\nvppctl -s " + vpp.getRunDir() + "/cli.sock"
+       vppcliFileName := "/usr/bin/vppcli"
+       vpp.container.createFile(vppcliFileName, cliContent)
+       vpp.container.exec("chmod 0755 " + vppcliFileName)
+
+       if *isVppDebug {
+               sig := make(chan os.Signal, 1)
+               signal.Notify(sig, syscall.SIGINT)
+               cont := make(chan bool, 1)
+               go func() {
+                       <-sig
+                       cont <- true
+               }()
+
+               vpp.container.execServer("su -c \"vpp -c " + startupFileName + " &> /proc/1/fd/1\"")
+               fmt.Println("run following command in different terminal:")
+               fmt.Println("docker exec -it " + vpp.container.name + " gdb -ex \"attach $(docker exec " + vpp.container.name + " pidof vpp)\"")
+               fmt.Println("Afterwards press CTRL+C to continue")
+               <-cont
+               fmt.Println("continuing...")
+       } else {
+               // Start VPP
+               vpp.container.execServer("su -c \"vpp -c " + startupFileName + " &> /proc/1/fd/1\"")
+       }
 
        // Connect to VPP and store the connection
-       sockAddress := vpp.container.GetHostWorkDir() + defaultApiSocketFilePath
+       sockAddress := vpp.container.getHostWorkDir() + defaultApiSocketFilePath
        conn, connEv, err := govpp.AsyncConnect(
                sockAddress,
                core.DefaultMaxReconnectAttempts,
@@ -153,13 +188,30 @@ func (vpp *VppInstance) vppctl(command string, arguments ...any) string {
        vppCliCommand := fmt.Sprintf(command, arguments...)
        containerExecCommand := fmt.Sprintf("docker exec --detach=false %[1]s vppctl -s %[2]s %[3]s",
                vpp.container.name, vpp.getCliSocket(), vppCliCommand)
-       vpp.Suite().log(containerExecCommand)
+       vpp.getSuite().log(containerExecCommand)
        output, err := exechelper.CombinedOutput(containerExecCommand)
-       vpp.Suite().assertNil(err)
+       vpp.getSuite().assertNil(err)
 
        return string(output)
 }
 
+func (vpp *VppInstance) GetSessionStat(stat string) int {
+       o := vpp.vppctl("show session stats")
+       vpp.getSuite().log(o)
+       for _, line := range strings.Split(o, "\n") {
+               if strings.Contains(line, stat) {
+                       tokens := strings.Split(strings.TrimSpace(line), " ")
+                       val, err := strconv.Atoi(tokens[0])
+                       if err != nil {
+                               vpp.getSuite().FailNow("failed to parse stat value %s", err)
+                               return 0
+                       }
+                       return val
+               }
+       }
+       return 0
+}
+
 func (vpp *VppInstance) waitForApp(appName string, timeout int) {
        for i := 0; i < timeout; i++ {
                o := vpp.vppctl("show app")
@@ -168,8 +220,7 @@ func (vpp *VppInstance) waitForApp(appName string, timeout int) {
                }
                time.Sleep(1 * time.Second)
        }
-       vpp.Suite().assertNil(1, "Timeout while waiting for app '%s'", appName)
-       return
+       vpp.getSuite().assertNil(1, "Timeout while waiting for app '%s'", appName)
 }
 
 func (vpp *VppInstance) createAfPacket(
@@ -179,20 +230,20 @@ func (vpp *VppInstance) createAfPacket(
                UseRandomHwAddr: true,
                HostIfName:      veth.Name(),
        }
-       if veth.HwAddress() != (MacAddress{}) {
+       if veth.hwAddress != (MacAddress{}) {
                createReq.UseRandomHwAddr = false
-               createReq.HwAddr = veth.HwAddress()
+               createReq.HwAddr = veth.hwAddress
        }
        createReply := &af_packet.AfPacketCreateV2Reply{}
 
        if err := vpp.apiChannel.SendRequest(createReq).ReceiveReply(createReply); err != nil {
                return 0, err
        }
-       veth.SetIndex(createReply.SwIfIndex)
+       veth.index = createReply.SwIfIndex
 
        // Set to up
        upReq := &interfaces.SwInterfaceSetFlags{
-               SwIfIndex: veth.Index(),
+               SwIfIndex: veth.index,
                Flags:     interface_types.IF_STATUS_API_FLAG_ADMIN_UP,
        }
        upReply := &interfaces.SwInterfaceSetFlagsReply{}
@@ -202,19 +253,19 @@ func (vpp *VppInstance) createAfPacket(
        }
 
        // Add address
-       if veth.AddressWithPrefix() == (AddressWithPrefix{}) {
+       if veth.addressWithPrefix() == (AddressWithPrefix{}) {
                var err error
                var ip4Address string
-               if ip4Address, err = veth.addresser.NewIp4Address(veth.Peer().networkNumber); err == nil {
-                       veth.SetAddress(ip4Address)
+               if ip4Address, err = veth.ip4AddrAllocator.NewIp4InterfaceAddress(veth.peer.networkNumber); err == nil {
+                       veth.ip4Address = ip4Address
                } else {
                        return 0, err
                }
        }
        addressReq := &interfaces.SwInterfaceAddDelAddress{
                IsAdd:     true,
-               SwIfIndex: veth.Index(),
-               Prefix:    veth.AddressWithPrefix(),
+               SwIfIndex: veth.index,
+               Prefix:    veth.addressWithPrefix(),
        }
        addressReply := &interfaces.SwInterfaceAddDelAddressReply{}
 
@@ -222,7 +273,7 @@ func (vpp *VppInstance) createAfPacket(
                return 0, err
        }
 
-       return veth.Index(), nil
+       return veth.index, nil
 }
 
 func (vpp *VppInstance) addAppNamespace(
@@ -266,7 +317,7 @@ func (vpp *VppInstance) createTap(
                HostIfNameSet:    true,
                HostIfName:       tap.Name(),
                HostIP4PrefixSet: true,
-               HostIP4Prefix:    tap.IP4AddressWithPrefix(),
+               HostIP4Prefix:    tap.ip4AddressWithPrefix(),
        }
        createTapReply := &tapv2.TapCreateV2Reply{}
 
@@ -279,7 +330,7 @@ func (vpp *VppInstance) createTap(
        addAddressReq := &interfaces.SwInterfaceAddDelAddress{
                IsAdd:     true,
                SwIfIndex: createTapReply.SwIfIndex,
-               Prefix:    tap.Peer().AddressWithPrefix(),
+               Prefix:    tap.peer.addressWithPrefix(),
        }
        addAddressReply := &interfaces.SwInterfaceAddDelAddressReply{}
 
@@ -303,10 +354,10 @@ func (vpp *VppInstance) createTap(
 
 func (vpp *VppInstance) saveLogs() {
        logTarget := vpp.container.getLogDirPath() + "vppinstance-" + vpp.container.name + ".log"
-       logSource := vpp.container.GetHostWorkDir() + defaultLogFilePath
+       logSource := vpp.container.getHostWorkDir() + defaultLogFilePath
        cmd := exec.Command("cp", logSource, logTarget)
-       vpp.Suite().T().Helper()
-       vpp.Suite().log(cmd.String())
+       vpp.getSuite().T().Helper()
+       vpp.getSuite().log(cmd.String())
        cmd.Run()
 }
 
@@ -314,3 +365,25 @@ func (vpp *VppInstance) disconnect() {
        vpp.connection.Disconnect()
        vpp.apiChannel.Close()
 }
+
+func (vpp *VppInstance) generateCpuConfig() string {
+       var c Stanza
+       var s string
+       if len(vpp.cpus) < 1 {
+               return ""
+       }
+       c.newStanza("cpu").
+               append(fmt.Sprintf("main-core %d", vpp.cpus[0]))
+       workers := vpp.cpus[1:]
+
+       if len(workers) > 0 {
+               for i := 0; i < len(workers); i++ {
+                       if i != 0 {
+                               s = s + ", "
+                       }
+                       s = s + fmt.Sprintf("%d", workers[i])
+               }
+               c.append(fmt.Sprintf("corelist-workers %s", s))
+       }
+       return c.close().toString()
+}