1
0
Fork 0

Merge branch 'master' of git://1984.lsi.us.es/net

hifive-unleashed-5.1
David S. Miller 2012-06-12 15:18:15 -07:00
commit 5aa04d3af8
3 changed files with 48 additions and 34 deletions

View File

@ -27,7 +27,12 @@ union hmark_ports {
__u16 src; __u16 src;
__u16 dst; __u16 dst;
} p16; } p16;
struct {
__be16 src;
__be16 dst;
} b16;
__u32 v32; __u32 v32;
__be32 b32;
}; };
struct xt_hmark_info { struct xt_hmark_info {

View File

@ -270,9 +270,8 @@ static int expect_rtp_rtcp(struct sk_buff *skb, struct nf_conn *ct,
return 0; return 0;
/* RTP port is even */ /* RTP port is even */
port &= htons(~1); rtp_port = port & ~htons(1);
rtp_port = port; rtcp_port = port | htons(1);
rtcp_port = htons(ntohs(port) + 1);
/* Create expect for RTP */ /* Create expect for RTP */
if ((rtp_exp = nf_ct_expect_alloc(ct)) == NULL) if ((rtp_exp = nf_ct_expect_alloc(ct)) == NULL)

View File

@ -32,13 +32,13 @@ MODULE_ALIAS("ipt_HMARK");
MODULE_ALIAS("ip6t_HMARK"); MODULE_ALIAS("ip6t_HMARK");
struct hmark_tuple { struct hmark_tuple {
u32 src; __be32 src;
u32 dst; __be32 dst;
union hmark_ports uports; union hmark_ports uports;
uint8_t proto; u8 proto;
}; };
static inline u32 hmark_addr6_mask(const __u32 *addr32, const __u32 *mask) static inline __be32 hmark_addr6_mask(const __be32 *addr32, const __be32 *mask)
{ {
return (addr32[0] & mask[0]) ^ return (addr32[0] & mask[0]) ^
(addr32[1] & mask[1]) ^ (addr32[1] & mask[1]) ^
@ -46,8 +46,8 @@ static inline u32 hmark_addr6_mask(const __u32 *addr32, const __u32 *mask)
(addr32[3] & mask[3]); (addr32[3] & mask[3]);
} }
static inline u32 static inline __be32
hmark_addr_mask(int l3num, const __u32 *addr32, const __u32 *mask) hmark_addr_mask(int l3num, const __be32 *addr32, const __be32 *mask)
{ {
switch (l3num) { switch (l3num) {
case AF_INET: case AF_INET:
@ -58,6 +58,22 @@ hmark_addr_mask(int l3num, const __u32 *addr32, const __u32 *mask)
return 0; return 0;
} }
static inline void hmark_swap_ports(union hmark_ports *uports,
const struct xt_hmark_info *info)
{
union hmark_ports hp;
u16 src, dst;
hp.b32 = (uports->b32 & info->port_mask.b32) | info->port_set.b32;
src = ntohs(hp.b16.src);
dst = ntohs(hp.b16.dst);
if (dst > src)
uports->v32 = (dst << 16) | src;
else
uports->v32 = (src << 16) | dst;
}
static int static int
hmark_ct_set_htuple(const struct sk_buff *skb, struct hmark_tuple *t, hmark_ct_set_htuple(const struct sk_buff *skb, struct hmark_tuple *t,
const struct xt_hmark_info *info) const struct xt_hmark_info *info)
@ -74,22 +90,19 @@ hmark_ct_set_htuple(const struct sk_buff *skb, struct hmark_tuple *t,
otuple = &ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple; otuple = &ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple;
rtuple = &ct->tuplehash[IP_CT_DIR_REPLY].tuple; rtuple = &ct->tuplehash[IP_CT_DIR_REPLY].tuple;
t->src = hmark_addr_mask(otuple->src.l3num, otuple->src.u3.all, t->src = hmark_addr_mask(otuple->src.l3num, otuple->src.u3.ip6,
info->src_mask.all); info->src_mask.ip6);
t->dst = hmark_addr_mask(otuple->src.l3num, rtuple->src.u3.all, t->dst = hmark_addr_mask(otuple->src.l3num, rtuple->src.u3.ip6,
info->dst_mask.all); info->dst_mask.ip6);
if (info->flags & XT_HMARK_FLAG(XT_HMARK_METHOD_L3)) if (info->flags & XT_HMARK_FLAG(XT_HMARK_METHOD_L3))
return 0; return 0;
t->proto = nf_ct_protonum(ct); t->proto = nf_ct_protonum(ct);
if (t->proto != IPPROTO_ICMP) { if (t->proto != IPPROTO_ICMP) {
t->uports.p16.src = otuple->src.u.all; t->uports.b16.src = otuple->src.u.all;
t->uports.p16.dst = rtuple->src.u.all; t->uports.b16.dst = rtuple->src.u.all;
t->uports.v32 = (t->uports.v32 & info->port_mask.v32) | hmark_swap_ports(&t->uports, info);
info->port_set.v32;
if (t->uports.p16.dst < t->uports.p16.src)
swap(t->uports.p16.dst, t->uports.p16.src);
} }
return 0; return 0;
@ -98,15 +111,19 @@ hmark_ct_set_htuple(const struct sk_buff *skb, struct hmark_tuple *t,
#endif #endif
} }
/* This hash function is endian independent, to ensure consistent hashing if
* the cluster is composed of big and little endian systems. */
static inline u32 static inline u32
hmark_hash(struct hmark_tuple *t, const struct xt_hmark_info *info) hmark_hash(struct hmark_tuple *t, const struct xt_hmark_info *info)
{ {
u32 hash; u32 hash;
u32 src = ntohl(t->src);
u32 dst = ntohl(t->dst);
if (t->dst < t->src) if (dst < src)
swap(t->src, t->dst); swap(src, dst);
hash = jhash_3words(t->src, t->dst, t->uports.v32, info->hashrnd); hash = jhash_3words(src, dst, t->uports.v32, info->hashrnd);
hash = hash ^ (t->proto & info->proto_mask); hash = hash ^ (t->proto & info->proto_mask);
return (((u64)hash * info->hmodulus) >> 32) + info->hoffset; return (((u64)hash * info->hmodulus) >> 32) + info->hoffset;
@ -126,11 +143,7 @@ hmark_set_tuple_ports(const struct sk_buff *skb, unsigned int nhoff,
if (skb_copy_bits(skb, nhoff, &t->uports, sizeof(t->uports)) < 0) if (skb_copy_bits(skb, nhoff, &t->uports, sizeof(t->uports)) < 0)
return; return;
t->uports.v32 = (t->uports.v32 & info->port_mask.v32) | hmark_swap_ports(&t->uports, info);
info->port_set.v32;
if (t->uports.p16.dst < t->uports.p16.src)
swap(t->uports.p16.dst, t->uports.p16.src);
} }
#if IS_ENABLED(CONFIG_IP6_NF_IPTABLES) #if IS_ENABLED(CONFIG_IP6_NF_IPTABLES)
@ -178,8 +191,8 @@ hmark_pkt_set_htuple_ipv6(const struct sk_buff *skb, struct hmark_tuple *t,
return -1; return -1;
} }
noicmp: noicmp:
t->src = hmark_addr6_mask(ip6->saddr.s6_addr32, info->src_mask.all); t->src = hmark_addr6_mask(ip6->saddr.s6_addr32, info->src_mask.ip6);
t->dst = hmark_addr6_mask(ip6->daddr.s6_addr32, info->dst_mask.all); t->dst = hmark_addr6_mask(ip6->daddr.s6_addr32, info->dst_mask.ip6);
if (info->flags & XT_HMARK_FLAG(XT_HMARK_METHOD_L3)) if (info->flags & XT_HMARK_FLAG(XT_HMARK_METHOD_L3))
return 0; return 0;
@ -255,11 +268,8 @@ hmark_pkt_set_htuple_ipv4(const struct sk_buff *skb, struct hmark_tuple *t,
} }
} }
t->src = (__force u32) ip->saddr; t->src = ip->saddr & info->src_mask.ip;
t->dst = (__force u32) ip->daddr; t->dst = ip->daddr & info->dst_mask.ip;
t->src &= info->src_mask.ip;
t->dst &= info->dst_mask.ip;
if (info->flags & XT_HMARK_FLAG(XT_HMARK_METHOD_L3)) if (info->flags & XT_HMARK_FLAG(XT_HMARK_METHOD_L3))
return 0; return 0;