mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/stable/linux.git
synced 2024-09-18 00:24:39 +00:00
10596608c4
Currently, there are two different methods to store an u16 integer to the u32 data register. For example: u32 *dest = ®s->data[priv->dreg]; 1. *dest = 0; *(u16 *) dest = val_u16; 2. *dest = val_u16; For method 1, the u16 value will be stored like this, either in big-endian or little-endian system: 0 15 31 +-+-+-+-+-+-+-+-+-+-+-+-+ | Value | 0 | +-+-+-+-+-+-+-+-+-+-+-+-+ For method 2, in little-endian system, the u16 value will be the same as listed above. But in big-endian system, the u16 value will be stored like this: 0 15 31 +-+-+-+-+-+-+-+-+-+-+-+-+ | 0 | Value | +-+-+-+-+-+-+-+-+-+-+-+-+ So later we use "memcmp(®s->data[priv->sreg], data, 2);" to do compare in nft_cmp, nft_lookup expr ..., method 2 will get the wrong result in big-endian system, as 0~15 bits will always be zero. For the similar reason, when loading an u16 value from the u32 data register, we should use "*(u16 *) sreg;" instead of "(u16)*sreg;", the 2nd method will get the wrong value in the big-endian system. So introduce some wrapper functions to store/load an u8 or u16 integer to/from the u32 data register, and use them in the right place. Signed-off-by: Liping Zhang <zlpnobody@gmail.com> Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
88 lines
2.4 KiB
C
88 lines
2.4 KiB
C
/*
|
|
* Copyright (c) 2014 Arturo Borrero Gonzalez <arturo@debian.org>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License version 2 as
|
|
* published by the Free Software Foundation.
|
|
*/
|
|
|
|
#include <linux/kernel.h>
|
|
#include <linux/init.h>
|
|
#include <linux/module.h>
|
|
#include <linux/netlink.h>
|
|
#include <linux/netfilter.h>
|
|
#include <linux/netfilter/nf_tables.h>
|
|
#include <net/netfilter/nf_tables.h>
|
|
#include <net/netfilter/nft_masq.h>
|
|
#include <net/netfilter/ipv4/nf_nat_masquerade.h>
|
|
|
|
static void nft_masq_ipv4_eval(const struct nft_expr *expr,
|
|
struct nft_regs *regs,
|
|
const struct nft_pktinfo *pkt)
|
|
{
|
|
struct nft_masq *priv = nft_expr_priv(expr);
|
|
struct nf_nat_range range;
|
|
|
|
memset(&range, 0, sizeof(range));
|
|
range.flags = priv->flags;
|
|
if (priv->sreg_proto_min) {
|
|
range.min_proto.all = (__force __be16)nft_reg_load16(
|
|
®s->data[priv->sreg_proto_min]);
|
|
range.max_proto.all = (__force __be16)nft_reg_load16(
|
|
®s->data[priv->sreg_proto_max]);
|
|
}
|
|
regs->verdict.code = nf_nat_masquerade_ipv4(pkt->skb, nft_hook(pkt),
|
|
&range, nft_out(pkt));
|
|
}
|
|
|
|
static void
|
|
nft_masq_ipv4_destroy(const struct nft_ctx *ctx, const struct nft_expr *expr)
|
|
{
|
|
nf_ct_netns_put(ctx->net, NFPROTO_IPV4);
|
|
}
|
|
|
|
static struct nft_expr_type nft_masq_ipv4_type;
|
|
static const struct nft_expr_ops nft_masq_ipv4_ops = {
|
|
.type = &nft_masq_ipv4_type,
|
|
.size = NFT_EXPR_SIZE(sizeof(struct nft_masq)),
|
|
.eval = nft_masq_ipv4_eval,
|
|
.init = nft_masq_init,
|
|
.destroy = nft_masq_ipv4_destroy,
|
|
.dump = nft_masq_dump,
|
|
.validate = nft_masq_validate,
|
|
};
|
|
|
|
static struct nft_expr_type nft_masq_ipv4_type __read_mostly = {
|
|
.family = NFPROTO_IPV4,
|
|
.name = "masq",
|
|
.ops = &nft_masq_ipv4_ops,
|
|
.policy = nft_masq_policy,
|
|
.maxattr = NFTA_MASQ_MAX,
|
|
.owner = THIS_MODULE,
|
|
};
|
|
|
|
static int __init nft_masq_ipv4_module_init(void)
|
|
{
|
|
int ret;
|
|
|
|
ret = nft_register_expr(&nft_masq_ipv4_type);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
nf_nat_masquerade_ipv4_register_notifier();
|
|
|
|
return ret;
|
|
}
|
|
|
|
static void __exit nft_masq_ipv4_module_exit(void)
|
|
{
|
|
nft_unregister_expr(&nft_masq_ipv4_type);
|
|
nf_nat_masquerade_ipv4_unregister_notifier();
|
|
}
|
|
|
|
module_init(nft_masq_ipv4_module_init);
|
|
module_exit(nft_masq_ipv4_module_exit);
|
|
|
|
MODULE_LICENSE("GPL");
|
|
MODULE_AUTHOR("Arturo Borrero Gonzalez <arturo@debian.org");
|
|
MODULE_ALIAS_NFT_AF_EXPR(AF_INET, "masq");
|