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

Add new two NFT_TUNNEL_IP6_SRC/DST match in nft_tunnel

Signed-off-by: wenxu <we...@ucloud.cn>
---
v4: new patch

 include/uapi/linux/netfilter/nf_tables.h |  2 ++
 net/netfilter/nft_tunnel.c               | 24 ++++++++++++++++++++++++
 2 files changed, 26 insertions(+)

diff --git a/include/uapi/linux/netfilter/nf_tables.h 
b/include/uapi/linux/netfilter/nf_tables.h
index 4f1e5ef..bfb051f 100644
--- a/include/uapi/linux/netfilter/nf_tables.h
+++ b/include/uapi/linux/netfilter/nf_tables.h
@@ -1767,6 +1767,8 @@ enum nft_tunnel_keys {
        NFT_TUNNEL_ID,
        NFT_TUNNEL_IP_SRC,
        NFT_TUNNEL_IP_DST,
+       NFT_TUNNEL_IP6_SRC,
+       NFT_TUNNEL_IP6_DST,
        __NFT_TUNNEL_MAX
 };
 #define NFT_TUNNEL_MAX (__NFT_TUNNEL_MAX - 1)
diff --git a/net/netfilter/nft_tunnel.c b/net/netfilter/nft_tunnel.c
index 64bda3d..58b6083 100644
--- a/net/netfilter/nft_tunnel.c
+++ b/net/netfilter/nft_tunnel.c
@@ -85,6 +85,26 @@ static void nft_tunnel_get_eval(const struct nft_expr *expr,
                else
                        regs->verdict.code = NFT_BREAK;
                break;
+       case NFT_TUNNEL_IP6_SRC:
+               if (!tun_info) {
+                       regs->verdict.code = NFT_BREAK;
+                       return;
+               }
+               if (nft_tunnel_mode_validate(priv->mode, tun_info->mode, true))
+                       memcpy(dest, &tun_info->key.u.ipv6.src, sizeof(struct 
in6_addr));
+               else
+                       regs->verdict.code = NFT_BREAK;
+               break;
+       case NFT_TUNNEL_IP6_DST:
+               if (!tun_info) {
+                       regs->verdict.code = NFT_BREAK;
+                       return;
+               }
+               if (nft_tunnel_mode_validate(priv->mode, tun_info->mode, true))
+                       memcpy(dest, &tun_info->key.u.ipv6.dst, sizeof(struct 
in6_addr));
+               else
+                       regs->verdict.code = NFT_BREAK;
+               break;
        default:
                WARN_ON(1);
                regs->verdict.code = NFT_BREAK;
@@ -118,6 +138,10 @@ static int nft_tunnel_get_init(const struct nft_ctx *ctx,
        case NFT_TUNNEL_IP_DST:
                len = sizeof(u32);
                break;
+       case NFT_TUNNEL_IP6_SRC:
+       case NFT_TUNNEL_IP6_DST:
+               len = sizeof(struct in6_addr);
+               break;
        default:
                return -EOPNOTSUPP;
        }
-- 
1.8.3.1

Reply via email to