mtd: ofpart: use for_each_child_of_node() macro

Use for_each_child_of_node() macro instead of open coding it.

Signed-off-by: Wei Yongjun <yongjun_wei@trendmicro.com.cn>
Acked-by: Huang Shijie <b32955@freescale.com>
Signed-off-by: David Woodhouse <David.Woodhouse@intel.com>

1 files changed, 2 insertions, 4 deletions

diff --git a/drivers/mtd/ofpart.c b/drivers/mtd/ofpart.c
index 7843a4491217..d64f8c30945f 100644
--- a/drivers/mtd/ofpart.c
+++ b/drivers/mtd/ofpart.c
@@ -43,9 +43,8 @@ static int parse_ofpart_partitions(struct mtd_info *master,
                 return 0;
 
         /* First count the subnodes */
-        pp = NULL;
         nr_parts = 0;
-        while ((pp = of_get_next_child(node, pp))) {
+        for_each_child_of_node(node,  pp) {
                 if (node_has_compatible(pp))
                         continue;
 
@@ -59,9 +58,8 @@ static int parse_ofpart_partitions(struct mtd_info *master,
         if (!*pparts)
                 return -ENOMEM;
 
-        pp = NULL;
         i = 0;
-        while ((pp = of_get_next_child(node, pp))) {
+        for_each_child_of_node(node,  pp) {
                 const __be32 *reg;
                 int len;
                 int a_cells, s_cells;