SocketClient.cpp revision df6c1b91e3813886070f35929583c30cfaead918 
1#include <alloca.h>
2#include <errno.h>
3#include <sys/types.h>
4#include <pthread.h>
5#include <string.h>
6
7#define LOG_TAG "SocketClient"
8#include <cutils/log.h>
9
10#include <sysutils/SocketClient.h>
11
12SocketClient::SocketClient(int socket) {
13    mSocket = socket;
14    pthread_mutex_init(&mWriteMutex, NULL);
15}
16
17int SocketClient::sendMsg(int code, char *msg, bool addErrno) {
18    char *buf;
19
20    if (addErrno) {
21        buf = (char *) alloca(strlen(msg) + strlen(strerror(errno)) + 8);
22        sprintf(buf, "%.3d %s (%s)", code, msg, strerror(errno));
23    } else {
24        buf = (char *) alloca(strlen(msg) + strlen("XXX "));
25        sprintf(buf, "%.3d %s", code, msg);
26    }
27    return sendMsg(buf);
28}
29
30int SocketClient::sendMsg(char *msg) {
31    if (mSocket < 0) {
32        errno = EHOSTUNREACH;
33        return -1;
34    }
35
36    char *bp;
37
38    if (msg[strlen(msg)] != '\n') {
39        bp = (char *) alloca(strlen(msg) + 1);
40        strcpy(bp, msg);
41        strcat(bp, "\n");
42    } else
43        bp = msg;
44
45    int rc = 0;
46    char *p = bp;
47    int brtw = strlen(bp);
48
49    pthread_mutex_lock(&mWriteMutex);
50    while(brtw) {
51        if ((rc = write(mSocket,p, brtw)) < 0) {
52            LOGW("Unable to send msg '%s' (%s)", msg, strerror(errno));
53            pthread_mutex_unlock(&mWriteMutex);
54            return -1;
55        } else if (!rc) {
56            LOGW("0 length write :(");
57            errno = EIO;
58            pthread_mutex_unlock(&mWriteMutex);
59            return -1;
60        }
61        p += rc;
62        brtw -= rc;
63    }
64    pthread_mutex_unlock(&mWriteMutex);
65    return 0;
66}
67