Re: [PATCH v2 13/21] x86/virt/seamldr: Abort updates if errors occurred midway

[Xu Yilun]

On Tue, Sep 30, 2025 at 07:52:57PM -0700, Chao Gao wrote:
> The TDX Module update process has multiple stages, each of which may
> encounter failures.
> 
> The current state machine of updates proceeds to the next stage
> regardless of errors. But continuing updates when errors occur midway
> is pointless.
> 
> Add support of transitioning directly to the final stage on errors,
> effectively aborting the update and skipping all remaining stages.
> 
> Signed-off-by: Chao Gao <chao.gao@xxxxxxxxx>
> Tested-by: Farrah Chen <farrah.chen@xxxxxxxxx>
> ---
>  arch/x86/virt/vmx/tdx/seamldr.c | 17 +++++++++++++++--
>  1 file changed, 15 insertions(+), 2 deletions(-)
> 
> diff --git a/arch/x86/virt/vmx/tdx/seamldr.c b/arch/x86/virt/vmx/tdx/seamldr.c
> index b074630d42e3..fca558b90f72 100644
> --- a/arch/x86/virt/vmx/tdx/seamldr.c
> +++ b/arch/x86/virt/vmx/tdx/seamldr.c
> @@ -235,6 +235,7 @@ enum tdp_state {
>  static struct {
>  	enum tdp_state state;
>  	atomic_t thread_ack;
> +	atomic_t failed;
>  } tdp_data;
>  
>  static void set_target_state(enum tdp_state state)
> @@ -249,8 +250,16 @@ static void set_target_state(enum tdp_state state)
>  /* Last one to ack a state moves to the next state. */
>  static void ack_state(void)
>  {
> -	if (atomic_dec_and_test(&tdp_data.thread_ack))
> -		set_target_state(tdp_data.state + 1);
> +	if (atomic_dec_and_test(&tdp_data.thread_ack)) {
> +		/*
> +		 * If an error occurred, abort the update by skipping to
> +		 * the final state
> +		 */
> +		if (atomic_read(&tdp_data.failed))
> +			set_target_state(TDP_DONE);

Could we immediately move to TDP_DONE once the error happens, i.e. no
need to get all ack for current state?

> +		else
> +			set_target_state(tdp_data.state + 1);
> +	}
>  }

----8<-----
  static void ack_state(void)
  {
        if (atomic_read(&tdp_data.failed)) {
                set_target_state(TDP_DONE);
		return;
	}

        if (atomic_dec_and_test(&tdp_data.thread_ack))
                 set_target_state(tdp_data.state + 1);
  }