net/virtio: fix incorrect cast of void *
[dpdk.git] / port / rte_swx_port_fd.c
1 /* SPDX-License-Identifier: BSD-3-Clause
2  * Copyright(c) 2021 Intel Corporation
3  */
4 #include <string.h>
5 #include <stdint.h>
6 #include <unistd.h>
7
8 #include <rte_mbuf.h>
9 #include <rte_malloc.h>
10 #include <rte_hexdump.h>
11
12 #include "rte_swx_port_fd.h"
13
14 #ifndef TRACE_LEVEL
15 #define TRACE_LEVEL 0
16 #endif
17
18 #if TRACE_LEVEL
19 #define TRACE(...) printf(__VA_ARGS__)
20 #else
21 #define TRACE(...)
22 #endif
23
24 /*
25  * FD Reader
26  */
27 struct reader {
28         struct {
29                 int fd;
30                 uint32_t mtu;
31                 uint32_t burst_size;
32                 struct rte_mempool *mempool;
33         } params;
34
35         struct rte_swx_port_in_stats stats;
36         struct rte_mbuf **pkts;
37         uint32_t n_pkts;
38         uint32_t pos;
39 };
40
41 static void *
42 reader_create(void *args)
43 {
44         struct rte_swx_port_fd_reader_params *conf = args;
45         struct reader *p;
46
47         /* Check input parameters. */
48         if (!conf || conf->fd < 0 || conf->mtu == 0 || !conf->mempool)
49                 return NULL;
50
51         /* Memory allocation. */
52         p = calloc(1, sizeof(struct reader));
53         if (!p)
54                 return NULL;
55
56         p->pkts = calloc(conf->burst_size, sizeof(struct rte_mbuf *));
57         if (!p->pkts) {
58                 free(p);
59                 return NULL;
60         }
61
62         /* Initialization. */
63         p->params.fd = conf->fd;
64         p->params.mtu = conf->mtu;
65         p->params.burst_size = conf->burst_size;
66         p->params.mempool = conf->mempool;
67
68         return p;
69 }
70
71 static void
72 reader_free(void *port)
73 {
74         struct reader *p = port;
75         uint32_t i;
76
77         if (!p)
78                 return;
79
80         for (i = 0; i < p->n_pkts; i++)
81                 rte_pktmbuf_free(p->pkts[i]);
82
83         free(p->pkts);
84         free(p);
85 }
86
87 static int
88 reader_pkt_rx(void *port, struct rte_swx_pkt *pkt)
89 {
90         struct reader *p = port;
91         struct rte_mbuf *m;
92         void *pkt_data;
93         ssize_t n_bytes;
94         uint32_t i, j;
95
96         if (p->n_pkts == p->pos) {
97                 if (rte_pktmbuf_alloc_bulk(p->params.mempool, p->pkts, p->params.burst_size) != 0)
98                         return 0;
99
100                 for (i = 0; i < p->params.burst_size; i++) {
101                         m = p->pkts[i];
102                         pkt_data = rte_pktmbuf_mtod(m, void *);
103                         n_bytes = read(p->params.fd, pkt_data, (size_t) p->params.mtu);
104
105                         if (n_bytes <= 0)
106                                 break;
107
108                         m->data_len = n_bytes;
109                         m->pkt_len = n_bytes;
110
111                         p->stats.n_pkts++;
112                         p->stats.n_bytes += n_bytes;
113                 }
114
115                 for (j = i; j < p->params.burst_size; j++)
116                         rte_pktmbuf_free(p->pkts[j]);
117
118                 p->n_pkts = i;
119                 p->pos = 0;
120
121                 if (!p->n_pkts)
122                         return 0;
123         }
124
125         m = p->pkts[p->pos++];
126         pkt->handle = m;
127         pkt->pkt = m->buf_addr;
128         pkt->offset = m->data_off;
129         pkt->length = m->pkt_len;
130
131         TRACE("[FD %u] Pkt %d (%u bytes at offset %u)\n",
132                 (uint32_t)p->params.fd,
133                 p->pos - 1,
134                 pkt->length,
135                 pkt->offset);
136
137         if (TRACE_LEVEL)
138                 rte_hexdump(stdout, NULL,
139                         &((uint8_t *)m->buf_addr)[m->data_off], m->data_len);
140
141         return 1;
142 }
143
144 static void
145 reader_stats_read(void *port, struct rte_swx_port_in_stats *stats)
146 {
147         struct reader *p = port;
148
149         memcpy(stats, &p->stats, sizeof(p->stats));
150 }
151
152 /*
153  * FD Writer
154  */
155 struct writer {
156         struct {
157                 int fd;
158                 uint32_t mtu;
159                 uint32_t burst_size;
160                 struct rte_mempool *mempool;
161         } params;
162
163         struct rte_swx_port_out_stats stats;
164         struct rte_mbuf **pkts;
165         uint32_t n_pkts;
166 };
167
168 static void *
169 writer_create(void *args)
170 {
171         struct rte_swx_port_fd_writer_params *conf = args;
172         struct writer *p;
173
174         /* Check input parameters. */
175         if (!conf)
176                 return NULL;
177
178         /* Memory allocation. */
179         p = calloc(1, sizeof(struct writer));
180         if (!p)
181                 return NULL;
182
183
184         p->pkts = calloc(conf->burst_size, sizeof(struct rte_mbuf *));
185         if (!p->pkts) {
186                 free(p);
187                 return NULL;
188         }
189
190         /* Initialization. */
191         p->params.fd = conf->fd;
192         p->params.burst_size = conf->burst_size;
193
194         return p;
195 }
196
197 static void
198 __writer_flush(struct writer *p)
199 {
200         struct rte_mbuf *pkt;
201         void *pkt_data;
202         size_t n_bytes;
203         ssize_t ret;
204         uint32_t i;
205
206         for (i = 0; i < p->n_pkts; i++) {
207                 pkt = p->pkts[i];
208                 pkt_data = rte_pktmbuf_mtod(pkt, void*);
209                 n_bytes = rte_pktmbuf_data_len(pkt);
210
211                 ret = write(p->params.fd, pkt_data, n_bytes);
212                 if (ret < 0)
213                         break;
214         }
215
216         TRACE("[FD %u] %u packets out\n",
217                 (uint32_t)p->params.fd,
218                 p->n_pkts);
219
220         for (i = 0; i < p->n_pkts; i++)
221                 rte_pktmbuf_free(p->pkts[i]);
222
223         p->n_pkts = 0;
224 }
225
226 static void
227 writer_pkt_tx(void *port, struct rte_swx_pkt *pkt)
228 {
229         struct writer *p = port;
230         struct rte_mbuf *m = pkt->handle;
231
232         TRACE("[FD %u] Pkt %u (%u bytes at offset %u)\n",
233                 (uint32_t)p->params.fd,
234                 p->n_pkts - 1,
235                 pkt->length,
236                 pkt->offset);
237
238         if (TRACE_LEVEL)
239                 rte_hexdump(stdout, NULL, &pkt->pkt[pkt->offset], pkt->length);
240
241         m->pkt_len = pkt->length;
242         m->data_len = (uint16_t)pkt->length;
243         m->data_off = (uint16_t)pkt->offset;
244
245         p->stats.n_pkts++;
246         p->stats.n_bytes += pkt->length;
247
248         p->pkts[p->n_pkts++] = m;
249         if (p->n_pkts == p->params.burst_size)
250                 __writer_flush(p);
251 }
252
253 static void
254 writer_flush(void *port)
255 {
256         struct writer *p = port;
257
258         if (p->n_pkts)
259                 __writer_flush(p);
260 }
261
262 static void
263 writer_free(void *port)
264 {
265         struct writer *p = port;
266
267         if (!p)
268                 return;
269
270         writer_flush(p);
271         free(p->pkts);
272         free(p);
273 }
274
275 static void
276 writer_stats_read(void *port, struct rte_swx_port_out_stats *stats)
277 {
278         struct writer *p = port;
279
280         memcpy(stats, &p->stats, sizeof(p->stats));
281 }
282
283 /*
284  * Summary of port operations
285  */
286 struct rte_swx_port_in_ops rte_swx_port_fd_reader_ops = {
287         .create = reader_create,
288         .free = reader_free,
289         .pkt_rx = reader_pkt_rx,
290         .stats_read = reader_stats_read,
291 };
292
293 struct rte_swx_port_out_ops rte_swx_port_fd_writer_ops = {
294         .create = writer_create,
295         .free = writer_free,
296         .pkt_tx = writer_pkt_tx,
297         .flush = writer_flush,
298         .stats_read = writer_stats_read,
299 };