San Mehat | fa644ff | 2009-05-08 11:15:53 -0700 | [diff] [blame] | 1 | #include <alloca.h> |
| 2 | #include <errno.h> |
Kenny Root | 30abb72 | 2010-09-14 14:26:12 -0700 | [diff] [blame] | 3 | #include <sys/socket.h> |
San Mehat | fa644ff | 2009-05-08 11:15:53 -0700 | [diff] [blame] | 4 | #include <sys/types.h> |
| 5 | #include <pthread.h> |
San Mehat | d768066 | 2009-05-12 11:16:59 -0700 | [diff] [blame] | 6 | #include <string.h> |
San Mehat | fa644ff | 2009-05-08 11:15:53 -0700 | [diff] [blame] | 7 | |
| 8 | #define LOG_TAG "SocketClient" |
| 9 | #include <cutils/log.h> |
| 10 | |
| 11 | #include <sysutils/SocketClient.h> |
| 12 | |
Robert Greenwalt | 8702bb1 | 2012-02-07 12:23:14 -0800 | [diff] [blame^] | 13 | SocketClient::SocketClient(int socket, bool owned) { |
| 14 | init(socket, owned, false); |
| 15 | } |
| 16 | |
| 17 | SocketClient::SocketClient(int socket, bool owned, bool useCmdNum) { |
| 18 | init(socket, owned, useCmdNum); |
| 19 | } |
| 20 | |
| 21 | void SocketClient::init(int socket, bool owned, bool useCmdNum) { |
| 22 | mSocket = socket; |
| 23 | mSocketOwned = owned; |
| 24 | mUseCmdNum = useCmdNum; |
San Mehat | fa644ff | 2009-05-08 11:15:53 -0700 | [diff] [blame] | 25 | pthread_mutex_init(&mWriteMutex, NULL); |
Brad Fitzpatrick | 648ebad | 2011-03-17 15:41:20 -0700 | [diff] [blame] | 26 | pthread_mutex_init(&mRefCountMutex, NULL); |
Robert Greenwalt | 8702bb1 | 2012-02-07 12:23:14 -0800 | [diff] [blame^] | 27 | mPid = -1; |
| 28 | mUid = -1; |
| 29 | mGid = -1; |
| 30 | mRefCount = 1; |
| 31 | mCmdNum = 0; |
Kenny Root | 30abb72 | 2010-09-14 14:26:12 -0700 | [diff] [blame] | 32 | |
| 33 | struct ucred creds; |
| 34 | socklen_t szCreds = sizeof(creds); |
| 35 | memset(&creds, 0, szCreds); |
| 36 | |
| 37 | int err = getsockopt(socket, SOL_SOCKET, SO_PEERCRED, &creds, &szCreds); |
| 38 | if (err == 0) { |
| 39 | mPid = creds.pid; |
| 40 | mUid = creds.uid; |
| 41 | mGid = creds.gid; |
| 42 | } |
San Mehat | fa644ff | 2009-05-08 11:15:53 -0700 | [diff] [blame] | 43 | } |
| 44 | |
Xianzhu Wang | 4520246 | 2011-09-29 12:59:55 +0800 | [diff] [blame] | 45 | SocketClient::~SocketClient() |
| 46 | { |
| 47 | if (mSocketOwned) { |
| 48 | close(mSocket); |
| 49 | } |
| 50 | } |
| 51 | |
San Mehat | db01754 | 2009-05-20 15:27:14 -0700 | [diff] [blame] | 52 | int SocketClient::sendMsg(int code, const char *msg, bool addErrno) { |
Robert Greenwalt | 8702bb1 | 2012-02-07 12:23:14 -0800 | [diff] [blame^] | 53 | return sendMsg(code, msg, addErrno, mUseCmdNum); |
| 54 | } |
| 55 | |
| 56 | int SocketClient::sendMsg(int code, const char *msg, bool addErrno, bool useCmdNum) { |
San Mehat | d768066 | 2009-05-12 11:16:59 -0700 | [diff] [blame] | 57 | char *buf; |
Robert Greenwalt | 8702bb1 | 2012-02-07 12:23:14 -0800 | [diff] [blame^] | 58 | int ret = 0; |
San Mehat | 03f0d27 | 2009-05-26 15:18:25 -0700 | [diff] [blame] | 59 | |
San Mehat | d768066 | 2009-05-12 11:16:59 -0700 | [diff] [blame] | 60 | if (addErrno) { |
Robert Greenwalt | 8702bb1 | 2012-02-07 12:23:14 -0800 | [diff] [blame^] | 61 | if (useCmdNum) { |
| 62 | ret = asprintf(&buf, "%d %d %s (%s)", code, getCmdNum(), msg, strerror(errno)); |
| 63 | } else { |
| 64 | ret = asprintf(&buf, "%d %s (%s)", code, msg, strerror(errno)); |
| 65 | } |
San Mehat | d768066 | 2009-05-12 11:16:59 -0700 | [diff] [blame] | 66 | } else { |
Robert Greenwalt | 8702bb1 | 2012-02-07 12:23:14 -0800 | [diff] [blame^] | 67 | if (useCmdNum) { |
| 68 | ret = asprintf(&buf, "%d %d %s", code, getCmdNum(), msg); |
| 69 | } else { |
| 70 | ret = asprintf(&buf, "%d %s", code, msg); |
| 71 | } |
San Mehat | d768066 | 2009-05-12 11:16:59 -0700 | [diff] [blame] | 72 | } |
David 'Digit' Turner | af61509 | 2011-01-17 02:34:15 +0100 | [diff] [blame] | 73 | /* Send the zero-terminated message */ |
Robert Greenwalt | 8702bb1 | 2012-02-07 12:23:14 -0800 | [diff] [blame^] | 74 | if (ret != -1) { |
| 75 | ret = sendMsg(buf); |
| 76 | free(buf); |
| 77 | } |
| 78 | return ret; |
San Mehat | d768066 | 2009-05-12 11:16:59 -0700 | [diff] [blame] | 79 | } |
| 80 | |
San Mehat | db01754 | 2009-05-20 15:27:14 -0700 | [diff] [blame] | 81 | int SocketClient::sendMsg(const char *msg) { |
San Mehat | fa644ff | 2009-05-08 11:15:53 -0700 | [diff] [blame] | 82 | if (mSocket < 0) { |
| 83 | errno = EHOSTUNREACH; |
| 84 | return -1; |
| 85 | } |
| 86 | |
San Mehat | c73a3a5 | 2009-06-15 14:06:03 -0700 | [diff] [blame] | 87 | // Send the message including null character |
Brad Fitzpatrick | 8c5669f | 2010-10-27 10:23:16 -0700 | [diff] [blame] | 88 | if (sendData(msg, strlen(msg) + 1) != 0) { |
| 89 | SLOGW("Unable to send msg '%s'", msg); |
| 90 | return -1; |
| 91 | } |
| 92 | return 0; |
| 93 | } |
| 94 | |
| 95 | int SocketClient::sendData(const void* data, int len) { |
San Mehat | d768066 | 2009-05-12 11:16:59 -0700 | [diff] [blame] | 96 | int rc = 0; |
Brad Fitzpatrick | 8c5669f | 2010-10-27 10:23:16 -0700 | [diff] [blame] | 97 | const char *p = (const char*) data; |
| 98 | int brtw = len; |
San Mehat | d768066 | 2009-05-12 11:16:59 -0700 | [diff] [blame] | 99 | |
Brad Fitzpatrick | 16ae478 | 2010-11-02 10:55:52 -0700 | [diff] [blame] | 100 | if (len == 0) { |
| 101 | return 0; |
| 102 | } |
| 103 | |
San Mehat | fa644ff | 2009-05-08 11:15:53 -0700 | [diff] [blame] | 104 | pthread_mutex_lock(&mWriteMutex); |
Brad Fitzpatrick | 8c5669f | 2010-10-27 10:23:16 -0700 | [diff] [blame] | 105 | while (brtw > 0) { |
David 'Digit' Turner | af61509 | 2011-01-17 02:34:15 +0100 | [diff] [blame] | 106 | rc = write(mSocket, p, brtw); |
| 107 | if (rc > 0) { |
| 108 | p += rc; |
| 109 | brtw -= rc; |
| 110 | continue; |
| 111 | } |
| 112 | |
| 113 | if (rc < 0 && errno == EINTR) |
| 114 | continue; |
| 115 | |
| 116 | pthread_mutex_unlock(&mWriteMutex); |
| 117 | if (rc == 0) { |
San Mehat | 7e8529a | 2010-03-25 09:31:42 -0700 | [diff] [blame] | 118 | SLOGW("0 length write :("); |
San Mehat | d768066 | 2009-05-12 11:16:59 -0700 | [diff] [blame] | 119 | errno = EIO; |
David 'Digit' Turner | af61509 | 2011-01-17 02:34:15 +0100 | [diff] [blame] | 120 | } else { |
| 121 | SLOGW("write error (%s)", strerror(errno)); |
San Mehat | d768066 | 2009-05-12 11:16:59 -0700 | [diff] [blame] | 122 | } |
David 'Digit' Turner | af61509 | 2011-01-17 02:34:15 +0100 | [diff] [blame] | 123 | return -1; |
San Mehat | fa644ff | 2009-05-08 11:15:53 -0700 | [diff] [blame] | 124 | } |
| 125 | pthread_mutex_unlock(&mWriteMutex); |
| 126 | return 0; |
| 127 | } |
Brad Fitzpatrick | 648ebad | 2011-03-17 15:41:20 -0700 | [diff] [blame] | 128 | |
| 129 | void SocketClient::incRef() { |
Brad Fitzpatrick | 4be4e69 | 2011-03-17 17:14:46 -0700 | [diff] [blame] | 130 | pthread_mutex_lock(&mRefCountMutex); |
| 131 | mRefCount++; |
| 132 | pthread_mutex_unlock(&mRefCountMutex); |
Brad Fitzpatrick | 648ebad | 2011-03-17 15:41:20 -0700 | [diff] [blame] | 133 | } |
| 134 | |
Brad Fitzpatrick | 4be4e69 | 2011-03-17 17:14:46 -0700 | [diff] [blame] | 135 | bool SocketClient::decRef() { |
| 136 | bool deleteSelf = false; |
| 137 | pthread_mutex_lock(&mRefCountMutex); |
| 138 | mRefCount--; |
| 139 | if (mRefCount == 0) { |
| 140 | deleteSelf = true; |
| 141 | } else if (mRefCount < 0) { |
| 142 | SLOGE("SocketClient refcount went negative!"); |
| 143 | } |
| 144 | pthread_mutex_unlock(&mRefCountMutex); |
| 145 | if (deleteSelf) { |
| 146 | delete this; |
| 147 | } |
| 148 | return deleteSelf; |
Brad Fitzpatrick | 648ebad | 2011-03-17 15:41:20 -0700 | [diff] [blame] | 149 | } |