#include "en_accel/macsec.h"
 #include "en_accel/macsec_fs.h"
 
+#define MLX5_MACSEC_EPN_SCOPE_MID 0x80000000L
+#define MLX5E_MACSEC_ASO_CTX_SZ MLX5_ST_SZ_BYTES(macsec_aso)
+
+enum mlx5_macsec_aso_event_arm {
+       MLX5E_ASO_EPN_ARM = BIT(0),
+};
+
+enum {
+       MLX5_MACSEC_ASO_REMOVE_FLOW_PKT_CNT_OFFSET,
+};
+
+struct mlx5e_macsec_handle {
+       struct mlx5e_macsec *macsec;
+       u32 obj_id;
+       u8 idx;
+};
+
+enum {
+       MLX5_MACSEC_EPN,
+};
+
+struct mlx5e_macsec_aso_out {
+       u8 event_arm;
+       u32 mode_param;
+};
+
+struct mlx5e_macsec_aso_in {
+       u8 mode;
+       u32 obj_id;
+};
+
+struct mlx5e_macsec_epn_state {
+       u32 epn_msb;
+       u8 epn_enabled;
+       u8 overlap;
+};
+
+struct mlx5e_macsec_async_work {
+       struct mlx5e_macsec *macsec;
+       struct mlx5_core_dev *mdev;
+       struct work_struct work;
+       u32 obj_id;
+};
+
 struct mlx5e_macsec_sa {
        bool active;
        u8  assoc_num;
        u32 enc_key_id;
        u32 next_pn;
        sci_t sci;
+       salt_t salt;
 
        struct rhash_head hash;
        u32 fs_id;
        union mlx5e_macsec_rule *macsec_rule;
        struct rcu_head rcu_head;
+       struct mlx5e_macsec_epn_state epn_state;
 };
 
 struct mlx5e_macsec_rx_sc;
 
        /* ASO */
        struct mlx5e_macsec_aso aso;
+
+       struct notifier_block nb;
+       struct workqueue_struct *wq;
 };
 
 struct mlx5_macsec_obj_attrs {
        __be64 sci;
        u32 enc_key_id;
        bool encrypt;
+       struct mlx5e_macsec_epn_state epn_state;
+       salt_t salt;
+       __be32 ssci;
+};
+
+struct mlx5_aso_ctrl_param {
+       u8   data_mask_mode;
+       u8   condition_0_operand;
+       u8   condition_1_operand;
+       u8   condition_0_offset;
+       u8   condition_1_offset;
+       u8   data_offset;
+       u8   condition_operand;
+       u32  condition_0_data;
+       u32  condition_0_mask;
+       u32  condition_1_data;
+       u32  condition_1_mask;
+       u64  bitwise_data;
+       u64  data_mask;
 };
 
 static int mlx5e_macsec_aso_reg_mr(struct mlx5_core_dev *mdev, struct mlx5e_macsec_aso *aso)
 
        MLX5_SET(macsec_offload_obj, obj, confidentiality_en, attrs->encrypt);
        MLX5_SET(macsec_offload_obj, obj, dekn, attrs->enc_key_id);
-       MLX5_SET64(macsec_offload_obj, obj, sci, (__force u64)(attrs->sci));
        MLX5_SET(macsec_offload_obj, obj, aso_return_reg, MLX5_MACSEC_ASO_REG_C_4_5);
        MLX5_SET(macsec_offload_obj, obj, macsec_aso_access_pd, attrs->aso_pdn);
        MLX5_SET(macsec_aso, aso_ctx, mode_parameter, attrs->next_pn);
 
