[linux.git] / net / netlabel / netlabel_mgmt.c

/*
 * NetLabel Management Support
 *
 * This file defines the management functions for the NetLabel system.  The
 * NetLabel system manages static and dynamic label mappings for network
 * protocols such as CIPSO and RIPSO.
 *
 * Author: Paul Moore <[email protected]>
 *
 */

/*
 * (c) Copyright Hewlett-Packard Development Company, L.P., 2006
 *
 * This program is free software;  you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY;  without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
 * the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program;  if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 *
 */

#include <linux/types.h>
#include <linux/socket.h>
#include <linux/string.h>
#include <linux/skbuff.h>
#include <net/sock.h>
#include <net/netlink.h>
#include <net/genetlink.h>
#include <net/netlabel.h>
#include <net/cipso_ipv4.h>

#include "netlabel_domainhash.h"
#include "netlabel_user.h"
#include "netlabel_mgmt.h"

/* Argument struct for netlbl_domhsh_walk() */
struct netlbl_domhsh_walk_arg {
	struct netlink_callback *nl_cb;
	struct sk_buff *skb;
	u32 seq;
};

/* NetLabel Generic NETLINK CIPSOv4 family */
static struct genl_family netlbl_mgmt_gnl_family = {
	.id = GENL_ID_GENERATE,
	.hdrsize = 0,
	.name = NETLBL_NLTYPE_MGMT_NAME,
	.version = NETLBL_PROTO_VERSION,
	.maxattr = NLBL_MGMT_A_MAX,
};

/* NetLabel Netlink attribute policy */
static struct nla_policy netlbl_mgmt_genl_policy[NLBL_MGMT_A_MAX + 1] = {
	[NLBL_MGMT_A_DOMAIN] = { .type = NLA_NUL_STRING },
	[NLBL_MGMT_A_PROTOCOL] = { .type = NLA_U32 },
	[NLBL_MGMT_A_VERSION] = { .type = NLA_U32 },
	[NLBL_MGMT_A_CV4DOI] = { .type = NLA_U32 },
};

/*
 * NetLabel Command Handlers
 */

/**
 * netlbl_mgmt_add - Handle an ADD message
 * @skb: the NETLINK buffer
 * @info: the Generic NETLINK info block
 *
 * Description:
 * Process a user generated ADD message and add the domains from the message
 * to the hash table.  See netlabel.h for a description of the message format.
 * Returns zero on success, negative values on failure.
 *
 */
static int netlbl_mgmt_add(struct sk_buff *skb, struct genl_info *info)
{
	int ret_val = -EINVAL;
	struct netlbl_dom_map *entry = NULL;
	size_t tmp_size;
	u32 tmp_val;

	if (!info->attrs[NLBL_MGMT_A_DOMAIN] ||
	    !info->attrs[NLBL_MGMT_A_PROTOCOL])
		goto add_failure;

	entry = kzalloc(sizeof(*entry), GFP_KERNEL);
	if (entry == NULL) {
		ret_val = -ENOMEM;
		goto add_failure;
	}
	tmp_size = nla_len(info->attrs[NLBL_MGMT_A_DOMAIN]);
	entry->domain = kmalloc(tmp_size, GFP_KERNEL);
	if (entry->domain == NULL) {
		ret_val = -ENOMEM;
		goto add_failure;
	}
	entry->type = nla_get_u32(info->attrs[NLBL_MGMT_A_PROTOCOL]);
	nla_strlcpy(entry->domain, info->attrs[NLBL_MGMT_A_DOMAIN], tmp_size);

	switch (entry->type) {
	case NETLBL_NLTYPE_UNLABELED:
		ret_val = netlbl_domhsh_add(entry, NETLINK_CB(skb).sid);
		break;
	case NETLBL_NLTYPE_CIPSOV4:
		if (!info->attrs[NLBL_MGMT_A_CV4DOI])
			goto add_failure;

		tmp_val = nla_get_u32(info->attrs[NLBL_MGMT_A_CV4DOI]);
		/* We should be holding a rcu_read_lock() here while we hold
		 * the result but since the entry will always be deleted when
		 * the CIPSO DOI is deleted we aren't going to keep the
		 * lock. */
		rcu_read_lock();
		entry->type_def.cipsov4 = cipso_v4_doi_getdef(tmp_val);
		if (entry->type_def.cipsov4 == NULL) {
			rcu_read_unlock();
			goto add_failure;
		}
		ret_val = netlbl_domhsh_add(entry, NETLINK_CB(skb).sid);
		rcu_read_unlock();
		break;
	default:
		goto add_failure;
	}
	if (ret_val != 0)
		goto add_failure;

	return 0;

add_failure:
	if (entry)
		kfree(entry->domain);
	kfree(entry);
	return ret_val;
}

/**
 * netlbl_mgmt_remove - Handle a REMOVE message
 * @skb: the NETLINK buffer
 * @info: the Generic NETLINK info block
 *
 * Description:
 * Process a user generated REMOVE message and remove the specified domain
 * mappings.  Returns zero on success, negative values on failure.
 *
 */
static int netlbl_mgmt_remove(struct sk_buff *skb, struct genl_info *info)
{
	char *domain;

	if (!info->attrs[NLBL_MGMT_A_DOMAIN])
		return -EINVAL;

	domain = nla_data(info->attrs[NLBL_MGMT_A_DOMAIN]);
	return netlbl_domhsh_remove(domain, NETLINK_CB(skb).sid);
}

/**
 * netlbl_mgmt_listall_cb - netlbl_domhsh_walk() callback for LISTALL
 * @entry: the domain mapping hash table entry
 * @arg: the netlbl_domhsh_walk_arg structure
 *
 * Description:
 * This function is designed to be used as a callback to the
 * netlbl_domhsh_walk() function for use in generating a response for a LISTALL
 * message.  Returns the size of the message on success, negative values on
 * failure.
 *
 */
