check_ide_smart: Add support for NetBSD

Thanks to Emmanuel Dreyfus for providing the patch.

4 files changed, 135 insertions, 8 deletions

diff --git a/NEWS b/NEWS
index b16c4e4..1668dc4 100644
--- a/NEWS
+++ b/NEWS
@@ -29,6 +29,7 @@ This file documents the major additions and syntax changes between releases.
         New check_mysql -f option to specify a client options file
         New check_mysql -g option to specify a client options group
         New check_snmp --offset option to allow for adding/substracting an offset value to sensor data
+        Let check_ide_smart support NetBSD
 
         FIXES
         Change the MAIL FROM command generated by check_smtp to be RFC compliant
diff --git a/THANKS.in b/THANKS.in
index 54315af..6118143 100644
--- a/THANKS.in
+++ b/THANKS.in
@@ -285,3 +285,4 @@ Roman Fiedler
 Fabio Rueda
 Gabriele Tozzi
 Sebastian Nohn
+Emmanuel Dreyfus
diff --git a/configure.in b/configure.in
index bbcbb71..de75e53 100644
--- a/configure.in
+++ b/configure.in
@@ -317,16 +317,34 @@ AS_IF([test "x$with_ldap" != "xno"], [
 ])
 
 dnl Check for headers used by check_ide_smart
-AC_CHECK_HEADER(linux/hdreg.h, FOUNDINCLUDE=yes, FOUNDINCLUDE=no)
-if test "$FOUNDINCLUDE" = "yes" ; then
-        AC_CHECK_HEADER(linux/types.h, FOUNDINCLUDE=yes, FOUNDINCLUDE=no)
-fi
+case $host in
+  *linux*)
+    AC_CHECK_HEADER(linux/hdreg.h, FOUNDINCLUDE=yes, FOUNDINCLUDE=no)
+    if test "$FOUNDINCLUDE" = "yes" ; then
+        AC_CHECK_HEADER(linux/types.h, FOUNDINCLUDE=yes, FOUNDINCLUDE=no)
+    fi
+    if test "$FOUNDINCLUDE" = "no" ; then
+        AC_MSG_WARN([Skipping check_ide_smart plugin.])
+        AC_MSG_WARN([check_ide_smart requires linux/hdreg.h and linux/types.h.])
+    fi
+  ;;
+  *netbsd*)
+    AC_CHECK_HEADER(dev/ata/atareg.h, FOUNDINCLUDE=yes, FOUNDINCLUDE=no)
+    if test "$FOUNDINCLUDE" = "yes" ; then
+        AC_CHECK_HEADER(dev/ic/wdcreg.h, FOUNDINCLUDE=yes, FOUNDINCLUDE=no)
+    fi
+    if test "$FOUNDINCLUDE" = "no" ; then
+        AC_MSG_WARN([Skipping check_ide_smart plugin.])
+        AC_MSG_WARN([check_ide_smart requires dev/ata/atareg.h and dev/ic/wdcreg.h])
+    fi
+  ;;
+  *)
+    AC_MSG_WARN([Skipping check_ide_smart plugin.])
+    AC_MSG_WARN([check_ide_smart works only on Linux and NetBSD])
+esac
 
 if test "$FOUNDINCLUDE" = "yes" ; then
         EXTRAS="$EXTRAS check_ide_smart"
-else
-        AC_MSG_WARN([Skipping check_ide_smart plugin.])
-        AC_MSG_WARN([check_ide_smart is linux specific. It requires linux/hdreg.h and linux/types.h.])
 fi
 
 dnl Check for mysql libraries
diff --git a/plugins/check_ide_smart.c b/plugins/check_ide_smart.c
index b942461..e1a75ce 100644
--- a/plugins/check_ide_smart.c
+++ b/plugins/check_ide_smart.c
@@ -46,8 +46,29 @@ void print_usage (void);
 #include <sys/stat.h>
 #include <sys/ioctl.h>
 #include <fcntl.h>
+#ifdef linux
 #include <linux/hdreg.h>
 #include <linux/types.h>
+
+#define OPEN_MODE O_RDONLY
+#endif /* linux */
+#ifdef __NetBSD__
+#include <sys/device.h>
+#include <sys/param.h>
+#include <sys/sysctl.h>
+#include <sys/videoio.h> /* for __u8 and friends */
+#include <sys/scsiio.h>
+#include <sys/ataio.h>
+#include <dev/ata/atareg.h>
+#include <dev/ic/wdcreg.h>
+
+#define SMART_ENABLE WDSM_ENABLE_OPS
+#define SMART_DISABLE WDSM_DISABLE_OPS
+#define SMART_IMMEDIATE_OFFLINE WDSM_EXEC_OFFL_IMM
+#define SMART_AUTO_OFFLINE 0xdb /* undefined in NetBSD headers */
+
+#define OPEN_MODE O_RDWR
+#endif /* __NetBSD__ */
 #include <errno.h>
         
 #define NR_ATTRIBUTES   30
