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>
		
			
				
	
	
		
			133 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			133 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * This file is part of the flashrom project.
 | 
						|
 *
 | 
						|
 * SPDX-License-Identifier: GPL-2.0-only
 | 
						|
 * SPDX-FileCopyrightText: 2022 Alexander Goncharov <chat@joursoir.net>
 | 
						|
 */
 | 
						|
 | 
						|
#include <stdlib.h>
 | 
						|
 | 
						|
#include "lifecycle.h"
 | 
						|
 | 
						|
#if CONFIG_CH341A_SPI == 1
 | 
						|
 | 
						|
/* Same macro as in ch341a_spi.c programmer. */
 | 
						|
#define WRITE_EP 0x02
 | 
						|
#define READ_EP 0x82
 | 
						|
 | 
						|
struct ch341a_spi_io_state {
 | 
						|
	struct libusb_transfer *transfer_out;
 | 
						|
	/*
 | 
						|
	 * Since the test transfers a data that fits in one CH341 packet, we
 | 
						|
	 * don't need an array of these transfers (as is done in the driver code).
 | 
						|
	 */
 | 
						|
	struct libusb_transfer *transfer_in;
 | 
						|
};
 | 
						|
 | 
						|
static struct libusb_transfer *ch341a_libusb_alloc_transfer(void *state, int iso_packets)
 | 
						|
{
 | 
						|
	return calloc(1, sizeof(struct libusb_transfer));
 | 
						|
}
 | 
						|
 | 
						|
/*
 | 
						|
 * The libusb code stores submitted transfers in their own context. But this
 | 
						|
 * function doesn't require a context pointer because libusb stores context
 | 
						|
 * pointers in libusb_transfer instances. Since our ch341 driver is using
 | 
						|
 * the default context, we store the transfer in our own.
 | 
						|
 */
 | 
						|
static int ch341a_libusb_submit_transfer(void *state, struct libusb_transfer *transfer)
 | 
						|
{
 | 
						|
	struct ch341a_spi_io_state *io_state = state;
 | 
						|
 | 
						|
	assert_true(transfer->endpoint == WRITE_EP || transfer->endpoint == READ_EP);
 | 
						|
 | 
						|
	if (transfer->endpoint == WRITE_EP) {
 | 
						|
		assert_null(io_state->transfer_out);
 | 
						|
		io_state->transfer_out = transfer;
 | 
						|
	} else if (transfer->endpoint == READ_EP) {
 | 
						|
		assert_null(io_state->transfer_in);
 | 
						|
		io_state->transfer_in = transfer;
 | 
						|
	}
 | 
						|
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static void ch341a_libusb_free_transfer(void *state, struct libusb_transfer *transfer)
 | 
						|
{
 | 
						|
	free(transfer);
 | 
						|
}
 | 
						|
 | 
						|
/*
 | 
						|
 * Handle submitted transfers by pretending that a transfer is completed and
 | 
						|
 * invoking its callback (that is the flashrom code).
 | 
						|
 */
 | 
						|
static int ch341a_libusb_handle_events_timeout(void *state, libusb_context *ctx, struct timeval *tv)
 | 
						|
{
 | 
						|
	struct ch341a_spi_io_state *io_state = state;
 | 
						|
 | 
						|
	if (io_state->transfer_out) {
 | 
						|
		io_state->transfer_out->status = LIBUSB_TRANSFER_COMPLETED;
 | 
						|
		io_state->transfer_out->actual_length = io_state->transfer_out->length;
 | 
						|
		io_state->transfer_out->callback(io_state->transfer_out);
 | 
						|
		io_state->transfer_out = NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	if (io_state->transfer_in) {
 | 
						|
		io_state->transfer_in->buffer[1] = reverse_byte(0xEF); /* WINBOND_NEX_ID */
 | 
						|
		io_state->transfer_in->buffer[2] = reverse_byte(0x40); /* WINBOND_NEX_W25Q128_V left byte */
 | 
						|
		io_state->transfer_in->buffer[3] = reverse_byte(0x18); /* WINBOND_NEX_W25Q128_V right byte */
 | 
						|
 | 
						|
		io_state->transfer_in->status = LIBUSB_TRANSFER_COMPLETED;
 | 
						|
		io_state->transfer_in->actual_length = io_state->transfer_in->length;
 | 
						|
		io_state->transfer_in->callback(io_state->transfer_in);
 | 
						|
		io_state->transfer_in = NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
void ch341a_spi_basic_lifecycle_test_success(void **state)
 | 
						|
{
 | 
						|
	struct ch341a_spi_io_state ch341a_spi_io_state = { 0 };
 | 
						|
	struct io_mock_fallback_open_state ch341a_spi_fallback_open_state = {
 | 
						|
		.noc = 0,
 | 
						|
		.paths = { NULL },
 | 
						|
	};
 | 
						|
	const struct io_mock ch341a_spi_io = {
 | 
						|
		.state = &ch341a_spi_io_state,
 | 
						|
		.libusb_alloc_transfer = &ch341a_libusb_alloc_transfer,
 | 
						|
		.libusb_submit_transfer = &ch341a_libusb_submit_transfer,
 | 
						|
		.libusb_free_transfer = &ch341a_libusb_free_transfer,
 | 
						|
		.libusb_handle_events_timeout = &ch341a_libusb_handle_events_timeout,
 | 
						|
		.fallback_open_state = &ch341a_spi_fallback_open_state,
 | 
						|
	};
 | 
						|
 | 
						|
	run_basic_lifecycle(state, &ch341a_spi_io, &programmer_ch341a_spi, "");
 | 
						|
}
 | 
						|
 | 
						|
void ch341a_spi_probe_lifecycle_test_success(void **state)
 | 
						|
{
 | 
						|
	struct ch341a_spi_io_state ch341a_spi_io_state = { 0 };
 | 
						|
	struct io_mock_fallback_open_state ch341a_spi_fallback_open_state = {
 | 
						|
		.noc = 0,
 | 
						|
		.paths = { NULL },
 | 
						|
	};
 | 
						|
	const struct io_mock ch341a_spi_io = {
 | 
						|
		.state = &ch341a_spi_io_state,
 | 
						|
		.libusb_alloc_transfer = &ch341a_libusb_alloc_transfer,
 | 
						|
		.libusb_submit_transfer = &ch341a_libusb_submit_transfer,
 | 
						|
		.libusb_free_transfer = &ch341a_libusb_free_transfer,
 | 
						|
		.libusb_handle_events_timeout = &ch341a_libusb_handle_events_timeout,
 | 
						|
		.fallback_open_state = &ch341a_spi_fallback_open_state,
 | 
						|
	};
 | 
						|
 | 
						|
	const char *expected_matched_names[1] = {"W25Q128.V"};
 | 
						|
	run_probe_v2_lifecycle(state, &ch341a_spi_io, &programmer_ch341a_spi, "", "W25Q128.V",
 | 
						|
				expected_matched_names, 1);
 | 
						|
}
 | 
						|
 | 
						|
#else
 | 
						|
	SKIP_TEST(ch341a_spi_basic_lifecycle_test_success)
 | 
						|
	SKIP_TEST(ch341a_spi_probe_lifecycle_test_success)
 | 
						|
#endif /* CONFIG_CH341A_SPI */
 |