static int netlbl_mgmt_listall_cb(struct netlbl_dom_map *entry, void *arg)
{
	int ret_val = -ENOMEM;
	struct netlbl_domhsh_walk_arg *cb_arg = arg;
	void *data;

	data = netlbl_netlink_hdr_put(cb_arg->skb,
				      NETLINK_CB(cb_arg->nl_cb->skb).pid,
				      cb_arg->seq,
				      netlbl_mgmt_gnl_family.id,
				      NLM_F_MULTI,
				      NLBL_MGMT_C_LISTALL);
	if (data == NULL)
		goto listall_cb_failure;

	ret_val = nla_put_string(cb_arg->skb,
				 NLBL_MGMT_A_DOMAIN,
				 entry->domain);
	if (ret_val != 0)
		goto listall_cb_failure;
	ret_val = nla_put_u32(cb_arg->skb, NLBL_MGMT_A_PROTOCOL, entry->type);
	if (ret_val != 0)
		goto listall_cb_failure;
	switch (entry->type) {
	case NETLBL_NLTYPE_CIPSOV4:
		ret_val = nla_put_u32(cb_arg->skb,
				      NLBL_MGMT_A_CV4DOI,
				      entry->type_def.cipsov4->doi);
		if (ret_val != 0)
			goto listall_cb_failure;
		break;
	}

	cb_arg->seq++;
	return genlmsg_end(cb_arg->skb, data);

listall_cb_failure:
	genlmsg_cancel(cb_arg->skb, data);
	return ret_val;
}

/**
 * netlbl_mgmt_listall - Handle a LISTALL message
 * @skb: the NETLINK buffer
 * @cb: the NETLINK callback
 *
 * Description:
 * Process a user generated LISTALL message and dumps the domain hash table in
 * a form suitable for use in a kernel generated LISTALL message.  Returns zero
 * on success, negative values on failure.
 *
 */
static int netlbl_mgmt_listall(struct sk_buff *skb,
			       struct netlink_callback *cb)
{
	struct netlbl_domhsh_walk_arg cb_arg;
	u32 skip_bkt = cb->args[0];
	u32 skip_chain = cb->args[1];

	cb_arg.nl_cb = cb;
	cb_arg.skb = skb;
	cb_arg.seq = cb->nlh->nlmsg_seq;

	netlbl_domhsh_walk(&skip_bkt,
			   &skip_chain,
			   netlbl_mgmt_listall_cb,
			   &cb_arg);

	cb->args[0] = skip_bkt;
	cb->args[1] = skip_chain;
	return skb->len;
}

/**
 * netlbl_mgmt_adddef - Handle an ADDDEF message
 * @skb: the NETLINK buffer
 * @info: the Generic NETLINK info block
 *
 * Description:
 * Process a user generated ADDDEF message and respond accordingly.  Returns
 * zero on success, negative values on failure.
 *
 */
static int netlbl_mgmt_adddef(struct sk_buff *skb, struct genl_info *info)
{
	int ret_val = -EINVAL;
	struct netlbl_dom_map *entry = NULL;
	u32 tmp_val;

	if (!info->attrs[NLBL_MGMT_A_PROTOCOL])
		goto adddef_failure;

	entry = kzalloc(sizeof(*entry), GFP_KERNEL);
	if (entry == NULL) {
		ret_val = -ENOMEM;
		goto adddef_failure;
	}
	entry->type = nla_get_u32(info->attrs[NLBL_MGMT_A_PROTOCOL]);

	switch (entry->type) {
	case NETLBL_NLTYPE_UNLABELED:
		ret_val = netlbl_domhsh_add_default(entry,
						    NETLINK_CB(skb).sid);
		break;
	case NETLBL_NLTYPE_CIPSOV4:
		if (!info->attrs[NLBL_MGMT_A_CV4DOI])
			goto adddef_failure;

		tmp_val = nla_get_u32(info->attrs[NLBL_MGMT_A_CV4DOI]);
		/* We should be holding a rcu_read_lock() here while we hold
		 * the result but since the entry will always be deleted when
		 * the CIPSO DOI is deleted we aren't going to keep the
		 * lock. */
		rcu_read_lock();
		entry->type_def.cipsov4 = cipso_v4_doi_getdef(tmp_val);
		if (entry->type_def.cipsov4 == NULL) {
			rcu_read_unlock();
			goto adddef_failure;
		}
		ret_val = netlbl_domhsh_add_default(entry,
						    NETLINK_CB(skb).sid);
		rcu_read_unlock();
		break;
	default:
		goto adddef_failure;
	}
	if (ret_val != 0)
		goto adddef_failure;

	return 0;

adddef_failure:
	kfree(entry);
	return ret_val;
}

/**
 * netlbl_mgmt_removedef - Handle a REMOVEDEF message
 * @skb: the NETLINK buffer
 * @info: the Generic NETLINK info block
 *
 * Description:
 * Process a user generated REMOVEDEF message and remove the default domain
 * mapping.  Returns zero on success, negative values on failure.
 *
 */
static int netlbl_mgmt_removedef(struct sk_buff *skb, struct genl_info *info)
{
	return netlbl_domhsh_remove_default(NETLINK_CB(skb).sid);
}

/**
 * netlbl_mgmt_listdef - Handle a LISTDEF message
 * @skb: the NETLINK buffer
 * @info: the Generic NETLINK info block
 *
 * Description:
 * Process a user generated LISTDEF message and dumps the default domain
 * mapping in a form suitable for use in a kernel generated LISTDEF message.
 * Returns zero on success, negative values on failure.
 *
 */
