ArmPkg/ArmScmiDxe: Fix the calculation of RequiredArraySize

As per the SCMI specification, section CLOCK_DESCRIBE_RATES mentions
that the value of num_rates_flags[11:0] in the response must be 3 if
the return format is the triplet. Due to the buggy firmware, this was
not noticed for long time. The firmware is now fixed resulting in
ClockDescribeRates() to fail with "Buffer Too Small" error as the
RequiredArraySize gets miscalculated as 72 instead of 24.

Fix the issue by reusing the logic for both the return format which
must work if num_rates_flags has correct value as expected from the
specification.

Cc: Girish Pathak <girish.pathak@arm.com>
Cc: Jeff Brasen <jbrasen@nvidia.com>
Reviewed-by: Pierre Gondois <pierre.gondois@arm.com>
Tested-by: Pierre Gondois <pierre.gondois@arm.com>
Reported-by: Sami Mujawar <sami.mujawar@arm.com>
Signed-off-by: Sudeep Holla <sudeep.holla@arm.com>
Reviewed-by: Sami Mujawar <sami.mujawar@arm.com>
Tested-by: Sami Mujawar <sami.mujawar@arm.com>

diff --git a/ArmPkg/Drivers/ArmScmiDxe/ScmiClockProtocol.c b/ArmPkg/Drivers/ArmScmiDxe/ScmiClockProtocol.c
index 12a7e6df5d4334424f7cd917046162dbe217bc0b..f092208a66018beaca3d76a9cbab1502dcc526bc 100644 (file)
--- a/ArmPkg/Drivers/ArmScmiDxe/ScmiClockProtocol.c
+++ b/ArmPkg/Drivers/ArmScmiDxe/ScmiClockProtocol.c
@@ -236,12 +236,7 @@ ClockDescribeRates (
       *TotalRates = NUM_RATES (DescribeRates->NumRatesFlags)\r
                     + NUM_REMAIN_RATES (DescribeRates->NumRatesFlags);\r
 \r
-      if (*Format == ScmiClockRateFormatDiscrete) {\r
-        RequiredArraySize = (*TotalRates) * sizeof (UINT64);\r
-      } else {\r
-        // We need to return triplet of 64 bit value for each rate\r
-        RequiredArraySize = (*TotalRates) * 3 * sizeof (UINT64);\r
-      }\r
+      RequiredArraySize = (*TotalRates) * sizeof (UINT64);\r
 \r
       if (RequiredArraySize > (*RateArraySize)) {\r
         *RateArraySize = RequiredArraySize;\r
@@ -259,23 +254,21 @@ ClockDescribeRates (
           ConvertTo64Bit (Rate->Low, Rate->High);\r
       }\r
     } else {\r
-      for (RateNo = 0; RateNo < NUM_RATES (DescribeRates->NumRatesFlags); RateNo++) {\r
-        // Linear clock rates from minimum to maximum in steps\r
-        // Minimum clock rate.\r
-        Rate                                    = &DescribeRates->Rates[RateOffset++];\r
-        RateArray[RateIndex].ContinuousRate.Min =\r
-          ConvertTo64Bit (Rate->Low, Rate->High);\r
-\r
-        Rate = &DescribeRates->Rates[RateOffset++];\r
-        // Maximum clock rate.\r
-        RateArray[RateIndex].ContinuousRate.Max =\r
-          ConvertTo64Bit (Rate->Low, Rate->High);\r
-\r
-        Rate = &DescribeRates->Rates[RateOffset++];\r
-        // Step.\r
-        RateArray[RateIndex++].ContinuousRate.Step =\r
-          ConvertTo64Bit (Rate->Low, Rate->High);\r
-      }\r
+      // Linear clock rates from minimum to maximum in steps\r
+      // Minimum clock rate.\r
+      Rate                                    = &DescribeRates->Rates[RateOffset++];\r
+      RateArray[RateIndex].ContinuousRate.Min =\r
+        ConvertTo64Bit (Rate->Low, Rate->High);\r
+\r
+      Rate = &DescribeRates->Rates[RateOffset++];\r
+      // Maximum clock rate.\r
+      RateArray[RateIndex].ContinuousRate.Max =\r
+        ConvertTo64Bit (Rate->Low, Rate->High);\r
+\r
+      Rate = &DescribeRates->Rates[RateOffset++];\r
+      // Step.\r
+      RateArray[RateIndex++].ContinuousRate.Step =\r
+        ConvertTo64Bit (Rate->Low, Rate->High);\r
     }\r
   } while (NUM_REMAIN_RATES (DescribeRates->NumRatesFlags) != 0);\r
 \r