From 7b37b95164a6bd9006dc8bd6d20f3f674834e4a2 Mon Sep 17 00:00:00 2001
From: cwx1272435 <caiyiming4@huawei.com>
Date: Fri, 6 Sep 2024 05:39:59 +0000
Subject: [PATCH] Add mutex locks to functions that lack mutex locks

Signed-off-by: cwx1272435 <caiyiming4@huawei.com>
---
 libpurgeablemem/cpp/src/purgeable_mem_base.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/libpurgeablemem/cpp/src/purgeable_mem_base.cpp b/libpurgeablemem/cpp/src/purgeable_mem_base.cpp
index 5fe2900..0c5a935 100644
--- a/libpurgeablemem/cpp/src/purgeable_mem_base.cpp
+++ b/libpurgeablemem/cpp/src/purgeable_mem_base.cpp
@@ -97,6 +97,7 @@ bool PurgeableMemBase::BeginRead()
 
 void PurgeableMemBase::EndRead()
 {
+    std::lock_guard<std::mutex> lock(dataLock_);
     if (isDataValid_) {
         Unpin();
     }
@@ -141,6 +142,7 @@ bool PurgeableMemBase::BeginWrite()
 
 void PurgeableMemBase::EndWrite()
 {
+    std::lock_guard<std::mutex> lock(dataLock_);
     PM_HILOG_DEBUG(LOG_CORE, "%{public}s %{public}s", __func__, ToString().c_str());
     Unpin();
 }
@@ -233,6 +235,7 @@ inline std::string PurgeableMemBase::ToString() const
 
 void PurgeableMemBase::SetRebuildSuccessCallback(std::function<void()> &callback)
 {
+    std::lock_guard<std::mutex> lock(dataLock_);
     if (builder_) {
         builder_->SetRebuildSuccessCallback(callback);
     }
@@ -240,11 +243,13 @@ void PurgeableMemBase::SetRebuildSuccessCallback(std::function<void()> &callback
 
 bool PurgeableMemBase::IsDataValid()
 {
+    std::lock_guard<std::mutex> lock(dataLock_);
     return isDataValid_;
 }
 
 void PurgeableMemBase::SetDataValid(bool target)
 {
+    std::lock_guard<std::mutex> lock(dataLock_);
     isDataValid_ = target;
 }
 } /* namespace PurgeableMem */
-- 
Gitee