// SPDX-License-Identifier: GPL-2.0-only
/*
 * MMC crypto engine (inline encryption) support
 *
 * Copyright 2020 Google LLC
 */

#include <linux/blk-crypto.h>
#include <linux/mmc/host.h>

#include "core.h"
#include "crypto.h"
#include "queue.h"

void mmc_crypto_set_initial_state(struct mmc_host *host)
{
 /* Reset might clear all keys, so reprogram all the keys. */
 if (host->caps2 & MMC_CAP2_CRYPTO)
  blk_crypto_reprogram_all_keys(&host->crypto_profile);
}

void mmc_crypto_setup_queue(struct request_queue *q, struct mmc_host *host)
{
 if (host->caps2 & MMC_CAP2_CRYPTO)
  blk_crypto_register(&host->crypto_profile, q);
}
EXPORT_SYMBOL_GPL(mmc_crypto_setup_queue);

void mmc_crypto_prepare_req(struct mmc_queue_req *mqrq)
{
 struct request *req = mmc_queue_req_to_req(mqrq);
 struct mmc_request *mrq = &mqrq->brq.mrq;
 struct blk_crypto_keyslot *keyslot;

 if (!req->crypt_ctx)
  return;

 mrq->crypto_ctx = req->crypt_ctx;

 keyslot = req->crypt_keyslot;
 if (keyslot)
  mrq->crypto_key_slot = blk_crypto_keyslot_index(keyslot);
}
EXPORT_SYMBOL_GPL(mmc_crypto_prepare_req);