| /* Shared library add-on to iptables for SCTP matching |
| * |
| * (C) 2003 by Harald Welte <laforge@gnumonks.org> |
| * |
| * This program is distributed under the terms of GNU GPL v2, 1991 |
| * |
| * libipt_ecn.c borrowed heavily from libipt_dscp.c |
| * |
| */ |
| #include <stdio.h> |
| #include <string.h> |
| #include <stdlib.h> |
| #include <getopt.h> |
| #include <netdb.h> |
| |
| #include <iptables.h> |
| #include <linux/netfilter_ipv4/ip_tables.h> |
| #include <linux/netfilter_ipv4/ipt_sctp.h> |
| |
| /* Initialize the match. */ |
| static void |
| init(struct ipt_entry_match *m, unsigned int *nfcache) |
| { |
| struct ipt_sctp_info *einfo = (struct ipt_sctp_info *)m->data; |
| |
| einfo->spts[1] = einfo->dpts[1] = 0xFFFF; |
| } |
| |
| static void help(void) |
| { |
| printf( |
| "SCTP match v%s options\n" |
| " --sctp-chunks [!] mask comp match when SCTP chunks & mask == comp\n" |
| " --source-port [!] port[:port]\n" |
| " --sport ...\n" |
| " match source port(s)" |
| " --destination-port [!] port[:port]\n" |
| " --dport ...\n\n", |
| IPTABLES_VERSION); |
| } |
| |
| static struct option opts[] = { |
| { .name = "source-port", .has_arg = 1, .flag = 0, .val = '1' }, |
| { .name = "sport", .has_arg = 1, .flag = 0, .val = '1' }, |
| { .name = "destination-port", .has_arg = 1, .flag = 0, .val = '2' }, |
| { .name = "dport", .has_arg = 1, .flag = 0, .val = '2' }, |
| { .name = "sctp-chunks", .has_arg = 1, .flag = 0, .val = '3' }, |
| { .name = 0 } |
| }; |
| |
| static int |
| service_to_port(const char *name) |
| { |
| struct servent *service; |
| |
| if ((service = getservbyname(name, "sctp")) != NULL) |
| return ntohs((unsigned short) service->s_port); |
| |
| return -1; |
| } |
| |
| static u_int16_t |
| parse_sctp_port(const char *port) |
| { |
| unsigned int portnum; |
| |
| if (string_to_number(port, 0, 65535, &portnum) != -1 || |
| (portnum = service_to_port(port)) != -1) |
| return (u_int16_t)portnum; |
| |
| exit_error(PARAMETER_PROBLEM, |
| "invalid TCP port/service `%s' specified", port); |
| } |
| |
| |
| static void |
| parse_sctp_ports(const char *portstring, u_int16_t *ports) |
| { |
| char *buffer; |
| char *cp; |
| |
| buffer = strdup(portstring); |
| if ((cp = strchr(buffer, ':')) == NULL) |
| ports[0] = ports[1] = parse_sctp_port(buffer); |
| else { |
| *cp = '\0'; |
| cp++; |
| |
| ports[0] = buffer[0] ? parse_sctp_port(buffer) : 0; |
| ports[1] = cp[0] ? parse_sctp_port(cp) : 0xFFFF; |
| |
| if (ports[0] > ports[1]) |
| exit_error(PARAMETER_PROBLEM, |
| "invalid portrange (min > max)"); |
| } |
| free(buffer); |
| } |
| |
| struct sctp_chunk_names { |
| const char *name; |
| unsigned int flag; |
| }; |
| |
| /* FIXME: */ |
| #define ALL_CHUNKS 0xabcdef |
| static struct sctp_chunk_names sctp_chunk_names[] |
| = { { .name = "DATA", .flag = (1 << 0) }, |
| { .name = "INIT", .flag = (1 << 1) }, |
| { .name = "INIT_ACK", .flag = (1 << 2) }, |
| { .name = "SACK", .flag = (1 << 3) }, |
| { .name = "HEARTBEAT", .flag = (1 << 4) }, |
| { .name = "HEARTBEAT_ACK", .flag = (1 << 5) }, |
| { .name = "ABORT", .flag = (1 << 6) }, |
| { .name = "SHUTDOWN", .flag = (1 << 7) }, |
| { .name = "SHUTDOWN_ACK", .flag = (1 << 8) }, |
| { .name = "ERROR", .flag = (1 << 9) }, |
| { .name = "COOKIE_ECHO", .flag = (1 << 10) }, |
| { .name = "COOKIE_ACK", .flag = (1 << 11) }, |
| { .name = "ECN_ECNE", .flag = (1 << 12) }, |
| { .name = "ECN_CWR", .flag = (1 << 13) }, |
| { .name = "SHUTDOWN_COMPLETE", .flag = (1 << 14) }, |
| { .name = "ASCONF", .flag = (1 << 31) }, |
| { .name = "ASCONF_ACK", .flag = (1 << 30) }, |
| { .name = "ALL", .flag = ALL_CHUNKS }, |
| { .name = "NONE", .flag = 0 }, |
| }; |
| |
| |
| static unsigned int |
| parse_sctp_chunk(const char *flags) |
| { |
| unsigned int ret = 0; |
| char *ptr; |
| char *buffer; |
| |
| buffer = strdup(flags); |
| |
| for (ptr = strtok(buffer, ","); ptr; ptr = strtok(NULL, ",")) { |
| unsigned int i; |
| int found = 0; |
| for (i = 0; |
| i < sizeof(sctp_chunk_names)/sizeof(struct sctp_chunk_names); |
| i++) { |
| if (strcasecmp(sctp_chunk_names[i].name, ptr) == 0) { |
| ret |= sctp_chunk_names[i].flag; |
| found = 1; |
| break; |
| } |
| } |
| if (!found) |
| exit_error(PARAMETER_PROBLEM, |
| "Unknown sctp chunk `%s'", ptr); |
| } |
| |
| free(buffer); |
| return ret; |
| } |
| |
| static void |
| parse_sctp_chunks(struct ipt_sctp_info *einfo, |
| const char *mask, |
| const char *cmp, |
| int invert) |
| { |
| einfo->chunks = parse_sctp_chunk(mask); |
| einfo->chunk_mask = parse_sctp_chunk(cmp); |
| |
| if (invert) |
| einfo->invflags |= IPT_SCTP_INV_CHUNKS; |
| } |
| |
| #define SCTP_SRC_PORTS 0x01 |
| #define SCTP_DST_PORTS 0x02 |
| #define SCTP_CHUNKS 0x03 |
| |
| static int |
| parse(int c, char **argv, int invert, unsigned int *flags, |
| const struct ipt_entry *entry, |
| unsigned int *nfcache, |
| struct ipt_entry_match **match) |
| { |
| struct ipt_sctp_info *einfo |
| = (struct ipt_sctp_info *)(*match)->data; |
| |
| switch (c) { |
| case '1': |
| if (*flags & SCTP_SRC_PORTS) |
| exit_error(PARAMETER_PROBLEM, |
| "Only one `--source-port' allowed"); |
| check_inverse(optarg, &invert, &optind, 0); |
| parse_sctp_ports(argv[optind-1], einfo->spts); |
| if (invert) |
| einfo->invflags |= IPT_SCTP_INV_SRCPT; |
| *flags |= SCTP_SRC_PORTS; |
| *nfcache |= NFC_IP_SRC_PT; |
| break; |
| |
| case '2': |
| if (*flags & SCTP_DST_PORTS) |
| exit_error(PARAMETER_PROBLEM, |
| "Only one `--destination-port' allowed"); |
| check_inverse(optarg, &invert, &optind, 0); |
| parse_sctp_ports(argv[optind-1], einfo->dpts); |
| if (invert) |
| einfo->invflags |= IPT_SCTP_INV_DSTPT; |
| *flags |= SCTP_DST_PORTS; |
| *nfcache |= NFC_IP_DST_PT; |
| break; |
| |
| case '3': |
| if (*flags & SCTP_CHUNKS) |
| exit_error(PARAMETER_PROBLEM, |
| "Only one `--sctp-chunks' allowed"); |
| check_inverse(optarg, &invert, &optind, 0); |
| |
| if (!argv[optind] |
| || argv[optind][0] == '-' || argv[optind][0] == '!') |
| exit_error(PARAMETER_PROBLEM, |
| "--sctp-chunks requires two args"); |
| |
| parse_sctp_chunks(einfo, argv[optind-1], argv[optind], invert); |
| optind++; |
| *flags |= SCTP_CHUNKS; |
| break; |
| default: |
| return 0; |
| } |
| |
| return 1; |
| } |
| |
| static void |
| final_check(unsigned int flags) |
| { |
| } |
| |
| static char * |
| port_to_service(int port) |
| { |
| struct servent *service; |
| |
| if ((service = getservbyport(htons(port), "sctp"))) |
| return service->s_name; |
| |
| return NULL; |
| } |
| |
| static void |
| print_port(u_int16_t port, int numeric) |
| { |
| char *service; |
| |
| if (numeric || (service = port_to_service(port)) == NULL) |
| printf("%u", port); |
| else |
| printf("%s", service); |
| } |
| |
| static void |
| print_ports(const char *name, u_int16_t min, u_int16_t max, |
| int invert, int numeric) |
| { |
| const char *inv = invert ? "!" : ""; |
| |
| if (min != 0 || max != 0xFFFF || invert) { |
| printf("%s", name); |
| if (min == max) { |
| printf(":%s", inv); |
| print_port(min, numeric); |
| } else { |
| printf("s:%s", inv); |
| print_port(min, numeric); |
| printf(":"); |
| print_port(max, numeric); |
| } |
| printf(" "); |
| } |
| } |
| |
| static void |
| print_chunk(u_int32_t chunks) |
| { |
| unsigned int have_flag = 0; |
| |
| while (chunks) { |
| unsigned int i; |
| |
| for (i = 0; (chunks & sctp_chunk_names[i].flag) == 0; i++); |
| |
| if (have_flag) |
| printf(","); |
| printf("%s", sctp_chunk_names[i].name); |
| have_flag = 1; |
| |
| chunks &= ~sctp_chunk_names[i].flag; |
| } |
| |
| if (!have_flag) |
| printf("NONE"); |
| } |
| |
| static void |
| print_chunks(u_int32_t mask, u_int32_t cmp, int invert, int numeric) |
| { |
| if (mask || invert) { |
| printf("flags:%s", invert ? "!" : ""); |
| if (numeric) |
| printf("0x%04X/0x%04X ", mask, cmp); |
| else { |
| print_chunk(mask); |
| printf("/"); |
| print_chunk(cmp); |
| printf(" "); |
| } |
| } |
| } |
| |
| /* Prints out the matchinfo. */ |
| static void |
| print(const struct ipt_ip *ip, |
| const struct ipt_entry_match *match, |
| int numeric) |
| { |
| const struct ipt_sctp_info *einfo = |
| (const struct ipt_sctp_info *)match->data; |
| |
| printf("sctp "); |
| |
| print_ports("spt", einfo->spts[0], einfo->spts[1], |
| einfo->invflags & IPT_SCTP_INV_SRCPT, |
| numeric); |
| print_ports("dpt", einfo->dpts[0], einfo->dpts[1], |
| einfo->invflags & IPT_SCTP_INV_DSTPT, |
| numeric); |
| |
| print_chunks(einfo->chunks, einfo->chunk_mask, |
| einfo->invflags & ~IPT_SCTP_INV_MASK, |
| numeric); |
| } |
| |
| /* Saves the union ipt_matchinfo in parsable form to stdout. */ |
| static void |
| save(const struct ipt_ip *ip, const struct ipt_entry_match *match) |
| { |
| const struct ipt_sctp_info *einfo = |
| (const struct ipt_sctp_info *)match->data; |
| |
| if (einfo->spts[0] != 0 |
| || einfo->spts[1] != 0xFFFF) { |
| if (einfo->invflags & IPT_SCTP_INV_SRCPT) |
| printf("! "); |
| if (einfo->spts[0] != einfo->spts[1]) |
| printf("--sport %u:%u ", |
| einfo->spts[0], einfo->spts[1]); |
| else |
| printf("--sport %u ", einfo->spts[0]); |
| } |
| |
| if (einfo->dpts[0] != 0 |
| || einfo->dpts[1] != 0xFFFF) { |
| if (einfo->invflags & IPT_SCTP_INV_DSTPT) |
| printf("! "); |
| if (einfo->dpts[0] != einfo->dpts[1]) |
| printf("--dport %u:%u ", |
| einfo->dpts[0], einfo->dpts[1]); |
| else |
| printf("--dport %u ", einfo->dpts[0]); |
| } |
| |
| if (einfo->chunks |
| || (einfo->invflags & IPT_SCTP_INV_CHUNKS)) { |
| if (einfo->invflags & IPT_SCTP_INV_CHUNKS) |
| printf("! "); |
| printf("--sctp-chunks "); |
| if (einfo->chunks != ALL_CHUNKS) { |
| print_chunk(einfo->chunks); |
| } |
| printf(" "); |
| print_chunk(einfo->chunk_mask); |
| printf(" "); |
| } |
| } |
| |
| static |
| struct iptables_match sctp |
| = { .name = "sctp", |
| .version = IPTABLES_VERSION, |
| .size = IPT_ALIGN(sizeof(struct ipt_sctp_info)), |
| .userspacesize = IPT_ALIGN(sizeof(struct ipt_sctp_info)), |
| .help = &help, |
| .init = &init, |
| .parse = &parse, |
| .final_check = &final_check, |
| .print = &print, |
| .save = &save, |
| .extra_opts = opts |
| }; |
| |
| void _init(void) |
| { |
| register_match(&sctp); |
| } |