in_devlink_port = mlxsw_core_port_devlink_port_get(mlxsw_sp->core,
                                                           local_port);
        skb_push(skb, ETH_HLEN);
-       devlink_trap_report(devlink, skb, trap_ctx, in_devlink_port);
+       devlink_trap_report(devlink, skb, trap_ctx, in_devlink_port, NULL);
        consume_skb(skb);
 }
 
        in_devlink_port = mlxsw_core_port_devlink_port_get(mlxsw_sp->core,
                                                           local_port);
        skb_push(skb, ETH_HLEN);
-       devlink_trap_report(devlink, skb, trap_ctx, in_devlink_port);
+       devlink_trap_report(devlink, skb, trap_ctx, in_devlink_port, NULL);
        skb_pull(skb, ETH_HLEN);
        skb->offload_fwd_mark = 1;
        netif_receive_skb(skb);
 
                 */
                local_bh_disable();
                devlink_trap_report(devlink, skb, nsim_trap_item->trap_ctx,
-                                   &nsim_dev_port->devlink_port);
+                                   &nsim_dev_port->devlink_port, NULL);
                local_bh_enable();
                consume_skb(skb);
        }
 
 #include <linux/workqueue.h>
 #include <linux/refcount.h>
 #include <net/net_namespace.h>
+#include <net/flow_offload.h>
 #include <uapi/linux/devlink.h>
 
 struct devlink_ops;
 void devlink_traps_unregister(struct devlink *devlink,
                              const struct devlink_trap *traps,
                              size_t traps_count);
-void devlink_trap_report(struct devlink *devlink,
-                        struct sk_buff *skb, void *trap_ctx,
-                        struct devlink_port *in_devlink_port);
+void devlink_trap_report(struct devlink *devlink, struct sk_buff *skb,
+                        void *trap_ctx, struct devlink_port *in_devlink_port,
+                        const struct flow_action_cookie *fa_cookie);
 void *devlink_trap_ctx_priv(void *trap_ctx);
 
 #if IS_ENABLED(CONFIG_NET_DEVLINK)
 
 static void
 devlink_trap_report_metadata_fill(struct net_dm_hw_metadata *hw_metadata,
                                  const struct devlink_trap_item *trap_item,
-                                 struct devlink_port *in_devlink_port)
+                                 struct devlink_port *in_devlink_port,
+                                 const struct flow_action_cookie *fa_cookie)
 {
        struct devlink_trap_group_item *group_item = trap_item->group_item;
 
        hw_metadata->trap_group_name = group_item->group->name;
        hw_metadata->trap_name = trap_item->trap->name;
+       hw_metadata->fa_cookie = fa_cookie;
 
        spin_lock(&in_devlink_port->type_lock);
        if (in_devlink_port->type == DEVLINK_PORT_TYPE_ETH)
  * @skb: Trapped packet.
  * @trap_ctx: Trap context.
  * @in_devlink_port: Input devlink port.
+ * @fa_cookie: Flow action cookie. Could be NULL.
  */
 void devlink_trap_report(struct devlink *devlink, struct sk_buff *skb,
-                        void *trap_ctx, struct devlink_port *in_devlink_port)
+                        void *trap_ctx, struct devlink_port *in_devlink_port,
+                        const struct flow_action_cookie *fa_cookie)
+
 {
        struct devlink_trap_item *trap_item = trap_ctx;
        struct net_dm_hw_metadata hw_metadata = {};
        devlink_trap_stats_update(trap_item->group_item->stats, skb->len);
 
        devlink_trap_report_metadata_fill(&hw_metadata, trap_item,
-                                         in_devlink_port);
+                                         in_devlink_port, fa_cookie);
        net_dm_hw_report(skb, &hw_metadata);
 }
 EXPORT_SYMBOL_GPL(devlink_trap_report);