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
116 *(current++) = '"';
117 while (arg < end) {
118 switch (*arg) {
119 case '\\':
120 case '"':
121 *(current++) = '\\'; // fallthrough
122 default:
123 *(current++) = *(arg++);
124 }
125 }
126 *(current++) = '"';
127 *(current++) = '\0';
128 result = (char *)realloc(result, current-result);
129 return result;
130 }
131
132
sendMsg(const char * msg)133 int SocketClient::sendMsg(const char *msg) {
134 if (mSocket < 0) {
135 errno = EHOSTUNREACH;
136 return -1;
137 }
138
139 // Send the message including null character
140 if (sendData(msg, strlen(msg) + 1) != 0) {
141 SLOGW("Unable to send msg '%s'", msg);
142 return -1;
143 }
144 return 0;
145 }
146
sendData(const void * data,int len)147 int SocketClient::sendData(const void *data, int len) {
148
149 pthread_mutex_lock(&mWriteMutex);
150 int rc = sendDataLocked(data, len);
151 pthread_mutex_unlock(&mWriteMutex);
152
153 return rc;
154 }
155
sendDataLocked(const void * data,int len)156 int SocketClient::sendDataLocked(const void *data, int len) {
157 int rc = 0;
158 const char *p = (const char*) data;
159 int brtw = len;
160
161 if (len == 0) {
162 return 0;
163 }
164
165 while (brtw > 0) {
166 rc = send(mSocket, p, brtw, MSG_NOSIGNAL);
167 if (rc > 0) {
168 p += rc;
169 brtw -= rc;
170 continue;
171 }
172
173 if (rc < 0 && errno == EINTR)
174 continue;
175
176 if (rc == 0) {
177 SLOGW("0 length write :(");
178 errno = EIO;
179 } else {
180 SLOGW("write error (%s)", strerror(errno));
181 }
182 return -1;
183 }
184 return 0;
185 }
186
incRef()187 void SocketClient::incRef() {
188 pthread_mutex_lock(&mRefCountMutex);
189 mRefCount++;
190 pthread_mutex_unlock(&mRefCountMutex);
191 }
192
decRef()193 bool SocketClient::decRef() {
194 bool deleteSelf = false;
195 pthread_mutex_lock(&mRefCountMutex);
196 mRefCount--;
197 if (mRefCount == 0) {
198 deleteSelf = true;
199 } else if (mRefCount < 0) {
200 SLOGE("SocketClient refcount went negative!");
201 }
202 pthread_mutex_unlock(&mRefCountMutex);
203 if (deleteSelf) {
204 delete this;
205 }
206 return deleteSelf;
207 }
208