1 files changed, 124 insertions, 0 deletions
diff --git a/extras/hs-test/vppinstance.go b/extras/hs-test/vppinstance.go
new file mode 100644
index 00000000000..c6d3935cc60
--- /dev/null
+++ b/extras/hs-test/vppinstance.go
@@ -0,0 +1,124 @@
+package main
+
+import (
+	"fmt"
+	"encoding/json"
+	"github.com/edwarnicke/exechelper"
+)
+
+const vppConfigTemplate = `unix {
+  nodaemon
+  log %[1]s/var/log/vpp/vpp.log
+  full-coredump
+  cli-listen %[1]s%[2]s
+  runtime-dir %[1]s/var/run
+  gid vpp
+}
+
+api-trace {
+  on
+}
+
+api-segment {
+  gid vpp
+}
+
+socksvr {
+  socket-name %[1]s/var/run/vpp/api.sock
+}
+
+statseg {
+  socket-name %[1]s/var/run/vpp/stats.sock
+}
+
+plugins {
+  plugin unittest_plugin.so { enable }
+  plugin dpdk_plugin.so { disable }
+  plugin crypto_aesni_plugin.so { enable }
+  plugin quic_plugin.so { enable }
+}
+
+`
+
+type VppInstance struct {
+	container *Container
+	config VppConfig
+	actionFuncName string
+}
+
+type VppConfig struct {
+	Variant string
+	CliSocketFilePath string
+}
+
+func (vc *VppConfig) getTemplate() string {
+	return fmt.Sprintf(vppConfigTemplate, "%[1]s", vc.CliSocketFilePath)
+}
+
+func (vpp *VppInstance) set2VethsServer() {
+	vpp.actionFuncName = "Configure2Veths"
+	vpp.config.Variant = "srv"
+}
+
+func (vpp *VppInstance) set2VethsClient() {
+	vpp.actionFuncName = "Configure2Veths"
+	vpp.config.Variant = "cln"
+}
+
+func (vpp *VppInstance) setCliSocket(filePath string) {
+	vpp.config.CliSocketFilePath = filePath
+}
+
+func (vpp *VppInstance) getCliSocket() string {
+	return fmt.Sprintf("/tmp/%s/%s", vpp.actionFuncName, vpp.config.CliSocketFilePath)
+}
+
+func (vpp *VppInstance) start() error {
+	if vpp.config.Variant == "" {
+		return fmt.Errorf("vpp start failed: variant must not be blank")
+	}
+	if vpp.actionFuncName == "" {
+		return fmt.Errorf("vpp start failed: action function name must not be blank")
+	}
+
+	serializedConfig, err := json.Marshal(vpp.config)
+	if err != nil {
+		return fmt.Errorf("vpp start failed: serializing configuration failed: %s", err)
+	}
+	args := fmt.Sprintf("%s '%s'", vpp.actionFuncName, string(serializedConfig))
+	_, err = hstExec(args, vpp.container.name)
+	if err != nil {
+		return fmt.Errorf("vpp start failed: %s", err)
+	}
+
+	return nil
+}
+
+func (vpp *VppInstance) vppctl(command string) (string, error) {
+	dockerExecCommand := fmt.Sprintf("docker exec --detach=false %[1]s vppctl -s %[2]s %[3]s",
+		vpp.container.name, vpp.getCliSocket(), command)
+	output, err := exechelper.CombinedOutput(dockerExecCommand)
+	if err != nil {
+		return "", fmt.Errorf("vppctl failed: %s", err)
+	}
+
+	return string(output), nil
+}
+
+func NewVppInstance(c *Container) *VppInstance {
+	vpp := new(VppInstance)
+	vpp.container = c
+	return vpp
+}
+
+func DeserializeVppConfig(input string) (VppConfig, error) {
+	var vppConfig VppConfig
+	err := json.Unmarshal([]byte(input), &vppConfig)
+	if err != nil {
+		// Since input is not a  valid JSON it is going be used as variant value
+		// for compatibility reasons
+		vppConfig.Variant = input
+		vppConfig.CliSocketFilePath = "/var/run/vpp/cli.sock"
+	}
+	return vppConfig, nil
+}