extensions/libipt_icmp.c

   1 /* Shared library add-on to iptables to add ICMP support. */
   2 #include <stdio.h>
   3 #include <netdb.h>
   4 #include <string.h>
   5 #include <stdlib.h>
   6 #include <getopt.h>
   7 #include <iptables.h>
   8 #include <linux/netfilter_ipv4/ip_tables.h>
   9
  10 /* special hack for icmp-type 'any':
  11  * Up to kernel <=2.4.20 the problem was:
  12  * '-p icmp ' matches all icmp packets
  13  * '-p icmp -m icmp' matches _only_ ICMP type 0 :(
  14  * This is now fixed by initializing the field * to icmp type 0xFF
  15  * See: https://bugzilla.netfilter.org/cgi-bin/bugzilla/show_bug.cgi?id=37
  16  */
  17
  18 struct icmp_names {
  19         const char *name;
  20         u_int8_t type;
  21         u_int8_t code_min, code_max;
  22 };
  23
  24 static const struct icmp_names icmp_codes[] = {
  25         { "any", 0xFF, 0, 0xFF },
  26         { "echo-reply", 0, 0, 0xFF },
  27         /* Alias */ { "pong", 0, 0, 0xFF },
  28
  29         { "destination-unreachable", 3, 0, 0xFF },
  30         {   "network-unreachable", 3, 0, 0 },
  31         {   "host-unreachable", 3, 1, 1 },
  32         {   "protocol-unreachable", 3, 2, 2 },
  33         {   "port-unreachable", 3, 3, 3 },
  34         {   "fragmentation-needed", 3, 4, 4 },
  35         {   "source-route-failed", 3, 5, 5 },
  36         {   "network-unknown", 3, 6, 6 },
  37         {   "host-unknown", 3, 7, 7 },
  38         {   "network-prohibited", 3, 9, 9 },
  39         {   "host-prohibited", 3, 10, 10 },
  40         {   "TOS-network-unreachable", 3, 11, 11 },
  41         {   "TOS-host-unreachable", 3, 12, 12 },
  42         {   "communication-prohibited", 3, 13, 13 },
  43         {   "host-precedence-violation", 3, 14, 14 },
  44         {   "precedence-cutoff", 3, 15, 15 },
  45
  46         { "source-quench", 4, 0, 0xFF },
  47
  48         { "redirect", 5, 0, 0xFF },
  49         {   "network-redirect", 5, 0, 0 },
  50         {   "host-redirect", 5, 1, 1 },
  51         {   "TOS-network-redirect", 5, 2, 2 },
  52         {   "TOS-host-redirect", 5, 3, 3 },
  53
  54         { "echo-request", 8, 0, 0xFF },
  55         /* Alias */ { "ping", 8, 0, 0xFF },
  56
  57         { "router-advertisement", 9, 0, 0xFF },
  58
  59         { "router-solicitation", 10, 0, 0xFF },
  60
  61         { "time-exceeded", 11, 0, 0xFF },
  62         /* Alias */ { "ttl-exceeded", 11, 0, 0xFF },
  63         {   "ttl-zero-during-transit", 11, 0, 0 },
  64         {   "ttl-zero-during-reassembly", 11, 1, 1 },
  65
  66         { "parameter-problem", 12, 0, 0xFF },
  67         {   "ip-header-bad", 12, 0, 0 },
  68         {   "required-option-missing", 12, 1, 1 },
  69
  70         { "timestamp-request", 13, 0, 0xFF },
  71
  72         { "timestamp-reply", 14, 0, 0xFF },
  73
  74         { "address-mask-request", 17, 0, 0xFF },
  75
  76         { "address-mask-reply", 18, 0, 0xFF }
  77 };
  78
  79 static void
  80 print_icmptypes()
  81 {
  82         unsigned int i;
  83         printf("Valid ICMP Types:");
  84
  85         for (i = 0; i < sizeof(icmp_codes)/sizeof(struct icmp_names); i++) {
  86                 if (i && icmp_codes[i].type == icmp_codes[i-1].type) {
  87                         if (icmp_codes[i].code_min == icmp_codes[i-1].code_min
  88                             && (icmp_codes[i].code_max
  89                                 == icmp_codes[i-1].code_max))
  90                                 printf(" (%s)", icmp_codes[i].name);
  91                         else
  92                                 printf("\n   %s", icmp_codes[i].name);
  93                 }
  94                 else
  95                         printf("\n%s", icmp_codes[i].name);
  96         }
  97         printf("\n");
  98 }
  99
 100 /* Function which prints out usage message. */
 101 static void
 102 help(void)
 103 {
 104         printf(
 105 "ICMP v%s options:\n"
 106 " --icmp-type [!] typename      match icmp type\n"
 107 "                               (or numeric type or type/code)\n"
 108 "\n", IPTABLES_VERSION);
 109         print_icmptypes();
 110 }
 111
 112 static struct option opts[] = {
 113         { "icmp-type", 1, 0, '1' },
 114         {0}
 115 };
 116
 117 static void
 118 parse_icmp(const char *icmptype, u_int8_t *type, u_int8_t code[])
 119 {
 120         unsigned int limit = sizeof(icmp_codes)/sizeof(struct icmp_names);
 121         unsigned int match = limit;
 122         unsigned int i;
 123
 124         for (i = 0; i < limit; i++) {
 125                 if (strncasecmp(icmp_codes[i].name, icmptype, strlen(icmptype))
 126                     == 0) {
 127                         if (match != limit)
 128                                 exit_error(PARAMETER_PROBLEM,
 129                                            "Ambiguous ICMP type `%s':"
 130                                            " `%s' or `%s'?",
 131                                            icmptype,
 132                                            icmp_codes[match].name,
 133                                            icmp_codes[i].name);
 134                         match = i;
 135                 }
 136         }
 137
 138         if (match != limit) {
 139                 *type = icmp_codes[match].type;
 140                 code[0] = icmp_codes[match].code_min;
 141                 code[1] = icmp_codes[match].code_max;
 142         } else {
 143                 char *slash;
 144                 char buffer[strlen(icmptype) + 1];
 145                 unsigned int number;
 146
 147                 strcpy(buffer, icmptype);
 148                 slash = strchr(buffer, '/');
 149
 150                 if (slash)
 151                         *slash = '\0';
 152
 153                 if (string_to_number(buffer, 0, 255, &number) == -1)
 154                         exit_error(PARAMETER_PROBLEM,
 155                                    "Invalid ICMP type `%s'\n", buffer);
 156                 *type = number;
 157                 if (slash) {
 158                         if (string_to_number(slash+1, 0, 255, &number) == -1)
 159                                 exit_error(PARAMETER_PROBLEM,
 160                                            "Invalid ICMP code `%s'\n",
 161                                            slash+1);
 162                         code[0] = code[1] = number;
 163                 } else {
 164                         code[0] = 0;
 165                         code[1] = 0xFF;
 166                 }
 167         }
 168 }
 169
 170 /* Initialize the match. */
 171 static void
 172 init(struct ipt_entry_match *m, unsigned int *nfcache)
 173 {
 174         struct ipt_icmp *icmpinfo = (struct ipt_icmp *)m->data;
 175
 176         icmpinfo->type = 0xFF;
 177         icmpinfo->code[1] = 0xFF;
 178 }
 179
 180 /* Function which parses command options; returns true if it
 181    ate an option */
 182 static int
 183 parse(int c, char **argv, int invert, unsigned int *flags,
 184       const struct ipt_entry *entry,
 185       unsigned int *nfcache,
 186       struct ipt_entry_match **match)
 187 {
 188         struct ipt_icmp *icmpinfo = (struct ipt_icmp *)(*match)->data;
 189
 190         switch (c) {
 191         case '1':
 192                 check_inverse(optarg, &invert, &optind, 0);
 193                 parse_icmp(argv[optind-1], &icmpinfo->type,
 194                            icmpinfo->code);
 195                 if (invert)
 196                         icmpinfo->invflags |= IPT_ICMP_INV;
 197                 break;
 198
 199         default:
 200                 return 0;
 201         }
 202
 203         return 1;
 204 }
 205
 206 static void print_icmptype(u_int8_t type,
 207                            u_int8_t code_min, u_int8_t code_max,
 208                            int invert,
 209                            int numeric)
 210 {
 211         if (!numeric) {
 212                 unsigned int i;
 213
 214                 for (i = 0;
 215                      i < sizeof(icmp_codes)/sizeof(struct icmp_names);
 216                      i++) {
 217                         if (icmp_codes[i].type == type
 218                             && icmp_codes[i].code_min == code_min
 219                             && icmp_codes[i].code_max == code_max)
 220                                 break;
 221                 }
 222
 223                 if (i != sizeof(icmp_codes)/sizeof(struct icmp_names)) {
 224                         printf("%s%s ",
 225                                invert ? "!" : "",
 226                                icmp_codes[i].name);
 227                         return;
 228                 }
 229         }
 230
 231         if (invert)
 232                 printf("!");
 233
 234         printf("type %u", type);
 235         if (code_min == 0 && code_max == 0xFF)
 236                 printf(" ");
 237         else if (code_min == code_max)
 238                 printf(" code %u ", code_min);
 239         else
 240                 printf(" codes %u-%u ", code_min, code_max);
 241 }
 242
 243 /* Prints out the union ipt_matchinfo. */
 244 static void
 245 print(const struct ipt_ip *ip,
 246       const struct ipt_entry_match *match,
 247       int numeric)
 248 {
 249         const struct ipt_icmp *icmp = (struct ipt_icmp *)match->data;
 250
 251         printf("icmp ");
 252         print_icmptype(icmp->type, icmp->code[0], icmp->code[1],
 253                        icmp->invflags & IPT_ICMP_INV,
 254                        numeric);
 255
 256         if (icmp->invflags & ~IPT_ICMP_INV)
 257                 printf("Unknown invflags: 0x%X ",
 258                        icmp->invflags & ~IPT_ICMP_INV);
 259 }
 260
 261 /* Saves the match in parsable form to stdout. */
 262 static void save(const struct ipt_ip *ip, const struct ipt_entry_match *match)
 263 {
 264         const struct ipt_icmp *icmp = (struct ipt_icmp *)match->data;
 265
 266         if (icmp->invflags & IPT_ICMP_INV)
 267                 printf("! ");
 268
 269         /* special hack for 'any' case */
 270         if (icmp->type == 0xFF) {
 271                 printf("--icmp-type any ");
 272         } else {
 273                 printf("--icmp-type %u", icmp->type);
 274                 if (icmp->code[0] != 0 || icmp->code[1] != 0xFF)
 275                         printf("/%u", icmp->code[0]);
 276                 printf(" ");
 277         }
 278 }
 279
 280 /* Final check; we don't care. */
 281 static void final_check(unsigned int flags)
 282 {
 283 }
 284
 285 static struct iptables_match icmp = {
 286         .next           = NULL,
 287         .name           = "icmp",
 288         .version        = IPTABLES_VERSION,
 289         .size           = IPT_ALIGN(sizeof(struct ipt_icmp)),
 290         .userspacesize  = IPT_ALIGN(sizeof(struct ipt_icmp)),
 291         .help           = &help,
 292         .init           = &init,
 293         .parse          = &parse,
 294         .final_check    = &final_check,
 295         .print          = &print,
 296         .save           = &save,
 297         .extra_opts     = opts
 298 };
 299
 300 void _init(void)
 301 {
 302         register_match(&icmp);
 303 }