diff --git a/CMakeLists.txt b/CMakeLists.txt
index f1b64445..9055f135 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -574,6 +574,7 @@ CHECK_INCLUDE_FILE(sys/types.h LWS_HAVE_SYS_TYPES_H)
 CHECK_INCLUDE_FILE(unistd.h LWS_HAVE_UNISTD_H)
 CHECK_INCLUDE_FILE(vfork.h LWS_HAVE_VFORK_H)
 CHECK_INCLUDE_FILE(sys/capability.h LWS_HAVE_SYS_CAPABILITY_H)
+CHECK_INCLUDE_FILE(malloc.h LWS_HAVE_MALLOC_H)
 
 CHECK_LIBRARY_EXISTS(cap cap_set_flag "" LWS_HAVE_LIBCAP) 
 
diff --git a/cmake/lws_config.h.in b/cmake/lws_config.h.in
index f8a72cf1..d74ef7e7 100644
--- a/cmake/lws_config.h.in
+++ b/cmake/lws_config.h.in
@@ -150,6 +150,8 @@
 #cmakedefine LWS_WITH_ACME
 #cmakedefine LWS_WITH_SELFTESTS
 
+#cmakedefine LWS_HAVE_MALLOC_H
+
 /* OpenSSL various APIs */
 
 #cmakedefine LWS_HAVE_TLS_CLIENT_METHOD
diff --git a/lib/tls/mbedtls/wrapper/include/platform/ssl_port.h b/lib/tls/mbedtls/wrapper/include/platform/ssl_port.h
index c4c26618..74c76343 100644
--- a/lib/tls/mbedtls/wrapper/include/platform/ssl_port.h
+++ b/lib/tls/mbedtls/wrapper/include/platform/ssl_port.h
@@ -19,12 +19,11 @@
  extern "C" {
 #endif
 
-/*
-#include "esp_types.h"
-#include "esp_log.h"
-*/
 #include "string.h"
+#include "stdlib.h"
+#if defined(LWS_HAVE_MALLOC_H)
 #include "malloc.h"
+#endif
 
 void *ssl_mem_zalloc(size_t size);