1 files changed, 10 insertions, 4 deletions

diff --git a/net/sched/cls_flow.c b/net/sched/cls_flow.c
index b4359924846c..76bc3a20ffdb 100644
--- a/net/sched/cls_flow.c
+++ b/net/sched/cls_flow.c
@@ -68,15 +68,21 @@ static inline u32 addr_fold(void *addr)
 
 static u32 flow_get_src(const struct sk_buff *skb, const struct flow_keys *flow)
 {
-        if (flow->addrs.src)
-                return ntohl(flow->addrs.src);
+        __be32 src = flow_get_u32_src(flow);
+
+        if (src)
+                return ntohl(src);
+
         return addr_fold(skb->sk);
 }
 
 static u32 flow_get_dst(const struct sk_buff *skb, const struct flow_keys *flow)
 {
-        if (flow->addrs.dst)
-                return ntohl(flow->addrs.dst);
+        __be32 dst = flow_get_u32_dst(flow);
+
+        if (dst)
+                return ntohl(dst);
+
         return addr_fold(skb_dst(skb)) ^ (__force u16) tc_skb_protocol(skb);
 }