diff options
Diffstat (limited to 'tests/nft-parsing-test.c')
| -rw-r--r-- | tests/nft-parsing-test.c | 283 |
1 files changed, 0 insertions, 283 deletions
diff --git a/tests/nft-parsing-test.c b/tests/nft-parsing-test.c deleted file mode 100644 index d25a73e..0000000 --- a/tests/nft-parsing-test.c +++ /dev/null @@ -1,283 +0,0 @@ -#include <stdio.h> -#include <stdlib.h> -#include <string.h> -#include <dirent.h> -#include <limits.h> -#include <errno.h> -#include <getopt.h> - -#include <libmnl/libmnl.h> /*nlmsghdr*/ -#include <libnftnl/ruleset.h> -#include <libnftnl/table.h> -#include <libnftnl/chain.h> -#include <libnftnl/rule.h> -#include <libnftnl/set.h> - -enum { - TEST_XML_RULESET, - TEST_JSON_RULESET, -}; - -static bool update = false; - -static void print_detail_error(char *a, char *b) -{ - int i; - int from = -1; - - for (i = 0; i < strlen(b); i++) { - if (from == -1 && a[i] != b[i]) { - from = i; - break; - - } - } - - if (from != -1) { - int k = from - 10; - - if (k < 0) - k = 0; - - fprintf(stderr, "from file: "); - for (i = k; i < from + 10; i++) - fprintf(stderr, "%c", a[i]); - - fprintf(stderr, "\nfrom snprintf: "); - for (i = k; i < from + 10; i++) - fprintf(stderr, "%c", b[i]); - - /* Don't look twice below this comment ;-) */ - fprintf(stderr, "\n "); - for (i = k; i < from + 10; i++) { - if (i == from) - fprintf(stderr, "^"); - else - fprintf(stderr, " "); - } - fprintf(stderr, "\n"); - } -} - -static int compare_test(uint32_t type, struct nftnl_ruleset *rs, - const char *filename, FILE *fp) -{ - char orig[4096]; - char out[4096]; - - switch (type) { - case TEST_XML_RULESET: - nftnl_ruleset_snprintf(out, sizeof(out), rs, - NFTNL_OUTPUT_XML, NFTNL_OF_EVENT_NEW); - break; - case TEST_JSON_RULESET: - nftnl_ruleset_snprintf(out, sizeof(out), rs, - NFTNL_OUTPUT_JSON, NFTNL_OF_EVENT_NEW); - break; - default: - errno = EINVAL; - return -1; - } - - rewind(fp); - fgets(orig, sizeof(orig), fp); - - if (strncmp(orig, out, strlen(out)) == 0) { - if (update) - printf("%s: No changes to update\n", filename); - return 0; - } - if (update) { - FILE *fout; - printf("%s: Updating test file\n", filename); - fout = fopen(filename, "w"); - if (fout == NULL) { - printf("unable to open file %s: %s\n", filename, - strerror(errno)); - return -1; - } - fprintf(fout, "%s\n", out); - fclose(fout); - return 0; - } - - printf("validating %s: ", filename); - printf("\033[31mFAILED\e[0m\n"); - print_detail_error(orig, out); - return -1; -} - -static int test_json(const char *filename, struct nftnl_parse_err *err) -{ - int ret = -1; - struct nftnl_ruleset *rs; - FILE *fp; - - fp = fopen(filename, "r"); - if (fp == NULL) { - printf("unable to open file %s: %s\n", filename, - strerror(errno)); - return -1; - } - - rs = nftnl_ruleset_alloc(); - if (rs == NULL) { - perror("nftnl_ruleset_alloc"); - return -1; - } - - if (nftnl_ruleset_parse_file(rs, NFTNL_PARSE_JSON, fp, err) == 0) - ret = compare_test(TEST_JSON_RULESET, rs, filename, fp); - else - goto failparsing; - - nftnl_ruleset_free(rs); - fclose(fp); - - return ret; - -failparsing: - fclose(fp); - printf("parsing %s: ", filename); - printf("\033[31mFAILED\e[0m (%s)\n", strerror(errno)); - nftnl_parse_perror("Reason", err); - return -1; -} - -static int execute_test(const char *dir_name) -{ - DIR *d; - struct dirent *dent; - char path[PATH_MAX]; - int ret = 0, exit_code = 0; - struct nftnl_parse_err *err; - - d = opendir(dir_name); - if (d == NULL) { - perror("opendir"); - exit(EXIT_FAILURE); - } - - err = nftnl_parse_err_alloc(); - if (err == NULL) { - perror("error"); - exit(EXIT_FAILURE); - } - - while ((dent = readdir(d)) != NULL) { - int len = strlen(dent->d_name); - - if (strcmp(dent->d_name, ".") == 0 || - strcmp(dent->d_name, "..") == 0) - continue; - - snprintf(path, sizeof(path), "%s/%s", dir_name, dent->d_name); - - if (strcmp(&dent->d_name[len-5], ".json") == 0) { - if ((ret = test_json(path, err)) == 0) { - if (!update) { - printf("parsing and validating %s: ", - path); - printf("\033[32mOK\e[0m\n"); - } - } - exit_code += ret; - } - } - - closedir(d); - nftnl_parse_err_free(err); - - if (exit_code != 0) - exit(EXIT_FAILURE); - - return 0; -} - -static int execute_test_file(const char *filename) -{ - char path[PATH_MAX]; - struct nftnl_parse_err *err; - int ret = 0, len; - - err = nftnl_parse_err_alloc(); - if (err == NULL) { - perror("error"); - exit(EXIT_FAILURE); - } - - snprintf(path, sizeof(path), "%s", filename); - - len = strlen(filename); - if (strcmp(&filename[len-5], ".json") == 0) { - if ((ret = test_json(path, err)) == 0) { - if (!update) { - printf("parsing and validating %s: ", - path); - printf("\033[32mOK\e[0m\n"); - } - } - nftnl_parse_err_free(err); - exit(EXIT_FAILURE); - } - - nftnl_parse_err_free(err); - - return 0; -} - -static void show_help(const char *name) -{ - printf( -"Usage: %s [option]\n" -"\n" -"Options:\n" -" -d/--dir <directory> Check test files from <directory>.\n" -" -u/--update <directory> Update test files from <directory>.\n" -" -f/--file <file> Check test file <file>\n" -"\n", - name); -} - -int main(int argc, char *argv[]) -{ - int val; - int ret = 0; - int option_index = 0; - static struct option long_options[] = { - { "dir", required_argument, 0, 'd' }, - { "update", required_argument, 0, 'u' }, - { "file", required_argument, 0, 'f' }, - { 0 } - }; - - if (argc != 3) { - show_help(argv[0]); - exit(EXIT_FAILURE); - } - - while (1) { - val = getopt_long(argc, argv, "d:u:f:", long_options, - &option_index); - - if (val == -1) - break; - - switch (val) { - case 'd': - ret = execute_test(optarg); - break; - case 'u': - update = true; - ret = execute_test(optarg); - break; - case 'f': - ret = execute_test_file(optarg); - break; - default: - show_help(argv[0]); - break; - } - } - return ret; -} |