mirror of https://github.com/hak5/openwrt-owl.git
303 lines
6.8 KiB
C
303 lines
6.8 KiB
C
/*
|
|
* lib/genl/ctrl.c Generic Netlink Controller
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation version 2.1
|
|
* of the License.
|
|
*
|
|
* Copyright (c) 2003-2008 Thomas Graf <tgraf@suug.ch>
|
|
*/
|
|
|
|
/**
|
|
* @ingroup genl_mngt
|
|
* @defgroup ctrl Controller
|
|
* @brief
|
|
*
|
|
* @{
|
|
*/
|
|
|
|
#include <netlink-generic.h>
|
|
#include <netlink/netlink.h>
|
|
#include <netlink/genl/genl.h>
|
|
#include <netlink/genl/family.h>
|
|
#include <netlink/genl/mngt.h>
|
|
#include <netlink/genl/ctrl.h>
|
|
#include <netlink/utils.h>
|
|
|
|
/** @cond SKIP */
|
|
#define CTRL_VERSION 0x0001
|
|
|
|
static struct nl_cache_ops genl_ctrl_ops;
|
|
/** @endcond */
|
|
|
|
static int ctrl_request_update(struct nl_cache *c, struct nl_sock *h)
|
|
{
|
|
return genl_send_simple(h, GENL_ID_CTRL, CTRL_CMD_GETFAMILY,
|
|
CTRL_VERSION, NLM_F_DUMP);
|
|
}
|
|
|
|
static struct nla_policy ctrl_policy[CTRL_ATTR_MAX+1] = {
|
|
[CTRL_ATTR_FAMILY_ID] = { .type = NLA_U16 },
|
|
[CTRL_ATTR_FAMILY_NAME] = { .type = NLA_STRING,
|
|
.maxlen = GENL_NAMSIZ },
|
|
[CTRL_ATTR_VERSION] = { .type = NLA_U32 },
|
|
[CTRL_ATTR_HDRSIZE] = { .type = NLA_U32 },
|
|
[CTRL_ATTR_MAXATTR] = { .type = NLA_U32 },
|
|
[CTRL_ATTR_OPS] = { .type = NLA_NESTED },
|
|
};
|
|
|
|
static struct nla_policy family_op_policy[CTRL_ATTR_OP_MAX+1] = {
|
|
[CTRL_ATTR_OP_ID] = { .type = NLA_U32 },
|
|
[CTRL_ATTR_OP_FLAGS] = { .type = NLA_U32 },
|
|
};
|
|
|
|
static int ctrl_msg_parser(struct nl_cache_ops *ops, struct genl_cmd *cmd,
|
|
struct genl_info *info, void *arg)
|
|
{
|
|
struct genl_family *family;
|
|
struct nl_parser_param *pp = arg;
|
|
int err;
|
|
|
|
family = genl_family_alloc();
|
|
if (family == NULL) {
|
|
err = -NLE_NOMEM;
|
|
goto errout;
|
|
}
|
|
|
|
if (info->attrs[CTRL_ATTR_FAMILY_NAME] == NULL) {
|
|
err = -NLE_MISSING_ATTR;
|
|
goto errout;
|
|
}
|
|
|
|
if (info->attrs[CTRL_ATTR_FAMILY_ID] == NULL) {
|
|
err = -NLE_MISSING_ATTR;
|
|
goto errout;
|
|
}
|
|
|
|
family->ce_msgtype = info->nlh->nlmsg_type;
|
|
genl_family_set_id(family,
|
|
nla_get_u16(info->attrs[CTRL_ATTR_FAMILY_ID]));
|
|
genl_family_set_name(family,
|
|
nla_get_string(info->attrs[CTRL_ATTR_FAMILY_NAME]));
|
|
|
|
if (info->attrs[CTRL_ATTR_VERSION]) {
|
|
uint32_t version = nla_get_u32(info->attrs[CTRL_ATTR_VERSION]);
|
|
genl_family_set_version(family, version);
|
|
}
|
|
|
|
if (info->attrs[CTRL_ATTR_HDRSIZE]) {
|
|
uint32_t hdrsize = nla_get_u32(info->attrs[CTRL_ATTR_HDRSIZE]);
|
|
genl_family_set_hdrsize(family, hdrsize);
|
|
}
|
|
|
|
if (info->attrs[CTRL_ATTR_MAXATTR]) {
|
|
uint32_t maxattr = nla_get_u32(info->attrs[CTRL_ATTR_MAXATTR]);
|
|
genl_family_set_maxattr(family, maxattr);
|
|
}
|
|
|
|
if (info->attrs[CTRL_ATTR_OPS]) {
|
|
struct nlattr *nla, *nla_ops;
|
|
int remaining;
|
|
|
|
nla_ops = info->attrs[CTRL_ATTR_OPS];
|
|
nla_for_each_nested(nla, nla_ops, remaining) {
|
|
struct nlattr *tb[CTRL_ATTR_OP_MAX+1];
|
|
int flags = 0, id;
|
|
|
|
err = nla_parse_nested(tb, CTRL_ATTR_OP_MAX, nla,
|
|
family_op_policy);
|
|
if (err < 0)
|
|
goto errout;
|
|
|
|
if (tb[CTRL_ATTR_OP_ID] == NULL) {
|
|
err = -NLE_MISSING_ATTR;
|
|
goto errout;
|
|
}
|
|
|
|
id = nla_get_u32(tb[CTRL_ATTR_OP_ID]);
|
|
|
|
if (tb[CTRL_ATTR_OP_FLAGS])
|
|
flags = nla_get_u32(tb[CTRL_ATTR_OP_FLAGS]);
|
|
|
|
err = genl_family_add_op(family, id, flags);
|
|
if (err < 0)
|
|
goto errout;
|
|
|
|
}
|
|
}
|
|
|
|
err = pp->pp_cb((struct nl_object *) family, pp);
|
|
errout:
|
|
genl_family_put(family);
|
|
return err;
|
|
}
|
|
|
|
/**
|
|
* @name Cache Management
|
|
* @{
|
|
*/
|
|
|
|
int genl_ctrl_alloc_cache(struct nl_sock *sock, struct nl_cache **result)
|
|
{
|
|
return nl_cache_alloc_and_fill(&genl_ctrl_ops, sock, result);
|
|
}
|
|
|
|
/**
|
|
* Look up generic netlink family by id in the provided cache.
|
|
* @arg cache Generic netlink family cache.
|
|
* @arg id Family identifier.
|
|
*
|
|
* Searches through the cache looking for a registered family
|
|
* matching the specified identifier. The caller will own a
|
|
* reference on the returned object which needs to be given
|
|
* back after usage using genl_family_put().
|
|
*
|
|
* @return Generic netlink family object or NULL if no match was found.
|
|
*/
|
|
struct genl_family *genl_ctrl_search(struct nl_cache *cache, int id)
|
|
{
|
|
struct genl_family *fam;
|
|
|
|
if (cache->c_ops != &genl_ctrl_ops)
|
|
BUG();
|
|
|
|
nl_list_for_each_entry(fam, &cache->c_items, ce_list) {
|
|
if (fam->gf_id == id) {
|
|
nl_object_get((struct nl_object *) fam);
|
|
return fam;
|
|
}
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
/**
|
|
* @name Resolver
|
|
* @{
|
|
*/
|
|
|
|
/**
|
|
* Look up generic netlink family by family name in the provided cache.
|
|
* @arg cache Generic netlink family cache.
|
|
* @arg name Family name.
|
|
*
|
|
* Searches through the cache looking for a registered family
|
|
* matching the specified name. The caller will own a reference
|
|
* on the returned object which needs to be given back after
|
|
* usage using genl_family_put().
|
|
*
|
|
* @return Generic netlink family object or NULL if no match was found.
|
|
*/
|
|
struct genl_family *genl_ctrl_search_by_name(struct nl_cache *cache,
|
|
const char *name)
|
|
{
|
|
struct genl_family *fam;
|
|
|
|
if (cache->c_ops != &genl_ctrl_ops)
|
|
BUG();
|
|
|
|
nl_list_for_each_entry(fam, &cache->c_items, ce_list) {
|
|
if (!strcmp(name, fam->gf_name)) {
|
|
nl_object_get((struct nl_object *) fam);
|
|
return fam;
|
|
}
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
/** @} */
|
|
|
|
/**
|
|
* Resolve generic netlink family name to its identifier
|
|
* @arg sk Netlink socket.
|
|
* @arg name Name of generic netlink family
|
|
*
|
|
* Resolves the generic netlink family name to its identifer and returns
|
|
* it.
|
|
*
|
|
* @return A positive identifier or a negative error code.
|
|
*/
|
|
int genl_ctrl_resolve(struct nl_sock *sk, const char *name)
|
|
{
|
|
struct nl_cache *cache;
|
|
struct genl_family *family;
|
|
int err;
|
|
|
|
if ((err = genl_ctrl_alloc_cache(sk, &cache)) < 0)
|
|
return err;
|
|
|
|
family = genl_ctrl_search_by_name(cache, name);
|
|
if (family == NULL) {
|
|
err = -NLE_OBJ_NOTFOUND;
|
|
goto errout;
|
|
}
|
|
|
|
err = genl_family_get_id(family);
|
|
genl_family_put(family);
|
|
errout:
|
|
nl_cache_free(cache);
|
|
|
|
return err;
|
|
}
|
|
|
|
/** @} */
|
|
|
|
static struct genl_cmd genl_cmds[] = {
|
|
{
|
|
.c_id = CTRL_CMD_NEWFAMILY,
|
|
.c_name = "NEWFAMILY" ,
|
|
.c_maxattr = CTRL_ATTR_MAX,
|
|
.c_attr_policy = ctrl_policy,
|
|
.c_msg_parser = ctrl_msg_parser,
|
|
},
|
|
{
|
|
.c_id = CTRL_CMD_DELFAMILY,
|
|
.c_name = "DELFAMILY" ,
|
|
},
|
|
{
|
|
.c_id = CTRL_CMD_GETFAMILY,
|
|
.c_name = "GETFAMILY" ,
|
|
},
|
|
{
|
|
.c_id = CTRL_CMD_NEWOPS,
|
|
.c_name = "NEWOPS" ,
|
|
},
|
|
{
|
|
.c_id = CTRL_CMD_DELOPS,
|
|
.c_name = "DELOPS" ,
|
|
},
|
|
};
|
|
|
|
static struct genl_ops genl_ops = {
|
|
.o_cmds = genl_cmds,
|
|
.o_ncmds = ARRAY_SIZE(genl_cmds),
|
|
};
|
|
|
|
/** @cond SKIP */
|
|
extern struct nl_object_ops genl_family_ops;
|
|
/** @endcond */
|
|
|
|
static struct nl_cache_ops genl_ctrl_ops = {
|
|
.co_name = "genl/family",
|
|
.co_hdrsize = GENL_HDRSIZE(0),
|
|
.co_msgtypes = GENL_FAMILY(GENL_ID_CTRL, "nlctrl"),
|
|
.co_genl = &genl_ops,
|
|
.co_protocol = NETLINK_GENERIC,
|
|
.co_request_update = ctrl_request_update,
|
|
.co_obj_ops = &genl_family_ops,
|
|
};
|
|
|
|
static void __init ctrl_init(void)
|
|
{
|
|
genl_register(&genl_ctrl_ops);
|
|
}
|
|
|
|
static void __exit ctrl_exit(void)
|
|
{
|
|
genl_unregister(&genl_ctrl_ops);
|
|
}
|
|
|
|
/** @} */
|