1 files changed, 18 insertions, 0 deletions
diff --git a/src/import/generic/memory/lib/utils/mcbist/gen_mss_mcbist_address.H b/src/import/generic/memory/lib/utils/mcbist/gen_mss_mcbist_address.H
index b42c0e39b..37e35c48d 100644
--- a/src/import/generic/memory/lib/utils/mcbist/gen_mss_mcbist_address.H
+++ b/src/import/generic/memory/lib/utils/mcbist/gen_mss_mcbist_address.H
@@ -99,6 +99,15 @@ class address
         }
 
         ///
+        /// @brief Construct an address from the provided buffer
+        /// @param[in] i_address fapi2::buffer address
+        ///
+        address(const fapi2::buffer<uint64_t>& i_address):
+            iv_address(i_address)
+        {
+        }
+
+        ///
         /// @brief Conversion operator to uint64_t
         /// @warn Right-aligns the address
         ///
@@ -412,6 +421,15 @@ class address
             return get_field<BANK_GROUP>();
         }
 
+        ///
+        /// @brief Get the buffer object for bit operations
+        /// @return fapi2::buffer<uint64_t>&
+        ///
+        inline fapi2::buffer<uint64_t> get_address_buffer() const
+        {
+            return iv_address;
+        }
+
     private:
         // We use a fapi2 buffer as it has static compile-time support
         fapi2::buffer<uint64_t> iv_address;