cdcad725ff1a052a60fde7441cd8e597541e85a6
[olsrd.git] / lib / arprefresh / src / olsrd_arprefresh.c
1
2 /*
3  * Copyright (c) 2007, Sven-Ola Tuecke <sven-ola-aet-gmx.de>
4  * Copyright (c) 2004, Andreas Tonnesen(andreto-at-olsr.org)
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  *
11  * * Redistributions of source code must retain the above copyright notice,
12  *   this list of conditions and the following disclaimer.
13  * * Redistributions in binary form must reproduce the above copyright notice,
14  *   this list of conditions and the following disclaimer in the documentation
15  *   and/or other materials provided with the distribution.
16  * * Neither the name of the UniK olsr daemon nor the names of its contributors
17  *   may be used to endorse or promote products derived from this software
18  *   without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
22  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
23  * IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
24  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
25  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
27  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
28  * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
29  * OF THE POSSIBILITY OF SUCH DAMAGE.
30  *
31  */
32
33 /*
34  * Plugin to refresh the local ARP cache from received OLSR broadcasts
35  */
36
37 #include <stdio.h>
38 #include <string.h>
39 #include <stdlib.h>
40 #include <fcntl.h>
41 #include <sys/ioctl.h>
42 #include <sys/socket.h>
43 #include <net/if.h>
44 #include <net/if_arp.h>
45 #ifndef __ANDROID__
46 #include <net/ethernet.h>
47 #endif /* __ANDROID__ */
48 #include <netinet/ip.h>
49 #include <netinet/udp.h>
50 #include <netpacket/packet.h>
51 #include <linux/types.h>
52 #include <linux/filter.h>
53 #include <unistd.h>
54
55 #include "olsrd_arprefresh.h"
56 #include "kernel_routes.h"
57 #include "scheduler.h"
58
59 #undef ARPREFRESH_DEBUG
60 #define PLUGIN_INTERFACE_VERSION 5
61
62 /****************************************************************************
63  *                Functions that the plugin MUST provide                    *
64  ****************************************************************************/
65
66 /**
67  * Plugin interface version
68  * Used by main olsrd to check plugin interface version
69  */
70 int
71 olsrd_plugin_interface_version(void)
72 {
73   return PLUGIN_INTERFACE_VERSION;
74 }
75
76 static const struct olsrd_plugin_parameters plugin_parameters[] = {
77 };
78
79 void
80 olsrd_get_plugin_parameters(const struct olsrd_plugin_parameters **params, int *size)
81 {
82   *params = plugin_parameters;
83   *size = sizeof(plugin_parameters) / sizeof(*plugin_parameters);
84 }
85
86 typedef struct {
87   struct ethhdr eth;
88   struct iphdr ip;
89   struct udphdr udp;
90 } __attribute__ ((packed)) arprefresh_buf;
91
92 static int arprefresh_sockfd = -1;
93 static const int arprefresh_portnum = 698;
94
95 /**
96  * Scheduled event to fetch gathered packets and update the ARP cache
97  * called from olsrd main thread
98  */
99 static void
100 olsr_arp_event(void *foo __attribute__ ((unused)))
101 {
102   if (0 <= arprefresh_sockfd) {
103     arprefresh_buf buf;
104     struct sockaddr_ll from;
105     socklen_t fromlen = sizeof(from);
106
107     /*
108      * Grab a single snapshot on the packet socket. This only works
109      * if not too much IP traffic is currently flowing through.
110      */
111     ssize_t size = recvfrom(arprefresh_sockfd, &buf, sizeof(buf),
112                             MSG_TRUNC, (struct sockaddr *)&from,
113                             &fromlen);
114
115     if (0 <= size && size >= (ssize_t) sizeof(buf)
116
117                                                  /*** &&
118                     ETH_P_IP == ntohs(buf.eth.h_proto) &&
119                     IPPROTO_UDP == buf.ip.protocol &&
120                     arprefresh_portnum == ntohs(buf.udp.source) &&
121                     arprefresh_portnum == ntohs(buf.udp.dest) ***/
122       ) {
123       union {
124         struct arpreq arp;
125         struct sockaddr_in in_pa;
126         struct sockaddr_in6 in_pa6;
127       } req;
128
129       memset(&req, 0, sizeof(req));
130       req.in_pa.sin_family = AF_INET;
131       memcpy(&req.in_pa.sin_addr, &buf.ip.saddr, sizeof(buf.ip.saddr));
132       req.arp.arp_ha.sa_family = AF_LOCAL;
133       memcpy(&req.arp.arp_ha.sa_data, &buf.eth.h_source, sizeof(buf.eth.h_source));
134       req.arp.arp_flags = ATF_COM;
135       if_indextoname(from.sll_ifindex, req.arp.arp_dev);
136 #ifdef ARPREFRESH_DEBUG
137       {
138         int i;
139         OLSR_PRINTF(0, "Refresh on %s, %s=", req.arp.arp_dev, inet_ntoa(*((struct in_addr *)&buf.ip.saddr)));
140         for (i = 0; i < (ssize_t) sizeof(buf.eth.h_source); i++) {
141           OLSR_PRINTF(0, "%02x%s", ((unsigned char *)&buf.eth.h_source)[i],
142                       i < (ssize_t) sizeof(buf.eth.h_source) - 1 ? ":" : "\n");
143         }
144       }
145 #endif /* ARPREFRESH_DEBUG */
146       if (ioctl(arprefresh_sockfd, SIOCSARP, &req) < 0) {
147         OLSR_PRINTF(1, "*** ARPREFRESH: SIOCSARP: %s\n", strerror(errno));
148         close(arprefresh_sockfd);
149         arprefresh_sockfd = -1;
150         return;
151       }
152     }
153   }
154 }
155
156 /**
157  * Initialize plugin
158  * Called after all parameters are passed
159  */
160 int
161 olsrd_plugin_init(void)
162 {
163   int ret = 0;
164   if (AF_INET == olsr_cnf->ip_version) {
165     int flags;
166     struct sock_fprog filter;
167     struct sock_filter BPF_code[] = {
168       /* tcpdump -s [sizeof(arprefresh_buf)] -ni lo udp and dst port [arprefresh_portnum] -dd */
169
170       /* Note: This program assumes NON-VLAN */
171       /* See also http://www.unix.com/man-page/FreeBSD/4/bpf/ */
172       /*      code                          jump-true jump-false generic-multiuse-field */
173              {BPF_LD  | BPF_H    | BPF_ABS , 0,        0,         0x0000000c},             // 0x28: A <- P[k:2], get 2-bytes from offset 0x0c : get Ethernet type
174
175 /* test6  */ {BPF_JMP | BPF_JEQ  | BPF_K   , 0,        4,         0x000086dd},             // 0x15: pc += (A == k) ? ipv6 : test4             : IPv6?
176 /* ipv6   */ {BPF_LD  | BPF_B    | BPF_ABS , 0,        0,         0x00000014},             // 0x30: A <- P[k:1], get 1-byte from offset 0x14  : get IPv6 protocol/next header
177              {BPF_JMP | BPF_JEQ  | BPF_K   , 0,        11,        0x00000011},             // 0x15: pc += (A == k) ? udp6 : ignore            : UDP?
178 /* udp6   */ {BPF_LD  | BPF_H    | BPF_ABS , 0,        0,         0x00000038},             // 0x28: A <- P[k:2], get 2-bytes from offset 0x38 : get UDP destination port
179              {BPF_JMP | BPF_JEQ  | BPF_K   , 8,        9,         arprefresh_portnum},     // 0x15: pc += (A == k) ? ok : ignore              : sent to port arprefresh_portnum?
180
181 /* test4  */ {BPF_JMP | BPF_JEQ  | BPF_K   , 0,        8,         0x00000800},             // 0x15: pc += (A == k) ? ipv4 : ignore            : IPv4?
182 /* ipv4   */ {BPF_LD  | BPF_B    | BPF_ABS , 0,        0,         0x00000017},             // 0x30: A <- P[k:1], get 1-byte from offset 0x17  : get IPv4 protocol
183              {BPF_JMP | BPF_JEQ  | BPF_K   , 0,        6,         0x00000011},             // 0x15: pc += (A == k) ? udp4 : ignore            : UDP?
184 /* udp4   */ {BPF_LD  | BPF_H    | BPF_ABS , 0,        0,         0x00000014},             // 0x28: A <- P[k:2], get 2-bytes from offset 0x14 : get fragment offset
185              {BPF_JMP | BPF_JSET | BPF_K   , 4,        0,         0x00001fff},             // 0x45: pc += (A & k) ? ignore : nofrag           : is this a fragment?
186 /* nofrag */ {BPF_LDX | BPF_B    | BPF_MSH , 0,        0,         0x0000000e},             // 0xb1: X <- 4*(P[k:1]&0xf)                       : get the IP header length in bytes
187              {BPF_LD  | BPF_H    | BPF_IND , 0,        0,         0x00000010},             // 0x48: A <- P[X+k:2]                             : get UDP destination port
188              {BPF_JMP | BPF_JEQ  | BPF_K   , 0,        1,         arprefresh_portnum},     // 0x15: pc += (A == k) ? ok : ignore              : sent to port arprefresh_portnum?
189
190 /* ok     */ {BPF_RET | BPF_K              , 0,        0,         sizeof(arprefresh_buf)}, //                                                 : accept sizeof(arprefresh_buf) bytes (all headers)
191 /* ignore */ {BPF_RET | BPF_K              , 0,        0,         0x00000000}              //                                                 : accept 0 bytes
192
193     };
194     filter.len = sizeof(BPF_code) / sizeof(BPF_code[0]);
195     filter.filter = BPF_code;
196     if (0 <= (arprefresh_sockfd = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_IP)))
197         && 0 <= (flags = fcntl(arprefresh_sockfd, F_GETFL))
198         && 0 <= fcntl(arprefresh_sockfd, F_SETFL, flags | O_NONBLOCK)
199         && 0 <= setsockopt(arprefresh_sockfd, SOL_SOCKET, SO_ATTACH_FILTER, &filter, sizeof(filter))) {
200       /* Register the ARP refresh event */
201       olsr_start_timer(2 * MSEC_PER_SEC, 0, OLSR_TIMER_PERIODIC, &olsr_arp_event, NULL, 0);
202       ret = 1;
203     } else {
204       OLSR_PRINTF(1, "*** ARPREFRESH: Cannot create non-blocking filtering packet socket: %s\n", strerror(errno));
205     }
206   } else {
207     OLSR_PRINTF(1, "*** ARPREFRESH: IPv6 not supported\n");
208   }
209   return ret;
210 }
211
212 /****************************************************************************
213  *       Optional private constructor and destructor functions              *
214  ****************************************************************************/
215
216 static void __attribute__ ((constructor)) my_init(void);
217 static void __attribute__ ((destructor)) my_fini(void);
218
219 static void
220 my_init(void)
221 {
222   printf("OLSRD arprefresh plugin by Sven-Ola\n");
223 }
224
225 /**
226  * Optional Private Destructor
227  */
228 static void
229 my_fini(void)
230 {
231   if (0 <= arprefresh_sockfd) {
232     close(arprefresh_sockfd);
233     arprefresh_sockfd = -1;
234   }
235 }
236
237 /*
238  * Local Variables:
239  * c-basic-offset: 2
240  * indent-tabs-mode: nil
241  * End:
242  */