bpf_struct.go 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199
  1. package modelse
  2. const (
  3. ETR_OK = 0
  4. ETR_NOTEXIST = -4
  5. ETR_UPDATE_MAP_FAILD = -9 /* update map failed */
  6. ETR_NOTSUPP = -14
  7. )
  8. const (
  9. // eBPF Map Name
  10. MAP_MEMBERS_OFFSET_NAME = "__members_offset"
  11. MAP_SOCKET_INFO_NAME = "__socket_info_map"
  12. MAP_TRACE_NAME = "__trace_map"
  13. MAP_PERF_SOCKET_DATA_NAME = "__socket_data"
  14. MAP_TRACE_CONF_NAME = "__trace_conf_map"
  15. MAP_TRACE_STATS_NAME = "__trace_stats_map"
  16. MAP_PROTO_FILTER_NAME = "__protocol_filter"
  17. MAP_ALLOW_PORT_BITMAP_NAME = "__allow_port_bitmap"
  18. MAP_ADAPT_KERN_UID_NAME = "__adapt_kern_uid_map"
  19. //Program jmp tables
  20. MAP_PROGS_JMP_KP_NAME = "__progs_jmp_kp_map"
  21. MAP_PROGS_JMP_TP_NAME = "__progs_jmp_tp_map"
  22. MAP_PROGS_JMP_UP_NAME = "__progs_jmp_up_map"
  23. MAP_PROC_INFO_MAP_NAME = "proc_info_map"
  24. // This prog is designed to handle data transfer
  25. PROGUP = "bpf_prog_up__"
  26. PROGKP = "bpf_prog_kp__"
  27. PROGTP = "bpf_prog_tp__"
  28. PROG_DATA_SUBMIT_NAME_FOR_KP = PROGKP + "data_submit"
  29. PROG_DATA_SUBMIT_NAME_FOR_TP = PROGTP + "data_submit"
  30. PROG_OUTPUT_DATA_NAME_FOR_KP = PROGKP + "output_data"
  31. PROG_OUTPUT_DATA_NAME_FOR_TP = PROGTP + "output_data"
  32. PROG_IO_EVENT_NAME_FOR_TP = PROGTP + "io_event"
  33. PROG_SAVE_SC_DATA_FOR_UP = PROGUP + "java_save_sc"
  34. PROG_JAVA_UPDATE_HEADER_FOR_UP = PROGUP + "java_update_header"
  35. PROG_GO_UPDATE_HEADER_FOR_UP = PROGUP + "go_update_header"
  36. )
  37. const (
  38. PROG_DATA_SUBMIT_TP_IDX = iota
  39. PROG_OUTPUT_DATA_TP_IDX
  40. PROG_IO_EVENT_TP_IDX
  41. PROG_TP_NUM
  42. )
  43. const (
  44. PROG_DATA_SUBMIT_KP_IDX = iota
  45. PROG_OUTPUT_DATA_KP_IDX
  46. PROG_KP_NUM
  47. )
  48. const (
  49. PROG_DATA_SAVE_UP_IDX = iota
  50. PROG_DATA_JAVA_UPDATE_HEADER_UP_IDX
  51. PROG_DATA_GO_UPDATE_HEADER_UP_IDX
  52. PROG_UPROBE_NUM
  53. )
  54. // 数据协议
  55. const (
  56. PROTO_UNKNOWN = 0
  57. PROTO_ORTHER = 1
  58. PROTO_HTTP1 = 20
  59. PROTO_HTTP2 = 21
  60. PROTO_TLS_HTTP1 = 22
  61. PROTO_TLS_HTTP2 = 23
  62. PROTO_DUBBO = 40
  63. PROTO_SOFARPC = 43
  64. PROTO_MYSQL = 60
  65. PROTO_POSTGRESQL = 61
  66. PROTO_REDIS = 80
  67. PROTO_KAFKA = 100
  68. PROTO_MQTT = 101
  69. PROTO_DNS = 120
  70. PROTO_NUM = 130
  71. )
  72. const (
  73. HASH_SIZE = 8
  74. )
  75. var EbpfConfigProtocolFilter [PROTO_NUM]uint32
  76. type TestStruct struct {
  77. Test_id uint32
  78. }
  79. type BpfOffsetParam struct {
  80. Ready uint32 `json:"ready"`
  81. TaskFilesOffset uint32 `json:"task__files_offset"`
  82. SockFlagsOffset uint32 `json:"sock__flags_offset"`
  83. TcpSockCopiedSeqOffset uint32 `json:"tcp_sock__copied_seq_offset"`
  84. TcpSockWriteSeqOffset uint32 `json:"tcp_sock__write_seq_offset"`
  85. StructFilesStructFdtOffset uint32 `json:"struct_files_struct_fdt_offset"`
  86. StructFilesPrivateDataOffset uint32 `json:"struct_files_private_data_offset"`
  87. StructFileFInodeOffset uint32 `json:"struct_file_f_inode_offset"`
  88. StructInodeIModeOffset uint32 `json:"struct_inode_i_mode_offset"`
  89. StructFileDentryOffset uint32 `json:"struct_file_dentry_offset"`
  90. StructDentryNameOffset uint32 `json:"struct_dentry_name_offset"`
  91. StructSockFamilyOffset uint32 `json:"struct_sock_family_offset"`
  92. StructSockSaddrOffset uint32 `json:"struct_sock_saddr_offset"`
  93. StructSockDaddrOffset uint32 `json:"struct_sock_daddr_offset"`
  94. StructSockIp6saddrOffset uint32 `json:"struct_sock_ip6saddr_offset"`
  95. StructSockIp6daddrOffset uint32 `json:"struct_sock_ip6daddr_offset"`
  96. StructSockDportOffset uint32 `json:"struct_sock_dport_offset"`
  97. StructSockSportOffset uint32 `json:"struct_sock_sport_offset"`
  98. StructSockSkcStateOffset uint32 `json:"struct_sock_skc_state_offset"`
  99. StructSockCommonIpv6onlyOffset uint32 `json:"struct_sock_common_ipv6only_offset"`
  100. }
  101. const (
  102. OFFSET_IDX_GOID_RUNTIME_G = iota
  103. OFFSET_IDX_CONN_TLS_CONN
  104. OFFSET_IDX_SYSFD_POLL_FD
  105. OFFSET_IDX_CONN_HTTP2_SERVER_CONN
  106. OFFSET_IDX_TCONN_HTTP2_CLIENT_CONN
  107. OFFSET_IDX_CC_HTTP2_CLIENT_CONN_READ_LOOP
  108. OFFSET_IDX_CONN_GRPC_HTTP2_CLIENT
  109. OFFSET_IDX_CONN_GRPC_HTTP2_SERVER
  110. OFFSET_IDX_FRAMER_GRPC_TRANSPORT_LOOPY_WRITER
  111. OFFSET_IDX_WRITER_GRPC_TRANSPORT_FRAMER
  112. OFFSET_IDX_CONN_GRPC_TRANSPORT_BUFWRITER
  113. OFFSET_IDX_SIDE_GRPC_TRANSPORT_LOOPY_WRITER
  114. OFFSET_IDX_FIELDS_HTTP2_META_HEADERS_FRAME
  115. OFFSET_IDX_STREAM_HTTP2_CLIENT_CONN
  116. OFFSET_IDX_STREAM_ID_HTTP2_FRAME_HEADER
  117. OFFSET_IDX_MAX
  118. )
  119. type EbpfTraceConf struct {
  120. SocketID uint64 // 会话标识
  121. CoroutineTraceID uint64 // 同一协程的数据转发关联
  122. ThreadTraceID uint64 // 同一进程/线程的数据转发关联,用于多事务流转场景
  123. DataLimitMax uint64 // Maximum number of data transfers
  124. GoTracingTimeout uint64
  125. IOEventCollectMode uint64
  126. IOEventMinimalDuartion uint64
  127. TotalCpus uint64
  128. HostID HashByte
  129. APPID HashByte
  130. }
  131. /*
  132. struct ebpf_proc_info {
  133. __u32 version;
  134. __u16 offsets[OFFSET_IDX_MAX];
  135. // In golang, itab represents type, and in interface, struct is represented
  136. // by the address of itab. We use itab to judge the structure type, and
  137. // find the fd representing the connection after multiple jumps. These
  138. // types are not available in Go ELF files without a symbol table.
  139. // Go 用 itab 表示类型, 在 interface 中通过 itab 确定具体的 struct, 并根据
  140. // struct 找到表示连接的 fd.
  141. __u64 net_TCPConn_itab;
  142. __u64 crypto_tls_Conn_itab; // TLS_HTTP1,TLS_HTTP2
  143. __u64 credentials_syscallConn_itab; // gRPC
  144. __u64 start_addr;
  145. __u64 end_addr;
  146. unsigned char instance_id[APM_INSTANCE_ID_SIZE];
  147. __u64 method_ptr_pos;
  148. __u64 url_ptr_pos;
  149. __u64 path_ptr_pos;
  150. __u64 status_code_pos;
  151. __u64 request_host_pos;
  152. __u64 request_proto_pos;
  153. __u64 ctx_ptr_pos;
  154. __u64 headers_ptr_pos;
  155. __u64 buckets_ptr_pos;
  156. } __attribute__((packed));
  157. */
  158. type EbpfProcInfo struct {
  159. Version uint32
  160. Offsets [OFFSET_IDX_MAX]uint16
  161. NetTCPConnItab uint64
  162. CryptoTLSConnItab uint64 // TLS_HTTP1,TLS_HTTP2
  163. CredentialsSyscallConnItab uint64 // gRPC
  164. StartAddr uint64
  165. EndAddr uint64
  166. InstanceId HashByte
  167. AppId HashByte
  168. CodeType uint16
  169. // go http
  170. MethodPtrPos uint64
  171. UrlPtrPos uint64
  172. PathPtrPos uint64
  173. StatusCodePos uint64
  174. RequestHostPos uint64
  175. ProtoPos uint64
  176. CtxPtrPos uint64
  177. HeadersPtrPos uint64
  178. BucketsPtrPos uint64
  179. }
  180. type allowPortBitmap struct {
  181. Bitmap [65536 / 8]uint8
  182. }