github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2026-05-21 08:33:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update main.go, socket_server_handlers.go, and 4 more files...

Browse Source
This commit is contained in:
RamiBerm
2021-05-24 17:06:03 +03:00
parent 2f8f253743
commit fbeb7fe9cd
6 changed files with 41 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
api/pkg/api/socket_server_handlers.go
View File

@@ -4,9 +4,9 @@ import (
"encoding/json"
"fmt"
"github.com/antoniodipinto/ikisocket"
"github.com/mitchellh/mapstructure"
"github.com/up9inc/mizu/shared"
"mizuserver/pkg/controllers"
"mizuserver/pkg/models"
"mizuserver/pkg/routes"
"mizuserver/pkg/tap"
)
@@ -55,39 +55,31 @@ func (h *RoutesEventHandlers) WebSocketError(ep *ikisocket.EventPayload) {
}
func (h *RoutesEventHandlers) WebSocketMessage(ep *ikisocket.EventPayload) {
if ep.Kws.GetAttribute("is_tapper") == true && h.SocketHarOutChannel != nil {
h.handleTapperMessage(ep)
} else {
h.handleControlMessage(ep)
}
}
func (h *RoutesEventHandlers) handleTapperMessage(ep *ikisocket.EventPayload) {
var tapOutput tap.OutputChannelItem
err := json.Unmarshal(ep.Data, &tapOutput)
if err != nil {
fmt.Printf("Could not unmarshal message received from tapper websocket %v", err)
} else {
h.SocketHarOutChannel <- &tapOutput
}
}
func (h *RoutesEventHandlers) handleControlMessage(ep *ikisocket.EventPayload) {
var socketMessage shared.MizuSocketMessage
err := json.Unmarshal(ep.Data, &socketMessage)
var socketMessageBase shared.WebSocketMessageMetadata
err := json.Unmarshal(ep.Data, &socketMessageBase)
if err != nil {
fmt.Printf("Could not unmarshal websocket message %v\n", err)
} else if socketMessage.MessageType == shared.TAPPING_STATUS_MESSAGE_TYPE {
var tapStatus shared.TapStatus
err := mapstructure.Decode(socketMessage.Data, &tapStatus)
} else if socketMessageBase.MessageType == shared.WebSocketMessageTypeTappedEntry {
var tappedEntryMessage models.WebSocketTappedEntryMessage
err := json.Unmarshal(ep.Data, &tappedEntryMessage)
if err != nil {
fmt.Printf("Could not decode map of message type %s %v", shared.TAPPING_STATUS_MESSAGE_TYPE, err)
fmt.Printf("Could not unmarshall message of message type %s %v", socketMessageBase.MessageType, err)
} else {
controllers.TapStatus = tapStatus
h.SocketHarOutChannel <- tappedEntryMessage.Data
}
} else if socketMessageBase.MessageType == shared.WebSocketMessageTypeUpdateStatus {
var statusMessage shared.WebSocketStatusMessage
err := json.Unmarshal(ep.Data, &statusMessage)
if err != nil {
fmt.Printf("Could not unmarshall message of message type %s %v", socketMessageBase.MessageType, err)
} else {
controllers.TapStatus = statusMessage.TappingStatus
broadcastToBrowserClients(ep.Data)
}
}
}
func removeSocketUUIDFromBrowserSlice(uuidToRemove string) {
newUUIDSlice := make([]string, 0, len(browserClientSocketUUIDs))
for _, uuid := range browserClientSocketUUIDs {