Fix for issue where program calls setLEDs to change the number of LEDs

1 files changed, 7 insertions, 0 deletions

diff --git a/src/platforms/esp/32/clockless_rmt_esp32.cpp b/src/platforms/esp/32/clockless_rmt_esp32.cpp
index 1b3b7d3f..c7691f69 100644
--- a/src/platforms/esp/32/clockless_rmt_esp32.cpp
+++ b/src/platforms/esp/32/clockless_rmt_esp32.cpp
@@ -84,10 +84,17 @@ ESP32RMTController::ESP32RMTController(int DATA_PIN, int T1, int T2, int T3, int
 //    the PixelController object until show is called.
 uint8_t * ESP32RMTController::getPixelBuffer(int size_in_bytes)
 {
+    // -- Free the old buffer if it will be too small
+    if (mPixelData != 0 and mSize < size_in_bytes) {
+        free(mPixelData);
+        mPixelData = 0;
+    }
+
     if (mPixelData == 0) {
         mSize = size_in_bytes;
         mPixelData = (uint8_t *) malloc(mSize);
     }
+
     return mPixelData;
 }