static int netlbl_mgmt_listdef(struct sk_buff *skb, struct genl_info *info)
{
	int ret_val = -ENOMEM;
	struct sk_buff *ans_skb = NULL;
	void *data;
	struct netlbl_dom_map *entry;

	ans_skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
	if (ans_skb == NULL)
		return -ENOMEM;
	data = netlbl_netlink_hdr_put(ans_skb,
				      info->snd_pid,
				      info->snd_seq,
				      netlbl_mgmt_gnl_family.id,
				      0,
				      NLBL_MGMT_C_LISTDEF);
	if (data == NULL)
		goto listdef_failure;

	rcu_read_lock();
	entry = netlbl_domhsh_getentry(NULL);
	if (entry == NULL) {
		ret_val = -ENOENT;
		goto listdef_failure_lock;
	}
	ret_val = nla_put_u32(ans_skb, NLBL_MGMT_A_PROTOCOL, entry->type);
	if (ret_val != 0)
		goto listdef_failure_lock;
	switch (entry->type) {
	case NETLBL_NLTYPE_CIPSOV4:
		ret_val = nla_put_u32(ans_skb,
				      NLBL_MGMT_A_CV4DOI,
				      entry->type_def.cipsov4->doi);
		if (ret_val != 0)
			goto listdef_failure_lock;
		break;
	}
	rcu_read_unlock();

	genlmsg_end(ans_skb, data);

	ret_val = genlmsg_unicast(ans_skb, info->snd_pid);
	if (ret_val != 0)
		goto listdef_failure;
	return 0;

listdef_failure_lock:
	rcu_read_unlock();
listdef_failure:
	kfree_skb(ans_skb);
	return ret_val;
}

/**
 * netlbl_mgmt_protocols_cb - Write an individual PROTOCOL message response
 * @skb: the skb to write to
 * @seq: the NETLINK sequence number
 * @cb: the NETLINK callback
 * @protocol: the NetLabel protocol to use in the message
 *
 * Description:
 * This function is to be used in conjunction with netlbl_mgmt_protocols() to
 * answer a application's PROTOCOLS message.  Returns the size of the message
 * on success, negative values on failure.
 *
 */
static int netlbl_mgmt_protocols_cb(struct sk_buff *skb,
				    struct netlink_callback *cb,
				    u32 protocol)
{
	int ret_val = -ENOMEM;
	void *data;

	data = netlbl_netlink_hdr_put(skb,
				      NETLINK_CB(cb->skb).pid,
				      cb->nlh->nlmsg_seq,
				      netlbl_mgmt_gnl_family.id,
				      NLM_F_MULTI,
				      NLBL_MGMT_C_PROTOCOLS);
	if (data == NULL)
		goto protocols_cb_failure;

	ret_val = nla_put_u32(skb, NLBL_MGMT_A_PROTOCOL, protocol);
	if (ret_val != 0)
		goto protocols_cb_failure;

	return genlmsg_end(skb, data);

protocols_cb_failure:
	genlmsg_cancel(skb, data);
	return ret_val;
}

/**
 * netlbl_mgmt_protocols - Handle a PROTOCOLS message
 * @skb: the NETLINK buffer
 * @cb: the NETLINK callback
 *
 * Description:
 * Process a user generated PROTOCOLS message and respond accordingly.
 *
 */
static int netlbl_mgmt_protocols(struct sk_buff *skb,
				 struct netlink_callback *cb)
{
	u32 protos_sent = cb->args[0];

	if (protos_sent == 0) {
		if (netlbl_mgmt_protocols_cb(skb,
					     cb,
					     NETLBL_NLTYPE_UNLABELED) < 0)
			goto protocols_return;
		protos_sent++;
	}
	if (protos_sent == 1) {
		if (netlbl_mgmt_protocols_cb(skb,
					     cb,
					     NETLBL_NLTYPE_CIPSOV4) < 0)
			goto protocols_return;
		protos_sent++;
	}

protocols_return:
	cb->args[0] = protos_sent;
	return skb->len;
}

/**
 * netlbl_mgmt_version - Handle a VERSION message
 * @skb: the NETLINK buffer
 * @info: the Generic NETLINK info block
 *
 * Description:
 * Process a user generated VERSION message and respond accordingly.  Returns
 * zero on success, negative values on failure.
 *
 */
static int netlbl_mgmt_version(struct sk_buff *skb, struct genl_info *info)
{
	int ret_val = -ENOMEM;
	struct sk_buff *ans_skb = NULL;
	void *data;

	ans_skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
	if (ans_skb == NULL)
		return -ENOMEM;
	data = netlbl_netlink_hdr_put(ans_skb,
				      info->snd_pid,
				      info->snd_seq,
				      netlbl_mgmt_gnl_family.id,
				      0,
				      NLBL_MGMT_C_VERSION);
	if (data == NULL)
		goto version_failure;

	ret_val = nla_put_u32(ans_skb,
			      NLBL_MGMT_A_VERSION,
			      NETLBL_PROTO_VERSION);
	if (ret_val != 0)
		goto version_failure;

	genlmsg_end(ans_skb, data);

	ret_val = genlmsg_unicast(ans_skb, info->snd_pid);
	if (ret_val != 0)
		goto version_failure;
	return 0;

version_failure:
	kfree_skb(ans_skb);
	return ret_val;
}


/*
 * NetLabel Generic NETLINK Command Definitions
 */

static struct genl_ops netlbl_mgmt_genl_c_add = {
	.cmd = NLBL_MGMT_C_ADD,
	.flags = GENL_ADMIN_PERM,
	.policy = netlbl_mgmt_genl_policy,
	.doit = netlbl_mgmt_add,
	.dumpit = NULL,
};

static struct genl_ops netlbl_mgmt_genl_c_remove = {
	.cmd = NLBL_MGMT_C_REMOVE,
	.flags = GENL_ADMIN_PERM,
	.policy = netlbl_mgmt_genl_policy,
	.doit = netlbl_mgmt_remove,
	.dumpit = NULL,
};

static struct genl_ops netlbl_mgmt_genl_c_listall = {
	.cmd = NLBL_MGMT_C_LISTALL,
	.flags = 0,
	.policy = netlbl_mgmt_genl_policy,
	.doit = NULL,
	.dumpit = netlbl_mgmt_listall,
};