+       /* Epn */
+       if (attrs->epn_state.epn_enabled) {
+               void *salt_p;
+               int i;
+
+               MLX5_SET(macsec_aso, aso_ctx, epn_event_arm, 1);
+               MLX5_SET(macsec_offload_obj, obj, epn_en, 1);
+               MLX5_SET(macsec_offload_obj, obj, epn_msb, attrs->epn_state.epn_msb);
+               MLX5_SET(macsec_offload_obj, obj, epn_overlap, attrs->epn_state.overlap);
+               MLX5_SET64(macsec_offload_obj, obj, sci, (__force u64)attrs->ssci);
+               salt_p = MLX5_ADDR_OF(macsec_offload_obj, obj, salt);
+               for (i = 0; i < 3 ; i++)
+                       memcpy((u32 *)salt_p + i, &attrs->salt.bytes[4 * (2 - i)], 4);
+               if (!is_tx)
+                       MLX5_SET(macsec_aso, aso_ctx, mode, MLX5_MACSEC_ASO_REPLAY_PROTECTION);
+       } else {
+               MLX5_SET64(macsec_offload_obj, obj, sci, (__force u64)(attrs->sci));
+       }
+
        MLX5_SET(macsec_aso, aso_ctx, valid, 0x1);
        if (is_tx)
                MLX5_SET(macsec_aso, aso_ctx, mode, MLX5_MACSEC_ASO_INC_SN);
        struct mlx5_core_dev *mdev = priv->mdev;
        struct mlx5_macsec_obj_attrs obj_attrs;
        union mlx5e_macsec_rule *macsec_rule;
+       struct macsec_key *key;
        int err;
 
        obj_attrs.next_pn = sa->next_pn;
        obj_attrs.enc_key_id = sa->enc_key_id;
        obj_attrs.encrypt = encrypt;
        obj_attrs.aso_pdn = macsec->aso.pdn;
+       obj_attrs.epn_state = sa->epn_state;
+
+       if (is_tx) {
+               obj_attrs.ssci = cpu_to_be32((__force u32)ctx->sa.tx_sa->ssci);
+               key = &ctx->sa.tx_sa->key;
+       } else {
+               obj_attrs.ssci = cpu_to_be32((__force u32)ctx->sa.rx_sa->ssci);
+               key = &ctx->sa.rx_sa->key;
+       }
+
+       memcpy(&obj_attrs.salt, &key->salt, sizeof(key->salt));
 
        err = mlx5e_macsec_create_object(mdev, &obj_attrs, is_tx, &sa->macsec_obj_id);
        if (err)
                return false;
        }
 
-       if (secy->xpn) {
-               netdev_err(netdev, "MACsec offload: xpn is not supported\n");
-               return false;
-       }
-
        if (secy->replay_protect) {
                netdev_err(netdev, "MACsec offload: replay protection is not supported\n");
                return false;
        return NULL;
 }
 
+static void update_macsec_epn(struct mlx5e_macsec_sa *sa, const struct macsec_key *key,
+                             const pn_t *next_pn_halves)
+{
+       struct mlx5e_macsec_epn_state *epn_state = &sa->epn_state;
+
+       sa->salt = key->salt;
+       epn_state->epn_enabled = 1;
+       epn_state->epn_msb = next_pn_halves->upper;
+       epn_state->overlap = next_pn_halves->lower < MLX5_MACSEC_EPN_SCOPE_MID ? 0 : 1;
+}
+
 static int mlx5e_macsec_add_txsa(struct macsec_context *ctx)
 {
        const struct macsec_tx_sc *tx_sc = &ctx->secy->tx_sc;
        tx_sa->next_pn = ctx_tx_sa->next_pn_halves.lower;
        tx_sa->sci = secy->sci;
        tx_sa->assoc_num = assoc_num;
+
+       if (secy->xpn)
+               update_macsec_epn(tx_sa, &ctx_tx_sa->key, &ctx_tx_sa->next_pn_halves);
+
        err = mlx5_create_encryption_key(mdev, ctx->sa.key, secy->key_len,
                                         MLX5_ACCEL_OBJ_MACSEC_KEY,
                                         &tx_sa->enc_key_id);
        rx_sa->assoc_num = assoc_num;
        rx_sa->fs_id = rx_sc->sc_xarray_element->fs_id;
 
+       if (ctx->secy->xpn)
+               update_macsec_epn(rx_sa, &ctx_rx_sa->key, &ctx_rx_sa->next_pn_halves);
+
        err = mlx5_create_encryption_key(mdev, ctx->sa.key, ctx->secy->key_len,
                                         MLX5_ACCEL_OBJ_MACSEC_KEY,
                                         &rx_sa->enc_key_id);
        return err;
 }
 
