#include <sys/param.h>
#include <sys/eventhandler.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/malloc.h>
#include <sys/mbuf.h>
#include <sys/errno.h>
#include <sys/proc.h>
#include <sys/syslog.h>
#include <sys/socket.h>
#include <sys/taskqueue.h>
#include <net/if.h>
#include <net/if_dl.h>
#include <net/if_types.h>
#include <net/if_arp.h>
#include <net/if_var.h>
#include <net/if_private.h>
#include <net/ethernet.h>
#include <net/if_bridgevar.h>
#include <net/vnet.h>
#include <netgraph/ng_message.h>
#include <netgraph/netgraph.h>
#include <netgraph/ng_parse.h>
#include <netgraph/ng_ether.h>
MODULE_VERSION(ng_ether, 1);
#define IFP2NG(ifp) ((ifp)->if_l2com)
struct private {
struct ifnet *ifp;
hook_p upper;
hook_p lower;
hook_p orphan;
u_char autoSrcAddr;
u_char promisc;
u_long hwassist;
u_int flags;
};
typedef struct private *priv_p;
extern void (*ng_ether_input_p)(struct ifnet *ifp, struct mbuf **mp);
extern void (*ng_ether_input_orphan_p)(struct ifnet *ifp, struct mbuf *m);
extern int (*ng_ether_output_p)(struct ifnet *ifp, struct mbuf **mp);
extern void (*ng_ether_attach_p)(struct ifnet *ifp);
extern void (*ng_ether_detach_p)(struct ifnet *ifp);
extern void (*ng_ether_link_state_p)(struct ifnet *ifp, int state);
static void ng_ether_input(struct ifnet *ifp, struct mbuf **mp);
static void ng_ether_input_orphan(struct ifnet *ifp, struct mbuf *m);
static int ng_ether_output(struct ifnet *ifp, struct mbuf **mp);
static void ng_ether_attach(struct ifnet *ifp);
static void ng_ether_detach(struct ifnet *ifp);
static void ng_ether_link_state(struct ifnet *ifp, int state);
static int ng_ether_rcv_lower(hook_p node, item_p item);
static int ng_ether_rcv_upper(hook_p node, item_p item);
static ng_constructor_t ng_ether_constructor;
static ng_rcvmsg_t ng_ether_rcvmsg;
static ng_shutdown_t ng_ether_shutdown;
static ng_newhook_t ng_ether_newhook;
static ng_rcvdata_t ng_ether_rcvdata;
static ng_disconnect_t ng_ether_disconnect;
static int ng_ether_mod_event(module_t mod, int event, void *data);
static eventhandler_tag ng_ether_ifnet_arrival_cookie;
static const struct ng_cmdlist ng_ether_cmdlist[] = {
{
NGM_ETHER_COOKIE,
NGM_ETHER_GET_IFNAME,
"getifname",
NULL,
&ng_parse_string_type
},
{
NGM_ETHER_COOKIE,
NGM_ETHER_GET_IFINDEX,
"getifindex",
NULL,
&ng_parse_int32_type
},
{
NGM_ETHER_COOKIE,
NGM_ETHER_GET_ENADDR,
"getenaddr",
NULL,
&ng_parse_enaddr_type
},
{
NGM_ETHER_COOKIE,
NGM_ETHER_SET_ENADDR,
"setenaddr",
&ng_parse_enaddr_type,
NULL
},
{
NGM_ETHER_COOKIE,
NGM_ETHER_GET_PROMISC,
"getpromisc",
NULL,
&ng_parse_int32_type
},
{
NGM_ETHER_COOKIE,
NGM_ETHER_SET_PROMISC,
"setpromisc",
&ng_parse_int32_type,
NULL
},
{
NGM_ETHER_COOKIE,
NGM_ETHER_GET_AUTOSRC,
"getautosrc",
NULL,
&ng_parse_int32_type
},
{
NGM_ETHER_COOKIE,
NGM_ETHER_SET_AUTOSRC,
"setautosrc",
&ng_parse_int32_type,
NULL
},
{
NGM_ETHER_COOKIE,
NGM_ETHER_ADD_MULTI,
"addmulti",
&ng_parse_enaddr_type,
NULL
},
{
NGM_ETHER_COOKIE,
NGM_ETHER_DEL_MULTI,
"delmulti",
&ng_parse_enaddr_type,
NULL
},
{
NGM_ETHER_COOKIE,
NGM_ETHER_DETACH,
"detach",
NULL,
NULL
},
{ 0 }
};
static struct ng_type ng_ether_typestruct = {
.version = NG_ABI_VERSION,
.name = NG_ETHER_NODE_TYPE,
.mod_event = ng_ether_mod_event,
.constructor = ng_ether_constructor,
.rcvmsg = ng_ether_rcvmsg,
.shutdown = ng_ether_shutdown,
.newhook = ng_ether_newhook,
.rcvdata = ng_ether_rcvdata,
.disconnect = ng_ether_disconnect,
.cmdlist = ng_ether_cmdlist,
};
NETGRAPH_INIT(ether, &ng_ether_typestruct);
static void
ng_ether_sanitize_ifname(const char *ifname, char *name)
{
int i;
for (i = 0; i < IFNAMSIZ; i++) {
if (ifname[i] == '.' || ifname[i] == ':')
name[i] = '_';
else
name[i] = ifname[i];
if (name[i] == '\0')
break;
}
}
static void
ng_ether_input(struct ifnet *ifp, struct mbuf **mp)
{
const node_p node = IFP2NG(ifp);
const priv_p priv = NG_NODE_PRIVATE(node);
int error;
if (priv->lower == NULL)
return;
NG_SEND_DATA_ONLY(error, priv->lower, *mp);
}
static void
ng_ether_input_orphan(struct ifnet *ifp, struct mbuf *m)
{
const node_p node = IFP2NG(ifp);
const priv_p priv = NG_NODE_PRIVATE(node);
int error;
if (priv->orphan == NULL) {
m_freem(m);
return;
}
NG_SEND_DATA_ONLY(error, priv->orphan, m);
}
static int
ng_ether_output(struct ifnet *ifp, struct mbuf **mp)
{
const node_p node = IFP2NG(ifp);
const priv_p priv = NG_NODE_PRIVATE(node);
int error = 0;
if (priv->upper == NULL)
return (0);
NG_OUTBOUND_THREAD_REF();
NG_SEND_DATA_ONLY(error, priv->upper, *mp);
NG_OUTBOUND_THREAD_UNREF();
return (error);
}
static void
ng_ether_attach(struct ifnet *ifp)
{
char name[IFNAMSIZ];
priv_p priv;
node_p node;
ng_ether_sanitize_ifname(ifp->if_xname, name);
if ((node = ng_name2noderef(NULL, name)) != NULL) {
NG_NODE_UNREF(node);
return;
}
KASSERT(!IFP2NG(ifp), ("%s: node already exists?", __func__));
if (ng_make_node_common(&ng_ether_typestruct, &node) != 0) {
log(LOG_ERR, "%s: can't %s for %s\n",
__func__, "create node", ifp->if_xname);
return;
}
priv = malloc(sizeof(*priv), M_NETGRAPH, M_NOWAIT | M_ZERO);
if (priv == NULL) {
log(LOG_ERR, "%s: can't %s for %s\n",
__func__, "allocate memory", ifp->if_xname);
NG_NODE_UNREF(node);
return;
}
NG_NODE_SET_PRIVATE(node, priv);
priv->ifp = ifp;
IFP2NG(ifp) = node;
priv->hwassist = ifp->if_hwassist;
if (ng_name_node(node, name) != 0)
log(LOG_WARNING, "%s: can't name node %s\n", __func__, name);
}
static void
ng_ether_detach(struct ifnet *ifp)
{
const node_p node = IFP2NG(ifp);
const priv_p priv = NG_NODE_PRIVATE(node);
taskqueue_drain(taskqueue_swi, &ifp->if_linktask);
NG_NODE_REALLY_DIE(node);
IFP2NG(ifp) = NULL;
priv->ifp = NULL;
ng_rmnode_self(node);
}
static void
ng_ether_link_state(struct ifnet *ifp, int state)
{
const node_p node = IFP2NG(ifp);
const priv_p priv = NG_NODE_PRIVATE(node);
struct ng_mesg *msg;
int cmd, dummy_error = 0;
if (state == LINK_STATE_UP)
cmd = NGM_LINK_IS_UP;
else if (state == LINK_STATE_DOWN)
cmd = NGM_LINK_IS_DOWN;
else
return;
if (priv->lower != NULL) {
NG_MKMESSAGE(msg, NGM_FLOW_COOKIE, cmd, 0, M_NOWAIT);
if (msg != NULL)
NG_SEND_MSG_HOOK(dummy_error, node, msg, priv->lower, 0);
}
if (priv->orphan != NULL) {
NG_MKMESSAGE(msg, NGM_FLOW_COOKIE, cmd, 0, M_NOWAIT);
if (msg != NULL)
NG_SEND_MSG_HOOK(dummy_error, node, msg, priv->orphan, 0);
}
}
static void
ng_ether_ifnet_arrival_event(void *arg __unused, struct ifnet *ifp)
{
char name[IFNAMSIZ];
node_p node;
if (ifp->if_type != IFT_ETHER &&
ifp->if_type != IFT_L2VLAN &&
ifp->if_type != IFT_BRIDGE)
return;
node = IFP2NG(ifp);
if (node == NULL)
return;
ng_ether_sanitize_ifname(ifp->if_xname, name);
if (ng_name_node(node, name) != 0)
log(LOG_WARNING, "%s: can't re-name node %s\n", __func__, name);
}
static int
ng_ether_constructor(node_p node)
{
return (EINVAL);
}
static int
ng_ether_newhook(node_p node, hook_p hook, const char *name)
{
const priv_p priv = NG_NODE_PRIVATE(node);
hook_p *hookptr;
if (strcmp(name, NG_ETHER_HOOK_DIVERT) == 0)
name = NG_ETHER_HOOK_LOWER;
if (strcmp(name, NG_ETHER_HOOK_UPPER) == 0) {
hookptr = &priv->upper;
NG_HOOK_SET_RCVDATA(hook, ng_ether_rcv_upper);
NG_HOOK_SET_TO_INBOUND(hook);
} else if (strcmp(name, NG_ETHER_HOOK_LOWER) == 0) {
hookptr = &priv->lower;
NG_HOOK_SET_RCVDATA(hook, ng_ether_rcv_lower);
} else if (strcmp(name, NG_ETHER_HOOK_ORPHAN) == 0) {
hookptr = &priv->orphan;
NG_HOOK_SET_RCVDATA(hook, ng_ether_rcv_lower);
} else
return (EINVAL);
if (*hookptr != NULL)
return (EISCONN);
if (hookptr == &priv->upper)
priv->ifp->if_hwassist = 0;
NG_HOOK_HI_STACK(hook);
*hookptr = hook;
return (0);
}
static int
ng_ether_rcvmsg(node_p node, item_p item, hook_p lasthook)
{
const priv_p priv = NG_NODE_PRIVATE(node);
struct ng_mesg *resp = NULL;
int error = 0;
struct ng_mesg *msg;
NGI_GET_MSG(item, msg);
switch (msg->header.typecookie) {
case NGM_ETHER_COOKIE:
switch (msg->header.cmd) {
case NGM_ETHER_GET_IFNAME:
NG_MKRESPONSE(resp, msg, IFNAMSIZ, M_NOWAIT);
if (resp == NULL) {
error = ENOMEM;
break;
}
strlcpy(resp->data, priv->ifp->if_xname, IFNAMSIZ);
break;
case NGM_ETHER_GET_IFINDEX:
NG_MKRESPONSE(resp, msg, sizeof(u_int32_t), M_NOWAIT);
if (resp == NULL) {
error = ENOMEM;
break;
}
*((u_int32_t *)resp->data) = priv->ifp->if_index;
break;
case NGM_ETHER_GET_ENADDR:
NG_MKRESPONSE(resp, msg, ETHER_ADDR_LEN, M_NOWAIT);
if (resp == NULL) {
error = ENOMEM;
break;
}
bcopy(IF_LLADDR(priv->ifp),
resp->data, ETHER_ADDR_LEN);
break;
case NGM_ETHER_SET_ENADDR:
{
if (msg->header.arglen != ETHER_ADDR_LEN) {
error = EINVAL;
break;
}
error = if_setlladdr(priv->ifp,
(u_char *)msg->data, ETHER_ADDR_LEN);
break;
}
case NGM_ETHER_GET_PROMISC:
NG_MKRESPONSE(resp, msg, sizeof(u_int32_t), M_NOWAIT);
if (resp == NULL) {
error = ENOMEM;
break;
}
*((u_int32_t *)resp->data) = priv->promisc;
break;
case NGM_ETHER_SET_PROMISC:
{
u_char want;
if (msg->header.arglen != sizeof(u_int32_t)) {
error = EINVAL;
break;
}
want = !!*((u_int32_t *)msg->data);
if (want ^ priv->promisc) {
if ((error = ifpromisc(priv->ifp, want)) != 0)
break;
priv->promisc = want;
}
break;
}
case NGM_ETHER_GET_AUTOSRC:
NG_MKRESPONSE(resp, msg, sizeof(u_int32_t), M_NOWAIT);
if (resp == NULL) {
error = ENOMEM;
break;
}
*((u_int32_t *)resp->data) = priv->autoSrcAddr;
break;
case NGM_ETHER_SET_AUTOSRC:
if (msg->header.arglen != sizeof(u_int32_t)) {
error = EINVAL;
break;
}
priv->autoSrcAddr = !!*((u_int32_t *)msg->data);
break;
case NGM_ETHER_ADD_MULTI:
{
struct sockaddr_dl sa_dl;
struct epoch_tracker et;
struct ifmultiaddr *ifma;
if (msg->header.arglen != ETHER_ADDR_LEN) {
error = EINVAL;
break;
}
bzero(&sa_dl, sizeof(struct sockaddr_dl));
sa_dl.sdl_len = sizeof(struct sockaddr_dl);
sa_dl.sdl_family = AF_LINK;
sa_dl.sdl_alen = ETHER_ADDR_LEN;
bcopy((void *)msg->data, LLADDR(&sa_dl),
ETHER_ADDR_LEN);
NET_EPOCH_ENTER(et);
ifma = if_findmulti(priv->ifp,
(struct sockaddr *)&sa_dl);
NET_EPOCH_EXIT(et);
if (ifma != NULL) {
error = EADDRINUSE;
} else {
error = if_addmulti(priv->ifp,
(struct sockaddr *)&sa_dl, &ifma);
}
break;
}
case NGM_ETHER_DEL_MULTI:
{
struct sockaddr_dl sa_dl;
if (msg->header.arglen != ETHER_ADDR_LEN) {
error = EINVAL;
break;
}
bzero(&sa_dl, sizeof(struct sockaddr_dl));
sa_dl.sdl_len = sizeof(struct sockaddr_dl);
sa_dl.sdl_family = AF_LINK;
sa_dl.sdl_alen = ETHER_ADDR_LEN;
bcopy((void *)msg->data, LLADDR(&sa_dl),
ETHER_ADDR_LEN);
error = if_delmulti(priv->ifp,
(struct sockaddr *)&sa_dl);
break;
}
case NGM_ETHER_DETACH:
ng_ether_detach(priv->ifp);
break;
default:
error = EINVAL;
break;
}
break;
default:
error = EINVAL;
break;
}
NG_RESPOND_MSG(error, node, item, resp);
NG_FREE_MSG(msg);
return (error);
}
static int
ng_ether_rcvdata(hook_p hook, item_p item)
{
NG_FREE_ITEM(item);
panic("%s: weird hook", __func__);
}
static int
ng_ether_rcv_lower(hook_p hook, item_p item)
{
struct mbuf *m;
const node_p node = NG_HOOK_NODE(hook);
const priv_p priv = NG_NODE_PRIVATE(node);
struct ifnet *const ifp = priv->ifp;
NGI_GET_M(item, m);
NG_FREE_ITEM(item);
if (!((ifp->if_flags & IFF_UP) &&
(ifp->if_drv_flags & IFF_DRV_RUNNING))) {
NG_FREE_M(m);
return (ENETDOWN);
}
if (m->m_pkthdr.len < sizeof(struct ether_header)) {
NG_FREE_M(m);
return (EINVAL);
}
if (m->m_len < sizeof(struct ether_header)
&& (m = m_pullup(m, sizeof(struct ether_header))) == NULL)
return (ENOBUFS);
if (priv->autoSrcAddr) {
if (!M_WRITABLE(m)
&& (m = m_pullup(m, sizeof(struct ether_header))) == NULL)
return (ENOBUFS);
bcopy(IF_LLADDR(ifp),
mtod(m, struct ether_header *)->ether_shost,
ETHER_ADDR_LEN);
}
return ether_output_frame(ifp, m);
}
static int
ng_ether_rcv_upper(hook_p hook, item_p item)
{
struct mbuf *m;
const node_p node = NG_HOOK_NODE(hook);
const priv_p priv = NG_NODE_PRIVATE(node);
struct ifnet *ifp = priv->ifp;
NGI_GET_M(item, m);
NG_FREE_ITEM(item);
if (m->m_pkthdr.len < sizeof(struct ether_header)) {
NG_FREE_M(m);
return (EINVAL);
}
if (m->m_len < sizeof(struct ether_header) &&
(m = m_pullup(m, sizeof(struct ether_header))) == NULL)
return (ENOBUFS);
m->m_pkthdr.rcvif = ifp;
if (ifp->if_bridge) {
BRIDGE_INPUT(ifp, m);
if (m == NULL)
return (0);
}
ether_demux(ifp, m);
return (0);
}
static int
ng_ether_shutdown(node_p node)
{
const priv_p priv = NG_NODE_PRIVATE(node);
if (node->nd_flags & NGF_REALLY_DIE) {
NG_NODE_SET_PRIVATE(node, NULL);
if (priv->ifp != NULL)
IFP2NG(priv->ifp) = NULL;
free(priv, M_NETGRAPH);
NG_NODE_UNREF(node);
return (0);
}
if (priv->promisc) {
(void)ifpromisc(priv->ifp, 0);
priv->promisc = 0;
}
NG_NODE_REVIVE(node);
return (0);
}
static int
ng_ether_disconnect(hook_p hook)
{
const priv_p priv = NG_NODE_PRIVATE(NG_HOOK_NODE(hook));
if (hook == priv->upper) {
priv->upper = NULL;
if (priv->ifp != NULL)
priv->ifp->if_hwassist = priv->hwassist;
} else if (hook == priv->lower)
priv->lower = NULL;
else if (hook == priv->orphan)
priv->orphan = NULL;
else
panic("%s: weird hook", __func__);
if ((NG_NODE_NUMHOOKS(NG_HOOK_NODE(hook)) == 0)
&& (NG_NODE_IS_VALID(NG_HOOK_NODE(hook))))
ng_rmnode_self(NG_HOOK_NODE(hook));
return (0);
}
static int
ng_ether_mod_event(module_t mod, int event, void *data)
{
int error = 0;
switch (event) {
case MOD_LOAD:
if (ng_ether_attach_p != NULL) {
error = EEXIST;
break;
}
ng_ether_attach_p = ng_ether_attach;
ng_ether_detach_p = ng_ether_detach;
ng_ether_output_p = ng_ether_output;
ng_ether_input_p = ng_ether_input;
ng_ether_input_orphan_p = ng_ether_input_orphan;
ng_ether_link_state_p = ng_ether_link_state;
ng_ether_ifnet_arrival_cookie =
EVENTHANDLER_REGISTER(ifnet_arrival_event,
ng_ether_ifnet_arrival_event, NULL, EVENTHANDLER_PRI_ANY);
break;
case MOD_UNLOAD:
EVENTHANDLER_DEREGISTER(ifnet_arrival_event,
ng_ether_ifnet_arrival_cookie);
ng_ether_attach_p = NULL;
ng_ether_detach_p = NULL;
ng_ether_output_p = NULL;
ng_ether_input_p = NULL;
ng_ether_input_orphan_p = NULL;
ng_ether_link_state_p = NULL;
break;
default:
error = EOPNOTSUPP;
break;
}
return (error);
}
static void
vnet_ng_ether_init(const void *unused)
{
struct ifnet *ifp;
if (ng_ether_attach_p != ng_ether_attach)
return;
IFNET_RLOCK();
CK_STAILQ_FOREACH(ifp, &V_ifnet, if_link) {
if (ifp->if_type == IFT_ETHER ||
ifp->if_type == IFT_L2VLAN ||
ifp->if_type == IFT_BRIDGE)
ng_ether_attach(ifp);
}
IFNET_RUNLOCK();
}
VNET_SYSINIT(vnet_ng_ether_init, SI_SUB_PSEUDO, SI_ORDER_ANY,
vnet_ng_ether_init, NULL);