/******************************************************************************
*
* (c) Copyright 2010-14 Xilinx, Inc. All rights reserved.
*
* This file contains confidential and proprietary information of Xilinx, Inc.
* and is protected under U.S. and international copyright and other
* intellectual property laws.
*
* DISCLAIMER
* This disclaimer is not a license and does not grant any rights to the
* materials distributed herewith. Except as otherwise provided in a valid
* license issued to you by Xilinx, and to the maximum extent permitted by
* applicable law: (1) THESE MATERIALS ARE MADE AVAILABLE "AS IS" AND WITH ALL
* FAULTS, AND XILINX HEREBY DISCLAIMS ALL WARRANTIES AND CONDITIONS, EXPRESS,
* IMPLIED, OR STATUTORY, INCLUDING BUT NOT LIMITED TO WARRANTIES OF
* MERCHANTABILITY, NON-INFRINGEMENT, OR FITNESS FOR ANY PARTICULAR PURPOSE
* and (2) Xilinx shall not be liable (whether in contract or tort, including
* negligence, or under any other theory of liability) for any loss or damage
* of any kind or nature related to, arising under or in connection with these
* materials, including for any direct, or any indirect, special, incidental,
* or consequential loss or damage (including loss of data, profits, goodwill,
* or any type of loss or damage suffered as a result of any action brought by
* a third party) even if such damage or loss was reasonably foreseeable or
* Xilinx had been advised of the possibility of the same.
*
* CRITICAL APPLICATIONS
* Xilinx products are not designed or intended to be fail-safe, or for use in
* any application requiring fail-safe performance, such as life-support or
* safety devices or systems, Class III medical devices, nuclear facilities,
* applications related to the deployment of airbags, or any other applications
* that could lead to death, personal injury, or severe property or
* environmental damage (individually and collectively, "Critical
* Applications"). Customer assumes the sole risk and liability of any use of
* Xilinx products in Critical Applications, subject only to applicable laws
* and regulations governing limitations on product liability.
*
* THIS COPYRIGHT NOTICE AND DISCLAIMER MUST BE RETAINED AS PART OF THIS FILE
* AT ALL TIMES.
*
******************************************************************************/
/*****************************************************************************/
/**
*
* @file intg_markblockbad_test.c
*
* This file contains the design example for using NAND driver (XNandPsu).
* This example tests the erase, read and write feature of the controller.
* The flash is erased and written. The data is
* read back and compared with the data written for correctness.
*
* @note None.
*
*
* <pre>
*
* MODIFICATION HISTORY:
*
* Ver   Who    Date	 Changes
* ----- -----  -------- -----------------------------------------------
* 1.0   sb    12/18/2014 First release
*
*
* </pre>
*
******************************************************************************/

/***************************** Include Files ********************************/

#include "intg.h"

/************************** Constant Definitions ****************************/

/**************************** Type Definitions ******************************/

/***************** Macros (Inline Functions) Definitions ********************/

/************************** Variable Definitions ****************************/

/************************** Function Prototypes *****************************/
s32 Mark_BlockBad_Test(XNandPsu * NandInstPtr);
/************************** Function Definitions ****************************/

/****************************************************************************/
/**
*
* Entry point to call the Mark Block Bad R/W test.
*
* @param	NandInstPtr - Instance to the nand driver.
* @param	TestLoops - Number of tests to execute.
*
* @return   Number of test failures.
*
* @note	 None.
*
*****************************************************************************/
int Intg_MarkBlockBadTest(XNandPsu * NandInstPtr, int TestLoops)
{

	s32 Status = XST_FAILURE;
	CT_TestReset("Module Mark Block Bad test");

	while(TestLoops--) {
		Status = Mark_BlockBad_Test(NandInstPtr);
		if (Status != XST_SUCCESS) {
			CT_LOG_FAILURE("Mark Block Bad Test Failed"
					" with %d mismatches\r\n", MismatchCounter);
			break;
		}
		CT_NotifyNextPass();

	}

	return(CT_GetTestFailures());
}

/****************************************************************************/
/**
*
* This function runs a test on the NAND flash device using the basic driver
* functions in polled mode.
* The function does the following tasks:
*   - Marks Blocks bad.
*   - Erase the blocks.
*   - Write data to the blocks.
*   - Read back the data from the blocks.
*   - Compare the data read against the data Written.
*
* @param	NandInstPtr - Instance to the nand driver.
*
* @return
*		- XST_SUCCESS if successful.
*		- XST_FAILURE if failed.
*
* @note
*		None
*
****************************************************************************/
s32 Mark_BlockBad_Test(XNandPsu * NandInstPtr)
{
	s32 Status = XST_FAILURE;
	s32 BlockNo;
	s32 i;
	u32 Index;
	u64 PageOff;
	u32 Length;
	u32 BlockSize = NandInstPtr->Geometry.BlockSize;
	u64 BlockOff;
	MismatchCounter = 0;

	PageOff = (u64)(TEST_PAGE_START * NandInstPtr->Geometry.BytesPerPage);
	Length = NandInstPtr->Geometry.BytesPerPage;

	/*
	 * Initialize the write buffer
	 */
	for (Index = 0; Index < Length;Index++) {
		WriteBuffer[Index] = 2U;
	}

	/*
	 * Marking blocks 1 & 3 as bad
	 */
	for (BlockNo = 0 ; BlockNo < 5 ; BlockNo++){
		if(BlockNo%2 == 0){
			continue;
		}
		Status = XNandPsu_MarkBlockBad(NandInstPtr,BlockNo);
		if(Status != XST_SUCCESS){
			goto Out;
		}
	}
	/*
	 * Performing Block Erase Read Write on Block 1,2,3
	 */
	for (BlockNo = TEST_BLOCK_START ; BlockNo < TEST_BLOCK_START + 3 ; BlockNo++ ){

		BlockOff = BlockNo * BlockSize;

		xil_printf("Erasing Block = %d \r\n", BlockNo);
		/*
		 * Erase the Block 1,2,3
		 */
		Status = XNandPsu_Erase(NandInstPtr, (u64)BlockOff, (u64)BlockSize);
		if (Status != XST_SUCCESS) {
			goto Out;
		}

		PageOff = BlockOff;

		for (i = 0; i < NandInstPtr->Geometry.PagesPerBlock; i++){
			/*
			 * Write to page offset
			 */
			Status = XNandPsu_Write(NandInstPtr, (u64)PageOff, (u64)Length,
					&WriteBuffer[0]);
			if (Status != XST_SUCCESS) {
				goto Out;
			}
			/*
			 * Read from the page after writing
			 */
			Status = XNandPsu_Read(NandInstPtr, (u64)PageOff, (u64)Length,
				&ReadBuffer[0]);
			if (Status != XST_SUCCESS) {
				goto Out;
			}
			/*
			 * Compare the results
			 */
			for (Index = 0U; Index < Length;Index++) {
				if (ReadBuffer[Index] != WriteBuffer[Index]) {
					MismatchCounter++;
					Status = XST_FAILURE;
				}
			}
			PageOff = PageOff + NandInstPtr->Geometry.BytesPerPage;
		}
	}

Out:
	return Status;
}