From f9620f7f7f24b68c403fab4b76c33a9bfced5e55 Mon Sep 17 00:00:00 2001
From: Zero Fanker <zcyfksn@vip.qq.com>
Date: Thu, 6 Jun 2024 00:32:01 -0400
Subject: [PATCH] #35, fixed taskforce unit delete .

---
 MissionEditor/TaskForce.cpp | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/MissionEditor/TaskForce.cpp b/MissionEditor/TaskForce.cpp
index 37d33e1..849478d 100644
--- a/MissionEditor/TaskForce.cpp
+++ b/MissionEditor/TaskForce.cpp
@@ -250,11 +250,15 @@ void CTaskForce::OnDeleteunit()
 	TruncSpace(tf);
 	auto sec = ini.TryGetSection(tf);
 	ASSERT(sec != nullptr);
-	char k[50];
-	itoa(u, k, 10);
+	CString numberStrToDelete;
+	numberStrToDelete.Format("%d", u);
+
+	if (!sec) {
+		return;
+	}
 
 	if (sec->Size() < 4) {
-		sec->RemoveByKey(k);
+		sec->RemoveByKey(numberStrToDelete);
 		m_Units.DeleteString(sel);
 		m_UnitType.SetWindowText("");
 		m_NumberOfUnits = atoi("0");
@@ -265,14 +269,12 @@ void CTaskForce::OnDeleteunit()
 	int lastpos = sec->Size() - 3;
 	char l[50];
 	itoa(lastpos, l, 10);
-	sec->SetString(k, sec->GetString(l));
-	sec->RemoveByKey(l);
+	sec->SetString(numberStrToDelete, sec->GetString(l));
 	sec->RemoveAt(lastpos);
 	m_Units.DeleteString(sel);
 
 
 	UpdateDialog();
-	return;
 }
 
 void CTaskForce::OnChangeNumberunits()