1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25#include "qemu/osdep.h"
26#include "clients.h"
27#include "qapi/error.h"
28#include "qemu/error-report.h"
29#include "qemu/iov.h"
30#include "qemu/module.h"
31#include "qemu/timer.h"
32#include "qapi/visitor.h"
33#include "net/filter.h"
34#include "qom/object.h"
35#include "sysemu/rtc.h"
36
37typedef struct DumpState {
38 int64_t start_ts;
39 int fd;
40 int pcap_caplen;
41} DumpState;
42
43#define PCAP_MAGIC 0xa1b2c3d4
44
45struct pcap_file_hdr {
46 uint32_t magic;
47 uint16_t version_major;
48 uint16_t version_minor;
49 int32_t thiszone;
50 uint32_t sigfigs;
51 uint32_t snaplen;
52 uint32_t linktype;
53};
54
55struct pcap_sf_pkthdr {
56 struct {
57 int32_t tv_sec;
58 int32_t tv_usec;
59 } ts;
60 uint32_t caplen;
61 uint32_t len;
62};
63
64static ssize_t dump_receive_iov(DumpState *s, const struct iovec *iov, int cnt,
65 int offset)
66{
67 struct pcap_sf_pkthdr hdr;
68 int64_t ts;
69 int caplen;
70 size_t size = iov_size(iov, cnt) - offset;
71 struct iovec dumpiov[cnt + 1];
72
73
74 if (s->fd < 0) {
75 return size;
76 }
77
78 ts = qemu_clock_get_us(QEMU_CLOCK_VIRTUAL);
79 caplen = size > s->pcap_caplen ? s->pcap_caplen : size;
80
81 hdr.ts.tv_sec = ts / 1000000 + s->start_ts;
82 hdr.ts.tv_usec = ts % 1000000;
83 hdr.caplen = caplen;
84 hdr.len = size;
85
86 dumpiov[0].iov_base = &hdr;
87 dumpiov[0].iov_len = sizeof(hdr);
88 cnt = iov_copy(&dumpiov[1], cnt, iov, cnt, offset, caplen);
89
90 if (writev(s->fd, dumpiov, cnt + 1) != sizeof(hdr) + caplen) {
91 error_report("network dump write error - stopping dump");
92 close(s->fd);
93 s->fd = -1;
94 }
95
96 return size;
97}
98
99static void dump_cleanup(DumpState *s)
100{
101 close(s->fd);
102 s->fd = -1;
103}
104
105static int net_dump_state_init(DumpState *s, const char *filename,
106 int len, Error **errp)
107{
108 struct pcap_file_hdr hdr;
109 struct tm tm;
110 int fd;
111
112 fd = open(filename, O_CREAT | O_TRUNC | O_WRONLY | O_BINARY, 0644);
113 if (fd < 0) {
114 error_setg_errno(errp, errno, "net dump: can't open %s", filename);
115 return -1;
116 }
117
118 hdr.magic = PCAP_MAGIC;
119 hdr.version_major = 2;
120 hdr.version_minor = 4;
121 hdr.thiszone = 0;
122 hdr.sigfigs = 0;
123 hdr.snaplen = len;
124 hdr.linktype = 1;
125
126 if (write(fd, &hdr, sizeof(hdr)) < sizeof(hdr)) {
127 error_setg_errno(errp, errno, "net dump write error");
128 close(fd);
129 return -1;
130 }
131
132 s->fd = fd;
133 s->pcap_caplen = len;
134
135 qemu_get_timedate(&tm, 0);
136 s->start_ts = mktime(&tm);
137
138 return 0;
139}
140
141#define TYPE_FILTER_DUMP "filter-dump"
142
143OBJECT_DECLARE_SIMPLE_TYPE(NetFilterDumpState, FILTER_DUMP)
144
145struct NetFilterDumpState {
146 NetFilterState nfs;
147 DumpState ds;
148 char *filename;
149 uint32_t maxlen;
150};
151
152static ssize_t filter_dump_receive_iov(NetFilterState *nf, NetClientState *sndr,
153 unsigned flags, const struct iovec *iov,
154 int iovcnt, NetPacketSent *sent_cb)
155{
156 NetFilterDumpState *nfds = FILTER_DUMP(nf);
157 int offset = qemu_get_using_vnet_hdr(nf->netdev) ?
158 qemu_get_vnet_hdr_len(nf->netdev) : 0;
159
160 dump_receive_iov(&nfds->ds, iov, iovcnt, offset);
161 return 0;
162}
163
164static void filter_dump_cleanup(NetFilterState *nf)
165{
166 NetFilterDumpState *nfds = FILTER_DUMP(nf);
167
168 dump_cleanup(&nfds->ds);
169}
170
171static void filter_dump_setup(NetFilterState *nf, Error **errp)
172{
173 NetFilterDumpState *nfds = FILTER_DUMP(nf);
174
175 if (!nfds->filename) {
176 error_setg(errp, "dump filter needs 'file' property set!");
177 return;
178 }
179
180 net_dump_state_init(&nfds->ds, nfds->filename, nfds->maxlen, errp);
181}
182
183static void filter_dump_get_maxlen(Object *obj, Visitor *v, const char *name,
184 void *opaque, Error **errp)
185{
186 NetFilterDumpState *nfds = FILTER_DUMP(obj);
187 uint32_t value = nfds->maxlen;
188
189 visit_type_uint32(v, name, &value, errp);
190}
191
192static void filter_dump_set_maxlen(Object *obj, Visitor *v, const char *name,
193 void *opaque, Error **errp)
194{
195 NetFilterDumpState *nfds = FILTER_DUMP(obj);
196 uint32_t value;
197
198 if (!visit_type_uint32(v, name, &value, errp)) {
199 return;
200 }
201 if (value == 0) {
202 error_setg(errp, "Property '%s.%s' doesn't take value '%u'",
203 object_get_typename(obj), name, value);
204 return;
205 }
206 nfds->maxlen = value;
207}
208
209static char *file_dump_get_filename(Object *obj, Error **errp)
210{
211 NetFilterDumpState *nfds = FILTER_DUMP(obj);
212
213 return g_strdup(nfds->filename);
214}
215
216static void file_dump_set_filename(Object *obj, const char *value, Error **errp)
217{
218 NetFilterDumpState *nfds = FILTER_DUMP(obj);
219
220 g_free(nfds->filename);
221 nfds->filename = g_strdup(value);
222}
223
224static void filter_dump_instance_init(Object *obj)
225{
226 NetFilterDumpState *nfds = FILTER_DUMP(obj);
227
228 nfds->maxlen = 65536;
229}
230
231static void filter_dump_instance_finalize(Object *obj)
232{
233 NetFilterDumpState *nfds = FILTER_DUMP(obj);
234
235 g_free(nfds->filename);
236}
237
238static void filter_dump_class_init(ObjectClass *oc, void *data)
239{
240 NetFilterClass *nfc = NETFILTER_CLASS(oc);
241
242 object_class_property_add(oc, "maxlen", "uint32", filter_dump_get_maxlen,
243 filter_dump_set_maxlen, NULL, NULL);
244 object_class_property_add_str(oc, "file", file_dump_get_filename,
245 file_dump_set_filename);
246
247 nfc->setup = filter_dump_setup;
248 nfc->cleanup = filter_dump_cleanup;
249 nfc->receive_iov = filter_dump_receive_iov;
250}
251
252static const TypeInfo filter_dump_info = {
253 .name = TYPE_FILTER_DUMP,
254 .parent = TYPE_NETFILTER,
255 .class_init = filter_dump_class_init,
256 .instance_init = filter_dump_instance_init,
257 .instance_finalize = filter_dump_instance_finalize,
258 .instance_size = sizeof(NetFilterDumpState),
259};
260
261static void filter_dump_register_types(void)
262{
263 type_register_static(&filter_dump_info);
264}
265
266type_init(filter_dump_register_types);
267
