Add simple RAM profiler

cesanta · Nov 28, 2024 · 7fffe0a · 7fffe0a
1 parent 8a42325
commit 7fffe0a
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/examples/arduino/w5500-mqtt/mongoose_config.h b/examples/arduino/w5500-mqtt/mongoose_config.h
@@ -15,8 +15,7 @@
 #define mkdir(a, b) (-1)
 #define MG_ENABLE_TCPIP_PRINT_DEBUG_STATS 0
 #define MG_ENABLE_CUSTOM_MILLIS 1
+#define MG_ENABLE_CUSTOM_RANDOM 1
+// #define MG_TLS MG_TLS_BUILTIN
 #define MG_IO_SIZE 128
 
-// Enable TLS
-// #define MG_TLS MG_TLS_BUILTIN
-// #define MG_ENABLE_CUSTOM_RANDOM 1
diff --git a/examples/arduino/w5500-mqtt/w5500-mqtt.ino b/examples/arduino/w5500-mqtt/w5500-mqtt.ino
@@ -20,6 +20,11 @@ struct mg_tcpip_if mif = {.mac = {2, 0, 1, 2, 3, 5}};  // Network interface
 uint64_t mg_millis(void) {
   return millis();
 }
+bool mg_random(void *buf, size_t len) {  // For TLS
+  uint8_t *p = (uint8_t *) buf;
+  while (len--) *p++ = (unsigned char) (rand() & 255);
+  return true;
+}
 
 void mqtt_publish(const char *message) {
   struct mg_mqtt_opts opts = {};
@@ -28,6 +33,15 @@ void mqtt_publish(const char *message) {
   if (mqtt_connection) mg_mqtt_pub(mqtt_connection, &opts);
 }
 
+// Crude function to get available RAM, for quick profiling
+size_t getFreeRAM(void) {
+  size_t size = 0, increment = 100;
+  void *p;
+  while ((p = malloc(size)) != NULL) free(p), size += increment;
+  return size;
+}
+
+// Implement LED control over MQTT: "on" and "off" commands
 void handle_command(struct mg_str msg) {
   if (msg.len == 3 && memcmp(msg.buf, "off", 3) == 0) {
     digitalWrite(LED_PIN, LOW);
@@ -36,6 +50,7 @@ void handle_command(struct mg_str msg) {
     digitalWrite(LED_PIN, HIGH);
     mqtt_publish("done - on");
   }
+  MG_INFO(("Free RAM: %u bytes", getFreeRAM()));
 }
 
 static void mqtt_ev_handler(struct mg_connection *c, int ev, void *ev_data) {