]> granicus.if.org Git - esp-idf/commitdiff
esp32: Add core dump printing to UART feature
authorAlexey Gerenkov <alexey@espressif.com>
Tue, 3 Jan 2017 11:16:41 +0000 (14:16 +0300)
committerAlexey Gerenkov <alexey@espressif.com>
Wed, 11 Jan 2017 17:51:28 +0000 (20:51 +0300)
components/esp32/core_dump.c
components/esp32/cpu_start.c
components/esp32/include/esp_panic.h
components/espcoredump/espcoredump.py
components/freertos/include/freertos/FreeRTOSConfig.h

index 165ec7419271d551f04038f7aa0adee0c6e2742f..eaab1871c303ee97fcd103990097fbe6df2d6384 100644 (file)
@@ -11,6 +11,7 @@
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
+#include <string.h>
 #include "freertos/FreeRTOS.h"
 #include "freertos/task.h"
 
 #include "esp_panic.h"
 #include "esp_partition.h"
 
-#ifdef ESP_PLATFORM
-// Uncomment this line to force output from this module
-#define LOG_LOCAL_LEVEL ESP_LOG_DEBUG
+#if CONFIG_ESP32_ENABLE_COREDUMP_TO_FLASH || CONFIG_ESP32_ENABLE_COREDUMP_TO_UART
 #include "esp_log.h"
-static const char* TAG = "esp_core_dump_init";
-#else
-#define ESP_LOGD(...)
-#endif
+const static char *TAG = "esp_core_dump";
 
 // TODO: allow user to set this in menuconfig or get tasks iteratively
 #define COREDUMP_MAX_TASKS_NUM    32
 
+typedef esp_err_t (*esp_core_dump_write_prepare_t)(void *priv, uint32_t *data_len);
+typedef esp_err_t (*esp_core_dump_write_start_t)(void *priv);
+typedef esp_err_t (*esp_core_dump_write_end_t)(void *priv);
+typedef esp_err_t (*esp_core_dump_flash_write_data_t)(void *priv, void * data, uint32_t data_len);
+
+typedef struct _core_dump_write_config_t
+{
+    esp_core_dump_write_prepare_t       prepare;
+    esp_core_dump_write_start_t         start;
+    esp_core_dump_write_end_t           end;
+    esp_core_dump_flash_write_data_t    write;
+    void *                              priv;
+
+} core_dump_write_config_t;
+
+static void esp_core_dump_write(XtExcFrame *frame, core_dump_write_config_t *write_cfg, int verb)
+{
+    union
+    {
+        uint8_t    data8[12];
+        uint32_t   data32[3];
+    } rom_data;
+    //const esp_partition_t *core_part;
+    esp_err_t err;
+    TaskSnapshot_t tasks[COREDUMP_MAX_TASKS_NUM];
+    UBaseType_t tcb_sz, task_num;
+    uint32_t data_len = 0, i, len;
+    //size_t off;
+
+    task_num = uxTaskGetSnapshotAll(tasks, COREDUMP_MAX_TASKS_NUM, &tcb_sz);
+    // take TCB padding into account, actual TCB size will be stored in header
+    if (tcb_sz % sizeof(uint32_t))
+        len = (tcb_sz / sizeof(uint32_t) + 1) * sizeof(uint32_t);
+    else
+        len = tcb_sz;
+    // header + tasknum*(tcb + stack start/end + tcb addr)
+    data_len = 3*sizeof(uint32_t) + task_num*(len + 2*sizeof(uint32_t) + sizeof(uint32_t *));
+    for (i = 0; i < task_num; i++) {
+        if (tasks[i].pxTCB == xTaskGetCurrentTaskHandle()) {
+            // set correct stack top for current task
+            tasks[i].pxTopOfStack = (StackType_t *)frame;
+            if (verb) {
+                esp_panicPutStr("Current task PC/A0/SP ");
+                esp_panicPutHex(frame->pc);
+                esp_panicPutStr(" ");
+                esp_panicPutHex(frame->a0);
+                esp_panicPutStr(" ");
+                esp_panicPutHex(frame->a1);
+                esp_panicPutStr("\r\n");
+            }
+        }
+#if( portSTACK_GROWTH < 0 )
+        len = (uint32_t)tasks[i].pxEndOfStack - (uint32_t)tasks[i].pxTopOfStack;
+#else
+        len = (uint32_t)tasks[i].pxTopOfStack - (uint32_t)tasks[i].pxEndOfStack;
+#endif
+        if (verb) {
+            esp_panicPutStr("stack len = ");
+            esp_panicPutHex(len);
+            esp_panicPutStr(" ");
+            esp_panicPutHex((int)tasks[i].pxTopOfStack);
+            esp_panicPutStr(" ");
+            esp_panicPutHex((int)tasks[i].pxEndOfStack);
+            esp_panicPutStr("\r\n");
+        }
+        // take stack padding into account
+        if (len % sizeof(uint32_t))
+            len = (len / sizeof(uint32_t) + 1) * sizeof(uint32_t);
+        data_len += len;
+    }
+
+    // prepare write
+    if (write_cfg->prepare) {
+        err = write_cfg->prepare(write_cfg->priv, &data_len);
+        if (err != ESP_OK) {
+            esp_panicPutStr("ERROR: Failed to prepare core dump ");
+            esp_panicPutHex(err);
+            esp_panicPutStr("!\r\n");
+            return;
+        }
+    }
+
+    if (verb) {
+        esp_panicPutStr("Core dump len =");
+        esp_panicPutHex(data_len);
+        esp_panicPutStr("\r\n");
+    }
+
+    // write start marker
+    if (write_cfg->start) {
+        err = write_cfg->start(write_cfg->priv);
+        if (err != ESP_OK) {
+            esp_panicPutStr("ERROR: Failed to start core dump ");
+            esp_panicPutHex(err);
+            esp_panicPutStr("!\r\n");
+            return;
+        }
+    }
+
+    // write header
+    rom_data.data32[0] = data_len;
+    rom_data.data32[1] = task_num;
+    rom_data.data32[2] = tcb_sz;
+    err = write_cfg->write(write_cfg->priv, &rom_data, 3*sizeof(uint32_t));
+    if (err != ESP_OK) {
+        esp_panicPutStr("ERROR: Failed to write core dump header ");
+        esp_panicPutHex(err);
+        esp_panicPutStr("!\r\n");
+        return;
+    }
+
+    // write tasks
+    for (i = 0; i < task_num; i++) {
+        if (verb) {
+            esp_panicPutStr("Dump task ");
+            esp_panicPutHex((int)tasks[i].pxTCB);
+            esp_panicPutStr("\r\n");
+        }
+        // save TCB address, stack base and stack top addr
+        rom_data.data32[0] = (uint32_t)tasks[i].pxTCB;
+        rom_data.data32[1] = (uint32_t)tasks[i].pxTopOfStack;
+        rom_data.data32[2] = (uint32_t)tasks[i].pxEndOfStack;
+        err = write_cfg->write(write_cfg->priv, &rom_data, 3*sizeof(uint32_t));
+        if (err != ESP_OK) {
+            esp_panicPutStr("ERROR: Failed to write task header ");
+            esp_panicPutHex(err);
+            esp_panicPutStr("!\r\n");
+            return;
+        }
+        // save TCB
+        err = write_cfg->write(write_cfg->priv, tasks[i].pxTCB, tcb_sz);
+        if (err != ESP_OK) {
+            esp_panicPutStr("ERROR: Failed to write task header ");
+            esp_panicPutHex(err);
+            esp_panicPutStr("!\r\n");
+            return;
+        }
+        // save task stack
+        /*int k;
+        for (k = 0; k < 8*4; k++) {
+            esp_panicPutStr("stack[");
+            esp_panicPutDec(k);
+            esp_panicPutStr("] = ");
+            esp_panicPutHex(((uint8_t *)tasks[i].pxTopOfStack)[k]);
+            esp_panicPutStr("\r\n");
+        }*/
+        err = write_cfg->write(write_cfg->priv,
+#if( portSTACK_GROWTH < 0 )
+                tasks[i].pxTopOfStack,
+                (uint32_t)tasks[i].pxEndOfStack - (uint32_t)tasks[i].pxTopOfStack
+#else
+                tasks[i].pxEndOfStack,
+                (uint32_t)tasks[i].pxTopOfStack - (uint32_t)tasks[i].pxEndOfStack
+#endif
+                );
+        if (err != ESP_OK) {
+            esp_panicPutStr("ERROR: Failed to write task header ");
+            esp_panicPutHex(err);
+            esp_panicPutStr("!\r\n");
+            return;
+        }
+    }
+
+    // write end marker
+    if (write_cfg->end) {
+        err = write_cfg->end(write_cfg->priv);
+        if (err != ESP_OK) {
+            esp_panicPutStr("ERROR: Failed to end core dump ");
+            esp_panicPutHex(err);
+            esp_panicPutStr("!\r\n");
+            return;
+        }
+    }
+}
+
 #if CONFIG_ESP32_ENABLE_COREDUMP_TO_FLASH
 
 // magic numbers to control core dump data consistency
 #define COREDUMP_FLASH_MAGIC_START    0xDEADBEEFUL
 #define COREDUMP_FLASH_MAGIC_END      0xACDCFEEDUL
 
