plugin_diskspace.c 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700
  1. // SPDX-License-Identifier: GPL-3.0-or-later
  2. #include "../proc.plugin/plugin_proc.h"
  3. #define PLUGIN_DISKSPACE_NAME "diskspace.plugin"
  4. #define DEFAULT_EXCLUDED_PATHS "/proc/* /sys/* /var/run/user/* /run/user/* /snap/* /var/lib/docker/*"
  5. #define DEFAULT_EXCLUDED_FILESYSTEMS "*gvfs *gluster* *s3fs *ipfs *davfs2 *httpfs *sshfs *gdfs *moosefs fusectl autofs"
  6. #define DEFAULT_EXCLUDED_FILESYSTEMS_INODES "msdosfs msdos vfat overlayfs aufs* *unionfs"
  7. #define CONFIG_SECTION_DISKSPACE "plugin:proc:diskspace"
  8. #define MAX_STAT_USEC 10000LU
  9. #define SLOW_UPDATE_EVERY 5
  10. static netdata_thread_t *diskspace_slow_thread = NULL;
  11. static struct mountinfo *disk_mountinfo_root = NULL;
  12. static int check_for_new_mountpoints_every = 15;
  13. static int cleanup_mount_points = 1;
  14. static inline void mountinfo_reload(int force) {
  15. static time_t last_loaded = 0;
  16. time_t now = now_realtime_sec();
  17. if(force || now - last_loaded >= check_for_new_mountpoints_every) {
  18. // mountinfo_free_all() can be called with NULL disk_mountinfo_root
  19. mountinfo_free_all(disk_mountinfo_root);
  20. // re-read mountinfo in case something changed
  21. disk_mountinfo_root = mountinfo_read(0);
  22. last_loaded = now;
  23. }
  24. }
  25. // Data to be stored in DICTIONARY dict_mountpoints used by do_disk_space_stats().
  26. // This DICTIONARY is used to lookup the settings of the mount point on each iteration.
  27. struct mount_point_metadata {
  28. int do_space;
  29. int do_inodes;
  30. int shown_error;
  31. int updated;
  32. int slow;
  33. DICTIONARY *chart_labels;
  34. size_t collected; // the number of times this has been collected
  35. RRDSET *st_space;
  36. RRDDIM *rd_space_used;
  37. RRDDIM *rd_space_avail;
  38. RRDDIM *rd_space_reserved;
  39. RRDSET *st_inodes;
  40. RRDDIM *rd_inodes_used;
  41. RRDDIM *rd_inodes_avail;
  42. RRDDIM *rd_inodes_reserved;
  43. };
  44. static DICTIONARY *dict_mountpoints = NULL;
  45. #define rrdset_obsolete_and_pointer_null(st) do { if(st) { rrdset_is_obsolete(st); (st) = NULL; } } while(st)
  46. int mount_point_cleanup(const char *name, void *entry, int slow) {
  47. (void)name;
  48. struct mount_point_metadata *mp = (struct mount_point_metadata *)entry;
  49. if(!mp) return 0;
  50. if (slow != mp->slow)
  51. return 0;
  52. if(likely(mp->updated)) {
  53. mp->updated = 0;
  54. return 0;
  55. }
  56. if(likely(cleanup_mount_points && mp->collected)) {
  57. mp->collected = 0;
  58. mp->updated = 0;
  59. mp->shown_error = 0;
  60. mp->rd_space_avail = NULL;
  61. mp->rd_space_used = NULL;
  62. mp->rd_space_reserved = NULL;
  63. mp->rd_inodes_avail = NULL;
  64. mp->rd_inodes_used = NULL;
  65. mp->rd_inodes_reserved = NULL;
  66. rrdset_obsolete_and_pointer_null(mp->st_space);
  67. rrdset_obsolete_and_pointer_null(mp->st_inodes);
  68. }
  69. return 0;
  70. }
  71. int mount_point_cleanup_cb(const DICTIONARY_ITEM *item, void *entry, void *data __maybe_unused) {
  72. const char *name = dictionary_acquired_item_name(item);
  73. return mount_point_cleanup(name, (struct mount_point_metadata *)entry, 0);
  74. }
  75. // a copy of basic mountinfo fields
  76. struct basic_mountinfo {
  77. char *persistent_id;
  78. char *root;
  79. char *mount_point;
  80. char *filesystem;
  81. struct basic_mountinfo *next;
  82. };
  83. static struct basic_mountinfo *slow_mountinfo_tmp_root = NULL;
  84. static netdata_mutex_t slow_mountinfo_mutex;
  85. static struct basic_mountinfo *basic_mountinfo_create_and_copy(struct mountinfo* mi)
  86. {
  87. struct basic_mountinfo *bmi = callocz(1, sizeof(struct basic_mountinfo));
  88. if (mi) {
  89. bmi->persistent_id = strdupz(mi->persistent_id);
  90. bmi->root = strdupz(mi->root);
  91. bmi->mount_point = strdupz(mi->mount_point);
  92. bmi->filesystem = strdupz(mi->filesystem);
  93. }
  94. return bmi;
  95. }
  96. static void add_basic_mountinfo(struct basic_mountinfo **root, struct mountinfo *mi)
  97. {
  98. if (!root)
  99. return;
  100. struct basic_mountinfo *bmi = basic_mountinfo_create_and_copy(mi);
  101. bmi->next = *root;
  102. *root = bmi;
  103. };
  104. static void free_basic_mountinfo(struct basic_mountinfo *bmi)
  105. {
  106. if (bmi) {
  107. freez(bmi->persistent_id);
  108. freez(bmi->root);
  109. freez(bmi->mount_point);
  110. freez(bmi->filesystem);
  111. freez(bmi);
  112. }
  113. };
  114. static void free_basic_mountinfo_list(struct basic_mountinfo *root)
  115. {
  116. struct basic_mountinfo *bmi = root, *next;
  117. while (bmi) {
  118. next = bmi->next;
  119. free_basic_mountinfo(bmi);
  120. bmi = next;
  121. }
  122. }
  123. static void calculate_values_and_show_charts(
  124. struct basic_mountinfo *mi,
  125. struct mount_point_metadata *m,
  126. struct statvfs *buff_statvfs,
  127. int update_every)
  128. {
  129. const char *family = mi->mount_point;
  130. const char *disk = mi->persistent_id;
  131. // logic found at get_fs_usage() in coreutils
  132. unsigned long bsize = (buff_statvfs->f_frsize) ? buff_statvfs->f_frsize : buff_statvfs->f_bsize;
  133. fsblkcnt_t bavail = buff_statvfs->f_bavail;
  134. fsblkcnt_t btotal = buff_statvfs->f_blocks;
  135. fsblkcnt_t bavail_root = buff_statvfs->f_bfree;
  136. fsblkcnt_t breserved_root = bavail_root - bavail;
  137. fsblkcnt_t bused = likely(btotal >= bavail_root) ? btotal - bavail_root : bavail_root - btotal;
  138. #ifdef NETDATA_INTERNAL_CHECKS
  139. if(unlikely(btotal != bavail + breserved_root + bused))
  140. collector_error("DISKSPACE: disk block statistics for '%s' (disk '%s') do not sum up: total = %llu, available = %llu, reserved = %llu, used = %llu", mi->mount_point, disk, (unsigned long long)btotal, (unsigned long long)bavail, (unsigned long long)breserved_root, (unsigned long long)bused);
  141. #endif
  142. // --------------------------------------------------------------------------
  143. fsfilcnt_t favail = buff_statvfs->f_favail;
  144. fsfilcnt_t ftotal = buff_statvfs->f_files;
  145. fsfilcnt_t favail_root = buff_statvfs->f_ffree;
  146. fsfilcnt_t freserved_root = favail_root - favail;
  147. fsfilcnt_t fused = ftotal - favail_root;
  148. if(m->do_inodes == CONFIG_BOOLEAN_AUTO && favail == (fsfilcnt_t)-1) {
  149. // this file system does not support inodes reporting
  150. // eg. cephfs
  151. m->do_inodes = CONFIG_BOOLEAN_NO;
  152. }
  153. #ifdef NETDATA_INTERNAL_CHECKS
  154. if(unlikely(btotal != bavail + breserved_root + bused))
  155. collector_error("DISKSPACE: disk inode statistics for '%s' (disk '%s') do not sum up: total = %llu, available = %llu, reserved = %llu, used = %llu", mi->mount_point, disk, (unsigned long long)ftotal, (unsigned long long)favail, (unsigned long long)freserved_root, (unsigned long long)fused);
  156. #endif
  157. int rendered = 0;
  158. if(m->do_space == CONFIG_BOOLEAN_YES || (m->do_space == CONFIG_BOOLEAN_AUTO &&
  159. (bavail || breserved_root || bused ||
  160. netdata_zero_metrics_enabled == CONFIG_BOOLEAN_YES))) {
  161. if(unlikely(!m->st_space) || m->st_space->update_every != update_every) {
  162. m->do_space = CONFIG_BOOLEAN_YES;
  163. m->st_space = rrdset_find_active_bytype_localhost("disk_space", disk);
  164. if(unlikely(!m->st_space || m->st_space->update_every != update_every)) {
  165. char title[4096 + 1];
  166. snprintfz(title, 4096, "Disk Space Usage");
  167. m->st_space = rrdset_create_localhost(
  168. "disk_space"
  169. , disk
  170. , NULL
  171. , family
  172. , "disk.space"
  173. , title
  174. , "GiB"
  175. , PLUGIN_DISKSPACE_NAME
  176. , NULL
  177. , NETDATA_CHART_PRIO_DISKSPACE_SPACE
  178. , update_every
  179. , RRDSET_TYPE_STACKED
  180. );
  181. }
  182. rrdset_update_rrdlabels(m->st_space, m->chart_labels);
  183. m->rd_space_avail = rrddim_add(m->st_space, "avail", NULL, (collected_number)bsize, 1024 * 1024 * 1024, RRD_ALGORITHM_ABSOLUTE);
  184. m->rd_space_used = rrddim_add(m->st_space, "used", NULL, (collected_number)bsize, 1024 * 1024 * 1024, RRD_ALGORITHM_ABSOLUTE);
  185. m->rd_space_reserved = rrddim_add(m->st_space, "reserved_for_root", "reserved for root", (collected_number)bsize, 1024 * 1024 * 1024, RRD_ALGORITHM_ABSOLUTE);
  186. }
  187. rrddim_set_by_pointer(m->st_space, m->rd_space_avail, (collected_number)bavail);
  188. rrddim_set_by_pointer(m->st_space, m->rd_space_used, (collected_number)bused);
  189. rrddim_set_by_pointer(m->st_space, m->rd_space_reserved, (collected_number)breserved_root);
  190. rrdset_done(m->st_space);
  191. rendered++;
  192. }
  193. if(m->do_inodes == CONFIG_BOOLEAN_YES || (m->do_inodes == CONFIG_BOOLEAN_AUTO &&
  194. (favail || freserved_root || fused ||
  195. netdata_zero_metrics_enabled == CONFIG_BOOLEAN_YES))) {
  196. if(unlikely(!m->st_inodes) || m->st_inodes->update_every != update_every) {
  197. m->do_inodes = CONFIG_BOOLEAN_YES;
  198. m->st_inodes = rrdset_find_active_bytype_localhost("disk_inodes", disk);
  199. if(unlikely(!m->st_inodes) || m->st_inodes->update_every != update_every) {
  200. char title[4096 + 1];
  201. snprintfz(title, 4096, "Disk Files (inodes) Usage");
  202. m->st_inodes = rrdset_create_localhost(
  203. "disk_inodes"
  204. , disk
  205. , NULL
  206. , family
  207. , "disk.inodes"
  208. , title
  209. , "inodes"
  210. , PLUGIN_DISKSPACE_NAME
  211. , NULL
  212. , NETDATA_CHART_PRIO_DISKSPACE_INODES
  213. , update_every
  214. , RRDSET_TYPE_STACKED
  215. );
  216. }
  217. rrdset_update_rrdlabels(m->st_inodes, m->chart_labels);
  218. m->rd_inodes_avail = rrddim_add(m->st_inodes, "avail", NULL, 1, 1, RRD_ALGORITHM_ABSOLUTE);
  219. m->rd_inodes_used = rrddim_add(m->st_inodes, "used", NULL, 1, 1, RRD_ALGORITHM_ABSOLUTE);
  220. m->rd_inodes_reserved = rrddim_add(m->st_inodes, "reserved_for_root", "reserved for root", 1, 1, RRD_ALGORITHM_ABSOLUTE);
  221. }
  222. rrddim_set_by_pointer(m->st_inodes, m->rd_inodes_avail, (collected_number)favail);
  223. rrddim_set_by_pointer(m->st_inodes, m->rd_inodes_used, (collected_number)fused);
  224. rrddim_set_by_pointer(m->st_inodes, m->rd_inodes_reserved, (collected_number)freserved_root);
  225. rrdset_done(m->st_inodes);
  226. rendered++;
  227. }
  228. if(likely(rendered))
  229. m->collected++;
  230. }
  231. static inline void do_disk_space_stats(struct mountinfo *mi, int update_every) {
  232. const char *disk = mi->persistent_id;
  233. static SIMPLE_PATTERN *excluded_mountpoints = NULL;
  234. static SIMPLE_PATTERN *excluded_filesystems = NULL;
  235. static SIMPLE_PATTERN *excluded_filesystems_inodes = NULL;
  236. usec_t slow_timeout = MAX_STAT_USEC * update_every;
  237. int do_space, do_inodes;
  238. if(unlikely(!dict_mountpoints)) {
  239. SIMPLE_PREFIX_MODE mode = SIMPLE_PATTERN_EXACT;
  240. if(config_move("plugin:proc:/proc/diskstats", "exclude space metrics on paths", CONFIG_SECTION_DISKSPACE, "exclude space metrics on paths") != -1) {
  241. // old configuration, enable backwards compatibility
  242. mode = SIMPLE_PATTERN_PREFIX;
  243. }
  244. excluded_mountpoints = simple_pattern_create(
  245. config_get(CONFIG_SECTION_DISKSPACE, "exclude space metrics on paths", DEFAULT_EXCLUDED_PATHS),
  246. NULL,
  247. mode,
  248. true);
  249. excluded_filesystems = simple_pattern_create(
  250. config_get(CONFIG_SECTION_DISKSPACE, "exclude space metrics on filesystems", DEFAULT_EXCLUDED_FILESYSTEMS),
  251. NULL,
  252. SIMPLE_PATTERN_EXACT,
  253. true);
  254. excluded_filesystems_inodes = simple_pattern_create(
  255. config_get(CONFIG_SECTION_DISKSPACE, "exclude inode metrics on filesystems", DEFAULT_EXCLUDED_FILESYSTEMS_INODES),
  256. NULL,
  257. SIMPLE_PATTERN_EXACT,
  258. true);
  259. dict_mountpoints = dictionary_create_advanced(DICT_OPTION_NONE, &dictionary_stats_category_collectors, 0);
  260. }
  261. struct mount_point_metadata *m = dictionary_get(dict_mountpoints, mi->mount_point);
  262. if(unlikely(!m)) {
  263. int slow = 0;
  264. char var_name[4096 + 1];
  265. snprintfz(var_name, 4096, "plugin:proc:diskspace:%s", mi->mount_point);
  266. int def_space = config_get_boolean_ondemand(CONFIG_SECTION_DISKSPACE, "space usage for all disks", CONFIG_BOOLEAN_AUTO);
  267. int def_inodes = config_get_boolean_ondemand(CONFIG_SECTION_DISKSPACE, "inodes usage for all disks", CONFIG_BOOLEAN_AUTO);
  268. if(unlikely(simple_pattern_matches(excluded_mountpoints, mi->mount_point))) {
  269. def_space = CONFIG_BOOLEAN_NO;
  270. def_inodes = CONFIG_BOOLEAN_NO;
  271. }
  272. if(unlikely(simple_pattern_matches(excluded_filesystems, mi->filesystem))) {
  273. def_space = CONFIG_BOOLEAN_NO;
  274. def_inodes = CONFIG_BOOLEAN_NO;
  275. }
  276. if (unlikely(simple_pattern_matches(excluded_filesystems_inodes, mi->filesystem))) {
  277. def_inodes = CONFIG_BOOLEAN_NO;
  278. }
  279. // check if the mount point is a directory #2407
  280. // but only when it is enabled by default #4491
  281. if(def_space != CONFIG_BOOLEAN_NO || def_inodes != CONFIG_BOOLEAN_NO) {
  282. usec_t start_time = now_monotonic_high_precision_usec();
  283. struct stat bs;
  284. if(stat(mi->mount_point, &bs) == -1) {
  285. collector_error("DISKSPACE: Cannot stat() mount point '%s' (disk '%s', filesystem '%s', root '%s')."
  286. , mi->mount_point
  287. , disk
  288. , mi->filesystem?mi->filesystem:""
  289. , mi->root?mi->root:""
  290. );
  291. def_space = CONFIG_BOOLEAN_NO;
  292. def_inodes = CONFIG_BOOLEAN_NO;
  293. }
  294. else {
  295. if((bs.st_mode & S_IFMT) != S_IFDIR) {
  296. collector_error("DISKSPACE: Mount point '%s' (disk '%s', filesystem '%s', root '%s') is not a directory."
  297. , mi->mount_point
  298. , disk
  299. , mi->filesystem?mi->filesystem:""
  300. , mi->root?mi->root:""
  301. );
  302. def_space = CONFIG_BOOLEAN_NO;
  303. def_inodes = CONFIG_BOOLEAN_NO;
  304. }
  305. }
  306. if ((now_monotonic_high_precision_usec() - start_time) > slow_timeout)
  307. slow = 1;
  308. }
  309. do_space = config_get_boolean_ondemand(var_name, "space usage", def_space);
  310. do_inodes = config_get_boolean_ondemand(var_name, "inodes usage", def_inodes);
  311. struct mount_point_metadata mp = {
  312. .do_space = do_space,
  313. .do_inodes = do_inodes,
  314. .shown_error = 0,
  315. .updated = 0,
  316. .slow = 0,
  317. .collected = 0,
  318. .st_space = NULL,
  319. .rd_space_avail = NULL,
  320. .rd_space_used = NULL,
  321. .rd_space_reserved = NULL,
  322. .st_inodes = NULL,
  323. .rd_inodes_avail = NULL,
  324. .rd_inodes_used = NULL,
  325. .rd_inodes_reserved = NULL
  326. };
  327. mp.chart_labels = rrdlabels_create();
  328. rrdlabels_add(mp.chart_labels, "mount_point", mi->mount_point, RRDLABEL_SRC_AUTO);
  329. rrdlabels_add(mp.chart_labels, "filesystem", mi->filesystem, RRDLABEL_SRC_AUTO);
  330. rrdlabels_add(mp.chart_labels, "mount_root", mi->root, RRDLABEL_SRC_AUTO);
  331. m = dictionary_set(dict_mountpoints, mi->mount_point, &mp, sizeof(struct mount_point_metadata));
  332. m->slow = slow;
  333. }
  334. if (m->slow) {
  335. add_basic_mountinfo(&slow_mountinfo_tmp_root, mi);
  336. return;
  337. }
  338. m->updated = 1;
  339. if(unlikely(m->do_space == CONFIG_BOOLEAN_NO && m->do_inodes == CONFIG_BOOLEAN_NO))
  340. return;
  341. if (unlikely(
  342. mi->flags & MOUNTINFO_READONLY &&
  343. !(mi->flags & MOUNTINFO_IS_IN_SYSD_PROTECTED_LIST) &&
  344. !m->collected &&
  345. m->do_space != CONFIG_BOOLEAN_YES &&
  346. m->do_inodes != CONFIG_BOOLEAN_YES))
  347. return;
  348. usec_t start_time = now_monotonic_high_precision_usec();
  349. struct statvfs buff_statvfs;
  350. if (statvfs(mi->mount_point, &buff_statvfs) < 0) {
  351. if(!m->shown_error) {
  352. collector_error("DISKSPACE: failed to statvfs() mount point '%s' (disk '%s', filesystem '%s', root '%s')"
  353. , mi->mount_point
  354. , disk
  355. , mi->filesystem?mi->filesystem:""
  356. , mi->root?mi->root:""
  357. );
  358. m->shown_error = 1;
  359. }
  360. return;
  361. }
  362. if ((now_monotonic_high_precision_usec() - start_time) > slow_timeout)
  363. m->slow = 1;
  364. m->shown_error = 0;
  365. struct basic_mountinfo bmi;
  366. bmi.mount_point = mi->mount_point;
  367. bmi.persistent_id = mi->persistent_id;
  368. bmi.filesystem = mi->filesystem;
  369. bmi.root = mi->root;
  370. calculate_values_and_show_charts(&bmi, m, &buff_statvfs, update_every);
  371. }
  372. static inline void do_slow_disk_space_stats(struct basic_mountinfo *mi, int update_every) {
  373. struct mount_point_metadata *m = dictionary_get(dict_mountpoints, mi->mount_point);
  374. m->updated = 1;
  375. struct statvfs buff_statvfs;
  376. if (statvfs(mi->mount_point, &buff_statvfs) < 0) {
  377. if(!m->shown_error) {
  378. collector_error("DISKSPACE: failed to statvfs() mount point '%s' (disk '%s', filesystem '%s', root '%s')"
  379. , mi->mount_point
  380. , mi->persistent_id
  381. , mi->filesystem?mi->filesystem:""
  382. , mi->root?mi->root:""
  383. );
  384. m->shown_error = 1;
  385. }
  386. return;
  387. }
  388. m->shown_error = 0;
  389. calculate_values_and_show_charts(mi, m, &buff_statvfs, update_every);
  390. }
  391. static void diskspace_slow_worker_cleanup(void *ptr)
  392. {
  393. UNUSED(ptr);
  394. collector_info("cleaning up...");
  395. worker_unregister();
  396. }
  397. #define WORKER_JOB_SLOW_MOUNTPOINT 0
  398. #define WORKER_JOB_SLOW_CLEANUP 1
  399. struct slow_worker_data {
  400. netdata_thread_t *slow_thread;
  401. int update_every;
  402. };
  403. void *diskspace_slow_worker(void *ptr)
  404. {
  405. struct slow_worker_data *data = (struct slow_worker_data *)ptr;
  406. worker_register("DISKSPACE_SLOW");
  407. worker_register_job_name(WORKER_JOB_SLOW_MOUNTPOINT, "mountpoint");
  408. worker_register_job_name(WORKER_JOB_SLOW_CLEANUP, "cleanup");
  409. struct basic_mountinfo *slow_mountinfo_root = NULL;
  410. int slow_update_every = data->update_every > SLOW_UPDATE_EVERY ? data->update_every : SLOW_UPDATE_EVERY;
  411. netdata_thread_cleanup_push(diskspace_slow_worker_cleanup, data->slow_thread);
  412. usec_t step = slow_update_every * USEC_PER_SEC;
  413. usec_t real_step = USEC_PER_SEC;
  414. heartbeat_t hb;
  415. heartbeat_init(&hb);
  416. while(service_running(SERVICE_COLLECTORS)) {
  417. worker_is_idle();
  418. heartbeat_next(&hb, USEC_PER_SEC);
  419. if (real_step < step) {
  420. real_step += USEC_PER_SEC;
  421. continue;
  422. }
  423. real_step = USEC_PER_SEC;
  424. usec_t start_time = now_monotonic_high_precision_usec();
  425. if (!dict_mountpoints)
  426. continue;
  427. if(unlikely(!service_running(SERVICE_COLLECTORS))) break;
  428. // --------------------------------------------------------------------------
  429. // disk space metrics
  430. worker_is_busy(WORKER_JOB_SLOW_MOUNTPOINT);
  431. netdata_mutex_lock(&slow_mountinfo_mutex);
  432. free_basic_mountinfo_list(slow_mountinfo_root);
  433. slow_mountinfo_root = slow_mountinfo_tmp_root;
  434. slow_mountinfo_tmp_root = NULL;
  435. netdata_mutex_unlock(&slow_mountinfo_mutex);
  436. struct basic_mountinfo *bmi;
  437. for(bmi = slow_mountinfo_root; bmi; bmi = bmi->next) {
  438. do_slow_disk_space_stats(bmi, slow_update_every);
  439. if(unlikely(!service_running(SERVICE_COLLECTORS))) break;
  440. }
  441. if(unlikely(!service_running(SERVICE_COLLECTORS))) break;
  442. worker_is_busy(WORKER_JOB_SLOW_CLEANUP);
  443. for(bmi = slow_mountinfo_root; bmi; bmi = bmi->next) {
  444. struct mount_point_metadata *m = dictionary_get(dict_mountpoints, bmi->mount_point);
  445. if (m)
  446. mount_point_cleanup(bmi->mount_point, m, 1);
  447. }
  448. usec_t dt = now_monotonic_high_precision_usec() - start_time;
  449. if (dt > step) {
  450. slow_update_every = (dt / USEC_PER_SEC) * 3 / 2;
  451. if (slow_update_every % SLOW_UPDATE_EVERY)
  452. slow_update_every += SLOW_UPDATE_EVERY - slow_update_every % SLOW_UPDATE_EVERY;
  453. step = slow_update_every * USEC_PER_SEC;
  454. }
  455. }
  456. netdata_thread_cleanup_pop(1);
  457. free_basic_mountinfo_list(slow_mountinfo_root);
  458. return NULL;
  459. }
  460. static void diskspace_main_cleanup(void *ptr) {
  461. rrd_collector_finished();
  462. worker_unregister();
  463. struct netdata_static_thread *static_thread = (struct netdata_static_thread *)ptr;
  464. static_thread->enabled = NETDATA_MAIN_THREAD_EXITING;
  465. collector_info("cleaning up...");
  466. if (diskspace_slow_thread) {
  467. netdata_thread_join(*diskspace_slow_thread, NULL);
  468. freez(diskspace_slow_thread);
  469. }
  470. free_basic_mountinfo_list(slow_mountinfo_tmp_root);
  471. static_thread->enabled = NETDATA_MAIN_THREAD_EXITED;
  472. }
  473. #define WORKER_JOB_MOUNTINFO 0
  474. #define WORKER_JOB_MOUNTPOINT 1
  475. #define WORKER_JOB_CLEANUP 2
  476. #if WORKER_UTILIZATION_MAX_JOB_TYPES < 3
  477. #error WORKER_UTILIZATION_MAX_JOB_TYPES has to be at least 3
  478. #endif
  479. void *diskspace_main(void *ptr) {
  480. worker_register("DISKSPACE");
  481. worker_register_job_name(WORKER_JOB_MOUNTINFO, "mountinfo");
  482. worker_register_job_name(WORKER_JOB_MOUNTPOINT, "mountpoint");
  483. worker_register_job_name(WORKER_JOB_CLEANUP, "cleanup");
  484. rrd_collector_started();
  485. netdata_thread_cleanup_push(diskspace_main_cleanup, ptr);
  486. cleanup_mount_points = config_get_boolean(CONFIG_SECTION_DISKSPACE, "remove charts of unmounted disks" , cleanup_mount_points);
  487. int update_every = (int)config_get_number(CONFIG_SECTION_DISKSPACE, "update every", localhost->rrd_update_every);
  488. if(update_every < localhost->rrd_update_every)
  489. update_every = localhost->rrd_update_every;
  490. check_for_new_mountpoints_every = (int)config_get_number(CONFIG_SECTION_DISKSPACE, "check for new mount points every", check_for_new_mountpoints_every);
  491. if(check_for_new_mountpoints_every < update_every)
  492. check_for_new_mountpoints_every = update_every;
  493. netdata_mutex_init(&slow_mountinfo_mutex);
  494. diskspace_slow_thread = mallocz(sizeof(netdata_thread_t));
  495. struct slow_worker_data slow_worker_data = {.slow_thread = diskspace_slow_thread, .update_every = update_every};
  496. netdata_thread_create(
  497. diskspace_slow_thread,
  498. "P[diskspace slow]",
  499. NETDATA_THREAD_OPTION_JOINABLE,
  500. diskspace_slow_worker,
  501. &slow_worker_data);
  502. usec_t step = update_every * USEC_PER_SEC;
  503. heartbeat_t hb;
  504. heartbeat_init(&hb);
  505. while(service_running(SERVICE_COLLECTORS)) {
  506. worker_is_idle();
  507. /* usec_t hb_dt = */ heartbeat_next(&hb, step);
  508. if(unlikely(!service_running(SERVICE_COLLECTORS))) break;
  509. // --------------------------------------------------------------------------
  510. // this is smart enough not to reload it every time
  511. worker_is_busy(WORKER_JOB_MOUNTINFO);
  512. mountinfo_reload(0);
  513. // --------------------------------------------------------------------------
  514. // disk space metrics
  515. netdata_mutex_lock(&slow_mountinfo_mutex);
  516. free_basic_mountinfo_list(slow_mountinfo_tmp_root);
  517. slow_mountinfo_tmp_root = NULL;
  518. struct mountinfo *mi;
  519. for(mi = disk_mountinfo_root; mi; mi = mi->next) {
  520. if(unlikely(mi->flags & (MOUNTINFO_IS_DUMMY | MOUNTINFO_IS_BIND)))
  521. continue;
  522. // exclude mounts made by ProtectHome and ProtectSystem systemd hardening options
  523. // https://github.com/netdata/netdata/issues/11498#issuecomment-950982878
  524. if(mi->flags & MOUNTINFO_READONLY && mi->flags & MOUNTINFO_IS_IN_SYSD_PROTECTED_LIST && !strcmp(mi->root, mi->mount_point))
  525. continue;
  526. worker_is_busy(WORKER_JOB_MOUNTPOINT);
  527. do_disk_space_stats(mi, update_every);
  528. if(unlikely(!service_running(SERVICE_COLLECTORS))) break;
  529. }
  530. netdata_mutex_unlock(&slow_mountinfo_mutex);
  531. if(unlikely(!service_running(SERVICE_COLLECTORS))) break;
  532. if(dict_mountpoints) {
  533. worker_is_busy(WORKER_JOB_CLEANUP);
  534. dictionary_walkthrough_read(dict_mountpoints, mount_point_cleanup_cb, NULL);
  535. }
  536. }
  537. worker_unregister();
  538. netdata_thread_cleanup_pop(1);
  539. return NULL;
  540. }