check_swap: stricter error handling

1 files changed, 6 insertions, 0 deletions

diff --git a/plugins/check_swap.c b/plugins/check_swap.c
index 94f41a55..e0c246db 100644
--- a/plugins/check_swap.c
+++ b/plugins/check_swap.c
@@ -27,6 +27,7 @@
  *
  *****************************************************************************/
 
+#include "common.h"
 #ifdef HAVE_DECL_SWAPCTL
 #       ifdef HAVE_SYS_PARAM_H
 #               include <sys/param.h>
@@ -83,6 +84,11 @@ int main(int argc, char **argv) {
 
         swap_result data = get_swap_data(config);
 
+        if (data.errorcode != STATE_OK) {
+                puts("SWAP UNKNOWN - Failed to retrieve Swap usage");
+                exit(STATE_UNKNOWN);
+        }
+
         double percent_used;
         /* if total_swap_mb == 0, let's not divide by 0 */
         if (data.metrics.total != 0) {