Implement randomx_get_cache_memory function

ArweaveTeam · Dec 6, 2024 · 150f16c · 150f16c
1 parent 9c4edf9
commit 150f16c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/randomx.cpp b/src/randomx.cpp
@@ -138,6 +138,11 @@ extern "C" {
 		}
 	}
 
+	void *randomx_get_cache_memory(randomx_cache *cache) {
+		assert(cache != nullptr);
+		return cache->memory;
+	}
+
 	void randomx_release_cache(randomx_cache* cache) {
 		assert(cache != nullptr);
 		cache->dealloc(cache);
@@ -416,8 +421,6 @@ extern "C" {
 			assert(blakeResult == 0);
 		}
 		machine->run(&tempHash);
-		unsigned char output[64];
-		machine->getFinalResult(output, RANDOMX_HASH_SIZE);
 
 #ifdef USE_CSR_INTRINSICS
 		_mm_setcsr(fpstate);

diff --git a/src/randomx.h b/src/randomx.h
@@ -119,6 +119,16 @@ RANDOMX_EXPORT randomx_cache *randomx_alloc_cache(randomx_flags flags);
 */
 RANDOMX_EXPORT void randomx_init_cache(randomx_cache *cache, const void *key, size_t keySize);
 
+/**
+ * Returns a pointer to the internal memory buffer of the cache structure. The size
+ * of the internal memory buffer is RANDOMX_ARGON_MEMORY * ArgonBlockSize.
+ *
+ * @param cache is a pointer to a previously allocated randomx_cache structure. Must not be NULL.
+ *
+ * @return Pointer to the internal memory buffer of the cache structure.
+*/
+RANDOMX_EXPORT void *randomx_get_cache_memory(randomx_cache *cache);
+
 /**
  * Releases all memory occupied by the randomx_cache structure.
  *