Commit | Line | Data |
---|---|---|
1da177e4 | 1 | /* |
cd1ae0e4 | 2 | * Copyright (C) 2002 - 2007 Jeff Dike (jdike@{addtoit,linux.intel}.com) |
1da177e4 LT |
3 | * Licensed under the GPL. |
4 | */ | |
5 | ||
1da177e4 LT |
6 | #include <errno.h> |
7 | #include <pcap.h> | |
cd1ae0e4 | 8 | #include <string.h> |
1da177e4 | 9 | #include <asm/types.h> |
37185b33 | 10 | #include <net_user.h> |
1da177e4 | 11 | #include "pcap_user.h" |
37185b33 | 12 | #include <um_malloc.h> |
1da177e4 | 13 | |
1da177e4 LT |
14 | #define PCAP_FD(p) (*(int *)(p)) |
15 | ||
f34d9d2d | 16 | static int pcap_user_init(void *data, void *dev) |
1da177e4 LT |
17 | { |
18 | struct pcap_data *pri = data; | |
19 | pcap_t *p; | |
20 | char errors[PCAP_ERRBUF_SIZE]; | |
21 | ||
b53f35a8 JD |
22 | p = pcap_open_live(pri->host_if, ETH_MAX_PACKET + ETH_HEADER_OTHER, |
23 | pri->promisc, 0, errors); | |
cd1ae0e4 | 24 | if (p == NULL) { |
7d98230a JD |
25 | printk(UM_KERN_ERR "pcap_user_init : pcap_open_live failed - " |
26 | "'%s'\n", errors); | |
f34d9d2d | 27 | return -EINVAL; |
1da177e4 LT |
28 | } |
29 | ||
30 | pri->dev = dev; | |
31 | pri->pcap = p; | |
f34d9d2d | 32 | return 0; |
1da177e4 LT |
33 | } |
34 | ||
35 | static int pcap_open(void *data) | |
36 | { | |
37 | struct pcap_data *pri = data; | |
38 | __u32 netmask; | |
39 | int err; | |
40 | ||
cd1ae0e4 | 41 | if (pri->pcap == NULL) |
56bd194b | 42 | return -ENODEV; |
1da177e4 | 43 | |
cd1ae0e4 | 44 | if (pri->filter != NULL) { |
1da177e4 | 45 | err = dev_netmask(pri->dev, &netmask); |
cd1ae0e4 | 46 | if (err < 0) { |
7d98230a | 47 | printk(UM_KERN_ERR "pcap_open : dev_netmask failed\n"); |
56bd194b | 48 | return -EIO; |
1da177e4 LT |
49 | } |
50 | ||
296cd66f | 51 | pri->compiled = uml_kmalloc(sizeof(struct bpf_program), |
cd1ae0e4 JD |
52 | UM_GFP_KERNEL); |
53 | if (pri->compiled == NULL) { | |
7d98230a | 54 | printk(UM_KERN_ERR "pcap_open : kmalloc failed\n"); |
56bd194b | 55 | return -ENOMEM; |
1da177e4 | 56 | } |
56bd194b | 57 | |
cd1ae0e4 JD |
58 | err = pcap_compile(pri->pcap, |
59 | (struct bpf_program *) pri->compiled, | |
1da177e4 | 60 | pri->filter, pri->optimize, netmask); |
cd1ae0e4 | 61 | if (err < 0) { |
7d98230a JD |
62 | printk(UM_KERN_ERR "pcap_open : pcap_compile failed - " |
63 | "'%s'\n", pcap_geterr(pri->pcap)); | |
cd1ae0e4 | 64 | goto out; |
1da177e4 LT |
65 | } |
66 | ||
67 | err = pcap_setfilter(pri->pcap, pri->compiled); | |
cd1ae0e4 | 68 | if (err < 0) { |
7d98230a JD |
69 | printk(UM_KERN_ERR "pcap_open : pcap_setfilter " |
70 | "failed - '%s'\n", pcap_geterr(pri->pcap)); | |
cd1ae0e4 | 71 | goto out; |
1da177e4 LT |
72 | } |
73 | } | |
56bd194b JD |
74 | |
75 | return PCAP_FD(pri->pcap); | |
cd1ae0e4 JD |
76 | |
77 | out: | |
78 | kfree(pri->compiled); | |
79 | return -EIO; | |
1da177e4 LT |
80 | } |
81 | ||
82 | static void pcap_remove(void *data) | |
83 | { | |
84 | struct pcap_data *pri = data; | |
85 | ||
cd1ae0e4 | 86 | if (pri->compiled != NULL) |
1da177e4 LT |
87 | pcap_freecode(pri->compiled); |
88 | ||
cd1ae0e4 | 89 | if (pri->pcap != NULL) |
7d98230a | 90 | pcap_close(pri->pcap); |
1da177e4 LT |
91 | } |
92 | ||
93 | struct pcap_handler_data { | |
94 | char *buffer; | |
95 | int len; | |
96 | }; | |
97 | ||
cd1ae0e4 | 98 | static void handler(u_char *data, const struct pcap_pkthdr *header, |
1da177e4 LT |
99 | const u_char *packet) |
100 | { | |
101 | int len; | |
102 | ||
103 | struct pcap_handler_data *hdata = (struct pcap_handler_data *) data; | |
104 | ||
105 | len = hdata->len < header->caplen ? hdata->len : header->caplen; | |
106 | memcpy(hdata->buffer, packet, len); | |
107 | hdata->len = len; | |
108 | } | |
109 | ||
110 | int pcap_user_read(int fd, void *buffer, int len, struct pcap_data *pri) | |
111 | { | |
112 | struct pcap_handler_data hdata = ((struct pcap_handler_data) | |
113 | { .buffer = buffer, | |
114 | .len = len }); | |
115 | int n; | |
116 | ||
117 | n = pcap_dispatch(pri->pcap, 1, handler, (u_char *) &hdata); | |
cd1ae0e4 | 118 | if (n < 0) { |
7d98230a JD |
119 | printk(UM_KERN_ERR "pcap_dispatch failed - %s\n", |
120 | pcap_geterr(pri->pcap)); | |
56bd194b | 121 | return -EIO; |
1da177e4 | 122 | } |
cd1ae0e4 | 123 | else if (n == 0) |
56bd194b JD |
124 | return 0; |
125 | return hdata.len; | |
1da177e4 LT |
126 | } |
127 | ||
5e7672ec | 128 | const struct net_user_info pcap_user_info = { |
1da177e4 LT |
129 | .init = pcap_user_init, |
130 | .open = pcap_open, | |
131 | .close = NULL, | |
132 | .remove = pcap_remove, | |
1da177e4 LT |
133 | .add_address = NULL, |
134 | .delete_address = NULL, | |
b53f35a8 JD |
135 | .mtu = ETH_MAX_PACKET, |
136 | .max_packet = ETH_MAX_PACKET + ETH_HEADER_OTHER, | |
1da177e4 | 137 | }; |