Merge branch 'next' of git://git.kernel.org/pub/scm/linux/kernel/git/rzhang/linux
[cascardo/linux.git] / net / netlabel / netlabel_kapi.c
1 /*
2  * NetLabel Kernel API
3  *
4  * This file defines the kernel API for the NetLabel system.  The NetLabel
5  * system manages static and dynamic label mappings for network protocols such
6  * as CIPSO and RIPSO.
7  *
8  * Author: Paul Moore <paul@paul-moore.com>
9  *
10  */
11
12 /*
13  * (c) Copyright Hewlett-Packard Development Company, L.P., 2006, 2008
14  *
15  * This program is free software;  you can redistribute it and/or modify
16  * it under the terms of the GNU General Public License as published by
17  * the Free Software Foundation; either version 2 of the License, or
18  * (at your option) any later version.
19  *
20  * This program is distributed in the hope that it will be useful,
21  * but WITHOUT ANY WARRANTY;  without even the implied warranty of
22  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
23  * the GNU General Public License for more details.
24  *
25  * You should have received a copy of the GNU General Public License
26  * along with this program;  if not, see <http://www.gnu.org/licenses/>.
27  *
28  */
29
30 #include <linux/init.h>
31 #include <linux/types.h>
32 #include <linux/slab.h>
33 #include <linux/audit.h>
34 #include <linux/in.h>
35 #include <linux/in6.h>
36 #include <net/ip.h>
37 #include <net/ipv6.h>
38 #include <net/netlabel.h>
39 #include <net/cipso_ipv4.h>
40 #include <asm/bug.h>
41 #include <linux/atomic.h>
42
43 #include "netlabel_domainhash.h"
44 #include "netlabel_unlabeled.h"
45 #include "netlabel_cipso_v4.h"
46 #include "netlabel_user.h"
47 #include "netlabel_mgmt.h"
48 #include "netlabel_addrlist.h"
49
50 /*
51  * Configuration Functions
52  */
53
54 /**
55  * netlbl_cfg_map_del - Remove a NetLabel/LSM domain mapping
56  * @domain: the domain mapping to remove
57  * @family: address family
58  * @addr: IP address
59  * @mask: IP address mask
60  * @audit_info: NetLabel audit information
61  *
62  * Description:
63  * Removes a NetLabel/LSM domain mapping.  A @domain value of NULL causes the
64  * default domain mapping to be removed.  Returns zero on success, negative
65  * values on failure.
66  *
67  */
68 int netlbl_cfg_map_del(const char *domain,
69                        u16 family,
70                        const void *addr,
71                        const void *mask,
72                        struct netlbl_audit *audit_info)
73 {
74         if (addr == NULL && mask == NULL) {
75                 return netlbl_domhsh_remove(domain, audit_info);
76         } else if (addr != NULL && mask != NULL) {
77                 switch (family) {
78                 case AF_INET:
79                         return netlbl_domhsh_remove_af4(domain, addr, mask,
80                                                         audit_info);
81                 default:
82                         return -EPFNOSUPPORT;
83                 }
84         } else
85                 return -EINVAL;
86 }
87
88 /**
89  * netlbl_cfg_unlbl_map_add - Add a new unlabeled mapping
90  * @domain: the domain mapping to add
91  * @family: address family
92  * @addr: IP address
93  * @mask: IP address mask
94  * @audit_info: NetLabel audit information
95  *
96  * Description:
97  * Adds a new unlabeled NetLabel/LSM domain mapping.  A @domain value of NULL
98  * causes a new default domain mapping to be added.  Returns zero on success,
99  * negative values on failure.
100  *
101  */
102 int netlbl_cfg_unlbl_map_add(const char *domain,
103                              u16 family,
104                              const void *addr,
105                              const void *mask,
106                              struct netlbl_audit *audit_info)
107 {
108         int ret_val = -ENOMEM;
109         struct netlbl_dom_map *entry;
110         struct netlbl_domaddr_map *addrmap = NULL;
111         struct netlbl_domaddr4_map *map4 = NULL;
112         struct netlbl_domaddr6_map *map6 = NULL;
113
114         entry = kzalloc(sizeof(*entry), GFP_ATOMIC);
115         if (entry == NULL)
116                 return -ENOMEM;
117         if (domain != NULL) {
118                 entry->domain = kstrdup(domain, GFP_ATOMIC);
119                 if (entry->domain == NULL)
120                         goto cfg_unlbl_map_add_failure;
121         }
122
123         if (addr == NULL && mask == NULL)
124                 entry->def.type = NETLBL_NLTYPE_UNLABELED;
125         else if (addr != NULL && mask != NULL) {
126                 addrmap = kzalloc(sizeof(*addrmap), GFP_ATOMIC);
127                 if (addrmap == NULL)
128                         goto cfg_unlbl_map_add_failure;
129                 INIT_LIST_HEAD(&addrmap->list4);
130                 INIT_LIST_HEAD(&addrmap->list6);
131
132                 switch (family) {
133                 case AF_INET: {
134                         const struct in_addr *addr4 = addr;
135                         const struct in_addr *mask4 = mask;
136                         map4 = kzalloc(sizeof(*map4), GFP_ATOMIC);
137                         if (map4 == NULL)
138                                 goto cfg_unlbl_map_add_failure;
139                         map4->def.type = NETLBL_NLTYPE_UNLABELED;
140                         map4->list.addr = addr4->s_addr & mask4->s_addr;
141                         map4->list.mask = mask4->s_addr;
142                         map4->list.valid = 1;
143                         ret_val = netlbl_af4list_add(&map4->list,
144                                                      &addrmap->list4);
145                         if (ret_val != 0)
146                                 goto cfg_unlbl_map_add_failure;
147                         break;
148                         }
149 #if IS_ENABLED(CONFIG_IPV6)
150                 case AF_INET6: {
151                         const struct in6_addr *addr6 = addr;
152                         const struct in6_addr *mask6 = mask;
153                         map6 = kzalloc(sizeof(*map6), GFP_ATOMIC);
154                         if (map6 == NULL)
155                                 goto cfg_unlbl_map_add_failure;
156                         map6->def.type = NETLBL_NLTYPE_UNLABELED;
157                         map6->list.addr = *addr6;
158                         map6->list.addr.s6_addr32[0] &= mask6->s6_addr32[0];
159                         map6->list.addr.s6_addr32[1] &= mask6->s6_addr32[1];
160                         map6->list.addr.s6_addr32[2] &= mask6->s6_addr32[2];
161                         map6->list.addr.s6_addr32[3] &= mask6->s6_addr32[3];
162                         map6->list.mask = *mask6;
163                         map6->list.valid = 1;
164                         ret_val = netlbl_af6list_add(&map6->list,
165                                                      &addrmap->list6);
166                         if (ret_val != 0)
167                                 goto cfg_unlbl_map_add_failure;
168                         break;
169                         }
170 #endif /* IPv6 */
171                 default:
172                         goto cfg_unlbl_map_add_failure;
173                 }
174
175                 entry->def.addrsel = addrmap;
176                 entry->def.type = NETLBL_NLTYPE_ADDRSELECT;
177         } else {
178                 ret_val = -EINVAL;
179                 goto cfg_unlbl_map_add_failure;
180         }
181
182         ret_val = netlbl_domhsh_add(entry, audit_info);
183         if (ret_val != 0)
184                 goto cfg_unlbl_map_add_failure;
185
186         return 0;
187
188 cfg_unlbl_map_add_failure:
189         kfree(entry->domain);
190         kfree(entry);
191         kfree(addrmap);
192         kfree(map4);
193         kfree(map6);
194         return ret_val;
195 }
196
197
198 /**
199  * netlbl_cfg_unlbl_static_add - Adds a new static label
200  * @net: network namespace
201  * @dev_name: interface name
202  * @addr: IP address in network byte order (struct in[6]_addr)
203  * @mask: address mask in network byte order (struct in[6]_addr)
204  * @family: address family
205  * @secid: LSM secid value for the entry
206  * @audit_info: NetLabel audit information
207  *
208  * Description:
209  * Adds a new NetLabel static label to be used when protocol provided labels
210  * are not present on incoming traffic.  If @dev_name is NULL then the default
211  * interface will be used.  Returns zero on success, negative values on failure.
212  *
213  */
214 int netlbl_cfg_unlbl_static_add(struct net *net,
215                                 const char *dev_name,
216                                 const void *addr,
217                                 const void *mask,
218                                 u16 family,
219                                 u32 secid,
220                                 struct netlbl_audit *audit_info)
221 {
222         u32 addr_len;
223
224         switch (family) {
225         case AF_INET:
226                 addr_len = sizeof(struct in_addr);
227                 break;
228 #if IS_ENABLED(CONFIG_IPV6)
229         case AF_INET6:
230                 addr_len = sizeof(struct in6_addr);
231                 break;
232 #endif /* IPv6 */
233         default:
234                 return -EPFNOSUPPORT;
235         }
236
237         return netlbl_unlhsh_add(net,
238                                  dev_name, addr, mask, addr_len,
239                                  secid, audit_info);
240 }
241
242 /**
243  * netlbl_cfg_unlbl_static_del - Removes an existing static label
244  * @net: network namespace
245  * @dev_name: interface name
246  * @addr: IP address in network byte order (struct in[6]_addr)
247  * @mask: address mask in network byte order (struct in[6]_addr)
248  * @family: address family
249  * @secid: LSM secid value for the entry
250  * @audit_info: NetLabel audit information
251  *
252  * Description:
253  * Removes an existing NetLabel static label used when protocol provided labels
254  * are not present on incoming traffic.  If @dev_name is NULL then the default
255  * interface will be used.  Returns zero on success, negative values on failure.
256  *
257  */
258 int netlbl_cfg_unlbl_static_del(struct net *net,
259                                 const char *dev_name,
260                                 const void *addr,
261                                 const void *mask,
262                                 u16 family,
263                                 struct netlbl_audit *audit_info)
264 {
265         u32 addr_len;
266
267         switch (family) {
268         case AF_INET:
269                 addr_len = sizeof(struct in_addr);
270                 break;
271 #if IS_ENABLED(CONFIG_IPV6)
272         case AF_INET6:
273                 addr_len = sizeof(struct in6_addr);
274                 break;
275 #endif /* IPv6 */
276         default:
277                 return -EPFNOSUPPORT;
278         }
279
280         return netlbl_unlhsh_remove(net,
281                                     dev_name, addr, mask, addr_len,
282                                     audit_info);
283 }
284
285 /**
286  * netlbl_cfg_cipsov4_add - Add a new CIPSOv4 DOI definition
287  * @doi_def: CIPSO DOI definition
288  * @audit_info: NetLabel audit information
289  *
290  * Description:
291  * Add a new CIPSO DOI definition as defined by @doi_def.  Returns zero on
292  * success and negative values on failure.
293  *
294  */
295 int netlbl_cfg_cipsov4_add(struct cipso_v4_doi *doi_def,
296                            struct netlbl_audit *audit_info)
297 {
298         return cipso_v4_doi_add(doi_def, audit_info);
299 }
300
301 /**
302  * netlbl_cfg_cipsov4_del - Remove an existing CIPSOv4 DOI definition
303  * @doi: CIPSO DOI
304  * @audit_info: NetLabel audit information
305  *
306  * Description:
307  * Remove an existing CIPSO DOI definition matching @doi.  Returns zero on
308  * success and negative values on failure.
309  *
310  */
311 void netlbl_cfg_cipsov4_del(u32 doi, struct netlbl_audit *audit_info)
312 {
313         cipso_v4_doi_remove(doi, audit_info);
314 }
315
316 /**
317  * netlbl_cfg_cipsov4_map_add - Add a new CIPSOv4 DOI mapping
318  * @doi: the CIPSO DOI
319  * @domain: the domain mapping to add
320  * @addr: IP address
321  * @mask: IP address mask
322  * @audit_info: NetLabel audit information
323  *
324  * Description:
325  * Add a new NetLabel/LSM domain mapping for the given CIPSO DOI to the NetLabel
326  * subsystem.  A @domain value of NULL adds a new default domain mapping.
327  * Returns zero on success, negative values on failure.
328  *
329  */
330 int netlbl_cfg_cipsov4_map_add(u32 doi,
331                                const char *domain,
332                                const struct in_addr *addr,
333                                const struct in_addr *mask,
334                                struct netlbl_audit *audit_info)
335 {
336         int ret_val = -ENOMEM;
337         struct cipso_v4_doi *doi_def;
338         struct netlbl_dom_map *entry;
339         struct netlbl_domaddr_map *addrmap = NULL;
340         struct netlbl_domaddr4_map *addrinfo = NULL;
341
342         doi_def = cipso_v4_doi_getdef(doi);
343         if (doi_def == NULL)
344                 return -ENOENT;
345
346         entry = kzalloc(sizeof(*entry), GFP_ATOMIC);
347         if (entry == NULL)
348                 goto out_entry;
349         if (domain != NULL) {
350                 entry->domain = kstrdup(domain, GFP_ATOMIC);
351                 if (entry->domain == NULL)
352                         goto out_domain;
353         }
354
355         if (addr == NULL && mask == NULL) {
356                 entry->def.cipso = doi_def;
357                 entry->def.type = NETLBL_NLTYPE_CIPSOV4;
358         } else if (addr != NULL && mask != NULL) {
359                 addrmap = kzalloc(sizeof(*addrmap), GFP_ATOMIC);
360                 if (addrmap == NULL)
361                         goto out_addrmap;
362                 INIT_LIST_HEAD(&addrmap->list4);
363                 INIT_LIST_HEAD(&addrmap->list6);
364
365                 addrinfo = kzalloc(sizeof(*addrinfo), GFP_ATOMIC);
366                 if (addrinfo == NULL)
367                         goto out_addrinfo;
368                 addrinfo->def.cipso = doi_def;
369                 addrinfo->def.type = NETLBL_NLTYPE_CIPSOV4;
370                 addrinfo->list.addr = addr->s_addr & mask->s_addr;
371                 addrinfo->list.mask = mask->s_addr;
372                 addrinfo->list.valid = 1;
373                 ret_val = netlbl_af4list_add(&addrinfo->list, &addrmap->list4);
374                 if (ret_val != 0)
375                         goto cfg_cipsov4_map_add_failure;
376
377                 entry->def.addrsel = addrmap;
378                 entry->def.type = NETLBL_NLTYPE_ADDRSELECT;
379         } else {
380                 ret_val = -EINVAL;
381                 goto out_addrmap;
382         }
383
384         ret_val = netlbl_domhsh_add(entry, audit_info);
385         if (ret_val != 0)
386                 goto cfg_cipsov4_map_add_failure;
387
388         return 0;
389
390 cfg_cipsov4_map_add_failure:
391         kfree(addrinfo);
392 out_addrinfo:
393         kfree(addrmap);
394 out_addrmap:
395         kfree(entry->domain);
396 out_domain:
397         kfree(entry);
398 out_entry:
399         cipso_v4_doi_putdef(doi_def);
400         return ret_val;
401 }
402
403 /*
404  * Security Attribute Functions
405  */
406
407 #define _CM_F_NONE      0x00000000
408 #define _CM_F_ALLOC     0x00000001
409 #define _CM_F_WALK      0x00000002
410
411 /**
412  * _netlbl_catmap_getnode - Get a individual node from a catmap
413  * @catmap: pointer to the category bitmap
414  * @offset: the requested offset
415  * @cm_flags: catmap flags, see _CM_F_*
416  * @gfp_flags: memory allocation flags
417  *
418  * Description:
419  * Iterate through the catmap looking for the node associated with @offset.
420  * If the _CM_F_ALLOC flag is set in @cm_flags and there is no associated node,
421  * one will be created and inserted into the catmap.  If the _CM_F_WALK flag is
422  * set in @cm_flags and there is no associated node, the next highest node will
423  * be returned.  Returns a pointer to the node on success, NULL on failure.
424  *
425  */
426 static struct netlbl_lsm_catmap *_netlbl_catmap_getnode(
427                                              struct netlbl_lsm_catmap **catmap,
428                                              u32 offset,
429                                              unsigned int cm_flags,
430                                              gfp_t gfp_flags)
431 {
432         struct netlbl_lsm_catmap *iter = *catmap;
433         struct netlbl_lsm_catmap *prev = NULL;
434
435         if (iter == NULL)
436                 goto catmap_getnode_alloc;
437         if (offset < iter->startbit)
438                 goto catmap_getnode_walk;
439         while (iter && offset >= (iter->startbit + NETLBL_CATMAP_SIZE)) {
440                 prev = iter;
441                 iter = iter->next;
442         }
443         if (iter == NULL || offset < iter->startbit)
444                 goto catmap_getnode_walk;
445
446         return iter;
447
448 catmap_getnode_walk:
449         if (cm_flags & _CM_F_WALK)
450                 return iter;
451 catmap_getnode_alloc:
452         if (!(cm_flags & _CM_F_ALLOC))
453                 return NULL;
454
455         iter = netlbl_catmap_alloc(gfp_flags);
456         if (iter == NULL)
457                 return NULL;
458         iter->startbit = offset & ~(NETLBL_CATMAP_SIZE - 1);
459
460         if (prev == NULL) {
461                 iter->next = *catmap;
462                 *catmap = iter;
463         } else {
464                 iter->next = prev->next;
465                 prev->next = iter;
466         }
467
468         return iter;
469 }
470
471 /**
472  * netlbl_catmap_walk - Walk a LSM secattr catmap looking for a bit
473  * @catmap: the category bitmap
474  * @offset: the offset to start searching at, in bits
475  *
476  * Description:
477  * This function walks a LSM secattr category bitmap starting at @offset and
478  * returns the spot of the first set bit or -ENOENT if no bits are set.
479  *
480  */
481 int netlbl_catmap_walk(struct netlbl_lsm_catmap *catmap, u32 offset)
482 {
483         struct netlbl_lsm_catmap *iter = catmap;
484         u32 idx;
485         u32 bit;
486         NETLBL_CATMAP_MAPTYPE bitmap;
487
488         iter = _netlbl_catmap_getnode(&catmap, offset, _CM_F_WALK, 0);
489         if (iter == NULL)
490                 return -ENOENT;
491         if (offset > iter->startbit) {
492                 offset -= iter->startbit;
493                 idx = offset / NETLBL_CATMAP_MAPSIZE;
494                 bit = offset % NETLBL_CATMAP_MAPSIZE;
495         } else {
496                 idx = 0;
497                 bit = 0;
498         }
499         bitmap = iter->bitmap[idx] >> bit;
500
501         for (;;) {
502                 if (bitmap != 0) {
503                         while ((bitmap & NETLBL_CATMAP_BIT) == 0) {
504                                 bitmap >>= 1;
505                                 bit++;
506                         }
507                         return iter->startbit +
508                                (NETLBL_CATMAP_MAPSIZE * idx) + bit;
509                 }
510                 if (++idx >= NETLBL_CATMAP_MAPCNT) {
511                         if (iter->next != NULL) {
512                                 iter = iter->next;
513                                 idx = 0;
514                         } else
515                                 return -ENOENT;
516                 }
517                 bitmap = iter->bitmap[idx];
518                 bit = 0;
519         }
520
521         return -ENOENT;
522 }
523
524 /**
525  * netlbl_catmap_walkrng - Find the end of a string of set bits
526  * @catmap: the category bitmap
527  * @offset: the offset to start searching at, in bits
528  *
529  * Description:
530  * This function walks a LSM secattr category bitmap starting at @offset and
531  * returns the spot of the first cleared bit or -ENOENT if the offset is past
532  * the end of the bitmap.
533  *
534  */
535 int netlbl_catmap_walkrng(struct netlbl_lsm_catmap *catmap, u32 offset)
536 {
537         struct netlbl_lsm_catmap *iter;
538         struct netlbl_lsm_catmap *prev = NULL;
539         u32 idx;
540         u32 bit;
541         NETLBL_CATMAP_MAPTYPE bitmask;
542         NETLBL_CATMAP_MAPTYPE bitmap;
543
544         iter = _netlbl_catmap_getnode(&catmap, offset, _CM_F_WALK, 0);
545         if (iter == NULL)
546                 return -ENOENT;
547         if (offset > iter->startbit) {
548                 offset -= iter->startbit;
549                 idx = offset / NETLBL_CATMAP_MAPSIZE;
550                 bit = offset % NETLBL_CATMAP_MAPSIZE;
551         } else {
552                 idx = 0;
553                 bit = 0;
554         }
555         bitmask = NETLBL_CATMAP_BIT << bit;
556
557         for (;;) {
558                 bitmap = iter->bitmap[idx];
559                 while (bitmask != 0 && (bitmap & bitmask) != 0) {
560                         bitmask <<= 1;
561                         bit++;
562                 }
563
564                 if (prev && idx == 0 && bit == 0)
565                         return prev->startbit + NETLBL_CATMAP_SIZE - 1;
566                 else if (bitmask != 0)
567                         return iter->startbit +
568                                 (NETLBL_CATMAP_MAPSIZE * idx) + bit - 1;
569                 else if (++idx >= NETLBL_CATMAP_MAPCNT) {
570                         if (iter->next == NULL)
571                                 return iter->startbit + NETLBL_CATMAP_SIZE - 1;
572                         prev = iter;
573                         iter = iter->next;
574                         idx = 0;
575                 }
576                 bitmask = NETLBL_CATMAP_BIT;
577                 bit = 0;
578         }
579
580         return -ENOENT;
581 }
582
583 /**
584  * netlbl_catmap_getlong - Export an unsigned long bitmap
585  * @catmap: pointer to the category bitmap
586  * @offset: pointer to the requested offset
587  * @bitmap: the exported bitmap
588  *
589  * Description:
590  * Export a bitmap with an offset greater than or equal to @offset and return
591  * it in @bitmap.  The @offset must be aligned to an unsigned long and will be
592  * updated on return if different from what was requested; if the catmap is
593  * empty at the requested offset and beyond, the @offset is set to (u32)-1.
594  * Returns zero on sucess, negative values on failure.
595  *
596  */
597 int netlbl_catmap_getlong(struct netlbl_lsm_catmap *catmap,
598                           u32 *offset,
599                           unsigned long *bitmap)
600 {
601         struct netlbl_lsm_catmap *iter;
602         u32 off = *offset;
603         u32 idx;
604
605         /* only allow aligned offsets */
606         if ((off & (BITS_PER_LONG - 1)) != 0)
607                 return -EINVAL;
608
609         if (off < catmap->startbit) {
610                 off = catmap->startbit;
611                 *offset = off;
612         }
613         iter = _netlbl_catmap_getnode(&catmap, off, _CM_F_NONE, 0);
614         if (iter == NULL) {
615                 *offset = (u32)-1;
616                 return 0;
617         }
618
619         if (off < iter->startbit) {
620                 off = iter->startbit;
621                 *offset = off;
622         } else
623                 off -= iter->startbit;
624
625         idx = off / NETLBL_CATMAP_MAPSIZE;
626         *bitmap = iter->bitmap[idx] >> (off % NETLBL_CATMAP_SIZE);
627
628         return 0;
629 }
630
631 /**
632  * netlbl_catmap_setbit - Set a bit in a LSM secattr catmap
633  * @catmap: pointer to the category bitmap
634  * @bit: the bit to set
635  * @flags: memory allocation flags
636  *
637  * Description:
638  * Set the bit specified by @bit in @catmap.  Returns zero on success,
639  * negative values on failure.
640  *
641  */
642 int netlbl_catmap_setbit(struct netlbl_lsm_catmap **catmap,
643                          u32 bit,
644                          gfp_t flags)
645 {
646         struct netlbl_lsm_catmap *iter;
647         u32 idx;
648
649         iter = _netlbl_catmap_getnode(catmap, bit, _CM_F_ALLOC, flags);
650         if (iter == NULL)
651                 return -ENOMEM;
652
653         bit -= iter->startbit;
654         idx = bit / NETLBL_CATMAP_MAPSIZE;
655         iter->bitmap[idx] |= NETLBL_CATMAP_BIT << (bit % NETLBL_CATMAP_MAPSIZE);
656
657         return 0;
658 }
659
660 /**
661  * netlbl_catmap_setrng - Set a range of bits in a LSM secattr catmap
662  * @catmap: pointer to the category bitmap
663  * @start: the starting bit
664  * @end: the last bit in the string
665  * @flags: memory allocation flags
666  *
667  * Description:
668  * Set a range of bits, starting at @start and ending with @end.  Returns zero
669  * on success, negative values on failure.
670  *
671  */
672 int netlbl_catmap_setrng(struct netlbl_lsm_catmap **catmap,
673                          u32 start,
674                          u32 end,
675                          gfp_t flags)
676 {
677         int rc = 0;
678         u32 spot = start;
679
680         while (rc == 0 && spot <= end) {
681                 if (((spot & (BITS_PER_LONG - 1)) != 0) &&
682                     ((end - spot) > BITS_PER_LONG)) {
683                         rc = netlbl_catmap_setlong(catmap,
684                                                    spot,
685                                                    (unsigned long)-1,
686                                                    flags);
687                         spot += BITS_PER_LONG;
688                 } else
689                         rc = netlbl_catmap_setbit(catmap, spot++, flags);
690         }
691
692         return rc;
693 }
694
695 /**
696  * netlbl_catmap_setlong - Import an unsigned long bitmap
697  * @catmap: pointer to the category bitmap
698  * @offset: offset to the start of the imported bitmap
699  * @bitmap: the bitmap to import
700  * @flags: memory allocation flags
701  *
702  * Description:
703  * Import the bitmap specified in @bitmap into @catmap, using the offset
704  * in @offset.  The offset must be aligned to an unsigned long.  Returns zero
705  * on success, negative values on failure.
706  *
707  */
708 int netlbl_catmap_setlong(struct netlbl_lsm_catmap **catmap,
709                           u32 offset,
710                           unsigned long bitmap,
711                           gfp_t flags)
712 {
713         struct netlbl_lsm_catmap *iter;
714         u32 idx;
715
716         /* only allow aligned offsets */
717         if ((offset & (BITS_PER_LONG - 1)) != 0)
718                 return -EINVAL;
719
720         iter = _netlbl_catmap_getnode(catmap, offset, _CM_F_ALLOC, flags);
721         if (iter == NULL)
722                 return -ENOMEM;
723
724         offset -= iter->startbit;
725         idx = offset / NETLBL_CATMAP_MAPSIZE;
726         iter->bitmap[idx] |= bitmap << (offset % NETLBL_CATMAP_MAPSIZE);
727
728         return 0;
729 }
730
731 /*
732  * LSM Functions
733  */
734
735 /**
736  * netlbl_enabled - Determine if the NetLabel subsystem is enabled
737  *
738  * Description:
739  * The LSM can use this function to determine if it should use NetLabel
740  * security attributes in it's enforcement mechanism.  Currently, NetLabel is
741  * considered to be enabled when it's configuration contains a valid setup for
742  * at least one labeled protocol (i.e. NetLabel can understand incoming
743  * labeled packets of at least one type); otherwise NetLabel is considered to
744  * be disabled.
745  *
746  */
747 int netlbl_enabled(void)
748 {
749         /* At some point we probably want to expose this mechanism to the user
750          * as well so that admins can toggle NetLabel regardless of the
751          * configuration */
752         return (atomic_read(&netlabel_mgmt_protocount) > 0);
753 }
754
755 /**
756  * netlbl_sock_setattr - Label a socket using the correct protocol
757  * @sk: the socket to label
758  * @family: protocol family
759  * @secattr: the security attributes
760  *
761  * Description:
762  * Attach the correct label to the given socket using the security attributes
763  * specified in @secattr.  This function requires exclusive access to @sk,
764  * which means it either needs to be in the process of being created or locked.
765  * Returns zero on success, -EDESTADDRREQ if the domain is configured to use
766  * network address selectors (can't blindly label the socket), and negative
767  * values on all other failures.
768  *
769  */
770 int netlbl_sock_setattr(struct sock *sk,
771                         u16 family,
772                         const struct netlbl_lsm_secattr *secattr)
773 {
774         int ret_val;
775         struct netlbl_dom_map *dom_entry;
776
777         rcu_read_lock();
778         dom_entry = netlbl_domhsh_getentry(secattr->domain);
779         if (dom_entry == NULL) {
780                 ret_val = -ENOENT;
781                 goto socket_setattr_return;
782         }
783         switch (family) {
784         case AF_INET:
785                 switch (dom_entry->def.type) {
786                 case NETLBL_NLTYPE_ADDRSELECT:
787                         ret_val = -EDESTADDRREQ;
788                         break;
789                 case NETLBL_NLTYPE_CIPSOV4:
790                         ret_val = cipso_v4_sock_setattr(sk,
791                                                         dom_entry->def.cipso,
792                                                         secattr);
793                         break;
794                 case NETLBL_NLTYPE_UNLABELED:
795                         ret_val = 0;
796                         break;
797                 default:
798                         ret_val = -ENOENT;
799                 }
800                 break;
801 #if IS_ENABLED(CONFIG_IPV6)
802         case AF_INET6:
803                 /* since we don't support any IPv6 labeling protocols right
804                  * now we can optimize everything away until we do */
805                 ret_val = 0;
806                 break;
807 #endif /* IPv6 */
808         default:
809                 ret_val = -EPROTONOSUPPORT;
810         }
811
812 socket_setattr_return:
813         rcu_read_unlock();
814         return ret_val;
815 }
816
817 /**
818  * netlbl_sock_delattr - Delete all the NetLabel labels on a socket
819  * @sk: the socket
820  *
821  * Description:
822  * Remove all the NetLabel labeling from @sk.  The caller is responsible for
823  * ensuring that @sk is locked.
824  *
825  */
826 void netlbl_sock_delattr(struct sock *sk)
827 {
828         cipso_v4_sock_delattr(sk);
829 }
830
831 /**
832  * netlbl_sock_getattr - Determine the security attributes of a sock
833  * @sk: the sock
834  * @secattr: the security attributes
835  *
836  * Description:
837  * Examines the given sock to see if any NetLabel style labeling has been
838  * applied to the sock, if so it parses the socket label and returns the
839  * security attributes in @secattr.  Returns zero on success, negative values
840  * on failure.
841  *
842  */
843 int netlbl_sock_getattr(struct sock *sk,
844                         struct netlbl_lsm_secattr *secattr)
845 {
846         int ret_val;
847
848         switch (sk->sk_family) {
849         case AF_INET:
850                 ret_val = cipso_v4_sock_getattr(sk, secattr);
851                 break;
852 #if IS_ENABLED(CONFIG_IPV6)
853         case AF_INET6:
854                 ret_val = -ENOMSG;
855                 break;
856 #endif /* IPv6 */
857         default:
858                 ret_val = -EPROTONOSUPPORT;
859         }
860
861         return ret_val;
862 }
863
864 /**
865  * netlbl_conn_setattr - Label a connected socket using the correct protocol
866  * @sk: the socket to label
867  * @addr: the destination address
868  * @secattr: the security attributes
869  *
870  * Description:
871  * Attach the correct label to the given connected socket using the security
872  * attributes specified in @secattr.  The caller is responsible for ensuring
873  * that @sk is locked.  Returns zero on success, negative values on failure.
874  *
875  */
876 int netlbl_conn_setattr(struct sock *sk,
877                         struct sockaddr *addr,
878                         const struct netlbl_lsm_secattr *secattr)
879 {
880         int ret_val;
881         struct sockaddr_in *addr4;
882         struct netlbl_dommap_def *entry;
883
884         rcu_read_lock();
885         switch (addr->sa_family) {
886         case AF_INET:
887                 addr4 = (struct sockaddr_in *)addr;
888                 entry = netlbl_domhsh_getentry_af4(secattr->domain,
889                                                    addr4->sin_addr.s_addr);
890                 if (entry == NULL) {
891                         ret_val = -ENOENT;
892                         goto conn_setattr_return;
893                 }
894                 switch (entry->type) {
895                 case NETLBL_NLTYPE_CIPSOV4:
896                         ret_val = cipso_v4_sock_setattr(sk,
897                                                         entry->cipso, secattr);
898                         break;
899                 case NETLBL_NLTYPE_UNLABELED:
900                         /* just delete the protocols we support for right now
901                          * but we could remove other protocols if needed */
902                         cipso_v4_sock_delattr(sk);
903                         ret_val = 0;
904                         break;
905                 default:
906                         ret_val = -ENOENT;
907                 }
908                 break;
909 #if IS_ENABLED(CONFIG_IPV6)
910         case AF_INET6:
911                 /* since we don't support any IPv6 labeling protocols right
912                  * now we can optimize everything away until we do */
913                 ret_val = 0;
914                 break;
915 #endif /* IPv6 */
916         default:
917                 ret_val = -EPROTONOSUPPORT;
918         }
919
920 conn_setattr_return:
921         rcu_read_unlock();
922         return ret_val;
923 }
924
925 /**
926  * netlbl_req_setattr - Label a request socket using the correct protocol
927  * @req: the request socket to label
928  * @secattr: the security attributes
929  *
930  * Description:
931  * Attach the correct label to the given socket using the security attributes
932  * specified in @secattr.  Returns zero on success, negative values on failure.
933  *
934  */
935 int netlbl_req_setattr(struct request_sock *req,
936                        const struct netlbl_lsm_secattr *secattr)
937 {
938         int ret_val;
939         struct netlbl_dommap_def *entry;
940
941         rcu_read_lock();
942         switch (req->rsk_ops->family) {
943         case AF_INET:
944                 entry = netlbl_domhsh_getentry_af4(secattr->domain,
945                                                    inet_rsk(req)->ir_rmt_addr);
946                 if (entry == NULL) {
947                         ret_val = -ENOENT;
948                         goto req_setattr_return;
949                 }
950                 switch (entry->type) {
951                 case NETLBL_NLTYPE_CIPSOV4:
952                         ret_val = cipso_v4_req_setattr(req,
953                                                        entry->cipso, secattr);
954                         break;
955                 case NETLBL_NLTYPE_UNLABELED:
956                         /* just delete the protocols we support for right now
957                          * but we could remove other protocols if needed */
958                         cipso_v4_req_delattr(req);
959                         ret_val = 0;
960                         break;
961                 default:
962                         ret_val = -ENOENT;
963                 }
964                 break;
965 #if IS_ENABLED(CONFIG_IPV6)
966         case AF_INET6:
967                 /* since we don't support any IPv6 labeling protocols right
968                  * now we can optimize everything away until we do */
969                 ret_val = 0;
970                 break;
971 #endif /* IPv6 */
972         default:
973                 ret_val = -EPROTONOSUPPORT;
974         }
975
976 req_setattr_return:
977         rcu_read_unlock();
978         return ret_val;
979 }
980
981 /**
982 * netlbl_req_delattr - Delete all the NetLabel labels on a socket
983 * @req: the socket
984 *
985 * Description:
986 * Remove all the NetLabel labeling from @req.
987 *
988 */
989 void netlbl_req_delattr(struct request_sock *req)
990 {
991         cipso_v4_req_delattr(req);
992 }
993
994 /**
995  * netlbl_skbuff_setattr - Label a packet using the correct protocol
996  * @skb: the packet
997  * @family: protocol family
998  * @secattr: the security attributes
999  *
1000  * Description:
1001  * Attach the correct label to the given packet using the security attributes
1002  * specified in @secattr.  Returns zero on success, negative values on failure.
1003  *
1004  */
1005 int netlbl_skbuff_setattr(struct sk_buff *skb,
1006                           u16 family,
1007                           const struct netlbl_lsm_secattr *secattr)
1008 {
1009         int ret_val;
1010         struct iphdr *hdr4;
1011         struct netlbl_dommap_def *entry;
1012
1013         rcu_read_lock();
1014         switch (family) {
1015         case AF_INET:
1016                 hdr4 = ip_hdr(skb);
1017                 entry = netlbl_domhsh_getentry_af4(secattr->domain,hdr4->daddr);
1018                 if (entry == NULL) {
1019                         ret_val = -ENOENT;
1020                         goto skbuff_setattr_return;
1021                 }
1022                 switch (entry->type) {
1023                 case NETLBL_NLTYPE_CIPSOV4:
1024                         ret_val = cipso_v4_skbuff_setattr(skb, entry->cipso,
1025                                                           secattr);
1026                         break;
1027                 case NETLBL_NLTYPE_UNLABELED:
1028                         /* just delete the protocols we support for right now
1029                          * but we could remove other protocols if needed */
1030                         ret_val = cipso_v4_skbuff_delattr(skb);
1031                         break;
1032                 default:
1033                         ret_val = -ENOENT;
1034                 }
1035                 break;
1036 #if IS_ENABLED(CONFIG_IPV6)
1037         case AF_INET6:
1038                 /* since we don't support any IPv6 labeling protocols right
1039                  * now we can optimize everything away until we do */
1040                 ret_val = 0;
1041                 break;
1042 #endif /* IPv6 */
1043         default:
1044                 ret_val = -EPROTONOSUPPORT;
1045         }
1046
1047 skbuff_setattr_return:
1048         rcu_read_unlock();
1049         return ret_val;
1050 }
1051
1052 /**
1053  * netlbl_skbuff_getattr - Determine the security attributes of a packet
1054  * @skb: the packet
1055  * @family: protocol family
1056  * @secattr: the security attributes
1057  *
1058  * Description:
1059  * Examines the given packet to see if a recognized form of packet labeling
1060  * is present, if so it parses the packet label and returns the security
1061  * attributes in @secattr.  Returns zero on success, negative values on
1062  * failure.
1063  *
1064  */
1065 int netlbl_skbuff_getattr(const struct sk_buff *skb,
1066                           u16 family,
1067                           struct netlbl_lsm_secattr *secattr)
1068 {
1069         switch (family) {
1070         case AF_INET:
1071                 if (CIPSO_V4_OPTEXIST(skb) &&
1072                     cipso_v4_skbuff_getattr(skb, secattr) == 0)
1073                         return 0;
1074                 break;
1075 #if IS_ENABLED(CONFIG_IPV6)
1076         case AF_INET6:
1077                 break;
1078 #endif /* IPv6 */
1079         }
1080
1081         return netlbl_unlabel_getattr(skb, family, secattr);
1082 }
1083
1084 /**
1085  * netlbl_skbuff_err - Handle a LSM error on a sk_buff
1086  * @skb: the packet
1087  * @error: the error code
1088  * @gateway: true if host is acting as a gateway, false otherwise
1089  *
1090  * Description:
1091  * Deal with a LSM problem when handling the packet in @skb, typically this is
1092  * a permission denied problem (-EACCES).  The correct action is determined
1093  * according to the packet's labeling protocol.
1094  *
1095  */
1096 void netlbl_skbuff_err(struct sk_buff *skb, int error, int gateway)
1097 {
1098         if (CIPSO_V4_OPTEXIST(skb))
1099                 cipso_v4_error(skb, error, gateway);
1100 }
1101
1102 /**
1103  * netlbl_cache_invalidate - Invalidate all of the NetLabel protocol caches
1104  *
1105  * Description:
1106  * For all of the NetLabel protocols that support some form of label mapping
1107  * cache, invalidate the cache.  Returns zero on success, negative values on
1108  * error.
1109  *
1110  */
1111 void netlbl_cache_invalidate(void)
1112 {
1113         cipso_v4_cache_invalidate();
1114 }
1115
1116 /**
1117  * netlbl_cache_add - Add an entry to a NetLabel protocol cache
1118  * @skb: the packet
1119  * @secattr: the packet's security attributes
1120  *
1121  * Description:
1122  * Add the LSM security attributes for the given packet to the underlying
1123  * NetLabel protocol's label mapping cache.  Returns zero on success, negative
1124  * values on error.
1125  *
1126  */
1127 int netlbl_cache_add(const struct sk_buff *skb,
1128                      const struct netlbl_lsm_secattr *secattr)
1129 {
1130         if ((secattr->flags & NETLBL_SECATTR_CACHE) == 0)
1131                 return -ENOMSG;
1132
1133         if (CIPSO_V4_OPTEXIST(skb))
1134                 return cipso_v4_cache_add(skb, secattr);
1135
1136         return -ENOMSG;
1137 }
1138
1139 /*
1140  * Protocol Engine Functions
1141  */
1142
1143 /**
1144  * netlbl_audit_start - Start an audit message
1145  * @type: audit message type
1146  * @audit_info: NetLabel audit information
1147  *
1148  * Description:
1149  * Start an audit message using the type specified in @type and fill the audit
1150  * message with some fields common to all NetLabel audit messages.  This
1151  * function should only be used by protocol engines, not LSMs.  Returns a
1152  * pointer to the audit buffer on success, NULL on failure.
1153  *
1154  */
1155 struct audit_buffer *netlbl_audit_start(int type,
1156                                         struct netlbl_audit *audit_info)
1157 {
1158         return netlbl_audit_start_common(type, audit_info);
1159 }
1160
1161 /*
1162  * Setup Functions
1163  */
1164
1165 /**
1166  * netlbl_init - Initialize NetLabel
1167  *
1168  * Description:
1169  * Perform the required NetLabel initialization before first use.
1170  *
1171  */
1172 static int __init netlbl_init(void)
1173 {
1174         int ret_val;
1175
1176         printk(KERN_INFO "NetLabel: Initializing\n");
1177         printk(KERN_INFO "NetLabel:  domain hash size = %u\n",
1178                (1 << NETLBL_DOMHSH_BITSIZE));
1179         printk(KERN_INFO "NetLabel:  protocols ="
1180                " UNLABELED"
1181                " CIPSOv4"
1182                "\n");
1183
1184         ret_val = netlbl_domhsh_init(NETLBL_DOMHSH_BITSIZE);
1185         if (ret_val != 0)
1186                 goto init_failure;
1187
1188         ret_val = netlbl_unlabel_init(NETLBL_UNLHSH_BITSIZE);
1189         if (ret_val != 0)
1190                 goto init_failure;
1191
1192         ret_val = netlbl_netlink_init();
1193         if (ret_val != 0)
1194                 goto init_failure;
1195
1196         ret_val = netlbl_unlabel_defconf();
1197         if (ret_val != 0)
1198                 goto init_failure;
1199         printk(KERN_INFO "NetLabel:  unlabeled traffic allowed by default\n");
1200
1201         return 0;
1202
1203 init_failure:
1204         panic("NetLabel: failed to initialize properly (%d)\n", ret_val);
1205 }
1206
1207 subsys_initcall(netlbl_init);