+typedef struct _core_dump_write_flash_data_t
+{
+    uint32_t    off;
+} core_dump_write_flash_data_t;
+
 // core dump partition start
 static uint32_t s_core_part_start;
 // core dump partition size
@@ -79,6 +255,115 @@ static uint32_t esp_core_dump_write_flash_padded(size_t off, uint8_t *data, uint
     return data_len;
 }
 
+static esp_err_t esp_core_dump_flash_write_prepare(void *priv, uint32_t *data_len)
+{
+    esp_err_t err;
+    uint32_t sec_num;
+    core_dump_write_flash_data_t *wr_data = (core_dump_write_flash_data_t *)priv;
+
+    esp_panicPutStr("Core dump len1 = ");
+    esp_panicPutHex(*data_len);
+    esp_panicPutStr("\r\n");
+
+    // add space for 2 magics. TODO: change to CRC
+    *data_len += 2*sizeof(uint32_t);
+    if (*data_len > s_core_part_size) {
+        esp_panicPutStr("ERROR: Not enough space to save core dump!");
+        return ESP_ERR_NO_MEM;
+    }
+
+    esp_panicPutStr("Core dump len2 = ");
+    esp_panicPutHex(*data_len);
+    esp_panicPutStr("\r\n");
+
+    wr_data->off = 0;
+
+    sec_num = *data_len / SPI_FLASH_SEC_SIZE;
+    if (*data_len % SPI_FLASH_SEC_SIZE)
+        sec_num++;
+    err = spi_flash_erase_range_panic(s_core_part_start + 0, sec_num * SPI_FLASH_SEC_SIZE);
+    if (err != ESP_OK) {
+        esp_panicPutStr("ERROR: Failed to erase flash ");
+        esp_panicPutHex(err);
+        esp_panicPutStr("!\r\n");
+        return err;
+    }
+
+    return err;
+}
+
+static esp_err_t esp_core_dump_flash_write_word(core_dump_write_flash_data_t *wr_data, uint32_t word)
+{
+    esp_err_t err = ESP_OK;
+    uint32_t  data32 = word;
+
+    err = spi_flash_write_panic(s_core_part_start + wr_data->off, &data32, sizeof(uint32_t));
+    if (err != ESP_OK) {
+        esp_panicPutStr("Failed to write to flash ");
+        esp_panicPutHex(err);
+        esp_panicPutStr("!\r\n");
+        return err;
+    }
+    wr_data->off += sizeof(uint32_t);
+
+    return err;
+}
+
+static esp_err_t esp_core_dump_flash_write_start(void *priv)
+{
+    core_dump_write_flash_data_t *wr_data = (core_dump_write_flash_data_t *)priv;
+    // save magic 1
+    return esp_core_dump_flash_write_word(wr_data, COREDUMP_FLASH_MAGIC_START);
+}
+
+static esp_err_t esp_core_dump_flash_write_end(void *priv)
+{
+    core_dump_write_flash_data_t *wr_data = (core_dump_write_flash_data_t *)priv;
+    uint32_t i;
+    union
+    {
+        uint8_t    data8[16];
+        uint32_t   data32[4];
+    } rom_data;
+
+    // TEST READ START
+    esp_err_t err = spi_flash_read_panic(s_core_part_start + 0, &rom_data, sizeof(rom_data));
+    if (err != ESP_OK) {
+        esp_panicPutStr("ERROR: Failed to read flash ");
+        esp_panicPutHex(err);
+        esp_panicPutStr("!\r\n");
+        return err;
+    }
+    else {
+        esp_panicPutStr("Data from flash:\r\n");
+        for (i = 0; i < sizeof(rom_data)/sizeof(rom_data.data32[0]); i++) {
+            esp_panicPutHex(rom_data.data32[i]);
+            esp_panicPutStr("\r\n");
+        }
+//            rom_data[4] = 0;
+//            esp_panicPutStr(rom_data);
+//        esp_panicPutStr("\r\n");
+    }
+    // TEST READ END
+
+    // save magic 2
+    return esp_core_dump_flash_write_word(wr_data, COREDUMP_FLASH_MAGIC_END);
+}
+
+static esp_err_t esp_core_dump_flash_write_data(void *priv, void * data, uint32_t data_len)
+{
+    esp_err_t err = ESP_OK;
+    core_dump_write_flash_data_t *wr_data = (core_dump_write_flash_data_t *)priv;
+
+    uint32_t len = esp_core_dump_write_flash_padded(s_core_part_start + wr_data->off, data, data_len);
+    if (len != data_len)
+        return ESP_FAIL;
+
+    wr_data->off += len;
+
+    return err;
+}
+
 /*
  * |   MAGIC1   |
  * |  TOTAL_LEN |  TASKS_NUM | TCB_SIZE |
@@ -90,6 +375,19 @@ static uint32_t esp_core_dump_write_flash_padded(size_t off, uint8_t *data, uint
  */
 void esp_core_dump_to_flash(XtExcFrame *frame)
 {
+#if 1
+    core_dump_write_config_t wr_cfg;
+    core_dump_write_flash_data_t wr_data;
+
+    wr_cfg.prepare = esp_core_dump_flash_write_prepare;
+    wr_cfg.start = esp_core_dump_flash_write_start;
+    wr_cfg.end = esp_core_dump_flash_write_end;
+    wr_cfg.write = esp_core_dump_flash_write_data;
+    wr_cfg.priv = &wr_data;
+
+    esp_panicPutStr("Save core dump to flash...\r\n");
+    esp_core_dump_write(frame, &wr_cfg, 1);
+#else
     union
     {
         uint8_t    data8[16];
@@ -245,14 +543,150 @@ void esp_core_dump_to_flash(XtExcFrame *frame)
         esp_panicPutStr("!\r\n");
         return;
     }
-
-    esp_panicPutStr("Core dump has been saved to flash partition.\r\n");
+#endif
+    esp_panicPutStr("Core dump has been saved to flash.\r\n");
 }
 #endif
 
 #if CONFIG_ESP32_ENABLE_COREDUMP_TO_UART
+#if 0
+#define BASE64_ENCODE_BODY(_src, _src_len, _dst)                                \
+    do {                                                                        \
+        static const char *b64 =                                                \
+            "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; \
+        int i, j, a, b, c;                                                      \
+                                                                                \
+        for (i = j = 0; i < _src_len; i += 3) {                                  \
+            a = _src[i];                                                         \
+            b = i + 1 >= _src_len ? 0 : _src[i + 1];                              \
+            c = i + 2 >= _src_len ? 0 : _src[i + 2];                              \
+                                                                                \
+            /*BASE64_OUT(b64[a >> 2], _dst[j]);*/                                            \
+            _dst[j++] = b64[a >> 2]; \
+            /*BASE64_OUT(b64[((a & 3) << 4) | (b >> 4)], _dst[j]);*/                         \
+            _dst[j++] = b64[((a & 3) << 4) | (b >> 4)]; \
+            j++;                                                                \
+            if (i + 1 < _src_len) {                                              \
+                BASE64_OUT(b64[(b & 15) << 2 | (c >> 6)], _dst[j]);                      \
+                j++;                                                                \
+            }                                                                   \
+            if (i + 2 < _src_len) {                                              \
+                BASE64_OUT(b64[c & 63], _dst[j]);                                        \
+                j++;                                                                \
+            }                                                                   \
+        }                                                                       \
+                                                                                \
+        while (j % 4 != 0) {                                                    \
+            BASE64_OUT('=', _dst);                                                    \
+        }                                                                       \
+        BASE64_FLUSH(_dst)                                                          \
+    } while(0)
+
+#define BASE64_OUT(ch, _dst) \
+  do {                 \
+    _dst = (ch);   \
+  } while (0)
+
+#define BASE64_FLUSH(_dst) \
+  do {                 \
+    _dst = '\0';   \
+  } while (0)
+#endif
+static void esp_core_dump_b64_encode(const uint8_t *src, uint32_t src_len, uint8_t *dst) {
+//  BASE64_ENCODE_BODY(src, src_len, dst);
+    static const char *b64 =
+        "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
+    int i, j, a, b, c;
+
+    for (i = j = 0; i < src_len; i += 3) {
+        a = src[i];
+        b = i + 1 >= src_len ? 0 : src[i + 1];
+        c = i + 2 >= src_len ? 0 : src[i + 2];
+
+        dst[j++] = b64[a >> 2];
+        dst[j++] = b64[((a & 3) << 4) | (b >> 4)];
+        if (i + 1 < src_len) {
+            dst[j++] = b64[(b & 0x0F) << 2 | (c >> 6)];
+        }
+        if (i + 2 < src_len) {
+            dst[j++] = b64[c & 0x3F];
+        }
+    }
+    while (j % 4 != 0) {
+        dst[j++] = '=';
+    }
+    dst[j++] = '\0';
+}
+
+/*static esp_err_t esp_core_dump_uart_write_prepare(void *priv, uint32_t *data_len)
+{
+    esp_err_t err = ESP_OK;
+    return err;
+}*/
+
+static esp_err_t esp_core_dump_uart_write_start(void *priv)
+{
+//    core_dump_write_flash_data_t *wr_data = (core_dump_write_flash_data_t *)priv;
+    esp_err_t err = ESP_OK;
+    esp_panicPutStr("================= CORE DUMP START =================\r\n");
+    return err;
+}
+
+static esp_err_t esp_core_dump_uart_write_end(void *priv)
+{
+//    core_dump_write_flash_data_t *wr_data = (core_dump_write_flash_data_t *)priv;
+    esp_err_t err = ESP_OK;
+    esp_panicPutStr("================= CORE DUMP END =================\r\n");
+    return err;
+}
+
+static esp_err_t esp_core_dump_uart_write_data(void *priv, void * data, uint32_t data_len)
+{
+//    core_dump_write_flash_data_t *wr_data = (core_dump_write_flash_data_t *)priv;
+    esp_err_t err = ESP_OK;
+    char buf[64 + 4], *addr = data;
+    char *end = addr + data_len;
+
+//    esp_panicPutStr("CORE DUMP SEC: ");
+//    esp_panicPutDec(data_len);
+//    esp_panicPutStr("bytes\r\n");
+
+    while (addr < end) {
+        size_t len = end - addr;
+        if (len > 48) len = 48;
+        /* Copy to stack to avoid alignment restrictions. */
+        char *tmp = buf + (sizeof(buf) - len);
+        memcpy(tmp, addr, len);
+        esp_core_dump_b64_encode((const uint8_t *)tmp, len, (uint8_t *)buf);
+        addr += len;
+        esp_panicPutStr(buf);
+//        for (size_t i = 0; buf[i] != '\0'; i++) {
+//            panicPutChar(buf[i]);
+//        }
+        //if (addr % 96 == 0)
+        esp_panicPutStr("\r\n");
+        /* Feed the Cerberus. */
+//        TIMERG0.wdt_wprotect = TIMG_WDT_WKEY_VALUE;
+//        TIMERG0.wdt_feed = 1;
+    }
+
+    return err;
+}
+
 void esp_core_dump_to_uart(XtExcFrame *frame)
 {
+    core_dump_write_config_t wr_cfg;
+    //core_dump_write_flash_data_t wr_data;
+
+    wr_cfg.prepare = NULL;//esp_core_dump_uart_write_prepare;
+    wr_cfg.start = esp_core_dump_uart_write_start;
+    wr_cfg.end = esp_core_dump_uart_write_end;
+    wr_cfg.write = esp_core_dump_uart_write_data;
+    wr_cfg.priv = NULL;
+
+    esp_panicPutStr("Save core dump to flash...\r\n");
+    esp_core_dump_write(frame, &wr_cfg, 0);
+    esp_panicPutStr("Core dump has been written to uart.\r\n");
 }
 #endif
 
@@ -261,6 +695,7 @@ void esp_core_dump_init()
 #if CONFIG_ESP32_ENABLE_COREDUMP_TO_FLASH
     const esp_partition_t *core_part;
 
+    ESP_LOGI(TAG, "Init core dump to flash");
     core_part = esp_partition_find_first(ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_DATA_COREDUMP, NULL);
     if (!core_part) {
         ESP_LOGE(TAG, "No core dump partition found!");
@@ -271,6 +706,9 @@ void esp_core_dump_init()
     s_core_part_size = core_part->size;
 #endif
 #if CONFIG_ESP32_ENABLE_COREDUMP_TO_UART
+    ESP_LOGI(TAG, "Init core dump to UART");
 #endif
 }
 
+#endif
+
index 95d5c5e6a2af05379806ba18ab4c212ab8cab9ea..1b8a95b75c3daa2f0ed9b7b100678e4ac736b353 100644 (file)
@@ -215,7 +215,7 @@ void start_cpu0_default(void)
     }
 #endif
 
