1/* Shared library add-on to iptables to add ESP support. */ 2#include <stdio.h> 3#include <netdb.h> 4#include <string.h> 5#include <stdlib.h> 6#include <getopt.h> 7#include <errno.h> 8#include <iptables.h> 9#include <linux/netfilter_ipv4/ipt_esp.h> 10 11/* Function which prints out usage message. */ 12static void 13help(void) 14{ 15 printf( 16"ESP v%s options:\n" 17" --espspi [!] spi[:spi]\n" 18" match spi (range)\n", 19IPTABLES_VERSION); 20} 21 22static struct option opts[] = { 23 { "espspi", 1, 0, '1' }, 24 {0} 25}; 26 27static u_int32_t 28parse_esp_spi(const char *spistr) 29{ 30 unsigned long int spi; 31 char* ep; 32 33 spi = strtoul(spistr,&ep,0) ; 34 35 if ( spistr == ep ) { 36 exit_error(PARAMETER_PROBLEM, 37 "ESP no valid digits in spi `%s'", spistr); 38 } 39 if ( spi == ULONG_MAX && errno == ERANGE ) { 40 exit_error(PARAMETER_PROBLEM, 41 "spi `%s' specified too big: would overflow", spistr); 42 } 43 if ( *spistr != '\0' && *ep != '\0' ) { 44 exit_error(PARAMETER_PROBLEM, 45 "ESP error parsing spi `%s'", spistr); 46 } 47 return (u_int32_t) spi; 48} 49 50static void 51parse_esp_spis(const char *spistring, u_int32_t *spis) 52{ 53 char *buffer; 54 char *cp; 55 56 buffer = strdup(spistring); 57 if ((cp = strchr(buffer, ':')) == NULL) 58 spis[0] = spis[1] = parse_esp_spi(buffer); 59 else { 60 *cp = '\0'; 61 cp++; 62 63 spis[0] = buffer[0] ? parse_esp_spi(buffer) : 0; 64 spis[1] = cp[0] ? parse_esp_spi(cp) : 0xFFFFFFFF; 65 if (spis[0] > spis[1]) 66 exit_error(PARAMETER_PROBLEM, 67 "Invalid ESP spi range: %s", spistring); 68 } 69 free(buffer); 70} 71 72/* Initialize the match. */ 73static void 74init(struct ipt_entry_match *m, unsigned int *nfcache) 75{ 76 struct ipt_esp *espinfo = (struct ipt_esp *)m->data; 77 78 espinfo->spis[1] = 0xFFFFFFFF; 79} 80 81#define ESP_SPI 0x01 82 83/* Function which parses command options; returns true if it 84 ate an option */ 85static int 86parse(int c, char **argv, int invert, unsigned int *flags, 87 const struct ipt_entry *entry, 88 unsigned int *nfcache, 89 struct ipt_entry_match **match) 90{ 91 struct ipt_esp *espinfo = (struct ipt_esp *)(*match)->data; 92 93 switch (c) { 94 case '1': 95 if (*flags & ESP_SPI) 96 exit_error(PARAMETER_PROBLEM, 97 "Only one `--espspi' allowed"); 98 check_inverse(optarg, &invert, &optind, 0); 99 parse_esp_spis(argv[optind-1], espinfo->spis); 100 if (invert) 101 espinfo->invflags |= IPT_ESP_INV_SPI; 102 *flags |= ESP_SPI; 103 break; 104 default: 105 return 0; 106 } 107 108 return 1; 109} 110 111/* Final check; we don't care. */ 112static void 113final_check(unsigned int flags) 114{ 115} 116 117static void 118print_spis(const char *name, u_int32_t min, u_int32_t max, 119 int invert) 120{ 121 const char *inv = invert ? "!" : ""; 122 123 if (min != 0 || max != 0xFFFFFFFF || invert) { 124 printf("%s", name); 125 if (min == max) { 126 printf(":%s", inv); 127 printf("%u", min); 128 } else { 129 printf("s:%s", inv); 130 printf("%u",min); 131 printf(":"); 132 printf("%u",max); 133 } 134 printf(" "); 135 } 136} 137 138/* Prints out the union ipt_matchinfo. */ 139static void 140print(const struct ipt_ip *ip, 141 const struct ipt_entry_match *match, int numeric) 142{ 143 const struct ipt_esp *esp = (struct ipt_esp *)match->data; 144 145 printf("esp "); 146 print_spis("spi", esp->spis[0], esp->spis[1], 147 esp->invflags & IPT_ESP_INV_SPI); 148 if (esp->invflags & ~IPT_ESP_INV_MASK) 149 printf("Unknown invflags: 0x%X ", 150 esp->invflags & ~IPT_ESP_INV_MASK); 151} 152 153/* Saves the union ipt_matchinfo in parsable form to stdout. */ 154static void save(const struct ipt_ip *ip, const struct ipt_entry_match *match) 155{ 156 const struct ipt_esp *espinfo = (struct ipt_esp *)match->data; 157 158 if (!(espinfo->spis[0] == 0 159 && espinfo->spis[1] == 0xFFFFFFFF)) { 160 printf("--espspi %s", 161 (espinfo->invflags & IPT_ESP_INV_SPI) ? "! " : ""); 162 if (espinfo->spis[0] 163 != espinfo->spis[1]) 164 printf("%u:%u ", 165 espinfo->spis[0], 166 espinfo->spis[1]); 167 else 168 printf("%u ", 169 espinfo->spis[0]); 170 } 171 172} 173 174static struct iptables_match esp = { 175 .next = NULL, 176 .name = "esp", 177 .version = IPTABLES_VERSION, 178 .size = IPT_ALIGN(sizeof(struct ipt_esp)), 179 .userspacesize = IPT_ALIGN(sizeof(struct ipt_esp)), 180 .help = &help, 181 .init = &init, 182 .parse = &parse, 183 .final_check = &final_check, 184 .print = &print, 185 .save = &save, 186 .extra_opts = opts 187}; 188 189void 190_init(void) 191{ 192 register_match(&esp); 193} 194