lib/utils_tcp.c

   1 /*****************************************************************************
   2 *
   3 * Library for check_tcp
   4 *
   5 * License: GPL
   6 * Copyright (c) 1999-2013 Nagios Plugins Development Team
   7 *
   8 * Description:
   9 *
  10 * This file contains utilities for check_tcp. These are tested by libtap
  11 *
  12 *
  13 * This program is free software: you can redistribute it and/or modify
  14 * it under the terms of the GNU General Public License as published by
  15 * the Free Software Foundation, either version 3 of the License, or
  16 * (at your option) any later version.
  17 *
  18 * This program is distributed in the hope that it will be useful,
  19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21 * GNU General Public License for more details.
  22 *
  23 * You should have received a copy of the GNU General Public License
  24 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  25 *
  26 *
  27 *****************************************************************************/
  28
  29 #include "common.h"
  30 #include "utils_tcp.h"
  31
  32 #define VERBOSE(message)                        \
  33         do {                                    \
  34                 if (flags & NP_MATCH_VERBOSE)   \
  35                         puts(message);          \
  36         } while (0)
  37
  38 enum np_match_result
  39 np_expect_match(char *status, char **server_expect, int expect_count, int flags)
  40 {
  41         int i, match = 0, partial = 0;
  42
  43         for (i = 0; i < expect_count; i++) {
  44                 if (flags & NP_MATCH_VERBOSE)
  45                         printf("looking for [%s] %s [%s]\n", server_expect[i],
  46                             (flags & NP_MATCH_EXACT) ?
  47                             "in beginning of" : "anywhere in",
  48                             status);
  49
  50                 if (flags & NP_MATCH_EXACT) {
  51                         if (strncmp(status, server_expect[i], strlen(server_expect[i])) == 0) {
  52                                 VERBOSE("found it");
  53                                 match++;
  54                                 continue;
  55                         } else if (strncmp(status, server_expect[i], strlen(status)) == 0) {
  56                                 VERBOSE("found a substring");
  57                                 partial++;
  58                                 continue;
  59                         }
  60                 } else if (strstr(status, server_expect[i]) != NULL) {
  61                                 VERBOSE("found it");
  62                                 match++;
  63                                 continue;
  64                 }
  65                 VERBOSE("couldn't find it");
  66         }
  67
  68         if ((flags & NP_MATCH_ALL && match == expect_count) ||
  69             (!(flags & NP_MATCH_ALL) && match >= 1))
  70                 return NP_MATCH_SUCCESS;
  71         else if (partial > 0 || !(flags & NP_MATCH_EXACT))
  72                 return NP_MATCH_RETRY;
  73         else
  74                 return NP_MATCH_FAILURE;
  75 }