1 #include <alloca.h>
2 #include <errno.h>
3 #include <sys/socket.h>
4 #include <sys/types.h>
5 #include <pthread.h>
6 #include <string.h>
7 #include <arpa/inet.h>
8
9 #define LOG_TAG "SocketClient"
10 #include <cutils/log.h>
11
12 #include <sysutils/SocketClient.h>
13
SocketClient(int socket,bool owned)14 SocketClient::SocketClient(int socket, bool owned) {
15 init(socket, owned, false);
16 }
17
SocketClient(int socket,bool owned,bool useCmdNum)18 SocketClient::SocketClient(int socket, bool owned, bool useCmdNum) {
19 init(socket, owned, useCmdNum);
20 }
21
init(int socket,bool owned,bool useCmdNum)22 void SocketClient::init(int socket, bool owned, bool useCmdNum) {
23 mSocket = socket;
24 mSocketOwned = owned;
25 mUseCmdNum = useCmdNum;
26 pthread_mutex_init(&mWriteMutex, NULL);
27 pthread_mutex_init(&mRefCountMutex, NULL);
28 mPid = -1;
29 mUid = -1;
30 mGid = -1;
31 mRefCount = 1;
32 mCmdNum = 0;
33
34 struct ucred creds;
35 socklen_t szCreds = sizeof(creds);
36 memset(&creds, 0, szCreds);
37
38 int err = getsockopt(socket, SOL_SOCKET, SO_PEERCRED, &creds, &szCreds);
39 if (err == 0) {
40 mPid = creds.pid;
41 mUid = creds.uid;
42 mGid = creds.gid;
43 }
44 }
45
~SocketClient()46 SocketClient::~SocketClient()
47 {
48 if (mSocketOwned) {
49 close(mSocket);
50 }
51 }
52
sendMsg(int code,const char * msg,bool addErrno)53 int SocketClient::sendMsg(int code, const char *msg, bool addErrno) {
54 return sendMsg(code, msg, addErrno, mUseCmdNum);
55 }
56
sendMsg(int code,const char * msg,bool addErrno,bool useCmdNum)57 int SocketClient::sendMsg(int code, const char *msg, bool addErrno, bool useCmdNum) {
58 char *buf;
59 int ret = 0;
60
61 if (addErrno) {
62 if (useCmdNum) {
63 ret = asprintf(&buf, "%d %d %s (%s)", code, getCmdNum(), msg, strerror(errno));
64 } else {
65 ret = asprintf(&buf, "%d %s (%s)", code, msg, strerror(errno));
66 }
67 } else {
68 if (useCmdNum) {
69 ret = asprintf(&buf, "%d %d %s", code, getCmdNum(), msg);
70 } else {
71 ret = asprintf(&buf, "%d %s", code, msg);
72 }
73 }
74 /* Send the zero-terminated message */
75 if (ret != -1) {
76 ret = sendMsg(buf);
77 free(buf);
78 }
79 return ret;
80 }
81
82 /** send 3-digit code, null, binary-length, binary data */
sendBinaryMsg(int code,const void * data,int len)83 int SocketClient::sendBinaryMsg(int code, const void *data, int len) {
84
85 /* 4 bytes for the code & null + 4 bytes for the len */
86 char buf[8];
87 /* Write the code */
88 snprintf(buf, 4, "%.3d", code);
89 /* Write the len */
90 uint32_t tmp = htonl(len);
91 memcpy(buf + 4, &tmp, sizeof(uint32_t));
92
93 pthread_mutex_lock(&mWriteMutex);
94 int result = sendDataLocked(buf, sizeof(buf));
95 if (result == 0 && len > 0) {
96 result = sendDataLocked(data, len);
97 }
98 pthread_mutex_unlock(&mWriteMutex);
99
100 return result;
101 }
102
103 // Sends the code (c-string null-terminated).
sendCode(int code)104 int SocketClient::sendCode(int code) {
105 char buf[4];
106 snprintf(buf, sizeof(buf), "%.3d", code);
107 return sendData(buf, sizeof(buf));
108 }
109
quoteArg(const char * arg)110 char *SocketClient::quoteArg(const char *arg) {
111 int len = strlen(arg);
112 char *result = (char *)malloc(len * 2 + 3);
113 char *current = result;
114 const char *end = arg + len;
115 char *oldresult;
116
117 if(result == NULL) {
118 SLOGW("malloc error (%s)", strerror(errno));
119 return NULL;
120 }
121
122 *(current++) = '"';
123 while (arg < end) {
124 switch (*arg) {
125 case '\\':
126 case '"':
127 *(current++) = '\\'; // fallthrough
128 default:
129 *(current++) = *(arg++);
130 }
131 }
132 *(current++) = '"';
133 *(current++) = '\0';
134 oldresult = result; // save pointer in case realloc fails
135 result = (char *)realloc(result, current-result);
136 return result ? result : oldresult;
137 }
138
139
sendMsg(const char * msg)140 int SocketClient::sendMsg(const char *msg) {
141 // Send the message including null character
142 if (sendData(msg, strlen(msg) + 1) != 0) {
143 SLOGW("Unable to send msg '%s'", msg);
144 return -1;
145 }
146 return 0;
147 }
148
sendData(const void * data,int len)149 int SocketClient::sendData(const void *data, int len) {
150
151 pthread_mutex_lock(&mWriteMutex);
152 int rc = sendDataLocked(data, len);
153 pthread_mutex_unlock(&mWriteMutex);
154
155 return rc;
156 }
157
sendDataLocked(const void * data,int len)158 int SocketClient::sendDataLocked(const void *data, int len) {
159 int rc = 0;
160 const char *p = (const char*) data;
161 int brtw = len;
162
163 if (mSocket < 0) {
164 errno = EHOSTUNREACH;
165 return -1;
166 }
167
168 if (len == 0) {
169 return 0;
170 }
171
172 while (brtw > 0) {
173 rc = send(mSocket, p, brtw, MSG_NOSIGNAL);
174 if (rc > 0) {
175 p += rc;
176 brtw -= rc;
177 continue;
178 }
179
180 if (rc < 0 && errno == EINTR)
181 continue;
182
183 if (rc == 0) {
184 SLOGW("0 length write :(");
185 errno = EIO;
186 } else {
187 SLOGW("write error (%s)", strerror(errno));
188 }
189 return -1;
190 }
191 return 0;
192 }
193
incRef()194 void SocketClient::incRef() {
195 pthread_mutex_lock(&mRefCountMutex);
196 mRefCount++;
197 pthread_mutex_unlock(&mRefCountMutex);
198 }
199
decRef()200 bool SocketClient::decRef() {
201 bool deleteSelf = false;
202 pthread_mutex_lock(&mRefCountMutex);
203 mRefCount--;
204 if (mRefCount == 0) {
205 deleteSelf = true;
206 } else if (mRefCount < 0) {
207 SLOGE("SocketClient refcount went negative!");
208 }
209 pthread_mutex_unlock(&mRefCountMutex);
210 if (deleteSelf) {
211 delete this;
212 }
213 return deleteSelf;
214 }
215