[CI] Make SystemCommands take a StringBuilderBase to avoid extra stac…

…k usage (#17984)
project-chip · May 13, 2022 · 1214366 · 1214366
1 parent 5bed9c9
commit 1214366
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/src/app/tests/suites/commands/system/SystemCommands.cpp b/src/app/tests/suites/commands/system/SystemCommands.cpp
@@ -16,8 +16,6 @@
  *
  */
 
-#include <lib/support/StringBuilder.h>
-
 #include "SystemCommands.h"
 
 namespace {
@@ -36,8 +34,8 @@ CHIP_ERROR SystemCommands::Start(uint16_t discriminator, uint16_t port, const ch
     constexpr const char * scriptName = "Start.py";
 
     char command[kCommandMaxLen];
-    ReturnErrorOnFailure(
-        CreateCommonCommandArgs(command, sizeof(command), scriptDir, scriptName, registerKey, discriminator, port, kvs));
+    chip::StringBuilderBase builder(command, sizeof(command));
+    ReturnErrorOnFailure(CreateCommonCommandArgs(builder, scriptDir, scriptName, registerKey, discriminator, port, kvs));
     return RunInternal(command);
 }
 
@@ -80,13 +78,12 @@ CHIP_ERROR SystemCommands::RunInternal(const char * command)
     return ContinueOnChipMainThread(CHIP_NO_ERROR);
 }
 
-CHIP_ERROR SystemCommands::CreateCommonCommandArgs(char * commandBuffer, size_t commandBufferSize, const char * scriptDir,
+CHIP_ERROR SystemCommands::CreateCommonCommandArgs(chip::StringBuilderBase & builder, const char * scriptDir,
                                                    const char * scriptName, const char * registerKey, uint16_t discriminator,
                                                    uint16_t port, const char * kvs)
 {
     VerifyOrReturnError(registerKey != nullptr, CHIP_ERROR_INVALID_KEY_ID);
 
-    chip::StringBuilder<kCommandMaxLen> builder;
     builder.Add(scriptDir);
     builder.Add(scriptName);
     builder.Add(" ");
@@ -109,7 +106,6 @@ CHIP_ERROR SystemCommands::CreateCommonCommandArgs(char * commandBuffer, size_t
         builder.Add(kvs);
     }
     VerifyOrReturnError(builder.Fit(), CHIP_ERROR_BUFFER_TOO_SMALL);
-    strncpy(commandBuffer, builder.c_str(), commandBufferSize);
 
     return CHIP_NO_ERROR;
 }
diff --git a/src/app/tests/suites/commands/system/SystemCommands.h b/src/app/tests/suites/commands/system/SystemCommands.h
@@ -19,6 +19,7 @@
 #pragma once
 
 #include <lib/support/CodeUtils.h>
+#include <lib/support/StringBuilder.h>
 
 class SystemCommands
 {
@@ -36,7 +37,6 @@ class SystemCommands
 
 private:
     CHIP_ERROR RunInternal(const char * command);
-    CHIP_ERROR CreateCommonCommandArgs(char * commandBuffer, size_t commandBufferSize, const char * scriptDir,
-                                       const char * scriptName, const char * registerKey, uint16_t discriminator, uint16_t port,
-                                       const char * kvs);
+    CHIP_ERROR CreateCommonCommandArgs(chip::StringBuilderBase & builder, const char * scriptDir, const char * scriptName,
+                                       const char * registerKey, uint16_t discriminator, uint16_t port, const char * kvs);
 };