static struct genl_ops netlbl_mgmt_genl_c_adddef = {
	.cmd = NLBL_MGMT_C_ADDDEF,
	.flags = GENL_ADMIN_PERM,
	.policy = netlbl_mgmt_genl_policy,
	.doit = netlbl_mgmt_adddef,
	.dumpit = NULL,
};

static struct genl_ops netlbl_mgmt_genl_c_removedef = {
	.cmd = NLBL_MGMT_C_REMOVEDEF,
	.flags = GENL_ADMIN_PERM,
	.policy = netlbl_mgmt_genl_policy,
	.doit = netlbl_mgmt_removedef,
	.dumpit = NULL,
};

static struct genl_ops netlbl_mgmt_genl_c_listdef = {
	.cmd = NLBL_MGMT_C_LISTDEF,
	.flags = 0,
	.policy = netlbl_mgmt_genl_policy,
	.doit = netlbl_mgmt_listdef,
	.dumpit = NULL,
};

static struct genl_ops netlbl_mgmt_genl_c_protocols = {
	.cmd = NLBL_MGMT_C_PROTOCOLS,
	.flags = 0,
	.policy = netlbl_mgmt_genl_policy,
	.doit = NULL,
	.dumpit = netlbl_mgmt_protocols,
};

static struct genl_ops netlbl_mgmt_genl_c_version = {
	.cmd = NLBL_MGMT_C_VERSION,
	.flags = 0,
	.policy = netlbl_mgmt_genl_policy,
	.doit = netlbl_mgmt_version,
	.dumpit = NULL,
};

/*
 * NetLabel Generic NETLINK Protocol Functions
 */

/**
 * netlbl_mgmt_genl_init - Register the NetLabel management component
 *
 * Description:
 * Register the NetLabel management component with the Generic NETLINK
 * mechanism.  Returns zero on success, negative values on failure.
 *
 */
int netlbl_mgmt_genl_init(void)
{
	int ret_val;

	ret_val = genl_register_family(&netlbl_mgmt_gnl_family);
	if (ret_val != 0)
		return ret_val;

	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
				    &netlbl_mgmt_genl_c_add);
	if (ret_val != 0)
		return ret_val;
	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
				    &netlbl_mgmt_genl_c_remove);
	if (ret_val != 0)
		return ret_val;
	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
				    &netlbl_mgmt_genl_c_listall);
	if (ret_val != 0)
		return ret_val;
	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
				    &netlbl_mgmt_genl_c_adddef);
	if (ret_val != 0)
		return ret_val;
	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
				    &netlbl_mgmt_genl_c_removedef);
	if (ret_val != 0)
		return ret_val;
	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
				    &netlbl_mgmt_genl_c_listdef);
	if (ret_val != 0)
		return ret_val;
	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
				    &netlbl_mgmt_genl_c_protocols);
	if (ret_val != 0)
		return ret_val;
	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
				    &netlbl_mgmt_genl_c_version);
	if (ret_val != 0)
		return ret_val;

	return 0;
}
Commit	Line	Data
d15c345f PM	1	/*
	2	* NetLabel Management Support
	3	*
	4	* This file defines the management functions for the NetLabel system. The
	5	* NetLabel system manages static and dynamic label mappings for network
	6	* protocols such as CIPSO and RIPSO.
	7	*
	8	* Author: Paul Moore <[email protected]>
	9	*
	10	*/
	11
	12	/*
	13	* (c) Copyright Hewlett-Packard Development Company, L.P., 2006
	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, write to the Free Software
	27	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	28	*
	29	*/
	30
	31	#include <linux/types.h>
	32	#include <linux/socket.h>
	33	#include <linux/string.h>
	34	#include <linux/skbuff.h>
	35	#include <net/sock.h>
	36	#include <net/netlink.h>
	37	#include <net/genetlink.h>
	38	#include <net/netlabel.h>
	39	#include <net/cipso_ipv4.h>
	40
	41	#include "netlabel_domainhash.h"
	42	#include "netlabel_user.h"
	43	#include "netlabel_mgmt.h"
	44
fd385855 PM	45	/* Argument struct for netlbl_domhsh_walk() */
	46	struct netlbl_domhsh_walk_arg {
	47	struct netlink_callback *nl_cb;
	48	struct sk_buff *skb;
	49	u32 seq;
	50	};
	51
d15c345f PM	52	/* NetLabel Generic NETLINK CIPSOv4 family */
	53	static struct genl_family netlbl_mgmt_gnl_family = {
	54	.id = GENL_ID_GENERATE,
	55	.hdrsize = 0,
	56	.name = NETLBL_NLTYPE_MGMT_NAME,
	57	.version = NETLBL_PROTO_VERSION,
fd385855	58	.maxattr = NLBL_MGMT_A_MAX,
d15c345f PM	59	};
d15c345f PM	60
fd385855 PM	61	/* NetLabel Netlink attribute policy */
	62	static struct nla_policy netlbl_mgmt_genl_policy[NLBL_MGMT_A_MAX + 1] = {
	63	[NLBL_MGMT_A_DOMAIN] = { .type = NLA_NUL_STRING },
	64	[NLBL_MGMT_A_PROTOCOL] = { .type = NLA_U32 },
	65	[NLBL_MGMT_A_VERSION] = { .type = NLA_U32 },
	66	[NLBL_MGMT_A_CV4DOI] = { .type = NLA_U32 },
	67	};
