1
2
3
4
5
6
7
8
9#include "qemu/osdep.h"
10#include "net/filter.h"
11#include "net/queue.h"
12#include "qapi/error.h"
13#include "qemu/timer.h"
14#include "qemu/iov.h"
15#include "qapi/qapi-builtin-visit.h"
16#include "qapi/qmp/qerror.h"
17#include "qom/object.h"
18
19#define TYPE_FILTER_BUFFER "filter-buffer"
20
21OBJECT_DECLARE_SIMPLE_TYPE(FilterBufferState, FILTER_BUFFER)
22
23struct FilterBufferState {
24 NetFilterState parent_obj;
25
26 NetQueue *incoming_queue;
27 uint32_t interval;
28 QEMUTimer release_timer;
29};
30
31static void filter_buffer_flush(NetFilterState *nf)
32{
33 FilterBufferState *s = FILTER_BUFFER(nf);
34
35 if (!qemu_net_queue_flush(s->incoming_queue)) {
36
37 qemu_net_queue_purge(s->incoming_queue, nf->netdev);
38 }
39}
40
41static void filter_buffer_release_timer(void *opaque)
42{
43 NetFilterState *nf = opaque;
44 FilterBufferState *s = FILTER_BUFFER(nf);
45
46
47
48
49
50
51
52 filter_buffer_flush(nf);
53
54 timer_mod(&s->release_timer,
55 qemu_clock_get_us(QEMU_CLOCK_VIRTUAL) + s->interval);
56}
57
58
59static ssize_t filter_buffer_receive_iov(NetFilterState *nf,
60 NetClientState *sender,
61 unsigned flags,
62 const struct iovec *iov,
63 int iovcnt,
64 NetPacketSent *sent_cb)
65{
66 FilterBufferState *s = FILTER_BUFFER(nf);
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82 qemu_net_queue_append_iov(s->incoming_queue, sender, flags,
83 iov, iovcnt, NULL);
84 return iov_size(iov, iovcnt);
85}
86
87static void filter_buffer_cleanup(NetFilterState *nf)
88{
89 FilterBufferState *s = FILTER_BUFFER(nf);
90
91 if (s->interval) {
92 timer_del(&s->release_timer);
93 }
94
95
96 if (s->incoming_queue) {
97 filter_buffer_flush(nf);
98 g_free(s->incoming_queue);
99 }
100}
101
102static void filter_buffer_setup_timer(NetFilterState *nf)
103{
104 FilterBufferState *s = FILTER_BUFFER(nf);
105
106 if (s->interval) {
107 timer_init_us(&s->release_timer, QEMU_CLOCK_VIRTUAL,
108 filter_buffer_release_timer, nf);
109
110 timer_mod(&s->release_timer,
111 qemu_clock_get_us(QEMU_CLOCK_VIRTUAL) + s->interval);
112 }
113}
114
115static void filter_buffer_setup(NetFilterState *nf, Error **errp)
116{
117 FilterBufferState *s = FILTER_BUFFER(nf);
118
119
120
121
122
123 if (!s->interval) {
124 error_setg(errp, QERR_INVALID_PARAMETER_VALUE, "interval",
125 "a non-zero interval");
126 return;
127 }
128
129 s->incoming_queue = qemu_new_net_queue(qemu_netfilter_pass_to_next, nf);
130 filter_buffer_setup_timer(nf);
131}
132
133static void filter_buffer_status_changed(NetFilterState *nf, Error **errp)
134{
135 FilterBufferState *s = FILTER_BUFFER(nf);
136
137 if (!nf->on) {
138 if (s->interval) {
139 timer_del(&s->release_timer);
140 }
141 filter_buffer_flush(nf);
142 } else {
143 filter_buffer_setup_timer(nf);
144 }
145}
146
147static void filter_buffer_get_interval(Object *obj, Visitor *v,
148 const char *name, void *opaque,
149 Error **errp)
150{
151 FilterBufferState *s = FILTER_BUFFER(obj);
152 uint32_t value = s->interval;
153
154 visit_type_uint32(v, name, &value, errp);
155}
156
157static void filter_buffer_set_interval(Object *obj, Visitor *v,
158 const char *name, void *opaque,
159 Error **errp)
160{
161 FilterBufferState *s = FILTER_BUFFER(obj);
162 uint32_t value;
163
164 if (!visit_type_uint32(v, name, &value, errp)) {
165 return;
166 }
167 if (!value) {
168 error_setg(errp, "Property '%s.%s' requires a positive value",
169 object_get_typename(obj), name);
170 return;
171 }
172 s->interval = value;
173}
174
175static void filter_buffer_class_init(ObjectClass *oc, void *data)
176{
177 NetFilterClass *nfc = NETFILTER_CLASS(oc);
178
179 object_class_property_add(oc, "interval", "uint32",
180 filter_buffer_get_interval,
181 filter_buffer_set_interval, NULL, NULL);
182
183 nfc->setup = filter_buffer_setup;
184 nfc->cleanup = filter_buffer_cleanup;
185 nfc->receive_iov = filter_buffer_receive_iov;
186 nfc->status_changed = filter_buffer_status_changed;
187}
188
189static const TypeInfo filter_buffer_info = {
190 .name = TYPE_FILTER_BUFFER,
191 .parent = TYPE_NETFILTER,
192 .class_init = filter_buffer_class_init,
193 .instance_size = sizeof(FilterBufferState),
194};
195
196static void register_types(void)
197{
198 type_register_static(&filter_buffer_info);
199}
200
201type_init(register_types);
202