2 files changed, 12 insertions, 4 deletions

diff --git a/include/net/ip6_fib.h b/include/net/ip6_fib.h
index 5a4a67b38712..40105738e2f6 100644
--- a/include/net/ip6_fib.h
+++ b/include/net/ip6_fib.h
@@ -195,6 +195,7 @@ struct fib6_result {
         struct fib6_info        *f6i;
         u32                     fib6_flags;
         u8                      fib6_type;
+        struct rt6_info         *rt6;
 };
 
 #define for_each_fib6_node_rt_rcu(fn)                                   \
diff --git a/net/ipv6/fib6_rules.c b/net/ipv6/fib6_rules.c
index dbedbe655c91..06d1b7763600 100644
--- a/net/ipv6/fib6_rules.c
+++ b/net/ipv6/fib6_rules.c
@@ -94,9 +94,11 @@ struct dst_entry *fib6_rule_lookup(struct net *net, struct flowi6 *fl6,
                                    int flags, pol_lookup_t lookup)
 {
         if (net->ipv6.fib6_has_custom_rules) {
+                struct fib6_result res = {};
                 struct fib_lookup_arg arg = {
                         .lookup_ptr = lookup,
                         .lookup_data = skb,
+                        .result = &res,
                         .flags = FIB_LOOKUP_NOREF,
                 };
 
@@ -106,8 +108,8 @@ struct dst_entry *fib6_rule_lookup(struct net *net, struct flowi6 *fl6,
                 fib_rules_lookup(net->ipv6.fib6_rules_ops,
                                  flowi6_to_flowi(fl6), flags, &arg);
 
-                if (arg.result)
-                        return arg.result;
+                if (res.rt6)
+                        return &res.rt6->dst;
         } else {
                 struct rt6_info *rt;
 
@@ -191,6 +193,7 @@ static int fib6_rule_action_alt(struct fib_rule *rule, struct flowi *flp,
 static int __fib6_rule_action(struct fib_rule *rule, struct flowi *flp,
                               int flags, struct fib_lookup_arg *arg)
 {
+        struct fib6_result *res = arg->result;
         struct flowi6 *flp6 = &flp->u.ip6;
         struct rt6_info *rt = NULL;
         struct fib6_table *table;
@@ -245,7 +248,7 @@ again:
 discard_pkt:
         dst_hold(&rt->dst);
 out:
-        arg->result = rt;
+        res->rt6 = rt;
         return err;
 }
 
@@ -260,9 +263,13 @@ static int fib6_rule_action(struct fib_rule *rule, struct flowi *flp,
 
 static bool fib6_rule_suppress(struct fib_rule *rule, struct fib_lookup_arg *arg)
 {
-        struct rt6_info *rt = (struct rt6_info *) arg->result;
+        struct fib6_result *res = arg->result;
+        struct rt6_info *rt = res->rt6;
         struct net_device *dev = NULL;
 
+        if (!rt)
+                return false;
+
         if (rt->rt6i_idev)
                 dev = rt->rt6i_idev->dev;