/* Copyright Statement:
 *
 * This software/firmware and related documentation ("MediaTek Software") are
 * protected under relevant copyright laws. The information contained herein is
 * confidential and proprietary to MediaTek Inc. and/or its licensors. Without
 * the prior written permission of MediaTek inc. and/or its licensors, any
 * reproduction, modification, use or disclosure of MediaTek Software, and
 * information contained herein, in whole or in part, shall be strictly
 * prohibited.
 *
 * MediaTek Inc. (C) 2016. All rights reserved.
 *
 * BY OPENING THIS FILE, RECEIVER HEREBY UNEQUIVOCALLY ACKNOWLEDGES AND AGREES
 * THAT THE SOFTWARE/FIRMWARE AND ITS DOCUMENTATIONS ("MEDIATEK SOFTWARE")
 * RECEIVED FROM MEDIATEK AND/OR ITS REPRESENTATIVES ARE PROVIDED TO RECEIVER
 * ON AN "AS-IS" BASIS ONLY. MEDIATEK EXPRESSLY DISCLAIMS ANY AND ALL
 * WARRANTIES, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE OR
 * NONINFRINGEMENT. NEITHER DOES MEDIATEK PROVIDE ANY WARRANTY WHATSOEVER WITH
 * RESPECT TO THE SOFTWARE OF ANY THIRD PARTY WHICH MAY BE USED BY,
 * INCORPORATED IN, OR SUPPLIED WITH THE MEDIATEK SOFTWARE, AND RECEIVER AGREES
 * TO LOOK ONLY TO SUCH THIRD PARTY FOR ANY WARRANTY CLAIM RELATING THERETO.
 * RECEIVER EXPRESSLY ACKNOWLEDGES THAT IT IS RECEIVER'S SOLE RESPONSIBILITY TO
 * OBTAIN FROM ANY THIRD PARTY ALL PROPER LICENSES CONTAINED IN MEDIATEK
 * SOFTWARE. MEDIATEK SHALL ALSO NOT BE RESPONSIBLE FOR ANY MEDIATEK SOFTWARE
 * RELEASES MADE TO RECEIVER'S SPECIFICATION OR TO CONFORM TO A PARTICULAR
 * STANDARD OR OPEN FORUM. RECEIVER'S SOLE AND EXCLUSIVE REMEDY AND MEDIATEK'S
 * ENTIRE AND CUMULATIVE LIABILITY WITH RESPECT TO THE MEDIATEK SOFTWARE
 * RELEASED HEREUNDER WILL BE, AT MEDIATEK'S OPTION, TO REVISE OR REPLACE THE
 * MEDIATEK SOFTWARE AT ISSUE, OR REFUND ANY SOFTWARE LICENSE FEES OR SERVICE
 * CHARGE PAID BY RECEIVER TO MEDIATEK FOR SUCH MEDIATEK SOFTWARE AT ISSUE.
 *
 * The following software/firmware and/or related documentation ("MediaTek
 * Software") have been modified by MediaTek Inc. All revisions are subject to
 * any receiver's applicable license agreements with MediaTek Inc.
 */

#include "typedefs.h"
#include "platform.h"
#include "bootctrl.h"
#include "ab_bootcheck.h"
#include "mmc_core.h"
#include <pal_log.h>

#define MOD "[AB]"

const char *ab_suffix = NULL;

void ab_switch_boot_part(u32 bootpart, const char *suffix)
{
    if (!rollback_slot(suffix) && ab_set_boot_part(bootpart)) {
        pal_log_err("Fail rollback to %s\n", suffix);
        ASSERT(0);
    } else { //success switch preloader to previous version, reboot
        pal_log_info("rollback to %s\n", suffix);
        mtk_arch_reset(1);
    }
}

void ab_ota_boot_check(void)
{
    ab_suffix = get_suffix();
    u8 ab_retry = get_retry_count(ab_suffix);
    boot_part_t bootpart = BOOT_PART_NONE;

    if (ab_get_boot_part(&bootpart)) {
        pal_log_err("%s invalid boot part: %d\n", MOD, bootpart);
    }
    if(bootpart == BOOT_PART_A) {
        pal_log_info("%s Current boot: Preloader_a\n", MOD);
        if(!memcmp(ab_suffix, BOOTCTRL_SUFFIX_B, 2)) {
            ab_switch_boot_part(BOOT_PART_B, BOOTCTRL_SUFFIX_B);
        }
    }
    else if(bootpart == BOOT_PART_B) {
        pal_log_info("%s Current boot: Preloader_b\n", MOD);
        if(!memcmp(ab_suffix, BOOTCTRL_SUFFIX_A, 2)) {
            ab_switch_boot_part(BOOT_PART_A, BOOTCTRL_SUFFIX_A);
        }
    }

    while (!check_valid_slot()) { //one or more valid slot
        pal_log_info("%s ab_suffix: %s, ab_retry: %d\n", MOD, ab_suffix, ab_retry);
        if (get_bootup_status(ab_suffix)) { //boot_successfully == 1
            pal_log_info("%s boot_success is 1!\n", MOD);
            return;
        } else {
            if (ab_retry > 0) {
                reduce_retry_count(ab_suffix);
                return;
            } else {
                mark_slot_invalid(ab_suffix);
                if (check_valid_slot() == -1) {
                    break;
                }
              //rollback suffix
              if (!memcmp(ab_suffix, BOOTCTRL_SUFFIX_A, 2)) {
                  ab_suffix = BOOTCTRL_SUFFIX_B;
                  ab_switch_boot_part(BOOT_PART_B, BOOTCTRL_SUFFIX_B);
              } else {
                  ab_suffix = BOOTCTRL_SUFFIX_A;
                  ab_switch_boot_part(BOOT_PART_A, BOOTCTRL_SUFFIX_A);
              }
              ab_retry = get_retry_count(ab_suffix);
            }
        }
    }
    g_boot_mode = RECOVERY_BOOT;
    pal_log_info("%s no valid slot! boot_mode: %d\n", MOD, g_boot_mode);
    return;
}