From 96bb992a2cff8145245b833fe581bb8e670ba85d Mon Sep 17 00:00:00 2001
From: Ratan Kaliani <ratankaliani@berkeley.edu>
Date: Wed, 7 Aug 2024 11:27:44 -0700
Subject: [PATCH] fix: helper binary build

---
 build/src/lib.rs  | 15 +++++++++++----
 helper/src/lib.rs | 12 ++++--------
 2 files changed, 15 insertions(+), 12 deletions(-)

diff --git a/build/src/lib.rs b/build/src/lib.rs
index 56de3ad217..2cc6a69d05 100644
--- a/build/src/lib.rs
+++ b/build/src/lib.rs
@@ -224,10 +224,17 @@ fn copy_elf_to_output_dir(
 
     // The ELF is written to a target folder specified by the program's package. If built with Docker,
     // includes /docker after HELPER_TARGET_SUBDIR.
-    let target_dir_suffix = if args.docker {
-        format!("{}/{}", HELPER_TARGET_SUBDIR, "docker")
+    let mut target_dir_suffix = HELPER_TARGET_SUBDIR.to_string();
+    if args.docker {
+        target_dir_suffix = format!("{}/{}", HELPER_TARGET_SUBDIR, "docker");
+    }
+
+    // The ELF's file name is the binary name if it's specified. Otherwise, it is the root package
+    // name.
+    let original_elf_file_name = if !args.binary.is_empty() {
+        args.binary.clone()
     } else {
-        HELPER_TARGET_SUBDIR.to_string()
+        root_package_name.unwrap().clone()
     };
 
     let original_elf_path = program_metadata
@@ -235,7 +242,7 @@ fn copy_elf_to_output_dir(
         .join(target_dir_suffix)
         .join(BUILD_TARGET)
         .join("release")
-        .join(root_package_name.unwrap());
+        .join(original_elf_file_name);
 
     // The order of precedence for the ELF name is:
     // 1. --elf_name flag
diff --git a/helper/src/lib.rs b/helper/src/lib.rs
index 523c81509d..82371b4ae7 100644
--- a/helper/src/lib.rs
+++ b/helper/src/lib.rs
@@ -61,14 +61,10 @@ fn execute_build_cmd(
     }
 
     // Build the program with the given arguments.
-    let path_output = if let Some(args) = args {
-        sp1_build::build_program(&args, Some(program_dir.as_ref().to_path_buf()))
-    } else {
-        sp1_build::build_program(
-            &BuildArgs::default(),
-            Some(program_dir.as_ref().to_path_buf()),
-        )
-    };
+    let path_output = sp1_build::build_program(
+        &args.unwrap_or_default(),
+        Some(program_dir.as_ref().to_path_buf()),
+    );
     if let Err(err) = path_output {
         panic!("Failed to build SP1 program: {}.", err);
     }