#include <linux/netfilter_bridge.h>
 #include <linux/seq_file.h>
 #include <linux/rcupdate.h>
+#include <linux/netfilter_ipv4.h>
+#include <linux/netfilter_ipv6.h>
 #include <net/protocol.h>
 #include <net/netfilter/nf_queue.h>
 #include <net/dst.h>
 {
        int status = -ENOENT;
        struct nf_queue_entry *entry = NULL;
-       const struct nf_afinfo *afinfo;
        const struct nf_queue_handler *qh;
        struct net *net = state->net;
+       unsigned int route_key_size;
 
        /* QUEUE == DROP if no one is waiting, to be safe. */
        qh = rcu_dereference(net->nf.queue_handler);
                goto err;
        }
 
-       afinfo = nf_get_afinfo(state->pf);
-       if (!afinfo)
-               goto err;
+       switch (state->pf) {
+       case AF_INET:
+               route_key_size = sizeof(struct ip_rt_info);
+               break;
+       case AF_INET6:
+               route_key_size = sizeof(struct ip6_rt_info);
+               break;
+       default:
+               route_key_size = 0;
+               break;
+       }
 
-       entry = kmalloc(sizeof(*entry) + afinfo->route_key_size, GFP_ATOMIC);
+       entry = kmalloc(sizeof(*entry) + route_key_size, GFP_ATOMIC);
        if (!entry) {
                status = -ENOMEM;
                goto err;
                .skb    = skb,
                .state  = *state,
                .hook_index = index,
-               .size   = sizeof(*entry) + afinfo->route_key_size,
+               .size   = sizeof(*entry) + route_key_size,
        };
 
        nf_queue_entry_get_refs(entry);