1 files changed, 16 insertions, 0 deletions
diff --git a/src/include/kernel/block.H b/src/include/kernel/block.H
index 94daf0b3a..2421f97f9 100644
--- a/src/include/kernel/block.H
+++ b/src/include/kernel/block.H
@@ -30,6 +30,8 @@
 #include <kernel/task.H>
 #include <kernel/vmmmgr.H>
 #include <kernel/blockmsghdlr.H>
+#include <kernel/msghandler.H>
+#include <kernel/ptmgr.H>
 
 class ShadowPTE;
 class Segment;
@@ -140,6 +142,20 @@ class Block
          */
         uint64_t findPhysicalAddress(uint64_t i_vaddr) const;
 
+        /**
+         * @brief Responsible for updating usage counts within the block [chain].
+         *
+         * @param[in] i_vaddr - Abbreviated Virtual Address of page
+         *                      (bottom 23 bits are zero)
+         * @param[in] i_stats - Usage statistics
+         *
+         * If the address is not within this block or a matching page cannot be
+         *  found, the block will attempt to make calls down the block-chain
+         *  if it exists.
+         */
+        void updateRefCount( uint64_t i_vaddr,
+			     PageTableManager::UsageStats_t i_stats );
+
         friend class Segment;
         friend class BaseSegment;
         friend class StackSegment;