BLEEndPoint ref count overflow check

project-chip · Mar 10, 2022 · e8cd248 · e8cd248
1 parent cd44f9b
commit e8cd248
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/ble/BLEEndPoint.cpp b/src/ble/BLEEndPoint.cpp
@@ -586,6 +586,16 @@ CHIP_ERROR BLEEndPoint::Init(BleLayer * bleLayer, BLE_CONNECTION_OBJECT connObj,
     return CHIP_NO_ERROR;
 }
 
+void BLEEndPoint::AddRef()
+{
+    if (mRefCount == UINT32_MAX)
+    {
+        return;
+    }
+
+    mRefCount++;
+}
+
 void BLEEndPoint::Release()
 {
     // Decrement the ref count.  When it reaches zero, NULL out the pointer to the chip::System::Layer

diff --git a/src/ble/BLEEndPoint.h b/src/ble/BLEEndPoint.h
@@ -114,7 +114,7 @@ class DLL_EXPORT BLEEndPoint
 
     uint32_t mRefCount;
 
-    void AddRef() { mRefCount++; }
+    void AddRef();
     void Release();
 
     // Private data members: