net/netfilter/xt_SECMARK.c

   1 /*
   2  * Module for modifying the secmark field of the skb, for use by
   3  * security subsystems.
   4  *
   5  * Based on the nfmark match by:
   6  * (C) 1999-2001 Marc Boucher <marc@mbsi.ca>
   7  *
   8  * (C) 2006 Red Hat, Inc., James Morris <jmorris@redhat.com>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License version 2 as
  12  * published by the Free Software Foundation.
  13  *
  14  */
  15 #include <linux/module.h>
  16 #include <linux/skbuff.h>
  17 #include <linux/selinux.h>
  18 #include <linux/netfilter/x_tables.h>
  19 #include <linux/netfilter/xt_SECMARK.h>
  20
  21 MODULE_LICENSE("GPL");
  22 MODULE_AUTHOR("James Morris <jmorris@redhat.com>");
  23 MODULE_DESCRIPTION("ip[6]tables SECMARK modification module");
  24 MODULE_ALIAS("ipt_SECMARK");
  25 MODULE_ALIAS("ip6t_SECMARK");
  26
  27 #define PFX "SECMARK: "
  28
  29 static u8 mode;
  30
  31 static unsigned int target(struct sk_buff **pskb, const struct net_device *in,
  32                            const struct net_device *out, unsigned int hooknum,
  33                            const struct xt_target *target,
  34                            const void *targinfo, void *userinfo)
  35 {
  36         u32 secmark = 0;
  37         const struct xt_secmark_target_info *info = targinfo;
  38
  39         BUG_ON(info->mode != mode);
  40
  41         switch (mode) {
  42         case SECMARK_MODE_SEL:
  43                 secmark = info->u.sel.selsid;
  44                 break;
  45
  46         default:
  47                 BUG();
  48         }
  49
  50         if ((*pskb)->secmark != secmark)
  51                 (*pskb)->secmark = secmark;
  52
  53         return XT_CONTINUE;
  54 }
  55
  56 static int checkentry_selinux(struct xt_secmark_target_info *info)
  57 {
  58         int err;
  59         struct xt_secmark_target_selinux_info *sel = &info->u.sel;
  60
  61         sel->selctx[SECMARK_SELCTX_MAX - 1] = '\0';
  62
  63         err = selinux_string_to_sid(sel->selctx, &sel->selsid);
  64         if (err) {
  65                 if (err == -EINVAL)
  66                         printk(KERN_INFO PFX "invalid SELinux context \'%s\'\n",
  67                                sel->selctx);
  68                 return 0;
  69         }
  70
  71         if (!sel->selsid) {
  72                 printk(KERN_INFO PFX "unable to map SELinux context \'%s\'\n",
  73                        sel->selctx);
  74                 return 0;
  75         }
  76
  77         err = selinux_relabel_packet_permission(sel->selsid);
  78         if (err) {
  79                 printk(KERN_INFO PFX "unable to obtain relabeling permission\n");
  80                 return 0;
  81         }
  82
  83         return 1;
  84 }
  85
  86 static int checkentry(const char *tablename, const void *entry,
  87                       const struct xt_target *target, void *targinfo,
  88                       unsigned int targinfosize, unsigned int hook_mask)
  89 {
  90         struct xt_secmark_target_info *info = targinfo;
  91
  92         if (mode && mode != info->mode) {
  93                 printk(KERN_INFO PFX "mode already set to %hu cannot mix with "
  94                        "rules for mode %hu\n", mode, info->mode);
  95                 return 0;
  96         }
  97
  98         switch (info->mode) {
  99         case SECMARK_MODE_SEL:
 100                 if (!checkentry_selinux(info))
 101                         return 0;
 102                 break;
 103
 104         default:
 105                 printk(KERN_INFO PFX "invalid mode: %hu\n", info->mode);
 106                 return 0;
 107         }
 108
 109         if (!mode)
 110                 mode = info->mode;
 111         return 1;
 112 }
 113
 114 static struct xt_target ipt_secmark_reg = {
 115         .name           = "SECMARK",
 116         .target         = target,
 117         .targetsize     = sizeof(struct xt_secmark_target_info),
 118         .table          = "mangle",
 119         .checkentry     = checkentry,
 120         .me             = THIS_MODULE,
 121         .family         = AF_INET,
 122         .revision       = 0,
 123 };
 124
 125 static struct xt_target ip6t_secmark_reg = {
 126         .name           = "SECMARK",
 127         .target         = target,
 128         .targetsize     = sizeof(struct xt_secmark_target_info),
 129         .table          = "mangle",
 130         .checkentry     = checkentry,
 131         .me             = THIS_MODULE,
 132         .family         = AF_INET6,
 133         .revision       = 0,
 134 };
 135
 136 static int __init xt_secmark_init(void)
 137 {
 138         int err;
 139
 140         err = xt_register_target(&ipt_secmark_reg);
 141         if (err)
 142                 return err;
 143
 144         err = xt_register_target(&ip6t_secmark_reg);
 145         if (err)
 146                 xt_unregister_target(&ipt_secmark_reg);
 147
 148         return err;
 149 }
 150
 151 static void __exit xt_secmark_fini(void)
 152 {
 153         xt_unregister_target(&ip6t_secmark_reg);
 154         xt_unregister_target(&ipt_secmark_reg);
 155 }
 156
 157 module_init(xt_secmark_init);
 158 module_exit(xt_secmark_fini);