3 files changed, 18 insertions, 20 deletions

diff --git a/include/net/act_api.h b/include/net/act_api.h
index 4be8b0daedf0..b18c699681ca 100644
--- a/include/net/act_api.h
+++ b/include/net/act_api.h
@@ -101,8 +101,8 @@ struct tc_action_ops {
                         struct netlink_ext_ack *);
         void    (*stats_update)(struct tc_action *, u64, u32, u64, bool);
         size_t  (*get_fill_size)(const struct tc_action *act);
-        struct net_device *(*get_dev)(const struct tc_action *a);
-        void    (*put_dev)(struct net_device *dev);
+        struct net_device *(*get_dev)(const struct tc_action *a,
+                                      tc_action_priv_destructor *destructor);
         struct psample_group *
         (*get_psample_group)(const struct tc_action *a,
                              tc_action_priv_destructor *destructor);
diff --git a/net/sched/act_mirred.c b/net/sched/act_mirred.c
index 9d1bf508075a..9ce073a05414 100644
--- a/net/sched/act_mirred.c
+++ b/net/sched/act_mirred.c
@@ -408,25 +408,31 @@ static struct notifier_block mirred_device_notifier = {
         .notifier_call = mirred_device_event,
 };
 
-static struct net_device *tcf_mirred_get_dev(const struct tc_action *a)
+static void tcf_mirred_dev_put(void *priv)
+{
+        struct net_device *dev = priv;
+
+        dev_put(dev);
+}
+
+static struct net_device *
+tcf_mirred_get_dev(const struct tc_action *a,
+                   tc_action_priv_destructor *destructor)
 {
         struct tcf_mirred *m = to_mirred(a);
         struct net_device *dev;
 
         rcu_read_lock();
         dev = rcu_dereference(m->tcfm_dev);
-        if (dev)
+        if (dev) {
                 dev_hold(dev);
+                *destructor = tcf_mirred_dev_put;
+        }
         rcu_read_unlock();
 
         return dev;
 }
 
-static void tcf_mirred_put_dev(struct net_device *dev)
-{
-        dev_put(dev);
-}
-
 static size_t tcf_mirred_get_fill_size(const struct tc_action *act)
 {
         return nla_total_size(sizeof(struct tc_mirred));
@@ -446,7 +452,6 @@ static struct tc_action_ops act_mirred_ops = {
         .get_fill_size  =       tcf_mirred_get_fill_size,
         .size           =       sizeof(struct tcf_mirred),
         .get_dev        =       tcf_mirred_get_dev,
-        .put_dev        =       tcf_mirred_put_dev,
 };
 
 static __net_init int mirred_init_net(struct net *net)
diff --git a/net/sched/cls_api.c b/net/sched/cls_api.c
index 60d44b14750a..32577c248968 100644
--- a/net/sched/cls_api.c
+++ b/net/sched/cls_api.c
@@ -3288,22 +3288,15 @@ void tc_cleanup_flow_action(struct flow_action *flow_action)
 }
 EXPORT_SYMBOL(tc_cleanup_flow_action);
 
-static void tcf_mirred_put_dev(void *priv)
-{
-        struct net_device *dev = priv;
-
-        dev_put(dev);
-}
-
 static void tcf_mirred_get_dev(struct flow_action_entry *entry,
                                const struct tc_action *act)
 {
-        entry->dev = tcf_mirred_dev(act);
+#ifdef CONFIG_NET_CLS_ACT
+        entry->dev = act->ops->get_dev(act, &entry->destructor);
         if (!entry->dev)
                 return;
-        dev_hold(entry->dev);
-        entry->destructor = tcf_mirred_put_dev;
         entry->destructor_priv = entry->dev;
+#endif
 }
 
 static void tcf_tunnel_encap_put_tunnel(void *priv)