/*
 * Copyright (C) 2016 The Android Open Source Project
 *
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation
 * files (the "Software"), to deal in the Software without
 * restriction, including without limitation the rights to use, copy,
 * modify, merge, publish, distribute, sublicense, and/or sell copies
 * of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <trusty/rpmb.h>
#include <trusty/trusty_dev.h>
#include <trusty/util.h>

#include <common.h>
#include <memalign.h>
#include <mmc.h>

void* rpmb_storage_get_ctx(void) {
    /* Unused for U-boot */
    return NULL;
}

void rpmb_storage_put_ctx(void* dev) {}

int rpmb_storage_send(void* rpmb_dev,
                      const void* rel_write_data,
                      size_t rel_write_size,
                      const void* write_data,
                      size_t write_size,
                      void* read_buf,
                      size_t read_size) {
    ALLOC_CACHE_ALIGN_BUFFER(uint8_t, rpmb_rel_write_data, rel_write_size);
    ALLOC_CACHE_ALIGN_BUFFER(uint8_t, rpmb_write_data, write_size);
    ALLOC_CACHE_ALIGN_BUFFER(uint8_t, rpmb_read_data, read_size);
    int ret = TRUSTY_ERR_NONE;
    struct mmc* mmc = find_mmc_device(mmc_get_env_dev());
    char original_part = mmc->block_dev.hwpart;

    /* Switch to RPMB partition */
    if (mmc->block_dev.hwpart != MMC_PART_RPMB) {
        ret = mmc_switch_part(mmc, MMC_PART_RPMB);
        if (ret) {
            trusty_error("failed to switch to RPMB partition\n");
            ret = TRUSTY_ERR_GENERIC;
            goto end;
        }
        mmc->block_dev.hwpart = MMC_PART_RPMB;
    }

    if (rel_write_size) {
        if (rel_write_size % MMC_BLOCK_SIZE) {
            trusty_error(
                    "rel_write_size is not a multiple of MMC_BLOCK_SIZE: %d\n",
                    rel_write_size);
            ret = TRUSTY_ERR_INVALID_ARGS;
            goto end;
        }
        trusty_memcpy(rpmb_rel_write_data, rel_write_data, rel_write_size);
        ret = mmc_rpmb_request(mmc, (const struct s_rpmb*)rpmb_rel_write_data,
                               rel_write_size / MMC_BLOCK_SIZE, true);
        if (ret) {
            trusty_error("failed to execute rpmb reliable write\n");
            goto end;
        }
    }
    if (write_size) {
        if (write_size % MMC_BLOCK_SIZE) {
            trusty_error("write_size is not a multiple of MMC_BLOCK_SIZE: %d\n",
                         write_size);
            ret = TRUSTY_ERR_INVALID_ARGS;
            goto end;
        }
        trusty_memcpy(rpmb_write_data, write_data, write_size);
        ret = mmc_rpmb_request(mmc, (const struct s_rpmb*)rpmb_write_data,
                               write_size / MMC_BLOCK_SIZE, false);
        if (ret) {
            trusty_error("failed to execute rpmb write\n");
            goto end;
        }
    }
    if (read_size) {
        if (read_size % MMC_BLOCK_SIZE) {
            trusty_error("read_size is not a multiple of MMC_BLOCK_SIZE: %d\n",
                         read_size);
            ret = TRUSTY_ERR_INVALID_ARGS;
            goto end;
        }
        ret = mmc_rpmb_response(mmc, (struct s_rpmb*)rpmb_read_data,
                                read_size / MMC_BLOCK_SIZE, 0);
        trusty_memcpy((void*)read_buf, rpmb_read_data, read_size);
        if (ret < 0) {
            trusty_error("failed to execute rpmb read\n");
        }
    }

end:
    /* Return to original partition */
    if (mmc->block_dev.hwpart != original_part) {
        if (mmc_switch_part(mmc, original_part) != 0) {
            trusty_error("failed to switch back to original partition\n");
            return TRUSTY_ERR_GENERIC;
        }
        mmc->block_dev.hwpart = original_part;
    }
    return ret;
}