use GNU fileutils for check_disk

git-svn-id: https://nagiosplug.svn.sourceforge.net/svnroot/nagiosplug/nagiosplug/trunk@430 f882894a-f735-0410-b71e-b25c423dba1c

1 files changed, 77 insertions, 143 deletions

diff --git a/plugins/check_disk.c b/plugins/check_disk.c
index 485ae82..2558f6d 100644
--- a/plugins/check_disk.c
+++ b/plugins/check_disk.c
@@ -55,18 +55,81 @@ const char *options = "\
     Print version information\n";
 
 #include "common.h"
+#if HAVE_INTTYPES_H
+# include <inttypes.h>
+#endif
+#include <assert.h>
 #include "popen.h"
 #include "utils.h"
 #include <stdarg.h>
+#include "../lib/fsusage.h"
+
+/* If nonzero, show inode information. */
+static int inode_format;
+
+/* If nonzero, show even filesystems with zero size or
+   uninteresting types. */
+static int show_all_fs;
+
+/* If nonzero, show only local filesystems.  */
+static int show_local_fs;
+
+/* If nonzero, output data for each filesystem corresponding to a
+   command line argument -- even if it's a dummy (automounter) entry.  */
+static int show_listed_fs;
+
+/* If positive, the units to use when printing sizes;
+   if negative, the human-readable base.  */
+static int output_block_size;
+
+/* If nonzero, invoke the `sync' system call before getting any usage data.
+   Using this option can make df very slow, especially with many or very
+   busy disks.  Note that this may make a difference on some systems --
+   SunOs4.1.3, for one.  It is *not* necessary on Linux.  */
+static int require_sync = 0;
+
+/* A filesystem type to display. */
+
+struct fs_type_list
+{
+  char *fs_name;
+  struct fs_type_list *fs_next;
+};
+
+/* Linked list of filesystem types to display.
+   If `fs_select_list' is NULL, list all types.
+   This table is generated dynamically from command-line options,
+   rather than hardcoding into the program what it thinks are the
+   valid filesystem types; let the user specify any filesystem type
+   they want to, and if there are any filesystems of that type, they
+   will be shown.
+
+   Some filesystem types:
+   4.2 4.3 ufs nfs swap ignore io vm efs dbg */
+
+static struct fs_type_list *fs_select_list;
+
+/* Linked list of filesystem types to omit.
+   If the list is empty, don't exclude any types.  */
+
+static struct fs_type_list *fs_exclude_list;
+
+/* Linked list of mounted filesystems. */
+static struct mount_entry *mount_list;
+
+/* For long options that have no equivalent short option, use a
+   non-character as a pseudo short option, starting with CHAR_MAX + 1.  */
+enum
+{
+  SYNC_OPTION = CHAR_MAX + 1,
+  NO_SYNC_OPTION,
+  BLOCK_SIZE_OPTION
+};
 
 #ifdef _AIX
  #pragma alloca
 #endif
 
-#if HAVE_INTTYPES_H
-# include <inttypes.h>
-#endif
-
 int process_arguments (int, char **);
 int validate_arguments (void);
 int check_disk (int usp, int free_disk);
@@ -99,151 +162,22 @@ main (int argc, char **argv)
         char mntp[MAX_INPUT_BUFFER];
         char *output = "";
 
-#ifdef HAVE_STRUCT_STATFS
-#ifdef HAVE_SYS_VFS_H
-#include <sys/vfs.h>
-#else
-#include <sys/param.h>
-#include <sys/mount.h>
-#endif
-        struct statfs buf;
-#endif
+        struct fs_usage fsp;
+        char *disk;
 
         if (process_arguments (argc, argv) != OK)
                 usage ("Could not parse arguments\n");
 
-#ifdef HAVE_STRUCT_STATFS
-
-        if (statfs (path, &buf) == -1) {
-                switch (errno)
-                        {
-#ifdef ENOTDIR
-                        case ENOTDIR:
-                                terminate (STATE_UNKNOWN, "A component of the path prefix is not a directory.\n");
-#endif
-#ifdef ENAMETOOLONG
-                        case ENAMETOOLONG:
-                                terminate (STATE_UNKNOWN, "path is too long.\n");
-#endif
-#ifdef ENOENT
-                        case ENOENT:
-                                terminate (STATE_UNKNOWN, "The file referred to by path does not exist.\n");
-#endif
-#ifdef EACCES
-                        case EACCES:
-                                terminate (STATE_UNKNOWN, "Search permission is denied for a component of the path prefix of path.\n");
-#endif
-#ifdef ELOOP
-                        case ELOOP:
-                                terminate (STATE_UNKNOWN, "Too many symbolic links were encountered in translating path.\n");
-#endif
-#ifdef EFAULT
-                        case EFAULT:
-                                terminate (STATE_UNKNOWN, "Buf or path points to an invalid address.\n");
-#endif
-#ifdef EIO
-                        case EIO:
-                                terminate (STATE_UNKNOWN, "An I/O error occurred while reading from or writing to the file system.\n");
-#endif
-#ifdef ENOMEM
-                        case ENOMEM:
-                                terminate (STATE_UNKNOWN, "Insufficient kernel memory was available.\n");
-#endif
-#ifdef ENOSYS
-                        case ENOSYS:
-                                terminate (STATE_UNKNOWN, "The  filesystem path is on does not support statfs.\n");
-#endif
-                        }
-        }
+        get_fs_usage (path, disk, &fsp);
 
