scripts: improve create-dfu-image.sh

Signed-off-by: Onur Özkan <work@onurozkan.dev>

scripts/analyze-fw.py

@@ -2,7 +2,7 @@
 """Show flash, static RAM and ELF-section use for embedded firmware.
 
 Usage:
-    python3 scripts/analyze_kaptan_firmware.py <firmware.elf> <memory.x>
+    scripts/analyze-fw.py <firmware.elf> <memory.x>
 """
 
 import re

scripts/create-dfu-image.sh

@@ -1,16 +1,31 @@
 #!/usr/bin/env bash
 set -euo pipefail
 
-if [[ $# -ne 2 ]]; then
-    printf 'Usage: %s <input.elf> <output.bin>\n' "$0" >&2
+if [[ $# -ne 1 ]]; then
+    printf 'Usage: %s <firmware.elf>\n' "$0" >&2
     exit 2
 fi
 
 readonly input_elf="$1"
-readonly output_image="$2"
+readonly firmware_name="$(basename "${input_elf%.elf}")"
+readonly board_name="${firmware_name#kaptan-}"
+readonly output_image="${input_elf%.elf}.bin"
+readonly script_directory="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
+readonly workspace_root="$(cd -- "$script_directory/.." && pwd)"
+readonly memory_x="$workspace_root/boards/$board_name/memory.x"
 readonly host="$(rustc -vV | sed -n 's/^host: //p')"
 readonly llvm_objcopy="$(rustc --print sysroot)/lib/rustlib/$host/bin/llvm-objcopy"
 
+if [[ ! -f "$input_elf" ]]; then
+    printf 'firmware ELF does not exist: %s\n' "$input_elf" >&2
+    exit 1
+fi
+
+if [[ ! -f "$memory_x" ]]; then
+    printf 'cannot find the linker layout for board %q: %s\n' "$board_name" "$memory_x" >&2
+    exit 1
+fi
+
 if [[ ! -x "$llvm_objcopy" ]]; then
     printf 'llvm-objcopy is unavailable; install it with: rustup component add llvm-tools-preview\n' >&2
     exit 1
@@ -18,3 +33,4 @@ fi
 
 "$llvm_objcopy" -O binary "$input_elf" "$output_image"
 printf 'DFU image: %s\n' "$output_image"
+"$script_directory/analyze-fw.py" "$input_elf" "$memory_x"