1 /* SPDX-License-Identifier: LGPL-2.1-only */
2 /*
3 * lib/cache.c Caching Module
4 *
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation version 2.1
8 * of the License.
9 *
10 * Copyright (c) 2003-2012 Thomas Graf <tgraf@suug.ch>
11 */
12
13 /**
14 * @ingroup cache_mngt
15 * @defgroup cache Cache
16 *
17 * @code
18 * Cache Management | | Type Specific Cache Operations
19 *
20 * | | +----------------+ +------------+
21 * | request update | | msg_parser |
22 * | | +----------------+ +------------+
23 * +- - - - -^- - - - - - - -^- -|- - - -
24 * nl_cache_update: | | | |
25 * 1) --------- co_request_update ------+ | |
26 * | | |
27 * 2) destroy old cache +----------- pp_cb ---------|---+
28 * | | |
29 * 3) ---------- nl_recvmsgs ----------+ +- cb_valid -+
30 * +--------------+ | | | |
31 * | nl_cache_add |<-----+ + - - -v- -|- - - - - - - - - - -
32 * +--------------+ | | +-------------+
33 * | nl_recvmsgs |
34 * | | +-----|-^-----+
35 * +---v-|---+
36 * | | | nl_recv |
37 * +---------+
38 * | | Core Netlink
39 * @endcode
40 *
41 * Related sections in the development guide:
42 * - @core_doc{core_cache, Caching System}
43 *
44 * @{
45 *
46 * Header
47 * ------
48 * ~~~~{.c}
49 * #include <netlink/cache.h>
50 * ~~~~
51 */
52
53 #include <netlink-private/netlink.h>
54 #include <netlink/netlink.h>
55 #include <netlink/cache.h>
56 #include <netlink/object.h>
57 #include <netlink/hashtable.h>
58 #include <netlink/utils.h>
59
60 /**
61 * @name Access Functions
62 * @{
63 */
64
65 /**
66 * Return the number of items in the cache
67 * @arg cache cache handle
68 */
nl_cache_nitems(struct nl_cache * cache)69 int nl_cache_nitems(struct nl_cache *cache)
70 {
71 return cache->c_nitems;
72 }
73
74 /**
75 * Return the number of items matching a filter in the cache
76 * @arg cache Cache object.
77 * @arg filter Filter object.
78 */
nl_cache_nitems_filter(struct nl_cache * cache,struct nl_object * filter)79 int nl_cache_nitems_filter(struct nl_cache *cache, struct nl_object *filter)
80 {
81 struct nl_object *obj;
82 int nitems = 0;
83
84 if (cache->c_ops == NULL)
85 BUG();
86
87 nl_list_for_each_entry(obj, &cache->c_items, ce_list) {
88 if (filter && !nl_object_match_filter(obj, filter))
89 continue;
90
91 nitems++;
92 }
93
94 return nitems;
95 }
96
97 /**
98 * Returns \b true if the cache is empty.
99 * @arg cache Cache to check
100 * @return \a true if the cache is empty, otherwise \b false is returned.
101 */
nl_cache_is_empty(struct nl_cache * cache)102 int nl_cache_is_empty(struct nl_cache *cache)
103 {
104 return nl_list_empty(&cache->c_items);
105 }
106
107 /**
108 * Return the operations set of the cache
109 * @arg cache cache handle
110 */
nl_cache_get_ops(struct nl_cache * cache)111 struct nl_cache_ops *nl_cache_get_ops(struct nl_cache *cache)
112 {
113 return cache->c_ops;
114 }
115
116 /**
117 * Return the first element in the cache
118 * @arg cache cache handle
119 */
nl_cache_get_first(struct nl_cache * cache)120 struct nl_object *nl_cache_get_first(struct nl_cache *cache)
121 {
122 if (nl_list_empty(&cache->c_items))
123 return NULL;
124
125 return nl_list_entry(cache->c_items.next,
126 struct nl_object, ce_list);
127 }
128
129 /**
130 * Return the last element in the cache
131 * @arg cache cache handle
132 */
nl_cache_get_last(struct nl_cache * cache)133 struct nl_object *nl_cache_get_last(struct nl_cache *cache)
134 {
135 if (nl_list_empty(&cache->c_items))
136 return NULL;
137
138 return nl_list_entry(cache->c_items.prev,
139 struct nl_object, ce_list);
140 }
141
142 /**
143 * Return the next element in the cache
144 * @arg obj current object
145 */
nl_cache_get_next(struct nl_object * obj)146 struct nl_object *nl_cache_get_next(struct nl_object *obj)
147 {
148 if (nl_list_at_tail(obj, &obj->ce_cache->c_items, ce_list))
149 return NULL;
150 else
151 return nl_list_entry(obj->ce_list.next,
152 struct nl_object, ce_list);
153 }
154
155 /**
156 * Return the previous element in the cache
157 * @arg obj current object
158 */
nl_cache_get_prev(struct nl_object * obj)159 struct nl_object *nl_cache_get_prev(struct nl_object *obj)
160 {
161 if (nl_list_at_head(obj, &obj->ce_cache->c_items, ce_list))
162 return NULL;
163 else
164 return nl_list_entry(obj->ce_list.prev,
165 struct nl_object, ce_list);
166 }
167
168 /** @} */
169
170 /**
171 * @name Cache Allocation/Deletion
172 * @{
173 */
174
175 /**
176 * Allocate new cache
177 * @arg ops Cache operations
178 *
179 * Allocate and initialize a new cache based on the cache operations
180 * provided.
181 *
182 * @return Allocated cache or NULL if allocation failed.
183 */
nl_cache_alloc(struct nl_cache_ops * ops)184 struct nl_cache *nl_cache_alloc(struct nl_cache_ops *ops)
185 {
186 struct nl_cache *cache;
187
188 cache = calloc(1, sizeof(*cache));
189 if (!cache)
190 return NULL;
191
192 nl_init_list_head(&cache->c_items);
193 cache->c_ops = ops;
194 cache->c_flags |= ops->co_flags;
195 cache->c_refcnt = 1;
196
197 /*
198 * If object type provides a hash keygen
199 * functions, allocate a hash table for the
200 * cache objects for faster lookups
201 */
202 if (ops->co_obj_ops->oo_keygen) {
203 int hashtable_size;
204
205 if (ops->co_hash_size)
206 hashtable_size = ops->co_hash_size;
207 else
208 hashtable_size = NL_MAX_HASH_ENTRIES;
209
210 cache->hashtable = nl_hash_table_alloc(hashtable_size);
211 }
212
213 NL_DBG(2, "Allocated cache %p <%s>.\n", cache, nl_cache_name(cache));
214
215 return cache;
216 }
217
218 /**
219 * Allocate new cache and fill it
220 * @arg ops Cache operations
221 * @arg sock Netlink socket
222 * @arg result Result pointer
223 *
224 * Allocate new cache and fill it. Equivalent to calling:
225 * @code
226 * cache = nl_cache_alloc(ops);
227 * nl_cache_refill(sock, cache);
228 * @endcode
229 *
230 * @see nl_cache_alloc
231 *
232 * @return 0 on success or a negative error code.
233 */
nl_cache_alloc_and_fill(struct nl_cache_ops * ops,struct nl_sock * sock,struct nl_cache ** result)234 int nl_cache_alloc_and_fill(struct nl_cache_ops *ops, struct nl_sock *sock,
235 struct nl_cache **result)
236 {
237 struct nl_cache *cache;
238 int err;
239
240 if (!(cache = nl_cache_alloc(ops)))
241 return -NLE_NOMEM;
242
243 if (sock && (err = nl_cache_refill(sock, cache)) < 0) {
244 nl_cache_free(cache);
245 return err;
246 }
247
248 *result = cache;
249 return 0;
250 }
251
252 /**
253 * Allocate new cache based on type name
254 * @arg kind Name of cache type
255 * @arg result Result pointer
256 *
257 * Lookup cache ops via nl_cache_ops_lookup() and allocate the cache
258 * by calling nl_cache_alloc(). Stores the allocated cache in the
259 * result pointer provided.
260 *
261 * @see nl_cache_alloc
262 *
263 * @return 0 on success or a negative error code.
264 */
nl_cache_alloc_name(const char * kind,struct nl_cache ** result)265 int nl_cache_alloc_name(const char *kind, struct nl_cache **result)
266 {
267 struct nl_cache_ops *ops;
268 struct nl_cache *cache;
269
270 ops = nl_cache_ops_lookup_safe(kind);
271 if (!ops)
272 return -NLE_NOCACHE;
273
274 cache = nl_cache_alloc(ops);
275 nl_cache_ops_put(ops);
276 if (!cache)
277 return -NLE_NOMEM;
278
279 *result = cache;
280 return 0;
281 }
282
283 /**
284 * Allocate new cache containing a subset of an existing cache
285 * @arg orig Original cache to base new cache on
286 * @arg filter Filter defining the subset to be filled into the new cache
287 *
288 * Allocates a new cache matching the type of the cache specified by
289 * \p orig. Iterates over the \p orig cache applying the specified
290 * \p filter and copies all objects that match to the new cache.
291 *
292 * The copied objects are clones but do not contain a reference to each
293 * other. Later modifications to objects in the original cache will
294 * not affect objects in the new cache.
295 *
296 * @return A newly allocated cache or NULL.
297 */
nl_cache_subset(struct nl_cache * orig,struct nl_object * filter)298 struct nl_cache *nl_cache_subset(struct nl_cache *orig,
299 struct nl_object *filter)
300 {
301 struct nl_cache *cache;
302 struct nl_object *obj;
303
304 if (!filter)
305 BUG();
306
307 cache = nl_cache_alloc(orig->c_ops);
308 if (!cache)
309 return NULL;
310
311 NL_DBG(2, "Filling subset of cache %p <%s> with filter %p into %p\n",
312 orig, nl_cache_name(orig), filter, cache);
313
314 nl_list_for_each_entry(obj, &orig->c_items, ce_list) {
315 if (!nl_object_match_filter(obj, filter))
316 continue;
317
318 nl_cache_add(cache, obj);
319 }
320
321 return cache;
322 }
323
324 /**
325 * Allocate new cache and copy the contents of an existing cache
326 * @arg cache Original cache to base new cache on
327 *
328 * Allocates a new cache matching the type of the cache specified by
329 * \p cache. Iterates over the \p cache cache and copies all objects
330 * to the new cache.
331 *
332 * The copied objects are clones but do not contain a reference to each
333 * other. Later modifications to objects in the original cache will
334 * not affect objects in the new cache.
335 *
336 * @return A newly allocated cache or NULL.
337 */
nl_cache_clone(struct nl_cache * cache)338 struct nl_cache *nl_cache_clone(struct nl_cache *cache)
339 {
340 struct nl_cache_ops *ops = nl_cache_get_ops(cache);
341 struct nl_cache *clone;
342 struct nl_object *obj;
343
344 clone = nl_cache_alloc(ops);
345 if (!clone)
346 return NULL;
347
348 NL_DBG(2, "Cloning %p into %p\n", cache, clone);
349
350 nl_list_for_each_entry(obj, &cache->c_items, ce_list)
351 nl_cache_add(clone, obj);
352
353 return clone;
354 }
355
356 /**
357 * Remove all objects of a cache.
358 * @arg cache Cache to clear
359 *
360 * The objects are unliked/removed from the cache by calling
361 * nl_cache_remove() on each object in the cache. If any of the objects
362 * to not contain any further references to them, those objects will
363 * be freed.
364 *
365 * Unlike with nl_cache_free(), the cache is not freed just emptied.
366 */
nl_cache_clear(struct nl_cache * cache)367 void nl_cache_clear(struct nl_cache *cache)
368 {
369 struct nl_object *obj, *tmp;
370
371 NL_DBG(2, "Clearing cache %p <%s>...\n", cache, nl_cache_name(cache));
372
373 nl_list_for_each_entry_safe(obj, tmp, &cache->c_items, ce_list)
374 nl_cache_remove(obj);
375 }
376
__nl_cache_free(struct nl_cache * cache)377 static void __nl_cache_free(struct nl_cache *cache)
378 {
379 nl_cache_clear(cache);
380
381 if (cache->hashtable)
382 nl_hash_table_free(cache->hashtable);
383
384 NL_DBG(2, "Freeing cache %p <%s>...\n", cache, nl_cache_name(cache));
385 free(cache);
386 }
387
388 /**
389 * Increase reference counter of cache
390 * @arg cache Cache
391 */
nl_cache_get(struct nl_cache * cache)392 void nl_cache_get(struct nl_cache *cache)
393 {
394 cache->c_refcnt++;
395
396 NL_DBG(3, "Incremented cache %p <%s> reference count to %d\n",
397 cache, nl_cache_name(cache), cache->c_refcnt);
398 }
399
400 /**
401 * Free a cache.
402 * @arg cache Cache to free.
403 *
404 * Calls nl_cache_clear() to remove all objects associated with the
405 * cache and frees the cache afterwards.
406 *
407 * @see nl_cache_clear()
408 */
nl_cache_free(struct nl_cache * cache)409 void nl_cache_free(struct nl_cache *cache)
410 {
411 if (!cache)
412 return;
413
414 cache->c_refcnt--;
415
416 NL_DBG(3, "Decremented cache %p <%s> reference count, %d remaining\n",
417 cache, nl_cache_name(cache), cache->c_refcnt);
418
419 if (cache->c_refcnt <= 0)
420 __nl_cache_free(cache);
421 }
422
nl_cache_put(struct nl_cache * cache)423 void nl_cache_put(struct nl_cache *cache)
424 {
425 nl_cache_free(cache);
426 }
427
428 /** @} */
429
430 /**
431 * @name Cache Modifications
432 * @{
433 */
434
__cache_add(struct nl_cache * cache,struct nl_object * obj)435 static int __cache_add(struct nl_cache *cache, struct nl_object *obj)
436 {
437 int ret;
438
439 obj->ce_cache = cache;
440
441 if (cache->hashtable) {
442 ret = nl_hash_table_add(cache->hashtable, obj);
443 if (ret < 0) {
444 obj->ce_cache = NULL;
445 return ret;
446 }
447 }
448
449 nl_list_add_tail(&obj->ce_list, &cache->c_items);
450 cache->c_nitems++;
451
452 NL_DBG(3, "Added object %p to cache %p <%s>, nitems %d\n",
453 obj, cache, nl_cache_name(cache), cache->c_nitems);
454
455 return 0;
456 }
457
458 /**
459 * Add object to cache.
460 * @arg cache Cache
461 * @arg obj Object to be added to the cache
462 *
463 * Adds the object \p obj to the specified \p cache. In case the object
464 * is already associated with another cache, the object is cloned before
465 * adding it to the cache. In this case, the sole reference to the object
466 * will be the one of the cache. Therefore clearing/freeing the cache
467 * will result in the object being freed again.
468 *
469 * If the object has not been associated with a cache yet, the reference
470 * counter of the object is incremented to account for the additional
471 * reference.
472 *
473 * The type of the object and cache must match, otherwise an error is
474 * returned (-NLE_OBJ_MISMATCH).
475 *
476 * @see nl_cache_move()
477 *
478 * @return 0 or a negative error code.
479 */
nl_cache_add(struct nl_cache * cache,struct nl_object * obj)480 int nl_cache_add(struct nl_cache *cache, struct nl_object *obj)
481 {
482 struct nl_object *new;
483 int ret = 0;
484
485 if (cache->c_ops->co_obj_ops != obj->ce_ops)
486 return -NLE_OBJ_MISMATCH;
487
488 if (!nl_list_empty(&obj->ce_list)) {
489 NL_DBG(3, "Object %p already in cache, cloning new object\n", obj);
490
491 new = nl_object_clone(obj);
492 if (!new)
493 return -NLE_NOMEM;
494 } else {
495 nl_object_get(obj);
496 new = obj;
497 }
498
499 ret = __cache_add(cache, new);
500 if (ret < 0)
501 nl_object_put(new);
502
503 return ret;
504 }
505
506 /**
507 * Move object from one cache to another
508 * @arg cache Cache to move object to.
509 * @arg obj Object subject to be moved
510 *
511 * Removes the the specified object \p obj from its associated cache
512 * and moves it to another cache.
513 *
514 * If the object is not associated with a cache, the function behaves
515 * just like nl_cache_add().
516 *
517 * The type of the object and cache must match, otherwise an error is
518 * returned (-NLE_OBJ_MISMATCH).
519 *
520 * @see nl_cache_add()
521 *
522 * @return 0 on success or a negative error code.
523 */
nl_cache_move(struct nl_cache * cache,struct nl_object * obj)524 int nl_cache_move(struct nl_cache *cache, struct nl_object *obj)
525 {
526 if (cache->c_ops->co_obj_ops != obj->ce_ops)
527 return -NLE_OBJ_MISMATCH;
528
529 NL_DBG(3, "Moving object %p from cache %p to cache %p\n",
530 obj, obj->ce_cache, cache);
531
532 /* Acquire reference, if already in a cache this will be
533 * reverted during removal */
534 nl_object_get(obj);
535
536 if (!nl_list_empty(&obj->ce_list))
537 nl_cache_remove(obj);
538
539 return __cache_add(cache, obj);
540 }
541
542 /**
543 * Remove object from cache.
544 * @arg obj Object to remove from cache
545 *
546 * Removes the object \c obj from the cache it is associated with. The
547 * reference counter of the object will be decremented. If the reference
548 * to the object was the only one remaining, the object will be freed.
549 *
550 * If no cache is associated with the object, this function is a NOP.
551 */
nl_cache_remove(struct nl_object * obj)552 void nl_cache_remove(struct nl_object *obj)
553 {
554 int ret;
555 struct nl_cache *cache = obj->ce_cache;
556
557 if (cache == NULL)
558 return;
559
560 if (cache->hashtable) {
561 ret = nl_hash_table_del(cache->hashtable, obj);
562 if (ret < 0)
563 NL_DBG(2, "Failed to delete %p from cache %p <%s>.\n",
564 obj, cache, nl_cache_name(cache));
565 }
566
567 nl_list_del(&obj->ce_list);
568 obj->ce_cache = NULL;
569 nl_object_put(obj);
570 cache->c_nitems--;
571
572 NL_DBG(2, "Deleted object %p from cache %p <%s>.\n",
573 obj, cache, nl_cache_name(cache));
574 }
575
576 /** @} */
577
578 /**
579 * @name Synchronization
580 * @{
581 */
582
583 /**
584 * Set synchronization arg1 of cache
585 * @arg cache Cache
586 * @arg arg argument
587 *
588 * Synchronization arguments are used to specify filters when
589 * requesting dumps from the kernel.
590 */
nl_cache_set_arg1(struct nl_cache * cache,int arg)591 void nl_cache_set_arg1(struct nl_cache *cache, int arg)
592 {
593 cache->c_iarg1 = arg;
594 }
595
596 /**
597 * Set synchronization arg2 of cache
598 * @arg cache Cache
599 * @arg arg argument
600 *
601 * Synchronization arguments are used to specify filters when
602 * requesting dumps from the kernel.
603 */
nl_cache_set_arg2(struct nl_cache * cache,int arg)604 void nl_cache_set_arg2(struct nl_cache *cache, int arg)
605 {
606 cache->c_iarg2 = arg;
607 }
608
609 /**
610 * Set cache flags
611 * @arg cache Cache
612 * @arg flags Flags
613 */
nl_cache_set_flags(struct nl_cache * cache,unsigned int flags)614 void nl_cache_set_flags(struct nl_cache *cache, unsigned int flags)
615 {
616 cache->c_flags |= flags;
617 }
618
619 /**
620 * Invoke the request-update operation
621 * @arg sk Netlink socket.
622 * @arg cache Cache
623 *
624 * This function causes the \e request-update function of the cache
625 * operations to be invoked. This usually causes a dump request to
626 * be sent over the netlink socket which triggers the kernel to dump
627 * all objects of a specific type to be dumped onto the netlink
628 * socket for pickup.
629 *
630 * The behaviour of this function depends on the implemenation of
631 * the \e request_update function of each individual type of cache.
632 *
633 * This function will not have any effects on the cache (unless the
634 * request_update implementation of the cache operations does so).
635 *
636 * Use nl_cache_pickup() to pick-up (read) the objects from the socket
637 * and fill them into the cache.
638 *
639 * @see nl_cache_pickup(), nl_cache_resync()
640 *
641 * @return 0 on success or a negative error code. Some implementations
642 * of co_request_update() return a positive number on success that is
643 * the number of bytes sent. Treat any non-negative number as success too.
644 */
nl_cache_request_full_dump(struct nl_sock * sk,struct nl_cache * cache)645 static int nl_cache_request_full_dump(struct nl_sock *sk,
646 struct nl_cache *cache)
647 {
648 if (sk->s_proto != cache->c_ops->co_protocol)
649 return -NLE_PROTO_MISMATCH;
650
651 if (cache->c_ops->co_request_update == NULL)
652 return -NLE_OPNOTSUPP;
653
654 NL_DBG(2, "Requesting update from kernel for cache %p <%s>\n",
655 cache, nl_cache_name(cache));
656
657 return cache->c_ops->co_request_update(cache, sk);
658 }
659
660 /** @cond SKIP */
661 struct update_xdata {
662 struct nl_cache_ops *ops;
663 struct nl_parser_param *params;
664 };
665
update_msg_parser(struct nl_msg * msg,void * arg)666 static int update_msg_parser(struct nl_msg *msg, void *arg)
667 {
668 struct update_xdata *x = arg;
669 int ret = 0;
670
671 ret = nl_cache_parse(x->ops, &msg->nm_src, msg->nm_nlh, x->params);
672 if (ret == -NLE_EXIST)
673 return NL_SKIP;
674 else
675 return ret;
676 }
677 /** @endcond */
678
679 /**
680 * Pick-up a netlink request-update with your own parser
681 * @arg sk Netlink socket
682 * @arg cache Cache
683 * @arg param Parser parameters
684 */
__cache_pickup(struct nl_sock * sk,struct nl_cache * cache,struct nl_parser_param * param)685 static int __cache_pickup(struct nl_sock *sk, struct nl_cache *cache,
686 struct nl_parser_param *param)
687 {
688 int err;
689 struct nl_cb *cb;
690 struct update_xdata x = {
691 .ops = cache->c_ops,
692 .params = param,
693 };
694
695 NL_DBG(2, "Picking up answer for cache %p <%s>\n",
696 cache, nl_cache_name(cache));
697
698 cb = nl_cb_clone(sk->s_cb);
699 if (cb == NULL)
700 return -NLE_NOMEM;
701
702 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, update_msg_parser, &x);
703
704 err = nl_recvmsgs(sk, cb);
705 if (err < 0)
706 NL_DBG(2, "While picking up for %p <%s>, recvmsgs() returned %d: %s\n",
707 cache, nl_cache_name(cache), err, nl_geterror(err));
708
709 nl_cb_put(cb);
710
711 return err;
712 }
713
pickup_checkdup_cb(struct nl_object * c,struct nl_parser_param * p)714 static int pickup_checkdup_cb(struct nl_object *c, struct nl_parser_param *p)
715 {
716 struct nl_cache *cache = (struct nl_cache *)p->pp_arg;
717 struct nl_object *old;
718
719 old = nl_cache_search(cache, c);
720 if (old) {
721 if (nl_object_update(old, c) == 0) {
722 nl_object_put(old);
723 return 0;
724 }
725
726 nl_cache_remove(old);
727 nl_object_put(old);
728 }
729
730 return nl_cache_add(cache, c);
731 }
732
pickup_cb(struct nl_object * c,struct nl_parser_param * p)733 static int pickup_cb(struct nl_object *c, struct nl_parser_param *p)
734 {
735 struct nl_cache *cache = p->pp_arg;
736
737 return nl_cache_add(cache, c);
738 }
739
__nl_cache_pickup(struct nl_sock * sk,struct nl_cache * cache,int checkdup)740 static int __nl_cache_pickup(struct nl_sock *sk, struct nl_cache *cache,
741 int checkdup)
742 {
743 struct nl_parser_param p;
744
745 p.pp_cb = checkdup ? pickup_checkdup_cb : pickup_cb;
746 p.pp_arg = cache;
747
748 if (sk->s_proto != cache->c_ops->co_protocol)
749 return -NLE_PROTO_MISMATCH;
750
751 return __cache_pickup(sk, cache, &p);
752 }
753
754 /**
755 * Pickup a netlink dump response and put it into a cache.
756 * @arg sk Netlink socket.
757 * @arg cache Cache to put items into.
758 *
759 * Waits for netlink messages to arrive, parses them and puts them into
760 * the specified cache.
761 *
762 * @return 0 on success or a negative error code.
763 */
nl_cache_pickup_checkdup(struct nl_sock * sk,struct nl_cache * cache)764 int nl_cache_pickup_checkdup(struct nl_sock *sk, struct nl_cache *cache)
765 {
766 return __nl_cache_pickup(sk, cache, 1);
767 }
768
769 /**
770 * Pickup a netlink dump response and put it into a cache.
771 * @arg sk Netlink socket.
772 * @arg cache Cache to put items into.
773 *
774 * Waits for netlink messages to arrive, parses them and puts them into
775 * the specified cache. If an old object with same key attributes is
776 * present in the cache, it is replaced with the new object.
777 * If the old object type supports an update operation, an update is
778 * attempted before a replace.
779 *
780 * @return 0 on success or a negative error code.
781 */
nl_cache_pickup(struct nl_sock * sk,struct nl_cache * cache)782 int nl_cache_pickup(struct nl_sock *sk, struct nl_cache *cache)
783 {
784 return __nl_cache_pickup(sk, cache, 0);
785 }
786
cache_include(struct nl_cache * cache,struct nl_object * obj,struct nl_msgtype * type,change_func_t cb,change_func_v2_t cb_v2,void * data)787 static int cache_include(struct nl_cache *cache, struct nl_object *obj,
788 struct nl_msgtype *type, change_func_t cb,
789 change_func_v2_t cb_v2, void *data)
790 {
791 struct nl_object *old;
792 struct nl_object *clone = NULL;
793 uint64_t diff = 0;
794
795 switch (type->mt_act) {
796 case NL_ACT_NEW:
797 case NL_ACT_DEL:
798 old = nl_cache_search(cache, obj);
799 if (old) {
800 if (cb_v2 && old->ce_ops->oo_update) {
801 clone = nl_object_clone(old);
802 diff = nl_object_diff64(old, obj);
803 }
804 /*
805 * Some objects types might support merging the new
806 * object with the old existing cache object.
807 * Handle them first.
808 */
809 if (nl_object_update(old, obj) == 0) {
810 if (cb_v2) {
811 cb_v2(cache, clone, obj, diff,
812 NL_ACT_CHANGE, data);
813 nl_object_put(clone);
814 } else if (cb)
815 cb(cache, old, NL_ACT_CHANGE, data);
816 nl_object_put(old);
817 return 0;
818 }
819 nl_object_put(clone);
820
821 nl_cache_remove(old);
822 if (type->mt_act == NL_ACT_DEL) {
823 if (cb_v2)
824 cb_v2(cache, old, NULL, 0, NL_ACT_DEL,
825 data);
826 else if (cb)
827 cb(cache, old, NL_ACT_DEL, data);
828 nl_object_put(old);
829 }
830 }
831
832 if (type->mt_act == NL_ACT_NEW) {
833 nl_cache_move(cache, obj);
834 if (old == NULL) {
835 if (cb_v2) {
836 cb_v2(cache, NULL, obj, 0, NL_ACT_NEW,
837 data);
838 } else if (cb)
839 cb(cache, obj, NL_ACT_NEW, data);
840 } else if (old) {
841 diff = 0;
842 if (cb || cb_v2)
843 diff = nl_object_diff64(old, obj);
844 if (diff && cb_v2) {
845 cb_v2(cache, old, obj, diff, NL_ACT_CHANGE,
846 data);
847 } else if (diff && cb)
848 cb(cache, obj, NL_ACT_CHANGE, data);
849
850 nl_object_put(old);
851 }
852 }
853 break;
854 default:
855 NL_DBG(2, "Unknown action associated to object %p\n", obj);
856 return 0;
857 }
858
859 return 0;
860 }
861
nl_cache_include(struct nl_cache * cache,struct nl_object * obj,change_func_t change_cb,void * data)862 int nl_cache_include(struct nl_cache *cache, struct nl_object *obj,
863 change_func_t change_cb, void *data)
864 {
865 struct nl_cache_ops *ops = cache->c_ops;
866 int i;
867
868 if (ops->co_obj_ops != obj->ce_ops)
869 return -NLE_OBJ_MISMATCH;
870
871 for (i = 0; ops->co_msgtypes[i].mt_id >= 0; i++)
872 if (ops->co_msgtypes[i].mt_id == obj->ce_msgtype)
873 return cache_include(cache, obj, &ops->co_msgtypes[i],
874 change_cb, NULL, data);
875
876 NL_DBG(3, "Object %p does not seem to belong to cache %p <%s>\n",
877 obj, cache, nl_cache_name(cache));
878
879 return -NLE_MSGTYPE_NOSUPPORT;
880 }
881
nl_cache_include_v2(struct nl_cache * cache,struct nl_object * obj,change_func_v2_t change_cb,void * data)882 int nl_cache_include_v2(struct nl_cache *cache, struct nl_object *obj,
883 change_func_v2_t change_cb, void *data)
884 {
885 struct nl_cache_ops *ops = cache->c_ops;
886 int i;
887
888 if (ops->co_obj_ops != obj->ce_ops)
889 return -NLE_OBJ_MISMATCH;
890
891 for (i = 0; ops->co_msgtypes[i].mt_id >= 0; i++)
892 if (ops->co_msgtypes[i].mt_id == obj->ce_msgtype)
893 return cache_include(cache, obj, &ops->co_msgtypes[i],
894 NULL, change_cb, data);
895
896 NL_DBG(3, "Object %p does not seem to belong to cache %p <%s>\n",
897 obj, cache, nl_cache_name(cache));
898
899 return -NLE_MSGTYPE_NOSUPPORT;
900 }
901
resync_cb(struct nl_object * c,struct nl_parser_param * p)902 static int resync_cb(struct nl_object *c, struct nl_parser_param *p)
903 {
904 struct nl_cache_assoc *ca = p->pp_arg;
905
906 if (ca->ca_change_v2)
907 return nl_cache_include_v2(ca->ca_cache, c, ca->ca_change_v2,
908 ca->ca_change_data);
909 else
910 return nl_cache_include(ca->ca_cache, c, ca->ca_change,
911 ca->ca_change_data);
912 }
913
nl_cache_resync(struct nl_sock * sk,struct nl_cache * cache,change_func_t change_cb,void * data)914 int nl_cache_resync(struct nl_sock *sk, struct nl_cache *cache,
915 change_func_t change_cb, void *data)
916 {
917 struct nl_object *obj, *next;
918 struct nl_af_group *grp;
919 struct nl_cache_assoc ca = {
920 .ca_cache = cache,
921 .ca_change = change_cb,
922 .ca_change_data = data,
923 };
924 struct nl_parser_param p = {
925 .pp_cb = resync_cb,
926 .pp_arg = &ca,
927 };
928 int err;
929
930 if (sk->s_proto != cache->c_ops->co_protocol)
931 return -NLE_PROTO_MISMATCH;
932
933 NL_DBG(1, "Resyncing cache %p <%s>...\n", cache, nl_cache_name(cache));
934
935 /* Mark all objects so we can see if some of them are obsolete */
936 nl_cache_mark_all(cache);
937
938 grp = cache->c_ops->co_groups;
939 do {
940 if (grp && grp->ag_group &&
941 (cache->c_flags & NL_CACHE_AF_ITER))
942 nl_cache_set_arg1(cache, grp->ag_family);
943
944 restart:
945 err = nl_cache_request_full_dump(sk, cache);
946 if (err < 0)
947 goto errout;
948
949 err = __cache_pickup(sk, cache, &p);
950 if (err == -NLE_DUMP_INTR)
951 goto restart;
952 else if (err < 0)
953 goto errout;
954
955 if (grp)
956 grp++;
957 } while (grp && grp->ag_group &&
958 (cache->c_flags & NL_CACHE_AF_ITER));
959
960 nl_list_for_each_entry_safe(obj, next, &cache->c_items, ce_list) {
961 if (nl_object_is_marked(obj)) {
962 nl_object_get(obj);
963 nl_cache_remove(obj);
964 if (change_cb)
965 change_cb(cache, obj, NL_ACT_DEL, data);
966 nl_object_put(obj);
967 }
968 }
969
970 NL_DBG(1, "Finished resyncing %p <%s>\n", cache, nl_cache_name(cache));
971
972 err = 0;
973 errout:
974 return err;
975 }
976
977 /** @} */
978
979 /**
980 * @name Parsing
981 * @{
982 */
983
984 /** @cond SKIP */
nl_cache_parse(struct nl_cache_ops * ops,struct sockaddr_nl * who,struct nlmsghdr * nlh,struct nl_parser_param * params)985 int nl_cache_parse(struct nl_cache_ops *ops, struct sockaddr_nl *who,
986 struct nlmsghdr *nlh, struct nl_parser_param *params)
987 {
988 int i, err;
989
990 if (!nlmsg_valid_hdr(nlh, ops->co_hdrsize))
991 return -NLE_MSG_TOOSHORT;
992
993 for (i = 0; ops->co_msgtypes[i].mt_id >= 0; i++) {
994 if (ops->co_msgtypes[i].mt_id == nlh->nlmsg_type) {
995 err = ops->co_msg_parser(ops, who, nlh, params);
996 if (err != -NLE_OPNOTSUPP)
997 goto errout;
998 }
999 }
1000
1001
1002 err = -NLE_MSGTYPE_NOSUPPORT;
1003 errout:
1004 return err;
1005 }
1006 /** @endcond */
1007
1008 /**
1009 * Parse a netlink message and add it to the cache.
1010 * @arg cache cache to add element to
1011 * @arg msg netlink message
1012 *
1013 * Parses a netlink message by calling the cache specific message parser
1014 * and adds the new element to the cache. If an old object with same key
1015 * attributes is present in the cache, it is replaced with the new object.
1016 * If the old object type supports an update operation, an update is
1017 * attempted before a replace.
1018 *
1019 * @return 0 or a negative error code.
1020 */
nl_cache_parse_and_add(struct nl_cache * cache,struct nl_msg * msg)1021 int nl_cache_parse_and_add(struct nl_cache *cache, struct nl_msg *msg)
1022 {
1023 struct nl_parser_param p = {
1024 .pp_cb = pickup_cb,
1025 .pp_arg = cache,
1026 };
1027
1028 return nl_cache_parse(cache->c_ops, NULL, nlmsg_hdr(msg), &p);
1029 }
1030
1031 /**
1032 * (Re)fill a cache with the contents in the kernel.
1033 * @arg sk Netlink socket.
1034 * @arg cache cache to update
1035 *
1036 * Clears the specified cache and fills it with the current state in
1037 * the kernel.
1038 *
1039 * @return 0 or a negative error code.
1040 */
nl_cache_refill(struct nl_sock * sk,struct nl_cache * cache)1041 int nl_cache_refill(struct nl_sock *sk, struct nl_cache *cache)
1042 {
1043 struct nl_af_group *grp;
1044 int err;
1045
1046 if (sk->s_proto != cache->c_ops->co_protocol)
1047 return -NLE_PROTO_MISMATCH;
1048
1049 nl_cache_clear(cache);
1050 grp = cache->c_ops->co_groups;
1051 do {
1052 if (grp && grp->ag_group &&
1053 (cache->c_flags & NL_CACHE_AF_ITER))
1054 nl_cache_set_arg1(cache, grp->ag_family);
1055
1056 restart:
1057 err = nl_cache_request_full_dump(sk, cache);
1058 if (err < 0)
1059 return err;
1060
1061 NL_DBG(2, "Updating cache %p <%s> for family %u, request sent, waiting for reply\n",
1062 cache, nl_cache_name(cache), grp ? grp->ag_family : AF_UNSPEC);
1063
1064 err = nl_cache_pickup(sk, cache);
1065 if (err == -NLE_DUMP_INTR) {
1066 NL_DBG(2, "Dump interrupted, restarting!\n");
1067 goto restart;
1068 } else if (err < 0)
1069 break;
1070
1071 if (grp)
1072 grp++;
1073 } while (grp && grp->ag_group &&
1074 (cache->c_flags & NL_CACHE_AF_ITER));
1075
1076 return err;
1077 }
1078
1079 /** @} */
1080
1081 /**
1082 * @name Utillities
1083 * @{
1084 */
__cache_fast_lookup(struct nl_cache * cache,struct nl_object * needle)1085 static struct nl_object *__cache_fast_lookup(struct nl_cache *cache,
1086 struct nl_object *needle)
1087 {
1088 struct nl_object *obj;
1089
1090 obj = nl_hash_table_lookup(cache->hashtable, needle);
1091 if (obj) {
1092 nl_object_get(obj);
1093 return obj;
1094 }
1095
1096 return NULL;
1097 }
1098
1099 /**
1100 * Search object in cache
1101 * @arg cache Cache
1102 * @arg needle Object to look for.
1103 *
1104 * Searches the cache for an object which matches the object \p needle.
1105 * The function nl_object_identical() is used to determine if the
1106 * objects match. If a matching object is found, the reference counter
1107 * is incremented and the object is returned.
1108 *
1109 * Therefore, if an object is returned, the reference to the object
1110 * must be returned by calling nl_object_put() after usage.
1111 *
1112 * @return Reference to object or NULL if not found.
1113 */
nl_cache_search(struct nl_cache * cache,struct nl_object * needle)1114 struct nl_object *nl_cache_search(struct nl_cache *cache,
1115 struct nl_object *needle)
1116 {
1117 struct nl_object *obj;
1118
1119 if (cache->hashtable)
1120 return __cache_fast_lookup(cache, needle);
1121
1122 nl_list_for_each_entry(obj, &cache->c_items, ce_list) {
1123 if (nl_object_identical(obj, needle)) {
1124 nl_object_get(obj);
1125 return obj;
1126 }
1127 }
1128
1129 return NULL;
1130 }
1131
1132 /**
1133 * Find object in cache
1134 * @arg cache Cache
1135 * @arg filter object acting as a filter
1136 *
1137 * Searches the cache for an object which matches the object filter.
1138 * If the filter attributes matches the object type id attributes,
1139 * and the cache supports hash lookups, a faster hashtable lookup
1140 * is used to return the object. Else, function nl_object_match_filter() is
1141 * used to determine if the objects match. If a matching object is
1142 * found, the reference counter is incremented and the object is returned.
1143 *
1144 * Therefore, if an object is returned, the reference to the object
1145 * must be returned by calling nl_object_put() after usage.
1146 *
1147 * @return Reference to object or NULL if not found.
1148 */
nl_cache_find(struct nl_cache * cache,struct nl_object * filter)1149 struct nl_object *nl_cache_find(struct nl_cache *cache,
1150 struct nl_object *filter)
1151 {
1152 struct nl_object *obj;
1153
1154 if (cache->c_ops == NULL)
1155 BUG();
1156
1157 if ((nl_object_get_id_attrs(filter) == filter->ce_mask)
1158 && cache->hashtable)
1159 return __cache_fast_lookup(cache, filter);
1160
1161 nl_list_for_each_entry(obj, &cache->c_items, ce_list) {
1162 if (nl_object_match_filter(obj, filter)) {
1163 nl_object_get(obj);
1164 return obj;
1165 }
1166 }
1167
1168 return NULL;
1169 }
1170
1171 /**
1172 * Mark all objects of a cache
1173 * @arg cache Cache
1174 *
1175 * Marks all objects of a cache by calling nl_object_mark() on each
1176 * object associated with the cache.
1177 */
nl_cache_mark_all(struct nl_cache * cache)1178 void nl_cache_mark_all(struct nl_cache *cache)
1179 {
1180 struct nl_object *obj;
1181
1182 NL_DBG(2, "Marking all objects in cache %p <%s>\n",
1183 cache, nl_cache_name(cache));
1184
1185 nl_list_for_each_entry(obj, &cache->c_items, ce_list)
1186 nl_object_mark(obj);
1187 }
1188
1189 /** @} */
1190
1191 /**
1192 * @name Dumping
1193 * @{
1194 */
1195
1196 /**
1197 * Dump all elements of a cache.
1198 * @arg cache cache to dump
1199 * @arg params dumping parameters
1200 *
1201 * Dumps all elements of the \a cache to the file descriptor \a fd.
1202 */
nl_cache_dump(struct nl_cache * cache,struct nl_dump_params * params)1203 void nl_cache_dump(struct nl_cache *cache, struct nl_dump_params *params)
1204 {
1205 nl_cache_dump_filter(cache, params, NULL);
1206 }
1207
1208 /**
1209 * Dump all elements of a cache (filtered).
1210 * @arg cache cache to dump
1211 * @arg params dumping parameters (optional)
1212 * @arg filter filter object
1213 *
1214 * Dumps all elements of the \a cache to the file descriptor \a fd
1215 * given they match the given filter \a filter.
1216 */
nl_cache_dump_filter(struct nl_cache * cache,struct nl_dump_params * params,struct nl_object * filter)1217 void nl_cache_dump_filter(struct nl_cache *cache,
1218 struct nl_dump_params *params,
1219 struct nl_object *filter)
1220 {
1221 int type = params ? params->dp_type : NL_DUMP_DETAILS;
1222 struct nl_object_ops *ops;
1223 struct nl_object *obj;
1224
1225 NL_DBG(2, "Dumping cache %p <%s> with filter %p\n",
1226 cache, nl_cache_name(cache), filter);
1227
1228 if (type > NL_DUMP_MAX || type < 0)
1229 BUG();
1230
1231 if (cache->c_ops == NULL)
1232 BUG();
1233
1234 ops = cache->c_ops->co_obj_ops;
1235 if (!ops->oo_dump[type])
1236 return;
1237
1238 if (params && params->dp_buf)
1239 memset(params->dp_buf, 0, params->dp_buflen);
1240
1241 nl_list_for_each_entry(obj, &cache->c_items, ce_list) {
1242 if (filter && !nl_object_match_filter(obj, filter))
1243 continue;
1244
1245 NL_DBG(4, "Dumping object %p...\n", obj);
1246 dump_from_ops(obj, params);
1247 }
1248 }
1249
1250 /** @} */
1251
1252 /**
1253 * @name Iterators
1254 * @{
1255 */
1256
1257 /**
1258 * Call a callback on each element of the cache.
1259 * @arg cache cache to iterate on
1260 * @arg cb callback function
1261 * @arg arg argument passed to callback function
1262 *
1263 * Calls a callback function \a cb on each element of the \a cache.
1264 * The argument \a arg is passed on the callback function.
1265 */
nl_cache_foreach(struct nl_cache * cache,void (* cb)(struct nl_object *,void *),void * arg)1266 void nl_cache_foreach(struct nl_cache *cache,
1267 void (*cb)(struct nl_object *, void *), void *arg)
1268 {
1269 nl_cache_foreach_filter(cache, NULL, cb, arg);
1270 }
1271
1272 /**
1273 * Call a callback on each element of the cache (filtered).
1274 * @arg cache cache to iterate on
1275 * @arg filter filter object
1276 * @arg cb callback function
1277 * @arg arg argument passed to callback function
1278 *
1279 * Calls a callback function \a cb on each element of the \a cache
1280 * that matches the \a filter. The argument \a arg is passed on
1281 * to the callback function.
1282 */
nl_cache_foreach_filter(struct nl_cache * cache,struct nl_object * filter,void (* cb)(struct nl_object *,void *),void * arg)1283 void nl_cache_foreach_filter(struct nl_cache *cache, struct nl_object *filter,
1284 void (*cb)(struct nl_object *, void *), void *arg)
1285 {
1286 struct nl_object *obj, *tmp;
1287
1288 if (cache->c_ops == NULL)
1289 BUG();
1290
1291 nl_list_for_each_entry_safe(obj, tmp, &cache->c_items, ce_list) {
1292 if (filter) {
1293 int diff = nl_object_match_filter(obj, filter);
1294
1295 NL_DBG(3, "%p<->%p object difference: %x\n",
1296 obj, filter, diff);
1297
1298 if (!diff)
1299 continue;
1300 }
1301
1302 /* Caller may hold obj for a long time */
1303 nl_object_get(obj);
1304
1305 cb(obj, arg);
1306
1307 nl_object_put(obj);
1308 }
1309 }
1310
1311 /** @} */
1312
1313 /** @} */
1314