return err;
 }
 
-/**
- * ice_devlink_flash_update - Update firmware stored in flash on the device
- * @devlink: pointer to devlink associated with device to update
- * @params: flash update parameters
- * @extack: netlink extended ACK structure
- *
- * Perform a device flash update. The bulk of the update logic is contained
- * within the ice_flash_pldm_image function.
- *
- * Returns: zero on success, or an error code on failure.
- */
-static int
-ice_devlink_flash_update(struct devlink *devlink,
-                        struct devlink_flash_update_params *params,
-                        struct netlink_ext_ack *extack)
-{
-       struct ice_pf *pf = devlink_priv(devlink);
-       struct ice_hw *hw = &pf->hw;
-       u8 preservation;
-       int err;
-
-       if (!params->overwrite_mask) {
-               /* preserve all settings and identifiers */
-               preservation = ICE_AQC_NVM_PRESERVE_ALL;
-       } else if (params->overwrite_mask == DEVLINK_FLASH_OVERWRITE_SETTINGS) {
-               /* overwrite settings, but preserve the vital device identifiers */
-               preservation = ICE_AQC_NVM_PRESERVE_SELECTED;
-       } else if (params->overwrite_mask == (DEVLINK_FLASH_OVERWRITE_SETTINGS |
-                                             DEVLINK_FLASH_OVERWRITE_IDENTIFIERS)) {
-               /* overwrite both settings and identifiers, preserve nothing */
-               preservation = ICE_AQC_NVM_NO_PRESERVATION;
-       } else {
-               NL_SET_ERR_MSG_MOD(extack, "Requested overwrite mask is not supported");
-               return -EOPNOTSUPP;
-       }
-
-       if (!hw->dev_caps.common_cap.nvm_unified_update) {
-               NL_SET_ERR_MSG_MOD(extack, "Current firmware does not support unified update");
-               return -EOPNOTSUPP;
-       }
-
-       err = ice_cancel_pending_update(pf, NULL, extack);
-       if (err)
-               return err;
-
-       devlink_flash_update_status_notify(devlink, "Preparing to flash", NULL, 0, 0);
-
-       return ice_flash_pldm_image(pf, params->fw, preservation, extack);
-}
-
 static const struct devlink_ops ice_devlink_ops = {
        .supported_flash_update_params = DEVLINK_SUPPORT_FLASH_UPDATE_OVERWRITE_MASK,
        .eswitch_mode_get = ice_eswitch_mode_get,
 
  *
  * Returns: zero on success, or a negative error code on failure.
  */
-int ice_cancel_pending_update(struct ice_pf *pf, const char *component,
-                             struct netlink_ext_ack *extack)
+static int
+ice_cancel_pending_update(struct ice_pf *pf, const char *component,
+                         struct netlink_ext_ack *extack)
 {
        struct devlink *devlink = priv_to_devlink(pf);
        struct device *dev = ice_pf_to_dev(pf);
 }
 
 /**
- * ice_flash_pldm_image - Write a PLDM-formatted firmware image to the device
- * @pf: private device driver structure
- * @fw: firmware object pointing to the relevant firmware file
- * @preservation: preservation level to request from firmware
+ * ice_devlink_flash_update - Write a firmware image to the device
+ * @devlink: pointer to devlink associated with the device to update
+ * @params: devlink flash update parameters
  * @extack: netlink extended ACK structure
  *
  * Parse the data for a given firmware file, verifying that it is a valid PLDM
  *
  * Returns: zero on success or a negative error code on failure.
  */
-int ice_flash_pldm_image(struct ice_pf *pf, const struct firmware *fw,
-                        u8 preservation, struct netlink_ext_ack *extack)
+int ice_devlink_flash_update(struct devlink *devlink,
+                            struct devlink_flash_update_params *params,
+                            struct netlink_ext_ack *extack)
 {
+       struct ice_pf *pf = devlink_priv(devlink);
        struct device *dev = ice_pf_to_dev(pf);
        struct ice_hw *hw = &pf->hw;
        struct ice_fwu_priv priv;
+       u8 preservation;
        int err;
 
-       switch (preservation) {
-       case ICE_AQC_NVM_PRESERVE_ALL:
-       case ICE_AQC_NVM_PRESERVE_SELECTED:
-       case ICE_AQC_NVM_NO_PRESERVATION:
-       case ICE_AQC_NVM_FACTORY_DEFAULT:
-               break;
-       default:
-               WARN(1, "Unexpected preservation level request %u", preservation);
-               return -EINVAL;
+       if (!params->overwrite_mask) {
+               /* preserve all settings and identifiers */
+               preservation = ICE_AQC_NVM_PRESERVE_ALL;
+       } else if (params->overwrite_mask == DEVLINK_FLASH_OVERWRITE_SETTINGS) {
+               /* overwrite settings, but preserve the vital device identifiers */
+               preservation = ICE_AQC_NVM_PRESERVE_SELECTED;
+       } else if (params->overwrite_mask == (DEVLINK_FLASH_OVERWRITE_SETTINGS |
+                                             DEVLINK_FLASH_OVERWRITE_IDENTIFIERS)) {
+               /* overwrite both settings and identifiers, preserve nothing */
+               preservation = ICE_AQC_NVM_NO_PRESERVATION;
+       } else {
+               NL_SET_ERR_MSG_MOD(extack, "Requested overwrite mask is not supported");
+               return -EOPNOTSUPP;
+       }
+
+       if (!hw->dev_caps.common_cap.nvm_unified_update) {
+               NL_SET_ERR_MSG_MOD(extack, "Current firmware does not support unified update");
+               return -EOPNOTSUPP;
        }
 
        memset(&priv, 0, sizeof(priv));
        priv.pf = pf;
        priv.activate_flags = preservation;
 
+       devlink_flash_update_status_notify(devlink, "Preparing to flash", NULL, 0, 0);
+
+       err = ice_cancel_pending_update(pf, NULL, extack);
+       if (err)
+               return err;
+
        err = ice_acquire_nvm(hw, ICE_RES_WRITE);
        if (err) {
                dev_err(dev, "Failed to acquire device flash lock, err %d aq_err %s\n",
                return err;
        }
 
-       err = pldmfw_flash_image(&priv.context, fw);
+       err = pldmfw_flash_image(&priv.context, params->fw);
        if (err == -ENOENT) {
                dev_err(dev, "Firmware image has no record matching this device\n");
                NL_SET_ERR_MSG_MOD(extack, "Firmware image has no record matching this device");