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