[shell/MainLoopDefault.cpp] Add a missing null check

project-chip · Aug 9, 2023 · 1089386 · 1089386
1 parent bb0031f
commit 1089386
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/lib/shell/MainLoopDefault.cpp b/src/lib/shell/MainLoopDefault.cpp
@@ -196,6 +196,18 @@ void Engine::RunMainLoop()
  while (true)
  {
  char * line = static_cast<char *>(Platform::MemoryAlloc(CHIP_SHELL_MAX_LINE_SIZE));
+ if (line == nullptr)
+ {
+ CHIP_ERROR error = CHIP_ERROR_NO_MEMORY;
+ char errorStr[160];
+ bool errorStrFound = FormatCHIPError(errorStr, sizeof(errorStr), error);
+ if (!errorStrFound)
+ {
+ errorStr[0] = 0;
+ }
+ streamer_printf(streamer_get(), "Error: %s\r\n", errorStr);
+ }
+
  if (ReadLine(line, CHIP_SHELL_MAX_LINE_SIZE) == 0u)
  {
  // Stop loop in case of empty read (Ctrl-D).