/*
 * Copyright(c) 2012-2022 Intel Corporation
 * Copyright(c) 2024 Huawei Technologies
 * SPDX-License-Identifier: BSD-3-Clause
 */

#include "ocf/ocf.h"
#include "../ocf_cache_priv.h"
#include "engine_wi.h"
#include "engine_common.h"
#include "engine_io.h"
#include "../concurrency/ocf_concurrency.h"
#include "../ocf_request.h"
#include "../utils/utils_cache_line.h"
#include "../metadata/metadata.h"

#define OCF_ENGINE_DEBUG_IO_NAME "wi"
#include "engine_debug.h"

static int _ocf_write_wi_next_pass(struct ocf_request *req)
{
	ocf_req_unlock_wr(ocf_cache_line_concurrency(req->cache), req);

	if (req->wi_second_pass) {
		req->complete(req, 0);
		ocf_req_put(req);

		return 0;
	}

	/* Perform second pass of write invalidate. It is necessary
	   only if concurrent I/O had inserted target LBAs to cache after
	   this request did traversation. These LBAs might have been
	   written by this request behind the concurrent I/O's back,
	   resulting in making these sectors effectively invalid.
	   In this case we must update these sectors metadata to
	   reflect this. However we won't know about this after we
	   traverse the request again - hence calling ocf_write_wi
	   again with req->wi_second_pass set to indicate that this
	   is a second pass (core write should be skipped). */
	req->wi_second_pass = true;
	ocf_write_wi(req);

	return 0;
}

static void _ocf_write_wi_io_flush_metadata(struct ocf_request *req, int error)
{
	if (error) {
		ocf_core_stats_cache_error_update(req->core, OCF_WRITE);
		ocf_engine_error(req, true, "Failed to write data to cache");
	}

	if (!error && !req->wi_second_pass && ocf_engine_is_miss(req)) {
		/* need another pass */
		ocf_queue_push_req_cb(req, _ocf_write_wi_next_pass,
				OCF_QUEUE_ALLOW_SYNC | OCF_QUEUE_PRIO_HIGH);
		return;
	}

	ocf_req_unlock_wr(ocf_cache_line_concurrency(req->cache), req);

	req->complete(req, error);

	ocf_req_put(req);
}

static int ocf_write_wi_update_and_flush_metadata(struct ocf_request *req)
{
	struct ocf_cache *cache = req->cache;

	if (!ocf_engine_mapped_count(req)) {
		/* jump directly to next pass */
		_ocf_write_wi_next_pass(req);
		return 0;
	}

	/* There are mapped cache line, need to remove them */

	ocf_hb_req_prot_lock_wr(req); /*- Metadata WR access ---------------*/

	/* Remove mapped cache lines from metadata */
	ocf_purge_map_info(req);

	ocf_hb_req_prot_unlock_wr(req); /*- END Metadata WR access ---------*/

	if (req->info.flush_metadata) {
		/* Request was dirty and need to flush metadata */
		ocf_metadata_flush_do_asynch(cache, req,
				_ocf_write_wi_io_flush_metadata);

		return 0;
	}

	_ocf_write_wi_io_flush_metadata(req, 0);

	return 0;
}

static void _ocf_write_wi_core_complete(struct ocf_request *req, int error)
{
	OCF_DEBUG_RQ(req, "Completion");

	if (error) {
		ocf_core_stats_core_error_update(req->core, OCF_WRITE);

		ocf_req_unlock_wr(ocf_cache_line_concurrency(req->cache), req);

		req->complete(req, error);

		ocf_req_put(req);
	} else {
		ocf_queue_push_req_cb(req,
				ocf_write_wi_update_and_flush_metadata,
				OCF_QUEUE_ALLOW_SYNC | OCF_QUEUE_PRIO_HIGH);
	}
}

static int _ocf_write_wi_core_write(struct ocf_request *req)
{
	/* Get OCF request - increase reference counter */
	ocf_req_get(req);

	if (req->info.dirty_any) {
		ocf_hb_req_prot_lock_rd(req);
		/* Need to clean, start it */
		ocf_engine_clean(req);
		ocf_hb_req_prot_unlock_rd(req);

		/* The processing shall be resumed once the async cleaning
		   ends */
		ocf_req_put(req);

		return 0;
	}

	OCF_DEBUG_RQ(req, "Submit");

	/* Submit write IO to the core */
	ocf_engine_forward_core_io_req(req, _ocf_write_wi_core_complete);

	/* Update statistics */
	ocf_engine_update_block_stats(req);

	ocf_core_stats_pt_block_update(req->core, req->part_id, req->rw,
			req->bytes);

	ocf_core_stats_request_pt_update(req->core, req->part_id, req->rw,
			req->info.hit_no, req->core_line_count);

	/* Put OCF request - decrease reference counter */
	ocf_req_put(req);

	return 0;
}

static void _ocf_write_wi_on_resume(struct ocf_request *req)
{
	OCF_DEBUG_RQ(req, "On resume");
	ocf_queue_push_req(req, OCF_QUEUE_ALLOW_SYNC | OCF_QUEUE_PRIO_HIGH);
}

int ocf_write_wi(struct ocf_request *req)
{
	int lock = OCF_LOCK_NOT_ACQUIRED;

	OCF_DEBUG_TRACE(req->cache);


	/* Get OCF request - increase reference counter */
	ocf_req_get(req);

	/* Set resume handler */
	req->engine_handler = req->wi_second_pass ?
			ocf_write_wi_update_and_flush_metadata :
			_ocf_write_wi_core_write;

	ocf_req_hash(req);
	ocf_hb_req_prot_lock_rd(req); /*- Metadata READ access, No eviction --------*/

	/* Travers to check if request is mapped fully */
	ocf_engine_traverse(req);

	if (ocf_engine_mapped_count(req)) {
		/* Some cache line are mapped, lock request for WRITE access */
		lock = ocf_req_async_lock_wr(
				ocf_cache_line_concurrency(req->cache),
				req, _ocf_write_wi_on_resume);
	} else {
		lock = OCF_LOCK_ACQUIRED;
	}

	ocf_hb_req_prot_unlock_rd(req); /*- END Metadata READ access----------------*/

	if (lock >= 0) {
		if (lock == OCF_LOCK_ACQUIRED) {
			req->engine_handler(req);
		} else {
			/* WR lock was not acquired, need to wait for resume */
			OCF_DEBUG_RQ(req, "NO LOCK");
		}
	} else {
		OCF_DEBUG_RQ(req, "LOCK ERROR %d", lock);
		req->complete(req, lock);
		ocf_req_put(req);
	}

	/* Put OCF request - decrease reference counter */
	ocf_req_put(req);

	return 0;
}