-#if CONFIG_ESP32_ENABLE_COREDUMP_TO_FLASH
+#if CONFIG_ESP32_ENABLE_COREDUMP_TO_FLASH || CONFIG_ESP32_ENABLE_COREDUMP_TO_UART
     esp_core_dump_init();
 #endif
 
index 25816d31f285ed510de3c8e07bfc3272f5edf8b3..3dd5d4b185ee00b34c57f3acad60873ae8b7f253 100644 (file)
@@ -24,7 +24,7 @@
  */
 
 void esp_set_breakpoint_if_jtag(void *fn);
-void esp_panicPutchar(char c);
+void esp_panicPutChar(char c);
 void esp_panicPutStr(const char *c);
 void esp_panicPutHex(int a);
 void esp_panicPutDec(int a);
index 065b323aa197faab3413dd51e15d58a0f31cd644..2f31d3f292617c1bcb642f3bf2254dea293f50c2 100755 (executable)
@@ -10,6 +10,7 @@ import tempfile
 import struct
 import array
 import errno
+import base64
 
 try:
     import esptool
@@ -23,10 +24,6 @@ except ImportError:
 
 __version__ = "0.1-dev"
 
-ESP32_COREDUMP_HDR_FMT              = '<4L'
-ESP32_COREDUMP_FLASH_MAGIC_START    = 0xDEADBEEF
-ESP32_COREDUMP_FLASH_MAGIC_END      = 0xACDCFEED
-
 
 class Struct(object):
     def __init__(self, buf=None):
@@ -77,7 +74,7 @@ class Elf32FileHeader(Struct):
         if buf is None:
             # Fill in sane ELF header for LSB32
             self.e_ident = "\x7fELF\1\1\1\0\0\0\0\0\0\0\0\0"
-            self.e_version = ESPCoreDumpFile.EV_CURRENT
+            self.e_version = ESPCoreDumpElfFile.EV_CURRENT
             self.e_ehsize = self.sizeof()
 
 
@@ -200,7 +197,7 @@ class ESPCoreDumpSection(esptool.ELFSection):
         return str
 
 
-class ESPCoreDumpFile(esptool.ELFFile):
+class ESPCoreDumpElfFile(esptool.ELFFile):
     # ELF file type
     ET_NONE             = 0x0 # No file type
     ET_REL              = 0x1 # Relocatable file
@@ -229,7 +226,7 @@ class ESPCoreDumpFile(esptool.ELFFile):
 
     def __init__(self, name=None):
         if name:
-            super(ESPCoreDumpFile, self).__init__(name)
+            super(ESPCoreDumpElfFile, self).__init__(name)
         else:
             self.sections = []
             self.program_segments = []
@@ -409,80 +406,12 @@ class ESPCoreDumpLoader(object):
     """
     TBD
     """
-    FLASH_READ_BLOCK_SZ = 0x2000
-    def __init__(self, off, path=None, chip='esp32', port=None, baud=None):
-#         print "esptool.__file__ %s" % esptool.__file__
-        if not path:
-            self.path = esptool.__file__
-            self.path = self.path[:-1]
-        else:
-            self.path =  path
-        self.port = port
-        self.baud = baud
-        self.chip = chip
-        self.fcores = []
-        self.fgdbcore = None 
-        self._load_coredump(off)
-        
-    def _load_coredump(self, off):
-        args = [self.path, '-c', self.chip]
-        if self.port:
-            args.extend(['-p', self.port])
-        if self.baud:
-            args.extend(['-b', str(self.baud)])
-        read_sz = self.FLASH_READ_BLOCK_SZ
-        read_off = off
-        args.extend(['read_flash', str(read_off), str(read_sz), ''])
-        try:
-            dump_sz = 0
-            tot_len = 0
-            while True:
-                fhnd,fname = tempfile.mkstemp()
-#                 print "tmpname %s" % fname
-#                 os.close(fhnd)
-                args[-1] = fname
-                et_out = subprocess.check_output(args)
-                print et_out
-    #             data = os.fdopen(fhnd, 'r').read(sz)
-                self.fcores.append(os.fdopen(fhnd, 'r'))
-                if dump_sz == 0:
-                    # read dump length from the first block
-                    dump_sz = self._read_core_dump_length(self.fcores[0])
-                tot_len += read_sz
-                if tot_len >= dump_sz:
-                    break
-                read_off += read_sz
-                if dump_sz - tot_len >= self.FLASH_READ_BLOCK_SZ:
-                    read_sz = self.FLASH_READ_BLOCK_SZ
-                else:
-                    read_sz = dump_sz - tot_len
-                args[-3] = str(read_off)
-                args[-2] = str(read_sz)
-                
-        except subprocess.CalledProcessError as e: 
-            print "esptool script execution failed with err %d" % e.returncode
-            print "Command ran: '%s'" % e.cmd
-            print "Command out:"
-            print e.output
-            self.cleanup()
-            return []
-
-    def _read_core_dump_length(self, f):
-        global ESP32_COREDUMP_HDR_FMT
-        global ESP32_COREDUMP_FLASH_MAGIC_START
-        print "Read core dump header from '%s'" % f.name
-        data = f.read(4*4)
-        mag1,tot_len,task_num,tcbsz = struct.unpack_from(ESP32_COREDUMP_HDR_FMT, data)
-        if mag1 != ESP32_COREDUMP_FLASH_MAGIC_START:
-            raise ESPCoreDumpLoaderError("Invalid start magic number!")
-        return tot_len
-                
-    def remove_tmp_file(self, fname):
-        try:
-            os.remove(fname)
-        except OSError as e:
-            if e.errno != errno.ENOENT:
-                print "Warning failed to remove temp file '%s'!" % fname
+    ESP32_COREDUMP_HDR_FMT      = '<3L'
+    ESP32_COREDUMP_HDR_SZ       = struct.calcsize(ESP32_COREDUMP_HDR_FMT)
+    ESP32_COREDUMP_TSK_HDR_FMT  = '<3L'
+    ESP32_COREDUMP_TSK_HDR_SZ   = struct.calcsize(ESP32_COREDUMP_TSK_HDR_FMT)
+    def __init__(self):
+        self.fcore = None
 
     def _get_registers_from_stack(self, data, grows_down):
     # from "gdb/xtensa-tdep.h"
@@ -582,51 +511,36 @@ class ESPCoreDumpLoader(object):
             regs[REG_AR_START_IDX + 0] = (regs[REG_AR_START_IDX + 0] & 0x3fffffff) | 0x40000000;
         return regs
 
+    def remove_tmp_file(self, fname):
+        try:
+            os.remove(fname)
+        except OSError as e:
+            if e.errno != errno.ENOENT:
+                print "Warning failed to remove temp file '%s'!" % fname
+
     def cleanup(self):
-#         if self.fgdbcore:
-#             self.fgdbcore.close()
-#            self.remove_tmp_file(self.fgdbcore.name)
-        for f in self.fcores:
-            if f:
-                f.close()
-                self.remove_tmp_file(f.name)
-
-    def get_corefile_from_flash(self):
+        if self.fcore:
+            self.fcore.close()
+            self.remove_tmp_file(self.fcore.name)
+
+    def create_corefile(self, core_fname=None, off=0):
         """ TBD
         """
-        global ESP32_COREDUMP_HDR_FMT
-        ESP32_COREDUMP_HDR_SZ = struct.calcsize(ESP32_COREDUMP_HDR_FMT)
-        ESP32_COREDUMP_TSK_HDR_FMT = '<LLL'
-        ESP32_COREDUMP_TSK_HDR_SZ = struct.calcsize(ESP32_COREDUMP_TSK_HDR_FMT)
-        ESP32_COREDUMP_MAGIC_FMT = '<L'
-        ESP32_COREDUMP_MAGIC_SZ = struct.calcsize(ESP32_COREDUMP_MAGIC_FMT)
-        no_progress = True #False
-        if no_progress:
-            flash_progress = None
-        else:
-            def flash_progress(progress, length):
-                msg = '%d (%d %%)' % (progress, progress * 100.0 / length)
-                padding = '\b' * len(msg)
-                if progress == length:
-                    padding = '\n'
-                sys.stdout.write(msg + padding)
-                sys.stdout.flush()
-    
-        core_off = 0
+        core_off = off
         print "Read core dump header"
