1 files changed, 25 insertions, 0 deletions
diff --git a/linux-core/drm_memrange.c b/linux-core/drm_memrange.c
index 0ae03655..e1d2233b 100644
--- a/linux-core/drm_memrange.c
+++ b/linux-core/drm_memrange.c
@@ -273,6 +273,31 @@ int drm_memrange_init(struct drm_memrange * mm, unsigned long start, unsigned lo
 	return drm_memrange_create_tail_node(mm, start, size);
 }
 
+/**
+ * Walks the list of allocated memory ranges and calls the callback on
+ * one.
+ */
+int drm_memrange_for_each(struct drm_memrange *mm,
+			  int (*callback)(struct drm_memrange_node *node,
+					  void *data),
+			  void *data)
+{
+	struct list_head *list, *next;
+
+	list_for_each_safe(list, next, &mm->ml_entry) {
+		struct drm_memrange_node *cur;
+		int ret;
+
+		cur = list_entry(list, struct drm_memrange_node, ml_entry);
+
+		ret = callback(cur, data);
+		if (ret != 0)
+			return ret;
+	}
+
+	return 0;
+}
+
 EXPORT_SYMBOL(drm_memrange_init);
 
 void drm_memrange_takedown(struct drm_memrange * mm)