d15c345f PM	68
	69	/*
	70	* NetLabel Command Handlers
	71	*/
	72
	73	/**
	74	* netlbl_mgmt_add - Handle an ADD message
	75	* @skb: the NETLINK buffer
	76	* @info: the Generic NETLINK info block
	77	*
	78	* Description:
	79	* Process a user generated ADD message and add the domains from the message
	80	* to the hash table. See netlabel.h for a description of the message format.
	81	* Returns zero on success, negative values on failure.
	82	*
	83	*/
	84	static int netlbl_mgmt_add(struct sk_buff skb, struct genl_info info)
	85	{
	86	int ret_val = -EINVAL;
d15c345f	87	struct netlbl_dom_map *entry = NULL;
fd385855	88	size_t tmp_size;
d15c345f	89	u32 tmp_val;
d15c345f	90
fd385855 PM	91	if (!info->attrs[NLBL_MGMT_A_DOMAIN] \|\|
fd385855 PM	92	!info->attrs[NLBL_MGMT_A_PROTOCOL])
d15c345f PM	93	goto add_failure;
d15c345f PM	94
fd385855 PM	95	entry = kzalloc(sizeof(*entry), GFP_KERNEL);
	96	if (entry == NULL) {
	97	ret_val = -ENOMEM;
	98	goto add_failure;
	99	}
	100	tmp_size = nla_len(info->attrs[NLBL_MGMT_A_DOMAIN]);
	101	entry->domain = kmalloc(tmp_size, GFP_KERNEL);
	102	if (entry->domain == NULL) {
	103	ret_val = -ENOMEM;
d15c345f	104	goto add_failure;
fd385855 PM	105	}
	106	entry->type = nla_get_u32(info->attrs[NLBL_MGMT_A_PROTOCOL]);
	107	nla_strlcpy(entry->domain, info->attrs[NLBL_MGMT_A_DOMAIN], tmp_size);
d15c345f	108
fd385855 PM	109	switch (entry->type) {
fd385855 PM	110	case NETLBL_NLTYPE_UNLABELED:
32f50cde	111	ret_val = netlbl_domhsh_add(entry, NETLINK_CB(skb).sid);
fd385855 PM	112	break;
	113	case NETLBL_NLTYPE_CIPSOV4:
	114	if (!info->attrs[NLBL_MGMT_A_CV4DOI])
d15c345f	115	goto add_failure;
d15c345f	116
fd385855 PM	117	tmp_val = nla_get_u32(info->attrs[NLBL_MGMT_A_CV4DOI]);
	118	/* We should be holding a rcu_read_lock() here while we hold
	119	* the result but since the entry will always be deleted when
	120	* the CIPSO DOI is deleted we aren't going to keep the
	121	* lock. */
	122	rcu_read_lock();
	123	entry->type_def.cipsov4 = cipso_v4_doi_getdef(tmp_val);
	124	if (entry->type_def.cipsov4 == NULL) {
d15c345f	125	rcu_read_unlock();
d15c345f	126	goto add_failure;
fd385855	127	}
32f50cde	128	ret_val = netlbl_domhsh_add(entry, NETLINK_CB(skb).sid);
fd385855 PM	129	rcu_read_unlock();
	130	break;
	131	default:
	132	goto add_failure;
d15c345f	133	}
fd385855 PM	134	if (ret_val != 0)
fd385855 PM	135	goto add_failure;
d15c345f	136
d15c345f PM	137	return 0;
	138
	139	add_failure:
	140	if (entry)
	141	kfree(entry->domain);
	142	kfree(entry);
d15c345f PM	143	return ret_val;
	144	}
	145
	146	/**
	147	* netlbl_mgmt_remove - Handle a REMOVE message
	148	* @skb: the NETLINK buffer
	149	* @info: the Generic NETLINK info block
	150	*
	151	* Description:
	152	* Process a user generated REMOVE message and remove the specified domain
	153	* mappings. Returns zero on success, negative values on failure.
	154	*
	155	*/
	156	static int netlbl_mgmt_remove(struct sk_buff skb, struct genl_info info)
	157	{
fd385855	158	char *domain;
d15c345f	159
fd385855 PM	160	if (!info->attrs[NLBL_MGMT_A_DOMAIN])
fd385855 PM	161	return -EINVAL;
d15c345f	162
fd385855	163	domain = nla_data(info->attrs[NLBL_MGMT_A_DOMAIN]);
32f50cde	164	return netlbl_domhsh_remove(domain, NETLINK_CB(skb).sid);
fd385855 PM	165	}
	166
	167	/**
	168	* netlbl_mgmt_listall_cb - netlbl_domhsh_walk() callback for LISTALL
	169	* @entry: the domain mapping hash table entry
	170	* @arg: the netlbl_domhsh_walk_arg structure
	171	*
	172	* Description:
	173	* This function is designed to be used as a callback to the
	174	* netlbl_domhsh_walk() function for use in generating a response for a LISTALL
	175	* message. Returns the size of the message on success, negative values on
	176	* failure.
	177	*
	178	*/
	179	static int netlbl_mgmt_listall_cb(struct netlbl_dom_map entry, void arg)
	180	{
	181	int ret_val = -ENOMEM;
	182	struct netlbl_domhsh_walk_arg *cb_arg = arg;
	183	void *data;
	184
	185	data = netlbl_netlink_hdr_put(cb_arg->skb,
	186	NETLINK_CB(cb_arg->nl_cb->skb).pid,
	187	cb_arg->seq,
	188	netlbl_mgmt_gnl_family.id,
	189	NLM_F_MULTI,
	190	NLBL_MGMT_C_LISTALL);
	191	if (data == NULL)
	192	goto listall_cb_failure;
	193
	194	ret_val = nla_put_string(cb_arg->skb,
	195	NLBL_MGMT_A_DOMAIN,
	196	entry->domain);
	197	if (ret_val != 0)
	198	goto listall_cb_failure;
	199	ret_val = nla_put_u32(cb_arg->skb, NLBL_MGMT_A_PROTOCOL, entry->type);
	200	if (ret_val != 0)
	201	goto listall_cb_failure;
	202	switch (entry->type) {
	203	case NETLBL_NLTYPE_CIPSOV4:
	204	ret_val = nla_put_u32(cb_arg->skb,
	205	NLBL_MGMT_A_CV4DOI,
	206	entry->type_def.cipsov4->doi);
d15c345f	207	if (ret_val != 0)
fd385855 PM	208	goto listall_cb_failure;
fd385855 PM	209	break;
d15c345f PM	210	}
d15c345f PM	211
fd385855 PM	212	cb_arg->seq++;
fd385855 PM	213	return genlmsg_end(cb_arg->skb, data);
d15c345f	214
fd385855 PM	215	listall_cb_failure:
fd385855 PM	216	genlmsg_cancel(cb_arg->skb, data);
d15c345f PM	217	return ret_val;
	218	}
	219
	220	/**
fd385855	221	* netlbl_mgmt_listall - Handle a LISTALL message
d15c345f	222	* @skb: the NETLINK buffer
fd385855	223	* @cb: the NETLINK callback
d15c345f PM	224	*
d15c345f PM	225	* Description:
fd385855 PM	226	* Process a user generated LISTALL message and dumps the domain hash table in
	227	* a form suitable for use in a kernel generated LISTALL message. Returns zero
	228	* on success, negative values on failure.
d15c345f PM	229	*
d15c345f PM	230	*/
fd385855 PM	231	static int netlbl_mgmt_listall(struct sk_buff *skb,
fd385855 PM	232	struct netlink_callback *cb)
d15c345f	233	{
fd385855 PM	234	struct netlbl_domhsh_walk_arg cb_arg;
	235	u32 skip_bkt = cb->args[0];
	236	u32 skip_chain = cb->args[1];
	237
	238	cb_arg.nl_cb = cb;
	239	cb_arg.skb = skb;
	240	cb_arg.seq = cb->nlh->nlmsg_seq;
	241
	242	netlbl_domhsh_walk(&skip_bkt,
	243	&skip_chain,
	244	netlbl_mgmt_listall_cb,
	245	&cb_arg);
	246
	247	cb->args[0] = skip_bkt;
	248	cb->args[1] = skip_chain;
	249	return skb->len;
d15c345f PM	250	}
	251
	252	/**
	253	* netlbl_mgmt_adddef - Handle an ADDDEF message
	254	* @skb: the NETLINK buffer
	255	* @info: the Generic NETLINK info block
	256	*
	257	* Description:
	258	* Process a user generated ADDDEF message and respond accordingly. Returns
	259	* zero on success, negative values on failure.
	260	*
	261	*/
	262	static int netlbl_mgmt_adddef(struct sk_buff skb, struct genl_info info)
	263	{
	264	int ret_val = -EINVAL;
d15c345f PM	265	struct netlbl_dom_map *entry = NULL;
	266	u32 tmp_val;
	267
fd385855	268	if (!info->attrs[NLBL_MGMT_A_PROTOCOL])
d15c345f	269	goto adddef_failure;
d15c345f PM	270
	271	entry = kzalloc(sizeof(*entry), GFP_KERNEL);
	272	if (entry == NULL) {
	273	ret_val = -ENOMEM;
	274	goto adddef_failure;
	275	}
fd385855	276	entry->type = nla_get_u32(info->attrs[NLBL_MGMT_A_PROTOCOL]);
d15c345f	277
d15c345f PM	278	switch (entry->type) {
d15c345f PM	279	case NETLBL_NLTYPE_UNLABELED:
32f50cde PM	280	ret_val = netlbl_domhsh_add_default(entry,
32f50cde PM	281	NETLINK_CB(skb).sid);
d15c345f PM	282	break;
d15c345f PM	283	case NETLBL_NLTYPE_CIPSOV4:
fd385855	284	if (!info->attrs[NLBL_MGMT_A_CV4DOI])
d15c345f	285	goto adddef_failure;
fd385855 PM	286
	287	tmp_val = nla_get_u32(info->attrs[NLBL_MGMT_A_CV4DOI]);
	288	/* We should be holding a rcu_read_lock() here while we hold
	289	* the result but since the entry will always be deleted when
	290	* the CIPSO DOI is deleted we aren't going to keep the
	291	* lock. */
d15c345f PM	292	rcu_read_lock();
	293	entry->type_def.cipsov4 = cipso_v4_doi_getdef(tmp_val);
	294	if (entry->type_def.cipsov4 == NULL) {
	295	rcu_read_unlock();
d15c345f PM	296	goto adddef_failure;
d15c345f PM	297	}
32f50cde PM	298	ret_val = netlbl_domhsh_add_default(entry,
32f50cde PM	299	NETLINK_CB(skb).sid);
d15c345f PM	300	rcu_read_unlock();
	301	break;
	302	default:
fd385855	303	goto adddef_failure;
d15c345f PM	304	}
	305	if (ret_val != 0)
	306	goto adddef_failure;
	307
d15c345f PM	308	return 0;
	309
	310	adddef_failure:
	311	kfree(entry);
d15c345f PM	312	return ret_val;
	313	}
	314
	315	/**
	316	* netlbl_mgmt_removedef - Handle a REMOVEDEF message
	317	* @skb: the NETLINK buffer
	318	* @info: the Generic NETLINK info block
	319	*
	320	* Description:
	321	* Process a user generated REMOVEDEF message and remove the default domain
	322	* mapping. Returns zero on success, negative values on failure.
	323	*
	324	*/
	325	static int netlbl_mgmt_removedef(struct sk_buff skb, struct genl_info info)
	326	{
32f50cde	327	return netlbl_domhsh_remove_default(NETLINK_CB(skb).sid);
d15c345f PM	328	}
	329
	330	/**
	331	* netlbl_mgmt_listdef - Handle a LISTDEF message
	332	* @skb: the NETLINK buffer
	333	* @info: the Generic NETLINK info block
	334	*
	335	* Description:
	336	* Process a user generated LISTDEF message and dumps the default domain
	337	* mapping in a form suitable for use in a kernel generated LISTDEF message.
	338	* Returns zero on success, negative values on failure.
	339	*
	340	*/
	341	static int netlbl_mgmt_listdef(struct sk_buff skb, struct genl_info info)
	342	{
	343	int ret_val = -ENOMEM;
fd385855 PM	344	struct sk_buff *ans_skb = NULL;
	345	void *data;
	346	struct netlbl_dom_map *entry;
d15c345f	347
fd385855	348	ans_skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
d15c345f	349	if (ans_skb == NULL)
fd385855 PM	350	return -ENOMEM;
	351	data = netlbl_netlink_hdr_put(ans_skb,
	352	info->snd_pid,
	353	info->snd_seq,
	354	netlbl_mgmt_gnl_family.id,
	355	0,
	356	NLBL_MGMT_C_LISTDEF);
	357	if (data == NULL)
d15c345f	358	goto listdef_failure;
d15c345f	359
fd385855 PM	360	rcu_read_lock();
	361	entry = netlbl_domhsh_getentry(NULL);
	362	if (entry == NULL) {
	363	ret_val = -ENOENT;
	364	goto listdef_failure_lock;
	365	}
	366	ret_val = nla_put_u32(ans_skb, NLBL_MGMT_A_PROTOCOL, entry->type);
d15c345f	367	if (ret_val != 0)
fd385855 PM	368	goto listdef_failure_lock;
	369	switch (entry->type) {
	370	case NETLBL_NLTYPE_CIPSOV4:
	371	ret_val = nla_put_u32(ans_skb,
	372	NLBL_MGMT_A_CV4DOI,
	373	entry->type_def.cipsov4->doi);
	374	if (ret_val != 0)
	375	goto listdef_failure_lock;
	376	break;
	377	}
	378	rcu_read_unlock();
d15c345f	379
fd385855 PM	380	genlmsg_end(ans_skb, data);
	381
	382	ret_val = genlmsg_unicast(ans_skb, info->snd_pid);
	383	if (ret_val != 0)
	384	goto listdef_failure;
d15c345f PM	385	return 0;
d15c345f PM	386
fd385855 PM	387	listdef_failure_lock:
fd385855 PM	388	rcu_read_unlock();
d15c345f	389	listdef_failure:
fd385855	390	kfree_skb(ans_skb);
d15c345f PM	391	return ret_val;
	392	}
	393
	394	/**
fd385855 PM	395	* netlbl_mgmt_protocols_cb - Write an individual PROTOCOL message response
	396	* @skb: the skb to write to
	397	* @seq: the NETLINK sequence number
	398	* @cb: the NETLINK callback
	399	* @protocol: the NetLabel protocol to use in the message
d15c345f PM	400	*
d15c345f PM	401	* Description:
fd385855 PM	402	* This function is to be used in conjunction with netlbl_mgmt_protocols() to
	403	* answer a application's PROTOCOLS message. Returns the size of the message
	404	* on success, negative values on failure.
d15c345f PM	405	*
d15c345f PM	406	*/
fd385855 PM	407	static int netlbl_mgmt_protocols_cb(struct sk_buff *skb,
	408	struct netlink_callback *cb,
	409	u32 protocol)
d15c345f PM	410	{
d15c345f PM	411	int ret_val = -ENOMEM;
fd385855 PM	412	void *data;
	413
	414	data = netlbl_netlink_hdr_put(skb,
	415	NETLINK_CB(cb->skb).pid,
	416	cb->nlh->nlmsg_seq,
	417	netlbl_mgmt_gnl_family.id,
	418	NLM_F_MULTI,
	419	NLBL_MGMT_C_PROTOCOLS);
	420	if (data == NULL)
	421	goto protocols_cb_failure;
	422
	423	ret_val = nla_put_u32(skb, NLBL_MGMT_A_PROTOCOL, protocol);
d15c345f	424	if (ret_val != 0)
fd385855	425	goto protocols_cb_failure;
d15c345f	426
fd385855	427	return genlmsg_end(skb, data);
d15c345f	428
fd385855 PM	429	protocols_cb_failure:
fd385855 PM	430	genlmsg_cancel(skb, data);
d15c345f PM	431	return ret_val;
	432	}
	433
fd385855 PM	434	/**
	435	* netlbl_mgmt_protocols - Handle a PROTOCOLS message
	436	* @skb: the NETLINK buffer
	437	* @cb: the NETLINK callback
	438	*
	439	* Description:
	440	* Process a user generated PROTOCOLS message and respond accordingly.
	441	*
	442	*/
	443	static int netlbl_mgmt_protocols(struct sk_buff *skb,
	444	struct netlink_callback *cb)
	445	{
	446	u32 protos_sent = cb->args[0];
	447
	448	if (protos_sent == 0) {
	449	if (netlbl_mgmt_protocols_cb(skb,
	450	cb,
	451	NETLBL_NLTYPE_UNLABELED) < 0)
	452	goto protocols_return;
	453	protos_sent++;
	454	}
	455	if (protos_sent == 1) {
	456	if (netlbl_mgmt_protocols_cb(skb,
	457	cb,
	458	NETLBL_NLTYPE_CIPSOV4) < 0)
	459	goto protocols_return;
	460	protos_sent++;
	461	}
	462
	463	protocols_return:
	464	cb->args[0] = protos_sent;
	465	return skb->len;
	466	}
	467
