From 595be5a143ddd754a28958596af542109822c7a8 Mon Sep 17 00:00:00 2001
From: zhangyanjiao <zhangyanjiao@espressif.com>
Date: Wed, 8 Jul 2020 20:44:42 +0800
Subject: [PATCH] esp_wifi: Add API to get available internal heap size

---
 components/esp32/esp_adapter.c        |  2 +-
 components/esp32/include/esp_system.h | 10 ++++++++++
 components/esp32/system_api.c         |  5 +++++
 3 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/components/esp32/esp_adapter.c b/components/esp32/esp_adapter.c
index 18c4b0181..25e81d201 100644
--- a/components/esp32/esp_adapter.c
+++ b/components/esp32/esp_adapter.c
@@ -551,7 +551,7 @@ wifi_osi_funcs_t g_wifi_osi_funcs = {
     ._task_get_max_priority = task_get_max_priority_wrapper,
     ._malloc = malloc,
     ._free = free,
-    ._get_free_heap_size = esp_get_free_heap_size,
+    ._get_free_heap_size = esp_get_free_internal_heap_size,
     ._rand = esp_random,
     ._dport_access_stall_other_cpu_start_wrap = esp_dport_access_stall_other_cpu_start_wrap,
     ._dport_access_stall_other_cpu_end_wrap = esp_dport_access_stall_other_cpu_end_wrap,
diff --git a/components/esp32/include/esp_system.h b/components/esp32/include/esp_system.h
index 1a51999d3..845ba6076 100644
--- a/components/esp32/include/esp_system.h
+++ b/components/esp32/include/esp_system.h
@@ -130,6 +130,16 @@ uint32_t system_get_time(void)  __attribute__ ((deprecated));
   */
 uint32_t esp_get_free_heap_size(void);
 
+/**
+ * @brief  Get the size of available internal heap.
+ *
+ * Note that the returned value may be larger than the maximum contiguous block
+ * which can be allocated.
+ *
+ * @return Available internal heap size, in bytes.
+ */
+uint32_t esp_get_free_internal_heap_size(void);
+
 /** @cond */
 /**
   * @brief  Get the size of available heap.
diff --git a/components/esp32/system_api.c b/components/esp32/system_api.c
index 435a2d748..416b890e7 100644
--- a/components/esp32/system_api.c
+++ b/components/esp32/system_api.c
@@ -363,6 +363,11 @@ uint32_t esp_get_free_heap_size( void )
     return heap_caps_get_free_size( MALLOC_CAP_DEFAULT );
 }
 
+uint32_t esp_get_free_internal_heap_size( void )
+{
+    return heap_caps_get_free_size( MALLOC_CAP_8BIT | MALLOC_CAP_DMA | MALLOC_CAP_INTERNAL );
+}
+
 uint32_t esp_get_minimum_free_heap_size( void )
 {
     return heap_caps_get_minimum_free_size( MALLOC_CAP_DEFAULT );