#define _trace_xdp_redirect_map(dev, xdp, fwd, map, idx)               \
         trace_xdp_redirect_map(dev, xdp, fwd ? fwd->ifindex : 0,       \
-                               0, map, idx);
+                               0, map, idx)
 
 #define _trace_xdp_redirect_map_err(dev, xdp, fwd, map, idx, err)      \
         trace_xdp_redirect_map_err(dev, xdp, fwd ? fwd->ifindex : 0,   \
-                                   err, map, idx);
+                                   err, map, idx)
 
 #endif /* _TRACE_XDP_H */
 
 
        struct redirect_info *ri = this_cpu_ptr(&redirect_info);
        const struct bpf_prog *map_owner = ri->map_owner;
        struct bpf_map *map = ri->map;
+       struct net_device *fwd = NULL;
        u32 index = ri->ifindex;
-       struct net_device *fwd;
        int err;
 
        ri->ifindex = 0;
        ri->map = NULL;
        ri->map_owner = NULL;
 
-       /* This is really only caused by a deliberately crappy
-        * BPF program, normally we would never hit that case,
-        * so no need to inform someone via tracepoints either,
-        * just bail out.
-        */
-       if (unlikely(map_owner != xdp_prog))
-               return -EINVAL;
+       if (unlikely(map_owner != xdp_prog)) {
+               err = -EFAULT;
+               map = NULL;
+               goto err;
+       }
 
        fwd = __dev_map_lookup_elem(map, index);
        if (!fwd) {
                            struct bpf_prog *xdp_prog)
 {
        struct redirect_info *ri = this_cpu_ptr(&redirect_info);
+       const struct bpf_prog *map_owner = ri->map_owner;
+       struct bpf_map *map = ri->map;
+       struct net_device *fwd = NULL;
        u32 index = ri->ifindex;
-       struct net_device *fwd;
        unsigned int len;
        int err = 0;
 
-       fwd = dev_get_by_index_rcu(dev_net(dev), index);
        ri->ifindex = 0;
+       ri->map = NULL;
+       ri->map_owner = NULL;
+
+       if (map) {
+               if (unlikely(map_owner != xdp_prog)) {
+                       err = -EFAULT;
+                       map = NULL;
+                       goto err;
+               }
+               fwd = __dev_map_lookup_elem(map, index);
+       } else {
+               fwd = dev_get_by_index_rcu(dev_net(dev), index);
+       }
        if (unlikely(!fwd)) {
                err = -EINVAL;
                goto err;
        }
 
        skb->dev = fwd;
-       _trace_xdp_redirect(dev, xdp_prog, index);
+       map ? _trace_xdp_redirect_map(dev, xdp_prog, fwd, map, index)
+               : _trace_xdp_redirect(dev, xdp_prog, index);
        return 0;
 err:
-       _trace_xdp_redirect_err(dev, xdp_prog, index, err);
+       map ? _trace_xdp_redirect_map_err(dev, xdp_prog, fwd, map, index, err)
+               : _trace_xdp_redirect_err(dev, xdp_prog, index, err);
        return err;
 }
 EXPORT_SYMBOL_GPL(xdp_do_generic_redirect);