fix leak; protect against malloc() fail

Author: mmoskal

main/main.c

@@ -24,6 +24,8 @@ int target_in_irq(void) {
 
 void flush_dmesg(void) {
     char *dmesgCopy = malloc(sizeof(codalLogStore));
+    if (!dmesgCopy)
+        return;
 
     uint32_t len;
 

main/rx.c

@@ -14,6 +14,9 @@ static int jd_rx_frame_received_core(jd_frame_t *frame, int is_loop) {
         return 0;
 
     jd_frame_t *copy = malloc(JD_FRAME_SIZE(frame));
+    if (!copy)
+        return -1;
+
     memcpy(copy, frame, JD_FRAME_SIZE(frame));
 
     if (!is_loop)

main/usb.c

@@ -109,6 +109,9 @@ static void hf2_send_buffer(uint8_t flag, const void *data, unsigned size, uint3
         size += 4;
 
     BufferEntry *ent = (BufferEntry *)malloc(sizeof(BufferEntry) + size);
+    if (!ent)
+        return;
+
     ent->size = size;
     ent->flag = flag;
     uint8_t *dst = ent->data;
@@ -121,8 +124,10 @@ static void hf2_send_buffer(uint8_t flag, const void *data, unsigned size, uint3
 
     memcpy(dst, data, size);
 
-    if (worker_run(fg_worker, send_buffer_core, ent) != 0)
+    if (worker_run(fg_worker, send_buffer_core, ent) != 0) {
         OVF_ERROR("HF2 queue full");
+        free(ent);
+    }
 }
 
 int hf2_send_event(uint32_t evId, const void *data, int size) {