GNU Linux-libre 5.4.207-gnu1
[releases.git] / net / netlabel / netlabel_domainhash.c
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * NetLabel Domain Hash Table
4  *
5  * This file manages the domain hash table that NetLabel uses to determine
6  * which network labeling protocol to use for a given domain.  The NetLabel
7  * system manages static and dynamic label mappings for network protocols such
8  * as CIPSO and RIPSO.
9  *
10  * Author: Paul Moore <paul@paul-moore.com>
11  */
12
13 /*
14  * (c) Copyright Hewlett-Packard Development Company, L.P., 2006, 2008
15  */
16
17 #include <linux/types.h>
18 #include <linux/rculist.h>
19 #include <linux/skbuff.h>
20 #include <linux/spinlock.h>
21 #include <linux/string.h>
22 #include <linux/audit.h>
23 #include <linux/slab.h>
24 #include <net/netlabel.h>
25 #include <net/cipso_ipv4.h>
26 #include <net/calipso.h>
27 #include <asm/bug.h>
28
29 #include "netlabel_mgmt.h"
30 #include "netlabel_addrlist.h"
31 #include "netlabel_calipso.h"
32 #include "netlabel_domainhash.h"
33 #include "netlabel_user.h"
34
35 struct netlbl_domhsh_tbl {
36         struct list_head *tbl;
37         u32 size;
38 };
39
40 /* Domain hash table */
41 /* updates should be so rare that having one spinlock for the entire hash table
42  * should be okay */
43 static DEFINE_SPINLOCK(netlbl_domhsh_lock);
44 #define netlbl_domhsh_rcu_deref(p) \
45         rcu_dereference_check(p, lockdep_is_held(&netlbl_domhsh_lock))
46 static struct netlbl_domhsh_tbl __rcu *netlbl_domhsh;
47 static struct netlbl_dom_map __rcu *netlbl_domhsh_def_ipv4;
48 static struct netlbl_dom_map __rcu *netlbl_domhsh_def_ipv6;
49
50 /*
51  * Domain Hash Table Helper Functions
52  */
53
54 /**
55  * netlbl_domhsh_free_entry - Frees a domain hash table entry
56  * @entry: the entry's RCU field
57  *
58  * Description:
59  * This function is designed to be used as a callback to the call_rcu()
60  * function so that the memory allocated to a hash table entry can be released
61  * safely.
62  *
63  */
64 static void netlbl_domhsh_free_entry(struct rcu_head *entry)
65 {
66         struct netlbl_dom_map *ptr;
67         struct netlbl_af4list *iter4;
68         struct netlbl_af4list *tmp4;
69 #if IS_ENABLED(CONFIG_IPV6)
70         struct netlbl_af6list *iter6;
71         struct netlbl_af6list *tmp6;
72 #endif /* IPv6 */
73
74         ptr = container_of(entry, struct netlbl_dom_map, rcu);
75         if (ptr->def.type == NETLBL_NLTYPE_ADDRSELECT) {
76                 netlbl_af4list_foreach_safe(iter4, tmp4,
77                                             &ptr->def.addrsel->list4) {
78                         netlbl_af4list_remove_entry(iter4);
79                         kfree(netlbl_domhsh_addr4_entry(iter4));
80                 }
81 #if IS_ENABLED(CONFIG_IPV6)
82                 netlbl_af6list_foreach_safe(iter6, tmp6,
83                                             &ptr->def.addrsel->list6) {
84                         netlbl_af6list_remove_entry(iter6);
85                         kfree(netlbl_domhsh_addr6_entry(iter6));
86                 }
87 #endif /* IPv6 */
88                 kfree(ptr->def.addrsel);
89         }
90         kfree(ptr->domain);
91         kfree(ptr);
92 }
93
94 /**
95  * netlbl_domhsh_hash - Hashing function for the domain hash table
96  * @domain: the domain name to hash
97  *
98  * Description:
99  * This is the hashing function for the domain hash table, it returns the
100  * correct bucket number for the domain.  The caller is responsible for
101  * ensuring that the hash table is protected with either a RCU read lock or the
102  * hash table lock.
103  *
104  */
105 static u32 netlbl_domhsh_hash(const char *key)
106 {
107         u32 iter;
108         u32 val;
109         u32 len;
110
111         /* This is taken (with slight modification) from
112          * security/selinux/ss/symtab.c:symhash() */
113
114         for (iter = 0, val = 0, len = strlen(key); iter < len; iter++)
115                 val = (val << 4 | (val >> (8 * sizeof(u32) - 4))) ^ key[iter];
116         return val & (netlbl_domhsh_rcu_deref(netlbl_domhsh)->size - 1);
117 }
118
119 static bool netlbl_family_match(u16 f1, u16 f2)
120 {
121         return (f1 == f2) || (f1 == AF_UNSPEC) || (f2 == AF_UNSPEC);
122 }
123
124 /**
125  * netlbl_domhsh_search - Search for a domain entry
126  * @domain: the domain
127  * @family: the address family
128  *
129  * Description:
130  * Searches the domain hash table and returns a pointer to the hash table
131  * entry if found, otherwise NULL is returned.  @family may be %AF_UNSPEC
132  * which matches any address family entries.  The caller is responsible for
133  * ensuring that the hash table is protected with either a RCU read lock or the
134  * hash table lock.
135  *
136  */
137 static struct netlbl_dom_map *netlbl_domhsh_search(const char *domain,
138                                                    u16 family)
139 {
140         u32 bkt;
141         struct list_head *bkt_list;
142         struct netlbl_dom_map *iter;
143
144         if (domain != NULL) {
145                 bkt = netlbl_domhsh_hash(domain);
146                 bkt_list = &netlbl_domhsh_rcu_deref(netlbl_domhsh)->tbl[bkt];
147                 list_for_each_entry_rcu(iter, bkt_list, list)
148                         if (iter->valid &&
149                             netlbl_family_match(iter->family, family) &&
150                             strcmp(iter->domain, domain) == 0)
151                                 return iter;
152         }
153
154         return NULL;
155 }
156
157 /**
158  * netlbl_domhsh_search_def - Search for a domain entry
159  * @domain: the domain
160  * @family: the address family
161  *
162  * Description:
163  * Searches the domain hash table and returns a pointer to the hash table
164  * entry if an exact match is found, if an exact match is not present in the
165  * hash table then the default entry is returned if valid otherwise NULL is
166  * returned.  @family may be %AF_UNSPEC which matches any address family
167  * entries.  The caller is responsible ensuring that the hash table is
168  * protected with either a RCU read lock or the hash table lock.
169  *
170  */
171 static struct netlbl_dom_map *netlbl_domhsh_search_def(const char *domain,
172                                                        u16 family)
173 {
174         struct netlbl_dom_map *entry;
175
176         entry = netlbl_domhsh_search(domain, family);
177         if (entry != NULL)
178                 return entry;
179         if (family == AF_INET || family == AF_UNSPEC) {
180                 entry = netlbl_domhsh_rcu_deref(netlbl_domhsh_def_ipv4);
181                 if (entry != NULL && entry->valid)
182                         return entry;
183         }
184         if (family == AF_INET6 || family == AF_UNSPEC) {
185                 entry = netlbl_domhsh_rcu_deref(netlbl_domhsh_def_ipv6);
186                 if (entry != NULL && entry->valid)
187                         return entry;
188         }
189
190         return NULL;
191 }
192
193 /**
194  * netlbl_domhsh_audit_add - Generate an audit entry for an add event
195  * @entry: the entry being added
196  * @addr4: the IPv4 address information
197  * @addr6: the IPv6 address information
198  * @result: the result code
199  * @audit_info: NetLabel audit information
200  *
201  * Description:
202  * Generate an audit record for adding a new NetLabel/LSM mapping entry with
203  * the given information.  Caller is responsible for holding the necessary
204  * locks.
205  *
206  */
207 static void netlbl_domhsh_audit_add(struct netlbl_dom_map *entry,
208                                     struct netlbl_af4list *addr4,
209                                     struct netlbl_af6list *addr6,
210                                     int result,
211                                     struct netlbl_audit *audit_info)
212 {
213         struct audit_buffer *audit_buf;
214         struct cipso_v4_doi *cipsov4 = NULL;
215         struct calipso_doi *calipso = NULL;
216         u32 type;
217
218         audit_buf = netlbl_audit_start_common(AUDIT_MAC_MAP_ADD, audit_info);
219         if (audit_buf != NULL) {
220                 audit_log_format(audit_buf, " nlbl_domain=%s",
221                                  entry->domain ? entry->domain : "(default)");
222                 if (addr4 != NULL) {
223                         struct netlbl_domaddr4_map *map4;
224                         map4 = netlbl_domhsh_addr4_entry(addr4);
225                         type = map4->def.type;
226                         cipsov4 = map4->def.cipso;
227                         netlbl_af4list_audit_addr(audit_buf, 0, NULL,
228                                                   addr4->addr, addr4->mask);
229 #if IS_ENABLED(CONFIG_IPV6)
230                 } else if (addr6 != NULL) {
231                         struct netlbl_domaddr6_map *map6;
232                         map6 = netlbl_domhsh_addr6_entry(addr6);
233                         type = map6->def.type;
234                         calipso = map6->def.calipso;
235                         netlbl_af6list_audit_addr(audit_buf, 0, NULL,
236                                                   &addr6->addr, &addr6->mask);
237 #endif /* IPv6 */
238                 } else {
239                         type = entry->def.type;
240                         cipsov4 = entry->def.cipso;
241                         calipso = entry->def.calipso;
242                 }
243                 switch (type) {
244                 case NETLBL_NLTYPE_UNLABELED:
245                         audit_log_format(audit_buf, " nlbl_protocol=unlbl");
246                         break;
247                 case NETLBL_NLTYPE_CIPSOV4:
248                         BUG_ON(cipsov4 == NULL);
249                         audit_log_format(audit_buf,
250                                          " nlbl_protocol=cipsov4 cipso_doi=%u",
251                                          cipsov4->doi);
252                         break;
253                 case NETLBL_NLTYPE_CALIPSO:
254                         BUG_ON(calipso == NULL);
255                         audit_log_format(audit_buf,
256                                          " nlbl_protocol=calipso calipso_doi=%u",
257                                          calipso->doi);
258                         break;
259                 }
260                 audit_log_format(audit_buf, " res=%u", result == 0 ? 1 : 0);
261                 audit_log_end(audit_buf);
262         }
263 }
264
265 /**
266  * netlbl_domhsh_validate - Validate a new domain mapping entry
267  * @entry: the entry to validate
268  *
269  * This function validates the new domain mapping entry to ensure that it is
270  * a valid entry.  Returns zero on success, negative values on failure.
271  *
272  */
273 static int netlbl_domhsh_validate(const struct netlbl_dom_map *entry)
274 {
275         struct netlbl_af4list *iter4;
276         struct netlbl_domaddr4_map *map4;
277 #if IS_ENABLED(CONFIG_IPV6)
278         struct netlbl_af6list *iter6;
279         struct netlbl_domaddr6_map *map6;
280 #endif /* IPv6 */
281
282         if (entry == NULL)
283                 return -EINVAL;
284
285         if (entry->family != AF_INET && entry->family != AF_INET6 &&
286             (entry->family != AF_UNSPEC ||
287              entry->def.type != NETLBL_NLTYPE_UNLABELED))
288                 return -EINVAL;
289
290         switch (entry->def.type) {
291         case NETLBL_NLTYPE_UNLABELED:
292                 if (entry->def.cipso != NULL || entry->def.calipso != NULL ||
293                     entry->def.addrsel != NULL)
294                         return -EINVAL;
295                 break;
296         case NETLBL_NLTYPE_CIPSOV4:
297                 if (entry->family != AF_INET ||
298                     entry->def.cipso == NULL)
299                         return -EINVAL;
300                 break;
301         case NETLBL_NLTYPE_CALIPSO:
302                 if (entry->family != AF_INET6 ||
303                     entry->def.calipso == NULL)
304                         return -EINVAL;
305                 break;
306         case NETLBL_NLTYPE_ADDRSELECT:
307                 netlbl_af4list_foreach(iter4, &entry->def.addrsel->list4) {
308                         map4 = netlbl_domhsh_addr4_entry(iter4);
309                         switch (map4->def.type) {
310                         case NETLBL_NLTYPE_UNLABELED:
311                                 if (map4->def.cipso != NULL)
312                                         return -EINVAL;
313                                 break;
314                         case NETLBL_NLTYPE_CIPSOV4:
315                                 if (map4->def.cipso == NULL)
316                                         return -EINVAL;
317                                 break;
318                         default:
319                                 return -EINVAL;
320                         }
321                 }
322 #if IS_ENABLED(CONFIG_IPV6)
323                 netlbl_af6list_foreach(iter6, &entry->def.addrsel->list6) {
324                         map6 = netlbl_domhsh_addr6_entry(iter6);
325                         switch (map6->def.type) {
326                         case NETLBL_NLTYPE_UNLABELED:
327                                 if (map6->def.calipso != NULL)
328                                         return -EINVAL;
329                                 break;
330                         case NETLBL_NLTYPE_CALIPSO:
331                                 if (map6->def.calipso == NULL)
332                                         return -EINVAL;
333                                 break;
334                         default:
335                                 return -EINVAL;
336                         }
337                 }
338 #endif /* IPv6 */
339                 break;
340         default:
341                 return -EINVAL;
342         }
343
344         return 0;
345 }
346
347 /*
348  * Domain Hash Table Functions
349  */
350
351 /**
352  * netlbl_domhsh_init - Init for the domain hash
353  * @size: the number of bits to use for the hash buckets
354  *
355  * Description:
356  * Initializes the domain hash table, should be called only by
357  * netlbl_user_init() during initialization.  Returns zero on success, non-zero
358  * values on error.
359  *
360  */
361 int __init netlbl_domhsh_init(u32 size)
362 {
363         u32 iter;
364         struct netlbl_domhsh_tbl *hsh_tbl;
365
366         if (size == 0)
367                 return -EINVAL;
368
369         hsh_tbl = kmalloc(sizeof(*hsh_tbl), GFP_KERNEL);
370         if (hsh_tbl == NULL)
371                 return -ENOMEM;
372         hsh_tbl->size = 1 << size;
373         hsh_tbl->tbl = kcalloc(hsh_tbl->size,
374                                sizeof(struct list_head),
375                                GFP_KERNEL);
376         if (hsh_tbl->tbl == NULL) {
377                 kfree(hsh_tbl);
378                 return -ENOMEM;
379         }
380         for (iter = 0; iter < hsh_tbl->size; iter++)
381                 INIT_LIST_HEAD(&hsh_tbl->tbl[iter]);
382
383         spin_lock(&netlbl_domhsh_lock);
384         rcu_assign_pointer(netlbl_domhsh, hsh_tbl);
385         spin_unlock(&netlbl_domhsh_lock);
386
387         return 0;
388 }
389
390 /**
391  * netlbl_domhsh_add - Adds a entry to the domain hash table
392  * @entry: the entry to add
393  * @audit_info: NetLabel audit information
394  *
395  * Description:
396  * Adds a new entry to the domain hash table and handles any updates to the
397  * lower level protocol handler (i.e. CIPSO).  @entry->family may be set to
398  * %AF_UNSPEC which will add an entry that matches all address families.  This
399  * is only useful for the unlabelled type and will only succeed if there is no
400  * existing entry for any address family with the same domain.  Returns zero
401  * on success, negative on failure.
402  *
403  */
404 int netlbl_domhsh_add(struct netlbl_dom_map *entry,
405                       struct netlbl_audit *audit_info)
406 {
407         int ret_val = 0;
408         struct netlbl_dom_map *entry_old, *entry_b;
409         struct netlbl_af4list *iter4;
410         struct netlbl_af4list *tmp4;
411 #if IS_ENABLED(CONFIG_IPV6)
412         struct netlbl_af6list *iter6;
413         struct netlbl_af6list *tmp6;
414 #endif /* IPv6 */
415
416         ret_val = netlbl_domhsh_validate(entry);
417         if (ret_val != 0)
418                 return ret_val;
419
420         /* XXX - we can remove this RCU read lock as the spinlock protects the
421          *       entire function, but before we do we need to fixup the
422          *       netlbl_af[4,6]list RCU functions to do "the right thing" with
423          *       respect to rcu_dereference() when only a spinlock is held. */
424         rcu_read_lock();
425         spin_lock(&netlbl_domhsh_lock);
426         if (entry->domain != NULL)
427                 entry_old = netlbl_domhsh_search(entry->domain, entry->family);
428         else
429                 entry_old = netlbl_domhsh_search_def(entry->domain,
430                                                      entry->family);
431         if (entry_old == NULL) {
432                 entry->valid = 1;
433
434                 if (entry->domain != NULL) {
435                         u32 bkt = netlbl_domhsh_hash(entry->domain);
436                         list_add_tail_rcu(&entry->list,
437                                     &rcu_dereference(netlbl_domhsh)->tbl[bkt]);
438                 } else {
439                         INIT_LIST_HEAD(&entry->list);
440                         switch (entry->family) {
441                         case AF_INET:
442                                 rcu_assign_pointer(netlbl_domhsh_def_ipv4,
443                                                    entry);
444                                 break;
445                         case AF_INET6:
446                                 rcu_assign_pointer(netlbl_domhsh_def_ipv6,
447                                                    entry);
448                                 break;
449                         case AF_UNSPEC:
450                                 if (entry->def.type !=
451                                     NETLBL_NLTYPE_UNLABELED) {
452                                         ret_val = -EINVAL;
453                                         goto add_return;
454                                 }
455                                 entry_b = kzalloc(sizeof(*entry_b), GFP_ATOMIC);
456                                 if (entry_b == NULL) {
457                                         ret_val = -ENOMEM;
458                                         goto add_return;
459                                 }
460                                 entry_b->family = AF_INET6;
461                                 entry_b->def.type = NETLBL_NLTYPE_UNLABELED;
462                                 entry_b->valid = 1;
463                                 entry->family = AF_INET;
464                                 rcu_assign_pointer(netlbl_domhsh_def_ipv4,
465                                                    entry);
466                                 rcu_assign_pointer(netlbl_domhsh_def_ipv6,
467                                                    entry_b);
468                                 break;
469                         default:
470                                 /* Already checked in
471                                  * netlbl_domhsh_validate(). */
472                                 ret_val = -EINVAL;
473                                 goto add_return;
474                         }
475                 }
476
477                 if (entry->def.type == NETLBL_NLTYPE_ADDRSELECT) {
478                         netlbl_af4list_foreach_rcu(iter4,
479                                                    &entry->def.addrsel->list4)
480                                 netlbl_domhsh_audit_add(entry, iter4, NULL,
481                                                         ret_val, audit_info);
482 #if IS_ENABLED(CONFIG_IPV6)
483                         netlbl_af6list_foreach_rcu(iter6,
484                                                    &entry->def.addrsel->list6)
485                                 netlbl_domhsh_audit_add(entry, NULL, iter6,
486                                                         ret_val, audit_info);
487 #endif /* IPv6 */
488                 } else
489                         netlbl_domhsh_audit_add(entry, NULL, NULL,
490                                                 ret_val, audit_info);
491         } else if (entry_old->def.type == NETLBL_NLTYPE_ADDRSELECT &&
492                    entry->def.type == NETLBL_NLTYPE_ADDRSELECT) {
493                 struct list_head *old_list4;
494                 struct list_head *old_list6;
495
496                 old_list4 = &entry_old->def.addrsel->list4;
497                 old_list6 = &entry_old->def.addrsel->list6;
498
499                 /* we only allow the addition of address selectors if all of
500                  * the selectors do not exist in the existing domain map */
501                 netlbl_af4list_foreach_rcu(iter4, &entry->def.addrsel->list4)
502                         if (netlbl_af4list_search_exact(iter4->addr,
503                                                         iter4->mask,
504                                                         old_list4)) {
505                                 ret_val = -EEXIST;
506                                 goto add_return;
507                         }
508 #if IS_ENABLED(CONFIG_IPV6)
509                 netlbl_af6list_foreach_rcu(iter6, &entry->def.addrsel->list6)
510                         if (netlbl_af6list_search_exact(&iter6->addr,
511                                                         &iter6->mask,
512                                                         old_list6)) {
513                                 ret_val = -EEXIST;
514                                 goto add_return;
515                         }
516 #endif /* IPv6 */
517
518                 netlbl_af4list_foreach_safe(iter4, tmp4,
519                                             &entry->def.addrsel->list4) {
520                         netlbl_af4list_remove_entry(iter4);
521                         iter4->valid = 1;
522                         ret_val = netlbl_af4list_add(iter4, old_list4);
523                         netlbl_domhsh_audit_add(entry_old, iter4, NULL,
524                                                 ret_val, audit_info);
525                         if (ret_val != 0)
526                                 goto add_return;
527                 }
528 #if IS_ENABLED(CONFIG_IPV6)
529                 netlbl_af6list_foreach_safe(iter6, tmp6,
530                                             &entry->def.addrsel->list6) {
531                         netlbl_af6list_remove_entry(iter6);
532                         iter6->valid = 1;
533                         ret_val = netlbl_af6list_add(iter6, old_list6);
534                         netlbl_domhsh_audit_add(entry_old, NULL, iter6,
535                                                 ret_val, audit_info);
536                         if (ret_val != 0)
537                                 goto add_return;
538                 }
539 #endif /* IPv6 */
540                 /* cleanup the new entry since we've moved everything over */
541                 netlbl_domhsh_free_entry(&entry->rcu);
542         } else
543                 ret_val = -EINVAL;
544
545 add_return:
546         spin_unlock(&netlbl_domhsh_lock);
547         rcu_read_unlock();
548         return ret_val;
549 }
550
551 /**
552  * netlbl_domhsh_add_default - Adds the default entry to the domain hash table
553  * @entry: the entry to add
554  * @audit_info: NetLabel audit information
555  *
556  * Description:
557  * Adds a new default entry to the domain hash table and handles any updates
558  * to the lower level protocol handler (i.e. CIPSO).  Returns zero on success,
559  * negative on failure.
560  *
561  */
562 int netlbl_domhsh_add_default(struct netlbl_dom_map *entry,
563                               struct netlbl_audit *audit_info)
564 {
565         return netlbl_domhsh_add(entry, audit_info);
566 }
567
568 /**
569  * netlbl_domhsh_remove_entry - Removes a given entry from the domain table
570  * @entry: the entry to remove
571  * @audit_info: NetLabel audit information
572  *
573  * Description:
574  * Removes an entry from the domain hash table and handles any updates to the
575  * lower level protocol handler (i.e. CIPSO).  Caller is responsible for
576  * ensuring that the RCU read lock is held.  Returns zero on success, negative
577  * on failure.
578  *
579  */
580 int netlbl_domhsh_remove_entry(struct netlbl_dom_map *entry,
581                                struct netlbl_audit *audit_info)
582 {
583         int ret_val = 0;
584         struct audit_buffer *audit_buf;
585         struct netlbl_af4list *iter4;
586         struct netlbl_domaddr4_map *map4;
587 #if IS_ENABLED(CONFIG_IPV6)
588         struct netlbl_af6list *iter6;
589         struct netlbl_domaddr6_map *map6;
590 #endif /* IPv6 */
591
592         if (entry == NULL)
593                 return -ENOENT;
594
595         spin_lock(&netlbl_domhsh_lock);
596         if (entry->valid) {
597                 entry->valid = 0;
598                 if (entry == rcu_dereference(netlbl_domhsh_def_ipv4))
599                         RCU_INIT_POINTER(netlbl_domhsh_def_ipv4, NULL);
600                 else if (entry == rcu_dereference(netlbl_domhsh_def_ipv6))
601                         RCU_INIT_POINTER(netlbl_domhsh_def_ipv6, NULL);
602                 else
603                         list_del_rcu(&entry->list);
604         } else
605                 ret_val = -ENOENT;
606         spin_unlock(&netlbl_domhsh_lock);
607
608         if (ret_val)
609                 return ret_val;
610
611         audit_buf = netlbl_audit_start_common(AUDIT_MAC_MAP_DEL, audit_info);
612         if (audit_buf != NULL) {
613                 audit_log_format(audit_buf,
614                                  " nlbl_domain=%s res=%u",
615                                  entry->domain ? entry->domain : "(default)",
616                                  ret_val == 0 ? 1 : 0);
617                 audit_log_end(audit_buf);
618         }
619
620         switch (entry->def.type) {
621         case NETLBL_NLTYPE_ADDRSELECT:
622                 netlbl_af4list_foreach_rcu(iter4, &entry->def.addrsel->list4) {
623                         map4 = netlbl_domhsh_addr4_entry(iter4);
624                         cipso_v4_doi_putdef(map4->def.cipso);
625                 }
626 #if IS_ENABLED(CONFIG_IPV6)
627                 netlbl_af6list_foreach_rcu(iter6, &entry->def.addrsel->list6) {
628                         map6 = netlbl_domhsh_addr6_entry(iter6);
629                         calipso_doi_putdef(map6->def.calipso);
630                 }
631 #endif /* IPv6 */
632                 break;
633         case NETLBL_NLTYPE_CIPSOV4:
634                 cipso_v4_doi_putdef(entry->def.cipso);
635                 break;
636 #if IS_ENABLED(CONFIG_IPV6)
637         case NETLBL_NLTYPE_CALIPSO:
638                 calipso_doi_putdef(entry->def.calipso);
639                 break;
640 #endif /* IPv6 */
641         }
642         call_rcu(&entry->rcu, netlbl_domhsh_free_entry);
643
644         return ret_val;
645 }
646
647 /**
648  * netlbl_domhsh_remove_af4 - Removes an address selector entry
649  * @domain: the domain
650  * @addr: IPv4 address
651  * @mask: IPv4 address mask
652  * @audit_info: NetLabel audit information
653  *
654  * Description:
655  * Removes an individual address selector from a domain mapping and potentially
656  * the entire mapping if it is empty.  Returns zero on success, negative values
657  * on failure.
658  *
659  */
660 int netlbl_domhsh_remove_af4(const char *domain,
661                              const struct in_addr *addr,
662                              const struct in_addr *mask,
663                              struct netlbl_audit *audit_info)
664 {
665         struct netlbl_dom_map *entry_map;
666         struct netlbl_af4list *entry_addr;
667         struct netlbl_af4list *iter4;
668 #if IS_ENABLED(CONFIG_IPV6)
669         struct netlbl_af6list *iter6;
670 #endif /* IPv6 */
671         struct netlbl_domaddr4_map *entry;
672
673         rcu_read_lock();
674
675         if (domain)
676                 entry_map = netlbl_domhsh_search(domain, AF_INET);
677         else
678                 entry_map = netlbl_domhsh_search_def(domain, AF_INET);
679         if (entry_map == NULL ||
680             entry_map->def.type != NETLBL_NLTYPE_ADDRSELECT)
681                 goto remove_af4_failure;
682
683         spin_lock(&netlbl_domhsh_lock);
684         entry_addr = netlbl_af4list_remove(addr->s_addr, mask->s_addr,
685                                            &entry_map->def.addrsel->list4);
686         spin_unlock(&netlbl_domhsh_lock);
687
688         if (entry_addr == NULL)
689                 goto remove_af4_failure;
690         netlbl_af4list_foreach_rcu(iter4, &entry_map->def.addrsel->list4)
691                 goto remove_af4_single_addr;
692 #if IS_ENABLED(CONFIG_IPV6)
693         netlbl_af6list_foreach_rcu(iter6, &entry_map->def.addrsel->list6)
694                 goto remove_af4_single_addr;
695 #endif /* IPv6 */
696         /* the domain mapping is empty so remove it from the mapping table */
697         netlbl_domhsh_remove_entry(entry_map, audit_info);
698
699 remove_af4_single_addr:
700         rcu_read_unlock();
701         /* yick, we can't use call_rcu here because we don't have a rcu head
702          * pointer but hopefully this should be a rare case so the pause
703          * shouldn't be a problem */
704         synchronize_rcu();
705         entry = netlbl_domhsh_addr4_entry(entry_addr);
706         cipso_v4_doi_putdef(entry->def.cipso);
707         kfree(entry);
708         return 0;
709
710 remove_af4_failure:
711         rcu_read_unlock();
712         return -ENOENT;
713 }
714
715 #if IS_ENABLED(CONFIG_IPV6)
716 /**
717  * netlbl_domhsh_remove_af6 - Removes an address selector entry
718  * @domain: the domain
719  * @addr: IPv6 address
720  * @mask: IPv6 address mask
721  * @audit_info: NetLabel audit information
722  *
723  * Description:
724  * Removes an individual address selector from a domain mapping and potentially
725  * the entire mapping if it is empty.  Returns zero on success, negative values
726  * on failure.
727  *
728  */
729 int netlbl_domhsh_remove_af6(const char *domain,
730                              const struct in6_addr *addr,
731                              const struct in6_addr *mask,
732                              struct netlbl_audit *audit_info)
733 {
734         struct netlbl_dom_map *entry_map;
735         struct netlbl_af6list *entry_addr;
736         struct netlbl_af4list *iter4;
737         struct netlbl_af6list *iter6;
738         struct netlbl_domaddr6_map *entry;
739
740         rcu_read_lock();
741
742         if (domain)
743                 entry_map = netlbl_domhsh_search(domain, AF_INET6);
744         else
745                 entry_map = netlbl_domhsh_search_def(domain, AF_INET6);
746         if (entry_map == NULL ||
747             entry_map->def.type != NETLBL_NLTYPE_ADDRSELECT)
748                 goto remove_af6_failure;
749
750         spin_lock(&netlbl_domhsh_lock);
751         entry_addr = netlbl_af6list_remove(addr, mask,
752                                            &entry_map->def.addrsel->list6);
753         spin_unlock(&netlbl_domhsh_lock);
754
755         if (entry_addr == NULL)
756                 goto remove_af6_failure;
757         netlbl_af4list_foreach_rcu(iter4, &entry_map->def.addrsel->list4)
758                 goto remove_af6_single_addr;
759         netlbl_af6list_foreach_rcu(iter6, &entry_map->def.addrsel->list6)
760                 goto remove_af6_single_addr;
761         /* the domain mapping is empty so remove it from the mapping table */
762         netlbl_domhsh_remove_entry(entry_map, audit_info);
763
764 remove_af6_single_addr:
765         rcu_read_unlock();
766         /* yick, we can't use call_rcu here because we don't have a rcu head
767          * pointer but hopefully this should be a rare case so the pause
768          * shouldn't be a problem */
769         synchronize_rcu();
770         entry = netlbl_domhsh_addr6_entry(entry_addr);
771         calipso_doi_putdef(entry->def.calipso);
772         kfree(entry);
773         return 0;
774
775 remove_af6_failure:
776         rcu_read_unlock();
777         return -ENOENT;
778 }
779 #endif /* IPv6 */
780
781 /**
782  * netlbl_domhsh_remove - Removes an entry from the domain hash table
783  * @domain: the domain to remove
784  * @family: address family
785  * @audit_info: NetLabel audit information
786  *
787  * Description:
788  * Removes an entry from the domain hash table and handles any updates to the
789  * lower level protocol handler (i.e. CIPSO).  @family may be %AF_UNSPEC which
790  * removes all address family entries.  Returns zero on success, negative on
791  * failure.
792  *
793  */
794 int netlbl_domhsh_remove(const char *domain, u16 family,
795                          struct netlbl_audit *audit_info)
796 {
797         int ret_val = -EINVAL;
798         struct netlbl_dom_map *entry;
799
800         rcu_read_lock();
801
802         if (family == AF_INET || family == AF_UNSPEC) {
803                 if (domain)
804                         entry = netlbl_domhsh_search(domain, AF_INET);
805                 else
806                         entry = netlbl_domhsh_search_def(domain, AF_INET);
807                 ret_val = netlbl_domhsh_remove_entry(entry, audit_info);
808                 if (ret_val && ret_val != -ENOENT)
809                         goto done;
810         }
811         if (family == AF_INET6 || family == AF_UNSPEC) {
812                 int ret_val2;
813
814                 if (domain)
815                         entry = netlbl_domhsh_search(domain, AF_INET6);
816                 else
817                         entry = netlbl_domhsh_search_def(domain, AF_INET6);
818                 ret_val2 = netlbl_domhsh_remove_entry(entry, audit_info);
819                 if (ret_val2 != -ENOENT)
820                         ret_val = ret_val2;
821         }
822 done:
823         rcu_read_unlock();
824
825         return ret_val;
826 }
827
828 /**
829  * netlbl_domhsh_remove_default - Removes the default entry from the table
830  * @family: address family
831  * @audit_info: NetLabel audit information
832  *
833  * Description:
834  * Removes/resets the default entry corresponding to @family from the domain
835  * hash table and handles any updates to the lower level protocol handler
836  * (i.e. CIPSO).  @family may be %AF_UNSPEC which removes all address family
837  * entries.  Returns zero on success, negative on failure.
838  *
839  */
840 int netlbl_domhsh_remove_default(u16 family, struct netlbl_audit *audit_info)
841 {
842         return netlbl_domhsh_remove(NULL, family, audit_info);
843 }
844
845 /**
846  * netlbl_domhsh_getentry - Get an entry from the domain hash table
847  * @domain: the domain name to search for
848  * @family: address family
849  *
850  * Description:
851  * Look through the domain hash table searching for an entry to match @domain,
852  * with address family @family, return a pointer to a copy of the entry or
853  * NULL.  The caller is responsible for ensuring that rcu_read_[un]lock() is
854  * called.
855  *
856  */
857 struct netlbl_dom_map *netlbl_domhsh_getentry(const char *domain, u16 family)
858 {
859         if (family == AF_UNSPEC)
860                 return NULL;
861         return netlbl_domhsh_search_def(domain, family);
862 }
863
864 /**
865  * netlbl_domhsh_getentry_af4 - Get an entry from the domain hash table
866  * @domain: the domain name to search for
867  * @addr: the IP address to search for
868  *
869  * Description:
870  * Look through the domain hash table searching for an entry to match @domain
871  * and @addr, return a pointer to a copy of the entry or NULL.  The caller is
872  * responsible for ensuring that rcu_read_[un]lock() is called.
873  *
874  */
875 struct netlbl_dommap_def *netlbl_domhsh_getentry_af4(const char *domain,
876                                                      __be32 addr)
877 {
878         struct netlbl_dom_map *dom_iter;
879         struct netlbl_af4list *addr_iter;
880
881         dom_iter = netlbl_domhsh_search_def(domain, AF_INET);
882         if (dom_iter == NULL)
883                 return NULL;
884
885         if (dom_iter->def.type != NETLBL_NLTYPE_ADDRSELECT)
886                 return &dom_iter->def;
887         addr_iter = netlbl_af4list_search(addr, &dom_iter->def.addrsel->list4);
888         if (addr_iter == NULL)
889                 return NULL;
890         return &(netlbl_domhsh_addr4_entry(addr_iter)->def);
891 }
892
893 #if IS_ENABLED(CONFIG_IPV6)
894 /**
895  * netlbl_domhsh_getentry_af6 - Get an entry from the domain hash table
896  * @domain: the domain name to search for
897  * @addr: the IP address to search for
898  *
899  * Description:
900  * Look through the domain hash table searching for an entry to match @domain
901  * and @addr, return a pointer to a copy of the entry or NULL.  The caller is
902  * responsible for ensuring that rcu_read_[un]lock() is called.
903  *
904  */
905 struct netlbl_dommap_def *netlbl_domhsh_getentry_af6(const char *domain,
906                                                    const struct in6_addr *addr)
907 {
908         struct netlbl_dom_map *dom_iter;
909         struct netlbl_af6list *addr_iter;
910
911         dom_iter = netlbl_domhsh_search_def(domain, AF_INET6);
912         if (dom_iter == NULL)
913                 return NULL;
914
915         if (dom_iter->def.type != NETLBL_NLTYPE_ADDRSELECT)
916                 return &dom_iter->def;
917         addr_iter = netlbl_af6list_search(addr, &dom_iter->def.addrsel->list6);
918         if (addr_iter == NULL)
919                 return NULL;
920         return &(netlbl_domhsh_addr6_entry(addr_iter)->def);
921 }
922 #endif /* IPv6 */
923
924 /**
925  * netlbl_domhsh_walk - Iterate through the domain mapping hash table
926  * @skip_bkt: the number of buckets to skip at the start
927  * @skip_chain: the number of entries to skip in the first iterated bucket
928  * @callback: callback for each entry
929  * @cb_arg: argument for the callback function
930  *
931  * Description:
932  * Interate over the domain mapping hash table, skipping the first @skip_bkt
933  * buckets and @skip_chain entries.  For each entry in the table call
934  * @callback, if @callback returns a negative value stop 'walking' through the
935  * table and return.  Updates the values in @skip_bkt and @skip_chain on
936  * return.  Returns zero on success, negative values on failure.
937  *
938  */
939 int netlbl_domhsh_walk(u32 *skip_bkt,
940                      u32 *skip_chain,
941                      int (*callback) (struct netlbl_dom_map *entry, void *arg),
942                      void *cb_arg)
943 {
944         int ret_val = -ENOENT;
945         u32 iter_bkt;
946         struct list_head *iter_list;
947         struct netlbl_dom_map *iter_entry;
948         u32 chain_cnt = 0;
949
950         rcu_read_lock();
951         for (iter_bkt = *skip_bkt;
952              iter_bkt < rcu_dereference(netlbl_domhsh)->size;
953              iter_bkt++, chain_cnt = 0) {
954                 iter_list = &rcu_dereference(netlbl_domhsh)->tbl[iter_bkt];
955                 list_for_each_entry_rcu(iter_entry, iter_list, list)
956                         if (iter_entry->valid) {
957                                 if (chain_cnt++ < *skip_chain)
958                                         continue;
959                                 ret_val = callback(iter_entry, cb_arg);
960                                 if (ret_val < 0) {
961                                         chain_cnt--;
962                                         goto walk_return;
963                                 }
964                         }
965         }
966
967 walk_return:
968         rcu_read_unlock();
969         *skip_bkt = iter_bkt;
970         *skip_chain = chain_cnt;
971         return ret_val;
972 }