1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
/*
* Copyright 2011, The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <stdlib.h>
#include <unistd.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <netinet/in.h>
#include <cutils/log.h>
#include "gltrace_transport.h"
namespace android {
namespace gltrace {
int acceptClientConnection(char *sockname) {
int serverSocket = socket(AF_LOCAL, SOCK_STREAM, 0);
if (serverSocket < 0) {
ALOGE("Error (%d) while creating socket. Check if app has network permissions.",
serverSocket);
return -1;
}
struct sockaddr_un server, client;
memset(&server, 0, sizeof server);
server.sun_family = AF_UNIX;
// the first byte of sun_path should be '\0' for abstract namespace
strcpy(server.sun_path + 1, sockname);
// note that sockaddr_len should be set to the exact size of the buffer that is used.
socklen_t sockaddr_len = sizeof(server.sun_family) + strlen(sockname) + 1;
if (bind(serverSocket, (struct sockaddr *) &server, sockaddr_len) < 0) {
close(serverSocket);
ALOGE("Failed to bind the server socket");
return -1;
}
if (listen(serverSocket, 1) < 0) {
close(serverSocket);
ALOGE("Failed to listen on server socket");
return -1;
}
ALOGD("gltrace::waitForClientConnection: server listening @ path %s", sockname);
int clientSocket = accept(serverSocket, (struct sockaddr *)&client, &sockaddr_len);
if (clientSocket < 0) {
close(serverSocket);
ALOGE("Failed to accept client connection");
return -1;
}
ALOGD("gltrace::waitForClientConnection: client connected.");
// do not accept any more incoming connections
close(serverSocket);
return clientSocket;
}
TCPStream::TCPStream(int socket) {
mSocket = socket;
pthread_mutex_init(&mSocketWriteMutex, NULL);
}
TCPStream::~TCPStream() {
pthread_mutex_destroy(&mSocketWriteMutex);
}
void TCPStream::closeStream() {
if (mSocket > 0) {
close(mSocket);
mSocket = 0;
}
}
int TCPStream::send(void *buf, size_t len) {
if (mSocket <= 0) {
return -1;
}
pthread_mutex_lock(&mSocketWriteMutex);
int n = write(mSocket, buf, len);
pthread_mutex_unlock(&mSocketWriteMutex);
return n;
}
int TCPStream::receive(void *data, size_t len) {
if (mSocket <= 0) {
return -1;
}
return read(mSocket, data, len);
}
BufferedOutputStream::BufferedOutputStream(TCPStream *stream, size_t bufferSize) {
mStream = stream;
mBufferSize = bufferSize;
mStringBuffer = "";
mStringBuffer.reserve(bufferSize);
}
int BufferedOutputStream::flush() {
if (mStringBuffer.size() == 0) {
return 0;
}
int n = mStream->send((void *)mStringBuffer.data(), mStringBuffer.size());
mStringBuffer.clear();
return n;
}
void BufferedOutputStream::enqueueMessage(GLMessage *msg) {
const uint32_t len = msg->ByteSize();
mStringBuffer.append((const char *)&len, sizeof(len)); // append header
msg->AppendToString(&mStringBuffer); // append message
}
int BufferedOutputStream::send(GLMessage *msg) {
enqueueMessage(msg);
if (mStringBuffer.size() > mBufferSize) {
return flush();
}
return 0;
}
}; // namespace gltrace
}; // namespace android
|
