| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743 |
- package ebpftracer
- import (
- "bytes"
- debugelf "debug/elf"
- "encoding/binary"
- "encoding/hex"
- "errors"
- "fmt"
- "os"
- "runtime"
- "strconv"
- "strings"
- "time"
- "github.com/cilium/ebpf"
- "github.com/cilium/ebpf/link"
- "github.com/cilium/ebpf/perf"
- "github.com/coroot/coroot-node-agent/common"
- "github.com/coroot/coroot-node-agent/ebpftracer/l7"
- "github.com/coroot/coroot-node-agent/ebpftracer/tracer"
- "github.com/coroot/coroot-node-agent/proc"
- "github.com/coroot/coroot-node-agent/utils"
- "golang.org/x/mod/semver"
- "golang.org/x/sys/unix"
- "inet.af/netaddr"
- "k8s.io/klog/v2"
- )
- /*
- #define TASK_COMM_LEN 16
- #define BURST_DATA_BUF_SIZE 8192 // For brust send buffer
- #include <linux/types.h>
- struct __tuple_t {
- __u8 daddr[16];
- __u8 rcv_saddr[16];
- __u8 addr_len;
- __u8 l4_protocol;
- __u16 dport;
- __u16 num;
- };
- struct __socket_data {
- __u32 pid;
- __u32 tgid;
- __u64 coroutine_id;
- __u8 source;
- __u8 comm[TASK_COMM_LEN];
- __u64 socket_id;
- struct __tuple_t tuple;
- __u32 extra_data;
- __u32 extra_data_count;
- __u32 tcp_seq;
- __u64 thread_trace_id;
- __u64 timestamp;
- __u8 direction: 1;
- __u8 msg_type: 7;
- __u64 syscall_len;
- __u64 data_seq;
- __u16 data_type;
- __u16 data_len;
- char data[BURST_DATA_BUF_SIZE];
- } __attribute__((packed));
- struct __socket_data_buffer {
- __u32 events_num;
- __u32 len;
- char data[32760];
- };
- */
- import "C"
- type SocketData C.struct___socket_data
- type SocketDataBuffer C.struct___socket_data_buffer
- const MaxPayloadSize = 1024
- type EventType uint32
- type EventReason uint32
- const (
- EventTypeProcessStart EventType = 1
- EventTypeProcessExit EventType = 2
- EventTypeConnectionOpen EventType = 3
- EventTypeConnectionClose EventType = 4
- EventTypeConnectionError EventType = 5
- EventTypeListenOpen EventType = 6
- EventTypeListenClose EventType = 7
- EventTypeFileOpen EventType = 8
- EventTypeTCPRetransmit EventType = 9
- EventTypeL7Request EventType = 10
- EventTypeFunEnt EventType = 11
- EventTypeFunRet EventType = 12
- EventReasonNone EventReason = 0
- EventReasonOOMKill EventReason = 1
- )
- type Event struct {
- Type EventType
- Reason EventReason
- Pid uint32
- SrcAddr netaddr.IPPort
- DstAddr netaddr.IPPort
- Fd uint64
- Timestamp uint64
- L7Request *l7.RequestData
- StackEvent *StackEvent
- }
- type perfMapType uint8
- const (
- perfMapTypeProcEvents perfMapType = 1
- perfMapTypeTCPEvents perfMapType = 2
- perfMapTypeFileEvents perfMapType = 3
- perfMapTypeL7Events perfMapType = 4
- perfMapTypeSocketEvents perfMapType = 5
- perfMapTypeEventQueue perfMapType = 6
- )
- type Tracer struct {
- kernelVersion string
- disableL7Tracing bool
- collection *ebpf.Collection
- readers map[string]*perf.Reader
- links []link.Link
- uprobes map[string]*ebpf.Program
- Symbols []debugelf.Symbol
- Uprobes []tracer.Uprobe
- UprobesMap map[string]tracer.Uprobe
- }
- func NewTracer(kernelVersion string, disableL7Tracing bool) *Tracer {
- if disableL7Tracing {
- klog.Infoln("L7 tracing is disabled")
- }
- return &Tracer{
- kernelVersion: kernelVersion,
- disableL7Tracing: disableL7Tracing,
- readers: map[string]*perf.Reader{},
- uprobes: map[string]*ebpf.Program{},
- }
- }
- func (t *Tracer) Run(events chan<- Event) error {
- if err := t.ebpf(events); err != nil {
- return err
- }
- if err := t.stack(); err != nil {
- return err
- }
- if err := t.init(events); err != nil {
- return err
- }
- return nil
- }
- func (t *Tracer) Close() {
- for _, p := range t.uprobes {
- _ = p.Close()
- }
- for _, l := range t.links {
- _ = l.Close()
- }
- for _, r := range t.readers {
- _ = r.Close()
- }
- t.collection.Close()
- }
- func (t *Tracer) init(ch chan<- Event) error {
- pids, err := proc.ListPids()
- if err != nil {
- return fmt.Errorf("failed to list pids: %w", err)
- }
- for _, pid := range pids {
- ch <- Event{Type: EventTypeProcessStart, Pid: pid}
- }
- fds, sockets := readFds(pids)
- for _, fd := range fds {
- ch <- Event{Type: EventTypeFileOpen, Pid: fd.pid, Fd: fd.fd}
- }
- listens := map[uint64]bool{}
- for _, s := range sockets {
- if s.Listen {
- listens[uint64(s.pid)<<32|uint64(s.SAddr.Port())] = true
- }
- }
- for _, s := range sockets {
- typ := EventTypeConnectionOpen
- if s.Listen {
- typ = EventTypeListenOpen
- } else if listens[uint64(s.pid)<<32|uint64(s.SAddr.Port())] || s.DAddr.Port() > s.SAddr.Port() { // inbound
- continue
- }
- ch <- Event{
- Type: typ,
- Pid: s.pid,
- Fd: s.fd,
- SrcAddr: s.SAddr,
- DstAddr: s.DAddr,
- }
- }
- return nil
- }
- type perfMap struct {
- name string
- perCPUBufferSizePages int
- typ perfMapType
- }
- func (t *Tracer) ebpf(ch chan<- Event) error {
- if _, ok := ebpfProg[runtime.GOARCH]; !ok {
- return fmt.Errorf("unsupported architecture: %s", runtime.GOARCH)
- }
- kv := "v" + common.KernelMajorMinor(t.kernelVersion)
- var prg []byte
- for _, p := range ebpfProg[runtime.GOARCH] {
- if semver.Compare(kv, p.v) >= 0 {
- prg = p.p
- break
- }
- }
- if len(prg) == 0 {
- return fmt.Errorf("unsupported kernel version: %s", t.kernelVersion)
- }
- _, debugFsErr := os.Stat("/sys/kernel/debug/tracing")
- _, traceFsErr := os.Stat("/sys/kernel/tracing")
- if debugFsErr != nil && traceFsErr != nil {
- return fmt.Errorf("kernel tracing is not available: debugfs or tracefs must be mounted")
- }
- collectionSpec, err := ebpf.LoadCollectionSpecFromReader(bytes.NewReader(prg))
- if err != nil {
- return fmt.Errorf("failed to load collection spec: %w", err)
- }
- _ = unix.Setrlimit(unix.RLIMIT_MEMLOCK, &unix.Rlimit{Cur: unix.RLIM_INFINITY, Max: unix.RLIM_INFINITY})
- tracer.PidFilter(collectionSpec)
- opts := &ebpf.CollectionOptions{MapReplacements: make(map[string]*ebpf.Map)}
- for _, spec := range collectionSpec.Maps {
- fmt.Println("maps:", spec.Name)
- }
- tracer.MapInit(collectionSpec, opts)
- // TODO 多进程
- tracer.SetConstants(collectionSpec)
- c, err := ebpf.NewCollectionWithOptions(collectionSpec, *opts)
- if err != nil {
- var verr *ebpf.VerifierError
- if errors.As(err, &verr) {
- klog.Errorf("----%+v", verr)
- }
- return fmt.Errorf("failed to load collection: %w", err)
- }
- tracer.Offset()
- t.collection = c
- perfMaps := []perfMap{
- {name: "proc_events", typ: perfMapTypeProcEvents, perCPUBufferSizePages: 4},
- {name: "tcp_listen_events", typ: perfMapTypeTCPEvents, perCPUBufferSizePages: 4},
- {name: "tcp_connect_events", typ: perfMapTypeTCPEvents, perCPUBufferSizePages: 8},
- {name: "tcp_retransmit_events", typ: perfMapTypeTCPEvents, perCPUBufferSizePages: 4},
- {name: "file_events", typ: perfMapTypeFileEvents, perCPUBufferSizePages: 4},
- {name: "event_queue", typ: perfMapTypeEventQueue, perCPUBufferSizePages: 32},
- }
- fmt.Println(len(collectionSpec.Programs))
- fmt.Println(len(c.Programs))
- tracer.MapInsert(c)
- if !t.disableL7Tracing {
- perfMaps = append(perfMaps, perfMap{name: "l7_events", typ: perfMapTypeL7Events, perCPUBufferSizePages: 32})
- }
- perfMaps = append(perfMaps, perfMap{name: tracer.MAP_PERF_SOCKET_DATA_NAME, typ: perfMapTypeSocketEvents, perCPUBufferSizePages: 64})
- fmt.Println("perfMaps start --")
- for _, pm := range perfMaps {
- fmt.Println("pm.namepm.name: ", pm.name)
- m, ok := t.collection.Maps[pm.name]
- if ok {
- r, err := perf.NewReader(m, pm.perCPUBufferSizePages*os.Getpagesize())
- if err != nil {
- t.Close()
- return fmt.Errorf("failed to create ebpf reader: %w", err)
- }
- t.readers[pm.name] = r
- // event监听
- go runEventsReader(pm.name, r, ch, pm.typ)
- }
- }
- fmt.Println("perfMaps end --")
- for _, programSpec := range collectionSpec.Programs {
- program := t.collection.Programs[programSpec.Name]
- fmt.Println("programSpecprogramSpec:--:", programSpec.Name, programSpec.SectionName, programSpec.Type)
- if t.disableL7Tracing {
- switch programSpec.Name {
- case "sys_enter_writev", "sys_enter_write", "sys_enter_sendto", "sys_enter_sendmsg", "sys_enter_sendmmsg":
- continue
- case "sys_enter_read", "sys_enter_readv", "sys_enter_recvfrom", "sys_enter_recvmsg":
- continue
- case "sys_exit_read", "sys_exit_readv", "sys_exit_recvfrom", "sys_exit_recvmsg":
- continue
- }
- }
- var l link.Link
- switch programSpec.Type {
- case ebpf.TracePoint:
- if strings.Contains(programSpec.SectionName, "prog") {
- continue
- }
- parts := strings.SplitN(programSpec.AttachTo, "/", 2)
- l, err = link.Tracepoint(parts[0], parts[1], program, nil)
- case ebpf.Kprobe:
- if strings.HasPrefix(programSpec.SectionName, "uprobe/") {
- fmt.Println("==============uprobe s")
- fmt.Println(programSpec.Name, programSpec.SectionName, programSpec.Type)
- fmt.Println("==============uprobe e")
- t.uprobes[programSpec.Name] = program
- continue
- }
- l, err = link.Kprobe(programSpec.AttachTo, program, nil)
- }
- if err != nil {
- t.Close()
- return fmt.Errorf("failed to link program: %w", err)
- }
- t.links = append(t.links, l)
- }
- return nil
- }
- func (t EventType) String() string {
- switch t {
- case EventTypeProcessStart:
- return "process-start"
- case EventTypeProcessExit:
- return "process-exit"
- case EventTypeConnectionOpen:
- return "connection-open"
- case EventTypeConnectionClose:
- return "connection-close"
- case EventTypeConnectionError:
- return "connection-error"
- case EventTypeListenOpen:
- return "listen-open"
- case EventTypeListenClose:
- return "listen-close"
- case EventTypeFileOpen:
- return "file-open"
- case EventTypeTCPRetransmit:
- return "tcp-retransmit"
- case EventTypeL7Request:
- return "l7-request"
- }
- return "unknown: " + strconv.Itoa(int(t))
- }
- func (t EventReason) String() string {
- switch t {
- case EventReasonNone:
- return "none"
- case EventReasonOOMKill:
- return "oom-kill"
- }
- return "unknown: " + strconv.Itoa(int(t))
- }
- type procEvent struct {
- Type EventType
- Pid uint32
- Reason uint32
- }
- type tcpEvent struct {
- Fd uint64
- Timestamp uint64
- Type EventType
- Pid uint32
- SPort uint16
- DPort uint16
- SAddr [16]byte
- DAddr [16]byte
- }
- type fileEvent struct {
- Type EventType
- Pid uint32
- Fd uint64
- }
- // struct l7_event in l7.c
- type l7Event struct {
- Fd uint64
- ConnectionTimestamp uint64
- Pid uint32
- Status uint32
- Duration uint64
- Protocol uint8
- Method uint8
- Padding uint16
- StatementId uint32
- PayloadSize uint64
- TraceId uint64
- StartAt uint64 // ns
- EndtAt uint64 // ns
- TraceStart uint32
- TraceEnd uint32
- EventCount uint32
- AssumedAppId utils.HashByte
- SpanId utils.HashByte
- TraceIdFrom utils.HashByte16
- CalledId utils.HashByte
- InstanceIdFrom utils.HashByte
- AppIdFrom utils.HashByte
- SpanIdFrom utils.HashByte
- }
- type SocketDataBufferddd struct {
- EventsNum uint32
- Len uint32
- Data [32760]byte
- }
- const (
- TASK_COMM_LEN = 16
- BURST_DATA_BUF_SIZE = 8192
- )
- type Tuple struct {
- Daddr [16]uint8
- RcvSaddr [16]uint8
- AddrLen uint8
- L4Protocol uint8
- Dport uint16
- Num uint16
- }
- type SocketDatadddd struct {
- Pid uint32 // 表示线程号 如果'pid == tgid'表示一个进程, 否则是线程
- Tgid uint32 // 进程号
- CoroutineID uint64
- Source uint8
- Comm [TASK_COMM_LEN]byte
- SocketID uint64
- Tuple Tuple
- ExtraData uint32
- ExtraDataCount uint32
- TcpSeq uint32
- ThreadTraceID uint64
- Timestamp uint64
- Direction uint8
- MsgType uint8
- SyscallLen uint64
- DataSeq uint64
- DataType uint16
- DataLen uint16
- Data [BURST_DATA_BUF_SIZE]byte
- }
- type StackEvent struct {
- Pid uint64
- TraceId uint64
- Goid uint64
- Ip uint64
- Bp uint64
- CallerIp uint64
- CallerBp uint64
- TimeNsStart uint64
- TimeNsEnd uint64
- // Nid uint64
- // Fpid uint64
- // Level uint64
- Location byte
- }
- type StackFunEvent struct {
- StackEvent StackEvent
- Uprobe *tracer.Uprobe
- Level int
- Pid int
- Nid int
- }
- func runEventsReader(name string, r *perf.Reader, ch chan<- Event, typ perfMapType) {
- for {
- rec, err := r.Read()
- if err != nil {
- if errors.Is(err, perf.ErrClosed) {
- break
- }
- continue
- }
- if rec.LostSamples > 0 {
- klog.Errorln(name, "lost samples:", rec.LostSamples)
- continue
- }
- var event Event
- switch typ {
- case perfMapTypeSocketEvents:
- //fmt.Println("perfMapTypeSocketEvents")
- // 假设 rec.RawSample 包含数据,类型为 []byte
- //rawData := rec.RawSample
- //fmt.Println("perfMapTypeSocketEvents2")
- //
- //// 创建一个 SocketDataBuffer 结构体实例
- //var buffer SocketDataBuffer
- //
- //// 创建一个字节缓冲区,并将数据填充到其中
- //reader := bytes.NewReader(rawData)
- //fmt.Println("perfMapTypeSocketEvents3")
- //fmt.Println(len(rawData))
- //// 使用 binary.Read 函数读取数据并解析为 SocketDataBuffer 结构体实例
- //if err := binary.Read(reader, binary.LittleEndian, &buffer); err != nil {
- // fmt.Println(reader.Len())
- // fmt.Println("Failed to read data:", err)
- // continue
- //}
- //fmt.Println("perfMapTypeSocketEvents4")
- //
- //// 打印解析后的数据
- //fmt.Println("EventsNum:", buffer.EventsNum)
- //fmt.Println("Len:", buffer.Len)
- //
- //// 打印 char data 的内容
- //fmt.Printf("Data: %s\n", string(buffer.Data[:buffer.Len])) // 仅打印实际长度的数据
- //socketDataBuffer := rec.RawSample
- // 解析 __socket_data_buffer
- //buf := (*SocketDataBuffer)(unsafe.Pointer(&rec.RawSample[0])) //nolint:gosec
- //
- //// 获取 char data[32760];
- //socketData := (*SocketData)(unsafe.Pointer(&buf.data[0])) //nolint:gosec
- //// todo
- //fmt.Printf("socketData.DataType:%d \n", (socketData.data_type))
- //fmt.Printf("socketData.DataLen:%d \n", (socketData.data_len))
- //
- //// 解析C结构体中的data字段
- //dataSlice := C.GoBytes(unsafe.Pointer(&socketData.data[0]), C.int(socketData.data_len))
- //// 打印或处理包含的数据
- //fmt.Printf("socketData.Payload:%v \n", string(dataSlice))
- /*todo */
- //socketData := (*(*[128]byte)(unsafe.Pointer(&eventC.line)))
- //dataPtr := unsafe.Pointer(&buf.data[0])
- //socketData := (*SocketData)(dataPtr)
- //reader2 := bytes.NewBuffer(rec.RawSample)
- // 222222
- //fmt.Println("socketData.Pid:", socketData.pid)
- //fmt.Println("socketData.Tgid:", socketData.tgid)
- //fmt.Println("socketData.CoroutineID:", socketData.coroutine_id)
- //fmt.Println("socketData.Source:", socketData.source)
- //
- //fmt.Printf("socketData.Comm: %s \n", socketData.comm)
- //fmt.Printf("socketData.SocketID :%v \n", socketData.socket_id)
- //fmt.Println("socketData.Tuple:", socketData.Tuple)
- //fmt.Println("socketData.ExtraData:", socketData.ExtraData)
- //fmt.Println("socketData.ExtraDataCount:", socketData.ExtraDataCount)
- //fmt.Println("socketData.TCPSeq:", socketData.TcpSeq)
- //fmt.Println("socketData.ThreadTraceID:", socketData.ThreadTraceID)
- //fmt.Println("socketData.Timestamp:", socketData.Timestamp)
- //fmt.Println("socketData.Direction:", socketData.Direction)
- //fmt.Println("socketData.MsgType:", socketData.MsgType)
- //fmt.Println("socketData.SyscallLen:", socketData.SyscallLen)
- //fmt.Println("socketData.DataSeq:", socketData.DataSeq)
- //socketData := &SocketData{}
- //reader := bytes.NewBuffer(rec.RawSample)
- //if err := binary.Read(reader, binary.LittleEndian, v); err != nil {
- // klog.Warningln("failed1 to read msg:", err)
- // continue
- //}
- //
- //var data []byte
- //payload := reader.Bytes()
- //switch {
- //case v.Len == 0:
- //case v.Len > 32760:
- // data = payload[:32760]
- //default:
- // data = payload[:v.Len]
- //}
- //////data2 := data[:v.Len]
- ////fmt.Println("perfMapTypeSocketEvents")
- //fmt.Println(v.EventsNum)
- //fmt.Println(v.Len)
- //fmt.Println(string(data))
- //
- //var data2 SocketData
- //reader2 := bytes.NewBuffer(data)
- //if err := binary.Read(reader2, binary.LittleEndian, data2); err != nil {
- // klog.Warningln("failed2 to read msg:", err)
- // continue
- //}
- //
- //fmt.Println(data2.Pid)
- //fmt.Println(data2.Tgid)
- //fmt.Println(string(v.Data))
- //continue
- case perfMapTypeL7Events:
- v := &l7Event{}
- reader := bytes.NewBuffer(rec.RawSample)
- if err := binary.Read(reader, binary.LittleEndian, v); err != nil {
- klog.Warningln("failed to read msg:", err)
- continue
- }
- //fmt.Println("v.TraceIdFrom")
- //fmt.Println(v.TraceIdFrom)
- //a := hex.EncodeToString(v.TraceIdFrom[:])
- //for _, b := range v.AssumedAppId {
- // fmt.Printf("v.AssumedAppId- %02\n", b)
- //}
- //fmt.Println(a)
- payload := reader.Bytes()
- req := &l7.RequestData{
- Protocol: l7.Protocol(v.Protocol),
- Status: l7.Status(v.Status),
- Duration: time.Duration(v.Duration),
- Method: l7.Method(v.Method),
- StatementId: v.StatementId,
- TraceId: v.TraceId,
- TraceStart: v.TraceStart,
- TraceEnd: v.TraceEnd,
- EventCount: v.EventCount,
- AssumedAppId: hex.EncodeToString(v.AssumedAppId[:]),
- SpanId: hex.EncodeToString(v.SpanId[:]),
- StartAt: v.StartAt,
- EndAt: v.EndtAt,
- }
- if v.TraceEnd == 1 {
- req.ParentSpanContext.TraceIdFrom = hex.EncodeToString(v.TraceIdFrom[:])
- req.ParentSpanContext.CalledId = hex.EncodeToString(v.CalledId[:])
- req.ParentSpanContext.InstanceIdFrom = hex.EncodeToString(v.InstanceIdFrom[:])
- req.ParentSpanContext.AppIdFrom = hex.EncodeToString(v.AppIdFrom[:])
- req.ParentSpanContext.SpanIdFrom = hex.EncodeToString(v.SpanIdFrom[:])
- }
- switch {
- case v.PayloadSize == 0:
- case v.PayloadSize > MaxPayloadSize:
- req.Payload = payload[:MaxPayloadSize]
- default:
- req.Payload = payload[:v.PayloadSize]
- }
- //fmt.Println("==========")
- //fmt.Println("req.Payload:", string(req.Payload))
- //fmt.Println("==========")
- event = Event{Type: EventTypeL7Request, Pid: v.Pid, Fd: v.Fd, Timestamp: v.ConnectionTimestamp, L7Request: req}
- case perfMapTypeFileEvents:
- v := &fileEvent{}
- if err := binary.Read(bytes.NewBuffer(rec.RawSample), binary.LittleEndian, v); err != nil {
- klog.Warningln("failed to read msg:", err)
- continue
- }
- event = Event{Type: v.Type, Pid: v.Pid, Fd: v.Fd}
- case perfMapTypeProcEvents:
- v := &procEvent{}
- if err := binary.Read(bytes.NewBuffer(rec.RawSample), binary.LittleEndian, v); err != nil {
- klog.Warningln("failed to read msg:", err)
- continue
- }
- event = Event{Type: v.Type, Reason: EventReason(v.Reason), Pid: v.Pid}
- case perfMapTypeTCPEvents:
- v := &tcpEvent{}
- if err := binary.Read(bytes.NewBuffer(rec.RawSample), binary.LittleEndian, v); err != nil {
- klog.Warningln("failed to read msg:", err)
- continue
- }
- event = Event{
- Type: v.Type,
- Pid: v.Pid,
- SrcAddr: ipPort(v.SAddr, v.SPort),
- DstAddr: ipPort(v.DAddr, v.DPort),
- Fd: v.Fd,
- Timestamp: v.Timestamp,
- }
- case perfMapTypeEventQueue:
- v := &StackEvent{}
- if err := binary.Read(bytes.NewBuffer(rec.RawSample), binary.LittleEndian, v); err != nil {
- klog.Warningln("failed to read msg:", err)
- continue
- }
- event = Event{
- Type: EventTypeFunEnt,
- StackEvent: v,
- }
- default:
- continue
- }
- ch <- event
- }
- }
- func ipPort(ip [16]byte, port uint16) netaddr.IPPort {
- i, _ := netaddr.FromStdIP(ip[:])
- return netaddr.IPPortFrom(i, port)
- }
- func (t *Tracer) InitKProcInfo(pid uint32, insID utils.ID) error {
- info := tracer.EbpfProcInfo{}
- info.InstanceId = insID.HashtVal
- // 获取内存地址
- allocDetails, err := tracer.Allocate(int(pid))
- if err == nil && allocDetails != nil {
- info.StartAddr = allocDetails.StartAddr
- info.EndAddr = allocDetails.EndAddr
- }
- //klog.Infoln("Major:", major)
- //klog.Infoln("Minor:", minor)
- //klog.Infoln("Revision:", revision)
- //klog.Infoln("goVersion", goVersion)
- //klog.Infoln("info.StartAddr", info.StartAddr)
- //klog.Infoln("info.EndAddr", info.EndAddr)
- _, err = tracer.UpdateProcInfoToMap(t.collection, pid, info)
- if err != nil {
- klog.Error("failed to update program info", err)
- }
- return err
- }
|