-        data = self.read_flash(core_off, ESP32_COREDUMP_HDR_SZ, flash_progress)
-        mag1,tot_len,task_num,tcbsz = struct.unpack_from(ESP32_COREDUMP_HDR_FMT, data)
+        data = self.read_data(core_off, self.ESP32_COREDUMP_HDR_SZ)
+        tot_len,task_num,tcbsz = struct.unpack_from(self.ESP32_COREDUMP_HDR_FMT, data)
         tcbsz_aligned = tcbsz
         if tcbsz_aligned % 4:
             tcbsz_aligned = 4*(tcbsz_aligned/4 + 1)
-        print "mag1=%x, tot_len=%d, task_num=%d, tcbsz=%d" % (mag1,tot_len,task_num,tcbsz)
-        core_off += ESP32_COREDUMP_HDR_SZ
-        core_elf = ESPCoreDumpFile()
+        print "tot_len=%d, task_num=%d, tcbsz=%d" % (tot_len,task_num,tcbsz)
+        core_off += self.ESP32_COREDUMP_HDR_SZ
+        core_elf = ESPCoreDumpElfFile()
         notes = b''
         for i in range(task_num):
             print "Read task[%d] header" % i
-            data = self.read_flash(core_off, ESP32_COREDUMP_TSK_HDR_SZ, flash_progress)
-            tcb_addr,stack_top,stack_end = struct.unpack_from(ESP32_COREDUMP_TSK_HDR_FMT, data)
+            data = self.read_data(core_off, self.ESP32_COREDUMP_TSK_HDR_SZ)
+            tcb_addr,stack_top,stack_end = struct.unpack_from(self.ESP32_COREDUMP_TSK_HDR_FMT, data)
             if stack_end > stack_top:
                 stack_len = stack_end - stack_top
                 stack_base = stack_top
@@ -639,23 +553,23 @@ class ESPCoreDumpLoader(object):
             if stack_len_aligned % 4:
                 stack_len_aligned = 4*(stack_len_aligned/4 + 1)
                 
-            core_off += ESP32_COREDUMP_TSK_HDR_SZ
+            core_off += self.ESP32_COREDUMP_TSK_HDR_SZ
             print "Read task[%d] TCB" % i
-            data = self.read_flash(core_off, tcbsz_aligned, flash_progress)
+            data = self.read_data(core_off, tcbsz_aligned)
             if tcbsz != tcbsz_aligned:
-                core_elf.add_program_segment(tcb_addr, data[:tcbsz - tcbsz_aligned], ESPCoreDumpFile.PT_LOAD, ESPCoreDumpSegment.PF_R | ESPCoreDumpSegment.PF_W)
+                core_elf.add_program_segment(tcb_addr, data[:tcbsz - tcbsz_aligned], ESPCoreDumpElfFile.PT_LOAD, ESPCoreDumpSegment.PF_R | ESPCoreDumpSegment.PF_W)
             else:
-                core_elf.add_program_segment(tcb_addr, data, ESPCoreDumpFile.PT_LOAD, ESPCoreDumpSegment.PF_R | ESPCoreDumpSegment.PF_W)
+                core_elf.add_program_segment(tcb_addr, data, ESPCoreDumpElfFile.PT_LOAD, ESPCoreDumpSegment.PF_R | ESPCoreDumpSegment.PF_W)
     #         print "tcb=%s" % data
             core_off += tcbsz_aligned
             print "Read task[%d] stack %d bytes" % (i,stack_len)
-            data = self.read_flash(core_off, stack_len_aligned, flash_progress)
+            data = self.read_data(core_off, stack_len_aligned)
     #         print "stk=%s" % data
             if stack_len != stack_len_aligned:
                 data = data[:stack_len - stack_len_aligned]
-            core_elf.add_program_segment(stack_base, data, ESPCoreDumpFile.PT_LOAD, ESPCoreDumpSegment.PF_R | ESPCoreDumpSegment.PF_W)
+            core_elf.add_program_segment(stack_base, data, ESPCoreDumpElfFile.PT_LOAD, ESPCoreDumpSegment.PF_R | ESPCoreDumpSegment.PF_W)
             core_off += stack_len_aligned
-    
+
             task_regs = self._get_registers_from_stack(data, stack_end > stack_top)
             prstatus = XtensaPrStatus()
             prstatus.pr_cursig = 0 # TODO: set sig only for current/failed task
@@ -663,33 +577,140 @@ class ESPCoreDumpLoader(object):
             note = Elf32NoteDesc("CORE", 1, prstatus.dump() + struct.pack("<%dL" % len(task_regs), *task_regs)).dump()
             print "NOTE_LEN %d" % len(note)
             notes += note
-    
-        print "Read core dump endmarker"
-        data = self.read_flash(core_off, ESP32_COREDUMP_MAGIC_SZ, flash_progress)
-        mag = struct.unpack_from(ESP32_COREDUMP_MAGIC_FMT, data)
-        print "mag2=%x" % (mag)
-    
+
         # add notes
-        core_elf.add_program_segment(0, notes, ESPCoreDumpFile.PT_NOTE, 0)
+        core_elf.add_program_segment(0, notes, ESPCoreDumpElfFile.PT_NOTE, 0)
     
-        core_elf.e_type = ESPCoreDumpFile.ET_CORE
-        core_elf.e_machine = ESPCoreDumpFile.EM_XTENSA
-        fhnd,fname = tempfile.mkstemp()
-        self.fgdbcore = os.fdopen(fhnd, 'wb')
-        core_elf.dump(self.fgdbcore)
-        return fname
-    ######################### END ###########################
-
-    def read_flash(self, off, sz, progress=None):
-#         print "read_flash: %x %d" % (off, sz)
-        id = off / self.FLASH_READ_BLOCK_SZ
-        if id >= len(self.fcores):
-            return ''
-        self.fcores[id].seek(off % self.FLASH_READ_BLOCK_SZ)
-        data = self.fcores[id].read(sz)
+        core_elf.e_type = ESPCoreDumpElfFile.ET_CORE
+        core_elf.e_machine = ESPCoreDumpElfFile.EM_XTENSA
+        if core_fname:
+            fce = open(core_fname, 'wb')
+        else:
+            fhnd,core_fname = tempfile.mkstemp()
+            fce = os.fdopen(fhnd, 'wb')
+        core_elf.dump(fce)
+        fce.close()
+        return core_fname
+
+    def read_data(self, off, sz):
+#         print "read_data: %x %d" % (off, sz)
+        self.fcore.seek(off)
+        data = self.fcore.read(sz)
 #         print "data1: %s" % data
         return data
 
