From: wenxu <we...@ucloud.cn>

With nf_conntrack_bridge function. The bridge family can do
conntrack it self. The flow offload function based on the
conntrack. This patch add bridge family operation in
nf_flow_table_core

Signed-off-by: wenxu <we...@ucloud.cn>
---
 include/net/netfilter/nf_flow_table.h | 31 +++++++++++++++++--
 net/netfilter/nf_flow_table_core.c    | 58 +++++++++++++++++++++++++++++------
 2 files changed, 78 insertions(+), 11 deletions(-)

diff --git a/include/net/netfilter/nf_flow_table.h 
b/include/net/netfilter/nf_flow_table.h
index d40d409..dcf197a 100644
--- a/include/net/netfilter/nf_flow_table.h
+++ b/include/net/netfilter/nf_flow_table.h
@@ -33,8 +33,23 @@ enum flow_offload_tuple_dir {
        FLOW_OFFLOAD_DIR_MAX = IP_CT_DIR_MAX
 };
 
+enum flow_offload_tuple_type {
+       FLOW_OFFLOAD_TYPE_INET,
+       FLOW_OFFLOAD_TYPE_BRIDGE,
+};
+
+struct dst_br_port {
+       struct net_device *dev;
+       u16     dst_vlan_tag;
+       u16     vlan_proto;
+};
+
 struct flow_offload_dst {
-       struct dst_entry                *dst_cache;
+       enum flow_offload_tuple_type type;
+       union {
+               struct dst_entry                *dst_cache;
+               struct dst_br_port              dst_port;
+       };
 };
 
 struct flow_offload_tuple {
@@ -52,6 +67,7 @@ struct flow_offload_tuple {
        };
 
        int                             iifidx;
+       u16                             vlan_tag;
 
        u8                              l3proto;
        u8                              l4proto;
@@ -89,8 +105,19 @@ struct nf_flow_route {
        } tuple[FLOW_OFFLOAD_DIR_MAX];
 };
 
