mctp: Add helper for address match checking

Currently, we have a couple of paths that check that an EID matches, or
the match value is MCTP_ADDR_ANY.

Rather than open coding this, add a little helper.

Signed-off-by: Jeremy Kerr <jk@codeconstruct.com.au>
Signed-off-by: David S. Miller <davem@davemloft.net>

diff --git a/include/net/mctp.h b/include/net/mctp.h
index 7e35ec7..706d329 100644 (file)
--- a/include/net/mctp.h
+++ b/include/net/mctp.h
@@ -45,6 +45,11 @@ static inline bool mctp_address_ok(mctp_eid_t eid)
        return eid >= 8 && eid < 255;
 }
 
+static inline bool mctp_address_matches(mctp_eid_t match, mctp_eid_t eid)
+{
+       return match == eid || match == MCTP_ADDR_ANY;
+}
+
 static inline struct mctp_hdr *mctp_hdr(struct sk_buff *skb)
 {
        return (struct mctp_hdr *)skb_network_header(skb);

diff --git a/net/mctp/route.c b/net/mctp/route.c
index 8d9f4ff..654467a 100644 (file)
--- a/net/mctp/route.c
+++ b/net/mctp/route.c
@@ -64,8 +64,7 @@ static struct mctp_sock *mctp_lookup_bind(struct net *net, struct sk_buff *skb)
                if (msk->bind_type != type)
                        continue;
 
-               if (msk->bind_addr != MCTP_ADDR_ANY &&
-                   msk->bind_addr != mh->dest)
+               if (!mctp_address_matches(msk->bind_addr, mh->dest))
                        continue;
 
                return msk;
@@ -616,9 +615,8 @@ static struct mctp_sk_key *mctp_alloc_local_tag(struct mctp_sock *msk,
                if (tmp->tag & MCTP_HDR_FLAG_TO)
                        continue;
 
-               if (!((tmp->peer_addr == daddr ||
-                      tmp->peer_addr == MCTP_ADDR_ANY) &&
-                      tmp->local_addr == saddr))
+               if (!(mctp_address_matches(tmp->peer_addr, daddr) &&
+                     tmp->local_addr == saddr))
                        continue;
 
                spin_lock(&tmp->lock);