mirror of
https://github.com/kubeshark/kubeshark.git
synced 2025-04-28 03:52:23 +00:00
108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/kubeshark/kubeshark/config"
|
|
"github.com/kubeshark/kubeshark/config/configStructs"
|
|
"github.com/kubeshark/kubeshark/internal/connect"
|
|
"github.com/kubeshark/kubeshark/kubernetes"
|
|
"github.com/kubeshark/kubeshark/misc"
|
|
"github.com/kubeshark/kubeshark/utils"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
func runProxy(block bool, noBrowser bool) {
|
|
kubernetesProvider, err := getKubernetesProviderForCli(false, false)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
exists, err := kubernetesProvider.DoesServiceExist(ctx, config.Config.Tap.Release.Namespace, kubernetes.FrontServiceName)
|
|
if err != nil {
|
|
log.Error().
|
|
Str("service", kubernetes.FrontServiceName).
|
|
Err(err).
|
|
Msg("Failed to found service!")
|
|
cancel()
|
|
return
|
|
}
|
|
|
|
if !exists {
|
|
log.Error().
|
|
Str("service", kubernetes.FrontServiceName).
|
|
Str("command", fmt.Sprintf("%s %s", misc.Program, tapCmd.Use)).
|
|
Msg("Service not found! You should run the command first:")
|
|
cancel()
|
|
return
|
|
}
|
|
|
|
exists, err = kubernetesProvider.DoesServiceExist(ctx, config.Config.Tap.Release.Namespace, kubernetes.HubServiceName)
|
|
if err != nil {
|
|
log.Error().
|
|
Str("service", kubernetes.HubServiceName).
|
|
Err(err).
|
|
Msg("Failed to found service!")
|
|
cancel()
|
|
return
|
|
}
|
|
|
|
if !exists {
|
|
log.Error().
|
|
Str("service", kubernetes.HubServiceName).
|
|
Str("command", fmt.Sprintf("%s %s", misc.Program, tapCmd.Use)).
|
|
Msg("Service not found! You should run the command first:")
|
|
cancel()
|
|
return
|
|
}
|
|
|
|
var establishedProxy bool
|
|
|
|
frontUrl := kubernetes.GetProxyOnPort(config.Config.Tap.Proxy.Front.Port)
|
|
response, err := http.Get(fmt.Sprintf("%s/", frontUrl))
|
|
if err == nil && response.StatusCode == 200 {
|
|
log.Info().
|
|
Str("service", kubernetes.FrontServiceName).
|
|
Int("port", int(config.Config.Tap.Proxy.Front.Port)).
|
|
Msg("Found a running service.")
|
|
|
|
okToOpen("Kubeshark", frontUrl, noBrowser)
|
|
} else {
|
|
startProxyReportErrorIfAny(
|
|
kubernetesProvider,
|
|
ctx,
|
|
kubernetes.FrontServiceName,
|
|
kubernetes.FrontPodName,
|
|
configStructs.ProxyFrontPortLabel,
|
|
config.Config.Tap.Proxy.Front.Port,
|
|
configStructs.ContainerPort,
|
|
"",
|
|
)
|
|
connector := connect.NewConnector(frontUrl, connect.DefaultRetries, connect.DefaultTimeout)
|
|
if err := connector.TestConnection(""); err != nil {
|
|
log.Error().Msg(fmt.Sprintf(utils.Red, "Couldn't connect to Front."))
|
|
return
|
|
}
|
|
|
|
establishedProxy = true
|
|
okToOpen("Kubeshark", frontUrl, noBrowser)
|
|
}
|
|
if establishedProxy && block {
|
|
utils.WaitForTermination(ctx, cancel)
|
|
}
|
|
|
|
}
|
|
|
|
func okToOpen(name string, url string, noBrowser bool) {
|
|
log.Info().Str("url", url).Msg(fmt.Sprintf(utils.Green, fmt.Sprintf("%s is available at:", name)))
|
|
|
|
if !config.Config.HeadlessMode && !noBrowser {
|
|
utils.OpenBrowser(url)
|
|
}
|
|
}
|