of/address: introduce of_address_count() helper

Introduce of_address_count() helper to count the IO resources
instead of open-coding it.

Signed-off-by: Yang Yingliang <yangyingliang@huawei.com>
Link: https://lore.kernel.org/r/20221008115617.3583890-2-yangyingliang@huawei.com
Signed-off-by: Rob Herring <robh@kernel.org>

diff --git a/include/linux/of_address.h b/include/linux/of_address.h
index 45598db..265f26e 100644 (file)
--- a/include/linux/of_address.h
+++ b/include/linux/of_address.h
@@ -154,4 +154,15 @@ static inline const __be32 *of_get_pci_address(struct device_node *dev, int bar_
        return __of_get_address(dev, -1, bar_no, size, flags);
 }
 
+static inline int of_address_count(struct device_node *np)
+{
+       struct resource res;
+       int count = 0;
+
+       while (of_address_to_resource(np, count, &res) == 0)
+               count++;
+
+       return count;
+}
+
 #endif /* __OF_ADDRESS_H */