samples/drivers/soc_flash_nrf5/src/main.c

/*
 * Copyright (c) 2016 Linaro Limited
 *               2016 Intel Corporation.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <zephyr.h>
#include <flash.h>
#include <device.h>
#if defined(CONFIG_STDOUT_CONSOLE)
#include <stdio.h>
#define PRINT           printf
#else
#include <misc/printk.h>
#define PRINT           printk
#endif

/* Offset between pages */
#define FLASH_TEST_OFFSET 0x40000
#define FLASH_PAGE_SIZE   4096
#define TEST_DATA_WORD_0  0x1122
#define TEST_DATA_WORD_1  0xaabb
#define TEST_DATA_WORD_2  0xabcd
#define TEST_DATA_WORD_3  0x1234
#define TEST_DATA_LEN     4

void main(void)
{
	struct device *flash_dev;
	uint32_t buf_array_1[4] = { TEST_DATA_WORD_0, TEST_DATA_WORD_1,
				TEST_DATA_WORD_2, TEST_DATA_WORD_3 };
	uint32_t buf_array_2[4] = { TEST_DATA_WORD_3, TEST_DATA_WORD_1,
				TEST_DATA_WORD_2, TEST_DATA_WORD_0 };
	uint32_t buf_word = 0;
	uint32_t i, offset;

	PRINT("\nNordic nRF5 Flash Testing\n");
	PRINT("=========================\n");

	flash_dev = device_get_binding("NRF5_FLASH");

	if (!flash_dev) {
		PRINT("Nordic nRF5 flash driver was not found!\n");
		return;
	}

	PRINT("\nTest 1: Flash erase page at 0x%x\n", FLASH_TEST_OFFSET);
	if (flash_erase(flash_dev, FLASH_TEST_OFFSET, FLASH_PAGE_SIZE) != 0) {
		PRINT("   Flash erase failed!\n");
	} else {
		PRINT("   Flash erase succeeded!\n");
	}

	PRINT("\nTest 2: Flash write (word array 1)\n");
	flash_write_protection_set(flash_dev, false);
	for (i = 0; i < TEST_DATA_LEN; i++) {
		offset = FLASH_TEST_OFFSET + (i << 2);
		PRINT("   Attempted to write %x at 0x%x\n", buf_array_1[i],
				offset);
		if (flash_write(flash_dev, offset, &buf_array_1[i],
					TEST_DATA_LEN) != 0) {
			PRINT("   Flash write failed!\n");
			return;
		}
		PRINT("   Attempted to read 0x%x\n", offset);
		if (flash_read(flash_dev, offset, &buf_word,
					TEST_DATA_LEN) != 0) {
			PRINT("   Flash read failed!\n");
			return;
		}
		PRINT("   Data read: %x\n", buf_word);
		if (buf_array_1[i] == buf_word) {
			PRINT("   Data read matches data written. Good!\n");
		} else {
			PRINT("   Data read does not match data written!\n");
		}
	}
	flash_write_protection_set(flash_dev, true);

	offset = FLASH_TEST_OFFSET - FLASH_PAGE_SIZE * 2;
	PRINT("\nTest 3: Flash erase (4 pages at 0x%x)\n", offset);
	if (flash_erase(flash_dev, offset, FLASH_PAGE_SIZE * 4) != 0) {
		PRINT("   Flash erase failed!\n");
	} else {
		PRINT("   Flash erase succeeded!\n");
	}

	PRINT("\nTest 4: Flash write (word array 2)\n");
	flash_write_protection_set(flash_dev, false);
	for (i = 0; i < TEST_DATA_LEN; i++) {
		offset = FLASH_TEST_OFFSET + (i << 2);
		PRINT("   Attempted to write %x at 0x%x\n", buf_array_2[i],
				offset);
		if (flash_write(flash_dev, offset, &buf_array_2[i],
					TEST_DATA_LEN) != 0) {
			PRINT("   Flash write failed!\n");
			return;
		}
		PRINT("   Attempted to read 0x%x\n", offset);
		if (flash_read(flash_dev, offset, &buf_word,
					TEST_DATA_LEN) != 0) {
			PRINT("   Flash read failed!\n");
			return;
		}
		PRINT("   Data read: %x\n", buf_word);
		if (buf_array_2[i] == buf_word) {
			PRINT("   Data read matches data written. Good!\n");
		} else {
			PRINT("   Data read does not match data written!\n");
		}
	}
	flash_write_protection_set(flash_dev, true);
}