1 | // SPDX-License-Identifier: GPL-2.0 |
2 | /* Copyright (C) B.A.T.M.A.N. contributors: |
3 | * |
4 | * Marek Lindner, Simon Wunderlich |
5 | */ |
6 | |
7 | #include "main.h" |
8 | |
9 | #include <linux/byteorder/generic.h> |
10 | #include <linux/container_of.h> |
11 | #include <linux/etherdevice.h> |
12 | #include <linux/gfp.h> |
13 | #include <linux/if_ether.h> |
14 | #include <linux/kref.h> |
15 | #include <linux/list.h> |
16 | #include <linux/lockdep.h> |
17 | #include <linux/netdevice.h> |
18 | #include <linux/pkt_sched.h> |
19 | #include <linux/rculist.h> |
20 | #include <linux/rcupdate.h> |
21 | #include <linux/skbuff.h> |
22 | #include <linux/slab.h> |
23 | #include <linux/spinlock.h> |
24 | #include <linux/stddef.h> |
25 | #include <linux/string.h> |
26 | #include <linux/types.h> |
27 | #include <uapi/linux/batadv_packet.h> |
28 | |
29 | #include "originator.h" |
30 | #include "send.h" |
31 | #include "tvlv.h" |
32 | |
33 | /** |
34 | * batadv_tvlv_handler_release() - release tvlv handler from lists and queue for |
35 | * free after rcu grace period |
36 | * @ref: kref pointer of the tvlv |
37 | */ |
38 | static void batadv_tvlv_handler_release(struct kref *ref) |
39 | { |
40 | struct batadv_tvlv_handler *tvlv_handler; |
41 | |
42 | tvlv_handler = container_of(ref, struct batadv_tvlv_handler, refcount); |
43 | kfree_rcu(tvlv_handler, rcu); |
44 | } |
45 | |
46 | /** |
47 | * batadv_tvlv_handler_put() - decrement the tvlv container refcounter and |
48 | * possibly release it |
49 | * @tvlv_handler: the tvlv handler to free |
50 | */ |
51 | static void batadv_tvlv_handler_put(struct batadv_tvlv_handler *tvlv_handler) |
52 | { |
53 | if (!tvlv_handler) |
54 | return; |
55 | |
56 | kref_put(kref: &tvlv_handler->refcount, release: batadv_tvlv_handler_release); |
57 | } |
58 | |
59 | /** |
60 | * batadv_tvlv_handler_get() - retrieve tvlv handler from the tvlv handler list |
61 | * based on the provided type and version (both need to match) |
62 | * @bat_priv: the bat priv with all the soft interface information |
63 | * @type: tvlv handler type to look for |
64 | * @version: tvlv handler version to look for |
65 | * |
66 | * Return: tvlv handler if found or NULL otherwise. |
67 | */ |
68 | static struct batadv_tvlv_handler * |
69 | batadv_tvlv_handler_get(struct batadv_priv *bat_priv, u8 type, u8 version) |
70 | { |
71 | struct batadv_tvlv_handler *tvlv_handler_tmp, *tvlv_handler = NULL; |
72 | |
73 | rcu_read_lock(); |
74 | hlist_for_each_entry_rcu(tvlv_handler_tmp, |
75 | &bat_priv->tvlv.handler_list, list) { |
76 | if (tvlv_handler_tmp->type != type) |
77 | continue; |
78 | |
79 | if (tvlv_handler_tmp->version != version) |
80 | continue; |
81 | |
82 | if (!kref_get_unless_zero(kref: &tvlv_handler_tmp->refcount)) |
83 | continue; |
84 | |
85 | tvlv_handler = tvlv_handler_tmp; |
86 | break; |
87 | } |
88 | rcu_read_unlock(); |
89 | |
90 | return tvlv_handler; |
91 | } |
92 | |
93 | /** |
94 | * batadv_tvlv_container_release() - release tvlv from lists and free |
95 | * @ref: kref pointer of the tvlv |
96 | */ |
97 | static void batadv_tvlv_container_release(struct kref *ref) |
98 | { |
99 | struct batadv_tvlv_container *tvlv; |
100 | |
101 | tvlv = container_of(ref, struct batadv_tvlv_container, refcount); |
102 | kfree(objp: tvlv); |
103 | } |
104 | |
105 | /** |
106 | * batadv_tvlv_container_put() - decrement the tvlv container refcounter and |
107 | * possibly release it |
108 | * @tvlv: the tvlv container to free |
109 | */ |
110 | static void batadv_tvlv_container_put(struct batadv_tvlv_container *tvlv) |
111 | { |
112 | if (!tvlv) |
113 | return; |
114 | |
115 | kref_put(kref: &tvlv->refcount, release: batadv_tvlv_container_release); |
116 | } |
117 | |
118 | /** |
119 | * batadv_tvlv_container_get() - retrieve tvlv container from the tvlv container |
120 | * list based on the provided type and version (both need to match) |
121 | * @bat_priv: the bat priv with all the soft interface information |
122 | * @type: tvlv container type to look for |
123 | * @version: tvlv container version to look for |
124 | * |
125 | * Has to be called with the appropriate locks being acquired |
126 | * (tvlv.container_list_lock). |
127 | * |
128 | * Return: tvlv container if found or NULL otherwise. |
129 | */ |
130 | static struct batadv_tvlv_container * |
131 | batadv_tvlv_container_get(struct batadv_priv *bat_priv, u8 type, u8 version) |
132 | { |
133 | struct batadv_tvlv_container *tvlv_tmp, *tvlv = NULL; |
134 | |
135 | lockdep_assert_held(&bat_priv->tvlv.container_list_lock); |
136 | |
137 | hlist_for_each_entry(tvlv_tmp, &bat_priv->tvlv.container_list, list) { |
138 | if (tvlv_tmp->tvlv_hdr.type != type) |
139 | continue; |
140 | |
141 | if (tvlv_tmp->tvlv_hdr.version != version) |
142 | continue; |
143 | |
144 | kref_get(kref: &tvlv_tmp->refcount); |
145 | tvlv = tvlv_tmp; |
146 | break; |
147 | } |
148 | |
149 | return tvlv; |
150 | } |
151 | |
152 | /** |
153 | * batadv_tvlv_container_list_size() - calculate the size of the tvlv container |
154 | * list entries |
155 | * @bat_priv: the bat priv with all the soft interface information |
156 | * |
157 | * Has to be called with the appropriate locks being acquired |
158 | * (tvlv.container_list_lock). |
159 | * |
160 | * Return: size of all currently registered tvlv containers in bytes. |
161 | */ |
162 | static u16 batadv_tvlv_container_list_size(struct batadv_priv *bat_priv) |
163 | { |
164 | struct batadv_tvlv_container *tvlv; |
165 | u16 tvlv_len = 0; |
166 | |
167 | lockdep_assert_held(&bat_priv->tvlv.container_list_lock); |
168 | |
169 | hlist_for_each_entry(tvlv, &bat_priv->tvlv.container_list, list) { |
170 | tvlv_len += sizeof(struct batadv_tvlv_hdr); |
171 | tvlv_len += ntohs(tvlv->tvlv_hdr.len); |
172 | } |
173 | |
174 | return tvlv_len; |
175 | } |
176 | |
177 | /** |
178 | * batadv_tvlv_container_remove() - remove tvlv container from the tvlv |
179 | * container list |
180 | * @bat_priv: the bat priv with all the soft interface information |
181 | * @tvlv: the to be removed tvlv container |
182 | * |
183 | * Has to be called with the appropriate locks being acquired |
184 | * (tvlv.container_list_lock). |
185 | */ |
186 | static void batadv_tvlv_container_remove(struct batadv_priv *bat_priv, |
187 | struct batadv_tvlv_container *tvlv) |
188 | { |
189 | lockdep_assert_held(&bat_priv->tvlv.container_list_lock); |
190 | |
191 | if (!tvlv) |
192 | return; |
193 | |
194 | hlist_del(n: &tvlv->list); |
195 | |
196 | /* first call to decrement the counter, second call to free */ |
197 | batadv_tvlv_container_put(tvlv); |
198 | batadv_tvlv_container_put(tvlv); |
199 | } |
200 | |
201 | /** |
202 | * batadv_tvlv_container_unregister() - unregister tvlv container based on the |
203 | * provided type and version (both need to match) |
204 | * @bat_priv: the bat priv with all the soft interface information |
205 | * @type: tvlv container type to unregister |
206 | * @version: tvlv container type to unregister |
207 | */ |
208 | void batadv_tvlv_container_unregister(struct batadv_priv *bat_priv, |
209 | u8 type, u8 version) |
210 | { |
211 | struct batadv_tvlv_container *tvlv; |
212 | |
213 | spin_lock_bh(lock: &bat_priv->tvlv.container_list_lock); |
214 | tvlv = batadv_tvlv_container_get(bat_priv, type, version); |
215 | batadv_tvlv_container_remove(bat_priv, tvlv); |
216 | spin_unlock_bh(lock: &bat_priv->tvlv.container_list_lock); |
217 | } |
218 | |
219 | /** |
220 | * batadv_tvlv_container_register() - register tvlv type, version and content |
221 | * to be propagated with each (primary interface) OGM |
222 | * @bat_priv: the bat priv with all the soft interface information |
223 | * @type: tvlv container type |
224 | * @version: tvlv container version |
225 | * @tvlv_value: tvlv container content |
226 | * @tvlv_value_len: tvlv container content length |
227 | * |
228 | * If a container of the same type and version was already registered the new |
229 | * content is going to replace the old one. |
230 | */ |
231 | void batadv_tvlv_container_register(struct batadv_priv *bat_priv, |
232 | u8 type, u8 version, |
233 | void *tvlv_value, u16 tvlv_value_len) |
234 | { |
235 | struct batadv_tvlv_container *tvlv_old, *tvlv_new; |
236 | |
237 | if (!tvlv_value) |
238 | tvlv_value_len = 0; |
239 | |
240 | tvlv_new = kzalloc(size: sizeof(*tvlv_new) + tvlv_value_len, GFP_ATOMIC); |
241 | if (!tvlv_new) |
242 | return; |
243 | |
244 | tvlv_new->tvlv_hdr.version = version; |
245 | tvlv_new->tvlv_hdr.type = type; |
246 | tvlv_new->tvlv_hdr.len = htons(tvlv_value_len); |
247 | |
248 | memcpy(tvlv_new + 1, tvlv_value, ntohs(tvlv_new->tvlv_hdr.len)); |
249 | INIT_HLIST_NODE(h: &tvlv_new->list); |
250 | kref_init(kref: &tvlv_new->refcount); |
251 | |
252 | spin_lock_bh(lock: &bat_priv->tvlv.container_list_lock); |
253 | tvlv_old = batadv_tvlv_container_get(bat_priv, type, version); |
254 | batadv_tvlv_container_remove(bat_priv, tvlv: tvlv_old); |
255 | |
256 | kref_get(kref: &tvlv_new->refcount); |
257 | hlist_add_head(n: &tvlv_new->list, h: &bat_priv->tvlv.container_list); |
258 | spin_unlock_bh(lock: &bat_priv->tvlv.container_list_lock); |
259 | |
260 | /* don't return reference to new tvlv_container */ |
261 | batadv_tvlv_container_put(tvlv: tvlv_new); |
262 | } |
263 | |
264 | /** |
265 | * batadv_tvlv_realloc_packet_buff() - reallocate packet buffer to accommodate |
266 | * requested packet size |
267 | * @packet_buff: packet buffer |
268 | * @packet_buff_len: packet buffer size |
269 | * @min_packet_len: requested packet minimum size |
270 | * @additional_packet_len: requested additional packet size on top of minimum |
271 | * size |
272 | * |
273 | * Return: true of the packet buffer could be changed to the requested size, |
274 | * false otherwise. |
275 | */ |
276 | static bool batadv_tvlv_realloc_packet_buff(unsigned char **packet_buff, |
277 | int *packet_buff_len, |
278 | int min_packet_len, |
279 | int additional_packet_len) |
280 | { |
281 | unsigned char *new_buff; |
282 | |
283 | new_buff = kmalloc(size: min_packet_len + additional_packet_len, GFP_ATOMIC); |
284 | |
285 | /* keep old buffer if kmalloc should fail */ |
286 | if (!new_buff) |
287 | return false; |
288 | |
289 | memcpy(new_buff, *packet_buff, min_packet_len); |
290 | kfree(objp: *packet_buff); |
291 | *packet_buff = new_buff; |
292 | *packet_buff_len = min_packet_len + additional_packet_len; |
293 | |
294 | return true; |
295 | } |
296 | |
297 | /** |
298 | * batadv_tvlv_container_ogm_append() - append tvlv container content to given |
299 | * OGM packet buffer |
300 | * @bat_priv: the bat priv with all the soft interface information |
301 | * @packet_buff: ogm packet buffer |
302 | * @packet_buff_len: ogm packet buffer size including ogm header and tvlv |
303 | * content |
304 | * @packet_min_len: ogm header size to be preserved for the OGM itself |
305 | * |
306 | * The ogm packet might be enlarged or shrunk depending on the current size |
307 | * and the size of the to-be-appended tvlv containers. |
308 | * |
309 | * Return: size of all appended tvlv containers in bytes. |
310 | */ |
311 | u16 batadv_tvlv_container_ogm_append(struct batadv_priv *bat_priv, |
312 | unsigned char **packet_buff, |
313 | int *packet_buff_len, int packet_min_len) |
314 | { |
315 | struct batadv_tvlv_container *tvlv; |
316 | struct batadv_tvlv_hdr *tvlv_hdr; |
317 | u16 tvlv_value_len; |
318 | void *tvlv_value; |
319 | bool ret; |
320 | |
321 | spin_lock_bh(lock: &bat_priv->tvlv.container_list_lock); |
322 | tvlv_value_len = batadv_tvlv_container_list_size(bat_priv); |
323 | |
324 | ret = batadv_tvlv_realloc_packet_buff(packet_buff, packet_buff_len, |
325 | min_packet_len: packet_min_len, additional_packet_len: tvlv_value_len); |
326 | |
327 | if (!ret) |
328 | goto end; |
329 | |
330 | if (!tvlv_value_len) |
331 | goto end; |
332 | |
333 | tvlv_value = (*packet_buff) + packet_min_len; |
334 | |
335 | hlist_for_each_entry(tvlv, &bat_priv->tvlv.container_list, list) { |
336 | tvlv_hdr = tvlv_value; |
337 | tvlv_hdr->type = tvlv->tvlv_hdr.type; |
338 | tvlv_hdr->version = tvlv->tvlv_hdr.version; |
339 | tvlv_hdr->len = tvlv->tvlv_hdr.len; |
340 | tvlv_value = tvlv_hdr + 1; |
341 | memcpy(tvlv_value, tvlv + 1, ntohs(tvlv->tvlv_hdr.len)); |
342 | tvlv_value = (u8 *)tvlv_value + ntohs(tvlv->tvlv_hdr.len); |
343 | } |
344 | |
345 | end: |
346 | spin_unlock_bh(lock: &bat_priv->tvlv.container_list_lock); |
347 | return tvlv_value_len; |
348 | } |
349 | |
350 | /** |
351 | * batadv_tvlv_call_handler() - parse the given tvlv buffer to call the |
352 | * appropriate handlers |
353 | * @bat_priv: the bat priv with all the soft interface information |
354 | * @tvlv_handler: tvlv callback function handling the tvlv content |
355 | * @packet_type: indicates for which packet type the TVLV handler is called |
356 | * @orig_node: orig node emitting the ogm packet |
357 | * @skb: the skb the TVLV handler is called for |
358 | * @tvlv_value: tvlv content |
359 | * @tvlv_value_len: tvlv content length |
360 | * |
361 | * Return: success if the handler was not found or the return value of the |
362 | * handler callback. |
363 | */ |
364 | static int batadv_tvlv_call_handler(struct batadv_priv *bat_priv, |
365 | struct batadv_tvlv_handler *tvlv_handler, |
366 | u8 packet_type, |
367 | struct batadv_orig_node *orig_node, |
368 | struct sk_buff *skb, void *tvlv_value, |
369 | u16 tvlv_value_len) |
370 | { |
371 | unsigned int tvlv_offset; |
372 | u8 *src, *dst; |
373 | |
374 | if (!tvlv_handler) |
375 | return NET_RX_SUCCESS; |
376 | |
377 | switch (packet_type) { |
378 | case BATADV_IV_OGM: |
379 | case BATADV_OGM2: |
380 | if (!tvlv_handler->ogm_handler) |
381 | return NET_RX_SUCCESS; |
382 | |
383 | if (!orig_node) |
384 | return NET_RX_SUCCESS; |
385 | |
386 | tvlv_handler->ogm_handler(bat_priv, orig_node, |
387 | BATADV_NO_FLAGS, |
388 | tvlv_value, tvlv_value_len); |
389 | tvlv_handler->flags |= BATADV_TVLV_HANDLER_OGM_CALLED; |
390 | break; |
391 | case BATADV_UNICAST_TVLV: |
392 | if (!skb) |
393 | return NET_RX_SUCCESS; |
394 | |
395 | if (!tvlv_handler->unicast_handler) |
396 | return NET_RX_SUCCESS; |
397 | |
398 | src = ((struct batadv_unicast_tvlv_packet *)skb->data)->src; |
399 | dst = ((struct batadv_unicast_tvlv_packet *)skb->data)->dst; |
400 | |
401 | return tvlv_handler->unicast_handler(bat_priv, src, |
402 | dst, tvlv_value, |
403 | tvlv_value_len); |
404 | case BATADV_MCAST: |
405 | if (!skb) |
406 | return NET_RX_SUCCESS; |
407 | |
408 | if (!tvlv_handler->mcast_handler) |
409 | return NET_RX_SUCCESS; |
410 | |
411 | tvlv_offset = (unsigned char *)tvlv_value - skb->data; |
412 | skb_set_network_header(skb, offset: tvlv_offset); |
413 | skb_set_transport_header(skb, offset: tvlv_offset + tvlv_value_len); |
414 | |
415 | return tvlv_handler->mcast_handler(bat_priv, skb); |
416 | } |
417 | |
418 | return NET_RX_SUCCESS; |
419 | } |
420 | |
421 | /** |
422 | * batadv_tvlv_containers_process() - parse the given tvlv buffer to call the |
423 | * appropriate handlers |
424 | * @bat_priv: the bat priv with all the soft interface information |
425 | * @packet_type: indicates for which packet type the TVLV handler is called |
426 | * @orig_node: orig node emitting the ogm packet |
427 | * @skb: the skb the TVLV handler is called for |
428 | * @tvlv_value: tvlv content |
429 | * @tvlv_value_len: tvlv content length |
430 | * |
431 | * Return: success when processing an OGM or the return value of all called |
432 | * handler callbacks. |
433 | */ |
434 | int batadv_tvlv_containers_process(struct batadv_priv *bat_priv, |
435 | u8 packet_type, |
436 | struct batadv_orig_node *orig_node, |
437 | struct sk_buff *skb, void *tvlv_value, |
438 | u16 tvlv_value_len) |
439 | { |
440 | struct batadv_tvlv_handler *tvlv_handler; |
441 | struct batadv_tvlv_hdr *tvlv_hdr; |
442 | u16 tvlv_value_cont_len; |
443 | u8 cifnotfound = BATADV_TVLV_HANDLER_OGM_CIFNOTFND; |
444 | int ret = NET_RX_SUCCESS; |
445 | |
446 | while (tvlv_value_len >= sizeof(*tvlv_hdr)) { |
447 | tvlv_hdr = tvlv_value; |
448 | tvlv_value_cont_len = ntohs(tvlv_hdr->len); |
449 | tvlv_value = tvlv_hdr + 1; |
450 | tvlv_value_len -= sizeof(*tvlv_hdr); |
451 | |
452 | if (tvlv_value_cont_len > tvlv_value_len) |
453 | break; |
454 | |
455 | tvlv_handler = batadv_tvlv_handler_get(bat_priv, |
456 | type: tvlv_hdr->type, |
457 | version: tvlv_hdr->version); |
458 | |
459 | ret |= batadv_tvlv_call_handler(bat_priv, tvlv_handler, |
460 | packet_type, orig_node, skb, |
461 | tvlv_value, |
462 | tvlv_value_len: tvlv_value_cont_len); |
463 | batadv_tvlv_handler_put(tvlv_handler); |
464 | tvlv_value = (u8 *)tvlv_value + tvlv_value_cont_len; |
465 | tvlv_value_len -= tvlv_value_cont_len; |
466 | } |
467 | |
468 | if (packet_type != BATADV_IV_OGM && |
469 | packet_type != BATADV_OGM2) |
470 | return ret; |
471 | |
472 | rcu_read_lock(); |
473 | hlist_for_each_entry_rcu(tvlv_handler, |
474 | &bat_priv->tvlv.handler_list, list) { |
475 | if (!tvlv_handler->ogm_handler) |
476 | continue; |
477 | |
478 | if ((tvlv_handler->flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND) && |
479 | !(tvlv_handler->flags & BATADV_TVLV_HANDLER_OGM_CALLED)) |
480 | tvlv_handler->ogm_handler(bat_priv, orig_node, |
481 | cifnotfound, NULL, 0); |
482 | |
483 | tvlv_handler->flags &= ~BATADV_TVLV_HANDLER_OGM_CALLED; |
484 | } |
485 | rcu_read_unlock(); |
486 | |
487 | return NET_RX_SUCCESS; |
488 | } |
489 | |
490 | /** |
491 | * batadv_tvlv_ogm_receive() - process an incoming ogm and call the appropriate |
492 | * handlers |
493 | * @bat_priv: the bat priv with all the soft interface information |
494 | * @batadv_ogm_packet: ogm packet containing the tvlv containers |
495 | * @orig_node: orig node emitting the ogm packet |
496 | */ |
497 | void batadv_tvlv_ogm_receive(struct batadv_priv *bat_priv, |
498 | struct batadv_ogm_packet *batadv_ogm_packet, |
499 | struct batadv_orig_node *orig_node) |
500 | { |
501 | void *tvlv_value; |
502 | u16 tvlv_value_len; |
503 | |
504 | if (!batadv_ogm_packet) |
505 | return; |
506 | |
507 | tvlv_value_len = ntohs(batadv_ogm_packet->tvlv_len); |
508 | if (!tvlv_value_len) |
509 | return; |
510 | |
511 | tvlv_value = batadv_ogm_packet + 1; |
512 | |
513 | batadv_tvlv_containers_process(bat_priv, packet_type: BATADV_IV_OGM, orig_node, NULL, |
514 | tvlv_value, tvlv_value_len); |
515 | } |
516 | |
517 | /** |
518 | * batadv_tvlv_handler_register() - register tvlv handler based on the provided |
519 | * type and version (both need to match) for ogm tvlv payload and/or unicast |
520 | * payload |
521 | * @bat_priv: the bat priv with all the soft interface information |
522 | * @optr: ogm tvlv handler callback function. This function receives the orig |
523 | * node, flags and the tvlv content as argument to process. |
524 | * @uptr: unicast tvlv handler callback function. This function receives the |
525 | * source & destination of the unicast packet as well as the tvlv content |
526 | * to process. |
527 | * @mptr: multicast packet tvlv handler callback function. This function |
528 | * receives the full skb to process, with the skb network header pointing |
529 | * to the current tvlv and the skb transport header pointing to the first |
530 | * byte after the current tvlv. |
531 | * @type: tvlv handler type to be registered |
532 | * @version: tvlv handler version to be registered |
533 | * @flags: flags to enable or disable TVLV API behavior |
534 | */ |
535 | void batadv_tvlv_handler_register(struct batadv_priv *bat_priv, |
536 | void (*optr)(struct batadv_priv *bat_priv, |
537 | struct batadv_orig_node *orig, |
538 | u8 flags, |
539 | void *tvlv_value, |
540 | u16 tvlv_value_len), |
541 | int (*uptr)(struct batadv_priv *bat_priv, |
542 | u8 *src, u8 *dst, |
543 | void *tvlv_value, |
544 | u16 tvlv_value_len), |
545 | int (*mptr)(struct batadv_priv *bat_priv, |
546 | struct sk_buff *skb), |
547 | u8 type, u8 version, u8 flags) |
548 | { |
549 | struct batadv_tvlv_handler *tvlv_handler; |
550 | |
551 | spin_lock_bh(lock: &bat_priv->tvlv.handler_list_lock); |
552 | |
553 | tvlv_handler = batadv_tvlv_handler_get(bat_priv, type, version); |
554 | if (tvlv_handler) { |
555 | spin_unlock_bh(lock: &bat_priv->tvlv.handler_list_lock); |
556 | batadv_tvlv_handler_put(tvlv_handler); |
557 | return; |
558 | } |
559 | |
560 | tvlv_handler = kzalloc(size: sizeof(*tvlv_handler), GFP_ATOMIC); |
561 | if (!tvlv_handler) { |
562 | spin_unlock_bh(lock: &bat_priv->tvlv.handler_list_lock); |
563 | return; |
564 | } |
565 | |
566 | tvlv_handler->ogm_handler = optr; |
567 | tvlv_handler->unicast_handler = uptr; |
568 | tvlv_handler->mcast_handler = mptr; |
569 | tvlv_handler->type = type; |
570 | tvlv_handler->version = version; |
571 | tvlv_handler->flags = flags; |
572 | kref_init(kref: &tvlv_handler->refcount); |
573 | INIT_HLIST_NODE(h: &tvlv_handler->list); |
574 | |
575 | kref_get(kref: &tvlv_handler->refcount); |
576 | hlist_add_head_rcu(n: &tvlv_handler->list, h: &bat_priv->tvlv.handler_list); |
577 | spin_unlock_bh(lock: &bat_priv->tvlv.handler_list_lock); |
578 | |
579 | /* don't return reference to new tvlv_handler */ |
580 | batadv_tvlv_handler_put(tvlv_handler); |
581 | } |
582 | |
583 | /** |
584 | * batadv_tvlv_handler_unregister() - unregister tvlv handler based on the |
585 | * provided type and version (both need to match) |
586 | * @bat_priv: the bat priv with all the soft interface information |
587 | * @type: tvlv handler type to be unregistered |
588 | * @version: tvlv handler version to be unregistered |
589 | */ |
590 | void batadv_tvlv_handler_unregister(struct batadv_priv *bat_priv, |
591 | u8 type, u8 version) |
592 | { |
593 | struct batadv_tvlv_handler *tvlv_handler; |
594 | |
595 | tvlv_handler = batadv_tvlv_handler_get(bat_priv, type, version); |
596 | if (!tvlv_handler) |
597 | return; |
598 | |
599 | batadv_tvlv_handler_put(tvlv_handler); |
600 | spin_lock_bh(lock: &bat_priv->tvlv.handler_list_lock); |
601 | hlist_del_rcu(n: &tvlv_handler->list); |
602 | spin_unlock_bh(lock: &bat_priv->tvlv.handler_list_lock); |
603 | batadv_tvlv_handler_put(tvlv_handler); |
604 | } |
605 | |
606 | /** |
607 | * batadv_tvlv_unicast_send() - send a unicast packet with tvlv payload to the |
608 | * specified host |
609 | * @bat_priv: the bat priv with all the soft interface information |
610 | * @src: source mac address of the unicast packet |
611 | * @dst: destination mac address of the unicast packet |
612 | * @type: tvlv type |
613 | * @version: tvlv version |
614 | * @tvlv_value: tvlv content |
615 | * @tvlv_value_len: tvlv content length |
616 | */ |
617 | void batadv_tvlv_unicast_send(struct batadv_priv *bat_priv, const u8 *src, |
618 | const u8 *dst, u8 type, u8 version, |
619 | void *tvlv_value, u16 tvlv_value_len) |
620 | { |
621 | struct batadv_unicast_tvlv_packet *unicast_tvlv_packet; |
622 | struct batadv_tvlv_hdr *tvlv_hdr; |
623 | struct batadv_orig_node *orig_node; |
624 | struct sk_buff *skb; |
625 | unsigned char *tvlv_buff; |
626 | unsigned int tvlv_len; |
627 | ssize_t hdr_len = sizeof(*unicast_tvlv_packet); |
628 | |
629 | orig_node = batadv_orig_hash_find(bat_priv, data: dst); |
630 | if (!orig_node) |
631 | return; |
632 | |
633 | tvlv_len = sizeof(*tvlv_hdr) + tvlv_value_len; |
634 | |
635 | skb = netdev_alloc_skb_ip_align(NULL, ETH_HLEN + hdr_len + tvlv_len); |
636 | if (!skb) |
637 | goto out; |
638 | |
639 | skb->priority = TC_PRIO_CONTROL; |
640 | skb_reserve(skb, ETH_HLEN); |
641 | tvlv_buff = skb_put(skb, len: sizeof(*unicast_tvlv_packet) + tvlv_len); |
642 | unicast_tvlv_packet = (struct batadv_unicast_tvlv_packet *)tvlv_buff; |
643 | unicast_tvlv_packet->packet_type = BATADV_UNICAST_TVLV; |
644 | unicast_tvlv_packet->version = BATADV_COMPAT_VERSION; |
645 | unicast_tvlv_packet->ttl = BATADV_TTL; |
646 | unicast_tvlv_packet->reserved = 0; |
647 | unicast_tvlv_packet->tvlv_len = htons(tvlv_len); |
648 | unicast_tvlv_packet->align = 0; |
649 | ether_addr_copy(dst: unicast_tvlv_packet->src, src); |
650 | ether_addr_copy(dst: unicast_tvlv_packet->dst, src: dst); |
651 | |
652 | tvlv_buff = (unsigned char *)(unicast_tvlv_packet + 1); |
653 | tvlv_hdr = (struct batadv_tvlv_hdr *)tvlv_buff; |
654 | tvlv_hdr->version = version; |
655 | tvlv_hdr->type = type; |
656 | tvlv_hdr->len = htons(tvlv_value_len); |
657 | tvlv_buff += sizeof(*tvlv_hdr); |
658 | memcpy(tvlv_buff, tvlv_value, tvlv_value_len); |
659 | |
660 | batadv_send_skb_to_orig(skb, orig_node, NULL); |
661 | out: |
662 | batadv_orig_node_put(orig_node); |
663 | } |
664 | |