d15c345f PM	468	/**
	469	* netlbl_mgmt_version - Handle a VERSION message
	470	* @skb: the NETLINK buffer
	471	* @info: the Generic NETLINK info block
	472	*
	473	* Description:
	474	* Process a user generated VERSION message and respond accordingly. Returns
	475	* zero on success, negative values on failure.
	476	*
	477	*/
	478	static int netlbl_mgmt_version(struct sk_buff skb, struct genl_info info)
	479	{
	480	int ret_val = -ENOMEM;
	481	struct sk_buff *ans_skb = NULL;
fd385855	482	void *data;
d15c345f	483
fd385855	484	ans_skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
d15c345f	485	if (ans_skb == NULL)
fd385855 PM	486	return -ENOMEM;
	487	data = netlbl_netlink_hdr_put(ans_skb,
	488	info->snd_pid,
	489	info->snd_seq,
	490	netlbl_mgmt_gnl_family.id,
	491	0,
	492	NLBL_MGMT_C_VERSION);
	493	if (data == NULL)
d15c345f PM	494	goto version_failure;
d15c345f PM	495
fd385855 PM	496	ret_val = nla_put_u32(ans_skb,
	497	NLBL_MGMT_A_VERSION,
	498	NETLBL_PROTO_VERSION);
d15c345f PM	499	if (ret_val != 0)
	500	goto version_failure;
	501
fd385855 PM	502	genlmsg_end(ans_skb, data);
	503
	504	ret_val = genlmsg_unicast(ans_skb, info->snd_pid);
d15c345f PM	505	if (ret_val != 0)
d15c345f PM	506	goto version_failure;
d15c345f PM	507	return 0;
	508
	509	version_failure:
	510	kfree_skb(ans_skb);
d15c345f PM	511	return ret_val;
	512	}
	513
	514
	515	/*
	516	* NetLabel Generic NETLINK Command Definitions
	517	*/
	518
	519	static struct genl_ops netlbl_mgmt_genl_c_add = {
	520	.cmd = NLBL_MGMT_C_ADD,
fd385855 PM	521	.flags = GENL_ADMIN_PERM,
fd385855 PM	522	.policy = netlbl_mgmt_genl_policy,
d15c345f PM	523	.doit = netlbl_mgmt_add,
	524	.dumpit = NULL,
	525	};
	526
	527	static struct genl_ops netlbl_mgmt_genl_c_remove = {
	528	.cmd = NLBL_MGMT_C_REMOVE,
fd385855 PM	529	.flags = GENL_ADMIN_PERM,
fd385855 PM	530	.policy = netlbl_mgmt_genl_policy,
d15c345f PM	531	.doit = netlbl_mgmt_remove,
	532	.dumpit = NULL,
	533	};
	534
