// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <ddk/debug.h>
#include <qcom/smc.h>

namespace qcom {

zx_status_t SmcCall(zx_handle_t h, zx_smc_parameters_t* params, zx_smc_result_t* result) {
    zxlogf(TRACE, "SMC params 0x%X 0x%lX 0x%lX 0x%lX 0x%lX 0x%lX\n", params->func_id,
           params->arg1, params->arg2, params->arg3, params->arg4, params->arg5);
    auto status = zx_smc_call(h, params, result);
    zxlogf(TRACE, "SMC results %ld 0x%lX 0x%lX 0x%lX\n", result->arg0, result->arg1,
           result->arg2, result->arg3);

    constexpr int total_retry_msecs = 2000;
    constexpr int busy_retry_msecs = 30;
    constexpr int busy_retries = total_retry_msecs / busy_retry_msecs;
    int busy_retry = busy_retries;
    while (status == ZX_OK && // Wait forever for smc_interrupted, limited for smc_busy replies.
           (result->arg0 == kSmcInterrupted || (result->arg0 == kSmcBusy && busy_retry--))) {
        if (result->arg0 == kSmcBusy) {
            zx_nanosleep(zx_deadline_after(ZX_MSEC(busy_retry_msecs)));
        }
        params->arg6 = result->arg6; // Pass optional session_id received via x6 back in retry.

        zxlogf(TRACE, "SMC params 0x%X 0x%lX 0x%lX 0x%lX 0x%lX 0x%lX\n", params->func_id,
               params->arg1, params->arg2, params->arg3, params->arg4, params->arg5);
        status = zx_smc_call(h, params, result);
        zxlogf(TRACE, "SMC busy_retry %d results %ld 0x%lX 0x%lX 0x%lX\n",
               busy_retries - busy_retry, result->arg0, result->arg1, result->arg2,
               result->arg3);
    }
    if (result->arg0 != 0) {
        zxlogf(ERROR, "%s error %d\n", __func__, static_cast<int>(result->arg0));
    }
    return status;
}
} // namespace qcom