31
|
1 |
|
|
2 |
#include "NetworkTCP.hh"
|
38
|
3 |
#include "Engine.hh"
|
31
|
4 |
|
|
5 |
#include <cstdlib>
|
|
6 |
#include <cassert>
|
|
7 |
|
|
8 |
NetworkBuffer::NetworkBuffer (NetworkSocket &socket, size_t size_hint) :
|
|
9 |
socket(socket), buf(0), size(0), offset(0) {
|
|
10 |
|
|
11 |
// allocate initial buffer
|
|
12 |
if ((buf = (char *) malloc(size_hint)) == NULL)
|
37
|
13 |
throw NetworkBufferError("malloc failed");
|
31
|
14 |
|
|
15 |
// remember size
|
|
16 |
size = size_hint;
|
|
17 |
}
|
32
|
18 |
|
37
|
19 |
NetworkBuffer::~NetworkBuffer (void) {
|
32
|
20 |
free(buf);
|
|
21 |
}
|
31
|
22 |
|
37
|
23 |
void NetworkBuffer::resize (size_t item_size) {
|
32
|
24 |
size_t new_size = size;
|
|
25 |
|
|
26 |
// grow new_size until item_size fits
|
|
27 |
while (offset + item_size > new_size)
|
|
28 |
new_size *= 2;
|
31
|
29 |
|
32
|
30 |
// grow if needed
|
|
31 |
if (new_size != size) {
|
|
32 |
// realloc buffer
|
|
33 |
if ((buf = (char *) realloc((void *) buf, new_size)) == NULL)
|
37
|
34 |
throw NetworkBufferError("realloc failed");
|
32
|
35 |
|
|
36 |
// update size
|
|
37 |
size = new_size;
|
|
38 |
|
|
39 |
} else if (new_size > (offset + item_size) * 4) {
|
|
40 |
// XXX: shrink?
|
|
41 |
}
|
31
|
42 |
}
|
|
43 |
|
32
|
44 |
void NetworkBuffer::trim (size_t prefix_size) {
|
|
45 |
// update offset
|
37
|
46 |
offset -= prefix_size;
|
32
|
47 |
|
|
48 |
// shift the buffer forwards from (buf + prefix) -> (buf), copying (old_offset - prefix) bytes
|
37
|
49 |
memmove(buf, buf + prefix_size, offset);
|
32
|
50 |
}
|
|
51 |
|
37
|
52 |
bool NetworkBuffer::try_read (size_t item_size) {
|
32
|
53 |
int ret;
|
|
54 |
size_t to_read = item_size;
|
|
55 |
|
|
56 |
// keept reads at at least NETWORK_CHUNK_SIZE bytes
|
|
57 |
if (to_read < NETWORK_TCP_CHUNK_SIZE)
|
|
58 |
to_read = NETWORK_TCP_CHUNK_SIZE;
|
|
59 |
|
|
60 |
// resize buffer if needed
|
|
61 |
resize(to_read);
|
|
62 |
|
|
63 |
// read once
|
|
64 |
try {
|
|
65 |
ret = socket.recv(buf + offset, to_read);
|
|
66 |
|
|
67 |
} catch (CL_Error &e) {
|
|
68 |
if (errno == EAGAIN)
|
|
69 |
return false;
|
38
|
70 |
|
32
|
71 |
else
|
38
|
72 |
throw NetworkSocketOSError(socket, "recv");
|
32
|
73 |
}
|
38
|
74 |
|
|
75 |
// handle EOF
|
|
76 |
if (ret == 0)
|
|
77 |
throw NetworkSocketEOFError(socket, "recv");
|
32
|
78 |
|
38
|
79 |
assert(ret >= 0);
|
37
|
80 |
|
32
|
81 |
// update offset
|
|
82 |
offset += ret;
|
|
83 |
|
|
84 |
// did we get enough?
|
37
|
85 |
if ((unsigned int) ret < item_size)
|
32
|
86 |
return false;
|
|
87 |
else
|
|
88 |
return true;
|
|
89 |
}
|
|
90 |
|
37
|
91 |
bool NetworkBuffer::peek_prefix (uint16_t &ref) {
|
32
|
92 |
if (offset < sizeof(uint16_t))
|
|
93 |
return false;
|
|
94 |
|
37
|
95 |
ref = ntohs(*((uint16_t *) (buf)));
|
32
|
96 |
|
|
97 |
return true;
|
|
98 |
}
|
|
99 |
|
37
|
100 |
bool NetworkBuffer::peek_prefix (uint32_t &ref) {
|
32
|
101 |
if (offset < sizeof(uint32_t))
|
|
102 |
return false;
|
|
103 |
|
37
|
104 |
ref = ntohl(*((uint32_t *) (buf)));
|
32
|
105 |
|
|
106 |
return true;
|
|
107 |
}
|
|
108 |
|
37
|
109 |
template <typename PrefixType> PrefixType NetworkBuffer::read_prefix (char *buf_ptr, size_t buf_max) {
|
32
|
110 |
PrefixType prefix = 0;
|
|
111 |
size_t missing = 0;
|
|
112 |
|
|
113 |
do {
|
|
114 |
// do we have the prefix?
|
|
115 |
if (peek_prefix(prefix)) {
|
|
116 |
// do we already have the payload?
|
|
117 |
if (offset >= sizeof(PrefixType) + prefix) {
|
|
118 |
break;
|
|
119 |
|
|
120 |
} else {
|
|
121 |
missing = (sizeof(PrefixType) + prefix) - offset;
|
|
122 |
}
|
|
123 |
|
|
124 |
} else {
|
|
125 |
missing = sizeof(PrefixType);
|
|
126 |
}
|
|
127 |
|
|
128 |
// sanity-check
|
|
129 |
assert(missing);
|
|
130 |
|
|
131 |
// try and read the missing data
|
|
132 |
if (try_read(missing) == false) {
|
|
133 |
// if unable to read what we need, return zero.
|
|
134 |
return 0;
|
|
135 |
}
|
|
136 |
|
|
137 |
// assess the situation again
|
|
138 |
} while (true);
|
|
139 |
|
|
140 |
// copy the data over, unless it's too large
|
|
141 |
if (prefix <= buf_max) {
|
|
142 |
memcpy(buf_ptr, buf, prefix);
|
|
143 |
|
|
144 |
// trim the bytes out
|
38
|
145 |
trim(sizeof(PrefixType) + prefix);
|
32
|
146 |
|
|
147 |
// return
|
|
148 |
return prefix;
|
|
149 |
|
|
150 |
} else {
|
|
151 |
// trim the bytes out
|
38
|
152 |
trim(sizeof(PrefixType) + prefix);
|
32
|
153 |
|
37
|
154 |
throw NetworkBufferError("recv prefix overflow");
|
32
|
155 |
}
|
|
156 |
}
|
|
157 |
|
|
158 |
void NetworkBuffer::push_write (char *buf_ptr, size_t buf_size) {
|
31
|
159 |
int ret;
|
|
160 |
|
|
161 |
// try and short-circuit writes unless we have already buffered data
|
|
162 |
if (offset == 0) {
|
|
163 |
try {
|
|
164 |
// attempt to send something
|
|
165 |
ret = socket.send(buf_ptr, buf_size);
|
|
166 |
|
|
167 |
} catch (CL_Error &e) {
|
|
168 |
// ignore EAGAIN, detect this by setting ret to -1
|
|
169 |
if (errno != EAGAIN)
|
38
|
170 |
throw NetworkSocketOSError(socket, "send");
|
31
|
171 |
|
|
172 |
ret = -1;
|
|
173 |
}
|
|
174 |
|
|
175 |
// if we managed to send something, adjust buf/size and buffer
|
|
176 |
if (ret > 0) {
|
37
|
177 |
// sanity-check
|
|
178 |
assert(buf_size >= (unsigned int) ret);
|
|
179 |
|
31
|
180 |
buf_ptr += ret;
|
|
181 |
buf_size -= ret;
|
|
182 |
|
|
183 |
// if that was all, we're done
|
|
184 |
if (buf_size == 0)
|
|
185 |
return;
|
|
186 |
}
|
|
187 |
}
|
|
188 |
|
32
|
189 |
// resize to fit buf_size more bytes
|
|
190 |
resize(buf_size);
|
31
|
191 |
|
|
192 |
// copy into our internal buffer
|
|
193 |
memcpy(buf + offset, buf_ptr, buf_size);
|
|
194 |
}
|
|
195 |
|
|
196 |
void NetworkBuffer::flush_write (void) {
|
|
197 |
int ret;
|
|
198 |
|
|
199 |
// ignore if we don't have any data buffered
|
|
200 |
if (offset == 0)
|
|
201 |
return;
|
|
202 |
|
|
203 |
// attempt to write as much as possible
|
|
204 |
try {
|
37
|
205 |
ret = socket.send(buf, offset);
|
31
|
206 |
|
|
207 |
} catch (CL_Error &e) {
|
|
208 |
// ignore EAGAIN and just return
|
|
209 |
if (errno == EAGAIN)
|
|
210 |
return;
|
|
211 |
|
|
212 |
else
|
38
|
213 |
throw NetworkSocketOSError(socket, "send");
|
31
|
214 |
}
|
|
215 |
|
32
|
216 |
// trim the buffer
|
|
217 |
trim(ret);
|
|
218 |
}
|
|
219 |
|
38
|
220 |
void NetworkBuffer::write_prefix (char *buf, uint16_t prefix) {
|
|
221 |
uint16_t nval = htons(prefix);
|
|
222 |
|
|
223 |
push_write((char*) &nval, sizeof(uint16_t));
|
32
|
224 |
push_write(buf, prefix);
|
|
225 |
}
|
31
|
226 |
|
38
|
227 |
void NetworkBuffer::write_prefix (char *buf, uint32_t prefix) {
|
|
228 |
uint32_t nval = htonl(prefix);
|
|
229 |
|
|
230 |
push_write((char*) &nval, sizeof(uint32_t));
|
|
231 |
push_write(buf, prefix);
|
|
232 |
}
|
31
|
233 |
|
37
|
234 |
NetworkTCPTransport::NetworkTCPTransport (NetworkSocket socket) :
|
|
235 |
socket(socket), in(socket, NETWORK_TCP_INITIAL_IN_BUF), out(socket, NETWORK_TCP_INITIAL_OUT_BUF) {
|
31
|
236 |
|
|
237 |
// connect signals
|
|
238 |
slots.connect(socket.sig_read_triggered(), this, &NetworkTCPTransport::on_read);
|
|
239 |
slots.connect(socket.sig_write_triggered(), this, &NetworkTCPTransport::on_write);
|
37
|
240 |
slots.connect(socket.sig_disconnected(), this, &NetworkTCPTransport::on_disconnected);
|
31
|
241 |
}
|
|
242 |
|
|
243 |
|
|
244 |
void NetworkTCPTransport::on_read (void) {
|
32
|
245 |
uint16_t prefix;
|
31
|
246 |
NetworkPacket packet;
|
32
|
247 |
|
|
248 |
// let the in stream read length-prefixed packets and pass them on to handle_packet
|
37
|
249 |
while ((prefix = in.read_prefix<uint16_t>(packet.get_buf(), packet.get_buf_size())) > 0) {
|
32
|
250 |
packet.set_data_size(prefix);
|
34
|
251 |
_sig_packet(packet);
|
32
|
252 |
}
|
31
|
253 |
}
|
|
254 |
|
|
255 |
void NetworkTCPTransport::on_write (void) {
|
|
256 |
// just flush the output buffer
|
|
257 |
out.flush_write();
|
|
258 |
}
|
|
259 |
|
|
260 |
void NetworkTCPTransport::on_disconnected (void) {
|
32
|
261 |
// pass right through
|
34
|
262 |
_sig_disconnect();
|
31
|
263 |
}
|
|
264 |
|
|
265 |
void NetworkTCPTransport::write_packet (const NetworkPacket &packet) {
|
32
|
266 |
uint16_t prefix = packet.get_data_size();
|
|
267 |
|
|
268 |
if (prefix != packet.get_data_size())
|
|
269 |
throw CL_Error("send prefix overflow");
|
38
|
270 |
|
|
271 |
try {
|
|
272 |
// just write to the output buffer
|
|
273 |
out.write_prefix((char *) packet.get_buf(), prefix);
|
32
|
274 |
|
38
|
275 |
} catch (Error &e) {
|
|
276 |
const char *err = e.what();
|
|
277 |
|
|
278 |
Engine::log(ERROR, "tcp.write_packet") << err;
|
|
279 |
|
|
280 |
throw;
|
|
281 |
}
|
31
|
282 |
}
|
|
283 |
|
|
284 |
NetworkTCPServer::NetworkTCPServer (const NetworkAddress &listen_addr) :
|
37
|
285 |
socket(CL_Socket::tcp, CL_Socket::ipv4) {
|
31
|
286 |
|
|
287 |
// bind
|
|
288 |
socket.bind(listen_addr);
|
|
289 |
|
|
290 |
// assign slots
|
|
291 |
slots.connect(socket.sig_read_triggered(), this, &NetworkTCPServer::on_accept);
|
|
292 |
|
|
293 |
// listen
|
|
294 |
socket.listen(NETWORK_LISTEN_BACKLOG);
|
38
|
295 |
|
|
296 |
// use nonblocking sockets
|
|
297 |
socket.set_nonblocking(true);
|
31
|
298 |
}
|
|
299 |
|
|
300 |
|
|
301 |
void NetworkTCPServer::on_accept (void) {
|
|
302 |
// accept a new socket
|
|
303 |
NetworkSocket client_sock = socket.accept();
|
|
304 |
|
|
305 |
// create a new NetworkTCPTransport
|
33
|
306 |
NetworkTCPTransport *client = buildTransport(client_sock);
|
|
307 |
|
31
|
308 |
// let our user handle it
|
34
|
309 |
_sig_client(client);
|
31
|
310 |
}
|
|
311 |
|
37
|
312 |
NetworkTCPTransport* NetworkTCPServer::buildTransport (CL_Socket &socket) {
|
|
313 |
return new NetworkTCPTransport(socket);
|
33
|
314 |
}
|
|
315 |
|
31
|
316 |
NetworkTCPClient::NetworkTCPClient (const NetworkAddress &connect_addr) :
|
37
|
317 |
NetworkTCPTransport(NetworkSocket(CL_Socket::tcp, CL_Socket::ipv4)) {
|
31
|
318 |
|
|
319 |
// connect
|
|
320 |
socket.connect(connect_addr);
|
38
|
321 |
|
|
322 |
// use nonblocking sockets
|
|
323 |
socket.set_nonblocking(true);
|
31
|
324 |
}
|