1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
/*
* This file is part of the flashrom project.
*
* Copyright 2022 Alexander Goncharov <chat@joursoir.net>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; version 2 of the License.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*/
#include "lifecycle.h"
#if CONFIG_CH341A_SPI == 1
/* Same macro as in ch341a_spi.c programmer. */
#define WRITE_EP 0x02
struct ch341a_spi_io_state {
struct libusb_transfer *transfer_out;
};
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);
assert_null(io_state->transfer_out);
io_state->transfer_out = 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;
}
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, "");
}
#else
SKIP_TEST(ch341a_spi_basic_lifecycle_test_success)
#endif /* CONFIG_CH341A_SPI */
|
