diff options
Diffstat (limited to 'userspace/libebtc/test/ebtc_test2.c')
| -rw-r--r-- | userspace/libebtc/test/ebtc_test2.c | 342 |
1 files changed, 342 insertions, 0 deletions
diff --git a/userspace/libebtc/test/ebtc_test2.c b/userspace/libebtc/test/ebtc_test2.c new file mode 100644 index 0000000..98ac461 --- /dev/null +++ b/userspace/libebtc/test/ebtc_test2.c @@ -0,0 +1,342 @@ +/* + * ==[ FILENAME: ebtc_test2.c ]================================================= + * + * Project + * + * Library for ethernet bridge tables. + * + * + * Description + * + * Test for this library. + * + * + * Copyright + * + * Copyright 2005 by Jens Götze + * All rights reserved. + * + * 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 2 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, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, + * USA. + * + * + * ============================================================================= + */ + + +#include <stdio.h> +#include <stdlib.h> +#include <util.h> +#include <libebtc.h> + + +int append_entry (const char *chain, ebtc_handle_t *handle) +{ + +/* ---- VAR ---- */ + + unsigned int size = 0; + + struct ebt_entry *entry; + + struct ebt_standard_target + *std_target; + + struct ebt_entry_target *target; + + + +/* ---- CODE ---- */ + + /* Allocate memory */ + + size += EBTC_ALIGN(sizeof(struct ebt_entry)); + size += EBTC_ALIGN(sizeof(struct ebt_standard_target)); + + entry = (struct ebt_entry *)malloc(size); + + if (!entry) { + + eprintf("Can't allocate memory\n"); + + return -1; + + } + + memset(entry, 0, size); + + size = EBTC_ALIGN(sizeof(*entry)); + std_target = (struct ebt_standard_target *)((char *)entry + size); + + /* Prepare entry */ + + entry->bitmask = EBT_ENTRY_OR_ENTRIES | EBT_NOPROTO; + + entry->watchers_offset = EBTC_ALIGN(sizeof(*entry)); + entry->target_offset = entry->watchers_offset; + entry->next_offset = entry->target_offset + EBTC_ALIGN(sizeof(*std_target)); + + /* Prepare target */ + + target = &std_target->target; + target->target_size = EBTC_ALIGN(sizeof(*std_target)); + target->target_size -= EBTC_ALIGN(sizeof(*target)); + + snprintf(target->u.name, EBT_FUNCTION_MAXNAMELEN, "standard"); + + std_target->verdict = EBT_CONTINUE; + + /* Append */ + + if (ebtc_append_entry(chain, entry, handle)) { + + eprintf("Can't append entry\n"); + + free(entry); + + return -1; + + } + + free(entry); + + return 0; + +} + + +void printlist (ebtc_handle_t *handle) +{ + +/* ---- VAR ---- */ + + const struct ebt_entry *rule; + + int i = 0; + + const struct ebt_counter + *counter; + + + +/* ---- CODE ---- */ + + rule = ebtc_first_rule("INPUT", handle); + + while (rule) { + + counter = ebtc_read_counter("INPUT", i++, handle); + + printf(" Entry jump to '%s' (bcnt = %lld; pcnt = %lld)\n", + ebtc_get_target(rule, handle), counter->bcnt, counter->pcnt); + + rule = ebtc_next_rule("INPUT", handle); + + } + +} + + +int test_run () +{ + +/* ---- VAR ---- */ + + int i; + + const char *chain; + + const char *policy; + + ebtc_handle_t handle; + + + +/* ---- CODE ---- */ + + /* Open ebtables handle */ + + handle = ebtc_init("filter", EBTC_INIT); + + if (!handle) { + + eprintf("Can't open ebtables (%s)\n", ebtc_strerror(NULL)); + + return 1; + + } + + /* Delete and append entry */ + + printf("Rule list in chain INPUT before delete\n"); + printlist(&handle); + printf("\n"); + + printf("Delete second entry\n"); + + if (ebtc_delete_entry("INPUT", 1, &handle)) { + + eprintf("Can't delete entry\n"); + + return 1; + + } + + printf("\n"); + + printf("Rule list in chain INPUT after delete\n"); + printlist(&handle); + printf("\n"); + + printf("Append entry\n"); + + if (append_entry("INPUT", &handle)) { + + eprintf("Can't append entry\n"); + + return 1; + + } + + printf("\n"); + + /* Wait for packets */ + + printf("Wait 5 seconds for network traffic...\n"); + sleep(5); + printf("\n"); + + /* Submit changes to kernel */ + + if (ebtc_commit(&handle)) { + + eprintf("Can't commit ebtables changes\n"); + + return 1; + + } + + /* Open ebtables handle */ + + handle = ebtc_init("filter", EBTC_INIT); + + if (!handle) { + + eprintf("Can't open ebtables (%s)\n", ebtc_strerror(NULL)); + + return 1; + + } + + /* List current entries */ + + printf("Rule list in chain INPUT after commit\n"); + printlist(&handle); + printf("\n"); + + /* Submit changes to kernel */ + + ebtc_free(&handle); + + return 0; + +} + + +int test_init () +{ + +/* ---- VAR ---- */ + + int i; + + const char *chain; + + const char *policy; + + ebtc_handle_t handle; + + + +/* ---- CODE ---- */ + + handle = ebtc_init("filter", EBTC_INIT_WITHFLUSH); + + if (!handle) { + + eprintf("Can't open ebtables (%s)\n", ebtc_strerror(NULL)); + + return 1; + + } + + printf("Append entry\n"); + + if (append_entry("INPUT", &handle)) { + + eprintf("Can't append entry\n"); + + return 1; + + } + + printf("Append entry\n"); + + if (append_entry("INPUT", &handle)) { + + eprintf("Can't append entry\n"); + + return 1; + + } + + printf("\n"); + + if (ebtc_commit(&handle)) { + + eprintf("Can't commit ebtables changes\n"); + + return 1; + + } + + return 0; + +} + + +int main (int argc, char **argv) +{ + +/* ---- CODE ---- */ + + if (argc != 2) { + + printf("Usage: progname <init|run>\n\n"); + + return 1; + + } + + if (!strcmp(argv[1], "init")) + return test_init(); + else if (!strcmp(argv[1], "run")) + return test_run(); + + return 0; + +} + + |