1
2
3
4
5
6
7
8
9
10
11
12
13
14
15#include "qemu/osdep.h"
16
17#include "hw/acpi/tpm.h"
18#include "libqtest.h"
19#include "tpm-util.h"
20#include "qapi/qmp/qdict.h"
21
22#define TIS_REG(LOCTY, REG) \
23 (TPM_TIS_ADDR_BASE + ((LOCTY) << 12) + REG)
24
25void tpm_util_crb_transfer(QTestState *s,
26 const unsigned char *req, size_t req_size,
27 unsigned char *rsp, size_t rsp_size)
28{
29 uint64_t caddr = qtest_readq(s, TPM_CRB_ADDR_BASE + A_CRB_CTRL_CMD_LADDR);
30 uint64_t raddr = qtest_readq(s, TPM_CRB_ADDR_BASE + A_CRB_CTRL_RSP_ADDR);
31
32 qtest_writeb(s, TPM_CRB_ADDR_BASE + A_CRB_LOC_CTRL, 1);
33
34 qtest_memwrite(s, caddr, req, req_size);
35
36 uint32_t sts, start = 1;
37 uint64_t end_time = g_get_monotonic_time() + 5 * G_TIME_SPAN_SECOND;
38 qtest_writel(s, TPM_CRB_ADDR_BASE + A_CRB_CTRL_START, start);
39 while (true) {
40 start = qtest_readl(s, TPM_CRB_ADDR_BASE + A_CRB_CTRL_START);
41 if ((start & 1) == 0) {
42 break;
43 }
44 if (g_get_monotonic_time() >= end_time) {
45 break;
46 }
47 };
48 start = qtest_readl(s, TPM_CRB_ADDR_BASE + A_CRB_CTRL_START);
49 g_assert_cmpint(start & 1, ==, 0);
50 sts = qtest_readl(s, TPM_CRB_ADDR_BASE + A_CRB_CTRL_STS);
51 g_assert_cmpint(sts & 1, ==, 0);
52
53 qtest_memread(s, raddr, rsp, rsp_size);
54}
55
56void tpm_util_tis_transfer(QTestState *s,
57 const unsigned char *req, size_t req_size,
58 unsigned char *rsp, size_t rsp_size)
59{
60 uint32_t sts;
61 uint16_t bcount;
62 size_t i;
63
64
65 qtest_writeb(s, TIS_REG(0, TPM_TIS_REG_ACCESS), TPM_TIS_ACCESS_REQUEST_USE);
66 qtest_writel(s, TIS_REG(0, TPM_TIS_REG_STS), TPM_TIS_STS_COMMAND_READY);
67
68 sts = qtest_readl(s, TIS_REG(0, TPM_TIS_REG_STS));
69 bcount = (sts >> 8) & 0xffff;
70 g_assert_cmpint(bcount, >=, req_size);
71
72
73 for (i = 0; i < req_size; i++) {
74 qtest_writeb(s, TIS_REG(0, TPM_TIS_REG_DATA_FIFO), req[i]);
75 }
76
77
78 qtest_writeb(s, TIS_REG(0, TPM_TIS_REG_STS), TPM_TIS_STS_TPM_GO);
79
80 uint64_t end_time = g_get_monotonic_time() + 50 * G_TIME_SPAN_SECOND;
81 do {
82 sts = qtest_readl(s, TIS_REG(0, TPM_TIS_REG_STS));
83 if ((sts & TPM_TIS_STS_DATA_AVAILABLE) != 0) {
84 break;
85 }
86 } while (g_get_monotonic_time() < end_time);
87
88 sts = qtest_readl(s, TIS_REG(0, TPM_TIS_REG_STS));
89 bcount = (sts >> 8) & 0xffff;
90
91 memset(rsp, 0, rsp_size);
92 for (i = 0; i < bcount; i++) {
93 rsp[i] = qtest_readb(s, TIS_REG(0, TPM_TIS_REG_DATA_FIFO));
94 }
95
96
97 qtest_writeb(s, TIS_REG(0, TPM_TIS_REG_ACCESS),
98 TPM_TIS_ACCESS_ACTIVE_LOCALITY);
99}
100
101void tpm_util_startup(QTestState *s, tx_func *tx)
102{
103 unsigned char buffer[1024];
104 unsigned char tpm_startup[] =
105 "\x80\x01\x00\x00\x00\x0c\x00\x00\x01\x44\x00\x00";
106 unsigned char tpm_startup_resp[] =
107 "\x80\x01\x00\x00\x00\x0a\x00\x00\x00\x00";
108
109 tx(s, tpm_startup, sizeof(tpm_startup), buffer, sizeof(buffer));
110
111 g_assert_cmpmem(buffer, sizeof(tpm_startup_resp),
112 tpm_startup_resp, sizeof(tpm_startup_resp));
113}
114
115void tpm_util_pcrextend(QTestState *s, tx_func *tx)
116{
117 unsigned char buffer[1024];
118 unsigned char tpm_pcrextend[] =
119 "\x80\x02\x00\x00\x00\x41\x00\x00\x01\x82\x00\x00\x00\x0a\x00\x00"
120 "\x00\x09\x40\x00\x00\x09\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00"
121 "\x0b\x74\x65\x73\x74\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
122 "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
123 "\x00";
124
125 unsigned char tpm_pcrextend_resp[] =
126 "\x80\x02\x00\x00\x00\x13\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
127 "\x01\x00\x00";
128
129 tx(s, tpm_pcrextend, sizeof(tpm_pcrextend), buffer, sizeof(buffer));
130
131 g_assert_cmpmem(buffer, sizeof(tpm_pcrextend_resp),
132 tpm_pcrextend_resp, sizeof(tpm_pcrextend_resp));
133}
134
135void tpm_util_pcrread(QTestState *s, tx_func *tx,
136 const unsigned char *exp_resp, size_t exp_resp_size)
137{
138 unsigned char buffer[1024];
139 unsigned char tpm_pcrread[] =
140 "\x80\x01\x00\x00\x00\x14\x00\x00\x01\x7e\x00\x00\x00\x01\x00\x0b"
141 "\x03\x00\x04\x00";
142
143 tx(s, tpm_pcrread, sizeof(tpm_pcrread), buffer, sizeof(buffer));
144
145 g_assert_cmpmem(buffer, exp_resp_size, exp_resp, exp_resp_size);
146}
147
148bool tpm_util_swtpm_has_tpm2(void)
149{
150 bool has_tpm2 = false;
151 char *out = NULL;
152 static const char *argv[] = {
153 "swtpm", "socket", "--help", NULL
154 };
155
156 if (!g_spawn_sync(NULL ,
157 (char **)argv,
158 NULL ,
159 G_SPAWN_SEARCH_PATH,
160 NULL ,
161 NULL ,
162 &out,
163 NULL ,
164 NULL ,
165 NULL)) {
166 return false;
167 }
168
169 if (strstr(out, "--tpm2")) {
170 has_tpm2 = true;
171 }
172
173 g_free(out);
174 return has_tpm2;
175}
176
177gboolean tpm_util_swtpm_start(const char *path, GPid *pid,
178 SocketAddress **addr, GError **error)
179{
180 char *swtpm_argv_tpmstate = g_strdup_printf("dir=%s", path);
181 char *swtpm_argv_ctrl = g_strdup_printf("type=unixio,path=%s/sock",
182 path);
183 gchar *swtpm_argv[] = {
184 g_strdup("swtpm"), g_strdup("socket"),
185 g_strdup("--tpmstate"), swtpm_argv_tpmstate,
186 g_strdup("--ctrl"), swtpm_argv_ctrl,
187 g_strdup("--tpm2"),
188 NULL
189 };
190 gboolean succ;
191 unsigned i;
192
193 *addr = g_new0(SocketAddress, 1);
194 (*addr)->type = SOCKET_ADDRESS_TYPE_UNIX;
195 (*addr)->u.q_unix.path = g_build_filename(path, "sock", NULL);
196
197 succ = g_spawn_async(NULL, swtpm_argv, NULL, G_SPAWN_SEARCH_PATH,
198 NULL, NULL, pid, error);
199
200 for (i = 0; swtpm_argv[i]; i++) {
201 g_free(swtpm_argv[i]);
202 }
203
204 return succ;
205}
206
207void tpm_util_swtpm_kill(GPid pid)
208{
209 int n;
210
211 if (!pid) {
212 return;
213 }
214
215 g_spawn_close_pid(pid);
216
217 n = kill(pid, 0);
218 if (n < 0) {
219 return;
220 }
221
222 kill(pid, SIGKILL);
223}
224
225void tpm_util_migrate(QTestState *who, const char *uri)
226{
227 QDict *rsp;
228
229 rsp = qtest_qmp(who,
230 "{ 'execute': 'migrate', 'arguments': { 'uri': %s } }",
231 uri);
232 g_assert(qdict_haskey(rsp, "return"));
233 qobject_unref(rsp);
234}
235
236void tpm_util_wait_for_migration_complete(QTestState *who)
237{
238 while (true) {
239 QDict *rsp_return;
240 bool completed;
241 const char *status;
242
243 qtest_qmp_send(who, "{ 'execute': 'query-migrate' }");
244 rsp_return = qtest_qmp_receive_success(who, NULL, NULL);
245 status = qdict_get_str(rsp_return, "status");
246 completed = strcmp(status, "completed") == 0;
247 g_assert_cmpstr(status, !=, "failed");
248 qobject_unref(rsp_return);
249 if (completed) {
250 return;
251 }
252 usleep(1000);
253 }
254}
255
256void tpm_util_migration_start_qemu(QTestState **src_qemu,
257 QTestState **dst_qemu,
258 SocketAddress *src_tpm_addr,
259 SocketAddress *dst_tpm_addr,
260 const char *miguri,
261 const char *ifmodel)
262{
263 char *src_qemu_args, *dst_qemu_args;
264
265 src_qemu_args = g_strdup_printf(
266 "-chardev socket,id=chr,path=%s "
267 "-tpmdev emulator,id=dev,chardev=chr "
268 "-device %s,tpmdev=dev ",
269 src_tpm_addr->u.q_unix.path, ifmodel);
270
271 *src_qemu = qtest_init(src_qemu_args);
272
273 dst_qemu_args = g_strdup_printf(
274 "-chardev socket,id=chr,path=%s "
275 "-tpmdev emulator,id=dev,chardev=chr "
276 "-device %s,tpmdev=dev "
277 "-incoming %s",
278 dst_tpm_addr->u.q_unix.path,
279 ifmodel, miguri);
280
281 *dst_qemu = qtest_init(dst_qemu_args);
282
283 free(src_qemu_args);
284 free(dst_qemu_args);
285}
286
