]>
Commit | Line | Data |
---|---|---|
f7105843 MM |
1 | /* |
2 | * Copyright (c) 2003-2008 Fabrice Bellard | |
3 | * Copyright (c) 2009 Red Hat, Inc. | |
4 | * | |
5 | * Permission is hereby granted, free of charge, to any person obtaining a copy | |
6 | * of this software and associated documentation files (the "Software"), to deal | |
7 | * in the Software without restriction, including without limitation the rights | |
8 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
9 | * copies of the Software, and to permit persons to whom the Software is | |
10 | * furnished to do so, subject to the following conditions: | |
11 | * | |
12 | * The above copyright notice and this permission notice shall be included in | |
13 | * all copies or substantial portions of the Software. | |
14 | * | |
15 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
16 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
17 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL | |
18 | * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
19 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
20 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | |
21 | * THE SOFTWARE. | |
22 | */ | |
23 | ||
e1144d00 | 24 | #include "net/queue.h" |
1de7afc9 | 25 | #include "qemu/queue.h" |
1422e32d | 26 | #include "net/net.h" |
f7105843 MM |
27 | |
28 | /* The delivery handler may only return zero if it will call | |
29 | * qemu_net_queue_flush() when it determines that it is once again able | |
30 | * to deliver packets. It must also call qemu_net_queue_purge() in its | |
31 | * cleanup path. | |
32 | * | |
33 | * If a sent callback is provided to send(), the caller must handle a | |
34 | * zero return from the delivery handler by not sending any more packets | |
35 | * until we have invoked the callback. Only in that case will we queue | |
36 | * the packet. | |
37 | * | |
38 | * If a sent callback isn't provided, we just drop the packet to avoid | |
39 | * unbounded queueing. | |
40 | */ | |
41 | ||
42 | struct NetPacket { | |
43 | QTAILQ_ENTRY(NetPacket) entry; | |
4e68f7a0 | 44 | NetClientState *sender; |
c0b8e49c | 45 | unsigned flags; |
f7105843 MM |
46 | int size; |
47 | NetPacketSent *sent_cb; | |
48 | uint8_t data[0]; | |
49 | }; | |
50 | ||
51 | struct NetQueue { | |
f7105843 | 52 | void *opaque; |
7d91ddd2 LR |
53 | uint32_t nq_maxlen; |
54 | uint32_t nq_count; | |
3e033a46 | 55 | NetQueueDeliverFunc *deliver; |
f7105843 MM |
56 | |
57 | QTAILQ_HEAD(packets, NetPacket) packets; | |
58 | ||
59 | unsigned delivering : 1; | |
60 | }; | |
61 | ||
3e033a46 | 62 | NetQueue *qemu_new_net_queue(NetQueueDeliverFunc *deliver, void *opaque) |
f7105843 MM |
63 | { |
64 | NetQueue *queue; | |
65 | ||
58889fe5 | 66 | queue = g_new0(NetQueue, 1); |
f7105843 | 67 | |
f7105843 | 68 | queue->opaque = opaque; |
7d91ddd2 LR |
69 | queue->nq_maxlen = 10000; |
70 | queue->nq_count = 0; | |
3e033a46 | 71 | queue->deliver = deliver; |
f7105843 MM |
72 | |
73 | QTAILQ_INIT(&queue->packets); | |
74 | ||
75 | queue->delivering = 0; | |
76 | ||
77 | return queue; | |
78 | } | |
79 | ||
80 | void qemu_del_net_queue(NetQueue *queue) | |
81 | { | |
82 | NetPacket *packet, *next; | |
83 | ||
84 | QTAILQ_FOREACH_SAFE(packet, &queue->packets, entry, next) { | |
85 | QTAILQ_REMOVE(&queue->packets, packet, entry); | |
7267c094 | 86 | g_free(packet); |
f7105843 MM |
87 | } |
88 | ||
7267c094 | 89 | g_free(queue); |
f7105843 MM |
90 | } |
91 | ||
06b5f36d SH |
92 | static void qemu_net_queue_append(NetQueue *queue, |
93 | NetClientState *sender, | |
94 | unsigned flags, | |
95 | const uint8_t *buf, | |
96 | size_t size, | |
97 | NetPacketSent *sent_cb) | |
f7105843 MM |
98 | { |
99 | NetPacket *packet; | |
100 | ||
7d91ddd2 LR |
101 | if (queue->nq_count >= queue->nq_maxlen && !sent_cb) { |
102 | return; /* drop if queue full and no callback */ | |
103 | } | |
7267c094 | 104 | packet = g_malloc(sizeof(NetPacket) + size); |
f7105843 | 105 | packet->sender = sender; |
c0b8e49c | 106 | packet->flags = flags; |
f7105843 MM |
107 | packet->size = size; |
108 | packet->sent_cb = sent_cb; | |
109 | memcpy(packet->data, buf, size); | |
110 | ||
7d91ddd2 | 111 | queue->nq_count++; |
f7105843 | 112 | QTAILQ_INSERT_TAIL(&queue->packets, packet, entry); |
f7105843 MM |
113 | } |
114 | ||
b68c7f76 YH |
115 | void qemu_net_queue_append_iov(NetQueue *queue, |
116 | NetClientState *sender, | |
117 | unsigned flags, | |
118 | const struct iovec *iov, | |
119 | int iovcnt, | |
120 | NetPacketSent *sent_cb) | |
f7105843 MM |
121 | { |
122 | NetPacket *packet; | |
123 | size_t max_len = 0; | |
124 | int i; | |
125 | ||
7d91ddd2 LR |
126 | if (queue->nq_count >= queue->nq_maxlen && !sent_cb) { |
127 | return; /* drop if queue full and no callback */ | |
128 | } | |
f7105843 MM |
129 | for (i = 0; i < iovcnt; i++) { |
130 | max_len += iov[i].iov_len; | |
131 | } | |
132 | ||
7267c094 | 133 | packet = g_malloc(sizeof(NetPacket) + max_len); |
f7105843 MM |
134 | packet->sender = sender; |
135 | packet->sent_cb = sent_cb; | |
c0b8e49c | 136 | packet->flags = flags; |
f7105843 MM |
137 | packet->size = 0; |
138 | ||
139 | for (i = 0; i < iovcnt; i++) { | |
140 | size_t len = iov[i].iov_len; | |
141 | ||
142 | memcpy(packet->data + packet->size, iov[i].iov_base, len); | |
143 | packet->size += len; | |
144 | } | |
145 | ||
7d91ddd2 | 146 | queue->nq_count++; |
f7105843 | 147 | QTAILQ_INSERT_TAIL(&queue->packets, packet, entry); |
f7105843 MM |
148 | } |
149 | ||
150 | static ssize_t qemu_net_queue_deliver(NetQueue *queue, | |
4e68f7a0 | 151 | NetClientState *sender, |
c0b8e49c | 152 | unsigned flags, |
f7105843 MM |
153 | const uint8_t *data, |
154 | size_t size) | |
155 | { | |
156 | ssize_t ret = -1; | |
fefe2a78 YH |
157 | struct iovec iov = { |
158 | .iov_base = (void *)data, | |
159 | .iov_len = size | |
160 | }; | |
f7105843 MM |
161 | |
162 | queue->delivering = 1; | |
3e033a46 | 163 | ret = queue->deliver(sender, flags, &iov, 1, queue->opaque); |
f7105843 MM |
164 | queue->delivering = 0; |
165 | ||
166 | return ret; | |
167 | } | |
168 | ||
169 | static ssize_t qemu_net_queue_deliver_iov(NetQueue *queue, | |
4e68f7a0 | 170 | NetClientState *sender, |
c0b8e49c | 171 | unsigned flags, |
f7105843 MM |
172 | const struct iovec *iov, |
173 | int iovcnt) | |
174 | { | |
175 | ssize_t ret = -1; | |
176 | ||
177 | queue->delivering = 1; | |
3e033a46 | 178 | ret = queue->deliver(sender, flags, iov, iovcnt, queue->opaque); |
f7105843 MM |
179 | queue->delivering = 0; |
180 | ||
181 | return ret; | |
182 | } | |
183 | ||
184 | ssize_t qemu_net_queue_send(NetQueue *queue, | |
4e68f7a0 | 185 | NetClientState *sender, |
c0b8e49c | 186 | unsigned flags, |
f7105843 MM |
187 | const uint8_t *data, |
188 | size_t size, | |
189 | NetPacketSent *sent_cb) | |
190 | { | |
191 | ssize_t ret; | |
192 | ||
691a4f3a | 193 | if (queue->delivering || !qemu_can_send_packet(sender)) { |
06b5f36d SH |
194 | qemu_net_queue_append(queue, sender, flags, data, size, sent_cb); |
195 | return 0; | |
f7105843 MM |
196 | } |
197 | ||
c0b8e49c | 198 | ret = qemu_net_queue_deliver(queue, sender, flags, data, size); |
839f368f | 199 | if (ret == 0) { |
c0b8e49c | 200 | qemu_net_queue_append(queue, sender, flags, data, size, sent_cb); |
f7105843 MM |
201 | return 0; |
202 | } | |
203 | ||
204 | qemu_net_queue_flush(queue); | |
205 | ||
206 | return ret; | |
207 | } | |
208 | ||
209 | ssize_t qemu_net_queue_send_iov(NetQueue *queue, | |
4e68f7a0 | 210 | NetClientState *sender, |
c0b8e49c | 211 | unsigned flags, |
f7105843 MM |
212 | const struct iovec *iov, |
213 | int iovcnt, | |
214 | NetPacketSent *sent_cb) | |
215 | { | |
216 | ssize_t ret; | |
217 | ||
691a4f3a | 218 | if (queue->delivering || !qemu_can_send_packet(sender)) { |
06b5f36d SH |
219 | qemu_net_queue_append_iov(queue, sender, flags, iov, iovcnt, sent_cb); |
220 | return 0; | |
f7105843 MM |
221 | } |
222 | ||
c0b8e49c | 223 | ret = qemu_net_queue_deliver_iov(queue, sender, flags, iov, iovcnt); |
839f368f | 224 | if (ret == 0) { |
c0b8e49c | 225 | qemu_net_queue_append_iov(queue, sender, flags, iov, iovcnt, sent_cb); |
f7105843 MM |
226 | return 0; |
227 | } | |
228 | ||
229 | qemu_net_queue_flush(queue); | |
230 | ||
231 | return ret; | |
232 | } | |
233 | ||
4e68f7a0 | 234 | void qemu_net_queue_purge(NetQueue *queue, NetClientState *from) |
f7105843 MM |
235 | { |
236 | NetPacket *packet, *next; | |
237 | ||
238 | QTAILQ_FOREACH_SAFE(packet, &queue->packets, entry, next) { | |
239 | if (packet->sender == from) { | |
240 | QTAILQ_REMOVE(&queue->packets, packet, entry); | |
7d91ddd2 | 241 | queue->nq_count--; |
07d80846 MT |
242 | if (packet->sent_cb) { |
243 | packet->sent_cb(packet->sender, 0); | |
244 | } | |
7267c094 | 245 | g_free(packet); |
f7105843 MM |
246 | } |
247 | } | |
248 | } | |
249 | ||
987a9b48 | 250 | bool qemu_net_queue_flush(NetQueue *queue) |
f7105843 MM |
251 | { |
252 | while (!QTAILQ_EMPTY(&queue->packets)) { | |
253 | NetPacket *packet; | |
254 | int ret; | |
255 | ||
256 | packet = QTAILQ_FIRST(&queue->packets); | |
257 | QTAILQ_REMOVE(&queue->packets, packet, entry); | |
7d91ddd2 | 258 | queue->nq_count--; |
f7105843 MM |
259 | |
260 | ret = qemu_net_queue_deliver(queue, | |
261 | packet->sender, | |
c0b8e49c | 262 | packet->flags, |
f7105843 MM |
263 | packet->data, |
264 | packet->size); | |
839f368f | 265 | if (ret == 0) { |
7d91ddd2 | 266 | queue->nq_count++; |
f7105843 | 267 | QTAILQ_INSERT_HEAD(&queue->packets, packet, entry); |
987a9b48 | 268 | return false; |
f7105843 MM |
269 | } |
270 | ||
271 | if (packet->sent_cb) { | |
272 | packet->sent_cb(packet->sender, ret); | |
273 | } | |
274 | ||
7267c094 | 275 | g_free(packet); |
f7105843 | 276 | } |
987a9b48 | 277 | return true; |
f7105843 | 278 | } |