Method walltime_bench_cmd

src/executor/wall_time/executor.rs:118–146

Prepare the benchmark command wrapped with the necessary environment variables for introspection and environment forwarding ahead of privilege escalation and isolation.

(
        config: &ExecutorConfig,
        execution_context: &ExecutionContext,
    )

    /// Prepare the benchmark command wrapped with the necessary environment variables for
    /// introspection and environment forwarding ahead of privilege escalation and isolation.
    fn walltime_bench_cmd(
        config: &ExecutorConfig,
        execution_context: &ExecutionContext,
    ) -> Result<(NamedTempFile, NamedTempFile, CommandBuilder)> {
        let path_value = build_path_env(config.enable_introspection)?;

        let mut extra_env = get_base_injected_env(
            RunnerMode::Walltime,
            &execution_context.profile_folder,
            &execution_context.config,
        );
        extra_env.insert("PATH".into(), path_value);

        // We have to write the benchmark command to a script, to ensure proper formatting
        // and to not have to manually escape everything.
        let mut script_file = NamedTempFile::new()?;
        script_file.write_all(get_bench_command(config)?.as_bytes())?;

        let mut bench_cmd = CommandBuilder::new("bash");
        bench_cmd.arg(script_file.path());
        let (mut bench_cmd, env_file) = wrap_with_env(bench_cmd, &extra_env)?;

        if let Some(cwd) = &config.working_directory {
            let abs_cwd = canonicalize(cwd)?;
            bench_cmd.current_dir(abs_cwd);
        }

        Ok((env_file, script_file, bench_cmd))
    }
}

#[async_trait(?Send)]