static bool mlx5e_ipsec_update_esn_state(struct mlx5e_ipsec_sa_entry *sa_entry)
 {
-       struct xfrm_replay_state_esn *replay_esn;
+       struct xfrm_state *x = sa_entry->x;
        u32 seq_bottom = 0;
+       u32 esn, esn_msb;
        u8 overlap;
 
-       replay_esn = sa_entry->x->replay_esn;
-       if (replay_esn->seq >= replay_esn->replay_window)
-               seq_bottom = replay_esn->seq - replay_esn->replay_window + 1;
+       switch (x->xso.type) {
+       case XFRM_DEV_OFFLOAD_PACKET:
+               switch (x->xso.dir) {
+               case XFRM_DEV_OFFLOAD_IN:
+                       esn = x->replay_esn->seq;
+                       esn_msb = x->replay_esn->seq_hi;
+                       break;
+               case XFRM_DEV_OFFLOAD_OUT:
+                       esn = x->replay_esn->oseq;
+                       esn_msb = x->replay_esn->oseq_hi;
+                       break;
+               default:
+                       WARN_ON(true);
+                       return false;
+               }
+               break;
+       case XFRM_DEV_OFFLOAD_CRYPTO:
+               /* Already parsed by XFRM core */
+               esn = x->replay_esn->seq;
+               break;
+       default:
+               WARN_ON(true);
+               return false;
+       }
 
        overlap = sa_entry->esn_state.overlap;
 
-       sa_entry->esn_state.esn = xfrm_replay_seqhi(sa_entry->x,
-                                                   htonl(seq_bottom));
+       if (esn >= x->replay_esn->replay_window)
+               seq_bottom = esn - x->replay_esn->replay_window + 1;
+
+       if (x->xso.type == XFRM_DEV_OFFLOAD_CRYPTO)
+               esn_msb = xfrm_replay_seqhi(x, htonl(seq_bottom));
+
+       sa_entry->esn_state.esn = esn;
+       sa_entry->esn_state.esn_msb = esn_msb;
 
        if (unlikely(overlap && seq_bottom < MLX5E_IPSEC_ESN_SCOPE_MID)) {
                sa_entry->esn_state.overlap = 0;
 
        /* esn */
        if (x->props.flags & XFRM_STATE_ESN) {
-               attrs->esn_trigger = true;
-               attrs->esn = sa_entry->esn_state.esn;
-               attrs->esn_overlap = sa_entry->esn_state.overlap;
-               attrs->replay_window = x->replay_esn->replay_window;
+               attrs->replay_esn.trigger = true;
+               attrs->replay_esn.esn = sa_entry->esn_state.esn;
+               attrs->replay_esn.esn_msb = sa_entry->esn_state.esn_msb;
+               attrs->replay_esn.overlap = sa_entry->esn_state.overlap;
        }
 
        attrs->dir = x->xso.dir;
 
        void *aso_ctx;
 
        aso_ctx = MLX5_ADDR_OF(ipsec_obj, obj, ipsec_aso);
-       if (attrs->esn_trigger) {
+       if (attrs->replay_esn.trigger) {
                MLX5_SET(ipsec_aso, aso_ctx, esn_event_arm, 1);
 
                if (attrs->dir == XFRM_DEV_OFFLOAD_IN) {
                        MLX5_SET(ipsec_aso, aso_ctx, window_sz,
-                                attrs->replay_window / 64);
+                                attrs->replay_esn.replay_window / 64);
                        MLX5_SET(ipsec_aso, aso_ctx, mode,
                                 MLX5_IPSEC_ASO_REPLAY_PROTECTION);
-                       }
+               }
+               MLX5_SET(ipsec_aso, aso_ctx, mode_parameter,
+                        attrs->replay_esn.esn);
        }
 
        /* ASO context */
        salt_iv_p = MLX5_ADDR_OF(ipsec_obj, obj, implicit_iv);
        memcpy(salt_iv_p, &aes_gcm->seq_iv, sizeof(aes_gcm->seq_iv));
        /* esn */
-       if (attrs->esn_trigger) {
+       if (attrs->replay_esn.trigger) {
                MLX5_SET(ipsec_obj, obj, esn_en, 1);
-               MLX5_SET(ipsec_obj, obj, esn_msb, attrs->esn);
-               MLX5_SET(ipsec_obj, obj, esn_overlap, attrs->esn_overlap);
+               MLX5_SET(ipsec_obj, obj, esn_msb, attrs->replay_esn.esn_msb);
+               MLX5_SET(ipsec_obj, obj, esn_overlap, attrs->replay_esn.overlap);
        }
 
        MLX5_SET(ipsec_obj, obj, dekn, sa_entry->enc_key_id);
        MLX5_SET64(ipsec_obj, obj, modify_field_select,
                   MLX5_MODIFY_IPSEC_BITMASK_ESN_OVERLAP |
                           MLX5_MODIFY_IPSEC_BITMASK_ESN_MSB);
-       MLX5_SET(ipsec_obj, obj, esn_msb, attrs->esn);
-       MLX5_SET(ipsec_obj, obj, esn_overlap, attrs->esn_overlap);
+       MLX5_SET(ipsec_obj, obj, esn_msb, attrs->replay_esn.esn_msb);
+       MLX5_SET(ipsec_obj, obj, esn_overlap, attrs->replay_esn.overlap);
 
        /* general object fields set */
        MLX5_SET(general_obj_in_cmd_hdr, in, opcode, MLX5_CMD_OP_MODIFY_GENERAL_OBJECT);
        struct mlx5_wqe_aso_ctrl_seg data = {};
 
        if (mode_param < MLX5E_IPSEC_ESN_SCOPE_MID) {
-               sa_entry->esn_state.esn++;
+               sa_entry->esn_state.esn_msb++;
                sa_entry->esn_state.overlap = 0;
        } else {
                sa_entry->esn_state.overlap = 1;
        if (ret)
                goto unlock;
 
-       if (attrs->esn_trigger &&
+       if (attrs->replay_esn.trigger &&
            !MLX5_GET(ipsec_aso, aso->ctx, esn_event_arm)) {
                u32 mode_param = MLX5_GET(ipsec_aso, aso->ctx, mode_parameter);