2011-12-06 Sebastian Huber <sebastian.huber@embedded-brains.de>

	* score/src/heapiterate.c, score/src/pheapiterate.c: New files.
	* score/Makefile.am: Reflect changes above.
	* score/include/rtems/score/heap.h: Declare _Heap_Iterate() and define
	Heap_Block_visitor.
	* score/include/rtems/score/protectedheap.h: Declare
	_Protected_heap_Iterate().
	* score/src/heapgetinfo.c: Use _Heap_Iterate().

3 files changed, 111 insertions, 24 deletions

diff --git a/cpukit/score/src/heapgetinfo.c b/cpukit/score/src/heapgetinfo.c
index bcb5a7f68d..872edc76ba 100644
--- a/cpukit/score/src/heapgetinfo.c
+++ b/cpukit/score/src/heapgetinfo.c
@@ -23,35 +23,32 @@
 
 #include <string.h>
 
-#include <rtems/system.h>
-#include <rtems/score/sysstate.h>
 #include <rtems/score/heap.h>
 
-void _Heap_Get_information(
-  Heap_Control            *the_heap,
-  Heap_Information_block  *the_info
+static bool _Heap_Get_information_visitor(
+  const Heap_Block *block __attribute__((unused)),
+  uintptr_t block_size,
+  bool block_is_used,
+  void *visitor_arg
 )
 {
-  Heap_Block *the_block = the_heap->first_block;
-  Heap_Block *const end = the_heap->last_block;
+  Heap_Information_block *info_block = visitor_arg;
+  Heap_Information *info = block_is_used ?
+    &info_block->Used : &info_block->Free;
 
-  memset(the_info, 0, sizeof(*the_info));
+  ++info->number;
+  info->total += block_size;
+  if ( info->largest < block_size )
+    info->largest = block_size;
 
-  while ( the_block != end ) {
-    uintptr_t const     the_size = _Heap_Block_size(the_block);
-    Heap_Block *const  next_block = _Heap_Block_at(the_block, the_size);
-    Heap_Information  *info;
-
-    if ( _Heap_Is_prev_used(next_block) )
-      info = &the_info->Used;
-    else
-      info = &the_info->Free;
-
-    info->number++;
-    info->total += the_size;
-    if ( info->largest < the_size )
-      info->largest = the_size;
+  return false;
+}
 
-    the_block = next_block;
-  }
+void _Heap_Get_information(
+  Heap_Control            *the_heap,
+  Heap_Information_block  *the_info
+)
+{
+  memset( the_info, 0, sizeof(*the_info) );
+  _Heap_Iterate( the_heap, _Heap_Get_information_visitor, the_info );
 }
diff --git a/cpukit/score/src/heapiterate.c b/cpukit/score/src/heapiterate.c
new file mode 100644
index 0000000000..b9100b9149
--- /dev/null
+++ b/cpukit/score/src/heapiterate.c
@@ -0,0 +1,50 @@
+/**
+ * @file
+ *
+ * @ingroup ScoreHeap
+ *
+ * @brief _Heap_Iterate() implementation.
+ */
+
+/*
+ * Copyright (c) 2011 embedded brains GmbH.  All rights reserved.
+ *
+ *  embedded brains GmbH
+ *  Obere Lagerstr. 30
+ *  82178 Puchheim
+ *  Germany
+ *  <rtems@embedded-brains.de>
+ *
+ * The license and distribution terms for this file may be
+ * found in the file LICENSE in this distribution or at
+ * http://www.rtems.com/license/LICENSE.
+ *
+ * $Id$
+ */
+
+#if HAVE_CONFIG_H
+  #include "config.h"
+#endif
+
+#include <rtems/score/heap.h>
+
+void _Heap_Iterate(
+  Heap_Control *heap,
+  Heap_Block_visitor visitor,
+  void *visitor_arg
+)
+{
+  Heap_Block *current = heap->first_block;
+  Heap_Block *end = heap->last_block;
+  bool stop = false;
+
+  while ( !stop && current != end ) {
+    uintptr_t size = _Heap_Block_size( current );
+    Heap_Block *next = _Heap_Block_at( current, size );
+    bool used = _Heap_Is_prev_used( next );
+
+    stop = (*visitor)( current, size, used, visitor_arg );
+
+    current = next;
+  }
+}
diff --git a/cpukit/score/src/pheapiterate.c b/cpukit/score/src/pheapiterate.c
new file mode 100644
index 0000000000..ec98f75ad2
--- /dev/null
+++ b/cpukit/score/src/pheapiterate.c
@@ -0,0 +1,40 @@
+/**
+ * @file
+ *
+ * @ingroup ScoreProtHeap
+ *
+ * @brief _Heap_Iterate() implementation.
+ */
+
+/*
+ * Copyright (c) 2011 embedded brains GmbH.  All rights reserved.
+ *
+ *  embedded brains GmbH
+ *  Obere Lagerstr. 30
+ *  82178 Puchheim
+ *  Germany
+ *  <rtems@embedded-brains.de>
+ *
+ * The license and distribution terms for this file may be
+ * found in the file LICENSE in this distribution or at
+ * http://www.rtems.com/license/LICENSE.
+ *
+ * $Id$
+ */
+
+#if HAVE_CONFIG_H
+  #include "config.h"
+#endif
+
+#include <rtems/score/protectedheap.h>
+
+void _Protected_heap_Iterate(
+  Heap_Control *heap,
+  Heap_Block_visitor visitor,
+  void *visitor_arg
+)
+{
+  _RTEMS_Lock_allocator();
+  _Heap_Iterate( heap, visitor, visitor_arg );
+  _RTEMS_Unlock_allocator();
+}