| /* |
| * useful_functions.c, January 2004 |
| * |
| * Random collection of functions that can be used by extensions. |
| * |
| * Author: Bart De Schuymer |
| * |
| * This code is stongly inspired on the iptables code which is |
| * Copyright (C) 1999 Paul `Rusty' Russell & Michael J. Neuling |
| * |
| * 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., 675 Mass Ave, Cambridge, MA 02139, USA. |
| */ |
| #include "include/ebtables_u.h" |
| #include "include/ethernetdb.h" |
| #include <errno.h> |
| #include <inttypes.h> |
| #include <limits.h> |
| #include <stdio.h> |
| #include <netinet/ether.h> |
| #include <string.h> |
| #include <stdlib.h> |
| #include <getopt.h> |
| #include <errno.h> |
| #include <sys/types.h> |
| #include <sys/socket.h> |
| #include <arpa/inet.h> |
| |
| |
| const unsigned char mac_type_unicast[ETH_ALEN] = {0,0,0,0,0,0}; |
| const unsigned char msk_type_unicast[ETH_ALEN] = {1,0,0,0,0,0}; |
| const unsigned char mac_type_multicast[ETH_ALEN] = {1,0,0,0,0,0}; |
| const unsigned char msk_type_multicast[ETH_ALEN] = {1,0,0,0,0,0}; |
| const unsigned char mac_type_broadcast[ETH_ALEN] = {255,255,255,255,255,255}; |
| const unsigned char msk_type_broadcast[ETH_ALEN] = {255,255,255,255,255,255}; |
| const unsigned char mac_type_bridge_group[ETH_ALEN] = {0x01,0x80,0xc2,0,0,0}; |
| const unsigned char msk_type_bridge_group[ETH_ALEN] = {255,255,255,255,255,255}; |
| |
| /* 0: default, print only 2 digits if necessary |
| * 2: always print 2 digits, a printed mac address |
| * then always has the same length */ |
| int ebt_printstyle_mac; |
| |
| void ebt_print_mac(const unsigned char *mac) |
| { |
| if (ebt_printstyle_mac == 2) { |
| int j; |
| for (j = 0; j < ETH_ALEN; j++) |
| printf("%02x%s", mac[j], |
| (j==ETH_ALEN-1) ? "" : ":"); |
| } else |
| printf("%s", ether_ntoa((struct ether_addr *) mac)); |
| } |
| |
| void ebt_print_mac_and_mask(const unsigned char *mac, const unsigned char *mask) |
| { |
| char hlpmsk[6] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff}; |
| |
| if (!memcmp(mac, mac_type_unicast, 6) && |
| !memcmp(mask, msk_type_unicast, 6)) |
| printf("Unicast"); |
| else if (!memcmp(mac, mac_type_multicast, 6) && |
| !memcmp(mask, msk_type_multicast, 6)) |
| printf("Multicast"); |
| else if (!memcmp(mac, mac_type_broadcast, 6) && |
| !memcmp(mask, msk_type_broadcast, 6)) |
| printf("Broadcast"); |
| else if (!memcmp(mac, mac_type_bridge_group, 6) && |
| !memcmp(mask, msk_type_bridge_group, 6)) |
| printf("BGA"); |
| else { |
| ebt_print_mac(mac); |
| if (memcmp(mask, hlpmsk, 6)) { |
| printf("/"); |
| ebt_print_mac(mask); |
| } |
| } |
| } |
| |
| /* Checks the type for validity and calls getethertypebynumber(). */ |
| struct ethertypeent *parseethertypebynumber(int type) |
| { |
| if (type < 1536) |
| ebt_print_error("Ethernet protocols have values >= 0x0600"); |
| if (type > 0xffff) |
| ebt_print_error("Ethernet protocols have values <= 0xffff"); |
| return getethertypebynumber(type); |
| } |
| |
| /* Put the mac address into 6 (ETH_ALEN) bytes returns 0 on success. */ |
| int ebt_get_mac_and_mask(const char *from, unsigned char *to, |
| unsigned char *mask) |
| { |
| char *p; |
| int i; |
| struct ether_addr *addr; |
| |
| if (strcasecmp(from, "Unicast") == 0) { |
| memcpy(to, mac_type_unicast, ETH_ALEN); |
| memcpy(mask, msk_type_unicast, ETH_ALEN); |
| return 0; |
| } |
| if (strcasecmp(from, "Multicast") == 0) { |
| memcpy(to, mac_type_multicast, ETH_ALEN); |
| memcpy(mask, msk_type_multicast, ETH_ALEN); |
| return 0; |
| } |
| if (strcasecmp(from, "Broadcast") == 0) { |
| memcpy(to, mac_type_broadcast, ETH_ALEN); |
| memcpy(mask, msk_type_broadcast, ETH_ALEN); |
| return 0; |
| } |
| if (strcasecmp(from, "BGA") == 0) { |
| memcpy(to, mac_type_bridge_group, ETH_ALEN); |
| memcpy(mask, msk_type_bridge_group, ETH_ALEN); |
| return 0; |
| } |
| if ( (p = strrchr(from, '/')) != NULL) { |
| *p = '\0'; |
| if (!(addr = ether_aton(p + 1))) |
| return -1; |
| memcpy(mask, addr, ETH_ALEN); |
| } else |
| memset(mask, 0xff, ETH_ALEN); |
| if (!(addr = ether_aton(from))) |
| return -1; |
| memcpy(to, addr, ETH_ALEN); |
| for (i = 0; i < ETH_ALEN; i++) |
| to[i] &= mask[i]; |
| return 0; |
| } |
| |
| /* 0: default |
| * 1: the inverse '!' of the option has already been specified */ |
| int ebt_invert = 0; |
| |
| /* |
| * Check if the inverse of the option is specified. This is used |
| * in the parse functions of the extensions and ebtables.c |
| */ |
| int _ebt_check_inverse(const char option[], int argc, char **argv) |
| { |
| if (!option) |
| return ebt_invert; |
| if (strcmp(option, "!") == 0) { |
| if (ebt_invert == 1) |
| ebt_print_error("Double use of '!' not allowed"); |
| if (optind >= argc) |
| optarg = NULL; |
| else |
| optarg = argv[optind]; |
| optind++; |
| ebt_invert = 1; |
| return 1; |
| } |
| return ebt_invert; |
| } |
| |
| /* Make sure the same option wasn't specified twice. This is used |
| * in the parse functions of the extensions and ebtables.c */ |
| void ebt_check_option(unsigned int *flags, unsigned int mask) |
| { |
| if (*flags & mask) |
| ebt_print_error("Multiple use of same option not allowed"); |
| *flags |= mask; |
| } |
| |
| /* Put the ip string into 4 bytes. */ |
| static int undot_ip(char *ip, unsigned char *ip2) |
| { |
| char *p, *q, *end; |
| long int onebyte; |
| int i; |
| char buf[20]; |
| |
| strncpy(buf, ip, sizeof(buf) - 1); |
| |
| p = buf; |
| for (i = 0; i < 3; i++) { |
| if ((q = strchr(p, '.')) == NULL) |
| return -1; |
| *q = '\0'; |
| onebyte = strtol(p, &end, 10); |
| if (*end != '\0' || onebyte > 255 || onebyte < 0) |
| return -1; |
| ip2[i] = (unsigned char)onebyte; |
| p = q + 1; |
| } |
| |
| onebyte = strtol(p, &end, 10); |
| if (*end != '\0' || onebyte > 255 || onebyte < 0) |
| return -1; |
| ip2[3] = (unsigned char)onebyte; |
| |
| return 0; |
| } |
| |
| /* Put the mask into 4 bytes. */ |
| static int ip_mask(char *mask, unsigned char *mask2) |
| { |
| char *end; |
| long int bits; |
| uint32_t mask22; |
| |
| if (undot_ip(mask, mask2)) { |
| /* not the /a.b.c.e format, maybe the /x format */ |
| bits = strtol(mask, &end, 10); |
| if (*end != '\0' || bits > 32 || bits < 0) |
| return -1; |
| if (bits != 0) { |
| mask22 = htonl(0xFFFFFFFF << (32 - bits)); |
| memcpy(mask2, &mask22, 4); |
| } else { |
| mask22 = 0xFFFFFFFF; |
| memcpy(mask2, &mask22, 4); |
| } |
| } |
| return 0; |
| } |
| |
| /* Set the ip mask and ip address. Callers should check ebt_errormsg[0]. |
| * The string pointed to by address can be altered. */ |
| void ebt_parse_ip_address(char *address, uint32_t *addr, uint32_t *msk) |
| { |
| char *p; |
| |
| /* first the mask */ |
| if ((p = strrchr(address, '/')) != NULL) { |
| *p = '\0'; |
| if (ip_mask(p + 1, (unsigned char *)msk)) { |
| ebt_print_error("Problem with the IP mask '%s'", p + 1); |
| return; |
| } |
| } else |
| *msk = 0xFFFFFFFF; |
| |
| if (undot_ip(address, (unsigned char *)addr)) { |
| ebt_print_error("Problem with the IP address '%s'", address); |
| return; |
| } |
| *addr = *addr & *msk; |
| } |
| |
| |
| /* Transform the ip mask into a string ready for output. */ |
| char *ebt_mask_to_dotted(uint32_t mask) |
| { |
| int i; |
| static char buf[20]; |
| uint32_t maskaddr, bits; |
| |
| maskaddr = ntohl(mask); |
| |
| /* don't print /32 */ |
| if (mask == 0xFFFFFFFFL) { |
| *buf = '\0'; |
| return buf; |
| } |
| |
| i = 32; |
| bits = 0xFFFFFFFEL; /* Case 0xFFFFFFFF has just been dealt with */ |
| while (--i >= 0 && maskaddr != bits) |
| bits <<= 1; |
| |
| if (i > 0) |
| sprintf(buf, "/%d", i); |
| else if (!i) |
| *buf = '\0'; |
| else |
| /* Mask was not a decent combination of 1's and 0's */ |
| sprintf(buf, "/%d.%d.%d.%d", ((unsigned char *)&mask)[0], |
| ((unsigned char *)&mask)[1], ((unsigned char *)&mask)[2], |
| ((unsigned char *)&mask)[3]); |
| |
| return buf; |
| } |
| |
| /* Most of the following code is derived from iptables */ |
| static void |
| in6addrcpy(struct in6_addr *dst, struct in6_addr *src) |
| { |
| memcpy(dst, src, sizeof(struct in6_addr)); |
| } |
| |
| int string_to_number_ll(const char *s, unsigned long long min, |
| unsigned long long max, unsigned long long *ret) |
| { |
| unsigned long long number; |
| char *end; |
| |
| /* Handle hex, octal, etc. */ |
| errno = 0; |
| number = strtoull(s, &end, 0); |
| if (*end == '\0' && end != s) { |
| /* we parsed a number, let's see if we want this */ |
| if (errno != ERANGE && min <= number && (!max || number <= max)) { |
| *ret = number; |
| return 0; |
| } |
| } |
| return -1; |
| } |
| |
| int string_to_number_l(const char *s, unsigned long min, unsigned long max, |
| unsigned long *ret) |
| { |
| int result; |
| unsigned long long number; |
| |
| result = string_to_number_ll(s, min, max, &number); |
| *ret = (unsigned long)number; |
| |
| return result; |
| } |
| |
| int string_to_number(const char *s, unsigned int min, unsigned int max, |
| unsigned int *ret) |
| { |
| int result; |
| unsigned long number; |
| |
| result = string_to_number_l(s, min, max, &number); |
| *ret = (unsigned int)number; |
| |
| return result; |
| } |
| |
| static struct in6_addr *numeric_to_addr(const char *num) |
| { |
| static struct in6_addr ap; |
| int err; |
| |
| if ((err=inet_pton(AF_INET6, num, &ap)) == 1) |
| return ≈ |
| return (struct in6_addr *)NULL; |
| } |
| |
| static struct in6_addr *parse_ip6_mask(char *mask) |
| { |
| static struct in6_addr maskaddr; |
| struct in6_addr *addrp; |
| unsigned int bits; |
| |
| if (mask == NULL) { |
| /* no mask at all defaults to 128 bits */ |
| memset(&maskaddr, 0xff, sizeof maskaddr); |
| return &maskaddr; |
| } |
| if ((addrp = numeric_to_addr(mask)) != NULL) |
| return addrp; |
| if (string_to_number(mask, 0, 128, &bits) == -1) |
| ebt_print_error("Invalid IPv6 Mask '%s' specified", mask); |
| if (bits != 0) { |
| char *p = (char *)&maskaddr; |
| memset(p, 0xff, bits / 8); |
| memset(p + (bits / 8) + 1, 0, (128 - bits) / 8); |
| p[bits / 8] = 0xff << (8 - (bits & 7)); |
| return &maskaddr; |
| } |
| |
| memset(&maskaddr, 0, sizeof maskaddr); |
| return &maskaddr; |
| } |
| |
| /* Set the ipv6 mask and address. Callers should check ebt_errormsg[0]. |
| * The string pointed to by address can be altered. */ |
| void ebt_parse_ip6_address(char *address, struct in6_addr *addr, |
| struct in6_addr *msk) |
| { |
| struct in6_addr *tmp_addr; |
| char buf[256]; |
| char *p; |
| int i; |
| int err; |
| |
| strncpy(buf, address, sizeof(buf) - 1); |
| /* first the mask */ |
| buf[sizeof(buf) - 1] = '\0'; |
| if ((p = strrchr(buf, '/')) != NULL) { |
| *p = '\0'; |
| tmp_addr = parse_ip6_mask(p + 1); |
| } else |
| tmp_addr = parse_ip6_mask(NULL); |
| in6addrcpy(msk, tmp_addr); |
| |
| /* if a null mask is given, the name is ignored, like in "any/0" */ |
| if (!memcmp(msk, &in6addr_any, sizeof(in6addr_any))) |
| strcpy(buf, "::"); |
| |
| if ((err=inet_pton(AF_INET6, buf, addr)) < 1) { |
| ebt_print_error("Invalid IPv6 Address '%s' specified", buf); |
| return; |
| } |
| |
| for (i = 0; i < 4; i++) |
| addr->s6_addr32[i] &= msk->s6_addr32[i]; |
| } |
| |
| /* Transform the ip6 addr into a string ready for output. */ |
| char *ebt_ip6_to_numeric(const struct in6_addr *addrp) |
| { |
| /* 0000:0000:0000:0000:0000:000.000.000.000 |
| * 0000:0000:0000:0000:0000:0000:0000:0000 */ |
| static char buf[50+1]; |
| return (char *)inet_ntop(AF_INET6, addrp, buf, sizeof(buf)); |
| } |
| |
| int ebt_ip6mask_to_cidr(const struct in6_addr *k) |
| { |
| unsigned int bits = 0; |
| uint32_t a, b, c, d; |
| |
| a = ntohl(k->s6_addr32[0]); |
| b = ntohl(k->s6_addr32[1]); |
| c = ntohl(k->s6_addr32[2]); |
| d = ntohl(k->s6_addr32[3]); |
| while (a & 0x80000000U) { |
| ++bits; |
| a <<= 1; |
| a |= (b >> 31) & 1; |
| b <<= 1; |
| b |= (c >> 31) & 1; |
| c <<= 1; |
| c |= (d >> 31) & 1; |
| d <<= 1; |
| } |
| if (a != 0 || b != 0 || c != 0 || d != 0) |
| return -1; |
| return bits; |
| } |
| |
| char *ebt_ip6_mask_to_string(const struct in6_addr *msk) |
| { |
| int l = ebt_ip6mask_to_cidr(msk); |
| static char buf[51+1]; |
| |
| if (l == 128) |
| *buf = '\0'; |
| else if (l == -1) |
| sprintf(buf, "/%s", ebt_ip6_to_numeric(msk)); |
| else |
| sprintf(buf, "/%d", l); |
| return buf; |
| } |
| |
| static char* |
| parse_num(const char *str, long min, long max, long *num) |
| { |
| char *end; |
| |
| errno = 0; |
| *num = strtol(str, &end, 10); |
| if (errno && (*num == LONG_MIN || *num == LONG_MAX)) { |
| ebt_print_error("Invalid number %s: %s", str, strerror(errno)); |
| return NULL; |
| } |
| if (min <= max) { |
| if (*num > max || *num < min) { |
| ebt_print_error("Value %ld out of range (%ld, %ld)", *num, min, max); |
| return NULL; |
| } |
| } |
| if (*num == 0 && str == end) |
| return NULL; |
| return end; |
| } |
| |
| static char * |
| parse_range(const char *str, long min, long max, long num[]) |
| { |
| char *next; |
| |
| next = parse_num(str, min, max, num); |
| if (next == NULL) |
| return NULL; |
| if (next && *next == ':') |
| next = parse_num(next+1, min, max, &num[1]); |
| else |
| num[1] = num[0]; |
| return next; |
| } |
| |
| int ebt_parse_icmp(const struct ebt_icmp_names *icmp_codes, size_t n_codes, |
| const char *icmptype, uint8_t type[], uint8_t code[]) |
| { |
| unsigned int match = n_codes; |
| unsigned int i; |
| long number[2]; |
| |
| for (i = 0; i < n_codes; i++) { |
| if (strncasecmp(icmp_codes[i].name, icmptype, strlen(icmptype))) |
| continue; |
| if (match != n_codes) |
| ebt_print_error("Ambiguous ICMP type `%s':" |
| " `%s' or `%s'?", |
| icmptype, icmp_codes[match].name, |
| icmp_codes[i].name); |
| match = i; |
| } |
| |
| if (match < n_codes) { |
| type[0] = type[1] = icmp_codes[match].type; |
| if (code) { |
| code[0] = icmp_codes[match].code_min; |
| code[1] = icmp_codes[match].code_max; |
| } |
| } else { |
| char *next = parse_range(icmptype, 0, 255, number); |
| if (!next) { |
| ebt_print_error("Unknown ICMP type `%s'", |
| icmptype); |
| return -1; |
| } |
| type[0] = (uint8_t) number[0]; |
| type[1] = (uint8_t) number[1]; |
| switch (*next) { |
| case 0: |
| if (code) { |
| code[0] = 0; |
| code[1] = 255; |
| } |
| return 0; |
| case '/': |
| if (code) { |
| next = parse_range(next+1, 0, 255, number); |
| code[0] = (uint8_t) number[0]; |
| code[1] = (uint8_t) number[1]; |
| if (next == NULL) |
| return -1; |
| if (next && *next == 0) |
| return 0; |
| } |
| /* fallthrough */ |
| default: |
| ebt_print_error("unknown character %c", *next); |
| return -1; |
| } |
| } |
| return 0; |
| } |
| |
| static void print_icmp_code(uint8_t *code) |
| { |
| if (!code) |
| return; |
| |
| if (code[0] == code[1]) |
| printf("/%"PRIu8 " ", code[0]); |
| else |
| printf("/%"PRIu8":%"PRIu8 " ", code[0], code[1]); |
| } |
| |
| void ebt_print_icmp_type(const struct ebt_icmp_names *icmp_codes, |
| size_t n_codes, uint8_t *type, uint8_t *code) |
| { |
| unsigned int i; |
| |
| if (type[0] != type[1]) { |
| printf("%"PRIu8 ":%" PRIu8, type[0], type[1]); |
| print_icmp_code(code); |
| return; |
| } |
| |
| for (i = 0; i < n_codes; i++) { |
| if (icmp_codes[i].type != type[0]) |
| continue; |
| |
| if (!code || (icmp_codes[i].code_min == code[0] && |
| icmp_codes[i].code_max == code[1])) { |
| printf("%s ", icmp_codes[i].name); |
| return; |
| } |
| } |
| printf("%"PRIu8, type[0]); |
| print_icmp_code(code); |
| } |
| |
| void ebt_print_icmp_types(const struct ebt_icmp_names *icmp_codes, |
| size_t n_codes) |
| { |
| unsigned int i; |
| |
| for (i = 0; i < n_codes; i++) { |
| if (i && icmp_codes[i].type == icmp_codes[i-1].type) { |
| if (icmp_codes[i].code_min == icmp_codes[i-1].code_min |
| && (icmp_codes[i].code_max |
| == icmp_codes[i-1].code_max)) |
| printf(" (%s)", icmp_codes[i].name); |
| else |
| printf("\n %s", icmp_codes[i].name); |
| } |
| else |
| printf("\n%s", icmp_codes[i].name); |
| } |
| printf("\n"); |
| } |