SocketClient.cpp revision 4be4e69f0128b7d9b0a29651ef4b79d806ae3ce7
1fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat#include <alloca.h> 2fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat#include <errno.h> 330abb7234de2a9caa1add4b00a189436f0b24560Kenny Root#include <sys/socket.h> 4fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat#include <sys/types.h> 5fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat#include <pthread.h> 6d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat#include <string.h> 7fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat 8fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat#define LOG_TAG "SocketClient" 9fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat#include <cutils/log.h> 10fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat 11fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat#include <sysutils/SocketClient.h> 12fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat 1330abb7234de2a9caa1add4b00a189436f0b24560Kenny RootSocketClient::SocketClient(int socket) 1430abb7234de2a9caa1add4b00a189436f0b24560Kenny Root : mSocket(socket) 1530abb7234de2a9caa1add4b00a189436f0b24560Kenny Root , mPid(-1) 1630abb7234de2a9caa1add4b00a189436f0b24560Kenny Root , mUid(-1) 1730abb7234de2a9caa1add4b00a189436f0b24560Kenny Root , mGid(-1) 18648ebad883e7825353c841950dd7d78664c238e6Brad Fitzpatrick , mRefCount(1) 1930abb7234de2a9caa1add4b00a189436f0b24560Kenny Root{ 20fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat pthread_mutex_init(&mWriteMutex, NULL); 21648ebad883e7825353c841950dd7d78664c238e6Brad Fitzpatrick pthread_mutex_init(&mRefCountMutex, NULL); 2230abb7234de2a9caa1add4b00a189436f0b24560Kenny Root 2330abb7234de2a9caa1add4b00a189436f0b24560Kenny Root struct ucred creds; 2430abb7234de2a9caa1add4b00a189436f0b24560Kenny Root socklen_t szCreds = sizeof(creds); 2530abb7234de2a9caa1add4b00a189436f0b24560Kenny Root memset(&creds, 0, szCreds); 2630abb7234de2a9caa1add4b00a189436f0b24560Kenny Root 2730abb7234de2a9caa1add4b00a189436f0b24560Kenny Root int err = getsockopt(socket, SOL_SOCKET, SO_PEERCRED, &creds, &szCreds); 2830abb7234de2a9caa1add4b00a189436f0b24560Kenny Root if (err == 0) { 2930abb7234de2a9caa1add4b00a189436f0b24560Kenny Root mPid = creds.pid; 3030abb7234de2a9caa1add4b00a189436f0b24560Kenny Root mUid = creds.uid; 3130abb7234de2a9caa1add4b00a189436f0b24560Kenny Root mGid = creds.gid; 3230abb7234de2a9caa1add4b00a189436f0b24560Kenny Root } 33fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat} 34fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat 35db017545796747115b8797f03e662b0f398a7c7bSan Mehatint SocketClient::sendMsg(int code, const char *msg, bool addErrno) { 36d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat char *buf; 37af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner const char* arg; 38af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner const char* fmt; 39af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner char tmp[1]; 40af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner int len; 4103f0d27f6c49530a91402ed42f8ca4b2fda04b9fSan Mehat 42d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat if (addErrno) { 43af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner fmt = "%.3d %s (%s)"; 44af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner arg = strerror(errno); 45d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat } else { 46af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner fmt = "%.3d %s"; 47af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner arg = NULL; 48d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat } 49af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner /* Measure length of required buffer */ 50af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner len = snprintf(tmp, sizeof tmp, fmt, code, msg, arg); 51af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner /* Allocate in the stack, then write to it */ 52af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner buf = (char*)alloca(len+1); 53af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner snprintf(buf, len+1, fmt, code, msg, arg); 54af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner /* Send the zero-terminated message */ 55d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat return sendMsg(buf); 56d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat} 57d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat 58db017545796747115b8797f03e662b0f398a7c7bSan Mehatint SocketClient::sendMsg(const char *msg) { 59fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat if (mSocket < 0) { 60fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat errno = EHOSTUNREACH; 61fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat return -1; 62fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat } 63fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat 64c73a3a5771a2d29d1bae666bfde12f751d66fc96San Mehat // Send the message including null character 658c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick if (sendData(msg, strlen(msg) + 1) != 0) { 668c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick SLOGW("Unable to send msg '%s'", msg); 678c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick return -1; 688c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick } 698c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick return 0; 708c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick} 718c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick 728c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrickint SocketClient::sendData(const void* data, int len) { 73d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat int rc = 0; 748c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick const char *p = (const char*) data; 758c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick int brtw = len; 76d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat 7716ae478af46100a435c2875afa020721a59a411aBrad Fitzpatrick if (len == 0) { 7816ae478af46100a435c2875afa020721a59a411aBrad Fitzpatrick return 0; 7916ae478af46100a435c2875afa020721a59a411aBrad Fitzpatrick } 8016ae478af46100a435c2875afa020721a59a411aBrad Fitzpatrick 81fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat pthread_mutex_lock(&mWriteMutex); 828c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick while (brtw > 0) { 83af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner rc = write(mSocket, p, brtw); 84af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner if (rc > 0) { 85af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner p += rc; 86af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner brtw -= rc; 87af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner continue; 88af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner } 89af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner 90af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner if (rc < 0 && errno == EINTR) 91af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner continue; 92af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner 93af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner pthread_mutex_unlock(&mWriteMutex); 94af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner if (rc == 0) { 957e8529a8b528fd30586aa037f15a31b29582c537San Mehat SLOGW("0 length write :("); 96d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat errno = EIO; 97af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner } else { 98af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner SLOGW("write error (%s)", strerror(errno)); 99d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat } 100af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner return -1; 101fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat } 102fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat pthread_mutex_unlock(&mWriteMutex); 103fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat return 0; 104fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat} 105648ebad883e7825353c841950dd7d78664c238e6Brad Fitzpatrick 106648ebad883e7825353c841950dd7d78664c238e6Brad Fitzpatrickvoid SocketClient::incRef() { 1074be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick pthread_mutex_lock(&mRefCountMutex); 1084be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick mRefCount++; 1094be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick pthread_mutex_unlock(&mRefCountMutex); 110648ebad883e7825353c841950dd7d78664c238e6Brad Fitzpatrick} 111648ebad883e7825353c841950dd7d78664c238e6Brad Fitzpatrick 1124be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrickbool SocketClient::decRef() { 1134be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick bool deleteSelf = false; 1144be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick pthread_mutex_lock(&mRefCountMutex); 1154be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick mRefCount--; 1164be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick if (mRefCount == 0) { 1174be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick deleteSelf = true; 1184be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick } else if (mRefCount < 0) { 1194be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick SLOGE("SocketClient refcount went negative!"); 1204be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick } 1214be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick pthread_mutex_unlock(&mRefCountMutex); 1224be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick if (deleteSelf) { 1234be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick delete this; 1244be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick } 1254be4e69f0128b7d9b0a29651ef4b79d806ae3ce7Brad Fitzpatrick return deleteSelf; 126648ebad883e7825353c841950dd7d78664c238e6Brad Fitzpatrick} 127