@@ -223,7 +244,7 @@ main (int argc, char *argv[])
                 return STATE_OK;
         }
 
-        fd = open (device, O_RDONLY);
+        fd = open (device, OPEN_MODE);
 
         if (fd < 0) {
                 printf (_("CRITICAL - Couldn't open device %s: %s\n"), device, strerror (errno));
@@ -284,6 +305,7 @@ get_offline_text (int status)
 int
 smart_read_values (int fd, values_t * values) 
 {
+#ifdef linux
         int e;
         __u8 args[4 + 512];
         args[0] = WIN_SMART;
@@ -296,6 +318,35 @@ smart_read_values (int fd, values_t * values)
                 return e;
         }
         memcpy (values, args + 4, 512);
+#endif /* linux */
+#ifdef __NetBSD__
+        struct atareq req;
+        unsigned char inbuf[DEV_BSIZE];
+
+        memset(&req, 0, sizeof(req));
+        req.timeout = 1000;
+        memset(&inbuf, 0, sizeof(inbuf));
+
+        req.flags = ATACMD_READ;
+        req.features = WDSM_RD_DATA;
+        req.command = WDCC_SMART;
+        req.databuf = (char *)inbuf;
+        req.datalen = sizeof(inbuf);
+        req.cylinder = WDSMART_CYL;
+
+        if (ioctl(fd, ATAIOCCOMMAND, &req) == 0) {
+                if (req.retsts != ATACMD_OK)
+                        errno = ENODEV;
+        }
+
+        if (errno != 0) {
+                int e = errno;
+                printf (_("CRITICAL - SMART_READ_VALUES: %s\n"), strerror (errno));
+                return e;
+        }
+
+        (void)memcpy(values, inbuf, 512);
+#endif /* __NetBSD__ */
         return 0;
 }
 
@@ -439,6 +490,7 @@ int
 smart_cmd_simple (int fd, enum SmartCommand command, __u8 val0, char show_error) 
 {
         int e = 0;
+#ifdef linux
         __u8 args[4];
         args[0] = WIN_SMART;
         args[1] = val0;
@@ -450,6 +502,31 @@ smart_cmd_simple (int fd, enum SmartCommand command, __u8 val0, char show_error)
                         printf (_("CRITICAL - %s: %s\n"), smart_command[command].text, strerror (errno));
                 }
         }
+#endif /* linux */
+#ifdef __NetBSD__
+        struct atareq req;
+
+        memset(&req, 0, sizeof(req));
+        req.timeout = 1000;
+        req.flags = ATACMD_READREG;
+        req.features = smart_command[command].value;
+        req.command = WDCC_SMART;
+        req.cylinder = WDSMART_CYL;
+        req.sec_count = val0;
+
+        if (ioctl(fd, ATAIOCCOMMAND, &req) == 0) {
+                if (req.retsts != ATACMD_OK)
+                        errno = ENODEV;
+                if (req.cylinder != WDSMART_CYL)
+                        errno = ENODEV;
+        }
+
+        if (errno != 0) {
+                e = errno;
+                printf (_("CRITICAL - %s: %s\n"), smart_command[command].text, strerror (errno));
+                return e;
+        }
+#endif /* __NetBSD__ */
         return e;
 }
 
@@ -458,6 +535,7 @@ smart_cmd_simple (int fd, enum SmartCommand command, __u8 val0, char show_error)
 int
 smart_read_thresholds (int fd, thresholds_t * thresholds) 
 {
+#ifdef linux
         int e;
         __u8 args[4 + 512];
         args[0] = WIN_SMART;
@@ -470,6 +548,35 @@ smart_read_thresholds (int fd, thresholds_t * thresholds)
                 return e;
         }
         memcpy (thresholds, args + 4, 512);
+#endif /* linux */
+#ifdef __NetBSD__
+        struct atareq req;
+        unsigned char inbuf[DEV_BSIZE];
+
+        memset(&req, 0, sizeof(req));
+        req.timeout = 1000;
+        memset(&inbuf, 0, sizeof(inbuf));
+
+        req.flags = ATACMD_READ;
+        req.features = WDSM_RD_THRESHOLDS;
+        req.command = WDCC_SMART;
+        req.databuf = (char *)inbuf;
+        req.datalen = sizeof(inbuf);
+        req.cylinder = WDSMART_CYL;
+
+        if (ioctl(fd, ATAIOCCOMMAND, &req) == 0) {
+                if (req.retsts != ATACMD_OK)
+                        errno = ENODEV;
+        }
+
+        if (errno != 0) {
+                int e = errno;
+                printf (_("CRITICAL - SMART_READ_THRESHOLDS: %s\n"), strerror (errno));
+                return e;
+        }
+
+        (void)memcpy(thresholds, inbuf, 512);
+#endif /* __NetBSD__ */
         return 0;
 }