On 3/12/2025 11:19 PM, mhkelle...@gmail.com wrote:
> From: Michael Kelley <mhkli...@outlook.com>
> 
> Update hypercall call sites to use the new hv_hvcall_*() functions
> to set up hypercall arguments. Since these functions zero the
> fixed portion of input memory, remove now redundant zero'ing of
> input fields.
> 
> hv_post_message() requires additional updates. The payload area is
> treated as an array to avoid wasting cycles on zero'ing it and
> then overwriting with memcpy(). To allow treatment as an array,
> the corresponding payload[] field is updated to have zero size.
> 
I'd prefer to leave the payload field as a fixed-sized array.
Changing it to a flexible array makes it look like that input is
for a variable-sized or rep hypercall, and it makes the surrounding
code in hv_post_message() more complex and inscrutable as a result.

I suggest leaving hv_post_message() alone, except for changing
hyperv_pcpu_input_arg -> hyperv_pcpu_arg, and perhaps a comment
explaining why hv_hvcall_input() isn't used there.

> Signed-off-by: Michael Kelley <mhkli...@outlook.com>
> ---
>  drivers/hv/hv.c           | 9 ++++++---
>  drivers/hv/hv_balloon.c   | 4 ++--
>  drivers/hv/hv_common.c    | 2 +-
>  drivers/hv/hv_proc.c      | 8 ++++----
>  drivers/hv/hyperv_vmbus.h | 2 +-
>  5 files changed, 14 insertions(+), 11 deletions(-)
> 
> diff --git a/drivers/hv/hv.c b/drivers/hv/hv.c
> index a38f84548bc2..e2dcbc816fc5 100644
> --- a/drivers/hv/hv.c
> +++ b/drivers/hv/hv.c
> @@ -66,7 +66,8 @@ int hv_post_message(union hv_connection_id connection_id,
>       if (hv_isolation_type_tdx() && ms_hyperv.paravisor_present)
>               aligned_msg = 
> this_cpu_ptr(hv_context.cpu_context)->post_msg_page;
>       else
> -             aligned_msg = *this_cpu_ptr(hyperv_pcpu_input_arg);
> +             hv_hvcall_in_array(&aligned_msg, sizeof(*aligned_msg),
> +                                sizeof(aligned_msg->payload[0]));
>  
>       aligned_msg->connectionid = connection_id;
>       aligned_msg->reserved = 0;
> @@ -80,8 +81,10 @@ int hv_post_message(union hv_connection_id connection_id,
>                                                 virt_to_phys(aligned_msg), 0);
>               else if (hv_isolation_type_snp())
>                       status = hv_ghcb_hypercall(HVCALL_POST_MESSAGE,
> -                                                aligned_msg, NULL,
> -                                                sizeof(*aligned_msg));
> +                                                aligned_msg,
> +                                                NULL,
> +                                                struct_size(aligned_msg, 
> payload,
> +                                                            
> HV_MESSAGE_PAYLOAD_QWORD_COUNT));

See my comment above, I'd prefer to leave this function mostly
alone to maintain readability.

>               else
>                       status = HV_STATUS_INVALID_PARAMETER;
>       } else {
> diff --git a/drivers/hv/hv_balloon.c b/drivers/hv/hv_balloon.c
> index fec2f18679e3..2def8b8794ee 100644
> --- a/drivers/hv/hv_balloon.c
> +++ b/drivers/hv/hv_balloon.c
> @@ -1582,14 +1582,14 @@ static int hv_free_page_report(struct 
> page_reporting_dev_info *pr_dev_info,
>       WARN_ON_ONCE(nents > HV_MEMORY_HINT_MAX_GPA_PAGE_RANGES);
>       WARN_ON_ONCE(sgl->length < (HV_HYP_PAGE_SIZE << page_reporting_order));
>       local_irq_save(flags);
> -     hint = *this_cpu_ptr(hyperv_pcpu_input_arg);
> +
> +     hv_hvcall_in_array(&hint, sizeof(*hint), sizeof(hint->ranges[0]));

We should ensure the returned batch size is large enough for
"nents".

>       if (!hint) {
>               local_irq_restore(flags);
>               return -ENOSPC;
>       }
>  
>       hint->heat_type = HV_EXTMEM_HEAT_HINT_COLD_DISCARD;
> -     hint->reserved = 0;
>       for_each_sg(sgl, sg, nents, i) {
>               union hv_gpa_page_range *range;
>  
> diff --git a/drivers/hv/hv_common.c b/drivers/hv/hv_common.c
> index 9804adb4cc56..a6b1cdfbc8d4 100644
> --- a/drivers/hv/hv_common.c
> +++ b/drivers/hv/hv_common.c
> @@ -293,7 +293,7 @@ void __init hv_get_partition_id(void)
>       u64 status, pt_id;
>  
>       local_irq_save(flags);
> -     output = *this_cpu_ptr(hyperv_pcpu_input_arg);
> +     hv_hvcall_inout(NULL, 0, &output, sizeof(*output));
>       status = hv_do_hypercall(HVCALL_GET_PARTITION_ID, NULL, &output);
>       pt_id = output->partition_id;
>       local_irq_restore(flags);
> diff --git a/drivers/hv/hv_proc.c b/drivers/hv/hv_proc.c
> index 2fae18e4f7d2..5c580ee1c23f 100644
> --- a/drivers/hv/hv_proc.c
> +++ b/drivers/hv/hv_proc.c
> @@ -73,7 +73,8 @@ int hv_call_deposit_pages(int node, u64 partition_id, u32 
> num_pages)
>  
>       local_irq_save(flags);
>  
> -     input_page = *this_cpu_ptr(hyperv_pcpu_input_arg);
> +     hv_hvcall_in_array(&input_page, sizeof(*input_page),
> +                        sizeof(input_page->gpa_page_list[0]));

We should ensure the returned batch size is large enough.

>  
>       input_page->partition_id = partition_id;
>  
> @@ -124,9 +125,8 @@ int hv_call_add_logical_proc(int node, u32 lp_index, u32 
> apic_id)
>       do {
>               local_irq_save(flags);
>  
> -             input = *this_cpu_ptr(hyperv_pcpu_input_arg);
>               /* We don't do anything with the output right now */
> -             output = *this_cpu_ptr(hyperv_pcpu_output_arg);
> +             hv_hvcall_inout(&input, sizeof(*input), &output, 
> sizeof(*output));
>  
>               input->lp_index = lp_index;
>               input->apic_id = apic_id;
> @@ -167,7 +167,7 @@ int hv_call_create_vp(int node, u64 partition_id, u32 
> vp_index, u32 flags)
>       do {
>               local_irq_save(irq_flags);
>  
> -             input = *this_cpu_ptr(hyperv_pcpu_input_arg);
> +             hv_hvcall_in(&input, sizeof(*input));
>  
>               input->partition_id = partition_id;
>               input->vp_index = vp_index;
> diff --git a/drivers/hv/hyperv_vmbus.h b/drivers/hv/hyperv_vmbus.h
> index 29780f3a7478..44b5e8330d9d 100644
> --- a/drivers/hv/hyperv_vmbus.h
> +++ b/drivers/hv/hyperv_vmbus.h
> @@ -101,7 +101,7 @@ struct hv_input_post_message {
>       u32 reserved;
>       u32 message_type;
>       u32 payload_size;
> -     u64 payload[HV_MESSAGE_PAYLOAD_QWORD_COUNT];
> +     u64 payload[];

See my comment above, I'd prefer to keep this how it is.

>  };
>  
>  

Thanks
Nuno


Reply via email to