Move emscripten_get_heap_size to native code

Author: sbc100

src/library.js

@@ -437,10 +437,6 @@ LibraryManager.library = {
     return -1;
   },
 
-  emscripten_get_heap_size: function() {
-    return HEAPU8.length;
-  },
-
 #if ABORTING_MALLOC
   $abortOnCannotGrowMemory: function(requestedSize) {
 #if ASSERTIONS
@@ -487,7 +483,7 @@ LibraryManager.library = {
   },
 #endif // ~TEST_MEMORY_GROWTH_FAILS
 
-  emscripten_resize_heap__deps: ['emscripten_get_heap_size'
+  __emscripten_resize_heap__deps: [
 #if ASSERTIONS == 2
   , 'emscripten_get_now'
 #endif
@@ -498,7 +494,7 @@ LibraryManager.library = {
   , '$emscripten_realloc_buffer'
 #endif
   ],
-  emscripten_resize_heap: function(requestedSize) {
+  __emscripten_resize_heap: function(oldSize, requestedSize) {
 #if CAN_ADDRESS_2GB
     requestedSize = requestedSize >>> 0;
 #endif
@@ -509,7 +505,6 @@ LibraryManager.library = {
     return false; // malloc will report failure
 #endif // ABORTING_MALLOC
 #else // ALLOW_MEMORY_GROWTH == 0
-    var oldSize = _emscripten_get_heap_size();
     // With pthreads, races can happen (another thread might increase the size in between), so return a failure, and let the caller retry.
 #if USE_PTHREADS
     if (requestedSize <= oldSize) {

system/include/emscripten/heap.h

@@ -7,6 +7,8 @@
 
 #pragma once
 
+#include <stddef.h>
+#include <stdint.h>
 #include <emscripten/emscripten.h>
 
 #define WASM_PAGE_SIZE 65536

system/lib/libc/emscripten_get_heap_size.c

@@ -0,0 +1,12 @@
+/*
+ * Copyright 2021 The Emscripten Authors.  All rights reserved.
+ * Emscripten is available under two separate licenses, the MIT license and the
+ * University of Illinois/NCSA Open Source License.  Both these licenses can be
+ * found in the LICENSE file.
+ */
+
+#include <emscripten/heap.h>
+
+size_t emscripten_get_heap_size() {
+  return __builtin_wasm_memory_size(0) * WASM_PAGE_SIZE;
+}

system/lib/libc/emscripten_resize_heap.c

@@ -0,0 +1,17 @@
+/*
+ * Copyright 2021 The Emscripten Authors.  All rights reserved.
+ * Emscripten is available under two separate licenses, the MIT license and the
+ * University of Illinois/NCSA Open Source License.  Both these licenses can be
+ * found in the LICENSE file.
+ */
+
+#include <emscripten/heap.h>
+
+int __emscripten_resize_heap(size_t requested_size, size_t current_size) EM_IMPORT(__emscripten_resize_heap);
+
+// This is just a small wrapper the JavaScript function `__emscripten_resize_heap`.
+// We pass in the current size to avoid reverse dependency on
+// 'emscripten_get_heap_size`
+int emscripten_resize_heap(size_t requested_size) {
+  return __emscripten_resize_heap(emscripten_get_heap_size(), requested_size);
+}

system/lib/standalone/standalone.c

@@ -117,9 +117,8 @@ void *emscripten_memcpy_big(void *restrict dest, const void *restrict src, size_
 
 extern void emscripten_notify_memory_growth(size_t memory_index);
 
-int emscripten_resize_heap(size_t size) {
+int __emscripten_resize_heap(size_t size, size_t old_size) {
 #ifdef __EMSCRIPTEN_MEMORY_GROWTH__
-  size_t old_size = __builtin_wasm_memory_size(0) * WASM_PAGE_SIZE;
   assert(old_size < size);
   ssize_t diff = (size - old_size + WASM_PAGE_SIZE - 1) / WASM_PAGE_SIZE;
   size_t result = __builtin_wasm_memory_grow(0, diff);

tests/other/metadce/hello_libcxx_O2.imports

@@ -1,6 +1,6 @@
+env.__emscripten_resize_heap
 env.abort
 env.emscripten_memcpy_big
-env.emscripten_resize_heap
 env.setTempRet0
 env.strftime_l
 wasi_snapshot_preview1.environ_get

tests/other/metadce/hello_libcxx_O2.sent

@@ -1,7 +1,7 @@
 __cxa_atexit
+__emscripten_resize_heap
 abort
 emscripten_memcpy_big
-emscripten_resize_heap
 environ_get
 environ_sizes_get
 fd_close

tests/other/metadce/hello_libcxx_O2_fexceptions.imports

@@ -7,10 +7,10 @@ env.__cxa_free_exception
 env.__cxa_rethrow
 env.__cxa_throw
 env.__cxa_uncaught_exceptions
+env.__emscripten_resize_heap
 env.__resumeException
 env.abort
 env.emscripten_memcpy_big
-env.emscripten_resize_heap
 env.getTempRet0
 env.invoke_diii
 env.invoke_fiii

tests/other/metadce/hello_libcxx_O2_fexceptions.sent

@@ -8,10 +8,10 @@ __cxa_free_exception
 __cxa_rethrow
 __cxa_throw
 __cxa_uncaught_exceptions
+__emscripten_resize_heap
 __resumeException
 abort
 emscripten_memcpy_big
-emscripten_resize_heap
 environ_get
 environ_sizes_get
 fd_close

tests/other/metadce/hello_libcxx_O2_fexceptions_DEMANGLE_SUPPORT.imports

@@ -7,10 +7,10 @@ env.__cxa_free_exception
 env.__cxa_rethrow
 env.__cxa_throw
 env.__cxa_uncaught_exceptions
+env.__emscripten_resize_heap
 env.__resumeException
 env.abort
 env.emscripten_memcpy_big
-env.emscripten_resize_heap
 env.getTempRet0
 env.invoke_diii
 env.invoke_fiii

tests/other/metadce/hello_libcxx_O2_fexceptions_DEMANGLE_SUPPORT.sent

@@ -8,10 +8,10 @@ __cxa_free_exception
 __cxa_rethrow
 __cxa_throw
 __cxa_uncaught_exceptions
+__emscripten_resize_heap
 __resumeException
 abort
 emscripten_memcpy_big
-emscripten_resize_heap
 environ_get
 environ_sizes_get
 fd_close

tests/other/metadce/hello_world_O3_MAIN_MODULE_2.imports

@@ -1,10 +1,10 @@
 GOT.mem.__heap_base
+env.__emscripten_resize_heap
 env.__indirect_function_table
 env.__memory_base
 env.__stack_pointer
 env.__table_base
 env.emscripten_memcpy_big
-env.emscripten_resize_heap
 env.memory
 env.setTempRet0
 wasi_snapshot_preview1.fd_write

tests/other/metadce/hello_world_O3_MAIN_MODULE_2.sent

@@ -1,9 +1,9 @@
+__emscripten_resize_heap
 __indirect_function_table
 __memory_base
 __stack_pointer
 __table_base
 emscripten_memcpy_big
-emscripten_resize_heap
 fd_write
 memory
 setTempRet0

tests/other/metadce/mem_O3_ALLOW_MEMORY_GROWTH_STANDALONE_WASM.funcs

@@ -1,6 +1,6 @@
+$__emscripten_resize_heap
 $__main_void
 $__wasm_call_ctors
 $_start
 $dlmalloc
-$emscripten_resize_heap
 $sbrk

tests/other/test_split_main_module.out

@@ -1,3 +1,3 @@
-wrote profile of 232 bytes (allocated 232 bytes)
+wrote profile of 240 bytes (allocated 240 bytes)
 Hello from main!
 Hello from lib!

tests/other/test_split_module.out

@@ -1,2 +1,2 @@
-wrote profile of 224 bytes (allocated 224 bytes)
+wrote profile of 232 bytes (allocated 232 bytes)
 Hello! answer: 42

tests/test_other.py

@@ -9981,6 +9981,7 @@ def test_split_module(self):
     self.assertNotIn('profile', result)
     self.assertIn('Hello! answer: 42', result)
 
+  @disabled('https://github.com/WebAssembly/binaryen/issues/3701')
   def test_split_main_module(self):
     initialTableSize = 16
 

tools/system_libs.py

@@ -736,7 +736,13 @@ def get_files(self):
 
     libc_files += files_in_path(
         path_components=['system', 'lib', 'libc'],
-        filenames=['extras.c', 'wasi-helpers.c', 'emscripten_pthread.c'])
+        filenames=[
+          'extras.c',
+          'wasi-helpers.c',
+          'emscripten_pthread.c',
+          'emscripten_get_heap_size.c',
+          'emscripten_resize_heap.c',
+        ])
 
     libc_files += files_in_path(
         path_components=['system', 'lib', 'pthread'],