package main import ( . "fd.io/hs-test/infra" "fmt" "github.com/edwarnicke/exechelper" . "github.com/onsi/ginkgo/v2" "os" ) func init() { RegisterNsTests(VppProxyHttpTcpTest, VppProxyHttpTlsTest, EnvoyProxyHttpTcpTest) } func testProxyHttpTcp(s *NsSuite, proto string) error { var outputFile string = s.ProcessIndex + "test" + s.Ppid + ".data" var srcFilePpid string = s.ProcessIndex + "httpTestFile" + s.Ppid const srcFileNoPpid = "httpTestFile" const fileSize string = "10M" stopServer := make(chan struct{}, 1) serverRunning := make(chan struct{}, 1) serverNetns := s.GetNetNamespaceByName("srv") clientNetns := s.GetNetNamespaceByName("cln") // create test file err := exechelper.Run(fmt.Sprintf("ip netns exec %s truncate -s %s %s", serverNetns, fileSize, srcFilePpid)) s.AssertNil(err, "failed to run truncate command: "+fmt.Sprint(err)) defer func() { os.Remove(srcFilePpid) }() s.Log("test file created...") go func() { defer GinkgoRecover() s.StartHttpServer(serverRunning, stopServer, ":666", serverNetns) }() // TODO better error handling and recovery <-serverRunning defer func(chan struct{}) { stopServer <- struct{}{} }(stopServer) s.Log("http server started...") clientVeth := s.GetInterfaceByName(ClientInterface) c := fmt.Sprintf("ip netns exec %s wget --no-proxy --retry-connrefused"+ " --retry-on-http-error=503 --tries=10 -O %s ", clientNetns, outputFile) if proto == "tls" { c += " --secure-protocol=TLSv1_3 --no-check-certificate https://" } c += fmt.Sprintf("%s:555/%s", clientVeth.Ip4AddressString(), srcFileNoPpid) s.Log(c) _, err = exechelper.CombinedOutput(c) defer func() { os.Remove(outputFile) }() s.AssertNil(err, "failed to run wget: '%s', cmd: %s", err, c) stopServer <- struct{}{} s.AssertNil(AssertFileSize(outputFile, srcFilePpid)) return nil } func configureVppProxy(s *NsSuite, proto string) { serverVeth := s.GetInterfaceByName(ServerInterface) clientVeth := s.GetInterfaceByName(ClientInterface) testVppProxy := s.GetContainerByName("vpp").VppInstance output := testVppProxy.Vppctl( "test proxy server server-uri %s://%s/555 client-uri tcp://%s/666", proto, clientVeth.Ip4AddressString(), serverVeth.Peer.Ip4AddressString(), ) s.Log("proxy configured: " + output) } func VppProxyHttpTcpTest(s *NsSuite) { proto := "tcp" configureVppProxy(s, proto) err := testProxyHttpTcp(s, proto) s.AssertNil(err, fmt.Sprint(err)) } func VppProxyHttpTlsTest(s *NsSuite) { proto := "tls" configureVppProxy(s, proto) err := testProxyHttpTcp(s, proto) s.AssertNil(err, fmt.Sprint(err)) } func configureEnvoyProxy(s *NsSuite) { envoyContainer := s.GetContainerByName("envoy") err := envoyContainer.Create() s.AssertNil(err, "Error creating envoy container: %s", err) serverVeth := s.GetInterfaceByName(ServerInterface) address := struct { Server string }{ Server: serverVeth.Peer.Ip4AddressString(), } envoyContainer.CreateConfig( "/etc/envoy/envoy.yaml", "resources/envoy/proxy.yaml", address, ) s.AssertNil(envoyContainer.Start()) } func EnvoyProxyHttpTcpTest(s *NsSuite) { configureEnvoyProxy(s) err := testProxyHttpTcp(s, "tcp") s.AssertNil(err, fmt.Sprint(err)) }