diff options
author | Shivani Bhardwaj <shivanib134@gmail.com> | 2016-01-23 02:25:55 +0530 |
---|---|---|
committer | Pablo Neira Ayuso <pablo@netfilter.org> | 2016-03-03 19:54:30 +0100 |
commit | cf8e0db8aacbafac5acb1f42afd62d8215ca5863 (patch) | |
tree | 7202b75c39b68c1a6c553a3102c7b40dce10a702 /src/netlink_linearize.c | |
parent | 45e5e4e92a2c882b22e95a807026611612d57729 (diff) |
src: Add support for masquerade port selection
Provide full support for masquerading by allowing port range selection, eg.
# nft add rule nat postrouting ip protocol tcp masquerade to :1024-10024
Signed-off-by: Shivani Bhardwaj <shivanib134@gmail.com>
Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
Diffstat (limited to 'src/netlink_linearize.c')
-rw-r--r-- | src/netlink_linearize.c | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/src/netlink_linearize.c b/src/netlink_linearize.c index 7ff3b004..07f70e06 100644 --- a/src/netlink_linearize.c +++ b/src/netlink_linearize.c @@ -916,12 +916,36 @@ static void netlink_gen_nat_stmt(struct netlink_linearize_ctx *ctx, static void netlink_gen_masq_stmt(struct netlink_linearize_ctx *ctx, const struct stmt *stmt) { + enum nft_registers pmin_reg, pmax_reg; struct nftnl_expr *nle; + int registers = 0; nle = alloc_nft_expr("masq"); if (stmt->masq.flags != 0) nftnl_expr_set_u32(nle, NFTNL_EXPR_MASQ_FLAGS, stmt->masq.flags); + if (stmt->masq.proto) { + pmin_reg = get_register(ctx, NULL); + registers++; + + if (stmt->masq.proto->ops->type == EXPR_RANGE) { + pmax_reg = get_register(ctx, NULL); + registers++; + + netlink_gen_expr(ctx, stmt->masq.proto->left, pmin_reg); + netlink_gen_expr(ctx, stmt->masq.proto->right, pmax_reg); + netlink_put_register(nle, NFTNL_EXPR_MASQ_REG_PROTO_MIN, pmin_reg); + netlink_put_register(nle, NFTNL_EXPR_MASQ_REG_PROTO_MAX, pmax_reg); + } else { + netlink_gen_expr(ctx, stmt->masq.proto, pmin_reg); + netlink_put_register(nle, NFTNL_EXPR_MASQ_REG_PROTO_MIN, pmin_reg); + } + } + + while (registers > 0) { + release_register(ctx, NULL); + registers--; + } nftnl_rule_add_expr(ctx->nlr, nle); } |