1 files changed, 3 insertions, 7 deletions

diff --git a/drivers/gpu/drm/drm_of.c b/drivers/gpu/drm/drm_of.c
index 16150a00c237..aaa130736bf8 100644
--- a/drivers/gpu/drm/drm_of.c
+++ b/drivers/gpu/drm/drm_of.c
@@ -43,14 +43,10 @@ static uint32_t drm_crtc_port_mask(struct drm_device *dev,
 uint32_t drm_of_find_possible_crtcs(struct drm_device *dev,
                                     struct device_node *port)
 {
-        struct device_node *remote_port, *ep = NULL;
+        struct device_node *remote_port, *ep;
         uint32_t possible_crtcs = 0;
 
-        do {
-                ep = of_graph_get_next_endpoint(port, ep);
-                if (!ep)
-                        break;
-
+        for_each_endpoint_of_node(port, ep) {
                 remote_port = of_graph_get_remote_port(ep);
                 if (!remote_port) {
                         of_node_put(ep);
@@ -60,7 +56,7 @@ uint32_t drm_of_find_possible_crtcs(struct drm_device *dev,
                 possible_crtcs |= drm_crtc_port_mask(dev, remote_port);
 
                 of_node_put(remote_port);
-        } while (1);
+        }
 
         return possible_crtcs;
 }