mirror of
				https://review.coreboot.org/flashrom.git
				synced 2025-11-04 07:00:39 +01:00 
			
		
		
		
	Change-Id: I3a8a8e59cbed871e0ecf953e547de56c0656e112 Signed-off-by: Anastasia Klimchuk <aklm@flashrom.org> Reviewed-on: https://review.coreboot.org/c/flashrom/+/89361 Reviewed-by: Stefan Reinauer <stefan.reinauer@coreboot.org> Reviewed-by: Peter Marheine <pmarheine@chromium.org> Reviewed-by: Antonio Vázquez Blanco <antoniovazquezblanco@gmail.com> Tested-by: build bot (Jenkins) <no-reply@coreboot.org>
		
			
				
	
	
		
			274 lines
		
	
	
		
			7.7 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			274 lines
		
	
	
		
			7.7 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * This file is part of the flashrom project.
 | 
						|
 *
 | 
						|
 * SPDX-License-Identifier: GPL-2.0-only
 | 
						|
 * SPDX-FileCopyrightText: 2020 Google LLC
 | 
						|
 */
 | 
						|
 | 
						|
#include <include/test.h>
 | 
						|
 | 
						|
#include "wraps.h"
 | 
						|
#include "tests.h"
 | 
						|
#include "programmer.h"
 | 
						|
#include "flashchips.h"
 | 
						|
#include "chipdrivers.h"
 | 
						|
#include "spi.h"
 | 
						|
 | 
						|
struct flashchip mock_chip = {
 | 
						|
	.vendor		= "Generic",
 | 
						|
	.name		= "unknown SPI chip (RDID)",
 | 
						|
	.bustype	= BUS_SPI,
 | 
						|
	.manufacture_id	= GENERIC_MANUF_ID,
 | 
						|
	.model_id	= GENERIC_DEVICE_ID,
 | 
						|
	.total_size	= 0,
 | 
						|
	.page_size	= 256,
 | 
						|
	.tested		= TEST_BAD_PREW,
 | 
						|
	.probe		= PROBE_SPI_RDID,
 | 
						|
	.write		= NO_WRITE_FUNC,
 | 
						|
};
 | 
						|
 | 
						|
/*
 | 
						|
 * This declaration is needed for visibility, so that wrap below could
 | 
						|
 * redirect to real function.
 | 
						|
 */
 | 
						|
int __real_spi_send_command(const struct flashctx *flash,
 | 
						|
		unsigned int writecnt, unsigned int readcnt,
 | 
						|
		const unsigned char *writearr, unsigned char *readarr);
 | 
						|
 | 
						|
