SocketClient.cpp revision af61509b50bc110b0c7c4691e37873cc0987ab5c
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) 1830abb7234de2a9caa1add4b00a189436f0b24560Kenny Root{ 19fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat pthread_mutex_init(&mWriteMutex, NULL); 2030abb7234de2a9caa1add4b00a189436f0b24560Kenny Root 2130abb7234de2a9caa1add4b00a189436f0b24560Kenny Root struct ucred creds; 2230abb7234de2a9caa1add4b00a189436f0b24560Kenny Root socklen_t szCreds = sizeof(creds); 2330abb7234de2a9caa1add4b00a189436f0b24560Kenny Root memset(&creds, 0, szCreds); 2430abb7234de2a9caa1add4b00a189436f0b24560Kenny Root 2530abb7234de2a9caa1add4b00a189436f0b24560Kenny Root int err = getsockopt(socket, SOL_SOCKET, SO_PEERCRED, &creds, &szCreds); 2630abb7234de2a9caa1add4b00a189436f0b24560Kenny Root if (err == 0) { 2730abb7234de2a9caa1add4b00a189436f0b24560Kenny Root mPid = creds.pid; 2830abb7234de2a9caa1add4b00a189436f0b24560Kenny Root mUid = creds.uid; 2930abb7234de2a9caa1add4b00a189436f0b24560Kenny Root mGid = creds.gid; 3030abb7234de2a9caa1add4b00a189436f0b24560Kenny Root } 31fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat} 32fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat 33db017545796747115b8797f03e662b0f398a7c7bSan Mehatint SocketClient::sendMsg(int code, const char *msg, bool addErrno) { 34d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat char *buf; 35af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner const char* arg; 36af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner const char* fmt; 37af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner char tmp[1]; 38af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner int len; 3903f0d27f6c49530a91402ed42f8ca4b2fda04b9fSan Mehat 40d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat if (addErrno) { 41af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner fmt = "%.3d %s (%s)"; 42af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner arg = strerror(errno); 43d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat } else { 44af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner fmt = "%.3d %s"; 45af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner arg = NULL; 46d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat } 47af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner /* Measure length of required buffer */ 48af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner len = snprintf(tmp, sizeof tmp, fmt, code, msg, arg); 49af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner /* Allocate in the stack, then write to it */ 50af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner buf = (char*)alloca(len+1); 51af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner snprintf(buf, len+1, fmt, code, msg, arg); 52af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner /* Send the zero-terminated message */ 53d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat return sendMsg(buf); 54d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat} 55d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat 56db017545796747115b8797f03e662b0f398a7c7bSan Mehatint SocketClient::sendMsg(const char *msg) { 57fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat if (mSocket < 0) { 58fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat errno = EHOSTUNREACH; 59fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat return -1; 60fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat } 61fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat 62c73a3a5771a2d29d1bae666bfde12f751d66fc96San Mehat // Send the message including null character 638c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick if (sendData(msg, strlen(msg) + 1) != 0) { 648c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick SLOGW("Unable to send msg '%s'", msg); 658c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick return -1; 668c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick } 678c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick return 0; 688c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick} 698c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick 708c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrickint SocketClient::sendData(const void* data, int len) { 71d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat int rc = 0; 728c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick const char *p = (const char*) data; 738c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick int brtw = len; 74d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat 7516ae478af46100a435c2875afa020721a59a411aBrad Fitzpatrick if (len == 0) { 7616ae478af46100a435c2875afa020721a59a411aBrad Fitzpatrick return 0; 7716ae478af46100a435c2875afa020721a59a411aBrad Fitzpatrick } 7816ae478af46100a435c2875afa020721a59a411aBrad Fitzpatrick 79fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat pthread_mutex_lock(&mWriteMutex); 808c5669f9f9a228efebf4059fd4ceace5cece578bBrad Fitzpatrick while (brtw > 0) { 81af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner rc = write(mSocket, p, brtw); 82af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner if (rc > 0) { 83af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner p += rc; 84af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner brtw -= rc; 85af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner continue; 86af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner } 87af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner 88af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner if (rc < 0 && errno == EINTR) 89af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner continue; 90af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner 91af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner pthread_mutex_unlock(&mWriteMutex); 92af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner if (rc == 0) { 937e8529a8b528fd30586aa037f15a31b29582c537San Mehat SLOGW("0 length write :("); 94d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat errno = EIO; 95af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner } else { 96af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner SLOGW("write error (%s)", strerror(errno)); 97d768066ef54270a0d3ccfccd50ae8238db5a2cddSan Mehat } 98af61509b50bc110b0c7c4691e37873cc0987ab5cDavid 'Digit' Turner return -1; 99fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat } 100fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat pthread_mutex_unlock(&mWriteMutex); 101fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat return 0; 102fa644ffe944c01a9b00f8d7676d58394fabee285San Mehat} 103