+struct nf_flow_forward {
+       struct {
+               struct dst_br_port      dst_port;
+               u16 vlan_tag;
+       } tuple[FLOW_OFFLOAD_DIR_MAX];
+};
+
 struct nf_flow_dst {
-       struct nf_flow_route route;
+       enum flow_offload_tuple_type type;
+       union {
+               struct nf_flow_route route;
+               struct nf_flow_forward forward;
+       };
 };
 
 struct flow_offload *flow_offload_alloc(struct nf_conn *ct,
diff --git a/net/netfilter/nf_flow_table_core.c 
b/net/netfilter/nf_flow_table_core.c
index 2bec409..08c1ca4 100644
--- a/net/netfilter/nf_flow_table_core.c
+++ b/net/netfilter/nf_flow_table_core.c
@@ -36,6 +36,21 @@ struct flow_offload_entry {
        return dst;
 }
 
+static struct dst_br_port *
+flow_offload_fill_bridge_dst(struct flow_offload_tuple *ft,
+                            struct nf_flow_forward *forward,
+                            enum flow_offload_tuple_dir dir)
+{
+       struct dst_br_port other_dst_port = forward->tuple[!dir].dst_port;
+       struct dst_br_port dst_port = forward->tuple[dir].dst_port;
+
+       ft->iifidx = other_dst_port.dev->ifindex;
+       ft->dst.dst_port = dst_port;
+       ft->vlan_tag = forward->tuple[dir].vlan_tag;
+
+       return &ft->dst.dst_port;
+}
+
 static void
 flow_offload_fill_dir(struct flow_offload *flow, struct nf_conn *ct,
                      struct nf_flow_dst *flow_dst,
@@ -43,16 +58,29 @@ struct flow_offload_entry {
 {
        struct flow_offload_tuple *ft = &flow->tuplehash[dir].tuple;
        struct nf_conntrack_tuple *ctt = &ct->tuplehash[dir].tuple;
+       struct dst_br_port *dst_port;
        struct dst_entry *dst;
 
-       dst = flow_offload_fill_inet_dst(ft, &flow_dst->route, dir);
+       switch (flow_dst->type) {
+       case FLOW_OFFLOAD_TYPE_INET:
+               dst = flow_offload_fill_inet_dst(ft, &flow_dst->route, dir);
+               break;
+       case FLOW_OFFLOAD_TYPE_BRIDGE:
+               dst_port = flow_offload_fill_bridge_dst(ft, &flow_dst->forward, 
dir);
+               break;
+       }
+
+       ft->dst.type = flow_dst->type;
        ft->dir = dir;
 
        switch (ctt->src.l3num) {
        case NFPROTO_IPV4:
                ft->src_v4 = ctt->src.u3.in;
                ft->dst_v4 = ctt->dst.u3.in;
-               ft->mtu = ip_dst_mtu_maybe_forward(dst, true);
+               if (flow_dst->type == FLOW_OFFLOAD_TYPE_INET)
+                       ft->mtu = ip_dst_mtu_maybe_forward(dst, true);
+               else
+                       ft->mtu = dst_port->dev->mtu;
                break;
        case NFPROTO_IPV6:
                ft->src_v6 = ctt->src.u3.in6;
@@ -67,13 +95,13 @@ struct flow_offload_entry {
        ft->dst_port = ctt->dst.u.tcp.port;
 }
 
-static int flow_offload_dst_hold(struct nf_flow_dst *flow_dst)
+static int flow_offload_dst_hold(struct nf_flow_route *route)
 {
-       if 
(!dst_hold_safe(flow_dst->route.tuple[FLOW_OFFLOAD_DIR_ORIGINAL].dst))
+       if (!dst_hold_safe(route->tuple[FLOW_OFFLOAD_DIR_ORIGINAL].dst))
                return -1;
 
-       if (!dst_hold_safe(flow_dst->route.tuple[FLOW_OFFLOAD_DIR_REPLY].dst)) {
-               
dst_release(flow_dst->route.tuple[FLOW_OFFLOAD_DIR_ORIGINAL].dst);
+       if (!dst_hold_safe(route->tuple[FLOW_OFFLOAD_DIR_REPLY].dst)) {
+               dst_release(route->tuple[FLOW_OFFLOAD_DIR_ORIGINAL].dst);
                return -1;
        }
 
@@ -96,7 +124,8 @@ struct flow_offload *
 
        flow = &entry->flow;
 
-       if (flow_offload_dst_hold(flow_dst))
+       if (flow_dst->type == FLOW_OFFLOAD_TYPE_INET &&
+           flow_offload_dst_hold(&flow_dst->route))
                goto err_dst_cache;
 
        entry->ct = ct;
@@ -156,8 +185,19 @@ static void flow_offload_fixup_ct_state(struct nf_conn *ct)
 
 static void flow_offload_dst_release(struct flow_offload *flow)
 {
-       
dst_release(flow->tuplehash[FLOW_OFFLOAD_DIR_ORIGINAL].tuple.dst.dst_cache);
-       
dst_release(flow->tuplehash[FLOW_OFFLOAD_DIR_REPLY].tuple.dst.dst_cache);
+       enum flow_offload_tuple_type type = 
flow->tuplehash[FLOW_OFFLOAD_DIR_ORIGINAL].tuple.dst.type;
+
+       switch (type) {
+       case FLOW_OFFLOAD_TYPE_INET:
+               
dst_release(flow->tuplehash[FLOW_OFFLOAD_DIR_ORIGINAL].tuple.dst.dst_cache);
+               
dst_release(flow->tuplehash[FLOW_OFFLOAD_DIR_REPLY].tuple.dst.dst_cache);
+               break;
+
+       case FLOW_OFFLOAD_TYPE_BRIDGE:
+               
dev_put(flow->tuplehash[FLOW_OFFLOAD_DIR_ORIGINAL].tuple.dst.dst_port.dev);
+               
dev_put(flow->tuplehash[FLOW_OFFLOAD_DIR_REPLY].tuple.dst.dst_port.dev);
+               break;
+       }
 }
 
 void flow_offload_free(struct flow_offload *flow)
-- 
1.8.3.1

Reply via email to