Home | History | Annotate | Download | only in genl
      1 /*
      2  * lib/genl/ctrl.c		Generic Netlink Controller
      3  *
      4  *	This library is free software; you can redistribute it and/or
      5  *	modify it under the terms of the GNU Lesser General Public
      6  *	License as published by the Free Software Foundation version 2.1
      7  *	of the License.
      8  *
      9  * Copyright (c) 2003-2008 Thomas Graf <tgraf (at) suug.ch>
     10  */
     11 
     12 /**
     13  * @ingroup genl_mngt
     14  * @defgroup ctrl Controller
     15  * @brief
     16  *
     17  * @{
     18  */
     19 
     20 #include <netlink-generic.h>
     21 #include <netlink/netlink.h>
     22 #include <netlink/genl/genl.h>
     23 #include <netlink/genl/family.h>
     24 #include <netlink/genl/mngt.h>
     25 #include <netlink/genl/ctrl.h>
     26 #include <netlink/utils.h>
     27 
     28 /** @cond SKIP */
     29 #define CTRL_VERSION		0x0001
     30 
     31 static struct nl_cache_ops genl_ctrl_ops;
     32 /** @endcond */
     33 
     34 static int ctrl_request_update(struct nl_cache *c, struct nl_sock *h)
     35 {
     36 	return genl_send_simple(h, GENL_ID_CTRL, CTRL_CMD_GETFAMILY,
     37 				CTRL_VERSION, NLM_F_DUMP);
     38 }
     39 
     40 static struct nla_policy ctrl_policy[CTRL_ATTR_MAX+1] = {
     41 	[CTRL_ATTR_FAMILY_ID]	= { .type = NLA_U16 },
     42 	[CTRL_ATTR_FAMILY_NAME]	= { .type = NLA_STRING,
     43 				    .maxlen = GENL_NAMSIZ },
     44 	[CTRL_ATTR_VERSION]	= { .type = NLA_U32 },
     45 	[CTRL_ATTR_HDRSIZE]	= { .type = NLA_U32 },
     46 	[CTRL_ATTR_MAXATTR]	= { .type = NLA_U32 },
     47 	[CTRL_ATTR_OPS]		= { .type = NLA_NESTED },
     48 };
     49 
     50 static struct nla_policy family_op_policy[CTRL_ATTR_OP_MAX+1] = {
     51 	[CTRL_ATTR_OP_ID]	= { .type = NLA_U32 },
     52 	[CTRL_ATTR_OP_FLAGS]	= { .type = NLA_U32 },
     53 };
     54 
     55 static int ctrl_msg_parser(struct nl_cache_ops *ops, struct genl_cmd *cmd,
     56 			   struct genl_info *info, void *arg)
     57 {
     58 	struct genl_family *family;
     59 	struct nl_parser_param *pp = arg;
     60 	int err;
     61 
     62 	family = genl_family_alloc();
     63 	if (family == NULL) {
     64 		err = -NLE_NOMEM;
     65 		goto errout;
     66 	}
     67 
     68 	if (info->attrs[CTRL_ATTR_FAMILY_NAME] == NULL) {
     69 		err = -NLE_MISSING_ATTR;
     70 		goto errout;
     71 	}
     72 
     73 	if (info->attrs[CTRL_ATTR_FAMILY_ID] == NULL) {
     74 		err = -NLE_MISSING_ATTR;
     75 		goto errout;
     76 	}
     77 
     78 	family->ce_msgtype = info->nlh->nlmsg_type;
     79 	genl_family_set_id(family,
     80 			   nla_get_u16(info->attrs[CTRL_ATTR_FAMILY_ID]));
     81 	genl_family_set_name(family,
     82 		     nla_get_string(info->attrs[CTRL_ATTR_FAMILY_NAME]));
     83 
     84 	if (info->attrs[CTRL_ATTR_VERSION]) {
     85 		uint32_t version = nla_get_u32(info->attrs[CTRL_ATTR_VERSION]);
     86 		genl_family_set_version(family, version);
     87 	}
     88 
     89 	if (info->attrs[CTRL_ATTR_HDRSIZE]) {
     90 		uint32_t hdrsize = nla_get_u32(info->attrs[CTRL_ATTR_HDRSIZE]);
     91 		genl_family_set_hdrsize(family, hdrsize);
     92 	}
     93 
     94 	if (info->attrs[CTRL_ATTR_MAXATTR]) {
     95 		uint32_t maxattr = nla_get_u32(info->attrs[CTRL_ATTR_MAXATTR]);
     96 		genl_family_set_maxattr(family, maxattr);
     97 	}
     98 
     99 	if (info->attrs[CTRL_ATTR_OPS]) {
    100 		struct nlattr *nla, *nla_ops;
    101 		int remaining;
    102 
    103 		nla_ops = info->attrs[CTRL_ATTR_OPS];
    104 		nla_for_each_nested(nla, nla_ops, remaining) {
    105 			struct nlattr *tb[CTRL_ATTR_OP_MAX+1];
    106 			int flags = 0, id;
    107 
    108 			err = nla_parse_nested(tb, CTRL_ATTR_OP_MAX, nla,
    109 					       family_op_policy);
    110 			if (err < 0)
    111 				goto errout;
    112 
    113 			if (tb[CTRL_ATTR_OP_ID] == NULL) {
    114 				err = -NLE_MISSING_ATTR;
    115 				goto errout;
    116 			}
    117 
    118 			id = nla_get_u32(tb[CTRL_ATTR_OP_ID]);
    119 
    120 			if (tb[CTRL_ATTR_OP_FLAGS])
    121 				flags = nla_get_u32(tb[CTRL_ATTR_OP_FLAGS]);
    122 
    123 			err = genl_family_add_op(family, id, flags);
    124 			if (err < 0)
    125 				goto errout;
    126 
    127 		}
    128 	}
    129 
    130 	err = pp->pp_cb((struct nl_object *) family, pp);
    131 errout:
    132 	genl_family_put(family);
    133 	return err;
    134 }
    135 
    136 /**
    137  * @name Cache Management
    138  * @{
    139  */
    140 
    141 int genl_ctrl_alloc_cache(struct nl_sock *sock, struct nl_cache **result)
    142 {
    143 	return nl_cache_alloc_and_fill(&genl_ctrl_ops, sock, result);
    144 }
    145 
    146 /**
    147  * Look up generic netlink family by id in the provided cache.
    148  * @arg cache		Generic netlink family cache.
    149  * @arg id		Family identifier.
    150  *
    151  * Searches through the cache looking for a registered family
    152  * matching the specified identifier. The caller will own a
    153  * reference on the returned object which needs to be given
    154  * back after usage using genl_family_put().
    155  *
    156  * @return Generic netlink family object or NULL if no match was found.
    157  */
    158 struct genl_family *genl_ctrl_search(struct nl_cache *cache, int id)
    159 {
    160 	struct genl_family *fam;
    161 
    162 	if (cache->c_ops != &genl_ctrl_ops)
    163 		BUG();
    164 
    165 	nl_list_for_each_entry(fam, &cache->c_items, ce_list) {
    166 		if (fam->gf_id == id) {
    167 			nl_object_get((struct nl_object *) fam);
    168 			return fam;
    169 		}
    170 	}
    171 
    172 	return NULL;
    173 }
    174 
    175 /**
    176  * @name Resolver
    177  * @{
    178  */
    179 
    180 /**
    181  * Look up generic netlink family by family name in the provided cache.
    182  * @arg cache		Generic netlink family cache.
    183  * @arg name		Family name.
    184  *
    185  * Searches through the cache looking for a registered family
    186  * matching the specified name. The caller will own a reference
    187  * on the returned object which needs to be given back after
    188  * usage using genl_family_put().
    189  *
    190  * @return Generic netlink family object or NULL if no match was found.
    191  */
    192 struct genl_family *genl_ctrl_search_by_name(struct nl_cache *cache,
    193 					    const char *name)
    194 {
    195 	struct genl_family *fam;
    196 
    197 	if (cache->c_ops != &genl_ctrl_ops)
    198 		BUG();
    199 
    200 	nl_list_for_each_entry(fam, &cache->c_items, ce_list) {
    201 		if (!strcmp(name, fam->gf_name)) {
    202 			nl_object_get((struct nl_object *) fam);
    203 			return fam;
    204 		}
    205 	}
    206 
    207 	return NULL;
    208 }
    209 
    210 /** @} */
    211 
    212 /**
    213  * Resolve generic netlink family name to its identifier
    214  * @arg sk		Netlink socket.
    215  * @arg name		Name of generic netlink family
    216  *
    217  * Resolves the generic netlink family name to its identifer and returns
    218  * it.
    219  *
    220  * @return A positive identifier or a negative error code.
    221  */
    222 int genl_ctrl_resolve(struct nl_sock *sk, const char *name)
    223 {
    224 	struct nl_cache *cache;
    225 	struct genl_family *family;
    226 	int err;
    227 
    228 	if ((err = genl_ctrl_alloc_cache(sk, &cache)) < 0)
    229 		return err;
    230 
    231 	family = genl_ctrl_search_by_name(cache, name);
    232 	if (family == NULL) {
    233 		err = -NLE_OBJ_NOTFOUND;
    234 		goto errout;
    235 	}
    236 
    237 	err = genl_family_get_id(family);
    238 	genl_family_put(family);
    239 errout:
    240 	nl_cache_free(cache);
    241 
    242 	return err;
    243 }
    244 
    245 /** @} */
    246 
    247 static struct genl_cmd genl_cmds[] = {
    248 	{
    249 		.c_id		= CTRL_CMD_NEWFAMILY,
    250 		.c_name		= "NEWFAMILY" ,
    251 		.c_maxattr	= CTRL_ATTR_MAX,
    252 		.c_attr_policy	= ctrl_policy,
    253 		.c_msg_parser	= ctrl_msg_parser,
    254 	},
    255 	{
    256 		.c_id		= CTRL_CMD_DELFAMILY,
    257 		.c_name		= "DELFAMILY" ,
    258 	},
    259 	{
    260 		.c_id		= CTRL_CMD_GETFAMILY,
    261 		.c_name		= "GETFAMILY" ,
    262 	},
    263 	{
    264 		.c_id		= CTRL_CMD_NEWOPS,
    265 		.c_name		= "NEWOPS" ,
    266 	},
    267 	{
    268 		.c_id		= CTRL_CMD_DELOPS,
    269 		.c_name		= "DELOPS" ,
    270 	},
    271 };
    272 
    273 static struct genl_ops genl_ops = {
    274 	.o_cmds			= genl_cmds,
    275 	.o_ncmds		= ARRAY_SIZE(genl_cmds),
    276 };
    277 
    278 /** @cond SKIP */
    279 extern struct nl_object_ops genl_family_ops;
    280 /** @endcond */
    281 
    282 static struct nl_cache_ops genl_ctrl_ops = {
    283 	.co_name		= "genl/family",
    284 	.co_hdrsize		= GENL_HDRSIZE(0),
    285 	.co_msgtypes		= GENL_FAMILY(GENL_ID_CTRL, "nlctrl"),
    286 	.co_genl		= &genl_ops,
    287 	.co_protocol		= NETLINK_GENERIC,
    288 	.co_request_update      = ctrl_request_update,
    289 	.co_obj_ops		= &genl_family_ops,
    290 };
    291 
    292 static void __init ctrl_init(void)
    293 {
    294 	genl_register(&genl_ctrl_ops);
    295 }
    296 
    297 static void __exit ctrl_exit(void)
    298 {
    299 	genl_unregister(&genl_ctrl_ops);
    300 }
    301 
    302 /** @} */
    303