/*-
 * Copyright (c) 2025, Samsung Electronics Co., Ltd.
 * Written by Jaeyoon Choi
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include "ufshci_private.h"

void
ufshci_ctrlr_cmd_send_task_mgmt_request(struct ufshci_controller *ctrlr,
    ufshci_cb_fn_t cb_fn, void *cb_arg, uint8_t function, uint8_t lun,
    uint8_t task_tag, uint8_t iid)
{
	struct ufshci_request *req;
	struct ufshci_task_mgmt_request_upiu *upiu;

	req = ufshci_allocate_request_vaddr(NULL, 0, M_NOWAIT, cb_fn, cb_arg);

	req->request_size = sizeof(struct ufshci_task_mgmt_request_upiu);
	req->response_size = sizeof(struct ufshci_task_mgmt_response_upiu);

	upiu = (struct ufshci_task_mgmt_request_upiu *)&req->request_upiu;
	memset(upiu, 0, req->request_size);
	upiu->header.trans_type =
	    UFSHCI_UPIU_TRANSACTION_CODE_TASK_MANAGEMENT_REQUEST;
	upiu->header.lun = lun;
	upiu->header.ext_iid_or_function = function;
	upiu->input_param1 = lun;
	upiu->input_param2 = task_tag;
	upiu->input_param3 = iid;

	ufshci_ctrlr_submit_task_mgmt_request(ctrlr, req);
}

void
ufshci_ctrlr_cmd_send_nop(struct ufshci_controller *ctrlr, ufshci_cb_fn_t cb_fn,
    void *cb_arg)
{
	struct ufshci_request *req;
	struct ufshci_nop_out_upiu *upiu;

	req = ufshci_allocate_request_vaddr(NULL, 0, M_WAITOK, cb_fn, cb_arg);

	req->request_size = sizeof(struct ufshci_nop_out_upiu);
	req->response_size = sizeof(struct ufshci_nop_in_upiu);
	req->is_admin = true;

	upiu = (struct ufshci_nop_out_upiu *)&req->request_upiu;
	memset(upiu, 0, req->request_size);
	upiu->header.trans_type = UFSHCI_UPIU_TRANSACTION_CODE_NOP_OUT;

	ufshci_ctrlr_submit_transfer_request(ctrlr, req);
}

void
ufshci_ctrlr_cmd_send_query_request(struct ufshci_controller *ctrlr,
    ufshci_cb_fn_t cb_fn, void *cb_arg, struct ufshci_query_param param)
{
	struct ufshci_request *req;
	struct ufshci_query_request_upiu *upiu;

	req = ufshci_allocate_request_vaddr(NULL, 0, M_WAITOK, cb_fn, cb_arg);

	req->request_size = sizeof(struct ufshci_query_request_upiu);
	req->response_size = sizeof(struct ufshci_query_response_upiu);
	req->is_admin = true;

	upiu = (struct ufshci_query_request_upiu *)&req->request_upiu;
	memset(upiu, 0, req->request_size);
	upiu->header.trans_type = UFSHCI_UPIU_TRANSACTION_CODE_QUERY_REQUEST;
	upiu->header.ext_iid_or_function = param.function;
	upiu->opcode = param.opcode;
	upiu->idn = param.type;
	upiu->index = param.index;
	upiu->selector = param.selector;
	upiu->value_64 = param.value;
	upiu->length = param.desc_size;

	ufshci_ctrlr_submit_transfer_request(ctrlr, req);
}