// SPDX-License-Identifier: GPL-2.0
/* Copyright(c) 2024  Realtek Corporation.*/

#include "../wifi.h"
#include "../rtl8192d/reg.h"
#include "../rtl8192d/def.h"
#include "../rtl8192d/fw_common.h"
#include "fw.h"

int rtl92du_download_fw(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        struct rtl_hal *rtlhal = rtl_hal(rtlpriv);
        enum version_8192d version = rtlhal->version;
        u8 *pfwheader;
        u8 *pfwdata;
        u32 fwsize;
        int err;

        if (rtlpriv->max_fw_size == 0 || !rtlhal->pfirmware)
                return 1;

        fwsize = rtlhal->fwsize;
        pfwheader = rtlhal->pfirmware;
        pfwdata = rtlhal->pfirmware;
        rtlhal->fw_version = (u16)GET_FIRMWARE_HDR_VERSION(pfwheader);
        rtlhal->fw_subversion = (u16)GET_FIRMWARE_HDR_SUB_VER(pfwheader);

        rtl_dbg(rtlpriv, COMP_INIT, DBG_LOUD,
                "FirmwareVersion(%d), FirmwareSubVersion(%d), Signature(%#x)\n",
                rtlhal->fw_version, rtlhal->fw_subversion,
                GET_FIRMWARE_HDR_SIGNATURE(pfwheader));

        if (IS_FW_HEADER_EXIST(pfwheader)) {
                rtl_dbg(rtlpriv, COMP_INIT, DBG_LOUD,
                        "Shift 32 bytes for FW header!!\n");
                pfwdata = pfwdata + 32;
                fwsize = fwsize - 32;
        }

        if (rtl92d_is_fw_downloaded(rtlpriv))
                goto exit;

        /* If 8051 is running in RAM code, driver should
         * inform Fw to reset by itself, or it will cause
         * download Fw fail.
         */
        if (rtl_read_byte(rtlpriv, REG_MCUFWDL) & BIT(7)) {
                rtl92d_firmware_selfreset(hw);
                rtl_write_byte(rtlpriv, REG_MCUFWDL, 0x00);
        }

        rtl92d_enable_fw_download(hw, true);
        rtl92d_write_fw(hw, version, pfwdata, fwsize);
        rtl92d_enable_fw_download(hw, false);

        err = rtl92d_fw_free_to_go(hw);
        if (err)
                pr_err("fw is not ready to run!\n");
exit:
        err = rtl92d_fw_init(hw);
        return err;
}