#include <linux/vmalloc.h>
 
 #include "core.h"
+#include "../mlxfw/mlxfw.h"
 
 struct mlxsw_linecard_ini_file {
        __le16 size;
        return linecard->name;
 }
 
+struct mlxsw_linecard_device_fw_info {
+       struct mlxfw_dev mlxfw_dev;
+       struct mlxsw_core *mlxsw_core;
+       struct mlxsw_linecard *linecard;
+};
+
+static int mlxsw_linecard_device_fw_component_query(struct mlxfw_dev *mlxfw_dev,
+                                                   u16 component_index,
+                                                   u32 *p_max_size,
+                                                   u8 *p_align_bits,
+                                                   u16 *p_max_write_size)
+{
+       struct mlxsw_linecard_device_fw_info *info =
+               container_of(mlxfw_dev, struct mlxsw_linecard_device_fw_info,
+                            mlxfw_dev);
+       struct mlxsw_linecard *linecard = info->linecard;
+       struct mlxsw_core *mlxsw_core = info->mlxsw_core;
+       char mddt_pl[MLXSW_REG_MDDT_LEN];
+       char *mcqi_pl;
+       int err;
+
+       mlxsw_reg_mddt_pack(mddt_pl, linecard->slot_index,
+                           linecard->device.index,
+                           MLXSW_REG_MDDT_METHOD_QUERY,
+                           MLXSW_REG(mcqi), &mcqi_pl);
+
+       mlxsw_reg_mcqi_pack(mcqi_pl, component_index);
+       err = mlxsw_reg_query(mlxsw_core, MLXSW_REG(mddt), mddt_pl);
+       if (err)
+               return err;
+       mlxsw_reg_mcqi_unpack(mcqi_pl, p_max_size, p_align_bits,
+                             p_max_write_size);
+
+       *p_align_bits = max_t(u8, *p_align_bits, 2);
+       *p_max_write_size = min_t(u16, *p_max_write_size,
+                                 MLXSW_REG_MCDA_MAX_DATA_LEN);
+       return 0;
+}
+
+static int mlxsw_linecard_device_fw_fsm_lock(struct mlxfw_dev *mlxfw_dev,
+                                            u32 *fwhandle)
+{
+       struct mlxsw_linecard_device_fw_info *info =
+               container_of(mlxfw_dev, struct mlxsw_linecard_device_fw_info,
+                            mlxfw_dev);
+       struct mlxsw_linecard *linecard = info->linecard;
+       struct mlxsw_core *mlxsw_core = info->mlxsw_core;
+       char mddt_pl[MLXSW_REG_MDDT_LEN];
+       u8 control_state;
+       char *mcc_pl;
+       int err;
+
+       mlxsw_reg_mddt_pack(mddt_pl, linecard->slot_index,
+                           linecard->device.index,
+                           MLXSW_REG_MDDT_METHOD_QUERY,
+                           MLXSW_REG(mcc), &mcc_pl);
+       mlxsw_reg_mcc_pack(mcc_pl, 0, 0, 0, 0);
+       err = mlxsw_reg_query(mlxsw_core, MLXSW_REG(mddt), mddt_pl);
+       if (err)
+               return err;
+
+       mlxsw_reg_mcc_unpack(mcc_pl, fwhandle, NULL, &control_state);
+       if (control_state != MLXFW_FSM_STATE_IDLE)
+               return -EBUSY;
+
+       mlxsw_reg_mddt_pack(mddt_pl, linecard->slot_index,
+                           linecard->device.index,
+                           MLXSW_REG_MDDT_METHOD_WRITE,
+                           MLXSW_REG(mcc), &mcc_pl);
+       mlxsw_reg_mcc_pack(mcc_pl, MLXSW_REG_MCC_INSTRUCTION_LOCK_UPDATE_HANDLE,
+                          0, *fwhandle, 0);
+       return mlxsw_reg_write(mlxsw_core, MLXSW_REG(mddt), mddt_pl);
+}
+
+static int
+mlxsw_linecard_device_fw_fsm_component_update(struct mlxfw_dev *mlxfw_dev,
+                                             u32 fwhandle,
+                                             u16 component_index,
+                                             u32 component_size)
+{
+       struct mlxsw_linecard_device_fw_info *info =
+               container_of(mlxfw_dev, struct mlxsw_linecard_device_fw_info,
+                            mlxfw_dev);
+       struct mlxsw_linecard *linecard = info->linecard;
+       struct mlxsw_core *mlxsw_core = info->mlxsw_core;
+       char mddt_pl[MLXSW_REG_MDDT_LEN];
+       char *mcc_pl;
+
+       mlxsw_reg_mddt_pack(mddt_pl, linecard->slot_index,
+                           linecard->device.index,
+                           MLXSW_REG_MDDT_METHOD_WRITE,
+                           MLXSW_REG(mcc), &mcc_pl);
+       mlxsw_reg_mcc_pack(mcc_pl, MLXSW_REG_MCC_INSTRUCTION_UPDATE_COMPONENT,
+                          component_index, fwhandle, component_size);
+       return mlxsw_reg_write(mlxsw_core, MLXSW_REG(mddt), mddt_pl);
+}
+
+static int
+mlxsw_linecard_device_fw_fsm_block_download(struct mlxfw_dev *mlxfw_dev,
+                                           u32 fwhandle, u8 *data,
+                                           u16 size, u32 offset)
+{
+       struct mlxsw_linecard_device_fw_info *info =
+               container_of(mlxfw_dev, struct mlxsw_linecard_device_fw_info,
+                            mlxfw_dev);
+       struct mlxsw_linecard *linecard = info->linecard;
+       struct mlxsw_core *mlxsw_core = info->mlxsw_core;
+       char mddt_pl[MLXSW_REG_MDDT_LEN];
+       char *mcda_pl;
+
+       mlxsw_reg_mddt_pack(mddt_pl, linecard->slot_index,
+                           linecard->device.index,
+                           MLXSW_REG_MDDT_METHOD_WRITE,
+                           MLXSW_REG(mcda), &mcda_pl);
+       mlxsw_reg_mcda_pack(mcda_pl, fwhandle, offset, size, data);
+       return mlxsw_reg_write(mlxsw_core, MLXSW_REG(mddt), mddt_pl);
+}
+
+static int
+mlxsw_linecard_device_fw_fsm_component_verify(struct mlxfw_dev *mlxfw_dev,
+                                             u32 fwhandle, u16 component_index)
+{
+       struct mlxsw_linecard_device_fw_info *info =
+               container_of(mlxfw_dev, struct mlxsw_linecard_device_fw_info,
+                            mlxfw_dev);
+       struct mlxsw_linecard *linecard = info->linecard;
+       struct mlxsw_core *mlxsw_core = info->mlxsw_core;
+       char mddt_pl[MLXSW_REG_MDDT_LEN];
+       char *mcc_pl;
+
+       mlxsw_reg_mddt_pack(mddt_pl, linecard->slot_index,
+                           linecard->device.index,
+                           MLXSW_REG_MDDT_METHOD_WRITE,
+                           MLXSW_REG(mcc), &mcc_pl);
+       mlxsw_reg_mcc_pack(mcc_pl, MLXSW_REG_MCC_INSTRUCTION_VERIFY_COMPONENT,
+                          component_index, fwhandle, 0);
+       return mlxsw_reg_write(mlxsw_core, MLXSW_REG(mddt), mddt_pl);
+}
+
+static int mlxsw_linecard_device_fw_fsm_activate(struct mlxfw_dev *mlxfw_dev,
+                                                u32 fwhandle)
+{
+       struct mlxsw_linecard_device_fw_info *info =
+               container_of(mlxfw_dev, struct mlxsw_linecard_device_fw_info,
+                            mlxfw_dev);
+       struct mlxsw_linecard *linecard = info->linecard;
+       struct mlxsw_core *mlxsw_core = info->mlxsw_core;
+       char mddt_pl[MLXSW_REG_MDDT_LEN];
+       char *mcc_pl;
+
+       mlxsw_reg_mddt_pack(mddt_pl, linecard->slot_index,
+                           linecard->device.index,
+                           MLXSW_REG_MDDT_METHOD_WRITE,
+                           MLXSW_REG(mcc), &mcc_pl);
+       mlxsw_reg_mcc_pack(mcc_pl, MLXSW_REG_MCC_INSTRUCTION_ACTIVATE,
+                          0, fwhandle, 0);
+       return mlxsw_reg_write(mlxsw_core, MLXSW_REG(mddt), mddt_pl);
+}
+
+static int
+mlxsw_linecard_device_fw_fsm_query_state(struct mlxfw_dev *mlxfw_dev,
+                                        u32 fwhandle,
+                                        enum mlxfw_fsm_state *fsm_state,
+                                        enum mlxfw_fsm_state_err *fsm_state_err)
+{
+       struct mlxsw_linecard_device_fw_info *info =
+               container_of(mlxfw_dev, struct mlxsw_linecard_device_fw_info,
+                            mlxfw_dev);
+       struct mlxsw_linecard *linecard = info->linecard;
+       struct mlxsw_core *mlxsw_core = info->mlxsw_core;
+       char mddt_pl[MLXSW_REG_MDDT_LEN];
+       u8 control_state;
+       u8 error_code;
+       char *mcc_pl;
+       int err;
+
+       mlxsw_reg_mddt_pack(mddt_pl, linecard->slot_index,
+                           linecard->device.index,
+                           MLXSW_REG_MDDT_METHOD_QUERY,
+                           MLXSW_REG(mcc), &mcc_pl);
+       mlxsw_reg_mcc_pack(mcc_pl, 0, 0, fwhandle, 0);
+       err = mlxsw_reg_query(mlxsw_core, MLXSW_REG(mddt), mddt_pl);
+       if (err)
+               return err;
+
+       mlxsw_reg_mcc_unpack(mcc_pl, NULL, &error_code, &control_state);
+       *fsm_state = control_state;
+       *fsm_state_err = min_t(enum mlxfw_fsm_state_err, error_code,
+                              MLXFW_FSM_STATE_ERR_MAX);
+       return 0;
+}
+
+static void mlxsw_linecard_device_fw_fsm_cancel(struct mlxfw_dev *mlxfw_dev,
+                                               u32 fwhandle)
+{
+       struct mlxsw_linecard_device_fw_info *info =
+               container_of(mlxfw_dev, struct mlxsw_linecard_device_fw_info,
+                            mlxfw_dev);
+       struct mlxsw_linecard *linecard = info->linecard;
+       struct mlxsw_core *mlxsw_core = info->mlxsw_core;
+       char mddt_pl[MLXSW_REG_MDDT_LEN];
+       char *mcc_pl;
+
+       mlxsw_reg_mddt_pack(mddt_pl, linecard->slot_index,
+                           linecard->device.index,
+                           MLXSW_REG_MDDT_METHOD_WRITE,
+                           MLXSW_REG(mcc), &mcc_pl);
+       mlxsw_reg_mcc_pack(mcc_pl, MLXSW_REG_MCC_INSTRUCTION_CANCEL,
+                          0, fwhandle, 0);
+       mlxsw_reg_write(mlxsw_core, MLXSW_REG(mddt), mddt_pl);
+}
+
+static void mlxsw_linecard_device_fw_fsm_release(struct mlxfw_dev *mlxfw_dev,
+                                                u32 fwhandle)
+{
+       struct mlxsw_linecard_device_fw_info *info =
+               container_of(mlxfw_dev, struct mlxsw_linecard_device_fw_info,
+                            mlxfw_dev);
+       struct mlxsw_linecard *linecard = info->linecard;
+       struct mlxsw_core *mlxsw_core = info->mlxsw_core;
+       char mddt_pl[MLXSW_REG_MDDT_LEN];
+       char *mcc_pl;
+
+       mlxsw_reg_mddt_pack(mddt_pl, linecard->slot_index,
+                           linecard->device.index,
+                           MLXSW_REG_MDDT_METHOD_WRITE,
+                           MLXSW_REG(mcc), &mcc_pl);
+       mlxsw_reg_mcc_pack(mcc_pl,
+                          MLXSW_REG_MCC_INSTRUCTION_RELEASE_UPDATE_HANDLE,
+                          0, fwhandle, 0);
+       mlxsw_reg_write(mlxsw_core, MLXSW_REG(mddt), mddt_pl);
+}
+
+static const struct mlxfw_dev_ops mlxsw_linecard_device_dev_ops = {
+       .component_query        = mlxsw_linecard_device_fw_component_query,
+       .fsm_lock               = mlxsw_linecard_device_fw_fsm_lock,
+       .fsm_component_update   = mlxsw_linecard_device_fw_fsm_component_update,
+       .fsm_block_download     = mlxsw_linecard_device_fw_fsm_block_download,
+       .fsm_component_verify   = mlxsw_linecard_device_fw_fsm_component_verify,
+       .fsm_activate           = mlxsw_linecard_device_fw_fsm_activate,
+       .fsm_query_state        = mlxsw_linecard_device_fw_fsm_query_state,
+       .fsm_cancel             = mlxsw_linecard_device_fw_fsm_cancel,
+       .fsm_release            = mlxsw_linecard_device_fw_fsm_release,
+};
+
+int mlxsw_linecard_flash_update(struct devlink *linecard_devlink,
+                               struct mlxsw_linecard *linecard,
+                               const struct firmware *firmware,
+                               struct netlink_ext_ack *extack)
+{
+       struct mlxsw_core *mlxsw_core = linecard->linecards->mlxsw_core;
+       struct mlxsw_linecard_device_fw_info info = {
+               .mlxfw_dev = {
+                       .ops = &mlxsw_linecard_device_dev_ops,
+                       .psid = linecard->device.info.psid,
+                       .psid_size = strlen(linecard->device.info.psid),
+                       .devlink = linecard_devlink,
+               },
+               .mlxsw_core = mlxsw_core,
+               .linecard = linecard,
+       };
+       int err;
+
+       mutex_lock(&linecard->lock);
+       if (!linecard->active) {
+               NL_SET_ERR_MSG_MOD(extack, "Only active line cards can be flashed");
+               err = -EINVAL;
+               goto unlock;
+       }
+       err = mlxsw_core_fw_flash(mlxsw_core, &info.mlxfw_dev,
+                                 firmware, extack);
+unlock:
+       mutex_unlock(&linecard->lock);
+       return err;
+}
+
 static int mlxsw_linecard_device_psid_get(struct mlxsw_linecard *linecard,
                                          u8 device_index, char *psid)
 {
                        return err;
 
                linecard->device.info = info;
+               linecard->device.index = device_index;
                flashable_found = true;
        } while (msg_seq);