fd385855 PM	535	static struct genl_ops netlbl_mgmt_genl_c_listall = {
fd385855 PM	536	.cmd = NLBL_MGMT_C_LISTALL,
d15c345f	537	.flags = 0,
fd385855 PM	538	.policy = netlbl_mgmt_genl_policy,
	539	.doit = NULL,
	540	.dumpit = netlbl_mgmt_listall,
d15c345f PM	541	};
	542
	543	static struct genl_ops netlbl_mgmt_genl_c_adddef = {
	544	.cmd = NLBL_MGMT_C_ADDDEF,
fd385855 PM	545	.flags = GENL_ADMIN_PERM,
fd385855 PM	546	.policy = netlbl_mgmt_genl_policy,
d15c345f PM	547	.doit = netlbl_mgmt_adddef,
	548	.dumpit = NULL,
	549	};
	550
	551	static struct genl_ops netlbl_mgmt_genl_c_removedef = {
	552	.cmd = NLBL_MGMT_C_REMOVEDEF,
fd385855 PM	553	.flags = GENL_ADMIN_PERM,
fd385855 PM	554	.policy = netlbl_mgmt_genl_policy,
d15c345f PM	555	.doit = netlbl_mgmt_removedef,
	556	.dumpit = NULL,
	557	};
	558
	559	static struct genl_ops netlbl_mgmt_genl_c_listdef = {
	560	.cmd = NLBL_MGMT_C_LISTDEF,
	561	.flags = 0,
fd385855	562	.policy = netlbl_mgmt_genl_policy,
d15c345f PM	563	.doit = netlbl_mgmt_listdef,
	564	.dumpit = NULL,
	565	};
	566
fd385855 PM	567	static struct genl_ops netlbl_mgmt_genl_c_protocols = {
fd385855 PM	568	.cmd = NLBL_MGMT_C_PROTOCOLS,
d15c345f	569	.flags = 0,
fd385855 PM	570	.policy = netlbl_mgmt_genl_policy,
	571	.doit = NULL,
	572	.dumpit = netlbl_mgmt_protocols,
d15c345f PM	573	};
	574
	575	static struct genl_ops netlbl_mgmt_genl_c_version = {
	576	.cmd = NLBL_MGMT_C_VERSION,
	577	.flags = 0,
fd385855	578	.policy = netlbl_mgmt_genl_policy,
d15c345f PM	579	.doit = netlbl_mgmt_version,
	580	.dumpit = NULL,
	581	};
	582
	583	/*
	584	* NetLabel Generic NETLINK Protocol Functions
	585	*/
	586
	587	/**
	588	* netlbl_mgmt_genl_init - Register the NetLabel management component
	589	*
	590	* Description:
	591	* Register the NetLabel management component with the Generic NETLINK
	592	* mechanism. Returns zero on success, negative values on failure.
	593	*
	594	*/
	595	int netlbl_mgmt_genl_init(void)
	596	{
	597	int ret_val;
	598
	599	ret_val = genl_register_family(&netlbl_mgmt_gnl_family);
	600	if (ret_val != 0)
	601	return ret_val;
	602
	603	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
	604	&netlbl_mgmt_genl_c_add);
	605	if (ret_val != 0)
	606	return ret_val;
	607	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
	608	&netlbl_mgmt_genl_c_remove);
	609	if (ret_val != 0)
	610	return ret_val;
	611	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
fd385855	612	&netlbl_mgmt_genl_c_listall);
d15c345f PM	613	if (ret_val != 0)
	614	return ret_val;
	615	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
	616	&netlbl_mgmt_genl_c_adddef);
	617	if (ret_val != 0)
	618	return ret_val;
	619	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
	620	&netlbl_mgmt_genl_c_removedef);
	621	if (ret_val != 0)
	622	return ret_val;
	623	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
	624	&netlbl_mgmt_genl_c_listdef);
	625	if (ret_val != 0)
	626	return ret_val;
	627	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
fd385855	628	&netlbl_mgmt_genl_c_protocols);
d15c345f PM	629	if (ret_val != 0)
	630	return ret_val;
	631	ret_val = genl_register_ops(&netlbl_mgmt_gnl_family,
	632	&netlbl_mgmt_genl_c_version);
	633	if (ret_val != 0)
	634	return ret_val;
	635
	636	return 0;
	637	}