[GitHub/mt8127/android_kernel_alcatel_ttab.git] / net / netfilter / xt_SECMARK.c

/*
 * Module for modifying the secmark field of the skb, for use by
 * security subsystems.
 *
 * Based on the nfmark match by:
 * (C) 1999-2001 Marc Boucher <marc@mbsi.ca>
 *
 * (C) 2006 Red Hat, Inc., James Morris <jmorris@redhat.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 */
#include <linux/module.h>
#include <linux/skbuff.h>
#include <linux/selinux.h>
#include <linux/netfilter/x_tables.h>
#include <linux/netfilter/xt_SECMARK.h>

MODULE_LICENSE("GPL");
MODULE_AUTHOR("James Morris <jmorris@redhat.com>");
MODULE_DESCRIPTION("ip[6]tables SECMARK modification module");
MODULE_ALIAS("ipt_SECMARK");
MODULE_ALIAS("ip6t_SECMARK");

#define PFX "SECMARK: "

static u8 mode;

static unsigned int target(struct sk_buff **pskb, const struct net_device *in,
			   const struct net_device *out, unsigned int hooknum,
			   const struct xt_target *target,
			   const void *targinfo)
{
	u32 secmark = 0;
	const struct xt_secmark_target_info *info = targinfo;

	BUG_ON(info->mode != mode);

	switch (mode) {
	case SECMARK_MODE_SEL:
		secmark = info->u.sel.selsid;
		break;

	default:
		BUG();
	}

	(*pskb)->secmark = secmark;
	return XT_CONTINUE;
}

static bool checkentry_selinux(struct xt_secmark_target_info *info)
{
	int err;
	struct xt_secmark_target_selinux_info *sel = &info->u.sel;

	sel->selctx[SECMARK_SELCTX_MAX - 1] = '\0';

	err = selinux_string_to_sid(sel->selctx, &sel->selsid);
	if (err) {
		if (err == -EINVAL)
			printk(KERN_INFO PFX "invalid SELinux context \'%s\'\n",
			       sel->selctx);
		return false;
	}

	if (!sel->selsid) {
		printk(KERN_INFO PFX "unable to map SELinux context \'%s\'\n",
		       sel->selctx);
		return false;
	}

	err = selinux_relabel_packet_permission(sel->selsid);
	if (err) {
		printk(KERN_INFO PFX "unable to obtain relabeling permission\n");
		return false;
	}

	return true;
}

static bool checkentry(const char *tablename, const void *entry,
		       const struct xt_target *target, void *targinfo,
		       unsigned int hook_mask)
{
	struct xt_secmark_target_info *info = targinfo;

	if (mode && mode != info->mode) {
		printk(KERN_INFO PFX "mode already set to %hu cannot mix with "
		       "rules for mode %hu\n", mode, info->mode);
		return false;
	}

	switch (info->mode) {
	case SECMARK_MODE_SEL:
		if (!checkentry_selinux(info))
			return false;
		break;

	default:
		printk(KERN_INFO PFX "invalid mode: %hu\n", info->mode);
		return false;
	}

	if (!mode)
		mode = info->mode;
	return true;
}

static struct xt_target xt_secmark_target[] __read_mostly = {
	{
		.name		= "SECMARK",
		.family		= AF_INET,
		.checkentry	= checkentry,
		.target		= target,
		.targetsize	= sizeof(struct xt_secmark_target_info),
		.table		= "mangle",
		.me		= THIS_MODULE,
	},
	{
		.name		= "SECMARK",
		.family		= AF_INET6,
		.checkentry	= checkentry,
		.target		= target,
		.targetsize	= sizeof(struct xt_secmark_target_info),
		.table		= "mangle",
		.me		= THIS_MODULE,
	},
};

static int __init xt_secmark_init(void)
{
	return xt_register_targets(xt_secmark_target,
				   ARRAY_SIZE(xt_secmark_target));
}

static void __exit xt_secmark_fini(void)
{
	xt_unregister_targets(xt_secmark_target, ARRAY_SIZE(xt_secmark_target));
}

