]> arthur.barton.de Git - netdata.git/blob - src/proc_self_mountinfo.c
several code optimizations
[netdata.git] / src / proc_self_mountinfo.c
1 #include "common.h"
2
3 // ----------------------------------------------------------------------------
4 // taken from gnulib/mountlist.c
5
6 #ifndef ME_REMOTE
7 /* A file system is "remote" if its Fs_name contains a ':'
8    or if (it is of type (smbfs or cifs) and its Fs_name starts with '//')
9    or Fs_name is equal to "-hosts" (used by autofs to mount remote fs).  */
10 # define ME_REMOTE(Fs_name, Fs_type)            \
11     (strchr (Fs_name, ':') != NULL              \
12      || ((Fs_name)[0] == '/'                    \
13          && (Fs_name)[1] == '/'                 \
14          && (strcmp (Fs_type, "smbfs") == 0     \
15              || strcmp (Fs_type, "cifs") == 0)) \
16      || (strcmp("-hosts", Fs_name) == 0))
17 #endif
18
19 #define ME_DUMMY_0(Fs_name, Fs_type)            \
20   (strcmp (Fs_type, "autofs") == 0              \
21    || strcmp (Fs_type, "proc") == 0             \
22    || strcmp (Fs_type, "subfs") == 0            \
23    /* for Linux 2.6/3.x */                      \
24    || strcmp (Fs_type, "debugfs") == 0          \
25    || strcmp (Fs_type, "devpts") == 0           \
26    || strcmp (Fs_type, "fusectl") == 0          \
27    || strcmp (Fs_type, "mqueue") == 0           \
28    || strcmp (Fs_type, "rpc_pipefs") == 0       \
29    || strcmp (Fs_type, "sysfs") == 0            \
30    /* FreeBSD, Linux 2.4 */                     \
31    || strcmp (Fs_type, "devfs") == 0            \
32    /* for NetBSD 3.0 */                         \
33    || strcmp (Fs_type, "kernfs") == 0           \
34    /* for Irix 6.5 */                           \
35    || strcmp (Fs_type, "ignore") == 0)
36
37 /* Historically, we have marked as "dummy" any file system of type "none",
38    but now that programs like du need to know about bind-mounted directories,
39    we grant an exception to any with "bind" in its list of mount options.
40    I.e., those are *not* dummy entries.  */
41 # define ME_DUMMY(Fs_name, Fs_type)             \
42   (ME_DUMMY_0 (Fs_name, Fs_type) || strcmp (Fs_type, "none") == 0)
43
44 // ----------------------------------------------------------------------------
45
46 // find the mount info with the given major:minor
47 // in the supplied linked list of mountinfo structures
48 struct mountinfo *mountinfo_find(struct mountinfo *root, unsigned long major, unsigned long minor) {
49     struct mountinfo *mi;
50
51     for(mi = root; mi ; mi = mi->next)
52         if(unlikely(mi->major == major && mi->minor == minor))
53             return mi;
54
55     return NULL;
56 }
57
58 // find the mount info with the given filesystem and mount_source
59 // in the supplied linked list of mountinfo structures
60 struct mountinfo *mountinfo_find_by_filesystem_mount_source(struct mountinfo *root, const char *filesystem, const char *mount_source) {
61     struct mountinfo *mi;
62     uint32_t filesystem_hash = simple_hash(filesystem), mount_source_hash = simple_hash(mount_source);
63
64     for(mi = root; mi ; mi = mi->next)
65         if(unlikely(mi->filesystem
66                 && mi->mount_source
67                 && mi->filesystem_hash == filesystem_hash
68                 && mi->mount_source_hash == mount_source_hash
69                 && !strcmp(mi->filesystem, filesystem)
70                 && !strcmp(mi->mount_source, mount_source)))
71             return mi;
72
73     return NULL;
74 }
75
76 struct mountinfo *mountinfo_find_by_filesystem_super_option(struct mountinfo *root, const char *filesystem, const char *super_options) {
77     struct mountinfo *mi;
78     uint32_t filesystem_hash = simple_hash(filesystem);
79
80     size_t solen = strlen(super_options);
81
82     for(mi = root; mi ; mi = mi->next)
83         if(unlikely(mi->filesystem
84                 && mi->super_options
85                 && mi->filesystem_hash == filesystem_hash
86                 && !strcmp(mi->filesystem, filesystem))) {
87
88             // super_options is a comma separated list
89             char *s = mi->super_options, *e;
90             while(*s) {
91                 e = s + 1;
92                 while(*e && *e != ',') e++;
93
94                 size_t len = e - s;
95                 if(unlikely(len == solen && !strncmp(s, super_options, len)))
96                     return mi;
97
98                 if(*e == ',') s = ++e;
99                 else s = e;
100             }
101         }
102
103     return NULL;
104 }
105
106
107 // free a linked list of mountinfo structures
108 void mountinfo_free(struct mountinfo *mi) {
109     if(unlikely(!mi))
110         return;
111
112     if(likely(mi->next))
113         mountinfo_free(mi->next);
114
115     freez(mi->root);
116     freez(mi->mount_point);
117     freez(mi->mount_options);
118     freez(mi->persistent_id);
119
120 /*
121     if(mi->optional_fields_count) {
122         int i;
123         for(i = 0; i < mi->optional_fields_count ; i++)
124             free(*mi->optional_fields[i]);
125     }
126     free(mi->optional_fields);
127 */
128     freez(mi->filesystem);
129     freez(mi->mount_source);
130     freez(mi->super_options);
131     freez(mi);
132 }
133
134 static char *strdupz_decoding_octal(const char *string) {
135     char *buffer = strdupz(string);
136
137     char *d = buffer;
138     const char *s = string;
139
140     while(*s) {
141         if(unlikely(*s == '\\')) {
142             s++;
143             if(likely(isdigit(*s) && isdigit(s[1]) && isdigit(s[2]))) {
144                 char c = *s++ - '0';
145                 c <<= 3;
146                 c |= *s++ - '0';
147                 c <<= 3;
148                 c |= *s++ - '0';
149                 *d++ = c;
150             }
151             else *d++ = '_';
152         }
153         else *d++ = *s++;
154     }
155     *d = '\0';
156
157     return buffer;
158 }
159
160 // read the whole mountinfo into a linked list
161 struct mountinfo *mountinfo_read() {
162     procfile *ff = NULL;
163
164     char filename[FILENAME_MAX + 1];
165     snprintfz(filename, FILENAME_MAX, "%s/proc/self/mountinfo", global_host_prefix);
166     ff = procfile_open(filename, " \t", PROCFILE_FLAG_DEFAULT);
167     if(unlikely(!ff)) {
168         snprintfz(filename, FILENAME_MAX, "%s/proc/1/mountinfo", global_host_prefix);
169         ff = procfile_open(filename, " \t", PROCFILE_FLAG_DEFAULT);
170         if(unlikely(!ff)) return NULL;
171     }
172
173     ff = procfile_readall(ff);
174     if(unlikely(!ff))
175         return NULL;
176
177     struct mountinfo *root = NULL, *last = NULL, *mi = NULL;
178
179     unsigned long l, lines = procfile_lines(ff);
180     for(l = 0; l < lines ;l++) {
181         if(unlikely(procfile_linewords(ff, l) < 5))
182             continue;
183
184         mi = mallocz(sizeof(struct mountinfo));
185
186         unsigned long w = 0;
187         mi->id = strtoul(procfile_lineword(ff, l, w), NULL, 10); w++;
188         mi->parentid = strtoul(procfile_lineword(ff, l, w), NULL, 10); w++;
189
190         char *major = procfile_lineword(ff, l, w), *minor; w++;
191         for(minor = major; *minor && *minor != ':' ;minor++) ;
192
193         if(unlikely(!*minor)) {
194             error("Cannot parse major:minor on '%s' at line %lu of '%s'", major, l + 1, filename);
195             freez(mi);
196             continue;
197         }
198
199         *minor = '\0';
200         minor++;
201
202         mi->flags = 0;
203
204         mi->major = strtoul(major, NULL, 10);
205         mi->minor = strtoul(minor, NULL, 10);
206
207         mi->root = strdupz(procfile_lineword(ff, l, w)); w++;
208         mi->root_hash = simple_hash(mi->root);
209
210         mi->mount_point = strdupz_decoding_octal(procfile_lineword(ff, l, w)); w++;
211         mi->mount_point_hash = simple_hash(mi->mount_point);
212
213         mi->persistent_id = strdupz(mi->mount_point);
214         netdata_fix_chart_id(mi->persistent_id);
215         mi->persistent_id_hash = simple_hash(mi->persistent_id);
216
217         mi->mount_options = strdupz(procfile_lineword(ff, l, w)); w++;
218
219         // count the optional fields
220 /*
221         unsigned long wo = w;
222 */
223         mi->optional_fields_count = 0;
224         char *s = procfile_lineword(ff, l, w);
225         while(*s && *s != '-') {
226             w++;
227             s = procfile_lineword(ff, l, w);
228             mi->optional_fields_count++;
229         }
230
231 /*
232         if(unlikely(mi->optional_fields_count)) {
233             // we have some optional fields
234             // read them into a new array of pointers;
235
236             mi->optional_fields = mallocz(mi->optional_fields_count * sizeof(char *));
237
238             int i;
239             for(i = 0; i < mi->optional_fields_count ; i++) {
240                 *mi->optional_fields[wo] = strdupz(procfile_lineword(ff, l, w));
241                 wo++;
242             }
243         }
244         else
245             mi->optional_fields = NULL;
246 */
247
248         if(likely(*s == '-')) {
249             w++;
250
251             mi->filesystem = strdupz(procfile_lineword(ff, l, w)); w++;
252             mi->filesystem_hash = simple_hash(mi->filesystem);
253
254             mi->mount_source = strdupz(procfile_lineword(ff, l, w)); w++;
255             mi->mount_source_hash = simple_hash(mi->mount_source);
256
257             mi->super_options = strdupz(procfile_lineword(ff, l, w)); w++;
258
259             if(unlikely(ME_DUMMY(mi->mount_source, mi->filesystem)))
260                 mi->flags |= MOUNTINFO_IS_DUMMY;
261
262             if(unlikely(ME_REMOTE(mi->mount_source, mi->filesystem)))
263                 mi->flags |= MOUNTINFO_IS_REMOTE;
264
265             // mark as BIND the duplicates (i.e. same filesystem + same source)
266             {
267                 struct stat buf;
268                 if(unlikely(stat(mi->mount_point, &buf) == -1)) {
269                     mi->st_dev = 0;
270                     mi->flags |= MOUNTINFO_NO_STAT;
271                 }
272                 else {
273                     mi->st_dev = buf.st_dev;
274
275                     struct mountinfo *mt;
276                     for(mt = root; mt; mt = mt->next) {
277                         if(unlikely(mt->st_dev == mi->st_dev && !(mi->flags & MOUNTINFO_NO_STAT))) {
278                             if(strlen(mi->mount_point) < strlen(mt->mount_point))
279                                 mt->flags |= MOUNTINFO_IS_SAME_DEV;
280                             else
281                                 mi->flags |= MOUNTINFO_IS_SAME_DEV;
282                         }
283                     }
284                 }
285             }
286         }
287         else {
288             mi->filesystem = NULL;
289             mi->filesystem_hash = 0;
290
291             mi->mount_source = NULL;
292             mi->mount_source_hash = 0;
293
294             mi->super_options = NULL;
295
296             mi->st_dev = 0;
297         }
298
299         // check if it has size
300         {
301             struct statvfs buff_statvfs;
302             if(unlikely(statvfs(mi->mount_point, &buff_statvfs) < 0)) {
303                 mi->flags |= MOUNTINFO_NO_STAT;
304             }
305             else if(unlikely(!buff_statvfs.f_blocks /* || !buff_statvfs.f_files */)) {
306                 mi->flags |= MOUNTINFO_NO_SIZE;
307             }
308         }
309
310         // link it
311         if(unlikely(!root))
312             root = mi;
313         else
314             last->next = mi;
315
316         last = mi;
317         mi->next = NULL;
318
319 /*
320 #ifdef NETDATA_INTERNAL_CHECKS
321         fprintf(stderr, "MOUNTINFO: %ld %ld %lu:%lu root '%s', persistent id '%s', mount point '%s', mount options '%s', filesystem '%s', mount source '%s', super options '%s'%s%s%s%s%s%s\n",
322              mi->id,
323              mi->parentid,
324              mi->major,
325              mi->minor,
326              mi->root,
327              mi->persistent_id,
328                 (mi->mount_point)?mi->mount_point:"",
329                 (mi->mount_options)?mi->mount_options:"",
330                 (mi->filesystem)?mi->filesystem:"",
331                 (mi->mount_source)?mi->mount_source:"",
332                 (mi->super_options)?mi->super_options:"",
333                 (mi->flags & MOUNTINFO_IS_DUMMY)?" DUMMY":"",
334                 (mi->flags & MOUNTINFO_IS_BIND)?" BIND":"",
335                 (mi->flags & MOUNTINFO_IS_REMOTE)?" REMOTE":"",
336                 (mi->flags & MOUNTINFO_NO_STAT)?" NOSTAT":"",
337                 (mi->flags & MOUNTINFO_NO_SIZE)?" NOSIZE":"",
338                 (mi->flags & MOUNTINFO_IS_SAME_DEV)?" SAMEDEV":""
339         );
340 #endif
341 */
342     }
343
344 /* find if the mount options have "bind" in them
345     {
346         FILE *fp = setmntent(MOUNTED, "r");
347         if (fp != NULL) {
348             struct mntent mntbuf;
349             struct mntent *mnt;
350             char buf[4096 + 1];
351
352             while ((mnt = getmntent_r(fp, &mntbuf, buf, 4096))) {
353                 char *bind = hasmntopt(mnt, "bind");
354                 if(unlikely(bind)) {
355                     struct mountinfo *mi;
356                     for(mi = root; mi ; mi = mi->next) {
357                         if(unlikely(strcmp(mnt->mnt_dir, mi->mount_point) == 0)) {
358                             fprintf(stderr, "Mount point '%s' is BIND\n", mi->mount_point);
359                             mi->flags |= MOUNTINFO_IS_BIND;
360                             break;
361                         }
362                     }
363
364 #ifdef NETDATA_INTERNAL_CHECKS
365                     if(unlikely(!mi)) {
366                         error("Mount point '%s' not found in /proc/self/mountinfo", mnt->mnt_dir);
367                     }
368 #endif
369                 }
370             }
371             endmntent(fp);
372         }
373     }
374 */
375
376     procfile_close(ff);
377     return root;
378 }