Commit 50e0e9b1 authored by Florian Westphal's avatar Florian Westphal Committed by Pablo Neira Ayuso

netfilter: connlimit: make same_source_net signed

currently returns 1 if they're the same.  Make it work like mem/strcmp
so it can be used as rbtree search function.
Reviewed-by: default avatarJesper Dangaard Brouer <brouer@redhat.com>
Signed-off-by: default avatarFlorian Westphal <fw@strlen.de>
Signed-off-by: default avatarPablo Neira Ayuso <pablo@netfilter.org>
parent 1442e750
...@@ -78,13 +78,14 @@ static inline bool already_closed(const struct nf_conn *conn) ...@@ -78,13 +78,14 @@ static inline bool already_closed(const struct nf_conn *conn)
return 0; return 0;
} }
static inline unsigned int static int
same_source_net(const union nf_inet_addr *addr, same_source_net(const union nf_inet_addr *addr,
const union nf_inet_addr *mask, const union nf_inet_addr *mask,
const union nf_inet_addr *u3, u_int8_t family) const union nf_inet_addr *u3, u_int8_t family)
{ {
if (family == NFPROTO_IPV4) { if (family == NFPROTO_IPV4) {
return (addr->ip & mask->ip) == (u3->ip & mask->ip); return ntohl(addr->ip & mask->ip) -
ntohl(u3->ip & mask->ip);
} else { } else {
union nf_inet_addr lh, rh; union nf_inet_addr lh, rh;
unsigned int i; unsigned int i;
...@@ -94,7 +95,7 @@ same_source_net(const union nf_inet_addr *addr, ...@@ -94,7 +95,7 @@ same_source_net(const union nf_inet_addr *addr,
rh.ip6[i] = u3->ip6[i] & mask->ip6[i]; rh.ip6[i] = u3->ip6[i] & mask->ip6[i];
} }
return memcmp(&lh.ip6, &rh.ip6, sizeof(lh.ip6)) == 0; return memcmp(&lh.ip6, &rh.ip6, sizeof(lh.ip6));
} }
} }
...@@ -143,7 +144,7 @@ static int count_hlist(struct net *net, ...@@ -143,7 +144,7 @@ static int count_hlist(struct net *net,
continue; continue;
} }
if (same_source_net(addr, mask, &conn->addr, family)) if (same_source_net(addr, mask, &conn->addr, family) == 0)
/* same source network -> be counted! */ /* same source network -> be counted! */
++matches; ++matches;
nf_ct_put(found_ct); nf_ct_put(found_ct);
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment