From c6f3da529f4889cd5de81b29399c105db1c03a6b Mon Sep 17 00:00:00 2001
From: Caisking <3449750381@qq.com>
Date: Fri, 5 Jul 2024 10:46:37 +0800
Subject: [PATCH] =?utf8?q?=E8=A1=A5=E4=BA=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 .../wearable/fd230app/nativeui/traing/src/TraingModel.cpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/code/sdk/application/wearable/fd230app/nativeui/traing/src/TraingModel.cpp b/code/sdk/application/wearable/fd230app/nativeui/traing/src/TraingModel.cpp
index 9c9f59b55..44bfd835a 100644
--- a/code/sdk/application/wearable/fd230app/nativeui/traing/src/TraingModel.cpp
+++ b/code/sdk/application/wearable/fd230app/nativeui/traing/src/TraingModel.cpp
@@ -1111,13 +1111,19 @@ namespace OHOS
             sprintf_s(file_path, sizeof(file_path), "%s%s.bin", TRAINING_PATH, timedata); // 防止越界
 
             FILE *fp = fopen(file_path, "rb+");
-            Trainingdata *data = (Trainingdata *)malloc(sizeof(Trainingdata));
             if (fp == NULL)
             {
                 WEARABLE_LOGE(WEARABLE_LOG_MODULE_APP, "file is NULL");
                 free(data); // 修复内存泄漏
                 return NULL;
             }
+            Trainingdata *data = (Trainingdata *)malloc(sizeof(Trainingdata));
+            if (data == NULL)
+            {
+                WEARABLE_LOGE(WEARABLE_LOG_MODULE_APP, "Trainingdata malloc is error");
+                fclose(fp);
+                return NULL;
+            }
             if (get_Trainingdata_data_num(fp) <= offset)
             {
                 WEARABLE_LOGE(WEARABLE_LOG_MODULE_APP, "file is NULL %d", get_Trainingdata_data_num(fp));
-- 
2.17.1