#include <linux/netfilter_ipv6/ip6_tables.h>
 #include <linux/netfilter/xt_hashlimit.h>
 #include <linux/mutex.h>
+#include <linux/kernel.h>
 
 MODULE_LICENSE("GPL");
 MODULE_AUTHOR("Harald Welte <laforge@netfilter.org>");
        }
 }
 
-static u64 user2rate_bytes(u64 user)
+static u64 user2rate_bytes(u32 user)
 {
        u64 r;
 
-       r = user ? 0xFFFFFFFFULL / user : 0xFFFFFFFFULL;
-       r = (r - 1) << 4;
+       r = user ? U32_MAX / user : U32_MAX;
+       r = (r - 1) << XT_HASHLIMIT_BYTE_SHIFT;
        return r;
 }
 
                dh->rateinfo.prev_window = 0;
                dh->rateinfo.current_rate = 0;
                if (hinfo->cfg.mode & XT_HASHLIMIT_BYTES) {
-                       dh->rateinfo.rate = user2rate_bytes(hinfo->cfg.avg);
+                       dh->rateinfo.rate =
+                               user2rate_bytes((u32)hinfo->cfg.avg);
                        if (hinfo->cfg.burst)
                                dh->rateinfo.burst =
                                        hinfo->cfg.burst * dh->rateinfo.rate;
 
        /* Check for overflow. */
        if (revision >= 3 && cfg->mode & XT_HASHLIMIT_RATE_MATCH) {
-               if (cfg->avg == 0) {
+               if (cfg->avg == 0 || cfg->avg > U32_MAX) {
                        pr_info("hashlimit invalid rate\n");
                        return -ERANGE;
                }