1
2
3
4
5
6
7
8
9
10
11
12
13#include "qemu/osdep.h"
14#include <glib/gstdio.h>
15
16#include "hw/tpm/tpm_ioctl.h"
17#include "io/channel-socket.h"
18#include "qapi/error.h"
19#include "tpm-emu.h"
20
21void tpm_emu_test_wait_cond(TestState *s)
22{
23 gint64 end_time = g_get_monotonic_time() + 5 * G_TIME_SPAN_SECOND;
24
25 g_mutex_lock(&s->data_mutex);
26
27 if (!s->data_cond_signal &&
28 !g_cond_wait_until(&s->data_cond, &s->data_mutex, end_time)) {
29 g_assert_not_reached();
30 }
31
32 s->data_cond_signal = false;
33
34 g_mutex_unlock(&s->data_mutex);
35}
36
37static void *tpm_emu_tpm_thread(void *data)
38{
39 TestState *s = data;
40 QIOChannel *ioc = s->tpm_ioc;
41
42 s->tpm_msg = g_new(struct tpm_hdr, 1);
43 while (true) {
44 int minhlen = sizeof(s->tpm_msg->tag) + sizeof(s->tpm_msg->len);
45
46 if (!qio_channel_read(ioc, (char *)s->tpm_msg, minhlen, &error_abort)) {
47 break;
48 }
49 s->tpm_msg->tag = be16_to_cpu(s->tpm_msg->tag);
50 s->tpm_msg->len = be32_to_cpu(s->tpm_msg->len);
51 g_assert_cmpint(s->tpm_msg->len, >=, minhlen);
52 g_assert_cmpint(s->tpm_msg->tag, ==, TPM2_ST_NO_SESSIONS);
53
54 s->tpm_msg = g_realloc(s->tpm_msg, s->tpm_msg->len);
55 qio_channel_read(ioc, (char *)&s->tpm_msg->code,
56 s->tpm_msg->len - minhlen, &error_abort);
57 s->tpm_msg->code = be32_to_cpu(s->tpm_msg->code);
58
59
60 s->tpm_msg->tag = cpu_to_be16(TPM2_ST_NO_SESSIONS);
61 s->tpm_msg->len = cpu_to_be32(sizeof(struct tpm_hdr));
62 s->tpm_msg->code = cpu_to_be32(TPM_RC_FAILURE);
63 qio_channel_write(ioc, (char *)s->tpm_msg, be32_to_cpu(s->tpm_msg->len),
64 &error_abort);
65 }
66
67 g_free(s->tpm_msg);
68 s->tpm_msg = NULL;
69 object_unref(OBJECT(s->tpm_ioc));
70 return NULL;
71}
72
73void *tpm_emu_ctrl_thread(void *data)
74{
75 TestState *s = data;
76 QIOChannelSocket *lioc = qio_channel_socket_new();
77 QIOChannel *ioc;
78
79 qio_channel_socket_listen_sync(lioc, s->addr, &error_abort);
80
81 g_mutex_lock(&s->data_mutex);
82 s->data_cond_signal = true;
83 g_mutex_unlock(&s->data_mutex);
84 g_cond_signal(&s->data_cond);
85
86 qio_channel_wait(QIO_CHANNEL(lioc), G_IO_IN);
87 ioc = QIO_CHANNEL(qio_channel_socket_accept(lioc, &error_abort));
88 g_assert(ioc);
89
90 {
91 uint32_t cmd = 0;
92 struct iovec iov = { .iov_base = &cmd, .iov_len = sizeof(cmd) };
93 int *pfd = NULL;
94 size_t nfd = 0;
95
96 qio_channel_readv_full(ioc, &iov, 1, &pfd, &nfd, &error_abort);
97 cmd = be32_to_cpu(cmd);
98 g_assert_cmpint(cmd, ==, CMD_SET_DATAFD);
99 g_assert_cmpint(nfd, ==, 1);
100 s->tpm_ioc = QIO_CHANNEL(qio_channel_socket_new_fd(*pfd, &error_abort));
101 g_free(pfd);
102
103 cmd = 0;
104 qio_channel_write(ioc, (char *)&cmd, sizeof(cmd), &error_abort);
105
106 s->emu_tpm_thread = g_thread_new(NULL, tpm_emu_tpm_thread, s);
107 }
108
109 while (true) {
110 uint32_t cmd;
111 ssize_t ret;
112
113 ret = qio_channel_read(ioc, (char *)&cmd, sizeof(cmd), NULL);
114 if (ret <= 0) {
115 break;
116 }
117
118 cmd = be32_to_cpu(cmd);
119 switch (cmd) {
120 case CMD_GET_CAPABILITY: {
121 ptm_cap cap = cpu_to_be64(0x3fff);
122 qio_channel_write(ioc, (char *)&cap, sizeof(cap), &error_abort);
123 break;
124 }
125 case CMD_INIT: {
126 ptm_init init;
127 qio_channel_read(ioc, (char *)&init.u.req, sizeof(init.u.req),
128 &error_abort);
129 init.u.resp.tpm_result = 0;
130 qio_channel_write(ioc, (char *)&init.u.resp, sizeof(init.u.resp),
131 &error_abort);
132 break;
133 }
134 case CMD_SHUTDOWN: {
135 ptm_res res = 0;
136 qio_channel_write(ioc, (char *)&res, sizeof(res), &error_abort);
137
138 g_thread_join(s->emu_tpm_thread);
139 break;
140 }
141 case CMD_STOP: {
142 ptm_res res = 0;
143 qio_channel_write(ioc, (char *)&res, sizeof(res), &error_abort);
144 break;
145 }
146 case CMD_SET_BUFFERSIZE: {
147 ptm_setbuffersize sbs;
148 qio_channel_read(ioc, (char *)&sbs.u.req, sizeof(sbs.u.req),
149 &error_abort);
150 sbs.u.resp.buffersize = sbs.u.req.buffersize ?: cpu_to_be32(4096);
151 sbs.u.resp.tpm_result = 0;
152 sbs.u.resp.minsize = cpu_to_be32(128);
153 sbs.u.resp.maxsize = cpu_to_be32(4096);
154 qio_channel_write(ioc, (char *)&sbs.u.resp, sizeof(sbs.u.resp),
155 &error_abort);
156 break;
157 }
158 case CMD_SET_LOCALITY: {
159 ptm_loc loc;
160
161 qio_channel_read(ioc, (char *)&loc, sizeof(loc), &error_abort);
162 g_assert_cmpint(loc.u.req.loc, ==, 0);
163 loc.u.resp.tpm_result = 0;
164 qio_channel_write(ioc, (char *)&loc, sizeof(loc), &error_abort);
165 break;
166 }
167 case CMD_GET_TPMESTABLISHED: {
168 ptm_est est = {
169 .u.resp.bit = 0,
170 };
171 qio_channel_write(ioc, (char *)&est, sizeof(est), &error_abort);
172 break;
173 }
174 default:
175 g_debug("unimplemented %u", cmd);
176 g_assert_not_reached();
177 }
178 }
179
180 object_unref(OBJECT(ioc));
181 object_unref(OBJECT(lioc));
182 return NULL;
183}
184