module_init(xt_secmark_init);
module_exit(xt_secmark_fini);
Commit	Line	Data
5e6874cd JM	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,
fe1cb108	34	const void *targinfo)
5e6874cd JM	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
2822b0d9	50	(*pskb)->secmark = secmark;
5e6874cd JM	51	return XT_CONTINUE;
	52	}
	53
e1931b78	54	static bool checkentry_selinux(struct xt_secmark_target_info *info)
5e6874cd JM	55	{
	56	int err;
	57	struct xt_secmark_target_selinux_info *sel = &info->u.sel;
601e68e1	58
a280b899	59	sel->selctx[SECMARK_SELCTX_MAX - 1] = '\0';
5e6874cd JM	60
	61	err = selinux_string_to_sid(sel->selctx, &sel->selsid);
	62	if (err) {
	63	if (err == -EINVAL)
	64	printk(KERN_INFO PFX "invalid SELinux context \'%s\'\n",
	65	sel->selctx);
e1931b78	66	return false;
5e6874cd JM	67	}
	68
	69	if (!sel->selsid) {
	70	printk(KERN_INFO PFX "unable to map SELinux context \'%s\'\n",
	71	sel->selctx);
e1931b78	72	return false;
5e6874cd JM	73	}
	74
	75	err = selinux_relabel_packet_permission(sel->selsid);
	76	if (err) {
	77	printk(KERN_INFO PFX "unable to obtain relabeling permission\n");
e1931b78	78	return false;
5e6874cd JM	79	}
5e6874cd JM	80
e1931b78	81	return true;
5e6874cd JM	82	}
5e6874cd JM	83
e1931b78 JE	84	static bool checkentry(const char tablename, const void entry,
	85	const struct xt_target target, void targinfo,
	86	unsigned int hook_mask)
5e6874cd JM	87	{
	88	struct xt_secmark_target_info *info = targinfo;
	89
	90	if (mode && mode != info->mode) {
	91	printk(KERN_INFO PFX "mode already set to %hu cannot mix with "
	92	"rules for mode %hu\n", mode, info->mode);
e1931b78	93	return false;
5e6874cd JM	94	}
	95
	96	switch (info->mode) {
	97	case SECMARK_MODE_SEL:
	98	if (!checkentry_selinux(info))
e1931b78	99	return false;
5e6874cd JM	100	break;
	101
	102	default:
	103	printk(KERN_INFO PFX "invalid mode: %hu\n", info->mode);
e1931b78	104	return false;
5e6874cd JM	105	}
	106
	107	if (!mode)
	108	mode = info->mode;
e1931b78	109	return true;
5e6874cd JM	110	}
5e6874cd JM	111
9f15c530	112	static struct xt_target xt_secmark_target[] __read_mostly = {
4470bbc7 PM	113	{
	114	.name = "SECMARK",
	115	.family = AF_INET,
	116	.checkentry = checkentry,
	117	.target = target,
	118	.targetsize = sizeof(struct xt_secmark_target_info),
	119	.table = "mangle",
	120	.me = THIS_MODULE,
	121	},
	122	{
	123	.name = "SECMARK",
	124	.family = AF_INET6,
	125	.checkentry = checkentry,
	126	.target = target,
	127	.targetsize = sizeof(struct xt_secmark_target_info),
	128	.table = "mangle",
	129	.me = THIS_MODULE,
	130	},
5e6874cd JM	131	};
	132
	133	static int __init xt_secmark_init(void)
	134	{
4470bbc7 PM	135	return xt_register_targets(xt_secmark_target,
4470bbc7 PM	136	ARRAY_SIZE(xt_secmark_target));
5e6874cd JM	137	}
	138
	139	static void __exit xt_secmark_fini(void)
	140	{
4470bbc7	141	xt_unregister_targets(xt_secmark_target, ARRAY_SIZE(xt_secmark_target));
5e6874cd JM	142	}
	143
	144	module_init(xt_secmark_init);
	145	module_exit(xt_secmark_fini);