-        usp = (buf.f_blocks - buf.f_bavail) / buf.f_blocks;
-        disk_result = check_disk (usp, buf.f_bavail);
+        usp = (fsp.fsu_blocks - fsp.fsu_bavail) / fsp.fsu_blocks;
+        disk_result = check_disk (usp, fsp.fsu_bavail);
         result = disk_result;
-        asprintf (&output, "%ld of %ld kB free (%ld-byte blocks)",
-                  buf.f_bavail*buf.f_bsize/1024, buf.f_blocks*buf.f_bsize/1024, buf.f_bsize);
-
-#else
-
-        asprintf (&command_line, "%s %s", DF_COMMAND, path);
-
-        if (verbose>0)
-                printf ("%s ==> ", command_line);
-
-        child_process = spopen (command_line);
-        if (child_process == NULL) {
-                printf ("Could not open pipe: %s\n", command_line);
-                return STATE_UNKNOWN;
-        }
-
-        child_stderr = fdopen (child_stderr_array[fileno (child_process)], "r");
-        if (child_stderr == NULL) {
-                printf ("Could not open stderr for %s\n", command_line);
-        }
-
-        while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process)) {
-
-                if (!index (input_buffer, '/'))
-                        continue;
-
-                /* Fixes AIX /proc fs which lists - for size values */
-                if (strstr (input_buffer, "/proc ") == input_buffer)
-                        continue;
-
-                if (sscanf (input_buffer, "%s %d %d %d %d%% %s", file_system,
-                     &total_disk, &used_disk, &free_disk, &usp, mntp) == 6 ||
-                    sscanf (input_buffer, "%s %*s %d %d %d %d%% %s", file_system,
-                                 &total_disk, &used_disk, &free_disk, &usp, mntp) == 6) {
-
-                        if (strcmp(exclude_device,file_system) == 0 ||
-                            strcmp(exclude_device,mntp) == 0) {
-                                if (verbose>0)
-                                        printf ("ignoring %s.", file_system);
-                                continue;
-                        }
-
-                        disk_result = check_disk (usp, free_disk);
-
-                        if (strcmp (file_system, "none") == 0)
-                                strncpy (file_system, mntp, MAX_INPUT_BUFFER-1);
-
-                        if (disk_result==STATE_OK && erronly && !verbose)
-                                continue;
-
-                        if (disk_result!=STATE_OK || verbose>=0) 
-                                asprintf (&output, "%s [%d kB (%d%%) free on %s]", output,
-                                          free_disk, 100 - usp, display_mntp ? mntp : file_system);
-
-                        result = max_state (result, disk_result);
-                }
-
-                else {
-                        printf ("Unable to read output:\n%s\n%s\n", command_line, input_buffer);
-                        return result;
-                }
-
-        }
-
-        /* If we get anything on stderr, at least set warning */
-        while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_stderr)) {
-                if (result != STATE_CRITICAL) {
-                        result = STATE_WARNING;
-                }
-        }
-
-        /* close stderr */
-        if (child_stderr) 
-                (void) fclose (child_stderr);
-
-        /* close the pipe */
-        if (spclose(child_process)!=0 && result!=STATE_CRITICAL)
-                        result = STATE_WARNING;
-
-        if (usp < 0)
-                terminate (result, "Disk \"%s\" not mounted or nonexistant\n", path);
-        else if (result == STATE_UNKNOWN)
-                terminate (result, "Unable to read output\n%s\n%s\n", command_line, input_buffer);
-
-#endif
+        asprintf (&output, "%llu of %llu kB (%2.0f%%) free (%d-byte blocks)",
+                  fsp.fsu_bavail*fsp.fsu_blocksize/1024,
+                  fsp.fsu_blocks*fsp.fsu_blocksize/1024,
+                  (double)fsp.fsu_bavail*100/fsp.fsu_blocks,
+                  fsp.fsu_blocksize);
 
         terminate (result, "DISK %s %s\n", state_text (result), output);
 }