Jeremy Kerr | 3d36ee2 | 2019-05-30 11:15:37 +0800 | [diff] [blame] | 1 | /* SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later */ |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 2 | |
| 3 | #include <assert.h> |
| 4 | #include <stdarg.h> |
| 5 | #include <stddef.h> |
| 6 | #include <stdint.h> |
| 7 | #include <stdio.h> |
| 8 | #include <stdlib.h> |
| 9 | #include <string.h> |
| 10 | |
| 11 | #undef pr_fmt |
| 12 | #define pr_fmt(fmt) "core: " fmt |
| 13 | |
| 14 | #include "libmctp.h" |
| 15 | #include "libmctp-alloc.h" |
| 16 | #include "libmctp-log.h" |
| 17 | |
| 18 | /* Internal data structures */ |
| 19 | |
| 20 | struct mctp_bus { |
| 21 | mctp_eid_t eid; |
| 22 | struct mctp_binding *binding; |
Jeremy Kerr | 1cd3118 | 2019-02-27 18:01:00 +0800 | [diff] [blame] | 23 | bool tx_enabled; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 24 | |
Jeremy Kerr | cc2458d | 2019-03-01 08:23:33 +0800 | [diff] [blame] | 25 | struct mctp_pktbuf *tx_queue_head; |
| 26 | struct mctp_pktbuf *tx_queue_tail; |
| 27 | |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 28 | /* todo: routing */ |
| 29 | }; |
| 30 | |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 31 | struct mctp_msg_ctx { |
| 32 | uint8_t src; |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 33 | uint8_t dest; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 34 | uint8_t tag; |
| 35 | uint8_t last_seq; |
| 36 | void *buf; |
| 37 | size_t buf_size; |
| 38 | size_t buf_alloc_size; |
| 39 | }; |
| 40 | |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 41 | struct mctp { |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 42 | int n_busses; |
| 43 | struct mctp_bus *busses; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 44 | |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 45 | /* Message RX callback */ |
| 46 | mctp_rx_fn message_rx; |
| 47 | void *message_rx_data; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 48 | |
| 49 | /* Message reassembly. |
| 50 | * @todo: flexible context count |
| 51 | */ |
| 52 | struct mctp_msg_ctx msg_ctxs[16]; |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 53 | |
| 54 | enum { |
| 55 | ROUTE_ENDPOINT, |
| 56 | ROUTE_BRIDGE, |
| 57 | } route_policy; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 58 | }; |
| 59 | |
| 60 | #ifndef BUILD_ASSERT |
| 61 | #define BUILD_ASSERT(x) \ |
| 62 | do { (void)sizeof(char[0-(!(x))]); } while (0) |
| 63 | #endif |
| 64 | |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 65 | #ifndef ARRAY_SIZE |
| 66 | #define ARRAY_SIZE(a) (sizeof(a) / sizeof(a[0])) |
| 67 | #endif |
| 68 | |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 69 | static int mctp_message_tx_on_bus(struct mctp *mctp, struct mctp_bus *bus, |
| 70 | mctp_eid_t src, mctp_eid_t dest, void *msg, size_t msg_len); |
| 71 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 72 | struct mctp_pktbuf *mctp_pktbuf_alloc(struct mctp_binding *binding, size_t len) |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 73 | { |
| 74 | struct mctp_pktbuf *buf; |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 75 | size_t size; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 76 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 77 | size = binding->pkt_size + binding->pkt_pad; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 78 | |
| 79 | /* todo: pools */ |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 80 | buf = __mctp_alloc(sizeof(*buf) + size); |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 81 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 82 | buf->size = size; |
| 83 | buf->start = binding->pkt_pad; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 84 | buf->end = buf->start + len; |
| 85 | buf->mctp_hdr_off = buf->start; |
Jeremy Kerr | dd109f1 | 2019-04-04 11:46:49 +0800 | [diff] [blame] | 86 | buf->next = NULL; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 87 | |
| 88 | return buf; |
| 89 | } |
| 90 | |
| 91 | void mctp_pktbuf_free(struct mctp_pktbuf *pkt) |
| 92 | { |
| 93 | __mctp_free(pkt); |
| 94 | } |
| 95 | |
| 96 | struct mctp_hdr *mctp_pktbuf_hdr(struct mctp_pktbuf *pkt) |
| 97 | { |
| 98 | return (void *)pkt->data + pkt->mctp_hdr_off; |
| 99 | } |
| 100 | |
| 101 | void *mctp_pktbuf_data(struct mctp_pktbuf *pkt) |
| 102 | { |
| 103 | return (void *)pkt->data + pkt->mctp_hdr_off + sizeof(struct mctp_hdr); |
| 104 | } |
| 105 | |
| 106 | uint8_t mctp_pktbuf_size(struct mctp_pktbuf *pkt) |
| 107 | { |
| 108 | return pkt->end - pkt->start; |
| 109 | } |
| 110 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 111 | void *mctp_pktbuf_alloc_start(struct mctp_pktbuf *pkt, size_t size) |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 112 | { |
| 113 | assert(size <= pkt->start); |
| 114 | pkt->start -= size; |
| 115 | return pkt->data + pkt->start; |
| 116 | } |
| 117 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 118 | void *mctp_pktbuf_alloc_end(struct mctp_pktbuf *pkt, size_t size) |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 119 | { |
| 120 | void *buf; |
| 121 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 122 | assert(size < (pkt->size - pkt->end)); |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 123 | buf = pkt->data + pkt->end; |
| 124 | pkt->end += size; |
| 125 | return buf; |
| 126 | } |
| 127 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 128 | int mctp_pktbuf_push(struct mctp_pktbuf *pkt, void *data, size_t len) |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 129 | { |
| 130 | void *p; |
| 131 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 132 | if (pkt->end + len > pkt->size) |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 133 | return -1; |
| 134 | |
| 135 | p = pkt->data + pkt->end; |
| 136 | |
| 137 | pkt->end += len; |
| 138 | memcpy(p, data, len); |
| 139 | |
| 140 | return 0; |
| 141 | } |
| 142 | |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 143 | /* Message reassembly */ |
| 144 | static struct mctp_msg_ctx *mctp_msg_ctx_lookup(struct mctp *mctp, |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 145 | uint8_t src, uint8_t dest, uint8_t tag) |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 146 | { |
| 147 | unsigned int i; |
| 148 | |
| 149 | /* @todo: better lookup, if we add support for more outstanding |
| 150 | * message contexts */ |
| 151 | for (i = 0; i < ARRAY_SIZE(mctp->msg_ctxs); i++) { |
| 152 | struct mctp_msg_ctx *ctx = &mctp->msg_ctxs[i]; |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 153 | if (ctx->src == src && ctx->dest == dest && ctx->tag == tag) |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 154 | return ctx; |
| 155 | } |
| 156 | |
| 157 | return NULL; |
| 158 | } |
| 159 | |
| 160 | static struct mctp_msg_ctx *mctp_msg_ctx_create(struct mctp *mctp, |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 161 | uint8_t src, uint8_t dest, uint8_t tag) |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 162 | { |
Jeremy Kerr | 11a234e | 2019-02-27 17:59:53 +0800 | [diff] [blame] | 163 | struct mctp_msg_ctx *ctx = NULL; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 164 | unsigned int i; |
| 165 | |
| 166 | for (i = 0; i < ARRAY_SIZE(mctp->msg_ctxs); i++) { |
| 167 | struct mctp_msg_ctx *tmp = &mctp->msg_ctxs[i]; |
| 168 | if (!tmp->src) { |
| 169 | ctx = tmp; |
| 170 | break; |
| 171 | } |
| 172 | } |
| 173 | |
| 174 | if (!ctx) |
| 175 | return NULL; |
| 176 | |
| 177 | ctx->src = src; |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 178 | ctx->dest = dest; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 179 | ctx->tag = tag; |
Jeremy Kerr | 9a3da81 | 2019-08-02 15:57:53 +0800 | [diff] [blame] | 180 | ctx->buf_size = 0; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 181 | |
| 182 | return ctx; |
| 183 | } |
| 184 | |
| 185 | static void mctp_msg_ctx_drop(struct mctp_msg_ctx *ctx) |
| 186 | { |
| 187 | ctx->src = 0; |
| 188 | } |
| 189 | |
| 190 | static void mctp_msg_ctx_reset(struct mctp_msg_ctx *ctx) |
| 191 | { |
| 192 | ctx->buf_size = 0; |
| 193 | } |
| 194 | |
| 195 | static int mctp_msg_ctx_add_pkt(struct mctp_msg_ctx *ctx, |
| 196 | struct mctp_pktbuf *pkt) |
| 197 | { |
| 198 | size_t len; |
| 199 | |
| 200 | len = mctp_pktbuf_size(pkt) - sizeof(struct mctp_hdr); |
| 201 | |
| 202 | if (ctx->buf_size + len > ctx->buf_alloc_size) { |
| 203 | size_t new_alloc_size; |
Andrew Jeffery | 00ecc6c | 2020-03-10 23:16:53 +1030 | [diff] [blame] | 204 | void *lbuf; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 205 | |
| 206 | /* @todo: finer-grained allocation, size limits */ |
| 207 | if (!ctx->buf_alloc_size) { |
| 208 | new_alloc_size = 4096; |
| 209 | } else { |
| 210 | new_alloc_size = ctx->buf_alloc_size * 2; |
| 211 | } |
Andrew Jeffery | 00ecc6c | 2020-03-10 23:16:53 +1030 | [diff] [blame] | 212 | |
| 213 | lbuf = __mctp_realloc(ctx->buf, new_alloc_size); |
| 214 | if (lbuf) { |
| 215 | ctx->buf = lbuf; |
| 216 | ctx->buf_alloc_size = new_alloc_size; |
| 217 | } else { |
| 218 | __mctp_free(ctx->buf); |
| 219 | return -1; |
| 220 | } |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 221 | } |
| 222 | |
| 223 | memcpy(ctx->buf + ctx->buf_size, mctp_pktbuf_data(pkt), len); |
| 224 | ctx->buf_size += len; |
| 225 | |
| 226 | return 0; |
| 227 | } |
| 228 | |
| 229 | /* Core API functions */ |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 230 | struct mctp *mctp_init(void) |
| 231 | { |
| 232 | struct mctp *mctp; |
| 233 | |
| 234 | mctp = __mctp_alloc(sizeof(*mctp)); |
| 235 | memset(mctp, 0, sizeof(*mctp)); |
| 236 | |
| 237 | return mctp; |
| 238 | } |
| 239 | |
Andrew Jeffery | fa56ca5 | 2020-03-10 23:18:22 +1030 | [diff] [blame] | 240 | void mctp_destroy(struct mctp *mctp) |
| 241 | { |
| 242 | int i; |
| 243 | |
| 244 | /* Cleanup message assembly contexts */ |
| 245 | for (i = 0; i < ARRAY_SIZE(mctp->msg_ctxs); i++) { |
| 246 | struct mctp_msg_ctx *tmp = &mctp->msg_ctxs[i]; |
| 247 | if (tmp->buf) |
| 248 | __mctp_free(tmp->buf); |
| 249 | } |
| 250 | |
| 251 | __mctp_free(mctp->busses); |
| 252 | __mctp_free(mctp); |
| 253 | } |
| 254 | |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 255 | int mctp_set_rx_all(struct mctp *mctp, mctp_rx_fn fn, void *data) |
| 256 | { |
| 257 | mctp->message_rx = fn; |
| 258 | mctp->message_rx_data = data; |
| 259 | return 0; |
| 260 | } |
| 261 | |
| 262 | static struct mctp_bus *find_bus_for_eid(struct mctp *mctp, |
| 263 | mctp_eid_t dest __attribute__((unused))) |
| 264 | { |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 265 | /* for now, just use the first bus. For full routing support, |
| 266 | * we will need a table of neighbours */ |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 267 | return &mctp->busses[0]; |
| 268 | } |
| 269 | |
Jeremy Kerr | 7520cec | 2019-03-01 07:13:18 +0800 | [diff] [blame] | 270 | int mctp_register_bus(struct mctp *mctp, |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 271 | struct mctp_binding *binding, |
| 272 | mctp_eid_t eid) |
| 273 | { |
Jeremy Kerr | 7520cec | 2019-03-01 07:13:18 +0800 | [diff] [blame] | 274 | /* todo: multiple busses */ |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 275 | assert(mctp->n_busses == 0); |
| 276 | mctp->n_busses = 1; |
| 277 | mctp->busses = __mctp_alloc(sizeof(struct mctp_bus)); |
James Feist | 62d7236 | 2019-12-13 13:43:32 -0800 | [diff] [blame] | 278 | memset(mctp->busses, 0, sizeof(struct mctp_bus)); |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 279 | mctp->busses[0].binding = binding; |
| 280 | mctp->busses[0].eid = eid; |
Jeremy Kerr | 7520cec | 2019-03-01 07:13:18 +0800 | [diff] [blame] | 281 | binding->bus = &mctp->busses[0]; |
Jeremy Kerr | 0a00dca | 2019-03-01 08:01:35 +0800 | [diff] [blame] | 282 | binding->mctp = mctp; |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 283 | mctp->route_policy = ROUTE_ENDPOINT; |
Jeremy Kerr | 3b36d17 | 2019-09-04 11:56:09 +0800 | [diff] [blame] | 284 | |
| 285 | if (binding->start) |
| 286 | binding->start(binding); |
| 287 | |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 288 | return 0; |
| 289 | } |
| 290 | |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 291 | int mctp_bridge_busses(struct mctp *mctp, |
| 292 | struct mctp_binding *b1, struct mctp_binding *b2) |
| 293 | { |
| 294 | assert(mctp->n_busses == 0); |
| 295 | mctp->busses = __mctp_alloc(2 * sizeof(struct mctp_bus)); |
James Feist | 62d7236 | 2019-12-13 13:43:32 -0800 | [diff] [blame] | 296 | memset(mctp->busses, 0, 2 * sizeof(struct mctp_bus)); |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 297 | mctp->n_busses = 2; |
| 298 | mctp->busses[0].binding = b1; |
| 299 | b1->bus = &mctp->busses[0]; |
| 300 | b1->mctp = mctp; |
| 301 | mctp->busses[1].binding = b2; |
| 302 | b2->bus = &mctp->busses[1]; |
| 303 | b2->mctp = mctp; |
| 304 | |
| 305 | mctp->route_policy = ROUTE_BRIDGE; |
Jeremy Kerr | 3b36d17 | 2019-09-04 11:56:09 +0800 | [diff] [blame] | 306 | |
| 307 | if (b1->start) |
| 308 | b1->start(b1); |
| 309 | |
| 310 | if (b2->start) |
| 311 | b2->start(b2); |
| 312 | |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 313 | return 0; |
| 314 | } |
| 315 | |
| 316 | static void mctp_rx(struct mctp *mctp, struct mctp_bus *bus, |
| 317 | mctp_eid_t src, mctp_eid_t dest, void *buf, size_t len) |
| 318 | { |
| 319 | if (mctp->route_policy == ROUTE_ENDPOINT && |
| 320 | dest == bus->eid && mctp->message_rx) |
| 321 | mctp->message_rx(src, mctp->message_rx_data, buf, len); |
| 322 | |
| 323 | if (mctp->route_policy == ROUTE_BRIDGE) { |
| 324 | int i; |
| 325 | |
| 326 | for (i = 0; i < mctp->n_busses; i++) { |
| 327 | struct mctp_bus *dest_bus = &mctp->busses[i]; |
| 328 | if (dest_bus == bus) |
| 329 | continue; |
| 330 | |
| 331 | mctp_message_tx_on_bus(mctp, dest_bus, |
| 332 | src, dest, buf, len); |
| 333 | } |
| 334 | |
| 335 | } |
| 336 | } |
| 337 | |
Jeremy Kerr | 0a00dca | 2019-03-01 08:01:35 +0800 | [diff] [blame] | 338 | void mctp_bus_rx(struct mctp_binding *binding, struct mctp_pktbuf *pkt) |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 339 | { |
Jeremy Kerr | 7520cec | 2019-03-01 07:13:18 +0800 | [diff] [blame] | 340 | struct mctp_bus *bus = binding->bus; |
Jeremy Kerr | 0a00dca | 2019-03-01 08:01:35 +0800 | [diff] [blame] | 341 | struct mctp *mctp = binding->mctp; |
Ed Tanous | c2def9f | 2019-02-21 08:33:08 -0800 | [diff] [blame] | 342 | uint8_t flags, exp_seq, seq, tag; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 343 | struct mctp_msg_ctx *ctx; |
| 344 | struct mctp_hdr *hdr; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 345 | size_t len; |
| 346 | void *p; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 347 | int rc; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 348 | |
Jeremy Kerr | 7520cec | 2019-03-01 07:13:18 +0800 | [diff] [blame] | 349 | assert(bus); |
| 350 | |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 351 | hdr = mctp_pktbuf_hdr(pkt); |
| 352 | |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 353 | /* small optimisation: don't bother reassembly if we're going to |
| 354 | * drop the packet in mctp_rx anyway */ |
| 355 | if (mctp->route_policy == ROUTE_ENDPOINT && hdr->dest != bus->eid) |
Jeremy Kerr | c1693af | 2019-08-05 14:30:59 +0800 | [diff] [blame] | 356 | goto out; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 357 | |
| 358 | flags = hdr->flags_seq_tag & (MCTP_HDR_FLAG_SOM | MCTP_HDR_FLAG_EOM); |
| 359 | tag = (hdr->flags_seq_tag >> MCTP_HDR_TAG_SHIFT) & MCTP_HDR_TAG_MASK; |
| 360 | seq = (hdr->flags_seq_tag >> MCTP_HDR_SEQ_SHIFT) & MCTP_HDR_SEQ_MASK; |
| 361 | |
| 362 | switch (flags) { |
| 363 | case MCTP_HDR_FLAG_SOM | MCTP_HDR_FLAG_EOM: |
| 364 | /* single-packet message - send straight up to rx function, |
| 365 | * no need to create a message context */ |
| 366 | len = pkt->end - pkt->mctp_hdr_off - sizeof(struct mctp_hdr); |
| 367 | p = pkt->data + pkt->mctp_hdr_off + sizeof(struct mctp_hdr), |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 368 | mctp_rx(mctp, bus, hdr->src, hdr->dest, p, len); |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 369 | break; |
| 370 | |
| 371 | case MCTP_HDR_FLAG_SOM: |
| 372 | /* start of a new message - start the new context for |
| 373 | * future message reception. If an existing context is |
| 374 | * already present, drop it. */ |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 375 | ctx = mctp_msg_ctx_lookup(mctp, hdr->src, hdr->dest, tag); |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 376 | if (ctx) { |
| 377 | mctp_msg_ctx_reset(ctx); |
| 378 | } else { |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 379 | ctx = mctp_msg_ctx_create(mctp, |
| 380 | hdr->src, hdr->dest, tag); |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 381 | } |
| 382 | |
| 383 | rc = mctp_msg_ctx_add_pkt(ctx, pkt); |
| 384 | if (rc) { |
| 385 | mctp_msg_ctx_drop(ctx); |
| 386 | } else { |
| 387 | ctx->last_seq = seq; |
| 388 | } |
| 389 | |
| 390 | break; |
| 391 | |
| 392 | case MCTP_HDR_FLAG_EOM: |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 393 | ctx = mctp_msg_ctx_lookup(mctp, hdr->src, hdr->dest, tag); |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 394 | if (!ctx) |
Jeremy Kerr | c1693af | 2019-08-05 14:30:59 +0800 | [diff] [blame] | 395 | goto out; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 396 | |
Ed Tanous | c2def9f | 2019-02-21 08:33:08 -0800 | [diff] [blame] | 397 | exp_seq = (ctx->last_seq + 1) % 4; |
| 398 | |
| 399 | if (exp_seq != seq) { |
| 400 | mctp_prdebug( |
| 401 | "Sequence number %d does not match expected %d", |
| 402 | seq, exp_seq); |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 403 | mctp_msg_ctx_drop(ctx); |
Jeremy Kerr | c1693af | 2019-08-05 14:30:59 +0800 | [diff] [blame] | 404 | goto out; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 405 | } |
| 406 | |
| 407 | rc = mctp_msg_ctx_add_pkt(ctx, pkt); |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 408 | if (!rc) |
| 409 | mctp_rx(mctp, bus, ctx->src, ctx->dest, |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 410 | ctx->buf, ctx->buf_size); |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 411 | |
| 412 | mctp_msg_ctx_drop(ctx); |
| 413 | break; |
Ed Tanous | c2def9f | 2019-02-21 08:33:08 -0800 | [diff] [blame] | 414 | |
| 415 | case 0: |
| 416 | /* Neither SOM nor EOM */ |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 417 | ctx = mctp_msg_ctx_lookup(mctp, hdr->src,hdr->dest, tag); |
Ed Tanous | c2def9f | 2019-02-21 08:33:08 -0800 | [diff] [blame] | 418 | if (!ctx) |
Jeremy Kerr | c1693af | 2019-08-05 14:30:59 +0800 | [diff] [blame] | 419 | goto out; |
Ed Tanous | c2def9f | 2019-02-21 08:33:08 -0800 | [diff] [blame] | 420 | |
| 421 | exp_seq = (ctx->last_seq + 1) % 4; |
| 422 | if (exp_seq != seq) { |
| 423 | mctp_prdebug( |
| 424 | "Sequence number %d does not match expected %d", |
| 425 | seq, exp_seq); |
| 426 | mctp_msg_ctx_drop(ctx); |
Jeremy Kerr | c1693af | 2019-08-05 14:30:59 +0800 | [diff] [blame] | 427 | goto out; |
Ed Tanous | c2def9f | 2019-02-21 08:33:08 -0800 | [diff] [blame] | 428 | } |
| 429 | |
| 430 | rc = mctp_msg_ctx_add_pkt(ctx, pkt); |
| 431 | if (rc) { |
| 432 | mctp_msg_ctx_drop(ctx); |
Jeremy Kerr | c1693af | 2019-08-05 14:30:59 +0800 | [diff] [blame] | 433 | goto out; |
Ed Tanous | c2def9f | 2019-02-21 08:33:08 -0800 | [diff] [blame] | 434 | } |
| 435 | ctx->last_seq = seq; |
| 436 | |
| 437 | break; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 438 | } |
Jeremy Kerr | c1693af | 2019-08-05 14:30:59 +0800 | [diff] [blame] | 439 | out: |
| 440 | mctp_pktbuf_free(pkt); |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 441 | } |
| 442 | |
Jeremy Kerr | 0a00dca | 2019-03-01 08:01:35 +0800 | [diff] [blame] | 443 | static int mctp_packet_tx(struct mctp_bus *bus, |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 444 | struct mctp_pktbuf *pkt) |
| 445 | { |
Jeremy Kerr | 1cd3118 | 2019-02-27 18:01:00 +0800 | [diff] [blame] | 446 | if (!bus->tx_enabled) |
| 447 | return -1; |
| 448 | |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 449 | return bus->binding->tx(bus->binding, pkt); |
| 450 | } |
| 451 | |
Jeremy Kerr | cc2458d | 2019-03-01 08:23:33 +0800 | [diff] [blame] | 452 | static void mctp_send_tx_queue(struct mctp_bus *bus) |
Jeremy Kerr | 1cd3118 | 2019-02-27 18:01:00 +0800 | [diff] [blame] | 453 | { |
| 454 | struct mctp_pktbuf *pkt; |
| 455 | |
Jeremy Kerr | cc2458d | 2019-03-01 08:23:33 +0800 | [diff] [blame] | 456 | while ((pkt = bus->tx_queue_head)) { |
Jeremy Kerr | 1cd3118 | 2019-02-27 18:01:00 +0800 | [diff] [blame] | 457 | int rc; |
| 458 | |
| 459 | rc = mctp_packet_tx(bus, pkt); |
| 460 | if (rc) |
| 461 | break; |
| 462 | |
Jeremy Kerr | cc2458d | 2019-03-01 08:23:33 +0800 | [diff] [blame] | 463 | bus->tx_queue_head = pkt->next; |
Jeremy Kerr | 1cd3118 | 2019-02-27 18:01:00 +0800 | [diff] [blame] | 464 | mctp_pktbuf_free(pkt); |
| 465 | } |
| 466 | |
Jeremy Kerr | cc2458d | 2019-03-01 08:23:33 +0800 | [diff] [blame] | 467 | if (!bus->tx_queue_head) |
| 468 | bus->tx_queue_tail = NULL; |
Jeremy Kerr | 1cd3118 | 2019-02-27 18:01:00 +0800 | [diff] [blame] | 469 | |
| 470 | } |
| 471 | |
| 472 | void mctp_binding_set_tx_enabled(struct mctp_binding *binding, bool enable) |
| 473 | { |
| 474 | struct mctp_bus *bus = binding->bus; |
| 475 | bus->tx_enabled = enable; |
| 476 | if (enable) |
Jeremy Kerr | cc2458d | 2019-03-01 08:23:33 +0800 | [diff] [blame] | 477 | mctp_send_tx_queue(bus); |
Jeremy Kerr | 1cd3118 | 2019-02-27 18:01:00 +0800 | [diff] [blame] | 478 | } |
| 479 | |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 480 | static int mctp_message_tx_on_bus(struct mctp *mctp, struct mctp_bus *bus, |
| 481 | mctp_eid_t src, mctp_eid_t dest, void *msg, size_t msg_len) |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 482 | { |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 483 | size_t max_payload_len, payload_len, p; |
Jeremy Kerr | 1cd3118 | 2019-02-27 18:01:00 +0800 | [diff] [blame] | 484 | struct mctp_pktbuf *pkt; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 485 | struct mctp_hdr *hdr; |
Jeremy Kerr | c855d7b | 2019-08-01 21:18:09 +0800 | [diff] [blame] | 486 | int i; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 487 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 488 | max_payload_len = bus->binding->pkt_size - sizeof(*hdr); |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 489 | |
Andrew Jeffery | 298865f | 2020-02-06 11:51:29 +1030 | [diff] [blame] | 490 | mctp_prdebug("%s: Generating packets for transmission of %zu byte message from %hhu to %hhu", |
| 491 | __func__, msg_len, src, dest); |
| 492 | |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 493 | /* queue up packets, each of max MCTP_MTU size */ |
Jeremy Kerr | c855d7b | 2019-08-01 21:18:09 +0800 | [diff] [blame] | 494 | for (p = 0, i = 0; p < msg_len; i++) { |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 495 | payload_len = msg_len - p; |
| 496 | if (payload_len > max_payload_len) |
| 497 | payload_len = max_payload_len; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 498 | |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 499 | pkt = mctp_pktbuf_alloc(bus->binding, |
| 500 | payload_len + sizeof(*hdr)); |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 501 | hdr = mctp_pktbuf_hdr(pkt); |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 502 | |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 503 | /* todo: tags */ |
| 504 | hdr->ver = bus->binding->version & 0xf; |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 505 | hdr->dest = dest; |
| 506 | hdr->src = src; |
Jeremy Kerr | c855d7b | 2019-08-01 21:18:09 +0800 | [diff] [blame] | 507 | hdr->flags_seq_tag = MCTP_HDR_FLAG_TO | |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 508 | (0 << MCTP_HDR_TAG_SHIFT); |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 509 | |
Jeremy Kerr | c855d7b | 2019-08-01 21:18:09 +0800 | [diff] [blame] | 510 | if (i == 0) |
| 511 | hdr->flags_seq_tag |= MCTP_HDR_FLAG_SOM; |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 512 | if (p + payload_len >= msg_len) |
Jeremy Kerr | c855d7b | 2019-08-01 21:18:09 +0800 | [diff] [blame] | 513 | hdr->flags_seq_tag |= MCTP_HDR_FLAG_EOM; |
| 514 | hdr->flags_seq_tag |= |
| 515 | (i & MCTP_HDR_SEQ_MASK) << MCTP_HDR_SEQ_SHIFT; |
| 516 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 517 | memcpy(mctp_pktbuf_data(pkt), msg + p, payload_len); |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 518 | |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 519 | /* add to tx queue */ |
Jeremy Kerr | cc2458d | 2019-03-01 08:23:33 +0800 | [diff] [blame] | 520 | if (bus->tx_queue_tail) |
| 521 | bus->tx_queue_tail->next = pkt; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 522 | else |
Jeremy Kerr | cc2458d | 2019-03-01 08:23:33 +0800 | [diff] [blame] | 523 | bus->tx_queue_head = pkt; |
| 524 | bus->tx_queue_tail = pkt; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 525 | |
Jeremy Kerr | df15f7e | 2019-08-05 15:41:19 +0800 | [diff] [blame] | 526 | p += payload_len; |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 527 | } |
| 528 | |
Andrew Jeffery | 298865f | 2020-02-06 11:51:29 +1030 | [diff] [blame] | 529 | mctp_prdebug("%s: Enqueued %d packets", __func__, i); |
| 530 | |
Jeremy Kerr | cc2458d | 2019-03-01 08:23:33 +0800 | [diff] [blame] | 531 | mctp_send_tx_queue(bus); |
Jeremy Kerr | 24db71f | 2019-02-07 21:37:35 +0800 | [diff] [blame] | 532 | |
| 533 | return 0; |
Jeremy Kerr | 4cdc200 | 2019-02-07 16:49:12 +0800 | [diff] [blame] | 534 | } |
Jeremy Kerr | 1a4ec3c | 2019-09-03 11:01:50 +0800 | [diff] [blame] | 535 | |
| 536 | int mctp_message_tx(struct mctp *mctp, mctp_eid_t eid, |
| 537 | void *msg, size_t msg_len) |
| 538 | { |
| 539 | struct mctp_bus *bus; |
| 540 | |
| 541 | bus = find_bus_for_eid(mctp, eid); |
| 542 | return mctp_message_tx_on_bus(mctp, bus, bus->eid, eid, msg, msg_len); |
| 543 | } |