int __wrap_spi_send_command(const struct flashctx *flash,
 | 
						|
		unsigned int writecnt, unsigned int readcnt,
 | 
						|
		const unsigned char *writearr, unsigned char *readarr)
 | 
						|
{
 | 
						|
	if (flash->chip != &mock_chip)
 | 
						|
		/*
 | 
						|
		 * Caller is some other test, redirecting to real function.
 | 
						|
		 * This test is the only one which uses wrap of spi_send_command,
 | 
						|
		 * all other tests use real function.
 | 
						|
		*/
 | 
						|
		return __real_spi_send_command(flash, writecnt, readcnt, writearr, readarr);
 | 
						|
 | 
						|
	if (!flash->progress_callback)
 | 
						|
		check_expected_ptr(flash);
 | 
						|
 | 
						|
	assert_int_equal(writecnt,    mock_type(int));
 | 
						|
	assert_int_equal(writearr[0], mock_type(int));
 | 
						|
 | 
						|
	int rcnt = mock_type(int);
 | 
						|
	assert_int_equal(readcnt, rcnt);
 | 
						|
	for (int i = 0; i < rcnt; i++)
 | 
						|
		readarr[i] = i;
 | 
						|
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static void spi_read_progress_cb(enum flashrom_progress_stage stage, size_t current, size_t total, void* user_data)
 | 
						|
{
 | 
						|
	uint32_t *cnt = (uint32_t *) user_data;
 | 
						|
	assert_int_equal(0x400, total);
 | 
						|
	switch (*cnt) {
 | 
						|
		case 0:
 | 
						|
			assert_int_equal(0x0, current);
 | 
						|
			break;
 | 
						|
		case 1:
 | 
						|
			assert_int_equal(0x100, current);
 | 
						|
			break;
 | 
						|
		case 2:
 | 
						|
			assert_int_equal(0x200, current);
 | 
						|
			break;
 | 
						|
		case 3:
 | 
						|
			assert_int_equal(0x300, current);
 | 
						|
			break;
 | 
						|
		case 4:
 | 
						|
			assert_int_equal(0x400, current);
 | 
						|
			break;
 | 
						|
		default:
 | 
						|
			fail();
 | 
						|
	}
 | 
						|
	(*cnt)++;
 | 
						|
}
 | 
						|
 | 
						|
void default_spi_read_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
	uint8_t buf[0x400] = { 0x0 };
 | 
						|
	uint32_t cnt = 0;
 | 
						|
	const unsigned int max_data_read = 0x100;
 | 
						|
	const unsigned int offset = 0x100;
 | 
						|
	struct registered_master mst = {
 | 
						|
		.spi.read = default_spi_read,
 | 
						|
		.spi.max_data_read = max_data_read
 | 
						|
	};
 | 
						|
 | 
						|
	/* setup initial test state */
 | 
						|
	struct flashctx flashctx = {
 | 
						|
		.chip = &mock_chip,
 | 
						|
		.mst = &mst
 | 
						|
	};
 | 
						|
	flashrom_set_progress_callback_v2(&flashctx, spi_read_progress_cb, (void *) &cnt);
 | 
						|
	init_progress(&flashctx, FLASHROM_PROGRESS_READ, 0x400);
 | 
						|
	for (int i = 0; i < 4; i++) {
 | 
						|
		will_return(__wrap_spi_send_command, JEDEC_WRDI);
 | 
						|
		will_return(__wrap_spi_send_command, JEDEC_READ);
 | 
						|
		will_return(__wrap_spi_send_command, max_data_read);
 | 
						|
	}
 | 
						|
	assert_int_equal(0, default_spi_read(&flashctx, buf, offset, sizeof(buf)));
 | 
						|
	assert_int_equal(5, cnt);
 | 
						|
 | 
						|
	flashrom_set_progress_callback_v2(&flashctx, NULL, NULL);
 | 
						|
}
 | 
						|
 | 
						|
void spi_write_enable_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
 | 
						|
	/* setup initial test state. */
 | 
						|
	struct flashctx flashctx = { .chip = &mock_chip };
 | 
						|
	expect_memory(__wrap_spi_send_command, flash,
 | 
						|
			&flashctx, sizeof(flashctx));
 | 
						|
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_WREN_OUTSIZE);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_WREN);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_WREN_INSIZE);
 | 
						|
	assert_int_equal(0, spi_write_enable(&flashctx));
 | 
						|
}
 | 
						|
 | 
						|
void spi_write_disable_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
 | 
						|
	/* setup initial test state. */
 | 
						|
	struct flashctx flashctx = { .chip = &mock_chip };
 | 
						|
	expect_memory(__wrap_spi_send_command, flash,
 | 
						|
			&flashctx, sizeof(flashctx));
 | 
						|
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_WRDI_OUTSIZE);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_WRDI);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_WRDI_INSIZE);
 | 
						|
	assert_int_equal(0, spi_write_disable(&flashctx));
 | 
						|
}
 | 
						|
 | 
						|
void probe_spi_rdid_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
 | 
						|
	/* setup initial test state. */
 | 
						|
	struct flashctx flashctx = { .chip = &mock_chip };
 | 
						|
	expect_memory(__wrap_spi_send_command, flash,
 | 
						|
			&flashctx, sizeof(flashctx));
 | 
						|
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RDID_OUTSIZE);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RDID);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RDID_INSIZE);
 | 
						|
	assert_int_equal(0, probe_spi_rdid(&flashctx));
 | 
						|
}
 | 
						|
 | 
						|
void probe_spi_rdid4_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
 | 
						|
	/* setup initial test state. */
 | 
						|
	struct flashctx flashctx = { .chip = &mock_chip };
 | 
						|
	expect_memory(__wrap_spi_send_command, flash,
 | 
						|
			&flashctx, sizeof(flashctx));
 | 
						|
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RDID_OUTSIZE);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RDID);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RDID_INSIZE + 1);
 | 
						|
	assert_int_equal(0, probe_spi_rdid4(&flashctx));
 | 
						|
}
 | 
						|
 | 
						|
