TRA-3186 partial cli integration and api+passive docker file

TRA-3186 partial cli integration and api+passive docker file

Dockerfile

@@ -0,0 +1,38 @@
+FROM golang:1.16-alpine AS builder
+# Set necessary environment variables needed for our image.
+ENV CGO_ENABLED=1 GOOS=linux GOARCH=amd64
+
+RUN apk add libpcap-dev gcc g++ make
+
+# Move to tapper working directory (/tap-build).
+WORKDIR /tap-build
+
+COPY tap/go.mod tap/go.sum ./
+RUN go mod download
+# Copy and build tapper code
+COPY tap/src ./
+RUN go build -ldflags="-s -w" -o passivetapper .
+
+# Move to api working directory (/api-build).
+WORKDIR ../api-build
+
+COPY api/go.mod api/go.sum ./
+RUN go mod download
+# Copy and build api code
+COPY api .
+RUN go build -ldflags="-s -w" -o apiserver .
+
+FROM alpine:3.13.5
+
+RUN apk add bash libpcap-dev
+
+WORKDIR /app
+
+# Copy binary and config files from /build to root folder of scratch container.
+COPY --from=builder ["/api-build/apiserver", "."]
+COPY --from=builder ["/tap-build/passivetapper", "."]
+
+COPY api/scripts/multi-runner.sh ./
+
+# this script runs both apiserver and passivetapper and exits either if one of them exits, preventing a scenario where the container runs without one process
+CMD "./multi-runner.sh"

api/Dockerfile

@@ -1,29 +0,0 @@
-FROM golang:1.16 AS builder
-
-# Move to working directory (/site).
-WORKDIR /build
-
-# Copy and download dependency using go mod.
-COPY go.mod go.sum ./
-RUN go mod download
-
-# Copy the code into the container.
-COPY . .
-
-# Set necessary environmet variables needed for our image and site the API server.
-ENV GOOS=linux GOARCH=amd64
-RUN go build -ldflags="-s -w" -o apiserver .
-
-FROM golang:1.16
-
-# Copy binary and config files from /site to root folder of scratch container.
-COPY --from=builder ["/build/apiserver", "/"]
-
-# Export necessary port.
-EXPOSE 8899
-
-COPY site /go/site
-COPY entries.db /go/entries.db
-
-# Command to run when starting the container.
-ENTRYPOINT ["/apiserver"]

api/scripts/multi-runner.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+
+# this script runs both executables and exits everything if one fails
+./apiserver &
+./passivetapper -i eth0 &
+wait -n
+pkill -P $$

cli/cmd/root.go

@@ -36,6 +36,8 @@ func init() {
 	rootCmd.Flags().StringVarP(&config.Configuration.Namespace, "namespace", "n", "", "Namespace selector")
 	rootCmd.Flags().BoolVarP(&config.Configuration.AllNamespaces, "all-namespaces", "A", false, "Select all namespaces")
 	rootCmd.Flags().StringVarP(&config.Configuration.KubeConfigPath, "kubeconfig", "k", "", "Path to kubeconfig file")
+	rootCmd.Flags().StringVarP(&config.Configuration.MizuImage, "mizu-image", "", "gcr.io/up9-docker-hub/mizu/develop/v1", "Custom image for mizu collector")
+	rootCmd.Flags().Uint16VarP(&config.Configuration.MizuPodPort, "mizu-port", "", 8899, "Port which mizu cli will attempt to forward from the mizu collector pod")
 }
 
 // Execute adds all child commands to the root command and sets flags appropriately.

cli/config/config.go

@@ -8,6 +8,8 @@ type Options struct {
 	Namespace        string
 	AllNamespaces    bool
 	KubeConfigPath   string
+	MizuImage        string
+	MizuPodPort      uint16
 }
 
 var Configuration = &Options{}

cli/mizu/mizuRunner.go

@@ -12,10 +12,6 @@ import (
 	"time"
 )
 
-var (
-	isPortForwarded = false
-)
-
 func Run(podRegex *regexp.Regexp) {
 	kubernetesProvider := kubernetes.NewProvider(config.Configuration.KubeConfigPath, config.Configuration.Namespace)
 	ctx, cancel := context.WithCancel(context.Background())
@@ -58,9 +54,7 @@ func watchPodsForTapping(ctx context.Context, kubernetesProvider *kubernetes.Pro
 }
 
 func createPodAndPortForward(ctx context.Context, kubernetesProvider *kubernetes.Provider, cancel context.CancelFunc, podName string) {
-	podImage := "kennethreitz/httpbin:latest"
+	pod, err := kubernetesProvider.CreatePod(ctx, podName, config.Configuration.MizuImage)
-
-	pod, err := kubernetesProvider.CreatePod(ctx, podName, podImage)
 	if err != nil {
 		fmt.Printf("error creating pod %s", err)
 		cancel()
@@ -82,7 +76,7 @@ func createPodAndPortForward(ctx context.Context, kubernetesProvider *kubernetes
 			if modifiedPod.Status.Phase == "Running" && !isPodReady {
 				isPodReady = true
 				var err error
-				portForward, err = kubernetes.NewPortForward(kubernetesProvider, kubernetesProvider.Namespace, podName, config.Configuration.DashboardPort, 80, cancel)
+				portForward, err = kubernetes.NewPortForward(kubernetesProvider, kubernetesProvider.Namespace, podName, config.Configuration.DashboardPort, config.Configuration.MizuPodPort, cancel)
 				if !config.Configuration.NoDashboard {
 					fmt.Printf("Dashboard is now available at http://localhost:%d\n", config.Configuration.DashboardPort)
 				}
@@ -92,7 +86,7 @@ func createPodAndPortForward(ctx context.Context, kubernetesProvider *kubernetes
 				}
 			}
 
-		case <- time.After(10 * time.Second):
+		case <- time.After(25 * time.Second):
 			if !isPodReady {
 				fmt.Printf("error: %s pod was not ready in time", podName)
 				cancel()