ebpf.c 68 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996
  1. // SPDX-License-Identifier: GPL-3.0-or-later
  2. #include <sys/time.h>
  3. #include <sys/resource.h>
  4. #include <ifaddrs.h>
  5. #include "ebpf.h"
  6. #include "ebpf_socket.h"
  7. #include "libnetdata/required_dummies.h"
  8. /*****************************************************************
  9. *
  10. * GLOBAL VARIABLES
  11. *
  12. *****************************************************************/
  13. char *ebpf_plugin_dir = PLUGINS_DIR;
  14. static char *ebpf_configured_log_dir = LOG_DIR;
  15. char *ebpf_algorithms[] = {"absolute", "incremental"};
  16. static int thread_finished = 0;
  17. int close_ebpf_plugin = 0;
  18. struct config collector_config = { .first_section = NULL,
  19. .last_section = NULL,
  20. .mutex = NETDATA_MUTEX_INITIALIZER,
  21. .index = { .avl_tree = { .root = NULL, .compar = appconfig_section_compare },
  22. .rwlock = AVL_LOCK_INITIALIZER } };
  23. int running_on_kernel = 0;
  24. int ebpf_nprocs;
  25. int isrh = 0;
  26. uint32_t finalized_threads = 1;
  27. pthread_mutex_t lock;
  28. pthread_mutex_t collect_data_mutex;
  29. pthread_cond_t collect_data_cond_var;
  30. ebpf_module_t ebpf_modules[] = {
  31. { .thread_name = "process", .config_name = "process", .enabled = 0, .start_routine = ebpf_process_thread,
  32. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  33. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0,
  34. .apps_routine = ebpf_process_create_apps_charts, .maps = NULL,
  35. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &process_config,
  36. .config_file = NETDATA_PROCESS_CONFIG_FILE,
  37. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4 | NETDATA_V5_10,
  38. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  39. { .thread_name = "socket", .config_name = "socket", .enabled = 0, .start_routine = ebpf_socket_thread,
  40. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  41. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0,
  42. .apps_routine = ebpf_socket_create_apps_charts, .maps = NULL,
  43. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &socket_config,
  44. .config_file = NETDATA_NETWORK_CONFIG_FILE,
  45. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  46. .load = EBPF_LOAD_LEGACY, .targets = socket_targets},
  47. { .thread_name = "cachestat", .config_name = "cachestat", .enabled = 0, .start_routine = ebpf_cachestat_thread,
  48. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  49. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0,
  50. .apps_routine = ebpf_cachestat_create_apps_charts, .maps = NULL,
  51. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &cachestat_config,
  52. .config_file = NETDATA_CACHESTAT_CONFIG_FILE,
  53. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18|
  54. NETDATA_V5_4 | NETDATA_V5_15 | NETDATA_V5_16,
  55. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  56. { .thread_name = "sync", .config_name = "sync", .enabled = 0, .start_routine = ebpf_sync_thread,
  57. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  58. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0, .apps_routine = NULL, .maps = NULL,
  59. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &sync_config,
  60. .config_file = NETDATA_SYNC_CONFIG_FILE,
  61. // All syscalls have the same kernels
  62. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  63. .load = EBPF_LOAD_LEGACY, .targets = sync_targets},
  64. { .thread_name = "dc", .config_name = "dc", .enabled = 0, .start_routine = ebpf_dcstat_thread,
  65. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  66. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0,
  67. .apps_routine = ebpf_dcstat_create_apps_charts, .maps = NULL,
  68. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &dcstat_config,
  69. .config_file = NETDATA_DIRECTORY_DCSTAT_CONFIG_FILE,
  70. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  71. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  72. { .thread_name = "swap", .config_name = "swap", .enabled = 0, .start_routine = ebpf_swap_thread,
  73. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  74. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0,
  75. .apps_routine = ebpf_swap_create_apps_charts, .maps = NULL,
  76. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &swap_config,
  77. .config_file = NETDATA_DIRECTORY_SWAP_CONFIG_FILE,
  78. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  79. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  80. { .thread_name = "vfs", .config_name = "vfs", .enabled = 0, .start_routine = ebpf_vfs_thread,
  81. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  82. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0,
  83. .apps_routine = ebpf_vfs_create_apps_charts, .maps = NULL,
  84. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &vfs_config,
  85. .config_file = NETDATA_DIRECTORY_VFS_CONFIG_FILE,
  86. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  87. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  88. { .thread_name = "filesystem", .config_name = "filesystem", .enabled = 0, .start_routine = ebpf_filesystem_thread,
  89. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  90. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0, .apps_routine = NULL, .maps = NULL,
  91. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &fs_config,
  92. .config_file = NETDATA_FILESYSTEM_CONFIG_FILE,
  93. //We are setting kernels as zero, because we load eBPF programs according the kernel running.
  94. .kernels = 0, .load = EBPF_LOAD_LEGACY, .targets = NULL },
  95. { .thread_name = "disk", .config_name = "disk", .enabled = 0, .start_routine = ebpf_disk_thread,
  96. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  97. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0, .apps_routine = NULL, .maps = NULL,
  98. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &disk_config,
  99. .config_file = NETDATA_DISK_CONFIG_FILE,
  100. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  101. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  102. { .thread_name = "mount", .config_name = "mount", .enabled = 0, .start_routine = ebpf_mount_thread,
  103. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  104. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0, .apps_routine = NULL, .maps = NULL,
  105. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &mount_config,
  106. .config_file = NETDATA_MOUNT_CONFIG_FILE,
  107. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  108. .load = EBPF_LOAD_LEGACY, .targets = mount_targets},
  109. { .thread_name = "fd", .config_name = "fd", .enabled = 0, .start_routine = ebpf_fd_thread,
  110. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  111. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0,
  112. .apps_routine = ebpf_fd_create_apps_charts, .maps = NULL,
  113. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &fd_config,
  114. .config_file = NETDATA_FD_CONFIG_FILE,
  115. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4 | NETDATA_V5_11,
  116. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  117. { .thread_name = "hardirq", .config_name = "hardirq", .enabled = 0, .start_routine = ebpf_hardirq_thread,
  118. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  119. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0, .apps_routine = NULL, .maps = NULL,
  120. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &hardirq_config,
  121. .config_file = NETDATA_HARDIRQ_CONFIG_FILE,
  122. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  123. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  124. { .thread_name = "softirq", .config_name = "softirq", .enabled = 0, .start_routine = ebpf_softirq_thread,
  125. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  126. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0, .apps_routine = NULL, .maps = NULL,
  127. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &softirq_config,
  128. .config_file = NETDATA_SOFTIRQ_CONFIG_FILE,
  129. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  130. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  131. { .thread_name = "oomkill", .config_name = "oomkill", .enabled = 0, .start_routine = ebpf_oomkill_thread,
  132. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  133. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0,
  134. .apps_routine = ebpf_oomkill_create_apps_charts, .maps = NULL,
  135. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &oomkill_config,
  136. .config_file = NETDATA_OOMKILL_CONFIG_FILE,
  137. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  138. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  139. { .thread_name = "shm", .config_name = "shm", .enabled = 0, .start_routine = ebpf_shm_thread,
  140. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  141. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0,
  142. .apps_routine = ebpf_shm_create_apps_charts, .maps = NULL,
  143. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &shm_config,
  144. .config_file = NETDATA_DIRECTORY_SHM_CONFIG_FILE,
  145. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  146. .load = EBPF_LOAD_LEGACY, .targets = shm_targets},
  147. { .thread_name = "mdflush", .config_name = "mdflush", .enabled = 0, .start_routine = ebpf_mdflush_thread,
  148. .update_every = EBPF_DEFAULT_UPDATE_EVERY, .global_charts = 1, .apps_charts = CONFIG_BOOLEAN_NO,
  149. .cgroup_charts = CONFIG_BOOLEAN_NO, .mode = MODE_ENTRY, .optional = 0, .apps_routine = NULL, .maps = NULL,
  150. .pid_map_size = ND_EBPF_DEFAULT_PID_SIZE, .names = NULL, .cfg = &mdflush_config,
  151. .config_file = NETDATA_DIRECTORY_MDFLUSH_CONFIG_FILE,
  152. .kernels = NETDATA_V3_10 | NETDATA_V4_14 | NETDATA_V4_16 | NETDATA_V4_18 | NETDATA_V5_4,
  153. .load = EBPF_LOAD_LEGACY, .targets = NULL},
  154. { .thread_name = NULL, .enabled = 0, .start_routine = NULL, .update_every = EBPF_DEFAULT_UPDATE_EVERY,
  155. .global_charts = 0, .apps_charts = CONFIG_BOOLEAN_NO, .cgroup_charts = CONFIG_BOOLEAN_NO,
  156. .mode = MODE_ENTRY, .optional = 0, .apps_routine = NULL, .maps = NULL, .pid_map_size = 0, .names = NULL,
  157. .cfg = NULL, .config_name = NULL, .kernels = 0, .load = EBPF_LOAD_LEGACY, .targets = NULL},
  158. };
  159. // Link with apps.plugin
  160. ebpf_process_stat_t *global_process_stat = NULL;
  161. // Link with cgroup.plugin
  162. netdata_ebpf_cgroup_shm_t shm_ebpf_cgroup = {NULL, NULL};
  163. int shm_fd_ebpf_cgroup = -1;
  164. sem_t *shm_sem_ebpf_cgroup = SEM_FAILED;
  165. pthread_mutex_t mutex_cgroup_shm;
  166. //Network viewer
  167. ebpf_network_viewer_options_t network_viewer_opt;
  168. // Statistic
  169. ebpf_plugin_stats_t plugin_statistics = {.core = 0, .legacy = 0, .running = 0, .threads = 0, .tracepoints = 0,
  170. .probes = 0, .retprobes = 0, .trampolines = 0};
  171. #ifdef LIBBPF_MAJOR_VERSION
  172. struct btf *default_btf = NULL;
  173. #endif
  174. char *btf_path = NULL;
  175. /*****************************************************************
  176. *
  177. * FUNCTIONS USED TO CLEAN MEMORY AND OPERATE SYSTEM FILES
  178. *
  179. *****************************************************************/
  180. /**
  181. * Clean Loaded Events
  182. *
  183. * This function cleans the events previous loaded on Linux.
  184. void clean_loaded_events()
  185. {
  186. int event_pid;
  187. for (event_pid = 0; ebpf_modules[event_pid].probes; event_pid++)
  188. clean_kprobe_events(NULL, (int)ebpf_modules[event_pid].thread_id, ebpf_modules[event_pid].probes);
  189. }
  190. */
  191. /**
  192. * Close the collector gracefully
  193. *
  194. * @param sig is the signal number used to close the collector
  195. */
  196. static void ebpf_exit(int sig)
  197. {
  198. close_ebpf_plugin = 1;
  199. static int remove_pid = 0;
  200. // When both threads were not finished case I try to go in front this address, the collector will crash
  201. if (!thread_finished) {
  202. return;
  203. }
  204. if (ebpf_modules[EBPF_MODULE_SOCKET_IDX].enabled) {
  205. ebpf_modules[EBPF_MODULE_SOCKET_IDX].enabled = 0;
  206. clean_socket_apps_structures();
  207. freez(socket_bandwidth_curr);
  208. }
  209. if (ebpf_modules[EBPF_MODULE_CACHESTAT_IDX].enabled) {
  210. ebpf_modules[EBPF_MODULE_CACHESTAT_IDX].enabled = 0;
  211. clean_cachestat_pid_structures();
  212. freez(cachestat_pid);
  213. }
  214. if (ebpf_modules[EBPF_MODULE_DCSTAT_IDX].enabled) {
  215. ebpf_modules[EBPF_MODULE_DCSTAT_IDX].enabled = 0;
  216. clean_dcstat_pid_structures();
  217. freez(dcstat_pid);
  218. }
  219. if (ebpf_modules[EBPF_MODULE_SWAP_IDX].enabled) {
  220. ebpf_modules[EBPF_MODULE_SWAP_IDX].enabled = 0;
  221. clean_swap_pid_structures();
  222. freez(swap_pid);
  223. }
  224. if (ebpf_modules[EBPF_MODULE_VFS_IDX].enabled) {
  225. ebpf_modules[EBPF_MODULE_VFS_IDX].enabled = 0;
  226. clean_vfs_pid_structures();
  227. freez(vfs_pid);
  228. }
  229. if (ebpf_modules[EBPF_MODULE_FD_IDX].enabled) {
  230. ebpf_modules[EBPF_MODULE_FD_IDX].enabled = 0;
  231. clean_fd_pid_structures();
  232. freez(fd_pid);
  233. }
  234. if (ebpf_modules[EBPF_MODULE_SHM_IDX].enabled) {
  235. ebpf_modules[EBPF_MODULE_SHM_IDX].enabled = 0;
  236. clean_shm_pid_structures();
  237. freez(shm_pid);
  238. }
  239. ebpf_close_cgroup_shm();
  240. ebpf_clean_cgroup_pids();
  241. /*
  242. int ret = fork();
  243. if (ret < 0) // error
  244. error("Cannot fork(), so I won't be able to clean %skprobe_events", NETDATA_DEBUGFS);
  245. else if (!ret) { // child
  246. int i;
  247. for (i = getdtablesize(); i >= 0; --i)
  248. close(i);
  249. int fd = open("/dev/null", O_RDWR, 0);
  250. if (fd != -1) {
  251. dup2(fd, STDIN_FILENO);
  252. dup2(fd, STDOUT_FILENO);
  253. dup2(fd, STDERR_FILENO);
  254. }
  255. if (fd > 2)
  256. close(fd);
  257. int sid = setsid();
  258. if (sid >= 0) {
  259. debug(D_EXIT, "Wait for father %d die", getpid());
  260. sleep_usec(200000); // Sleep 200 milliseconds to father dies.
  261. clean_loaded_events();
  262. } else {
  263. error("Cannot become session id leader, so I won't try to clean kprobe_events.\n");
  264. }
  265. } else { // parent
  266. exit(0);
  267. }
  268. */
  269. #ifdef LIBBPF_MAJOR_VERSION
  270. if (default_btf) {
  271. btf__free(default_btf);
  272. default_btf = NULL;
  273. }
  274. #endif
  275. if (!remove_pid) {
  276. remove_pid = 1;
  277. char filename[FILENAME_MAX + 1];
  278. ebpf_pid_file(filename, FILENAME_MAX);
  279. if (unlink(filename))
  280. error("Cannot remove PID file %s", filename);
  281. }
  282. exit(sig);
  283. }
  284. /*****************************************************************
  285. *
  286. * FUNCTIONS TO CREATE CHARTS
  287. *
  288. *****************************************************************/
  289. /**
  290. * Get a value from a structure.
  291. *
  292. * @param basis it is the first address of the structure
  293. * @param offset it is the offset of the data you want to access.
  294. * @return
  295. */
  296. collected_number get_value_from_structure(char *basis, size_t offset)
  297. {
  298. collected_number *value = (collected_number *)(basis + offset);
  299. collected_number ret = (collected_number)llabs(*value);
  300. // this reset is necessary to avoid keep a constant value while processing is not executing a task
  301. *value = 0;
  302. return ret;
  303. }
  304. /**
  305. * Write begin command on standard output
  306. *
  307. * @param family the chart family name
  308. * @param name the chart name
  309. */
  310. void write_begin_chart(char *family, char *name)
  311. {
  312. printf("BEGIN %s.%s\n", family, name);
  313. }
  314. /**
  315. * Write END command on stdout.
  316. */
  317. inline void write_end_chart()
  318. {
  319. printf("END\n");
  320. }
  321. /**
  322. * Write set command on standard output
  323. *
  324. * @param dim the dimension name
  325. * @param value the value for the dimension
  326. */
  327. void write_chart_dimension(char *dim, long long value)
  328. {
  329. printf("SET %s = %lld\n", dim, value);
  330. }
  331. /**
  332. * Call the necessary functions to create a chart.
  333. *
  334. * @param name the chart name
  335. * @param family the chart family
  336. * @param move the pointer with the values that will be published
  337. * @param end the number of values that will be written on standard output
  338. *
  339. * @return It returns a variable that maps the charts that did not have zero values.
  340. */
  341. void write_count_chart(char *name, char *family, netdata_publish_syscall_t *move, uint32_t end)
  342. {
  343. write_begin_chart(family, name);
  344. uint32_t i = 0;
  345. while (move && i < end) {
  346. write_chart_dimension(move->name, move->ncall);
  347. move = move->next;
  348. i++;
  349. }
  350. write_end_chart();
  351. }
  352. /**
  353. * Call the necessary functions to create a chart.
  354. *
  355. * @param name the chart name
  356. * @param family the chart family
  357. * @param move the pointer with the values that will be published
  358. * @param end the number of values that will be written on standard output
  359. */
  360. void write_err_chart(char *name, char *family, netdata_publish_syscall_t *move, int end)
  361. {
  362. write_begin_chart(family, name);
  363. int i = 0;
  364. while (move && i < end) {
  365. write_chart_dimension(move->name, move->nerr);
  366. move = move->next;
  367. i++;
  368. }
  369. write_end_chart();
  370. }
  371. /**
  372. * Write charts
  373. *
  374. * Write the current information to publish the charts.
  375. *
  376. * @param family chart family
  377. * @param chart chart id
  378. * @param dim dimension name
  379. * @param v1 value.
  380. */
  381. void ebpf_one_dimension_write_charts(char *family, char *chart, char *dim, long long v1)
  382. {
  383. write_begin_chart(family, chart);
  384. write_chart_dimension(dim, v1);
  385. write_end_chart();
  386. }
  387. /**
  388. * Call the necessary functions to create a chart.
  389. *
  390. * @param chart the chart name
  391. * @param family the chart family
  392. * @param dwrite the dimension name
  393. * @param vwrite the value for previous dimension
  394. * @param dread the dimension name
  395. * @param vread the value for previous dimension
  396. *
  397. * @return It returns a variable that maps the charts that did not have zero values.
  398. */
  399. void write_io_chart(char *chart, char *family, char *dwrite, long long vwrite, char *dread, long long vread)
  400. {
  401. write_begin_chart(family, chart);
  402. write_chart_dimension(dwrite, vwrite);
  403. write_chart_dimension(dread, vread);
  404. write_end_chart();
  405. }
  406. /**
  407. * Write chart cmd on standard output
  408. *
  409. * @param type chart type
  410. * @param id chart id
  411. * @param title chart title
  412. * @param units units label
  413. * @param family group name used to attach the chart on dashboard
  414. * @param charttype chart type
  415. * @param context chart context
  416. * @param order chart order
  417. * @param update_every update interval used by plugin
  418. * @param module chart module name, this is the eBPF thread.
  419. */
  420. void ebpf_write_chart_cmd(char *type, char *id, char *title, char *units, char *family,
  421. char *charttype, char *context, int order, int update_every, char *module)
  422. {
  423. printf("CHART %s.%s '' '%s' '%s' '%s' '%s' '%s' %d %d '' 'ebpf.plugin' '%s'\n",
  424. type,
  425. id,
  426. title,
  427. units,
  428. (family)?family:"",
  429. (context)?context:"",
  430. (charttype)?charttype:"",
  431. order,
  432. update_every,
  433. module);
  434. }
  435. /**
  436. * Write chart cmd on standard output
  437. *
  438. * @param type chart type
  439. * @param id chart id
  440. * @param title chart title
  441. * @param units units label
  442. * @param family group name used to attach the chart on dashboard
  443. * @param charttype chart type
  444. * @param context chart context
  445. * @param order chart order
  446. * @param update_every value to overwrite the update frequency set by the server.
  447. */
  448. void ebpf_write_chart_obsolete(char *type, char *id, char *title, char *units, char *family,
  449. char *charttype, char *context, int order, int update_every)
  450. {
  451. printf("CHART %s.%s '' '%s' '%s' '%s' '%s' '%s' %d %d 'obsolete'\n",
  452. type,
  453. id,
  454. title,
  455. units,
  456. (family)?family:"",
  457. (context)?context:"",
  458. (charttype)?charttype:"",
  459. order,
  460. update_every);
  461. }
  462. /**
  463. * Write the dimension command on standard output
  464. *
  465. * @param name the dimension name
  466. * @param id the dimension id
  467. * @param algo the dimension algorithm
  468. */
  469. void ebpf_write_global_dimension(char *name, char *id, char *algorithm)
  470. {
  471. printf("DIMENSION %s %s %s 1 1\n", name, id, algorithm);
  472. }
  473. /**
  474. * Call ebpf_write_global_dimension to create the dimensions for a specific chart
  475. *
  476. * @param ptr a pointer to a structure of the type netdata_publish_syscall_t
  477. * @param end the number of dimensions for the structure ptr
  478. */
  479. void ebpf_create_global_dimension(void *ptr, int end)
  480. {
  481. netdata_publish_syscall_t *move = ptr;
  482. int i = 0;
  483. while (move && i < end) {
  484. ebpf_write_global_dimension(move->name, move->dimension, move->algorithm);
  485. move = move->next;
  486. i++;
  487. }
  488. }
  489. /**
  490. * Call write_chart_cmd to create the charts
  491. *
  492. * @param type chart type
  493. * @param id chart id
  494. * @param title chart title
  495. * @param units axis label
  496. * @param family group name used to attach the chart on dashboard
  497. * @param context chart context
  498. * @param charttype chart type
  499. * @param order order number of the specified chart
  500. * @param ncd a pointer to a function called to create dimensions
  501. * @param move a pointer for a structure that has the dimensions
  502. * @param end number of dimensions for the chart created
  503. * @param update_every update interval used with chart.
  504. * @param module chart module name, this is the eBPF thread.
  505. */
  506. void ebpf_create_chart(char *type,
  507. char *id,
  508. char *title,
  509. char *units,
  510. char *family,
  511. char *context,
  512. char *charttype,
  513. int order,
  514. void (*ncd)(void *, int),
  515. void *move,
  516. int end,
  517. int update_every,
  518. char *module)
  519. {
  520. ebpf_write_chart_cmd(type, id, title, units, family, charttype, context, order, update_every, module);
  521. if (ncd) {
  522. ncd(move, end);
  523. }
  524. }
  525. /**
  526. * Create charts on apps submenu
  527. *
  528. * @param id the chart id
  529. * @param title the value displayed on vertical axis.
  530. * @param units the value displayed on vertical axis.
  531. * @param family Submenu that the chart will be attached on dashboard.
  532. * @param charttype chart type
  533. * @param order the chart order
  534. * @param algorithm the algorithm used by dimension
  535. * @param root structure used to create the dimensions.
  536. * @param update_every update interval used by plugin
  537. * @param module chart module name, this is the eBPF thread.
  538. */
  539. void ebpf_create_charts_on_apps(char *id, char *title, char *units, char *family, char *charttype, int order,
  540. char *algorithm, struct target *root, int update_every, char *module)
  541. {
  542. struct target *w;
  543. ebpf_write_chart_cmd(NETDATA_APPS_FAMILY, id, title, units, family, charttype, NULL, order,
  544. update_every, module);
  545. for (w = root; w; w = w->next) {
  546. if (unlikely(w->exposed))
  547. fprintf(stdout, "DIMENSION %s '' %s 1 1\n", w->name, algorithm);
  548. }
  549. }
  550. /**
  551. * Call the necessary functions to create a name.
  552. *
  553. * @param family family name
  554. * @param name chart name
  555. * @param hist0 histogram values
  556. * @param dimensions dimension values.
  557. * @param end number of bins that will be sent to Netdata.
  558. *
  559. * @return It returns a variable that maps the charts that did not have zero values.
  560. */
  561. void write_histogram_chart(char *family, char *name, const netdata_idx_t *hist, char **dimensions, uint32_t end)
  562. {
  563. write_begin_chart(family, name);
  564. uint32_t i;
  565. for (i = 0; i < end; i++) {
  566. write_chart_dimension(dimensions[i], (long long) hist[i]);
  567. }
  568. write_end_chart();
  569. fflush(stdout);
  570. }
  571. /*****************************************************************
  572. *
  573. * FUNCTIONS TO DEFINE OPTIONS
  574. *
  575. *****************************************************************/
  576. /**
  577. * Define labels used to generate charts
  578. *
  579. * @param is structure with information about number of calls made for a function.
  580. * @param pio structure used to generate charts.
  581. * @param dim a pointer for the dimensions name
  582. * @param name a pointer for the tensor with the name of the functions.
  583. * @param algorithm a vector with the algorithms used to make the charts
  584. * @param end the number of elements in the previous 4 arguments.
  585. */
  586. void ebpf_global_labels(netdata_syscall_stat_t *is, netdata_publish_syscall_t *pio, char **dim,
  587. char **name, int *algorithm, int end)
  588. {
  589. int i;
  590. netdata_syscall_stat_t *prev = NULL;
  591. netdata_publish_syscall_t *publish_prev = NULL;
  592. for (i = 0; i < end; i++) {
  593. if (prev) {
  594. prev->next = &is[i];
  595. }
  596. prev = &is[i];
  597. pio[i].dimension = dim[i];
  598. pio[i].name = name[i];
  599. pio[i].algorithm = strdupz(ebpf_algorithms[algorithm[i]]);
  600. if (publish_prev) {
  601. publish_prev->next = &pio[i];
  602. }
  603. publish_prev = &pio[i];
  604. }
  605. }
  606. /**
  607. * Define thread mode for all ebpf program.
  608. *
  609. * @param lmode the mode that will be used for them.
  610. */
  611. static inline void ebpf_set_thread_mode(netdata_run_mode_t lmode)
  612. {
  613. int i;
  614. for (i = 0; ebpf_modules[i].thread_name; i++) {
  615. ebpf_modules[i].mode = lmode;
  616. }
  617. }
  618. /**
  619. * Enable specific charts selected by user.
  620. *
  621. * @param em the structure that will be changed
  622. * @param disable_apps the status about the apps charts.
  623. * @param disable_cgroup the status about the cgroups charts.
  624. */
  625. static inline void ebpf_enable_specific_chart(struct ebpf_module *em, int disable_apps, int disable_cgroup)
  626. {
  627. em->enabled = CONFIG_BOOLEAN_YES;
  628. // oomkill stores data inside apps submenu, so it always need to have apps_enabled for plugin to create
  629. // its chart, without this comparison eBPF.plugin will try to store invalid data when apps is disabled.
  630. if (!disable_apps || !strcmp(em->thread_name, "oomkill")) {
  631. em->apps_charts = CONFIG_BOOLEAN_YES;
  632. }
  633. if (!disable_cgroup) {
  634. em->cgroup_charts = CONFIG_BOOLEAN_YES;
  635. }
  636. em->global_charts = CONFIG_BOOLEAN_YES;
  637. }
  638. /**
  639. * Enable all charts
  640. *
  641. * @param apps what is the current status of apps
  642. * @param cgroups what is the current status of cgroups
  643. */
  644. static inline void ebpf_enable_all_charts(int apps, int cgroups)
  645. {
  646. int i;
  647. for (i = 0; ebpf_modules[i].thread_name; i++) {
  648. ebpf_enable_specific_chart(&ebpf_modules[i], apps, cgroups);
  649. }
  650. }
  651. /**
  652. * Disable all Global charts
  653. *
  654. * Disable charts
  655. */
  656. static inline void disable_all_global_charts()
  657. {
  658. int i;
  659. for (i = 0; ebpf_modules[i].thread_name; i++) {
  660. ebpf_modules[i].enabled = 0;
  661. ebpf_modules[i].global_charts = 0;
  662. }
  663. }
  664. /**
  665. * Enable the specified chart group
  666. *
  667. * @param idx the index of ebpf_modules that I am enabling
  668. * @param disable_apps should I keep apps charts?
  669. */
  670. static inline void ebpf_enable_chart(int idx, int disable_apps, int disable_cgroup)
  671. {
  672. int i;
  673. for (i = 0; ebpf_modules[i].thread_name; i++) {
  674. if (i == idx) {
  675. ebpf_enable_specific_chart(&ebpf_modules[i], disable_apps, disable_cgroup);
  676. break;
  677. }
  678. }
  679. }
  680. /**
  681. * Disable APPs
  682. *
  683. * Disable charts for apps loading only global charts.
  684. */
  685. static inline void ebpf_disable_apps()
  686. {
  687. int i;
  688. for (i = 0; ebpf_modules[i].thread_name; i++) {
  689. ebpf_modules[i].apps_charts = 0;
  690. }
  691. }
  692. /**
  693. * Disable Cgroups
  694. *
  695. * Disable charts for apps loading only global charts.
  696. */
  697. static inline void ebpf_disable_cgroups()
  698. {
  699. int i;
  700. for (i = 0; ebpf_modules[i].thread_name; i++) {
  701. ebpf_modules[i].cgroup_charts = 0;
  702. }
  703. }
  704. /**
  705. * Update Disabled Plugins
  706. *
  707. * This function calls ebpf_update_stats to update statistics for collector.
  708. *
  709. * @param em a pointer to `struct ebpf_module`
  710. */
  711. void ebpf_update_disabled_plugin_stats(ebpf_module_t *em)
  712. {
  713. pthread_mutex_lock(&lock);
  714. ebpf_update_stats(&plugin_statistics, em);
  715. pthread_mutex_unlock(&lock);
  716. }
  717. /**
  718. * Print help on standard error for user knows how to use the collector.
  719. */
  720. void ebpf_print_help()
  721. {
  722. const time_t t = time(NULL);
  723. struct tm ct;
  724. struct tm *test = localtime_r(&t, &ct);
  725. int year;
  726. if (test)
  727. year = ct.tm_year;
  728. else
  729. year = 0;
  730. fprintf(stderr,
  731. "\n"
  732. " Netdata ebpf.plugin %s\n"
  733. " Copyright (C) 2016-%d Costa Tsaousis <costa@tsaousis.gr>\n"
  734. " Released under GNU General Public License v3 or later.\n"
  735. " All rights reserved.\n"
  736. "\n"
  737. " This eBPF.plugin is a data collector plugin for netdata.\n"
  738. "\n"
  739. " This plugin only accepts long options with one or two dashes. The available command line options are:\n"
  740. "\n"
  741. " SECONDS Set the data collection frequency.\n"
  742. "\n"
  743. " [-]-help Show this help.\n"
  744. "\n"
  745. " [-]-version Show software version.\n"
  746. "\n"
  747. " [-]-global Disable charts per application and cgroup.\n"
  748. "\n"
  749. " [-]-all Enable all chart groups (global, apps, and cgroup), unless -g is also given.\n"
  750. "\n"
  751. " [-]-cachestat Enable charts related to process run time.\n"
  752. "\n"
  753. " [-]-dcstat Enable charts related to directory cache.\n"
  754. "\n"
  755. " [-]-disk Enable charts related to disk monitoring.\n"
  756. "\n"
  757. " [-]-filesystem Enable chart related to filesystem run time.\n"
  758. "\n"
  759. " [-]-hardirq Enable chart related to hard IRQ latency.\n"
  760. "\n"
  761. " [-]-mdflush Enable charts related to multi-device flush.\n"
  762. "\n"
  763. " [-]-mount Enable charts related to mount monitoring.\n"
  764. "\n"
  765. " [-]-net Enable network viewer charts.\n"
  766. "\n"
  767. " [-]-oomkill Enable chart related to OOM kill tracking.\n"
  768. "\n"
  769. " [-]-process Enable charts related to process run time.\n"
  770. "\n"
  771. " [-]-return Run the collector in return mode.\n"
  772. "\n"
  773. " [-]-shm Enable chart related to shared memory tracking.\n"
  774. "\n"
  775. " [-]-softirq Enable chart related to soft IRQ latency.\n"
  776. "\n"
  777. " [-]-sync Enable chart related to sync run time.\n"
  778. "\n"
  779. " [-]-swap Enable chart related to swap run time.\n"
  780. "\n"
  781. " [-]-vfs Enable chart related to vfs run time.\n"
  782. "\n"
  783. " [-]-legacy Load legacy eBPF programs.\n"
  784. "\n"
  785. " [-]-core Use CO-RE when available(Working in progress).\n"
  786. "\n",
  787. VERSION,
  788. (year >= 116) ? year + 1900 : 2020);
  789. }
  790. /*****************************************************************
  791. *
  792. * TRACEPOINT MANAGEMENT FUNCTIONS
  793. *
  794. *****************************************************************/
  795. /**
  796. * Enable a tracepoint.
  797. *
  798. * @return 0 on success, -1 on error.
  799. */
  800. int ebpf_enable_tracepoint(ebpf_tracepoint_t *tp)
  801. {
  802. int test = ebpf_is_tracepoint_enabled(tp->class, tp->event);
  803. // err?
  804. if (test == -1) {
  805. return -1;
  806. }
  807. // disabled?
  808. else if (test == 0) {
  809. // enable it then.
  810. if (ebpf_enable_tracing_values(tp->class, tp->event)) {
  811. return -1;
  812. }
  813. }
  814. // enabled now or already was.
  815. tp->enabled = true;
  816. return 0;
  817. }
  818. /**
  819. * Disable a tracepoint if it's enabled.
  820. *
  821. * @return 0 on success, -1 on error.
  822. */
  823. int ebpf_disable_tracepoint(ebpf_tracepoint_t *tp)
  824. {
  825. int test = ebpf_is_tracepoint_enabled(tp->class, tp->event);
  826. // err?
  827. if (test == -1) {
  828. return -1;
  829. }
  830. // enabled?
  831. else if (test == 1) {
  832. // disable it then.
  833. if (ebpf_disable_tracing_values(tp->class, tp->event)) {
  834. return -1;
  835. }
  836. }
  837. // disable now or already was.
  838. tp->enabled = false;
  839. return 0;
  840. }
  841. /**
  842. * Enable multiple tracepoints on a list of tracepoints which end when the
  843. * class is NULL.
  844. *
  845. * @return the number of successful enables.
  846. */
  847. uint32_t ebpf_enable_tracepoints(ebpf_tracepoint_t *tps)
  848. {
  849. uint32_t cnt = 0;
  850. for (int i = 0; tps[i].class != NULL; i++) {
  851. if (ebpf_enable_tracepoint(&tps[i]) == -1) {
  852. infoerr("failed to enable tracepoint %s:%s",
  853. tps[i].class, tps[i].event);
  854. }
  855. else {
  856. cnt += 1;
  857. }
  858. }
  859. return cnt;
  860. }
  861. /*****************************************************************
  862. *
  863. * AUXILIARY FUNCTIONS USED DURING INITIALIZATION
  864. *
  865. *****************************************************************/
  866. /**
  867. * Read Local Ports
  868. *
  869. * Parse /proc/net/{tcp,udp} and get the ports Linux is listening.
  870. *
  871. * @param filename the proc file to parse.
  872. * @param proto is the magic number associated to the protocol file we are reading.
  873. */
  874. static void read_local_ports(char *filename, uint8_t proto)
  875. {
  876. procfile *ff = procfile_open(filename, " \t:", PROCFILE_FLAG_DEFAULT);
  877. if (!ff)
  878. return;
  879. ff = procfile_readall(ff);
  880. if (!ff)
  881. return;
  882. size_t lines = procfile_lines(ff), l;
  883. netdata_passive_connection_t values = {.counter = 0, .tgid = 0, .pid = 0};
  884. for(l = 0; l < lines ;l++) {
  885. size_t words = procfile_linewords(ff, l);
  886. // This is header or end of file
  887. if (unlikely(words < 14))
  888. continue;
  889. // https://elixir.bootlin.com/linux/v5.7.8/source/include/net/tcp_states.h
  890. // 0A = TCP_LISTEN
  891. if (strcmp("0A", procfile_lineword(ff, l, 5)))
  892. continue;
  893. // Read local port
  894. uint16_t port = (uint16_t)strtol(procfile_lineword(ff, l, 2), NULL, 16);
  895. update_listen_table(htons(port), proto, &values);
  896. }
  897. procfile_close(ff);
  898. }
  899. /**
  900. * Read Local addresseses
  901. *
  902. * Read the local address from the interfaces.
  903. */
  904. static void read_local_addresses()
  905. {
  906. struct ifaddrs *ifaddr, *ifa;
  907. if (getifaddrs(&ifaddr) == -1) {
  908. error("Cannot get the local IP addresses, it is no possible to do separation between inbound and outbound connections");
  909. return;
  910. }
  911. char *notext = { "No text representation" };
  912. for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
  913. if (ifa->ifa_addr == NULL)
  914. continue;
  915. if ((ifa->ifa_addr->sa_family != AF_INET) && (ifa->ifa_addr->sa_family != AF_INET6))
  916. continue;
  917. ebpf_network_viewer_ip_list_t *w = callocz(1, sizeof(ebpf_network_viewer_ip_list_t));
  918. int family = ifa->ifa_addr->sa_family;
  919. w->ver = (uint8_t) family;
  920. char text[INET6_ADDRSTRLEN];
  921. if (family == AF_INET) {
  922. struct sockaddr_in *in = (struct sockaddr_in*) ifa->ifa_addr;
  923. w->first.addr32[0] = in->sin_addr.s_addr;
  924. w->last.addr32[0] = in->sin_addr.s_addr;
  925. if (inet_ntop(AF_INET, w->first.addr8, text, INET_ADDRSTRLEN)) {
  926. w->value = strdupz(text);
  927. w->hash = simple_hash(text);
  928. } else {
  929. w->value = strdupz(notext);
  930. w->hash = simple_hash(notext);
  931. }
  932. } else {
  933. struct sockaddr_in6 *in6 = (struct sockaddr_in6*) ifa->ifa_addr;
  934. memcpy(w->first.addr8, (void *)&in6->sin6_addr, sizeof(struct in6_addr));
  935. memcpy(w->last.addr8, (void *)&in6->sin6_addr, sizeof(struct in6_addr));
  936. if (inet_ntop(AF_INET6, w->first.addr8, text, INET_ADDRSTRLEN)) {
  937. w->value = strdupz(text);
  938. w->hash = simple_hash(text);
  939. } else {
  940. w->value = strdupz(notext);
  941. w->hash = simple_hash(notext);
  942. }
  943. }
  944. fill_ip_list((family == AF_INET)?&network_viewer_opt.ipv4_local_ip:&network_viewer_opt.ipv6_local_ip,
  945. w,
  946. "selector");
  947. }
  948. freeifaddrs(ifaddr);
  949. }
  950. /**
  951. * Start Pthread Variable
  952. *
  953. * This function starts all pthread variables.
  954. *
  955. * @return It returns 0 on success and -1.
  956. */
  957. int ebpf_start_pthread_variables()
  958. {
  959. pthread_mutex_init(&lock, NULL);
  960. pthread_mutex_init(&collect_data_mutex, NULL);
  961. if (pthread_cond_init(&collect_data_cond_var, NULL)) {
  962. thread_finished++;
  963. error("Cannot start conditional variable to control Apps charts.");
  964. return -1;
  965. }
  966. return 0;
  967. }
  968. /**
  969. * Allocate the vectors used for all threads.
  970. */
  971. static void ebpf_allocate_common_vectors()
  972. {
  973. all_pids = callocz((size_t)pid_max, sizeof(struct pid_stat *));
  974. global_process_stat = callocz((size_t)ebpf_nprocs, sizeof(ebpf_process_stat_t));
  975. }
  976. /**
  977. * Define how to load the ebpf programs
  978. *
  979. * @param ptr the option given by users
  980. */
  981. static inline void how_to_load(char *ptr)
  982. {
  983. if (!strcasecmp(ptr, EBPF_CFG_LOAD_MODE_RETURN))
  984. ebpf_set_thread_mode(MODE_RETURN);
  985. else if (!strcasecmp(ptr, EBPF_CFG_LOAD_MODE_DEFAULT))
  986. ebpf_set_thread_mode(MODE_ENTRY);
  987. else
  988. error("the option %s for \"ebpf load mode\" is not a valid option.", ptr);
  989. }
  990. /**
  991. * Update interval
  992. *
  993. * Update default interval with value from user
  994. *
  995. * @param update_every value to overwrite the update frequency set by the server.
  996. */
  997. static void ebpf_update_interval(int update_every)
  998. {
  999. int i;
  1000. int value = (int) appconfig_get_number(&collector_config, EBPF_GLOBAL_SECTION, EBPF_CFG_UPDATE_EVERY,
  1001. update_every);
  1002. for (i = 0; ebpf_modules[i].thread_name; i++) {
  1003. ebpf_modules[i].update_every = value;
  1004. }
  1005. }
  1006. /**
  1007. * Update PID table size
  1008. *
  1009. * Update default size with value from user
  1010. */
  1011. static void ebpf_update_table_size()
  1012. {
  1013. int i;
  1014. uint32_t value = (uint32_t) appconfig_get_number(&collector_config, EBPF_GLOBAL_SECTION,
  1015. EBPF_CFG_PID_SIZE, ND_EBPF_DEFAULT_PID_SIZE);
  1016. for (i = 0; ebpf_modules[i].thread_name; i++) {
  1017. ebpf_modules[i].pid_map_size = value;
  1018. }
  1019. }
  1020. /**
  1021. * Set Load mode
  1022. *
  1023. * @param load default load mode.
  1024. */
  1025. static inline void ebpf_set_load_mode(netdata_ebpf_load_mode_t load)
  1026. {
  1027. #ifdef LIBBPF_MAJOR_VERSION
  1028. if (load == EBPF_LOAD_CORE || load == EBPF_LOAD_PLAY_DICE) {
  1029. load = (!default_btf) ? EBPF_LOAD_LEGACY : EBPF_LOAD_CORE;
  1030. }
  1031. #else
  1032. load = EBPF_LOAD_LEGACY;
  1033. #endif
  1034. int i;
  1035. for (i = 0; ebpf_modules[i].thread_name; i++) {
  1036. // TO DO: Use `load` variable after we change all threads.
  1037. ebpf_modules[i].load = EBPF_LOAD_LEGACY; // load ;
  1038. }
  1039. }
  1040. /**
  1041. * Update mode
  1042. *
  1043. * @param str value read from configuration file.
  1044. */
  1045. static inline void epbf_update_load_mode(char *str)
  1046. {
  1047. netdata_ebpf_load_mode_t load = epbf_convert_string_to_load_mode(str);
  1048. ebpf_set_load_mode(load);
  1049. }
  1050. #ifdef LIBBPF_MAJOR_VERSION
  1051. /**
  1052. * Set default btf file
  1053. *
  1054. * Load the default BTF file on environment.
  1055. */
  1056. static void ebpf_set_default_btf_file()
  1057. {
  1058. char path[PATH_MAX + 1];
  1059. snprintfz(path, PATH_MAX, "%s/vmlinux", btf_path);
  1060. default_btf = ebpf_parse_btf_file(path);
  1061. if (!default_btf)
  1062. info("Your environment does not have BTF file %s/vmlinux. The plugin will work with 'legacy' code.",
  1063. btf_path);
  1064. }
  1065. #endif
  1066. /**
  1067. * Read collector values
  1068. *
  1069. * @param disable_apps variable to store information related to apps.
  1070. * @param disable_cgroups variable to store information related to cgroups.
  1071. * @param update_every value to overwrite the update frequency set by the server.
  1072. */
  1073. static void read_collector_values(int *disable_apps, int *disable_cgroups, int update_every)
  1074. {
  1075. // Read global section
  1076. char *value;
  1077. if (appconfig_exists(&collector_config, EBPF_GLOBAL_SECTION, "load")) // Backward compatibility
  1078. value = appconfig_get(&collector_config, EBPF_GLOBAL_SECTION, "load",
  1079. EBPF_CFG_LOAD_MODE_DEFAULT);
  1080. else
  1081. value = appconfig_get(&collector_config, EBPF_GLOBAL_SECTION, EBPF_CFG_LOAD_MODE,
  1082. EBPF_CFG_LOAD_MODE_DEFAULT);
  1083. how_to_load(value);
  1084. btf_path = appconfig_get(&collector_config, EBPF_GLOBAL_SECTION, EBPF_CFG_PROGRAM_PATH,
  1085. EBPF_DEFAULT_BTF_FILE);
  1086. #ifdef LIBBPF_MAJOR_VERSION
  1087. ebpf_set_default_btf_file();
  1088. #endif
  1089. value = appconfig_get(&collector_config, EBPF_GLOBAL_SECTION, EBPF_CFG_TYPE_FORMAT, EBPF_CFG_DEFAULT_PROGRAM);
  1090. epbf_update_load_mode(value);
  1091. ebpf_update_interval(update_every);
  1092. ebpf_update_table_size();
  1093. // This is kept to keep compatibility
  1094. uint32_t enabled = appconfig_get_boolean(&collector_config, EBPF_GLOBAL_SECTION, "disable apps",
  1095. CONFIG_BOOLEAN_NO);
  1096. if (!enabled) {
  1097. // Apps is a positive sentence, so we need to invert the values to disable apps.
  1098. enabled = appconfig_get_boolean(&collector_config, EBPF_GLOBAL_SECTION, EBPF_CFG_APPLICATION,
  1099. CONFIG_BOOLEAN_YES);
  1100. enabled = (enabled == CONFIG_BOOLEAN_NO)?CONFIG_BOOLEAN_YES:CONFIG_BOOLEAN_NO;
  1101. }
  1102. *disable_apps = (int)enabled;
  1103. // Cgroup is a positive sentence, so we need to invert the values to disable apps.
  1104. // We are using the same pattern for cgroup and apps
  1105. enabled = appconfig_get_boolean(&collector_config, EBPF_GLOBAL_SECTION, EBPF_CFG_CGROUP, CONFIG_BOOLEAN_NO);
  1106. *disable_cgroups = (enabled == CONFIG_BOOLEAN_NO)?CONFIG_BOOLEAN_YES:CONFIG_BOOLEAN_NO;
  1107. // Read ebpf programs section
  1108. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION,
  1109. ebpf_modules[EBPF_MODULE_PROCESS_IDX].config_name, CONFIG_BOOLEAN_YES);
  1110. int started = 0;
  1111. if (enabled) {
  1112. ebpf_enable_chart(EBPF_MODULE_PROCESS_IDX, *disable_apps, *disable_cgroups);
  1113. started++;
  1114. }
  1115. // This is kept to keep compatibility
  1116. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "network viewer",
  1117. CONFIG_BOOLEAN_NO);
  1118. if (!enabled)
  1119. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION,
  1120. ebpf_modules[EBPF_MODULE_SOCKET_IDX].config_name,
  1121. CONFIG_BOOLEAN_NO);
  1122. if (enabled) {
  1123. ebpf_enable_chart(EBPF_MODULE_SOCKET_IDX, *disable_apps, *disable_cgroups);
  1124. // Read network viewer section if network viewer is enabled
  1125. // This is kept here to keep backward compatibility
  1126. parse_network_viewer_section(&collector_config);
  1127. parse_service_name_section(&collector_config);
  1128. started++;
  1129. }
  1130. // This is kept to keep compatibility
  1131. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "network connection monitoring",
  1132. CONFIG_BOOLEAN_NO);
  1133. if (!enabled)
  1134. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "network connections",
  1135. CONFIG_BOOLEAN_NO);
  1136. ebpf_modules[EBPF_MODULE_SOCKET_IDX].optional = (int)enabled;
  1137. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "cachestat",
  1138. CONFIG_BOOLEAN_NO);
  1139. if (enabled) {
  1140. ebpf_enable_chart(EBPF_MODULE_CACHESTAT_IDX, *disable_apps, *disable_cgroups);
  1141. started++;
  1142. }
  1143. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "sync",
  1144. CONFIG_BOOLEAN_YES);
  1145. if (enabled) {
  1146. ebpf_enable_chart(EBPF_MODULE_SYNC_IDX, *disable_apps, *disable_cgroups);
  1147. started++;
  1148. }
  1149. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "dcstat",
  1150. CONFIG_BOOLEAN_NO);
  1151. if (enabled) {
  1152. ebpf_enable_chart(EBPF_MODULE_DCSTAT_IDX, *disable_apps, *disable_cgroups);
  1153. started++;
  1154. }
  1155. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "swap",
  1156. CONFIG_BOOLEAN_NO);
  1157. if (enabled) {
  1158. ebpf_enable_chart(EBPF_MODULE_SWAP_IDX, *disable_apps, *disable_cgroups);
  1159. started++;
  1160. }
  1161. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "vfs",
  1162. CONFIG_BOOLEAN_NO);
  1163. if (enabled) {
  1164. ebpf_enable_chart(EBPF_MODULE_VFS_IDX, *disable_apps, *disable_cgroups);
  1165. started++;
  1166. }
  1167. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "filesystem",
  1168. CONFIG_BOOLEAN_NO);
  1169. if (enabled) {
  1170. ebpf_enable_chart(EBPF_MODULE_FILESYSTEM_IDX, *disable_apps, *disable_cgroups);
  1171. started++;
  1172. }
  1173. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "disk",
  1174. CONFIG_BOOLEAN_NO);
  1175. if (enabled) {
  1176. ebpf_enable_chart(EBPF_MODULE_DISK_IDX, *disable_apps, *disable_cgroups);
  1177. started++;
  1178. }
  1179. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "mount",
  1180. CONFIG_BOOLEAN_YES);
  1181. if (enabled) {
  1182. ebpf_enable_chart(EBPF_MODULE_MOUNT_IDX, *disable_apps, *disable_cgroups);
  1183. started++;
  1184. }
  1185. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "fd",
  1186. CONFIG_BOOLEAN_YES);
  1187. if (enabled) {
  1188. ebpf_enable_chart(EBPF_MODULE_FD_IDX, *disable_apps, *disable_cgroups);
  1189. started++;
  1190. }
  1191. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "hardirq",
  1192. CONFIG_BOOLEAN_YES);
  1193. if (enabled) {
  1194. ebpf_enable_chart(EBPF_MODULE_HARDIRQ_IDX, *disable_apps, *disable_cgroups);
  1195. started++;
  1196. }
  1197. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "softirq",
  1198. CONFIG_BOOLEAN_YES);
  1199. if (enabled) {
  1200. ebpf_enable_chart(EBPF_MODULE_SOFTIRQ_IDX, *disable_apps, *disable_cgroups);
  1201. started++;
  1202. }
  1203. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "oomkill",
  1204. CONFIG_BOOLEAN_YES);
  1205. if (enabled) {
  1206. ebpf_enable_chart(EBPF_MODULE_OOMKILL_IDX, *disable_apps, *disable_cgroups);
  1207. started++;
  1208. }
  1209. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "shm",
  1210. CONFIG_BOOLEAN_YES);
  1211. if (enabled) {
  1212. ebpf_enable_chart(EBPF_MODULE_SHM_IDX, *disable_apps, *disable_cgroups);
  1213. started++;
  1214. }
  1215. enabled = appconfig_get_boolean(&collector_config, EBPF_PROGRAMS_SECTION, "mdflush",
  1216. CONFIG_BOOLEAN_NO);
  1217. if (enabled) {
  1218. ebpf_enable_chart(EBPF_MODULE_MDFLUSH_IDX, *disable_apps, *disable_cgroups);
  1219. started++;
  1220. }
  1221. if (!started){
  1222. ebpf_enable_all_charts(*disable_apps, *disable_cgroups);
  1223. // Read network viewer section
  1224. // This is kept here to keep backward compatibility
  1225. parse_network_viewer_section(&collector_config);
  1226. parse_service_name_section(&collector_config);
  1227. }
  1228. }
  1229. /**
  1230. * Load collector config
  1231. *
  1232. * @param path the path where the file ebpf.conf is stored.
  1233. * @param disable_apps variable to store the information about apps plugin status.
  1234. * @param disable_cgroups variable to store the information about cgroups plugin status.
  1235. * @param update_every value to overwrite the update frequency set by the server.
  1236. *
  1237. * @return 0 on success and -1 otherwise.
  1238. */
  1239. static int load_collector_config(char *path, int *disable_apps, int *disable_cgroups, int update_every)
  1240. {
  1241. char lpath[4096];
  1242. snprintf(lpath, 4095, "%s/%s", path, NETDATA_EBPF_CONFIG_FILE);
  1243. if (!appconfig_load(&collector_config, lpath, 0, NULL)) {
  1244. snprintf(lpath, 4095, "%s/%s", path, NETDATA_EBPF_OLD_CONFIG_FILE);
  1245. if (!appconfig_load(&collector_config, lpath, 0, NULL)) {
  1246. return -1;
  1247. }
  1248. }
  1249. read_collector_values(disable_apps, disable_cgroups, update_every);
  1250. return 0;
  1251. }
  1252. /**
  1253. * Set global variables reading environment variables
  1254. */
  1255. void set_global_variables()
  1256. {
  1257. // Get environment variables
  1258. ebpf_plugin_dir = getenv("NETDATA_PLUGINS_DIR");
  1259. if (!ebpf_plugin_dir)
  1260. ebpf_plugin_dir = PLUGINS_DIR;
  1261. ebpf_user_config_dir = getenv("NETDATA_USER_CONFIG_DIR");
  1262. if (!ebpf_user_config_dir)
  1263. ebpf_user_config_dir = CONFIG_DIR;
  1264. ebpf_stock_config_dir = getenv("NETDATA_STOCK_CONFIG_DIR");
  1265. if (!ebpf_stock_config_dir)
  1266. ebpf_stock_config_dir = LIBCONFIG_DIR;
  1267. ebpf_configured_log_dir = getenv("NETDATA_LOG_DIR");
  1268. if (!ebpf_configured_log_dir)
  1269. ebpf_configured_log_dir = LOG_DIR;
  1270. ebpf_nprocs = (int)sysconf(_SC_NPROCESSORS_ONLN);
  1271. if (ebpf_nprocs > NETDATA_MAX_PROCESSOR) {
  1272. ebpf_nprocs = NETDATA_MAX_PROCESSOR;
  1273. }
  1274. isrh = get_redhat_release();
  1275. pid_max = get_system_pid_max();
  1276. running_on_kernel = ebpf_get_kernel_version();
  1277. }
  1278. /**
  1279. * Load collector config
  1280. *
  1281. * @param lmode the mode that will be used for them.
  1282. */
  1283. static inline void ebpf_load_thread_config()
  1284. {
  1285. int i;
  1286. for (i = 0; ebpf_modules[i].thread_name; i++) {
  1287. ebpf_update_module(&ebpf_modules[i]);
  1288. }
  1289. }
  1290. /**
  1291. * Parse arguments given from user.
  1292. *
  1293. * @param argc the number of arguments
  1294. * @param argv the pointer to the arguments
  1295. */
  1296. static void ebpf_parse_args(int argc, char **argv)
  1297. {
  1298. int disable_apps = 0;
  1299. int disable_cgroups = 1;
  1300. int freq = 0;
  1301. int option_index = 0;
  1302. uint64_t select_threads = 0;
  1303. static struct option long_options[] = {
  1304. {"process", no_argument, 0, 0 },
  1305. {"net", no_argument, 0, 0 },
  1306. {"cachestat", no_argument, 0, 0 },
  1307. {"sync", no_argument, 0, 0 },
  1308. {"dcstat", no_argument, 0, 0 },
  1309. {"swap", no_argument, 0, 0 },
  1310. {"vfs", no_argument, 0, 0 },
  1311. {"filesystem", no_argument, 0, 0 },
  1312. {"disk", no_argument, 0, 0 },
  1313. {"mount", no_argument, 0, 0 },
  1314. {"filedescriptor", no_argument, 0, 0 },
  1315. {"hardirq", no_argument, 0, 0 },
  1316. {"softirq", no_argument, 0, 0 },
  1317. {"oomkill", no_argument, 0, 0 },
  1318. {"shm", no_argument, 0, 0 },
  1319. {"mdflush", no_argument, 0, 0 },
  1320. /* INSERT NEW THREADS BEFORE THIS COMMENT TO KEEP COMPATIBILITY WITH enum ebpf_module_indexes */
  1321. {"all", no_argument, 0, 0 },
  1322. {"version", no_argument, 0, 0 },
  1323. {"help", no_argument, 0, 0 },
  1324. {"global", no_argument, 0, 0 },
  1325. {"return", no_argument, 0, 0 },
  1326. {"legacy", no_argument, 0, 0 },
  1327. {"core", no_argument, 0, 0 },
  1328. {0, 0, 0, 0}
  1329. };
  1330. memset(&network_viewer_opt, 0, sizeof(network_viewer_opt));
  1331. network_viewer_opt.max_dim = NETDATA_NV_CAP_VALUE;
  1332. if (argc > 1) {
  1333. int n = (int)str2l(argv[1]);
  1334. if (n > 0) {
  1335. freq = n;
  1336. }
  1337. }
  1338. if (!freq)
  1339. freq = EBPF_DEFAULT_UPDATE_EVERY;
  1340. if (load_collector_config(ebpf_user_config_dir, &disable_apps, &disable_cgroups, freq)) {
  1341. info(
  1342. "Does not have a configuration file inside `%s/ebpf.d.conf. It will try to load stock file.",
  1343. ebpf_user_config_dir);
  1344. if (load_collector_config(ebpf_stock_config_dir, &disable_apps, &disable_cgroups, freq)) {
  1345. info("Does not have a stock file. It is starting with default options.");
  1346. }
  1347. }
  1348. ebpf_load_thread_config();
  1349. while (1) {
  1350. int c = getopt_long_only(argc, argv, "", long_options, &option_index);
  1351. if (c == -1)
  1352. break;
  1353. switch (option_index) {
  1354. case EBPF_MODULE_PROCESS_IDX: {
  1355. select_threads |= 1<<EBPF_MODULE_PROCESS_IDX;
  1356. #ifdef NETDATA_INTERNAL_CHECKS
  1357. info("EBPF enabling \"PROCESS\" charts, because it was started with the option \"[-]-process\".");
  1358. #endif
  1359. break;
  1360. }
  1361. case EBPF_MODULE_SOCKET_IDX: {
  1362. select_threads |= 1<<EBPF_MODULE_SOCKET_IDX;
  1363. #ifdef NETDATA_INTERNAL_CHECKS
  1364. info("EBPF enabling \"NET\" charts, because it was started with the option \"[-]-net\".");
  1365. #endif
  1366. break;
  1367. }
  1368. case EBPF_MODULE_CACHESTAT_IDX: {
  1369. select_threads |= 1<<EBPF_MODULE_CACHESTAT_IDX;
  1370. #ifdef NETDATA_INTERNAL_CHECKS
  1371. info("EBPF enabling \"CACHESTAT\" charts, because it was started with the option \"[-]-cachestat\".");
  1372. #endif
  1373. break;
  1374. }
  1375. case EBPF_MODULE_SYNC_IDX: {
  1376. select_threads |= 1<<EBPF_MODULE_SYNC_IDX;
  1377. #ifdef NETDATA_INTERNAL_CHECKS
  1378. info("EBPF enabling \"SYNC\" chart, because it was started with the option \"[-]-sync\".");
  1379. #endif
  1380. break;
  1381. }
  1382. case EBPF_MODULE_DCSTAT_IDX: {
  1383. select_threads |= 1<<EBPF_MODULE_DCSTAT_IDX;
  1384. #ifdef NETDATA_INTERNAL_CHECKS
  1385. info("EBPF enabling \"DCSTAT\" charts, because it was started with the option \"[-]-dcstat\".");
  1386. #endif
  1387. break;
  1388. }
  1389. case EBPF_MODULE_SWAP_IDX: {
  1390. select_threads |= 1<<EBPF_MODULE_SWAP_IDX;
  1391. #ifdef NETDATA_INTERNAL_CHECKS
  1392. info("EBPF enabling \"SWAP\" chart, because it was started with the option \"[-]-swap\".");
  1393. #endif
  1394. break;
  1395. }
  1396. case EBPF_MODULE_VFS_IDX: {
  1397. select_threads |= 1<<EBPF_MODULE_VFS_IDX;
  1398. #ifdef NETDATA_INTERNAL_CHECKS
  1399. info("EBPF enabling \"VFS\" chart, because it was started with the option \"[-]-vfs\".");
  1400. #endif
  1401. break;
  1402. }
  1403. case EBPF_MODULE_FILESYSTEM_IDX: {
  1404. select_threads |= 1<<EBPF_MODULE_FILESYSTEM_IDX;
  1405. #ifdef NETDATA_INTERNAL_CHECKS
  1406. info("EBPF enabling \"FILESYSTEM\" chart, because it was started with the option \"[-]-filesystem\".");
  1407. #endif
  1408. break;
  1409. }
  1410. case EBPF_MODULE_DISK_IDX: {
  1411. select_threads |= 1<<EBPF_MODULE_DISK_IDX;
  1412. #ifdef NETDATA_INTERNAL_CHECKS
  1413. info("EBPF enabling \"DISK\" chart, because it was started with the option \"[-]-disk\".");
  1414. #endif
  1415. break;
  1416. }
  1417. case EBPF_MODULE_MOUNT_IDX: {
  1418. select_threads |= 1<<EBPF_MODULE_MOUNT_IDX;
  1419. #ifdef NETDATA_INTERNAL_CHECKS
  1420. info("EBPF enabling \"MOUNT\" chart, because it was started with the option \"[-]-mount\".");
  1421. #endif
  1422. break;
  1423. }
  1424. case EBPF_MODULE_FD_IDX: {
  1425. select_threads |= 1<<EBPF_MODULE_FD_IDX;
  1426. #ifdef NETDATA_INTERNAL_CHECKS
  1427. info("EBPF enabling \"FILEDESCRIPTOR\" chart, because it was started with the option \"[-]-filedescriptor\".");
  1428. #endif
  1429. break;
  1430. }
  1431. case EBPF_MODULE_HARDIRQ_IDX: {
  1432. select_threads |= 1<<EBPF_MODULE_HARDIRQ_IDX;
  1433. #ifdef NETDATA_INTERNAL_CHECKS
  1434. info("EBPF enabling \"HARDIRQ\" chart, because it was started with the option \"[-]-hardirq\".");
  1435. #endif
  1436. break;
  1437. }
  1438. case EBPF_MODULE_SOFTIRQ_IDX: {
  1439. select_threads |= 1<<EBPF_MODULE_SOFTIRQ_IDX;
  1440. #ifdef NETDATA_INTERNAL_CHECKS
  1441. info("EBPF enabling \"SOFTIRQ\" chart, because it was started with the option \"[-]-softirq\".");
  1442. #endif
  1443. break;
  1444. }
  1445. case EBPF_MODULE_OOMKILL_IDX: {
  1446. select_threads |= 1<<EBPF_MODULE_OOMKILL_IDX;
  1447. #ifdef NETDATA_INTERNAL_CHECKS
  1448. info("EBPF enabling \"OOMKILL\" chart, because it was started with the option \"[-]-oomkill\".");
  1449. #endif
  1450. break;
  1451. }
  1452. case EBPF_MODULE_SHM_IDX: {
  1453. select_threads |= 1<<EBPF_MODULE_SHM_IDX;
  1454. #ifdef NETDATA_INTERNAL_CHECKS
  1455. info("EBPF enabling \"SHM\" chart, because it was started with the option \"[-]-shm\".");
  1456. #endif
  1457. break;
  1458. }
  1459. case EBPF_MODULE_MDFLUSH_IDX: {
  1460. select_threads |= 1<<EBPF_MODULE_MDFLUSH_IDX;
  1461. #ifdef NETDATA_INTERNAL_CHECKS
  1462. info("EBPF enabling \"MDFLUSH\" chart, because it was started with the option \"[-]-mdflush\".");
  1463. #endif
  1464. break;
  1465. }
  1466. case EBPF_OPTION_ALL_CHARTS: {
  1467. disable_apps = 0;
  1468. disable_cgroups = 0;
  1469. #ifdef NETDATA_INTERNAL_CHECKS
  1470. info("EBPF running with all chart groups, because it was started with the option \"[-]-all\".");
  1471. #endif
  1472. break;
  1473. }
  1474. case EBPF_OPTION_VERSION: {
  1475. printf("ebpf.plugin %s\n", VERSION);
  1476. exit(0);
  1477. }
  1478. case EBPF_OPTION_HELP: {
  1479. ebpf_print_help();
  1480. exit(0);
  1481. }
  1482. case EBPF_OPTION_GLOBAL_CHART: {
  1483. disable_apps = 1;
  1484. disable_cgroups = 1;
  1485. #ifdef NETDATA_INTERNAL_CHECKS
  1486. info("EBPF running with global chart group, because it was started with the option \"[-]-global\".");
  1487. #endif
  1488. break;
  1489. }
  1490. case EBPF_OPTION_RETURN_MODE: {
  1491. ebpf_set_thread_mode(MODE_RETURN);
  1492. #ifdef NETDATA_INTERNAL_CHECKS
  1493. info("EBPF running in \"RETURN\" mode, because it was started with the option \"[-]-return\".");
  1494. #endif
  1495. break;
  1496. }
  1497. case EBPF_OPTION_LEGACY: {
  1498. ebpf_set_load_mode(EBPF_LOAD_LEGACY);
  1499. #ifdef NETDATA_INTERNAL_CHECKS
  1500. info("EBPF running with \"LEGACY\" code, because it was started with the option \"[-]-legacy\".");
  1501. #endif
  1502. break;
  1503. }
  1504. case EBPF_OPTION_CORE: {
  1505. ebpf_set_load_mode(EBPF_LOAD_CORE);
  1506. #ifdef NETDATA_INTERNAL_CHECKS
  1507. info("EBPF running with \"CO-RE\" code, because it was started with the option \"[-]-core\".");
  1508. #endif
  1509. break;
  1510. }
  1511. default: {
  1512. break;
  1513. }
  1514. }
  1515. }
  1516. if (disable_apps || disable_cgroups) {
  1517. if (disable_apps)
  1518. ebpf_disable_apps();
  1519. if (disable_cgroups)
  1520. ebpf_disable_cgroups();
  1521. }
  1522. if (select_threads) {
  1523. disable_all_global_charts();
  1524. uint64_t idx;
  1525. for (idx = 0; idx < EBPF_OPTION_ALL_CHARTS; idx++) {
  1526. if (select_threads & 1<<idx)
  1527. ebpf_enable_specific_chart(&ebpf_modules[idx], disable_apps, disable_cgroups);
  1528. }
  1529. }
  1530. // Load apps_groups.conf
  1531. if (ebpf_read_apps_groups_conf(
  1532. &apps_groups_default_target, &apps_groups_root_target, ebpf_user_config_dir, "groups")) {
  1533. info("Cannot read process groups configuration file '%s/apps_groups.conf'. Will try '%s/apps_groups.conf'",
  1534. ebpf_user_config_dir, ebpf_stock_config_dir);
  1535. if (ebpf_read_apps_groups_conf(
  1536. &apps_groups_default_target, &apps_groups_root_target, ebpf_stock_config_dir, "groups")) {
  1537. error("Cannot read process groups '%s/apps_groups.conf'. There are no internal defaults. Failing.",
  1538. ebpf_stock_config_dir);
  1539. thread_finished++;
  1540. ebpf_exit(1);
  1541. }
  1542. } else
  1543. info("Loaded config file '%s/apps_groups.conf'", ebpf_user_config_dir);
  1544. }
  1545. /*****************************************************************
  1546. *
  1547. * COLLECTOR ENTRY POINT
  1548. *
  1549. *****************************************************************/
  1550. /**
  1551. * Update PID file
  1552. *
  1553. * Update the content of PID file
  1554. *
  1555. * @param filename is the full name of the file.
  1556. * @param pid that identifies the process
  1557. */
  1558. static void ebpf_update_pid_file(char *filename, pid_t pid)
  1559. {
  1560. FILE *fp = fopen(filename, "w");
  1561. if (!fp)
  1562. return;
  1563. fprintf(fp, "%d", pid);
  1564. fclose(fp);
  1565. }
  1566. /**
  1567. * Get Process Name
  1568. *
  1569. * Get process name from /proc/PID/status
  1570. *
  1571. * @param pid that identifies the process
  1572. */
  1573. static char *ebpf_get_process_name(pid_t pid)
  1574. {
  1575. char *name = NULL;
  1576. char filename[FILENAME_MAX + 1];
  1577. snprintfz(filename, FILENAME_MAX, "/proc/%d/status", pid);
  1578. procfile *ff = procfile_open(filename, " \t", PROCFILE_FLAG_DEFAULT);
  1579. if(unlikely(!ff)) {
  1580. error("Cannot open %s", filename);
  1581. return name;
  1582. }
  1583. ff = procfile_readall(ff);
  1584. if(unlikely(!ff))
  1585. return name;
  1586. unsigned long i, lines = procfile_lines(ff);
  1587. for(i = 0; i < lines ; i++) {
  1588. char *cmp = procfile_lineword(ff, i, 0);
  1589. if (!strcmp(cmp, "Name:")) {
  1590. name = strdupz(procfile_lineword(ff, i, 1));
  1591. break;
  1592. }
  1593. }
  1594. procfile_close(ff);
  1595. return name;
  1596. }
  1597. /**
  1598. * Read Previous PID
  1599. *
  1600. * @param filename is the full name of the file.
  1601. *
  1602. * @return It returns the PID used during previous execution on success or 0 otherwise
  1603. */
  1604. static pid_t ebpf_read_previous_pid(char *filename)
  1605. {
  1606. FILE *fp = fopen(filename, "r");
  1607. if (!fp)
  1608. return 0;
  1609. char buffer[64];
  1610. size_t length = fread(buffer, sizeof(*buffer), 63, fp);
  1611. pid_t old_pid = 0;
  1612. if (length) {
  1613. if (length > 63)
  1614. length = 63;
  1615. buffer[length] = '\0';
  1616. old_pid = (pid_t)str2uint32_t(buffer);
  1617. }
  1618. fclose(fp);
  1619. return old_pid;
  1620. }
  1621. /**
  1622. * Kill previous process
  1623. *
  1624. * Kill previous process whether it was not closed.
  1625. *
  1626. * @param filename is the full name of the file.
  1627. * @param pid that identifies the process
  1628. */
  1629. static void ebpf_kill_previous_process(char *filename, pid_t pid)
  1630. {
  1631. pid_t old_pid = ebpf_read_previous_pid(filename);
  1632. if (!old_pid)
  1633. return;
  1634. // Process is not running
  1635. char *prev_name = ebpf_get_process_name(old_pid);
  1636. if (!prev_name)
  1637. return;
  1638. char *current_name = ebpf_get_process_name(pid);
  1639. if (!strcmp(prev_name, current_name))
  1640. kill(old_pid, SIGKILL);
  1641. freez(prev_name);
  1642. freez(current_name);
  1643. // wait few microseconds before start new plugin
  1644. sleep_usec(USEC_PER_MS * 300);
  1645. }
  1646. /**
  1647. * PID file
  1648. *
  1649. * Write the filename for PID inside the given vector.
  1650. *
  1651. * @param filename vector where we will store the name.
  1652. * @param length number of bytes available in filename vector
  1653. */
  1654. void ebpf_pid_file(char *filename, size_t length)
  1655. {
  1656. snprintfz(filename, length, "%s%s/ebpf.d/ebpf.pid", netdata_configured_host_prefix, ebpf_plugin_dir);
  1657. }
  1658. /**
  1659. * Manage PID
  1660. *
  1661. * This function kills another instance of eBPF whether it is necessary and update the file content.
  1662. *
  1663. * @param pid that identifies the process
  1664. */
  1665. static void ebpf_manage_pid(pid_t pid)
  1666. {
  1667. char filename[FILENAME_MAX + 1];
  1668. ebpf_pid_file(filename, FILENAME_MAX);
  1669. ebpf_kill_previous_process(filename, pid);
  1670. ebpf_update_pid_file(filename, pid);
  1671. }
  1672. /**
  1673. * Entry point
  1674. *
  1675. * @param argc the number of arguments
  1676. * @param argv the pointer to the arguments
  1677. *
  1678. * @return it returns 0 on success and another integer otherwise
  1679. */
  1680. int main(int argc, char **argv)
  1681. {
  1682. set_global_variables();
  1683. ebpf_parse_args(argc, argv);
  1684. ebpf_manage_pid(getpid());
  1685. if (!has_condition_to_run(running_on_kernel)) {
  1686. error("The current collector cannot run on this kernel.");
  1687. return 2;
  1688. }
  1689. if (!am_i_running_as_root()) {
  1690. error(
  1691. "ebpf.plugin should either run as root (now running with uid %u, euid %u) or have special capabilities..",
  1692. (unsigned int)getuid(), (unsigned int)geteuid());
  1693. return 3;
  1694. }
  1695. // set name
  1696. program_name = "ebpf.plugin";
  1697. // disable syslog
  1698. error_log_syslog = 0;
  1699. // set errors flood protection to 100 logs per hour
  1700. error_log_errors_per_period = 100;
  1701. error_log_throttle_period = 3600;
  1702. struct rlimit r = { RLIM_INFINITY, RLIM_INFINITY };
  1703. if (setrlimit(RLIMIT_MEMLOCK, &r)) {
  1704. error("Setrlimit(RLIMIT_MEMLOCK)");
  1705. return 4;
  1706. }
  1707. signal(SIGINT, ebpf_exit);
  1708. signal(SIGTERM, ebpf_exit);
  1709. signal(SIGPIPE, ebpf_exit);
  1710. if (ebpf_start_pthread_variables()) {
  1711. thread_finished++;
  1712. error("Cannot start mutex to control overall charts.");
  1713. ebpf_exit(5);
  1714. }
  1715. ebpf_allocate_common_vectors();
  1716. #ifdef LIBBPF_MAJOR_VERSION
  1717. libbpf_set_strict_mode(LIBBPF_STRICT_ALL);
  1718. #endif
  1719. read_local_addresses();
  1720. read_local_ports("/proc/net/tcp", IPPROTO_TCP);
  1721. read_local_ports("/proc/net/tcp6", IPPROTO_TCP);
  1722. read_local_ports("/proc/net/udp", IPPROTO_UDP);
  1723. read_local_ports("/proc/net/udp6", IPPROTO_UDP);
  1724. struct netdata_static_thread ebpf_threads[] = {
  1725. {"EBPF PROCESS", NULL, NULL, 1,
  1726. NULL, NULL, ebpf_modules[EBPF_MODULE_PROCESS_IDX].start_routine},
  1727. {"EBPF SOCKET" , NULL, NULL, 1,
  1728. NULL, NULL, ebpf_modules[EBPF_MODULE_SOCKET_IDX].start_routine},
  1729. {"EBPF CACHESTAT" , NULL, NULL, 1,
  1730. NULL, NULL, ebpf_modules[EBPF_MODULE_CACHESTAT_IDX].start_routine},
  1731. {"EBPF SYNC" , NULL, NULL, 1,
  1732. NULL, NULL, ebpf_modules[EBPF_MODULE_SYNC_IDX].start_routine},
  1733. {"EBPF DCSTAT" , NULL, NULL, 1,
  1734. NULL, NULL, ebpf_modules[EBPF_MODULE_DCSTAT_IDX].start_routine},
  1735. {"EBPF SWAP" , NULL, NULL, 1,
  1736. NULL, NULL, ebpf_modules[EBPF_MODULE_SWAP_IDX].start_routine},
  1737. {"EBPF VFS" , NULL, NULL, 1,
  1738. NULL, NULL, ebpf_modules[EBPF_MODULE_VFS_IDX].start_routine},
  1739. {"EBPF FILESYSTEM" , NULL, NULL, 1,
  1740. NULL, NULL, ebpf_modules[EBPF_MODULE_FILESYSTEM_IDX].start_routine},
  1741. {"EBPF DISK" , NULL, NULL, 1,
  1742. NULL, NULL, ebpf_modules[EBPF_MODULE_DISK_IDX].start_routine},
  1743. {"EBPF MOUNT" , NULL, NULL, 1,
  1744. NULL, NULL, ebpf_modules[EBPF_MODULE_MOUNT_IDX].start_routine},
  1745. {"EBPF FD" , NULL, NULL, 1,
  1746. NULL, NULL, ebpf_modules[EBPF_MODULE_FD_IDX].start_routine},
  1747. {"EBPF HARDIRQ" , NULL, NULL, 1,
  1748. NULL, NULL, ebpf_modules[EBPF_MODULE_HARDIRQ_IDX].start_routine},
  1749. {"EBPF SOFTIRQ" , NULL, NULL, 1,
  1750. NULL, NULL, ebpf_modules[EBPF_MODULE_SOFTIRQ_IDX].start_routine},
  1751. {"EBPF OOMKILL" , NULL, NULL, 1,
  1752. NULL, NULL, ebpf_modules[EBPF_MODULE_OOMKILL_IDX].start_routine},
  1753. {"EBPF SHM" , NULL, NULL, 1,
  1754. NULL, NULL, ebpf_modules[EBPF_MODULE_SHM_IDX].start_routine},
  1755. {"EBPF MDFLUSH" , NULL, NULL, 1,
  1756. NULL, NULL, ebpf_modules[EBPF_MODULE_MDFLUSH_IDX].start_routine},
  1757. {NULL , NULL, NULL, 0,
  1758. NULL, NULL, NULL}
  1759. };
  1760. //clean_loaded_events();
  1761. int i;
  1762. for (i = 0; ebpf_threads[i].name != NULL; i++) {
  1763. struct netdata_static_thread *st = &ebpf_threads[i];
  1764. st->thread = mallocz(sizeof(netdata_thread_t));
  1765. ebpf_module_t *em = &ebpf_modules[i];
  1766. em->thread_id = i;
  1767. netdata_thread_create(st->thread, st->name, NETDATA_THREAD_OPTION_JOINABLE, st->start_routine, em);
  1768. }
  1769. for (i = 0; ebpf_threads[i].name != NULL; i++) {
  1770. struct netdata_static_thread *st = &ebpf_threads[i];
  1771. netdata_thread_join(*st->thread, NULL);
  1772. }
  1773. thread_finished++;
  1774. ebpf_exit(0);
  1775. return 0;
  1776. }