Merge "SocketClient: add optional reference counting"

1 files changed, 21 insertions, 0 deletions

diff --git a/libsysutils/src/SocketClient.cpp b/libsysutils/src/SocketClient.cpp
index a6aed265d..6d4dff45d 100644
--- a/libsysutils/src/SocketClient.cpp
+++ b/libsysutils/src/SocketClient.cpp
@@ -15,8 +15,10 @@ SocketClient::SocketClient(int socket)
         , mPid(-1)
         , mUid(-1)
         , mGid(-1)
+        , mRefCount(1)
 {
     pthread_mutex_init(&mWriteMutex, NULL);
+    pthread_mutex_init(&mRefCountMutex, NULL);
 
     struct ucred creds;
     socklen_t szCreds = sizeof(creds);
@@ -100,3 +102,22 @@ int SocketClient::sendData(const void* data, int len) {
     pthread_mutex_unlock(&mWriteMutex);
     return 0;
 }
+
+void SocketClient::incRef() {
+  pthread_mutex_lock(&mRefCountMutex);
+  mRefCount++;
+  pthread_mutex_unlock(&mRefCountMutex);
+}
+
+void SocketClient::decRef() {
+  bool deleteSelf = false;
+  pthread_mutex_lock(&mRefCountMutex);
+  mRefCount--;
+  if (mRefCount == 0) {
+      deleteSelf = true;
+  }
+  pthread_mutex_unlock(&mRefCountMutex);
+  if (deleteSelf) {
+      delete this;
+  }
+}