tracer.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749
  1. package ebpftracer
  2. import (
  3. "bytes"
  4. debugelf "debug/elf"
  5. "encoding/binary"
  6. "encoding/hex"
  7. "errors"
  8. "fmt"
  9. "os"
  10. "strconv"
  11. "strings"
  12. "time"
  13. "github.com/cilium/ebpf"
  14. "github.com/cilium/ebpf/link"
  15. "github.com/cilium/ebpf/perf"
  16. "github.com/coroot/coroot-node-agent/common"
  17. "github.com/coroot/coroot-node-agent/ebpftracer/l7"
  18. "github.com/coroot/coroot-node-agent/ebpftracer/tracer"
  19. "github.com/coroot/coroot-node-agent/proc"
  20. "github.com/coroot/coroot-node-agent/utils"
  21. klog "github.com/sirupsen/logrus"
  22. "golang.org/x/sys/unix"
  23. "inet.af/netaddr"
  24. )
  25. /*
  26. #define TASK_COMM_LEN 16
  27. #define BURST_DATA_BUF_SIZE 8192 // For brust send buffer
  28. #include <linux/types.h>
  29. struct __tuple_t {
  30. __u8 daddr[16];
  31. __u8 rcv_saddr[16];
  32. __u8 addr_len;
  33. __u8 l4_protocol;
  34. __u16 dport;
  35. __u16 num;
  36. };
  37. struct __socket_data {
  38. __u32 pid;
  39. __u32 tgid;
  40. __u64 coroutine_id;
  41. __u8 source;
  42. __u8 comm[TASK_COMM_LEN];
  43. __u64 socket_id;
  44. struct __tuple_t tuple;
  45. __u32 extra_data;
  46. __u32 extra_data_count;
  47. __u32 tcp_seq;
  48. __u64 thread_trace_id;
  49. __u64 timestamp;
  50. __u8 direction: 1;
  51. __u8 msg_type: 7;
  52. __u64 syscall_len;
  53. __u64 data_seq;
  54. __u16 data_type;
  55. __u16 data_len;
  56. char data[BURST_DATA_BUF_SIZE];
  57. } __attribute__((packed));
  58. struct __socket_data_buffer {
  59. __u32 events_num;
  60. __u32 len;
  61. char data[32760];
  62. };
  63. */
  64. import "C"
  65. type SocketData C.struct___socket_data
  66. type SocketDataBuffer C.struct___socket_data_buffer
  67. const MaxPayloadSize = 1024
  68. type EventType uint32
  69. type EventReason uint32
  70. const (
  71. EventTypeProcessStart EventType = 1
  72. EventTypeProcessExit EventType = 2
  73. EventTypeConnectionOpen EventType = 3
  74. EventTypeConnectionClose EventType = 4
  75. EventTypeConnectionError EventType = 5
  76. EventTypeListenOpen EventType = 6
  77. EventTypeListenClose EventType = 7
  78. EventTypeFileOpen EventType = 8
  79. EventTypeTCPRetransmit EventType = 9
  80. EventTypeL7Request EventType = 10
  81. EventTypeFunEnt EventType = 11
  82. EventTypeFunRet EventType = 12
  83. EventReasonNone EventReason = 0
  84. EventReasonOOMKill EventReason = 1
  85. )
  86. type Event struct {
  87. Type EventType
  88. Reason EventReason
  89. Pid uint32
  90. SrcAddr netaddr.IPPort
  91. DstAddr netaddr.IPPort
  92. Fd uint64
  93. Timestamp uint64
  94. L7Request *l7.RequestData
  95. StackEvent *StackEvent
  96. }
  97. type perfMapType uint8
  98. const (
  99. perfMapTypeProcEvents perfMapType = 1
  100. perfMapTypeTCPEvents perfMapType = 2
  101. perfMapTypeFileEvents perfMapType = 3
  102. perfMapTypeL7Events perfMapType = 4
  103. perfMapTypeSocketEvents perfMapType = 5
  104. perfMapTypeEventQueue perfMapType = 6
  105. )
  106. type Tracer struct {
  107. kernelVersion string
  108. disableL7Tracing bool
  109. disableE2ETracing bool
  110. disableStackTracing bool
  111. collection *ebpf.Collection
  112. readers map[string]*perf.Reader
  113. links []link.Link
  114. uprobes map[string]*ebpf.Program
  115. Symbols []debugelf.Symbol
  116. Uprobes []tracer.Uprobe
  117. UprobesMap map[string]tracer.Uprobe
  118. }
  119. func NewTracer(kernelVersion string, disableL7Tracing, disableE2ETracing, disableStackTracing bool) *Tracer {
  120. if disableL7Tracing {
  121. klog.Infoln("L7 tracing is disabled")
  122. }
  123. if disableE2ETracing {
  124. klog.Infoln("L7 tracing is disabled")
  125. }
  126. if disableStackTracing {
  127. klog.Infoln("L7 stack is disabled")
  128. }
  129. return &Tracer{
  130. kernelVersion: kernelVersion,
  131. disableL7Tracing: disableL7Tracing,
  132. disableE2ETracing: disableE2ETracing,
  133. disableStackTracing: disableStackTracing,
  134. readers: map[string]*perf.Reader{},
  135. uprobes: map[string]*ebpf.Program{},
  136. }
  137. }
  138. func (t *Tracer) Run(events chan<- Event) error {
  139. if err := t.ebpf(events); err != nil {
  140. return err
  141. }
  142. if err := t.init(events); err != nil {
  143. return err
  144. }
  145. return nil
  146. }
  147. func (t *Tracer) Close() {
  148. for _, p := range t.uprobes {
  149. _ = p.Close()
  150. }
  151. for _, l := range t.links {
  152. _ = l.Close()
  153. }
  154. for _, r := range t.readers {
  155. _ = r.Close()
  156. }
  157. t.collection.Close()
  158. }
  159. func (t *Tracer) init(ch chan<- Event) error {
  160. pids, err := proc.ListPids()
  161. if err != nil {
  162. return fmt.Errorf("failed to list pids: %w", err)
  163. }
  164. for _, pid := range pids {
  165. ch <- Event{Type: EventTypeProcessStart, Pid: pid}
  166. }
  167. fds, sockets := readFds(pids)
  168. for _, fd := range fds {
  169. ch <- Event{Type: EventTypeFileOpen, Pid: fd.pid, Fd: fd.fd}
  170. }
  171. listens := map[uint64]bool{}
  172. for _, s := range sockets {
  173. if s.Listen {
  174. listens[uint64(s.pid)<<32|uint64(s.SAddr.Port())] = true
  175. }
  176. }
  177. for _, s := range sockets {
  178. typ := EventTypeConnectionOpen
  179. if s.Listen {
  180. typ = EventTypeListenOpen
  181. } else if listens[uint64(s.pid)<<32|uint64(s.SAddr.Port())] || s.DAddr.Port() > s.SAddr.Port() { // inbound
  182. continue
  183. }
  184. ch <- Event{
  185. Type: typ,
  186. Pid: s.pid,
  187. Fd: s.fd,
  188. SrcAddr: s.SAddr,
  189. DstAddr: s.DAddr,
  190. }
  191. }
  192. return nil
  193. }
  194. type perfMap struct {
  195. name string
  196. perCPUBufferSizePages int
  197. typ perfMapType
  198. }
  199. func (t *Tracer) ebpf(ch chan<- Event) error {
  200. kv := "v" + common.KernelMajorMinor(t.kernelVersion)
  201. path, prg, err := EbpfCode(kv)
  202. klog.Infof("kv is %s, kernel version: %s path: %s", kv, t.kernelVersion, path)
  203. if len(prg) == 0 || err != nil {
  204. return fmt.Errorf("kv is %s, unsupported kernel version: %s path: %s err:<%v>", kv, t.kernelVersion, path, err)
  205. }
  206. _, debugFsErr := os.Stat("/sys/kernel/debug/tracing")
  207. _, traceFsErr := os.Stat("/sys/kernel/tracing")
  208. if debugFsErr != nil && traceFsErr != nil {
  209. return fmt.Errorf("kernel tracing is not available: debugfs or tracefs must be mounted")
  210. }
  211. collectionSpec, err := ebpf.LoadCollectionSpecFromReader(bytes.NewReader(prg))
  212. if err != nil {
  213. return fmt.Errorf("failed to load collection spec: %w", err)
  214. }
  215. _ = unix.Setrlimit(unix.RLIMIT_MEMLOCK, &unix.Rlimit{Cur: unix.RLIM_INFINITY, Max: unix.RLIM_INFINITY})
  216. tracer.PidFilter(collectionSpec)
  217. opts := &ebpf.CollectionOptions{MapReplacements: make(map[string]*ebpf.Map)}
  218. klog.Infof("[start] Look eBPF .maps")
  219. for _, spec := range collectionSpec.Maps {
  220. klog.Infoln(spec.Name)
  221. }
  222. klog.Infof("[end] Look eBPF .maps")
  223. tracer.MapInit(collectionSpec, opts)
  224. // TODO 多进程
  225. // tracer.SetConstants(collectionSpec)
  226. c, err := ebpf.NewCollectionWithOptions(collectionSpec, *opts)
  227. if err != nil {
  228. var verr *ebpf.VerifierError
  229. if errors.As(err, &verr) {
  230. klog.Errorf("----%+v", verr)
  231. }
  232. return fmt.Errorf("failed to load collection: %w", err)
  233. }
  234. tracer.Offset()
  235. t.collection = c
  236. perfMaps := []perfMap{
  237. {name: "proc_events", typ: perfMapTypeProcEvents, perCPUBufferSizePages: 4},
  238. {name: "tcp_listen_events", typ: perfMapTypeTCPEvents, perCPUBufferSizePages: 4},
  239. {name: "tcp_connect_events", typ: perfMapTypeTCPEvents, perCPUBufferSizePages: 8},
  240. {name: "tcp_retransmit_events", typ: perfMapTypeTCPEvents, perCPUBufferSizePages: 4},
  241. {name: "file_events", typ: perfMapTypeFileEvents, perCPUBufferSizePages: 4},
  242. {name: "event_queue", typ: perfMapTypeEventQueue, perCPUBufferSizePages: 32},
  243. }
  244. tracer.MapInsert(c)
  245. if !t.DisableL7Tracing() {
  246. perfMaps = append(perfMaps, perfMap{name: "l7_events", typ: perfMapTypeL7Events, perCPUBufferSizePages: 32})
  247. }
  248. perfMaps = append(perfMaps, perfMap{name: tracer.MAP_PERF_SOCKET_DATA_NAME, typ: perfMapTypeSocketEvents, perCPUBufferSizePages: 64})
  249. klog.Infof("[start] Look eBPF perf_maps")
  250. for _, pm := range perfMaps {
  251. klog.Infoln(pm.name)
  252. m, ok := t.collection.Maps[pm.name]
  253. if ok {
  254. r, err := perf.NewReader(m, pm.perCPUBufferSizePages*os.Getpagesize())
  255. if err != nil {
  256. t.Close()
  257. return fmt.Errorf("failed to create ebpf reader: %w", err)
  258. }
  259. t.readers[pm.name] = r
  260. // event监听
  261. go runEventsReader(pm.name, r, ch, pm.typ)
  262. }
  263. }
  264. klog.Infof("[end] Look eBPF perf_maps")
  265. klog.Infof("[start] Look eBPF specPrograms")
  266. for _, programSpec := range collectionSpec.Programs {
  267. program := t.collection.Programs[programSpec.Name]
  268. klog.Infof("%s:[%s]", programSpec.SectionName, programSpec.Name)
  269. if t.DisableL7Tracing() {
  270. switch programSpec.Name {
  271. case "sys_enter_writev", "sys_enter_write", "sys_enter_sendto", "sys_enter_sendmsg", "sys_enter_sendmmsg":
  272. continue
  273. case "sys_enter_read", "sys_enter_readv", "sys_enter_recvfrom", "sys_enter_recvmsg":
  274. continue
  275. case "sys_exit_read", "sys_exit_readv", "sys_exit_recvfrom", "sys_exit_recvmsg":
  276. continue
  277. }
  278. }
  279. var l link.Link
  280. switch programSpec.Type {
  281. case ebpf.TracePoint:
  282. if strings.Contains(programSpec.SectionName, "prog") {
  283. continue
  284. }
  285. parts := strings.SplitN(programSpec.AttachTo, "/", 2)
  286. l, err = link.Tracepoint(parts[0], parts[1], program, nil)
  287. case ebpf.Kprobe:
  288. if strings.HasPrefix(programSpec.SectionName, "uprobe/") {
  289. t.uprobes[programSpec.Name] = program
  290. continue
  291. }
  292. l, err = link.Kprobe(programSpec.AttachTo, program, nil)
  293. }
  294. if err != nil {
  295. t.Close()
  296. return fmt.Errorf("failed to link program: %w", err)
  297. }
  298. t.links = append(t.links, l)
  299. }
  300. klog.Infof("[end] Look eBPF specPrograms")
  301. return nil
  302. }
  303. func (t EventType) String() string {
  304. switch t {
  305. case EventTypeProcessStart:
  306. return "process-start"
  307. case EventTypeProcessExit:
  308. return "process-exit"
  309. case EventTypeConnectionOpen:
  310. return "connection-open"
  311. case EventTypeConnectionClose:
  312. return "connection-close"
  313. case EventTypeConnectionError:
  314. return "connection-error"
  315. case EventTypeListenOpen:
  316. return "listen-open"
  317. case EventTypeListenClose:
  318. return "listen-close"
  319. case EventTypeFileOpen:
  320. return "file-open"
  321. case EventTypeTCPRetransmit:
  322. return "tcp-retransmit"
  323. case EventTypeL7Request:
  324. return "l7-request"
  325. }
  326. return "unknown: " + strconv.Itoa(int(t))
  327. }
  328. func (t EventReason) String() string {
  329. switch t {
  330. case EventReasonNone:
  331. return "none"
  332. case EventReasonOOMKill:
  333. return "oom-kill"
  334. }
  335. return "unknown: " + strconv.Itoa(int(t))
  336. }
  337. type procEvent struct {
  338. Type EventType
  339. Pid uint32
  340. Reason uint32
  341. }
  342. type tcpEvent struct {
  343. Fd uint64
  344. Timestamp uint64
  345. Type EventType
  346. Pid uint32
  347. SPort uint16
  348. DPort uint16
  349. SAddr [16]byte
  350. DAddr [16]byte
  351. }
  352. type fileEvent struct {
  353. Type EventType
  354. Pid uint32
  355. Fd uint64
  356. }
  357. // struct l7_event in l7.c
  358. type l7Event struct {
  359. Fd uint64
  360. ConnectionTimestamp uint64
  361. Pid uint32
  362. Status uint32
  363. Duration uint64
  364. Protocol uint8
  365. Method uint8
  366. Padding uint16
  367. StatementId uint32
  368. PayloadSize uint64
  369. TraceId uint64
  370. StartAt uint64 // ns
  371. EndtAt uint64 // ns
  372. TraceStart uint32
  373. TraceEnd uint32
  374. EventCount uint32
  375. AssumedAppId utils.HashByte
  376. SpanId utils.HashByte
  377. TraceIdFrom utils.HashByte16
  378. CalledId utils.HashByte
  379. InstanceIdFrom utils.HashByte
  380. AppIdFrom utils.HashByte
  381. SpanIdFrom utils.HashByte
  382. }
  383. type SocketDataBufferddd struct {
  384. EventsNum uint32
  385. Len uint32
  386. Data [32760]byte
  387. }
  388. const (
  389. TASK_COMM_LEN = 16
  390. BURST_DATA_BUF_SIZE = 8192
  391. )
  392. type Tuple struct {
  393. Daddr [16]uint8
  394. RcvSaddr [16]uint8
  395. AddrLen uint8
  396. L4Protocol uint8
  397. Dport uint16
  398. Num uint16
  399. }
  400. type SocketDatadddd struct {
  401. Pid uint32 // 表示线程号 如果'pid == tgid'表示一个进程, 否则是线程
  402. Tgid uint32 // 进程号
  403. CoroutineID uint64
  404. Source uint8
  405. Comm [TASK_COMM_LEN]byte
  406. SocketID uint64
  407. Tuple Tuple
  408. ExtraData uint32
  409. ExtraDataCount uint32
  410. TcpSeq uint32
  411. ThreadTraceID uint64
  412. Timestamp uint64
  413. Direction uint8
  414. MsgType uint8
  415. SyscallLen uint64
  416. DataSeq uint64
  417. DataType uint16
  418. DataLen uint16
  419. Data [BURST_DATA_BUF_SIZE]byte
  420. }
  421. type StackEvent struct {
  422. Pid uint64
  423. TraceId uint64
  424. Goid uint64
  425. Ip uint64
  426. Bp uint64
  427. CallerIp uint64
  428. CallerBp uint64
  429. TimeNsStart uint64
  430. TimeNsEnd uint64
  431. // Nid uint64
  432. // Fpid uint64
  433. // Level uint64
  434. Location byte
  435. }
  436. type StackFunEvent struct {
  437. StackEvent StackEvent
  438. Uprobe *tracer.Uprobe
  439. }
  440. func runEventsReader(name string, r *perf.Reader, ch chan<- Event, typ perfMapType) {
  441. for {
  442. rec, err := r.Read()
  443. if err != nil {
  444. if errors.Is(err, perf.ErrClosed) {
  445. break
  446. }
  447. continue
  448. }
  449. if rec.LostSamples > 0 {
  450. klog.Errorln(name, "lost samples:", rec.LostSamples)
  451. continue
  452. }
  453. var event Event
  454. switch typ {
  455. case perfMapTypeSocketEvents:
  456. //fmt.Println("perfMapTypeSocketEvents")
  457. // 假设 rec.RawSample 包含数据,类型为 []byte
  458. //rawData := rec.RawSample
  459. //fmt.Println("perfMapTypeSocketEvents2")
  460. //
  461. //// 创建一个 SocketDataBuffer 结构体实例
  462. //var buffer SocketDataBuffer
  463. //
  464. //// 创建一个字节缓冲区,并将数据填充到其中
  465. //reader := bytes.NewReader(rawData)
  466. //fmt.Println("perfMapTypeSocketEvents3")
  467. //fmt.Println(len(rawData))
  468. //// 使用 binary.Read 函数读取数据并解析为 SocketDataBuffer 结构体实例
  469. //if err := binary.Read(reader, binary.LittleEndian, &buffer); err != nil {
  470. // fmt.Println(reader.Len())
  471. // fmt.Println("Failed to read data:", err)
  472. // continue
  473. //}
  474. //fmt.Println("perfMapTypeSocketEvents4")
  475. //
  476. //// 打印解析后的数据
  477. //fmt.Println("EventsNum:", buffer.EventsNum)
  478. //fmt.Println("Len:", buffer.Len)
  479. //
  480. //// 打印 char data 的内容
  481. //fmt.Printf("Data: %s\n", string(buffer.Data[:buffer.Len])) // 仅打印实际长度的数据
  482. //socketDataBuffer := rec.RawSample
  483. // 解析 __socket_data_buffer
  484. //buf := (*SocketDataBuffer)(unsafe.Pointer(&rec.RawSample[0])) //nolint:gosec
  485. //
  486. //// 获取 char data[32760];
  487. //socketData := (*SocketData)(unsafe.Pointer(&buf.data[0])) //nolint:gosec
  488. //// todo
  489. //fmt.Printf("socketData.DataType:%d \n", (socketData.data_type))
  490. //fmt.Printf("socketData.DataLen:%d \n", (socketData.data_len))
  491. //
  492. //// 解析C结构体中的data字段
  493. //dataSlice := C.GoBytes(unsafe.Pointer(&socketData.data[0]), C.int(socketData.data_len))
  494. //// 打印或处理包含的数据
  495. //fmt.Printf("socketData.Payload:%v \n", string(dataSlice))
  496. /*todo */
  497. //socketData := (*(*[128]byte)(unsafe.Pointer(&eventC.line)))
  498. //dataPtr := unsafe.Pointer(&buf.data[0])
  499. //socketData := (*SocketData)(dataPtr)
  500. //reader2 := bytes.NewBuffer(rec.RawSample)
  501. // 222222
  502. //fmt.Println("socketData.Pid:", socketData.pid)
  503. //fmt.Println("socketData.Tgid:", socketData.tgid)
  504. //fmt.Println("socketData.CoroutineID:", socketData.coroutine_id)
  505. //fmt.Println("socketData.Source:", socketData.source)
  506. //
  507. //fmt.Printf("socketData.Comm: %s \n", socketData.comm)
  508. //fmt.Printf("socketData.SocketID :%v \n", socketData.socket_id)
  509. //fmt.Println("socketData.Tuple:", socketData.Tuple)
  510. //fmt.Println("socketData.ExtraData:", socketData.ExtraData)
  511. //fmt.Println("socketData.ExtraDataCount:", socketData.ExtraDataCount)
  512. //fmt.Println("socketData.TCPSeq:", socketData.TcpSeq)
  513. //fmt.Println("socketData.ThreadTraceID:", socketData.ThreadTraceID)
  514. //fmt.Println("socketData.Timestamp:", socketData.Timestamp)
  515. //fmt.Println("socketData.Direction:", socketData.Direction)
  516. //fmt.Println("socketData.MsgType:", socketData.MsgType)
  517. //fmt.Println("socketData.SyscallLen:", socketData.SyscallLen)
  518. //fmt.Println("socketData.DataSeq:", socketData.DataSeq)
  519. //socketData := &SocketData{}
  520. //reader := bytes.NewBuffer(rec.RawSample)
  521. //if err := binary.Read(reader, binary.LittleEndian, v); err != nil {
  522. // klog.Warningln("failed1 to read msg:", err)
  523. // continue
  524. //}
  525. //
  526. //var data []byte
  527. //payload := reader.Bytes()
  528. //switch {
  529. //case v.Len == 0:
  530. //case v.Len > 32760:
  531. // data = payload[:32760]
  532. //default:
  533. // data = payload[:v.Len]
  534. //}
  535. //////data2 := data[:v.Len]
  536. ////fmt.Println("perfMapTypeSocketEvents")
  537. //fmt.Println(v.EventsNum)
  538. //fmt.Println(v.Len)
  539. //fmt.Println(string(data))
  540. //
  541. //var data2 SocketData
  542. //reader2 := bytes.NewBuffer(data)
  543. //if err := binary.Read(reader2, binary.LittleEndian, data2); err != nil {
  544. // klog.Warningln("failed2 to read msg:", err)
  545. // continue
  546. //}
  547. //
  548. //fmt.Println(data2.Pid)
  549. //fmt.Println(data2.Tgid)
  550. //fmt.Println(string(v.Data))
  551. //continue
  552. case perfMapTypeL7Events:
  553. v := &l7Event{}
  554. reader := bytes.NewBuffer(rec.RawSample)
  555. if err := binary.Read(reader, binary.LittleEndian, v); err != nil {
  556. klog.Warningln("failed to read msg:", err)
  557. continue
  558. }
  559. //fmt.Println("v.TraceIdFrom")
  560. //fmt.Println(v.TraceIdFrom)
  561. //a := hex.EncodeToString(v.TraceIdFrom[:])
  562. //for _, b := range v.AssumedAppId {
  563. // fmt.Printf("v.AssumedAppId- %02\n", b)
  564. //}
  565. //fmt.Println(a)
  566. payload := reader.Bytes()
  567. req := &l7.RequestData{
  568. Protocol: l7.Protocol(v.Protocol),
  569. Status: l7.Status(v.Status),
  570. Duration: time.Duration(v.Duration),
  571. Method: l7.Method(v.Method),
  572. StatementId: v.StatementId,
  573. TraceId: v.TraceId,
  574. TraceStart: v.TraceStart,
  575. TraceEnd: v.TraceEnd,
  576. EventCount: v.EventCount,
  577. AssumedAppId: hex.EncodeToString(v.AssumedAppId[:]),
  578. SpanId: hex.EncodeToString(v.SpanId[:]),
  579. StartAt: v.StartAt,
  580. EndAt: v.EndtAt,
  581. }
  582. if v.TraceEnd == 1 {
  583. req.ParentSpanContext.TraceIdFrom = hex.EncodeToString(v.TraceIdFrom[:])
  584. req.ParentSpanContext.CalledId = hex.EncodeToString(v.CalledId[:])
  585. req.ParentSpanContext.InstanceIdFrom = hex.EncodeToString(v.InstanceIdFrom[:])
  586. req.ParentSpanContext.AppIdFrom = hex.EncodeToString(v.AppIdFrom[:])
  587. req.ParentSpanContext.SpanIdFrom = hex.EncodeToString(v.SpanIdFrom[:])
  588. }
  589. switch {
  590. case v.PayloadSize == 0:
  591. case v.PayloadSize > MaxPayloadSize:
  592. req.Payload = payload[:MaxPayloadSize]
  593. default:
  594. req.Payload = payload[:v.PayloadSize]
  595. }
  596. //fmt.Println("==========")
  597. //fmt.Println("req.Payload:", string(req.Payload))
  598. //fmt.Println("==========")
  599. event = Event{Type: EventTypeL7Request, Pid: v.Pid, Fd: v.Fd, Timestamp: v.ConnectionTimestamp, L7Request: req}
  600. case perfMapTypeFileEvents:
  601. v := &fileEvent{}
  602. if err := binary.Read(bytes.NewBuffer(rec.RawSample), binary.LittleEndian, v); err != nil {
  603. klog.Warningln("failed to read msg:", err)
  604. continue
  605. }
  606. event = Event{Type: v.Type, Pid: v.Pid, Fd: v.Fd}
  607. case perfMapTypeProcEvents:
  608. v := &procEvent{}
  609. if err := binary.Read(bytes.NewBuffer(rec.RawSample), binary.LittleEndian, v); err != nil {
  610. klog.Warningln("failed to read msg:", err)
  611. continue
  612. }
  613. event = Event{Type: v.Type, Reason: EventReason(v.Reason), Pid: v.Pid}
  614. case perfMapTypeTCPEvents:
  615. v := &tcpEvent{}
  616. if err := binary.Read(bytes.NewBuffer(rec.RawSample), binary.LittleEndian, v); err != nil {
  617. klog.Warningln("failed to read msg:", err)
  618. continue
  619. }
  620. event = Event{
  621. Type: v.Type,
  622. Pid: v.Pid,
  623. SrcAddr: ipPort(v.SAddr, v.SPort),
  624. DstAddr: ipPort(v.DAddr, v.DPort),
  625. Fd: v.Fd,
  626. Timestamp: v.Timestamp,
  627. }
  628. case perfMapTypeEventQueue:
  629. v := &StackEvent{}
  630. if err := binary.Read(bytes.NewBuffer(rec.RawSample), binary.LittleEndian, v); err != nil {
  631. klog.Warningln("failed to read msg:", err)
  632. continue
  633. }
  634. event = Event{
  635. Type: EventTypeFunEnt,
  636. StackEvent: v,
  637. }
  638. default:
  639. continue
  640. }
  641. ch <- event
  642. }
  643. }
  644. func ipPort(ip [16]byte, port uint16) netaddr.IPPort {
  645. i, _ := netaddr.FromStdIP(ip[:])
  646. return netaddr.IPPortFrom(i, port)
  647. }
  648. func (t *Tracer) InitKProcInfo(pid uint32, insID utils.ID, codeType uint16) error {
  649. var err error
  650. info := tracer.EbpfProcInfo{}
  651. info.InstanceId = insID.HashtVal
  652. // 获取内存地址
  653. //allocDetails, err := tracer.Allocate(int(pid))
  654. //if err == nil && allocDetails != nil {
  655. // info.StartAddr = allocDetails.StartAddr
  656. // info.EndAddr = allocDetails.EndAddr
  657. //}
  658. info.CodeType = codeType
  659. //klog.Infoln("Major:", major)
  660. //klog.Infoln("Minor:", minor)
  661. //klog.Infoln("Revision:", revision)
  662. //klog.Infoln("goVersion", goVersion)
  663. //klog.Infoln("info.StartAddr", info.StartAddr)
  664. //klog.Infoln("info.EndAddr", info.EndAddr)
  665. _, err = tracer.UpdateProcInfoToMap(t.collection, pid, info)
  666. if err != nil {
  667. klog.Error("failed to update program info", err)
  668. }
  669. return err
  670. }
  671. // TODO check language
  672. func (t *Tracer) DisableL7Tracing() bool {
  673. return t.disableL7Tracing
  674. }
  675. func (t *Tracer) DisableE2ETracing() bool {
  676. return t.disableE2ETracing
  677. }
  678. func (t *Tracer) DisableStackTracing() bool {
  679. return t.disableStackTracing
  680. }