summaryrefslogtreecommitdiffstats
path: root/lib/utils_tcp.c
blob: daae1d54f5a05ccdfe9c40a9b739957bee16c13a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
/*****************************************************************************
 *
 * Library for check_tcp
 *
 * License: GPL
 * Copyright (c) 1999-2024 Monitoring Plugins Development Team
 *
 * Description:
 *
 * This file contains utilities for check_tcp. These are tested by libtap
 *
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 *
 *****************************************************************************/

#include "common.h"
#include "utils_tcp.h"

#define VERBOSE(message)                                                                                                                   \
	do {                                                                                                                                   \
		if (flags & NP_MATCH_VERBOSE)                                                                                                      \
			puts(message);                                                                                                                 \
	} while (0)

enum np_match_result np_expect_match(char *status, char **server_expect, int expect_count, int flags) {
	int i, match = 0, partial = 0;

	for (i = 0; i < expect_count; i++) {
		if (flags & NP_MATCH_VERBOSE)
			printf("looking for [%s] %s [%s]\n", server_expect[i], (flags & NP_MATCH_EXACT) ? "in beginning of" : "anywhere in", status);

		if (flags & NP_MATCH_EXACT) {
			if (strncmp(status, server_expect[i], strlen(server_expect[i])) == 0) {
				VERBOSE("found it");
				match++;
				continue;
			} else if (strncmp(status, server_expect[i], strlen(status)) == 0) {
				VERBOSE("found a substring");
				partial++;
				continue;
			}
		} else if (strstr(status, server_expect[i]) != NULL) {
			VERBOSE("found it");
			match++;
			continue;
		}
		VERBOSE("couldn't find it");
	}

	if ((flags & NP_MATCH_ALL && match == expect_count) || (!(flags & NP_MATCH_ALL) && match >= 1))
		return NP_MATCH_SUCCESS;
	else if (partial > 0 || !(flags & NP_MATCH_EXACT))
		return NP_MATCH_RETRY;
	else
		return NP_MATCH_FAILURE;
}