shell: flashdev: close fds on allocation failure

2026-10-06 10:59:48 +08:00 · 2026-09-25 02:05:33 +00:00
parent 05bab5136d
commit 124690270d
1 changed files with 6 additions and 1 deletions
@@ -312,7 +312,12 @@ int flashdev_shell_write(

  /* Create buffer */
  buffer = calloc(1, 0x1000);
-
+  if (buffer == NULL) {
+    close(flash);
+    close(file);
+    printf("Failed to allocate write buffer\n");
+    return -1;
+  }
  /* Write file to flash device in 0x1000 byte chunks */
  offset = 0;
  while (offset != length) {