summaryrefslogtreecommitdiffstats
path: root/tests/conntrack/test-conntrack.c
blob: ec66e35d014407f6f50135aa958f1d1d0298533f (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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
/* 
 * Very simple test-tool for the command line tool `conntrack'.
 * This code is released under GPLv2 or any later at your option.
 *
 * gcc test-conntrack.c -o test
 *
 * Do not forget that you need *root* or CAP_NET_ADMIN capabilities ;-)
 *
 * (c) 2008 Pablo Neira Ayuso <pablo@netfilter.org>
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <dirent.h>

#define CT_PROG "/usr/sbin/conntrack"

int main()
{
	int ret, ok = 0, bad = 0, line;
	FILE *fp;
	char buf[1024];
	struct dirent **dents;
	struct dirent *dent;
	char file[1024];
	int i,n;

	n = scandir("testsuite", &dents, NULL, alphasort);

	for (i = 0; i < n; i++) {
		dent = dents[i];

		sprintf(file, "testsuite/%s", dent->d_name);

		line = 0;

		fp = fopen(file, "r");
		if (fp == NULL) {
			perror("cannot find testsuite file");
			exit(EXIT_FAILURE);
		}

		while (fgets(buf, sizeof(buf), fp)) {
			char tmp[1024] = CT_PROG, *res;
			tmp[strlen(CT_PROG)] = ' ';

			line++;

			if (buf[0] == '#' || buf[0] == ' ')
				continue;

			res = strchr(buf, ';');
			if (!res) {
				printf("malformed file %s at line %d\n", 
					dent->d_name, line);
				exit(EXIT_FAILURE);
			}
			*res = '\0';
			res+=2;

			strcpy(tmp + strlen(CT_PROG) + 1, buf);
			printf("(%d) Executing: %s\n", line, tmp);

			fflush(stdout);
			ret = system(tmp);

			if (WIFEXITED(ret) &&
			    WEXITSTATUS(ret) == EXIT_SUCCESS) {
			    	if (res[0] == 'O' &&
				    res[1] == 'K')
					ok++;
				else {
					bad++;
					printf("^----- BAD\n");
				}
			} else {
				if (res[0] == 'B' &&
				    res[1] == 'A' &&
				    res[2] == 'D')
					ok++;
				else {
					bad++;
					printf("^----- BAD\n");
				}
			}
			printf("=====\n");
		}
		fclose(fp);
	}

	for (i = 0; i < n; i++)
		free(dents[i]);

	free(dents);

	fprintf(stdout, "OK: %d BAD: %d\n", ok, bad);
}