+static void macsec_build_accel_attrs(struct mlx5e_macsec_sa *sa,
+                                    struct mlx5_macsec_obj_attrs *attrs)
+{
+       attrs->epn_state.epn_msb = sa->epn_state.epn_msb;
+       attrs->epn_state.overlap = sa->epn_state.overlap;
+}
+
+static void macsec_aso_build_wqe_ctrl_seg(struct mlx5e_macsec_aso *macsec_aso,
+                                         struct mlx5_wqe_aso_ctrl_seg *aso_ctrl,
+                                         struct mlx5_aso_ctrl_param *param)
+{
+       memset(aso_ctrl, 0, sizeof(*aso_ctrl));
+       if (macsec_aso->umr->dma_addr) {
+               aso_ctrl->va_l  = cpu_to_be32(macsec_aso->umr->dma_addr | ASO_CTRL_READ_EN);
+               aso_ctrl->va_h  = cpu_to_be32((u64)macsec_aso->umr->dma_addr >> 32);
+               aso_ctrl->l_key = cpu_to_be32(macsec_aso->umr->mkey);
+       }
+
+       if (!param)
+               return;
+
+       aso_ctrl->data_mask_mode = param->data_mask_mode << 6;
+       aso_ctrl->condition_1_0_operand = param->condition_1_operand |
+                                               param->condition_0_operand << 4;
+       aso_ctrl->condition_1_0_offset = param->condition_1_offset |
+                                               param->condition_0_offset << 4;
+       aso_ctrl->data_offset_condition_operand = param->data_offset |
+                                               param->condition_operand << 6;
+       aso_ctrl->condition_0_data = cpu_to_be32(param->condition_0_data);
+       aso_ctrl->condition_0_mask = cpu_to_be32(param->condition_0_mask);
+       aso_ctrl->condition_1_data = cpu_to_be32(param->condition_1_data);
+       aso_ctrl->condition_1_mask = cpu_to_be32(param->condition_1_mask);
+       aso_ctrl->bitwise_data = cpu_to_be64(param->bitwise_data);
+       aso_ctrl->data_mask = cpu_to_be64(param->data_mask);
+}
+
+static int mlx5e_macsec_modify_obj(struct mlx5_core_dev *mdev, struct mlx5_macsec_obj_attrs *attrs,
+                                  u32 macsec_id)
+{
+       u32 in[MLX5_ST_SZ_DW(modify_macsec_obj_in)] = {};
+       u32 out[MLX5_ST_SZ_DW(query_macsec_obj_out)];
+       u64 modify_field_select = 0;
+       void *obj;
+       int err;
+
+       /* General object fields set */
+       MLX5_SET(general_obj_in_cmd_hdr, in, opcode, MLX5_CMD_OP_QUERY_GENERAL_OBJECT);
+       MLX5_SET(general_obj_in_cmd_hdr, in, obj_type, MLX5_GENERAL_OBJECT_TYPES_MACSEC);
+       MLX5_SET(general_obj_in_cmd_hdr, in, obj_id, macsec_id);
+       err = mlx5_cmd_exec(mdev, in, sizeof(in), out, sizeof(out));
+       if (err) {
+               mlx5_core_err(mdev, "Query MACsec object failed (Object id %d), err = %d\n",
+                             macsec_id, err);
+               return err;
+       }
+
+       obj = MLX5_ADDR_OF(query_macsec_obj_out, out, macsec_object);
+       modify_field_select = MLX5_GET64(macsec_offload_obj, obj, modify_field_select);
+
+       /* EPN */
+       if (!(modify_field_select & MLX5_MODIFY_MACSEC_BITMASK_EPN_OVERLAP) ||
+           !(modify_field_select & MLX5_MODIFY_MACSEC_BITMASK_EPN_MSB)) {
+               mlx5_core_dbg(mdev, "MACsec object field is not modifiable (Object id %d)\n",
+                             macsec_id);
+               return -EOPNOTSUPP;
+       }
+
+       obj = MLX5_ADDR_OF(modify_macsec_obj_in, in, macsec_object);
+       MLX5_SET64(macsec_offload_obj, obj, modify_field_select,
+                  MLX5_MODIFY_MACSEC_BITMASK_EPN_OVERLAP | MLX5_MODIFY_MACSEC_BITMASK_EPN_MSB);
+       MLX5_SET(macsec_offload_obj, obj, epn_msb, attrs->epn_state.epn_msb);
+       MLX5_SET(macsec_offload_obj, obj, epn_overlap, attrs->epn_state.overlap);
+
+       /* General object fields set */
+       MLX5_SET(general_obj_in_cmd_hdr, in, opcode, MLX5_CMD_OP_MODIFY_GENERAL_OBJECT);
+
+       return mlx5_cmd_exec(mdev, in, sizeof(in), out, sizeof(out));
+}
+
+static void macsec_aso_build_ctrl(struct mlx5e_macsec_aso *aso,
+                                 struct mlx5_wqe_aso_ctrl_seg *aso_ctrl,
+                                 struct mlx5e_macsec_aso_in *in)
+{
+       struct mlx5_aso_ctrl_param param = {};
+
+       param.data_mask_mode = MLX5_ASO_DATA_MASK_MODE_BITWISE_64BIT;
+       param.condition_0_operand = MLX5_ASO_ALWAYS_TRUE;
+       param.condition_1_operand = MLX5_ASO_ALWAYS_TRUE;
+       if (in->mode == MLX5_MACSEC_EPN) {
+               param.data_offset = MLX5_MACSEC_ASO_REMOVE_FLOW_PKT_CNT_OFFSET;
+               param.bitwise_data = BIT_ULL(54);
+               param.data_mask = param.bitwise_data;
+       }
+       macsec_aso_build_wqe_ctrl_seg(aso, aso_ctrl, ¶m);
+}
+
+static int macsec_aso_set_arm_event(struct mlx5_core_dev *mdev, struct mlx5e_macsec *macsec,
+                                   struct mlx5e_macsec_aso_in *in)
+{
+       struct mlx5e_macsec_aso *aso;
+       struct mlx5_aso_wqe *aso_wqe;
+       struct mlx5_aso *maso;
+       int err;
+
+       aso = &macsec->aso;
+       maso = aso->maso;
+
+       mutex_lock(&aso->aso_lock);
+       aso_wqe = mlx5_aso_get_wqe(maso);
+       mlx5_aso_build_wqe(maso, MLX5_MACSEC_ASO_DS_CNT, aso_wqe, in->obj_id,
+                          MLX5_ACCESS_ASO_OPC_MOD_MACSEC);
+       macsec_aso_build_ctrl(aso, &aso_wqe->aso_ctrl, in);
+       mlx5_aso_post_wqe(maso, false, &aso_wqe->ctrl);
+       err = mlx5_aso_poll_cq(maso, false, 10);
+       mutex_unlock(&aso->aso_lock);
+
+       return err;
+}
+
+static int macsec_aso_query(struct mlx5_core_dev *mdev, struct mlx5e_macsec *macsec,
+                           struct mlx5e_macsec_aso_in *in, struct mlx5e_macsec_aso_out *out)
+{
+       struct mlx5e_macsec_aso *aso;
+       struct mlx5_aso_wqe *aso_wqe;
+       struct mlx5_aso *maso;
+       int err;
+
+       aso = &macsec->aso;
+       maso = aso->maso;
+
+       mutex_lock(&aso->aso_lock);
+
+       aso_wqe = mlx5_aso_get_wqe(maso);
+       mlx5_aso_build_wqe(maso, MLX5_MACSEC_ASO_DS_CNT, aso_wqe, in->obj_id,
+                          MLX5_ACCESS_ASO_OPC_MOD_MACSEC);
+       macsec_aso_build_wqe_ctrl_seg(aso, &aso_wqe->aso_ctrl, NULL);
+
+       mlx5_aso_post_wqe(maso, false, &aso_wqe->ctrl);
+       err = mlx5_aso_poll_cq(maso, false, 10);
+       if (err)
+               goto err_out;
+
+       if (MLX5_GET(macsec_aso, aso->umr->ctx, epn_event_arm))
+               out->event_arm |= MLX5E_ASO_EPN_ARM;
+
+       out->mode_param = MLX5_GET(macsec_aso, aso->umr->ctx, mode_parameter);
+
+err_out:
+       mutex_unlock(&aso->aso_lock);
+       return err;
+}
+
+static struct mlx5e_macsec_sa *get_macsec_tx_sa_from_obj_id(const struct mlx5e_macsec *macsec,
+                                                           const u32 obj_id)
+{
+       const struct list_head *device_list;
+       struct mlx5e_macsec_sa *macsec_sa;
+       struct mlx5e_macsec_device *iter;
+       int i;
+
+       device_list = &macsec->macsec_device_list_head;
+
+       list_for_each_entry(iter, device_list, macsec_device_list_element) {
+               for (i = 0; i < MACSEC_NUM_AN; ++i) {
+                       macsec_sa = iter->tx_sa[i];
+                       if (!macsec_sa || !macsec_sa->active)
+                               continue;
+                       if (macsec_sa->macsec_obj_id == obj_id)
+                               return macsec_sa;
+               }
+       }
+
+       return NULL;
+}
+
+static struct mlx5e_macsec_sa *get_macsec_rx_sa_from_obj_id(const struct mlx5e_macsec *macsec,
+                                                           const u32 obj_id)
+{
+       const struct list_head *device_list, *sc_list;
+       struct mlx5e_macsec_rx_sc *mlx5e_rx_sc;
+       struct mlx5e_macsec_sa *macsec_sa;
+       struct mlx5e_macsec_device *iter;
+       int i;
+
+       device_list = &macsec->macsec_device_list_head;
+
+       list_for_each_entry(iter, device_list, macsec_device_list_element) {
+               sc_list = &iter->macsec_rx_sc_list_head;
+               list_for_each_entry(mlx5e_rx_sc, sc_list, rx_sc_list_element) {
+                       for (i = 0; i < MACSEC_NUM_AN; ++i) {
+                               macsec_sa = mlx5e_rx_sc->rx_sa[i];
+                               if (!macsec_sa || !macsec_sa->active)
+                                       continue;
+                               if (macsec_sa->macsec_obj_id == obj_id)
+                                       return macsec_sa;
+                       }
+               }
+       }
+
+       return NULL;
+}
+
+static void macsec_epn_update(struct mlx5e_macsec *macsec, struct mlx5_core_dev *mdev,
+                             struct mlx5e_macsec_sa *sa, u32 obj_id, u32 mode_param)
+{
+       struct mlx5_macsec_obj_attrs attrs = {};
+       struct mlx5e_macsec_aso_in in = {};
+
+       /* When the bottom of the replay protection window (mode_param) crosses 2^31 (half sequence
+        * number wraparound) hence mode_param > MLX5_MACSEC_EPN_SCOPE_MID the SW should update the
+        * esn_overlap to OLD (1).
+        * When the bottom of the replay protection window (mode_param) crosses 2^32 (full sequence
+        * number wraparound) hence mode_param < MLX5_MACSEC_EPN_SCOPE_MID since it did a
+        * wraparound, the SW should update the esn_overlap to NEW (0), and increment the esn_msb.
+        */
+
+       if (mode_param < MLX5_MACSEC_EPN_SCOPE_MID) {
+               sa->epn_state.epn_msb++;
+               sa->epn_state.overlap = 0;
+       } else {
+               sa->epn_state.overlap = 1;
+       }
+
+       macsec_build_accel_attrs(sa, &attrs);
+       mlx5e_macsec_modify_obj(mdev, &attrs, obj_id);
+
+       /* Re-set EPN arm event */
+       in.obj_id = obj_id;
+       in.mode = MLX5_MACSEC_EPN;
+       macsec_aso_set_arm_event(mdev, macsec, &in);
+}
+
+static void macsec_async_event(struct work_struct *work)
+{
+       struct mlx5e_macsec_async_work *async_work;
+       struct mlx5e_macsec_aso_out out = {};
+       struct mlx5e_macsec_aso_in in = {};
+       struct mlx5e_macsec_sa *macsec_sa;
+       struct mlx5e_macsec *macsec;
+       struct mlx5_core_dev *mdev;
+       u32 obj_id;
+
+       async_work = container_of(work, struct mlx5e_macsec_async_work, work);
+       macsec = async_work->macsec;
+       mdev = async_work->mdev;
+       obj_id = async_work->obj_id;
+       macsec_sa = get_macsec_tx_sa_from_obj_id(macsec, obj_id);
+       if (!macsec_sa) {
+               macsec_sa = get_macsec_rx_sa_from_obj_id(macsec, obj_id);
+               if (!macsec_sa) {
+                       mlx5_core_dbg(mdev, "MACsec SA is not found (SA object id %d)\n", obj_id);
+                       goto out_async_work;
+               }
+       }
+
+       /* Query MACsec ASO context */
+       in.obj_id = obj_id;
+       macsec_aso_query(mdev, macsec, &in, &out);
+
+       /* EPN case */
+       if (macsec_sa->epn_state.epn_enabled && !(out.event_arm & MLX5E_ASO_EPN_ARM))
+               macsec_epn_update(macsec, mdev, macsec_sa, obj_id, out.mode_param);
+
+out_async_work:
+       kfree(async_work);
+}
+
+static int macsec_obj_change_event(struct notifier_block *nb, unsigned long event, void *data)
+{
+       struct mlx5e_macsec *macsec = container_of(nb, struct mlx5e_macsec, nb);
+       struct mlx5e_macsec_async_work *async_work;
+       struct mlx5_eqe_obj_change *obj_change;
+       struct mlx5_eqe *eqe = data;
+       u16 obj_type;
+       u32 obj_id;
+
+       if (event != MLX5_EVENT_TYPE_OBJECT_CHANGE)
+               return NOTIFY_DONE;
+
+       obj_change = &eqe->data.obj_change;
+       obj_type = be16_to_cpu(obj_change->obj_type);
+       obj_id = be32_to_cpu(obj_change->obj_id);
+
+       if (obj_type != MLX5_GENERAL_OBJECT_TYPES_MACSEC)
+               return NOTIFY_DONE;
+
+       async_work = kzalloc(sizeof(*async_work), GFP_ATOMIC);
+       if (!async_work)
+               return NOTIFY_DONE;
+
+       async_work->macsec = macsec;
+       async_work->mdev = macsec->mdev;
+       async_work->obj_id = obj_id;
+
+       INIT_WORK(&async_work->work, macsec_async_event);
+
+       WARN_ON(!queue_work(macsec->wq, &async_work->work));
+
+       return NOTIFY_OK;
+}
+
 static int mlx5e_macsec_aso_init(struct mlx5e_macsec_aso *aso, struct mlx5_core_dev *mdev)
 {
        struct mlx5_aso *maso;
                goto err_aso;
        }
 