+
+class ESPCoreDumpFileLoader(ESPCoreDumpLoader):
+    """ TBD
+    """
+    def __init__(self, path, b64 = False):
+        super(ESPCoreDumpFileLoader, self).__init__()
+        self.fcore = self._load_coredump(path, b64)
+        
+    def _load_coredump(self, path, b64):
+        if b64:
+            fhnd,fname = tempfile.mkstemp()
+            print "tmpname %s" % fname
+            fcore = os.fdopen(fhnd, 'wb')
+            fb64 = open(path, 'r')
+            try:
+                while True:
+                    line = fb64.readline()
+                    if len(line) == 0:
+                        break
+                    data = base64.b64decode(line.rstrip('\r\n'))#, validate=True)
+                    fcore.write(data)
+                fcore.close()
+                fcore = open(fname, 'r')
+            finally:
+                fb64.close()
+        else:
+            fcore = open(path, 'r')
+        return fcore
+
+
+class ESPCoreDumpFlashLoader(ESPCoreDumpLoader):
+    """ TBD
+    """
+    ESP32_COREDUMP_FLASH_MAGIC_START    = 0xDEADBEEF
+    ESP32_COREDUMP_FLASH_MAGIC_END      = 0xACDCFEED
+    ESP32_COREDUMP_FLASH_MAGIC_FMT      = '<L'
+    ESP32_COREDUMP_FLASH_MAGIC_SZ       = struct.calcsize(ESP32_COREDUMP_FLASH_MAGIC_FMT)
+    ESP32_COREDUMP_FLASH_HDR_FMT        = '<4L'
+    ESP32_COREDUMP_FLASH_HDR_SZ         = struct.calcsize(ESP32_COREDUMP_FLASH_HDR_FMT)
+
+    def __init__(self, off, tool_path=None, chip='esp32', port=None, baud=None):
+#         print "esptool.__file__ %s" % esptool.__file__
+        super(ESPCoreDumpFlashLoader, self).__init__()
+        if not tool_path:
+            self.path = esptool.__file__
+            self.path = self.path[:-1]
+        else:
+            self.path =  tool_path
+        self.port = port
+        self.baud = baud
+        self.chip = chip
+        self.dump_sz = 0
+        self.fcore = self._load_coredump(off)
+        
+    def _load_coredump(self, off):
+        args = [self.path, '-c', self.chip]
+        if self.port:
+            args.extend(['-p', self.port])
+        if self.baud:
+            args.extend(['-b', str(self.baud)])
+        args.extend(['read_flash', str(off), str(self.ESP32_COREDUMP_FLASH_HDR_SZ), ''])
+
+        fname = None
+        try:
+            fhnd,fname = tempfile.mkstemp()
+            args[-1] = fname
+            # read core dump length
+            et_out = subprocess.check_output(args)
+            print et_out
+            f = os.fdopen(fhnd, 'r')
+            self.dump_sz = self._read_core_dump_length(f)
+            # read core dump
+            args[-2] = str(self. dump_sz)
+            et_out = subprocess.check_output(args)
+            print et_out
+        except subprocess.CalledProcessError as e: 
+            print "esptool script execution failed with err %d" % e.returncode
+            print "Command ran: '%s'" % e.cmd
+            print "Command out:"
+            print e.output
+            self.remove_tmp_file(fname)
+            raise e
+        return f
+
+    def _read_core_dump_length(self, f):
+        print "Read core dump header from '%s'" % f.name
+        data = f.read(4*4)
+        mag1,tot_len,task_num,tcbsz = struct.unpack_from(self.ESP32_COREDUMP_FLASH_HDR_FMT, data)
+        if mag1 != self.ESP32_COREDUMP_FLASH_MAGIC_START:
+            raise ESPCoreDumpLoaderError("Invalid start magic number!")
+        return tot_len
+
+    def create_corefile(self, core_fname=None):
+        """ TBD
+        """
+        print "Read core dump start marker"
+        data = self.read_data(0, self.ESP32_COREDUMP_FLASH_MAGIC_SZ)
+        mag1, = struct.unpack_from(self.ESP32_COREDUMP_FLASH_MAGIC_FMT, data)
+        if mag1 != self.ESP32_COREDUMP_FLASH_MAGIC_START:
+            print "Invalid start marker %x" % mag1
+            return None
+
+        print "Read core dump end marker"
+        data = self.read_data(self.dump_sz-self.ESP32_COREDUMP_FLASH_MAGIC_SZ, self.ESP32_COREDUMP_FLASH_MAGIC_SZ)
+        mag2, = struct.unpack_from(self.ESP32_COREDUMP_FLASH_MAGIC_FMT, data)
+        if mag2 != self.ESP32_COREDUMP_FLASH_MAGIC_END:
+            print "Invalid end marker %x" % mag2
+            return None
+        
+        return super(ESPCoreDumpFlashLoader, self).create_corefile(core_fname, off=self.ESP32_COREDUMP_FLASH_MAGIC_SZ)
+
+
 class GDBMIOutRecordHandler(object):
     """ TBD
     """
@@ -772,17 +793,22 @@ class GDBMIStreamConsoleHandler(GDBMIOutStreamHandler):
 def dbg_corefile(args):
     """ TBD
     """
-    print "dbg_corefile %s %s %s" % (args.gdb, args.prog, args.core)
+    print "dbg_corefile %s %s %s %s" % (args.gdb, args.prog, args.core, args.save_core)
     loader = None
     if not args.core:
-        loader = ESPCoreDumpLoader(args.off, port=args.port)
-        core_fname = loader.get_corefile_from_flash()
-        loader.fgdbcore.close()
-#         core_fname = 'esp_core.elf'
+        loader = ESPCoreDumpFlashLoader(args.off, port=args.port)
+        core_fname = loader.create_corefile(args.save_core)
+        if not core_fname:
+            print "Failed to create corefile!"
+            return
     else:
         core_fname = args.core
-#     print core_fname
-#     return
+        if args.core_format and args.core_format != 'elf':
+            loader = ESPCoreDumpFileLoader(core_fname, args.core_format == 'b64')
+            core_fname = loader.create_corefile(args.save_core)
+            if not core_fname:
+                print "Failed to create corefile!"
+                return
 
     p = subprocess.Popen(
             bufsize = 0,
@@ -794,8 +820,10 @@ def dbg_corefile(args):
             close_fds = True
             )
     p.wait()
+    
     if loader:
-        loader.remove_tmp_file(loader.fgdbcore.name)
+        if not args.core and not args.save_core:
+            loader.remove_tmp_file(core_fname)
         loader.cleanup()
     print 'Done!'
 
@@ -831,11 +859,19 @@ def info_corefile(args):
 
     loader = None
     if not args.core:
-        loader = ESPCoreDumpLoader(args.off, port=args.port)
-        core_fname = loader.get_corefile_from_flash()
-        loader.fgdbcore.close()
+        loader = ESPCoreDumpFlashLoader(args.off, port=args.port)
+        core_fname = loader.create_corefile(args.save_core)
+        if not core_fname:
+            print "Failed to create corefile!"
+            return
     else:
         core_fname = args.core
+        if args.core_format and args.core_format != 'elf':
+            loader = ESPCoreDumpFileLoader(core_fname, args.core_format == 'b64')
+            core_fname = loader.create_corefile(args.save_core)
+            if not core_fname:
+                print "Failed to create corefile!"
+                return
 
     handlers = {}
     handlers[GDBMIResultHandler.TAG] = GDBMIResultHandler(verbose=False)
@@ -855,8 +891,8 @@ def info_corefile(args):
             )
 
     gdbmi_read2prompt(p.stdout, handlers)
-    exe_elf = ESPCoreDumpFile(args.prog)
-    core_elf = ESPCoreDumpFile(core_fname)
+    exe_elf = ESPCoreDumpElfFile(args.prog)
+    core_elf = ESPCoreDumpElfFile(core_fname)
     merged_segs = []#[(s, 0) for s in exe_elf.sections if s.flags & (esptool.ELFSection.SHF_ALLOC | esptool.ELFSection.SHF_WRITE)]
     for s in exe_elf.sections:
         merged = False
@@ -940,9 +976,12 @@ def info_corefile(args):
     p.terminate()
     p.stdin.close()
     p.stdout.close()
+    
     if loader:
-        loader.remove_tmp_file(loader.fgdbcore.name)
+        if not args.core and not args.save_core:
+            loader.remove_tmp_file(core_fname)
         loader.cleanup()
+    print 'Done!'
         
 
 def main():
@@ -978,16 +1017,20 @@ def main():
         help='Starts GDB debugging session with specified corefile')
     parser_debug_coredump.add_argument('--gdb', '-g', help='Path to gdb', default='xtensa-esp32-elf-gdb')
     parser_debug_coredump.add_argument('--core', '-c', help='Path to core dump file (if skipped core dump will be read from flash)', type=str)
+    parser_debug_coredump.add_argument('--core-format', '-t', help='(elf, raw or b64). File specified with "-c" is an ELF ("elf"), raw (raw) or base64-encoded (b64) binary', type=str, default='elf')
     parser_debug_coredump.add_argument('--off', '-o', help='Ofsset of coredump partition in flash (type "make partition_table" to see).', type=int, default=0x110000)
+    parser_debug_coredump.add_argument('--save-core', '-s', help='Save core to file. Othwerwise temporary core file will be deleted. Ignored with "-c"', type=str)
     parser_debug_coredump.add_argument('prog', help='Path to program\'s ELF binary', type=str)
 
     parser_info_coredump = subparsers.add_parser(
         'info_corefile',
         help='Print core dump info from file')
     parser_info_coredump.add_argument('--gdb', '-g', help='Path to gdb', default='xtensa-esp32-elf-gdb')
-    parser_info_coredump.add_argument('--print-mem', '-m', help='Print memory dump', action='store_true')
     parser_info_coredump.add_argument('--core', '-c', help='Path to core dump file (if skipped core dump will be read from flash)', type=str)
+    parser_info_coredump.add_argument('--core-format', '-t', help='(elf, raw or b64). File specified with "-c" is an ELF ("elf"), raw (raw) or base64-encoded (b64) binary', type=str, default='elf')
     parser_info_coredump.add_argument('--off', '-o', help='Ofsset of coredump partition in flash (type "make partition_table" to see).', type=int, default=0x110000)
+    parser_info_coredump.add_argument('--save-core', '-s', help='Save core to file. Othwerwise temporary core file will be deleted. Does not work with "-c"', type=str)
+    parser_info_coredump.add_argument('--print-mem', '-m', help='Print memory dump', action='store_true')
     parser_info_coredump.add_argument('prog', help='Path to program\'s ELF binary', type=str)
 
     # internal sanity check - every operation matches a module function of the same name
index 2bedaa02e856626dec669fa32035371ffd491eff..7c9d8fd972b25e681e735eb9297d20f16ed04ab6 100644 (file)
 #define configXT_BOARD                      1   /* Board mode */
 #define configXT_SIMULATOR                                     0
 
-#if CONFIG_ESP32_ENABLE_COREDUMP_TO_FLASH | CONFIG_ESP32_ENABLE_COREDUMP_TO_FLASH
+#if CONFIG_ESP32_ENABLE_COREDUMP_TO_FLASH | CONFIG_ESP32_ENABLE_COREDUMP_TO_UART
 #define configENABLE_TASK_SNAPSHOT                     1
 #endif