github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-09-08 14:01:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

WIP

Browse Source
This commit is contained in:
RamiBerm
2021-05-24 12:17:26 +03:00
parent c1409251e2
commit 2f8f253743
4 changed files with 20 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/main.go
View File

@@ -107,11 +107,7 @@ func pipeChannelToSocket(connection *websocket.Conn, messageDataChannel <-chan *
}
for messageData := range messageDataChannel {
socketMessage := shared.MizuSocketMessage{
MessageType: shared.TAPPED_MESSAGE_TYPE,
Data: messageData,
}
marshaledData, err := json.Marshal(socketMessage)
marshaledData, err := json.Marshal(messageData)
if err != nil {
fmt.Printf("error converting message to json %s, (%v,%+v)\n", err, err, err)
continue

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

@@ -55,18 +55,28 @@ 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)
if err != nil {
fmt.Printf("Could not unmarshal websocket message %v\n", err)
} else if socketMessage.MessageType == shared.TAPPED_MESSAGE_TYPE {
var tapOutput tap.OutputChannelItem
err := mapstructure.Decode(socketMessage.Data, &tapOutput)
if err != nil {
fmt.Printf("Could not decode map of message type %s %v", shared.TAPPED_MESSAGE_TYPE, err)
} else {
h.SocketHarOutChannel <- &tapOutput
}
} else if socketMessage.MessageType == shared.TAPPING_STATUS_MESSAGE_TYPE {
var tapStatus shared.TapStatus
err := mapstructure.Decode(socketMessage.Data, &tapStatus)