1 /**
2 * collectd - src/utils_mount.c
3 * Copyright (C) 2005,2006 Niki W. Waibel
4 *
5 * This program is free software; you can redistribute it and/
6 * or modify it under the terms of the GNU General Public Li-
7 * cence as published by the Free Software Foundation; either
8 * version 2 of the Licence, or any later version.
9 *
10 * This program is distributed in the hope that it will be use-
11 * ful, but WITHOUT ANY WARRANTY; without even the implied war-
12 * ranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
13 * See the GNU General Public Licence for more details.
14 *
15 * You should have received a copy of the GNU General Public
16 * Licence along with this program; if not, write to the Free
17 * Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139,
18 * USA.
19 *
20 * Author:
21 * Niki W. Waibel <niki.waibel@gmx.net>
22 **/
26 #include "common.h"
27 #if HAVE_XFS_XQM_H
28 # include <xfs/xqm.h>
29 #define XFS_SUPER_MAGIC_STR "XFSB"
30 #define XFS_SUPER_MAGIC2_STR "BSFX"
31 #endif
33 #include "utils_debug.h"
34 #include "utils_mount.h"
36 #if HAVE_GETVFSSTAT
37 # if HAVE_SYS_TYPES_H
38 # include <sys/types.h>
39 # endif
40 # if HAVE_SYS_STATVFS_H
41 # include <sys/statvfs.h>
42 # endif
43 /* #endif HAVE_GETVFSSTAT */
45 #elif HAVE_GETFSSTAT
46 # if HAVE_SYS_PARAM_H
47 # include <sys/param.h>
48 # endif
49 # if HAVE_SYS_UCRED_H
50 # include <sys/ucred.h>
51 # endif
52 # if HAVE_SYS_MOUNT_H
53 # include <sys/mount.h>
54 # endif
55 #endif /* HAVE_GETFSSTAT */
57 #if HAVE_MNTENT_H
58 # include <mntent.h>
59 #endif
60 #if HAVE_SYS_MNTTAB_H
61 # include <sys/mnttab.h>
62 #endif
64 #if HAVE_PATHS_H
65 # include <paths.h>
66 #endif
68 #ifdef COLLECTD_MNTTAB
69 # undef COLLECTD_MNTTAB
70 #endif
72 #if defined(_PATH_MOUNTED) /* glibc */
73 # define COLLECTD_MNTTAB _PATH_MOUNTED
74 #elif defined(MNTTAB) /* Solaris */
75 # define COLLECTD_MNTTAB MNTTAB
76 #elif defined(MNT_MNTTAB)
77 # define COLLECTD_MNTTAB MNT_MNTTAB
78 #elif defined(MNTTABNAME)
79 # define COLLECTD_MNTTAB MNTTABNAME
80 #elif defined(KMTAB)
81 # define COLLECTD_MNTTAB KMTAB
82 #else
83 # define COLLECTD_MNTTAB "/etc/mnttab"
84 #endif
86 /* *** *** *** ********************************************* *** *** *** */
87 /* *** *** *** *** *** *** private functions *** *** *** *** *** *** */
88 /* *** *** *** ********************************************* *** *** *** */
90 /* stolen from quota-3.13 (quota-tools) */
92 #define PROC_PARTITIONS "/proc/partitions"
93 #define DEVLABELDIR "/dev"
94 #define UUID 1
95 #define VOL 2
97 static struct uuidCache_s {
98 struct uuidCache_s *next;
99 char uuid[16];
100 char *label;
101 char *device;
102 } *uuidCache = NULL;
104 #define EXT2_SUPER_MAGIC 0xEF53
105 struct ext2_super_block {
106 unsigned char s_dummy1[56];
107 unsigned char s_magic[2];
108 unsigned char s_dummy2[46];
109 unsigned char s_uuid[16];
110 char s_volume_name[16];
111 };
112 #define ext2magic(s) ((unsigned int)s.s_magic[0] \
113 + (((unsigned int)s.s_magic[1]) << 8))
115 #if HAVE_XFS_XQM_H
116 struct xfs_super_block {
117 unsigned char s_magic[4];
118 unsigned char s_dummy[28];
119 unsigned char s_uuid[16];
120 unsigned char s_dummy2[60];
121 char s_fsname[12];
122 };
123 #endif /* HAVE_XFS_XQM_H */
125 #define REISER_SUPER_MAGIC "ReIsEr2Fs"
126 struct reiserfs_super_block {
127 unsigned char s_dummy1[52];
128 unsigned char s_magic[10];
129 unsigned char s_dummy2[22];
130 unsigned char s_uuid[16];
131 char s_volume_name[16];
132 };
134 /* for now, only ext2 and xfs are supported */
135 static int
136 get_label_uuid(const char *device, char **label, char *uuid)
137 {
138 /* start with ext2 and xfs tests, taken from mount_guess_fstype */
139 /* should merge these later */
140 int fd, rv = 1;
141 size_t namesize;
142 struct ext2_super_block e2sb;
143 #if HAVE_XFS_XQM_H
144 struct xfs_super_block xfsb;
145 #endif
146 struct reiserfs_super_block reisersb;
148 fd = open(device, O_RDONLY);
149 if(fd == -1) {
150 return rv;
151 }
153 if(lseek(fd, 1024, SEEK_SET) == 1024
154 && read(fd, (char *)&e2sb, sizeof(e2sb)) == sizeof(e2sb)
155 && ext2magic(e2sb) == EXT2_SUPER_MAGIC) {
156 memcpy(uuid, e2sb.s_uuid, sizeof(e2sb.s_uuid));
157 namesize = sizeof(e2sb.s_volume_name);
158 *label = smalloc(namesize + 1);
159 sstrncpy(*label, e2sb.s_volume_name, namesize);
160 rv = 0;
161 #if HAVE_XFS_XQM_H
162 } else if(lseek(fd, 0, SEEK_SET) == 0
163 && read(fd, (char *)&xfsb, sizeof(xfsb)) == sizeof(xfsb)
164 && (strncmp((char *)&xfsb.s_magic, XFS_SUPER_MAGIC_STR, 4) == 0 ||
165 strncmp((char *)&xfsb.s_magic, XFS_SUPER_MAGIC2_STR, 4) == 0)) {
166 memcpy(uuid, xfsb.s_uuid, sizeof(xfsb.s_uuid));
167 namesize = sizeof(xfsb.s_fsname);
168 *label = smalloc(namesize + 1);
169 sstrncpy(*label, xfsb.s_fsname, namesize);
170 rv = 0;
171 #endif /* HAVE_XFS_XQM_H */
172 } else if(lseek(fd, 65536, SEEK_SET) == 65536
173 && read(fd, (char *)&reisersb, sizeof(reisersb)) == sizeof(reisersb)
174 && !strncmp((char *)&reisersb.s_magic, REISER_SUPER_MAGIC, 9)) {
175 memcpy(uuid, reisersb.s_uuid, sizeof(reisersb.s_uuid));
176 namesize = sizeof(reisersb.s_volume_name);
177 *label = smalloc(namesize + 1);
178 sstrncpy(*label, reisersb.s_volume_name, namesize);
179 rv = 0;
180 }
181 close(fd);
182 return rv;
183 }
185 static void
186 uuidcache_addentry(char *device, char *label, char *uuid)
187 {
188 struct uuidCache_s *last;
190 if(!uuidCache) {
191 last = uuidCache = smalloc(sizeof(*uuidCache));
192 } else {
193 for(last = uuidCache; last->next; last = last->next);
194 last->next = smalloc(sizeof(*uuidCache));
195 last = last->next;
196 }
197 last->next = NULL;
198 last->device = device;
199 last->label = label;
200 memcpy(last->uuid, uuid, sizeof(last->uuid));
201 }
203 static void
204 uuidcache_init(void)
205 {
206 char line[100];
207 char *s;
208 int ma, mi, sz;
209 static char ptname[100];
210 FILE *procpt;
211 char uuid[16], *label = NULL;
212 char device[110];
213 int firstPass;
214 int handleOnFirst;
216 if(uuidCache) {
217 return;
218 }
220 procpt = fopen(PROC_PARTITIONS, "r");
221 if(procpt == NULL) {
222 return;
223 }
225 for(firstPass = 1; firstPass >= 0; firstPass--) {
226 fseek(procpt, 0, SEEK_SET);
227 while(fgets(line, sizeof(line), procpt)) {
228 if(sscanf(line, " %d %d %d %[^\n ]",
229 &ma, &mi, &sz, ptname) != 4)
230 {
231 continue;
232 }
234 /* skip extended partitions (heuristic: size 1) */
235 if(sz == 1) {
236 continue;
237 }
239 /* look only at md devices on first pass */
240 handleOnFirst = !strncmp(ptname, "md", 2);
241 if(firstPass != handleOnFirst) {
242 continue;
243 }
245 /* skip entire disk (minor 0, 64, ... on ide;
246 0, 16, ... on sd) */
247 /* heuristic: partition name ends in a digit */
249 for(s = ptname; *s; s++);
251 if(isdigit((int)s[-1])) {
252 /*
253 * Note: this is a heuristic only - there is no reason
254 * why these devices should live in /dev.
255 * Perhaps this directory should be specifiable by option.
256 * One might for example have /devlabel with links to /dev
257 * for the devices that may be accessed in this way.
258 * (This is useful, if the cdrom on /dev/hdc must not
259 * be accessed.)
260 */
261 snprintf(device, sizeof(device), "%s/%s",
262 DEVLABELDIR, ptname);
263 if(!get_label_uuid(device, &label, uuid)) {
264 uuidcache_addentry(sstrdup(device),
265 label, uuid);
266 }
267 }
268 }
269 }
270 fclose(procpt);
271 }
273 static unsigned char
274 fromhex(char c)
275 {
276 if(isdigit((int)c)) {
277 return (c - '0');
278 } else if(islower((int)c)) {
279 return (c - 'a' + 10);
280 } else {
281 return (c - 'A' + 10);
282 }
283 }
285 static char *
286 get_spec_by_x(int n, const char *t)
287 {
288 struct uuidCache_s *uc;
290 uuidcache_init();
291 uc = uuidCache;
293 while(uc) {
294 switch(n) {
295 case UUID:
296 if(!memcmp(t, uc->uuid, sizeof(uc->uuid))) {
297 return sstrdup(uc->device);
298 }
299 break;
300 case VOL:
301 if(!strcmp(t, uc->label)) {
302 return sstrdup(uc->device);
303 }
304 break;
305 }
306 uc = uc->next;
307 }
308 return NULL;
309 }
311 static char *
312 get_spec_by_uuid(const char *s)
313 {
314 char uuid[16];
315 int i;
317 if(strlen(s) != 36
318 || s[8] != '-' || s[13] != '-' || s[18] != '-' || s[23] != '-') {
319 goto bad_uuid;
320 }
322 for(i=0; i<16; i++) {
323 if(*s == '-') {
324 s++;
325 }
326 if(!isxdigit((int)s[0]) || !isxdigit((int)s[1])) {
327 goto bad_uuid;
328 }
329 uuid[i] = ((fromhex(s[0]) << 4) | fromhex(s[1]));
330 s += 2;
331 }
332 return get_spec_by_x(UUID, uuid);
334 bad_uuid:
335 DBG("Found an invalid UUID: %s", s);
336 return NULL;
337 }
339 static char *get_spec_by_volume_label(const char *s)
340 {
341 return get_spec_by_x (VOL, s);
342 }
344 static char *get_device_name(const char *optstr)
345 {
346 char *rc;
348 if (optstr == NULL)
349 {
350 return (NULL);
351 }
352 else if (strncmp (optstr, "UUID=", 5) == 0)
353 {
354 DBG ("TODO: check UUID= code!");
355 rc = get_spec_by_uuid (optstr + 5);
356 }
357 else if (strncmp (optstr, "LABEL=", 6) == 0)
358 {
359 DBG ("TODO: check LABEL= code!");
360 rc = get_spec_by_volume_label (optstr + 6);
361 }
362 else
363 {
364 rc = sstrdup (optstr);
365 }
367 if(!rc)
368 {
369 DBG ("Error checking device name: optstr = %s", optstr);
370 }
371 return rc;
372 }
374 /* What weird OS is this..? I can't find any info with google :/ -octo */
375 #if HAVE_LISTMNTENT && 0
376 static cu_mount_t *cu_mount_listmntent (void)
377 {
378 cu_mount_t *last = *list;
379 struct tabmntent *p;
380 struct mntent *mnt;
382 struct tabmntent *mntlist;
383 if(listmntent(&mntlist, COLLECTD_MNTTAB, NULL, NULL) < 0) {
384 DBG("calling listmntent() failed: %s", strerror(errno));
385 }
387 for(p = mntlist; p; p = p->next) {
388 char *loop = NULL, *device = NULL;
390 mnt = p->ment;
391 loop = cu_mount_getoptionvalue(mnt->mnt_opts, "loop=");
392 if(loop == NULL) { /* no loop= mount */
393 device = get_device_name(mnt->mnt_fsname);
394 if(device == NULL) {
395 DBG("can't get devicename for fs (%s) %s (%s)"
396 ": ignored", mnt->mnt_type,
397 mnt->mnt_dir, mnt->mnt_fsname);
398 continue;
399 }
400 } else {
401 device = loop;
402 }
403 if(*list == NULL) {
404 *list = (cu_mount_t *)smalloc(sizeof(cu_mount_t));
405 last = *list;
406 } else {
407 while(last->next != NULL) { /* is last really last? */
408 last = last->next;
409 }
410 last->next = (cu_mount_t *)smalloc(sizeof(cu_mount_t));
411 last = last->next;
412 }
413 last->dir = sstrdup(mnt->mnt_dir);
414 last->spec_device = sstrdup(mnt->mnt_fsname);
415 last->device = device;
416 last->type = sstrdup(mnt->mnt_type);
417 last->options = sstrdup(mnt->mnt_opts);
418 last->next = NULL;
419 } /* for(p = mntlist; p; p = p->next) */
421 return(last);
422 } /* cu_mount_t *cu_mount_listmntent(void) */
423 /* #endif HAVE_LISTMNTENT */
425 /* 4.4BSD and Mac OS X (getfsstat) or NetBSD (getvfsstat) */
426 #elif HAVE_GETVFSSTAT || HAVE_GETFSSTAT
427 static cu_mount_t *cu_mount_getfsstat (void)
428 {
429 #if HAVE_GETVFSSTAT
430 # define STRUCT_STATFS struct statvfs
431 # define CMD_STATFS getvfsstat
432 # define FLAGS_STATFS ST_NOWAIT
433 /* #endif HAVE_GETVFSSTAT */
434 #elif HAVE_GETFSSTAT
435 # define STRUCT_STATFS struct statfs
436 # define CMD_STATFS getfsstat
437 # define FLAGS_STATFS MNT_NOWAIT
438 #endif /* HAVE_GETFSSTAT */
440 int bufsize;
441 STRUCT_STATFS *buf;
443 int num;
444 int i;
446 cu_mount_t *first = NULL;
447 cu_mount_t *last = NULL;
448 cu_mount_t *new = NULL;
450 /* Get the number of mounted file systems */
451 if ((bufsize = CMD_STATFS (NULL, 0, FLAGS_STATFS)) < 1)
452 {
453 DBG ("getv?fsstat failed: %s", strerror (errno));
454 return (NULL);
455 }
457 if ((buf = (STRUCT_STATFS *) malloc (bufsize * sizeof (STRUCT_STATFS)))
458 == NULL)
459 return (NULL);
460 memset (buf, '\0', bufsize * sizeof (STRUCT_STATFS));
462 /* The bufsize needs to be passed in bytes. Really. This is not in the
463 * manpage.. -octo */
464 if ((num = CMD_STATFS (buf, bufsize * sizeof (STRUCT_STATFS), FLAGS_STATFS)) < 1)
465 {
466 DBG ("getv?fsstat failed: %s", strerror (errno));
467 free (buf);
468 return (NULL);
469 }
471 for (i = 0; i < num; i++)
472 {
473 if ((new = malloc (sizeof (cu_mount_t))) == NULL)
474 break;
475 memset (new, '\0', sizeof (cu_mount_t));
477 /* Copy values from `struct mnttab' */
478 new->dir = sstrdup (buf[i].f_mntonname);
479 new->spec_device = sstrdup (buf[i].f_mntfromname);
480 new->type = sstrdup (buf[i].f_fstypename);
481 new->options = NULL;
482 new->device = get_device_name (new->options);
483 new->next = NULL;
485 /* Append to list */
486 if (first == NULL)
487 {
488 first = new;
489 last = new;
490 }
491 else
492 {
493 last->next = new;
494 last = new;
495 }
496 }
498 free (buf);
500 return (first);
501 }
502 /* #endif HAVE_GETVFSSTAT || HAVE_GETFSSTAT */
504 /* Solaris (SunOS 10): int getmntent(FILE *fp, struct mnttab *mp); */
505 #elif HAVE_TWO_GETMNTENT || HAVE_GEN_GETMNTENT || HAVE_SUN_GETMNTENT
506 static cu_mount_t *cu_mount_gen_getmntent (void)
507 {
508 struct mnttab mt;
509 FILE *fp;
511 cu_mount_t *first = NULL;
512 cu_mount_t *last = NULL;
513 cu_mount_t *new = NULL;
515 DBG ("(void); COLLECTD_MNTTAB = %s", COLLECTD_MNTTAB);
517 if ((fp = fopen (COLLECTD_MNTTAB, "r")) == NULL)
518 {
519 syslog (LOG_ERR, "fopen (%s): %s", COLLECTD_MNTTAB, strerror (errno));
520 return (NULL);
521 }
523 while (getmntent (fp, &mt) == 0)
524 {
525 if ((new = malloc (sizeof (cu_mount_t))) == NULL)
526 break;
527 memset (new, '\0', sizeof (cu_mount_t));
529 /* Copy values from `struct mnttab' */
530 new->dir = sstrdup (mt.mnt_mountp);
531 new->spec_device = sstrdup (mt.mnt_special);
532 new->type = sstrdup (mt.mnt_fstype);
533 new->options = sstrdup (mt.mnt_mntopts);
534 new->device = get_device_name (new->options);
535 new->next = NULL;
537 /* Append to list */
538 if (first == NULL)
539 {
540 first = new;
541 last = new;
542 }
543 else
544 {
545 last->next = new;
546 last = new;
547 }
548 }
550 fclose (fp);
552 return (first);
553 } /* static cu_mount_t *cu_mount_gen_getmntent (void) */
554 /* #endif HAVE_TWO_GETMNTENT || HAVE_GEN_GETMNTENT || HAVE_SUN_GETMNTENT */
556 #elif HAVE_SEQ_GETMNTENT
557 #warn "This version of `getmntent' hat not yet been implemented!"
558 /* #endif HAVE_SEQ_GETMNTENT */
560 #elif HAVE_ONE_GETMNTENT
561 static cu_mount_t *cu_mount_getmntent (void)
562 {
563 FILE *fp;
564 struct mntent *me;
566 cu_mount_t *first = NULL;
567 cu_mount_t *last = NULL;
568 cu_mount_t *new = NULL;
570 DBG ("(void); COLLECTD_MNTTAB = %s", COLLECTD_MNTTAB);
572 if ((fp = setmntent (COLLECTD_MNTTAB, "r")) == NULL)
573 {
574 syslog (LOG_ERR, "setmntent (%s): %s", COLLECTD_MNTTAB, strerror (errno));
575 return (NULL);
576 }
578 while ((me = getmntent (fp)) != NULL)
579 {
580 if ((new = malloc (sizeof (cu_mount_t))) == NULL)
581 break;
582 memset (new, '\0', sizeof (cu_mount_t));
584 /* Copy values from `struct mntent *' */
585 new->dir = sstrdup (me->mnt_dir);
586 new->spec_device = sstrdup (me->mnt_fsname);
587 new->type = sstrdup (me->mnt_type);
588 new->options = sstrdup (me->mnt_opts);
589 new->device = get_device_name (new->options);
590 new->next = NULL;
592 DBG ("new = {dir = %s, spec_device = %s, type = %s, options = %s, device = %s}",
593 new->dir, new->spec_device, new->type, new->options, new->device);
595 /* Append to list */
596 if (first == NULL)
597 {
598 first = new;
599 last = new;
600 }
601 else
602 {
603 last->next = new;
604 last = new;
605 }
606 }
608 endmntent (fp);
610 DBG ("return (0x%p)", (void *) first);
612 return (first);
613 }
614 #endif /* HAVE_ONE_GETMNTENT */
616 /* *** *** *** ******************************************** *** *** *** */
617 /* *** *** *** *** *** *** public functions *** *** *** *** *** *** */
618 /* *** *** *** ******************************************** *** *** *** */
620 cu_mount_t *cu_mount_getlist(cu_mount_t **list)
621 {
622 cu_mount_t *new;
623 cu_mount_t *first = NULL;
624 cu_mount_t *last = NULL;
626 if (list == NULL)
627 return (NULL);
629 if (*list != NULL)
630 {
631 first = *list;
632 last = first;
633 while (last->next != NULL)
634 last = last->next;
635 }
637 #if HAVE_LISTMNTENT && 0
638 new = cu_mount_listmntent ();
639 #elif HAVE_GETVFSSTAT || HAVE_GETFSSTAT
640 new = cu_mount_getfsstat ();
641 #elif HAVE_GEN_GETMNTENT
642 new = cu_mount_gen_getmntent ();
643 #elif HAVE_ONE_GETMNTENT
644 new = cu_mount_getmntent ();
645 #else
646 new = NULL;
647 #endif
649 if (first != NULL)
650 {
651 last->next = new;
652 }
653 else
654 {
655 first = new;
656 last = new;
657 *list = first;
658 }
660 while ((last != NULL) && (last->next != NULL))
661 last = last->next;
663 return (last);
664 } /* cu_mount_t *cu_mount_getlist(cu_mount_t **list) */
666 void cu_mount_freelist (cu_mount_t *list)
667 {
668 cu_mount_t *this;
669 cu_mount_t *next;
671 DBG ("(list = 0x%p)", (void *) list);
673 for (this = list; this != NULL; this = next)
674 {
675 next = this->next;
677 sfree (this->dir);
678 sfree (this->spec_device);
679 sfree (this->device);
680 sfree (this->type);
681 sfree (this->options);
682 sfree (this);
683 }
684 } /* void cu_mount_freelist(cu_mount_t *list) */
686 char *
687 cu_mount_checkoption(char *line, char *keyword, int full)
688 {
689 char *line2, *l2;
690 int l = strlen(keyword);
691 char *p1, *p2;
693 if(line == NULL || keyword == NULL) {
694 return NULL;
695 }
696 if(full != 0) {
697 full = 1;
698 }
700 line2 = sstrdup(line);
701 l2 = line2;
702 while(*l2 != '\0') {
703 if(*l2 == ',') {
704 *l2 = '\0';
705 }
706 l2++;
707 }
709 p1 = line - 1;
710 p2 = strchr(line, ',');
711 do {
712 if(strncmp(line2+(p1-line)+1, keyword, l+full) == 0) {
713 free(line2);
714 return p1+1;
715 }
716 p1 = p2;
717 if(p1 != NULL) {
718 p2 = strchr(p1+1, ',');
719 }
720 } while(p1 != NULL);
722 free(line2);
723 return NULL;
724 } /* char *cu_mount_checkoption(char *line, char *keyword, int full) */
726 char *
727 cu_mount_getoptionvalue(char *line, char *keyword)
728 {
729 char *r;
731 r = cu_mount_checkoption(line, keyword, 0);
732 if(r != NULL) {
733 char *p;
734 r += strlen(keyword);
735 p = strchr(r, ',');
736 if(p == NULL) {
737 if(strlen(r) == 0) {
738 return NULL;
739 }
740 return sstrdup(r);
741 } else {
742 char *m;
743 if((p-r) == 1) {
744 return NULL;
745 }
746 m = (char *)smalloc(p-r+1);
747 sstrncpy(m, r, p-r+1);
748 return m;
749 }
750 }
751 return r;
752 } /* char *cu_mount_getoptionvalue(char *line, char *keyword) */
756 int
757 cu_mount_type(const char *type)
758 {
759 if(strcmp(type, "ext3") == 0) return CUMT_EXT3;
760 if(strcmp(type, "ext2") == 0) return CUMT_EXT2;
761 if(strcmp(type, "ufs") == 0) return CUMT_UFS;
762 if(strcmp(type, "vxfs") == 0) return CUMT_VXFS;
763 if(strcmp(type, "zfs") == 0) return CUMT_ZFS;
764 return CUMT_UNKNOWN;
765 } /* int cu_mount_type(const char *type) */