EmbeddedPkg/AndroidFastboot: eliminate deprecated string function calls
[mirror_edk2.git] / EmbeddedPkg / Application / AndroidFastboot / AndroidFastbootApp.c
index 9ddc34f..c5e8a7e 100644 (file)
@@ -99,7 +99,7 @@ HandleDownload (
   IN CHAR8 *NumBytesString\r
   )\r
 {\r
-  CHAR8       Response[12] = "DATA";\r
+  CHAR8       Response[13];\r
   CHAR16      OutputString[FASTBOOT_STRING_MAX_LENGTH];\r
 \r
   // Argument is 8-character ASCII string hex representation of number of bytes\r
@@ -127,8 +127,10 @@ HandleDownload (
   if (mDataBuffer == NULL) {\r
     SEND_LITERAL ("FAILNot enough memory");\r
   } else {\r
-    AsciiStrnCpy (Response + 4, NumBytesString, 8);\r
-    mTransport->Send (sizeof(Response), Response, &mFatalSendErrorEvent);\r
+    ZeroMem (Response, sizeof Response);\r
+    AsciiSPrint (Response, sizeof Response, "DATA%x",\r
+      (UINT32)mNumDataBytes);\r
+    mTransport->Send (sizeof Response - 1, Response, &mFatalSendErrorEvent);\r
 \r
     mState = ExpectDataState;\r
     mBytesReceivedSoFar = 0;\r
@@ -257,8 +259,7 @@ AcceptCmd (
   }\r
 \r
   // Commands aren't null-terminated. Let's get a null-terminated version.\r
-  AsciiStrnCpy (Command, Data, Size);\r
-  Command[Size] = '\0';\r
+  AsciiStrnCpyS (Command, sizeof Command, Data, Size);\r
 \r
   // Parse command\r
   if (MATCH_CMD_LITERAL ("getvar", Command)) {\r