mirror of
https://github.com/kubeshark/kubeshark.git
synced 2025-08-25 11:59:35 +00:00
106 lines
2.6 KiB
Go
106 lines
2.6 KiB
Go
package controllers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/up9inc/mizu/shared"
|
|
"github.com/up9inc/mizu/shared/logger"
|
|
"mizuserver/pkg/api"
|
|
"mizuserver/pkg/holder"
|
|
"mizuserver/pkg/providers"
|
|
"mizuserver/pkg/providers/tappedPods"
|
|
"mizuserver/pkg/providers/tappers"
|
|
"mizuserver/pkg/up9"
|
|
"mizuserver/pkg/validation"
|
|
"net/http"
|
|
)
|
|
|
|
func HealthCheck(c *gin.Context) {
|
|
tappersStatus := make([]*shared.TapperStatus, 0)
|
|
for _, value := range tappers.GetStatus() {
|
|
tappersStatus = append(tappersStatus, value)
|
|
}
|
|
|
|
response := shared.HealthResponse{
|
|
TappedPods: tappedPods.Get(),
|
|
ConnectedTappersCount: tappers.GetConnectedCount(),
|
|
TappersStatus: tappersStatus,
|
|
}
|
|
c.JSON(http.StatusOK, response)
|
|
}
|
|
|
|
func PostTappedPods(c *gin.Context) {
|
|
var requestTappedPods []*shared.PodInfo
|
|
if err := c.Bind(&requestTappedPods); err != nil {
|
|
c.JSON(http.StatusBadRequest, err)
|
|
return
|
|
}
|
|
|
|
logger.Log.Infof("[Status] POST request: %d tapped pods", len(requestTappedPods))
|
|
tappedPods.Set(requestTappedPods)
|
|
broadcastTappedPodsStatus()
|
|
}
|
|
|
|
func broadcastTappedPodsStatus() {
|
|
tappedPodsStatus := tappedPods.GetTappedPodsStatus()
|
|
|
|
message := shared.CreateWebSocketStatusMessage(tappedPodsStatus)
|
|
if jsonBytes, err := json.Marshal(message); err != nil {
|
|
logger.Log.Errorf("Could not Marshal message %v", err)
|
|
} else {
|
|
api.BroadcastToBrowserClients(jsonBytes)
|
|
}
|
|
}
|
|
|
|
func PostTapperStatus(c *gin.Context) {
|
|
tapperStatus := &shared.TapperStatus{}
|
|
if err := c.Bind(tapperStatus); err != nil {
|
|
c.JSON(http.StatusBadRequest, err)
|
|
return
|
|
}
|
|
|
|
if err := validation.Validate(tapperStatus); err != nil {
|
|
c.JSON(http.StatusBadRequest, err)
|
|
return
|
|
}
|
|
|
|
logger.Log.Infof("[Status] POST request, tapper status: %v", tapperStatus)
|
|
tappers.SetStatus(tapperStatus)
|
|
broadcastTappedPodsStatus()
|
|
}
|
|
|
|
func GetConnectedTappersCount(c *gin.Context) {
|
|
c.JSON(http.StatusOK, tappers.GetConnectedCount())
|
|
}
|
|
|
|
func GetAuthStatus(c *gin.Context) {
|
|
authStatus, err := providers.GetAuthStatus()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, authStatus)
|
|
}
|
|
|
|
func GetTappingStatus(c *gin.Context) {
|
|
tappedPodsStatus := tappedPods.GetTappedPodsStatus()
|
|
c.JSON(http.StatusOK, tappedPodsStatus)
|
|
}
|
|
|
|
func AnalyzeInformation(c *gin.Context) {
|
|
c.JSON(http.StatusOK, up9.GetAnalyzeInfo())
|
|
}
|
|
|
|
func GetGeneralStats(c *gin.Context) {
|
|
c.JSON(http.StatusOK, providers.GetGeneralStats())
|
|
}
|
|
|
|
func GetRecentTLSLinks(c *gin.Context) {
|
|
c.JSON(http.StatusOK, providers.GetAllRecentTLSAddresses())
|
|
}
|
|
|
|
func GetCurrentResolvingInformation(c *gin.Context) {
|
|
c.JSON(http.StatusOK, holder.GetResolver().GetMap())
|
|
}
|