+       macsec->wq = alloc_ordered_workqueue("mlx5e_macsec_%s", 0, priv->netdev->name);
+       if (!macsec->wq) {
+               err = -ENOMEM;
+               goto err_wq;
+       }
+
        xa_init_flags(&macsec->sc_xarray, XA_FLAGS_ALLOC1);
 
        priv->macsec = macsec;
 
        macsec->macsec_fs = macsec_fs;
 
+       macsec->nb.notifier_call = macsec_obj_change_event;
+       mlx5_notifier_register(mdev, &macsec->nb);
+
        mlx5_core_dbg(mdev, "MACsec attached to netdevice\n");
 
        return 0;
 
 err_out:
+       destroy_workqueue(macsec->wq);
+err_wq:
        mlx5e_macsec_aso_cleanup(&macsec->aso, priv->mdev);
 err_aso:
        rhashtable_destroy(&macsec->sci_hash);
        if (!macsec)
                return;
 
+       mlx5_notifier_unregister(mdev, &macsec->nb);
+
        mlx5e_macsec_fs_cleanup(macsec->macsec_fs);
 
+       /* Cleanup workqueue */
+       destroy_workqueue(macsec->wq);
+
        mlx5e_macsec_aso_cleanup(&macsec->aso, mdev);
 
        priv->macsec = NULL;