net/ipv6/netfilter/ip6t_mh.c

   1 /*
   2  * Copyright (C)2006 USAGI/WIDE Project
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License version 2 as
   6  * published by the Free Software Foundation.
   7  *
   8  * Author:
   9  *      Masahide NAKAMURA @USAGI <masahide.nakamura.cz@hitachi.com>
  10  *
  11  * Based on net/netfilter/xt_tcpudp.c
  12  *
  13  */
  14 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  15 #include <linux/types.h>
  16 #include <linux/module.h>
  17 #include <net/ip.h>
  18 #include <linux/ipv6.h>
  19 #include <net/ipv6.h>
  20 #include <net/mip6.h>
  21
  22 #include <linux/netfilter/x_tables.h>
  23 #include <linux/netfilter_ipv6/ip6t_mh.h>
  24
  25 MODULE_DESCRIPTION("Xtables: IPv6 Mobility Header match");
  26 MODULE_LICENSE("GPL");
  27
  28 /* Returns 1 if the type is matched by the range, 0 otherwise */
  29 static inline bool
  30 type_match(u_int8_t min, u_int8_t max, u_int8_t type, bool invert)
  31 {
  32         return (type >= min && type <= max) ^ invert;
  33 }
  34
  35 static bool mh_mt6(const struct sk_buff *skb,
  36                    const struct xt_action_param *par)
  37 {
  38         struct ip6_mh _mh;
  39         const struct ip6_mh *mh;
  40         const struct ip6t_mh *mhinfo = par->matchinfo;
  41
  42         /* Must not be a fragment. */
  43         if (par->fragoff != 0)
  44                 return false;
  45
  46         mh = skb_header_pointer(skb, par->thoff, sizeof(_mh), &_mh);
  47         if (mh == NULL) {
  48                 /* We've been asked to examine this packet, and we
  49                    can't.  Hence, no choice but to drop. */
  50                 pr_debug("Dropping evil MH tinygram.\n");
  51                 *par->hotdrop = true;
  52                 return false;
  53         }
  54
  55         if (mh->ip6mh_proto != IPPROTO_NONE) {
  56                 pr_debug("Dropping invalid MH Payload Proto: %u\n",
  57                          mh->ip6mh_proto);
  58                 *par->hotdrop = true;
  59                 return false;
  60         }
  61
  62         return type_match(mhinfo->types[0], mhinfo->types[1], mh->ip6mh_type,
  63                           !!(mhinfo->invflags & IP6T_MH_INV_TYPE));
  64 }
  65
  66 static int mh_mt6_check(const struct xt_mtchk_param *par)
  67 {
  68         const struct ip6t_mh *mhinfo = par->matchinfo;
  69
  70         /* Must specify no unknown invflags */
  71         return (mhinfo->invflags & ~IP6T_MH_INV_MASK) ? -EINVAL : 0;
  72 }
  73
  74 static struct xt_match mh_mt6_reg __read_mostly = {
  75         .name           = "mh",
  76         .family         = NFPROTO_IPV6,
  77         .checkentry     = mh_mt6_check,
  78         .match          = mh_mt6,
  79         .matchsize      = sizeof(struct ip6t_mh),
  80         .proto          = IPPROTO_MH,
  81         .me             = THIS_MODULE,
  82 };
  83
  84 static int __init mh_mt6_init(void)
  85 {
  86         return xt_register_match(&mh_mt6_reg);
  87 }
  88
  89 static void __exit mh_mt6_exit(void)
  90 {
  91         xt_unregister_match(&mh_mt6_reg);
  92 }
  93
  94 module_init(mh_mt6_init);
  95 module_exit(mh_mt6_exit);