void probe_spi_rems_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
 | 
						|
	/* setup initial test state. */
 | 
						|
	struct flashctx flashctx = { .chip = &mock_chip };
 | 
						|
	expect_memory(__wrap_spi_send_command, flash,
 | 
						|
			&flashctx, sizeof(flashctx));
 | 
						|
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_REMS_OUTSIZE);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_REMS);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_REMS_INSIZE);
 | 
						|
	assert_int_equal(0, probe_spi_rems(&flashctx));
 | 
						|
}
 | 
						|
 | 
						|
void probe_spi_res1_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
 | 
						|
	/* setup initial test state. */
 | 
						|
	struct flashctx flashctx = { .chip = &mock_chip };
 | 
						|
	expect_memory(__wrap_spi_send_command, flash,
 | 
						|
			&flashctx, sizeof(flashctx));
 | 
						|
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RES_OUTSIZE);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RES);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RES_INSIZE + 1);
 | 
						|
	assert_int_equal(0, probe_spi_res2(&flashctx));
 | 
						|
}
 | 
						|
 | 
						|
void probe_spi_res2_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
 | 
						|
	/* setup initial test state. */
 | 
						|
	clear_spi_id_cache();
 | 
						|
	struct flashctx flashctx = { .chip = &mock_chip };
 | 
						|
	expect_memory(__wrap_spi_send_command, flash,
 | 
						|
			&flashctx, sizeof(flashctx));
 | 
						|
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RES_OUTSIZE);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RES);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RES_INSIZE + 1);
 | 
						|
	assert_int_equal(0, probe_spi_res2(&flashctx));
 | 
						|
}
 | 
						|
 | 
						|
void probe_spi_res3_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
 | 
						|
	/* setup initial test state. */
 | 
						|
	struct flashctx flashctx = { .chip = &mock_chip };
 | 
						|
	expect_memory(__wrap_spi_send_command, flash,
 | 
						|
			&flashctx, sizeof(flashctx));
 | 
						|
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RES_OUTSIZE);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RES);
 | 
						|
	will_return(__wrap_spi_send_command, JEDEC_RES_INSIZE + 2);
 | 
						|
	assert_int_equal(0, probe_spi_res3(&flashctx));
 | 
						|
}
 | 
						|
 | 
						|
void probe_spi_at25f_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
 | 
						|
	/* setup initial test state. */
 | 
						|
	struct flashctx flashctx = { .chip = &mock_chip };
 | 
						|
	expect_memory(__wrap_spi_send_command, flash,
 | 
						|
			&flashctx, sizeof(flashctx));
 | 
						|
 | 
						|
	will_return(__wrap_spi_send_command, AT25F_RDID_OUTSIZE);
 | 
						|
	will_return(__wrap_spi_send_command, AT25F_RDID);
 | 
						|
	will_return(__wrap_spi_send_command, AT25F_RDID_INSIZE);
 | 
						|
	assert_int_equal(0, probe_spi_at25f(&flashctx));
 | 
						|
}
 | 
						|
 | 
						|
/* spi95.c */
 | 
						|
void probe_spi_st95_test_success(void **state)
 | 
						|
{
 | 
						|
	(void) state; /* unused */
 | 
						|
 | 
						|
	/* setup initial test state. */
 | 
						|
	struct flashctx flashctx = { .chip = &mock_chip };
 | 
						|
	expect_memory(__wrap_spi_send_command, flash,
 | 
						|
			&flashctx, sizeof(flashctx));
 | 
						|
 | 
						|
	/* chip total size < 64K. */
 | 
						|
	uint32_t rdid_outsize = ST_M95_RDID_2BA_OUTSIZE; // 16 bit address
 | 
						|
 | 
						|
	will_return(__wrap_spi_send_command, rdid_outsize);
 | 
						|
	will_return(__wrap_spi_send_command, ST_M95_RDID);
 | 
						|
	will_return(__wrap_spi_send_command, ST_M95_RDID_INSIZE);
 | 
						|
	assert_int_equal(0, probe_spi_st95(&flashctx));
 | 
						|
}
 |