mirror of
https://github.com/Ysurac/openmptcprouter-feeds.git
synced 2025-03-09 15:40:03 +00:00
fix
This commit is contained in:
suyuan
parent
d25235c692
commit
59d8d433ab
5 changed files with 0 additions and 2472 deletions
|
|
@ -1,10 +0,0 @@
|
||||||
obj-$(CONFIG_FAST_CLASSIFIER) += fast-classifier.o
|
|
||||||
|
|
||||||
ifeq ($(SFE_SUPPORT_IPV6),)
|
|
||||||
SFE_SUPPORT_IPV6=y
|
|
||||||
endif
|
|
||||||
ccflags-$(SFE_SUPPORT_IPV6) += -DSFE_SUPPORT_IPV6
|
|
||||||
|
|
||||||
ccflags-y += -I$(obj)/../shortcut-fe
|
|
||||||
|
|
||||||
obj ?= .
|
|
||||||
File diff suppressed because it is too large
Load diff
|
|
@ -1,57 +0,0 @@
|
||||||
/*
|
|
||||||
* User space header to send message to the fast classifier
|
|
||||||
*
|
|
||||||
* Copyright (c) 2013,2016 The Linux Foundation. All rights reserved.
|
|
||||||
* Permission to use, copy, modify, and/or distribute this software for
|
|
||||||
* any purpose with or without fee is hereby granted, provided that the
|
|
||||||
* above copyright notice and this permission notice appear in all copies.
|
|
||||||
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
||||||
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
||||||
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
||||||
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
||||||
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
||||||
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
|
|
||||||
* OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include <linux/if_ether.h>
|
|
||||||
|
|
||||||
#define FAST_CLASSIFIER_GENL_VERSION (1)
|
|
||||||
#define FAST_CLASSIFIER_GENL_NAME "FC"
|
|
||||||
#define FAST_CLASSIFIER_GENL_MCGRP "FC_MCGRP"
|
|
||||||
#define FAST_CLASSIFIER_GENL_HDRSIZE (0)
|
|
||||||
|
|
||||||
enum {
|
|
||||||
FAST_CLASSIFIER_A_UNSPEC,
|
|
||||||
FAST_CLASSIFIER_A_TUPLE,
|
|
||||||
__FAST_CLASSIFIER_A_MAX,
|
|
||||||
};
|
|
||||||
|
|
||||||
#define FAST_CLASSIFIER_A_MAX (__FAST_CLASSIFIER_A_MAX - 1)
|
|
||||||
|
|
||||||
enum {
|
|
||||||
FAST_CLASSIFIER_C_UNSPEC,
|
|
||||||
FAST_CLASSIFIER_C_OFFLOAD,
|
|
||||||
FAST_CLASSIFIER_C_OFFLOADED,
|
|
||||||
FAST_CLASSIFIER_C_DONE,
|
|
||||||
__FAST_CLASSIFIER_C_MAX,
|
|
||||||
};
|
|
||||||
|
|
||||||
#define FAST_CLASSIFIER_C_MAX (__FAST_CLASSIFIER_C_MAX - 1)
|
|
||||||
|
|
||||||
struct fast_classifier_tuple {
|
|
||||||
unsigned short ethertype;
|
|
||||||
unsigned char proto;
|
|
||||||
union {
|
|
||||||
struct in_addr in;
|
|
||||||
struct in6_addr in6;
|
|
||||||
} src_saddr;
|
|
||||||
union {
|
|
||||||
struct in_addr in;
|
|
||||||
struct in6_addr in6;
|
|
||||||
} dst_saddr;
|
|
||||||
unsigned short sport;
|
|
||||||
unsigned short dport;
|
|
||||||
unsigned char smac[ETH_ALEN];
|
|
||||||
unsigned char dmac[ETH_ALEN];
|
|
||||||
};
|
|
||||||
|
|
@ -1,281 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2016 The Linux Foundation. All rights reserved.
|
|
||||||
* Permission to use, copy, modify, and/or distribute this software for
|
|
||||||
* any purpose with or without fee is hereby granted, provided that the
|
|
||||||
* above copyright notice and this permission notice appear in all copies.
|
|
||||||
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
||||||
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
||||||
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
||||||
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
||||||
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
||||||
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
|
|
||||||
* OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include <netlink/genl/genl.h>
|
|
||||||
#include <netlink/genl/ctrl.h>
|
|
||||||
#include <errno.h>
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <signal.h>
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
|
|
||||||
#define NL_CLASSIFIER_GENL_VERSION 1
|
|
||||||
#define NL_CLASSIFIER_GENL_FAMILY "FC"
|
|
||||||
#define NL_CLASSIFIER_GENL_GROUP "FC_MCGRP"
|
|
||||||
#define NL_CLASSIFIER_GENL_HDRSIZE 0
|
|
||||||
|
|
||||||
enum NL_CLASSIFIER_CMD {
|
|
||||||
NL_CLASSIFIER_CMD_UNSPEC,
|
|
||||||
NL_CLASSIFIER_CMD_ACCEL,
|
|
||||||
NL_CLASSIFIER_CMD_ACCEL_OK,
|
|
||||||
NL_CLASSIFIER_CMD_CONNECTION_CLOSED,
|
|
||||||
NL_CLASSIFIER_CMD_MAX,
|
|
||||||
};
|
|
||||||
|
|
||||||
enum NL_CLASSIFIER_ATTR {
|
|
||||||
NL_CLASSIFIER_ATTR_UNSPEC,
|
|
||||||
NL_CLASSIFIER_ATTR_TUPLE,
|
|
||||||
NL_CLASSIFIER_ATTR_MAX,
|
|
||||||
};
|
|
||||||
|
|
||||||
union nl_classifier_tuple_ip {
|
|
||||||
struct in_addr in;
|
|
||||||
struct in6_addr in6;
|
|
||||||
};
|
|
||||||
|
|
||||||
struct nl_classifier_tuple {
|
|
||||||
unsigned short af;
|
|
||||||
unsigned char proto;
|
|
||||||
union nl_classifier_tuple_ip src_ip;
|
|
||||||
union nl_classifier_tuple_ip dst_ip;
|
|
||||||
unsigned short sport;
|
|
||||||
unsigned short dport;
|
|
||||||
unsigned char smac[6];
|
|
||||||
unsigned char dmac[6];
|
|
||||||
};
|
|
||||||
|
|
||||||
struct nl_classifier_instance {
|
|
||||||
struct nl_sock *sock;
|
|
||||||
int family_id;
|
|
||||||
int group_id;
|
|
||||||
int stop;
|
|
||||||
};
|
|
||||||
|
|
||||||
struct nl_classifier_instance nl_cls_inst;
|
|
||||||
|
|
||||||
static struct nla_policy nl_classifier_genl_policy[(NL_CLASSIFIER_ATTR_MAX+1)] = {
|
|
||||||
[NL_CLASSIFIER_ATTR_TUPLE] = { .type = NLA_UNSPEC },
|
|
||||||
};
|
|
||||||
|
|
||||||
void nl_classifier_dump_nl_tuple(struct nl_classifier_tuple *tuple)
|
|
||||||
{
|
|
||||||
char ip_str[64];
|
|
||||||
|
|
||||||
printf("protocol = %s\n", (tuple->proto == IPPROTO_UDP) ? "udp" : ((tuple->proto == IPPROTO_TCP) ? "tcp" : "unknown"));
|
|
||||||
printf("source ip = %s\n", inet_ntop(tuple->af, &tuple->src_ip, ip_str, sizeof(ip_str)));
|
|
||||||
printf("destination ip = %s\n", inet_ntop(tuple->af, &tuple->dst_ip, ip_str, sizeof(ip_str)));
|
|
||||||
printf("source port = %d\n", ntohs(tuple->sport));
|
|
||||||
printf("destination port = %d\n", ntohs(tuple->dport));
|
|
||||||
}
|
|
||||||
|
|
||||||
int nl_classifier_msg_recv(struct nl_msg *msg, void *arg)
|
|
||||||
{
|
|
||||||
struct nlmsghdr *nlh = nlmsg_hdr(msg);
|
|
||||||
struct genlmsghdr *gnlh = nlmsg_data(nlh);
|
|
||||||
struct nlattr *attrs[(NL_CLASSIFIER_ATTR_MAX+1)];
|
|
||||||
|
|
||||||
genlmsg_parse(nlh, NL_CLASSIFIER_GENL_HDRSIZE, attrs, NL_CLASSIFIER_ATTR_MAX, nl_classifier_genl_policy);
|
|
||||||
|
|
||||||
switch (gnlh->cmd) {
|
|
||||||
case NL_CLASSIFIER_CMD_ACCEL_OK:
|
|
||||||
printf("Acceleration successful:\n");
|
|
||||||
nl_classifier_dump_nl_tuple(nla_data(attrs[NL_CLASSIFIER_ATTR_TUPLE]));
|
|
||||||
return NL_OK;
|
|
||||||
case NL_CLASSIFIER_CMD_CONNECTION_CLOSED:
|
|
||||||
printf("Connection is closed:\n");
|
|
||||||
nl_classifier_dump_nl_tuple(nla_data(attrs[NL_CLASSIFIER_ATTR_TUPLE]));
|
|
||||||
return NL_OK;
|
|
||||||
default:
|
|
||||||
printf("nl classifier received unknow message %d\n", gnlh->cmd);
|
|
||||||
}
|
|
||||||
|
|
||||||
return NL_SKIP;
|
|
||||||
}
|
|
||||||
|
|
||||||
void nl_classifier_offload(struct nl_classifier_instance *inst,
|
|
||||||
unsigned char proto, unsigned long *src_saddr,
|
|
||||||
unsigned long *dst_saddr, unsigned short sport,
|
|
||||||
unsigned short dport, int af)
|
|
||||||
{
|
|
||||||
struct nl_msg *msg;
|
|
||||||
int ret;
|
|
||||||
struct nl_classifier_tuple classifier_msg;
|
|
||||||
|
|
||||||
memset(&classifier_msg, 0, sizeof(classifier_msg));
|
|
||||||
classifier_msg.af = af;
|
|
||||||
classifier_msg.proto = proto;
|
|
||||||
memcpy(&classifier_msg.src_ip, src_saddr, (af == AF_INET ? 4 : 16));
|
|
||||||
memcpy(&classifier_msg.dst_ip, dst_saddr, (af == AF_INET ? 4 : 16));
|
|
||||||
classifier_msg.sport = sport;
|
|
||||||
classifier_msg.dport = dport;
|
|
||||||
|
|
||||||
msg = nlmsg_alloc();
|
|
||||||
if (!msg) {
|
|
||||||
printf("Unable to allocate message\n");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
genlmsg_put(msg, NL_AUTO_PID, NL_AUTO_SEQ, inst->family_id,
|
|
||||||
NL_CLASSIFIER_GENL_HDRSIZE, NLM_F_REQUEST,
|
|
||||||
NL_CLASSIFIER_CMD_ACCEL, NL_CLASSIFIER_GENL_VERSION);
|
|
||||||
nla_put(msg, NL_CLASSIFIER_ATTR_TUPLE, sizeof(classifier_msg), &classifier_msg);
|
|
||||||
|
|
||||||
ret = nl_send_auto(inst->sock, msg);
|
|
||||||
if (ret < 0) {
|
|
||||||
printf("send netlink message failed.\n");
|
|
||||||
nlmsg_free(msg);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
nlmsg_free(msg);
|
|
||||||
printf("nl classifier offload connection successful\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
int nl_classifier_init(struct nl_classifier_instance *inst)
|
|
||||||
{
|
|
||||||
int ret;
|
|
||||||
|
|
||||||
inst->sock = nl_socket_alloc();
|
|
||||||
if (!inst->sock) {
|
|
||||||
printf("Unable to allocation socket.\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
genl_connect(inst->sock);
|
|
||||||
|
|
||||||
inst->family_id = genl_ctrl_resolve(inst->sock, NL_CLASSIFIER_GENL_FAMILY);
|
|
||||||
if (inst->family_id < 0) {
|
|
||||||
printf("Unable to resolve family %s\n", NL_CLASSIFIER_GENL_FAMILY);
|
|
||||||
goto init_failed;
|
|
||||||
}
|
|
||||||
|
|
||||||
inst->group_id = genl_ctrl_resolve_grp(inst->sock, NL_CLASSIFIER_GENL_FAMILY, NL_CLASSIFIER_GENL_GROUP);
|
|
||||||
if (inst->group_id < 0) {
|
|
||||||
printf("Unable to resolve mcast group %s\n", NL_CLASSIFIER_GENL_GROUP);
|
|
||||||
goto init_failed;
|
|
||||||
}
|
|
||||||
|
|
||||||
ret = nl_socket_add_membership(inst->sock, inst->group_id);
|
|
||||||
if (ret < 0) {
|
|
||||||
printf("Unable to add membership\n");
|
|
||||||
goto init_failed;
|
|
||||||
}
|
|
||||||
|
|
||||||
nl_socket_disable_seq_check(inst->sock);
|
|
||||||
nl_socket_modify_cb(inst->sock, NL_CB_VALID, NL_CB_CUSTOM, nl_classifier_msg_recv, NULL);
|
|
||||||
|
|
||||||
printf("nl classifier init successful\n");
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
init_failed:
|
|
||||||
if (inst->sock) {
|
|
||||||
nl_close(inst->sock);
|
|
||||||
nl_socket_free(inst->sock);
|
|
||||||
inst->sock = NULL;
|
|
||||||
}
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
void nl_classifier_exit(struct nl_classifier_instance *inst)
|
|
||||||
{
|
|
||||||
if (inst->sock) {
|
|
||||||
nl_close(inst->sock);
|
|
||||||
nl_socket_free(inst->sock);
|
|
||||||
inst->sock = NULL;
|
|
||||||
}
|
|
||||||
printf("nl classifier exit successful\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
int nl_classifier_parse_arg(int argc, char *argv[], unsigned char *proto, unsigned long *src_saddr,
|
|
||||||
unsigned long *dst_saddr, unsigned short *sport, unsigned short *dport, int *af)
|
|
||||||
{
|
|
||||||
int ret;
|
|
||||||
unsigned short port;
|
|
||||||
|
|
||||||
if (argc < 7) {
|
|
||||||
printf("help: nl_classifier <v4|v6> <udp|tcp> <source ip> <destination ip> <source port> <destination port>\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (0 == strncmp(argv[1], "v4", 2)) {
|
|
||||||
*af = AF_INET;
|
|
||||||
} else if (0 == strncmp(argv[1], "v6", 2)) {
|
|
||||||
*af = AF_INET6;
|
|
||||||
} else {
|
|
||||||
printf("Address family is not supported");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (0 == strncmp(argv[2], "udp", 3)) {
|
|
||||||
*proto = IPPROTO_UDP;
|
|
||||||
} else if (0 == strncmp(argv[2], "tcp", 3)) {
|
|
||||||
*proto = IPPROTO_TCP;
|
|
||||||
} else {
|
|
||||||
printf("Protocol is not supported");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
ret = inet_pton(*af, argv[3], src_saddr);
|
|
||||||
if (ret <= 0) {
|
|
||||||
printf("source ip has wrong format\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
ret = inet_pton(*af, argv[4], dst_saddr);
|
|
||||||
if (ret <= 0) {
|
|
||||||
printf("destination ip has wrong format\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
port = strtol(argv[5], NULL, 0);
|
|
||||||
*sport = htons(port);
|
|
||||||
port = strtol(argv[6], NULL, 0);
|
|
||||||
*dport = htons(port);
|
|
||||||
|
|
||||||
printf("nl classifier parse arguments successful\n");
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int main(int argc, char *argv[])
|
|
||||||
{
|
|
||||||
struct nl_classifier_instance *inst = &nl_cls_inst;
|
|
||||||
unsigned char proto;
|
|
||||||
unsigned long src_addr[4];
|
|
||||||
unsigned long dst_addr[4];
|
|
||||||
unsigned short sport;
|
|
||||||
unsigned short dport;
|
|
||||||
int af;
|
|
||||||
int ret;
|
|
||||||
|
|
||||||
ret = nl_classifier_parse_arg(argc, argv, &proto, src_addr, dst_addr, &sport, &dport, &af);
|
|
||||||
if (ret < 0) {
|
|
||||||
printf("Failed to parse arguments\n");
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
ret = nl_classifier_init(inst);
|
|
||||||
if (ret < 0) {
|
|
||||||
printf("Unable to init generic netlink\n");
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
nl_classifier_offload(inst, proto, src_addr, dst_addr, sport, dport, af);
|
|
||||||
|
|
||||||
/* main loop to listen on message */
|
|
||||||
while (!inst->stop) {
|
|
||||||
nl_recvmsgs_default(inst->sock);
|
|
||||||
}
|
|
||||||
|
|
||||||
nl_classifier_exit(inst);
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
@ -1,232 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2013,2016 The Linux Foundation. All rights reserved.
|
|
||||||
* Permission to use, copy, modify, and/or distribute this software for
|
|
||||||
* any purpose with or without fee is hereby granted, provided that the
|
|
||||||
* above copyright notice and this permission notice appear in all copies.
|
|
||||||
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
||||||
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
||||||
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
||||||
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
||||||
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
||||||
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
|
|
||||||
* OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include <netlink/genl/genl.h>
|
|
||||||
#include <netlink/genl/ctrl.h>
|
|
||||||
#include <errno.h>
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
|
|
||||||
#include <fast-classifier.h>
|
|
||||||
|
|
||||||
static struct nl_sock *sock;
|
|
||||||
static struct nl_sock *sock_event;
|
|
||||||
static int family;
|
|
||||||
static int grp_id;
|
|
||||||
|
|
||||||
static struct nla_policy fast_classifier_genl_policy[FAST_CLASSIFIER_A_MAX + 1] = {
|
|
||||||
[FAST_CLASSIFIER_A_TUPLE] = { .type = NLA_UNSPEC },
|
|
||||||
};
|
|
||||||
|
|
||||||
void dump_fc_tuple(struct fast_classifier_tuple *fc_msg)
|
|
||||||
{
|
|
||||||
char src_str[INET_ADDRSTRLEN];
|
|
||||||
char dst_str[INET_ADDRSTRLEN];
|
|
||||||
|
|
||||||
printf("TUPLE: %d, %s, %s, %d, %d"
|
|
||||||
" SMAC=%02x:%02x:%02x:%02x:%02x:%02x",
|
|
||||||
" DMAC=%02x:%02x:%02x:%02x:%02x:%02x\n",
|
|
||||||
fc_msg->proto,
|
|
||||||
inet_ntop(AF_INET,
|
|
||||||
&fc_msg->src_saddr.in.s_addr,
|
|
||||||
src_str,
|
|
||||||
INET_ADDRSTRLEN),
|
|
||||||
inet_ntop(AF_INET,
|
|
||||||
&fc_msg->dst_saddr.in.s_addr,
|
|
||||||
dst_str,
|
|
||||||
INET_ADDRSTRLEN),
|
|
||||||
fc_msg->sport, fc_msg->dport,
|
|
||||||
fc_msg->smac[0], fc_msg->smac[1], fc_msg->smac[2],
|
|
||||||
fc_msg->smac[3], fc_msg->smac[4], fc_msg->smac[5],
|
|
||||||
fc_msg->dmac[0], fc_msg->dmac[1], fc_msg->dmac[2],
|
|
||||||
fc_msg->dmac[3], fc_msg->dmac[4], fc_msg->dmac[5]);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int parse_cb(struct nl_msg *msg, void *arg)
|
|
||||||
{
|
|
||||||
struct nlmsghdr *nlh = nlmsg_hdr(msg);
|
|
||||||
struct genlmsghdr *gnlh = nlmsg_data(nlh);
|
|
||||||
struct nlattr *attrs[FAST_CLASSIFIER_A_MAX];
|
|
||||||
|
|
||||||
genlmsg_parse(nlh, 0, attrs, FAST_CLASSIFIER_A_MAX, fast_classifier_genl_policy);
|
|
||||||
|
|
||||||
switch (gnlh->cmd) {
|
|
||||||
case FAST_CLASSIFIER_C_OFFLOADED:
|
|
||||||
printf("Got a offloaded message\n");
|
|
||||||
dump_fc_tuple(nla_data(attrs[FAST_CLASSIFIER_A_TUPLE]));
|
|
||||||
return NL_OK;
|
|
||||||
case FAST_CLASSIFIER_C_DONE:
|
|
||||||
printf("Got a done message\n");
|
|
||||||
dump_fc_tuple(nla_data(attrs[FAST_CLASSIFIER_A_TUPLE]));
|
|
||||||
return NL_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
return NL_SKIP;
|
|
||||||
}
|
|
||||||
|
|
||||||
int fast_classifier_init(void)
|
|
||||||
{
|
|
||||||
int err;
|
|
||||||
|
|
||||||
sock = nl_socket_alloc();
|
|
||||||
if (!sock) {
|
|
||||||
printf("Unable to allocation socket.\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
genl_connect(sock);
|
|
||||||
|
|
||||||
sock_event = nl_socket_alloc();
|
|
||||||
if (!sock_event) {
|
|
||||||
nl_close(sock);
|
|
||||||
nl_socket_free(sock);
|
|
||||||
printf("Unable to allocation socket.\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
genl_connect(sock_event);
|
|
||||||
|
|
||||||
family = genl_ctrl_resolve(sock, FAST_CLASSIFIER_GENL_NAME);
|
|
||||||
if (family < 0) {
|
|
||||||
nl_close(sock_event);
|
|
||||||
nl_close(sock);
|
|
||||||
nl_socket_free(sock);
|
|
||||||
nl_socket_free(sock_event);
|
|
||||||
printf("Unable to resolve family\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
grp_id = genl_ctrl_resolve_grp(sock, FAST_CLASSIFIER_GENL_NAME,
|
|
||||||
FAST_CLASSIFIER_GENL_MCGRP);
|
|
||||||
if (grp_id < 0) {
|
|
||||||
printf("Unable to resolve mcast group\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
err = nl_socket_add_membership(sock_event, grp_id);
|
|
||||||
if (err < 0) {
|
|
||||||
printf("Unable to add membership\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
nl_socket_disable_seq_check(sock_event);
|
|
||||||
nl_socket_modify_cb(sock_event, NL_CB_VALID, NL_CB_CUSTOM, parse_cb, NULL);
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
void fast_classifier_close(void)
|
|
||||||
{
|
|
||||||
nl_close(sock_event);
|
|
||||||
nl_close(sock);
|
|
||||||
nl_socket_free(sock_event);
|
|
||||||
nl_socket_free(sock);
|
|
||||||
}
|
|
||||||
|
|
||||||
void fast_classifier_ipv4_offload(unsigned char proto, unsigned long src_saddr,
|
|
||||||
unsigned long dst_saddr, unsigned short sport,
|
|
||||||
unsigned short dport)
|
|
||||||
{
|
|
||||||
struct nl_msg *msg;
|
|
||||||
int ret;
|
|
||||||
#ifdef DEBUG
|
|
||||||
char src_str[INET_ADDRSTRLEN];
|
|
||||||
char dst_str[INET_ADDRSTRLEN];
|
|
||||||
#endif
|
|
||||||
struct fast_classifier_tuple fc_msg;
|
|
||||||
|
|
||||||
#ifdef DEBUG
|
|
||||||
printf("DEBUG: would offload: %d, %s, %s, %d, %d\n", proto,
|
|
||||||
inet_ntop(AF_INET, &src_saddr, src_str, INET_ADDRSTRLEN),
|
|
||||||
inet_ntop(AF_INET, &dst_saddr, dst_str, INET_ADDRSTRLEN),
|
|
||||||
sport, dport);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
fc_msg.proto = proto;
|
|
||||||
fc_msg.src_saddr.in.s_addr = src_saddr;
|
|
||||||
fc_msg.dst_saddr.in.s_addr = dst_saddr;
|
|
||||||
fc_msg.sport = sport;
|
|
||||||
fc_msg.dport = dport;
|
|
||||||
fc_msg.smac[0] = 'a';
|
|
||||||
fc_msg.smac[1] = 'b';
|
|
||||||
fc_msg.smac[2] = 'c';
|
|
||||||
fc_msg.smac[3] = 'd';
|
|
||||||
fc_msg.smac[4] = 'e';
|
|
||||||
fc_msg.smac[5] = 'f';
|
|
||||||
fc_msg.dmac[0] = 'f';
|
|
||||||
fc_msg.dmac[1] = 'e';
|
|
||||||
fc_msg.dmac[2] = 'd';
|
|
||||||
fc_msg.dmac[3] = 'c';
|
|
||||||
fc_msg.dmac[4] = 'b';
|
|
||||||
fc_msg.dmac[5] = 'a';
|
|
||||||
|
|
||||||
if (fast_classifier_init() < 0) {
|
|
||||||
printf("Unable to init generic netlink\n");
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
msg = nlmsg_alloc();
|
|
||||||
if (!msg) {
|
|
||||||
nl_socket_free(sock);
|
|
||||||
printf("Unable to allocate message\n");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
genlmsg_put(msg, NL_AUTO_PID, NL_AUTO_SEQ, family,
|
|
||||||
FAST_CLASSIFIER_GENL_HDRSIZE, NLM_F_REQUEST,
|
|
||||||
FAST_CLASSIFIER_C_OFFLOAD, FAST_CLASSIFIER_GENL_VERSION);
|
|
||||||
nla_put(msg, 1, sizeof(fc_msg), &fc_msg);
|
|
||||||
|
|
||||||
ret = nl_send_auto_complete(sock, msg);
|
|
||||||
|
|
||||||
nlmsg_free(msg);
|
|
||||||
if (ret < 0) {
|
|
||||||
printf("nlmsg_free failed");
|
|
||||||
nl_close(sock);
|
|
||||||
nl_socket_free(sock);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
ret = nl_wait_for_ack(sock);
|
|
||||||
if (ret < 0) {
|
|
||||||
printf("wait for ack failed");
|
|
||||||
nl_close(sock);
|
|
||||||
nl_socket_free(sock);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void fast_classifier_listen_for_messages(void)
|
|
||||||
{
|
|
||||||
printf("waiting for netlink events\n");
|
|
||||||
|
|
||||||
while (1) {
|
|
||||||
nl_recvmsgs_default(sock_event);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
int main(int argc, char *argv[])
|
|
||||||
{
|
|
||||||
if (fast_classifier_init() < 0) {
|
|
||||||
printf("Unable to init generic netlink\n");
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
fast_classifier_ipv4_offload('a', 0, 0, 0, 0);
|
|
||||||
|
|
||||||
/* this never returns */
|
|
||||||
fast_classifier_listen_for_messages();
|
|
||||||
|
|
||||||
fast_classifier_close();
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue