feat: add malloc failure check

merbanan · Dec 25, 2023 · ae98d2c · ae98d2c
1 parent fa8dfc5
commit ae98d2c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/devices/compustar_1wg3r.c b/src/devices/compustar_1wg3r.c
@@ -60,6 +60,10 @@ static int compustar_1wg3r_decode(r_device *decoder, bitbuffer_t *bitbuffer)
 {
     int rows_data_idx = -1;
     data_t **rows_data = malloc(bitbuffer->num_rows * sizeof(data_t *));
+    if (!rows_data) {
+        WARN_MALLOC("compustar_1wg3r_decode()");
+        return -1; // NOTE: returns error on alloc failure.
+    }
 
     // loop through all of the rows and only return unique valid results
     // programming mode will send a sequence of key presses all in one message