Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 3eb0897d authored by Brad Fitzpatrick's avatar Brad Fitzpatrick Committed by Android (Google) Code Review
Browse files

Merge "SocketClient: add optional reference counting"

parents 9b53946d 648ebad8

include/sysutils/SocketClient.h

+11 −0
Original line number Diff line number Diff line
@@ -19,6 +19,10 @@ class SocketClient { 
    /* Peer group ID */

    gid_t mGid;



    /* Reference count (starts at 1) */

    pthread_mutex_t mRefCountMutex;

    int mRefCount;



public:

    SocketClient(int sock);

    virtual ~SocketClient() {}
@@ -34,6 +38,13 @@ public: 


    // Sending binary data:

    int sendData(const void *data, int len);



    // Optional reference counting.  Reference count starts at 1.  If

    // it's decremented to 0, it deletes itself.

    // SocketListener creates a SocketClient (at refcount 1) and calls

    // decRef() when it's done with the client.

    void incRef();

    void decRef();

};



typedef android::List<SocketClient *> SocketClientCollection;

libsysutils/src/SocketClient.cpp

+21 −0
Original line number Diff line number Diff line
@@ -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;

  }

}

libsysutils/src/SocketListener.cpp

+2 −2
Original line number Diff line number Diff line
@@ -55,7 +55,7 @@ SocketListener::~SocketListener() { 
    }

    SocketClientCollection::iterator it;

    for (it = mClients->begin(); it != mClients->end();) {

        delete (*it);

        (*it)->decRef();

        it = mClients->erase(it);

    }

    delete mClients;
@@ -226,7 +226,7 @@ void SocketListener::runListener() { 
                pthread_mutex_unlock(&mClientsLock);

                /* Destroy the client */

                close(c->getSocket());

                delete c;

                c->decRef();

            }

        }

    }