github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-06-25 07:45:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Determine the endianness on runtime if it's possible in Go (default little-endian) #run_acceptance_tests

Browse Source
This commit is contained in:
M. Mert Yildiran 2022-06-20 14:13:45 +03:00
parent 8ea163bcef
commit a2c83c6040
No known key found for this signature in database
GPG Key ID: D42ADB236521BF7A
4 changed files with 34 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tap/passive_tapper.go
View File

@ -9,6 +9,7 @@
package tap
import (
"encoding/binary"
"encoding/json"
"flag"
"fmt"
@ -17,6 +18,7 @@ import (
"strconv"
"strings"
"time"
"unsafe"
"github.com/shirou/gopsutil/cpu"
"github.com/struCoder/pidusage"
@ -298,8 +300,10 @@ func startTlsTapper(extension *api.Extension, outputItems chan *api.OutputChanne
OutputChannel: outputItems,
}
go tls.PollForLogging()
go tls.Poll(emitter, options, streamsMap)
byteOrder := getByteOrder()
go tls.PollForLogging(byteOrder)
go tls.Poll(byteOrder, emitter, options, streamsMap)
return &tls
}
@ -319,3 +323,20 @@ func buildIgnoredPortsList(ignoredPorts string) []uint16 {
return result
}
func getByteOrder() (byteOrder binary.ByteOrder) {
buf := [2]byte{}
*(*uint16)(unsafe.Pointer(&buf[0])) = uint16(0xABCD)
switch buf {
case [2]byte{0xCD, 0xAB}:
byteOrder = binary.LittleEndian
case [2]byte{0xAB, 0xCD}:
byteOrder = binary.BigEndian
default:
logger.Log.Warning("Could not determine native endianness. Defaulting to little-endian")
byteOrder = binary.LittleEndian
}
return
}

4
tap/tlstapper/bpf_logger.go
View File

@ -52,7 +52,7 @@ func (p *bpfLogger) close() error {
return p.logReader.Close()
}
func (p *bpfLogger) poll() {
func (p *bpfLogger) poll(byteOrder binary.ByteOrder) {
logger.Log.Infof("Start polling for bpf logs")
for {
@ -76,7 +76,7 @@ func (p *bpfLogger) poll() {
var log logMessage
if err := binary.Read(buffer, binary.LittleEndian, &log); err != nil {
if err := binary.Read(buffer, byteOrder, &log); err != nil {
LogError(errors.Errorf("Error parsing log %v", err))
continue
}

8
tap/tlstapper/tls_poller.go
View File

@ -75,11 +75,11 @@ func (p *tlsPoller) close() error {
return p.chunksReader.Close()
}
func (p *tlsPoller) poll(emitter api.Emitter, options *api.TrafficFilteringOptions, streamsMap api.TcpStreamMap) {
func (p *tlsPoller) poll(byteOrder binary.ByteOrder, emitter api.Emitter, options *api.TrafficFilteringOptions, streamsMap api.TcpStreamMap) {
// tlsTapperTlsChunk is generated by bpf2go.
chunks := make(chan *tlsTapperTlsChunk)
go p.pollChunksPerfBuffer(chunks)
go p.pollChunksPerfBuffer(byteOrder, chunks)
for {
select {
@ -97,7 +97,7 @@ func (p *tlsPoller) poll(emitter api.Emitter, options *api.TrafficFilteringOptio
}
}
func (p *tlsPoller) pollChunksPerfBuffer(chunks chan<- *tlsTapperTlsChunk) {
func (p *tlsPoller) pollChunksPerfBuffer(byteOrder binary.ByteOrder, chunks chan<- *tlsTapperTlsChunk) {
logger.Log.Infof("Start polling for tls events")
for {
@ -123,7 +123,7 @@ func (p *tlsPoller) pollChunksPerfBuffer(chunks chan<- *tlsTapperTlsChunk) {
var chunk tlsTapperTlsChunk
if err := binary.Read(buffer, binary.LittleEndian, &chunk); err != nil {
if err := binary.Read(buffer, byteOrder, &chunk); err != nil {
LogError(errors.Errorf("Error parsing chunk %v", err))
continue
}

9
tap/tlstapper/tls_tapper.go
View File

@ -1,6 +1,7 @@
package tlstapper
import (
"encoding/binary"
"strconv"
"sync"
@ -58,12 +59,12 @@ func (t *TlsTapper) Init(chunksBufferSize int, logBufferSize int, procfs string,
return t.poller.init(&t.bpfObjects, chunksBufferSize)
}
func (t *TlsTapper) Poll(emitter api.Emitter, options *api.TrafficFilteringOptions, streamsMap api.TcpStreamMap) {
t.poller.poll(emitter, options, streamsMap)
func (t *TlsTapper) Poll(byteOrder binary.ByteOrder, emitter api.Emitter, options *api.TrafficFilteringOptions, streamsMap api.TcpStreamMap) {
t.poller.poll(byteOrder, emitter, options, streamsMap)
}
func (t *TlsTapper) PollForLogging() {
t.bpfLogger.poll()
func (t *TlsTapper) PollForLogging(byteOrder binary.ByteOrder) {
t.bpfLogger.poll(byteOrder)
}
func (t *TlsTapper) GlobalSsllibTap(sslLibrary string) error {