Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(80)

Side by Side Diff: runtime/bin/sync_socket.cc

Issue 2803543006: Added synchronous socket implementation to dart:io. (Closed)
Patch Set: Addressed second round of comments Created 3 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 // Copyright (c) 2017, the Dart project authors. Please see the AUTHORS file
2 // for details. All rights reserved. Use of this source code is governed by a
3 // BSD-style license that can be found in the LICENSE file.
4
5 #if !defined(DART_IO_DISABLED)
6
7 #include "bin/sync_socket.h"
8
9 #include "bin/dartutils.h"
10 #include "bin/io_buffer.h"
11 #include "bin/isolate_data.h"
12 #include "bin/lockers.h"
13 #include "bin/thread.h"
14 #include "bin/utils.h"
15
16 #include "include/dart_api.h"
17
18 #include "platform/globals.h"
19 #include "platform/utils.h"
20
21 #define DART_CHECK_ERROR_AND_CLEANUP(handle, ptr) \
22 do { \
23 if (Dart_IsError((handle))) { \
24 delete (ptr); \
25 Dart_SetReturnValue(args, (handle)); \
26 return; \
27 } \
28 } while (0)
29
30 #define DART_CHECK_ERROR(handle) \
31 do { \
32 if (Dart_IsError((handle))) { \
33 Dart_SetReturnValue(args, (handle)); \
34 return; \
35 } \
36 } while (0)
37
38 namespace dart {
39 namespace bin {
40
41 static const int kSocketIdNativeField = 0;
42
43 void FUNCTION_NAME(SynchronousSocket_LookupRequest)(Dart_NativeArguments args) {
44 if (Dart_GetNativeArgumentCount(args) != 2) {
45 Dart_SetReturnValue(
46 args, DartUtils::NewDartArgumentError("Invalid argument count."));
47 return;
48 }
49
50 char* host = NULL;
zra 2017/04/11 16:19:40 nit: Maybe declare this closer to where it's used.
bkonyi 2017/04/11 18:11:17 Done.
51 char* peer = NULL;
52 int64_t type = 0;
zra 2017/04/11 16:19:41 ditto
bkonyi 2017/04/11 18:11:18 Done.
53
54 Dart_Handle host_arg =
55 Dart_GetNativeStringArgument(args, 0, reinterpret_cast<void**>(&peer));
56 if (Dart_IsError(host_arg)) {
zra 2017/04/11 16:19:40 You can use your macro here.
bkonyi 2017/04/11 18:11:18 Done.
57 Dart_SetReturnValue(args, host_arg);
58 return;
59 }
60
61 host_arg = Dart_StringToCString(host_arg, const_cast<const char**>(&host));
62 if (Dart_IsError(host_arg)) {
zra 2017/04/11 16:19:40 ditto
bkonyi 2017/04/11 18:11:18 Done.
63 Dart_SetReturnValue(args, host_arg);
64 return;
65 }
66
67 Dart_Handle port_error = Dart_GetNativeIntegerArgument(args, 1, &type);
68 if (Dart_IsError(port_error)) {
zra 2017/04/11 16:19:40 ditto
bkonyi 2017/04/11 18:11:18 Done.
69 Dart_SetReturnValue(args, port_error);
70 return;
71 }
72
73 OSError* os_error = NULL;
74 AddressList<SocketAddress>* addresses =
75 SocketBase::LookupAddress(host, type, &os_error);
76
zra 2017/04/11 16:19:40 nit: This whitespace is allowed by the style guide
bkonyi 2017/04/11 18:11:18 Makes sense. I've removed the whitespace.
77 if (addresses == NULL) {
78 Dart_SetReturnValue(args, DartUtils::NewDartOSError(os_error));
79 return;
80 }
81
82 Dart_Handle array = Dart_NewList(addresses->count());
83 Dart_Handle error;
zra 2017/04/11 16:19:40 nit: It's a little confusing to declare this betwe
bkonyi 2017/04/11 18:11:18 I agree. I've moved the declaration to the first t
84 DART_CHECK_ERROR_AND_CLEANUP(array, addresses);
85
86 for (intptr_t i = 0; i < addresses->count(); i++) {
87 SocketAddress* addr = addresses->GetAt(i);
88 Dart_Handle entry = Dart_NewList(3);
89 DART_CHECK_ERROR_AND_CLEANUP(entry, addresses);
90
91 Dart_Handle type = Dart_NewInteger(addr->GetType());
92 DART_CHECK_ERROR_AND_CLEANUP(type, addresses);
93 error = Dart_ListSetAt(entry, 0, type);
94 DART_CHECK_ERROR_AND_CLEANUP(error, addresses);
95
96 Dart_Handle as_string = Dart_NewStringFromCString(addr->as_string());
97 DART_CHECK_ERROR_AND_CLEANUP(as_string, addresses);
98 error = Dart_ListSetAt(entry, 1, as_string);
99 DART_CHECK_ERROR_AND_CLEANUP(error, addresses);
100
101 RawAddr raw = addr->addr();
102 Dart_Handle data = SocketAddress::ToTypedData(raw);
103 DART_CHECK_ERROR_AND_CLEANUP(data, addresses);
104
105 error = Dart_ListSetAt(entry, 2, data);
106 DART_CHECK_ERROR_AND_CLEANUP(error, addresses);
107 error = Dart_ListSetAt(array, i, entry);
108 DART_CHECK_ERROR_AND_CLEANUP(error, addresses);
109 }
110 delete addresses;
111 Dart_SetReturnValue(args, array);
112 return;
113 }
114
115
116 void FUNCTION_NAME(SynchronousSocket_CreateConnectSync)(
117 Dart_NativeArguments args) {
118 RawAddr addr;
119 SocketAddress::GetSockAddr(Dart_GetNativeArgument(args, 1), &addr);
120 Dart_Handle port_arg = Dart_GetNativeArgument(args, 2);
121 DART_CHECK_ERROR(port_arg);
122 int64_t port = DartUtils::GetInt64ValueCheckRange(port_arg, 0, 65535);
123 SocketAddress::SetAddrPort(&addr, static_cast<intptr_t>(port));
124 intptr_t socket = SynchronousSocket::CreateConnect(addr);
125 if (socket >= 0) {
126 SynchronousSocket::SetSocketIdNativeField(Dart_GetNativeArgument(args, 0),
127 socket);
128 Dart_SetBooleanReturnValue(args, true);
129 } else {
130 Dart_SetReturnValue(args, DartUtils::NewDartOSError());
131 }
132 }
133
134
135 void FUNCTION_NAME(SynchronousSocket_WriteList)(Dart_NativeArguments args) {
136 SynchronousSocket* socket = SynchronousSocket::GetSocketIdNativeField(
137 Dart_GetNativeArgument(args, 0));
138 Dart_Handle buffer_obj = Dart_GetNativeArgument(args, 1);
139 if (!Dart_IsList(buffer_obj)) {
140 Dart_SetReturnValue(args, DartUtils::NewDartArgumentError(
141 "First parameter must be a List<int>"));
142 return;
143 }
144 intptr_t offset = DartUtils::GetIntptrValue(Dart_GetNativeArgument(args, 2));
145 intptr_t length = DartUtils::GetIntptrValue(Dart_GetNativeArgument(args, 3));
146 Dart_TypedData_Type type;
147 uint8_t* buffer = NULL;
148 intptr_t len;
149 Dart_Handle result = Dart_TypedDataAcquireData(
150 buffer_obj, &type, reinterpret_cast<void**>(&buffer), &len);
151 DART_CHECK_ERROR(result);
152 ASSERT((offset + length) <= len);
153 buffer += offset;
154 intptr_t bytes_written = SocketBase::Write(socket->fd(), buffer, length);
155 if (bytes_written >= 0) {
156 Dart_TypedDataReleaseData(buffer_obj);
157 Dart_SetIntegerReturnValue(args, bytes_written);
158 } else {
159 // Extract OSError before we release data, as it may override the error.
160 OSError os_error;
161 Dart_TypedDataReleaseData(buffer_obj);
162 Dart_SetReturnValue(args, DartUtils::NewDartOSError(&os_error));
163 }
164 }
165
166
167 void FUNCTION_NAME(SynchronousSocket_ReadList)(Dart_NativeArguments args) {
168 SynchronousSocket* socket = SynchronousSocket::GetSocketIdNativeField(
169 Dart_GetNativeArgument(args, 0));
170 Dart_Handle buffer_obj = Dart_GetNativeArgument(args, 1);
171 if (!Dart_IsList(buffer_obj)) {
172 Dart_SetReturnValue(args, DartUtils::NewDartArgumentError(
173 "First parameter must be a List<int>"));
174 return;
175 }
176 intptr_t offset = DartUtils::GetIntptrValue(Dart_GetNativeArgument(args, 2));
177 intptr_t bytes = DartUtils::GetIntptrValue(Dart_GetNativeArgument(args, 3));
178 intptr_t array_len = 0;
179
180 Dart_Handle result = Dart_ListLength(buffer_obj, &array_len);
181 DART_CHECK_ERROR(result);
182
183 uint8_t* buffer = Dart_ScopeAllocate(bytes);
184 intptr_t bytes_read = SocketBase::Read(socket->fd(), buffer, bytes);
185 if (bytes_read >= 0) {
186 result = Dart_ListSetAsBytes(buffer_obj, offset, buffer, bytes_read);
187 DART_CHECK_ERROR(result);
188 Dart_SetIntegerReturnValue(args, bytes_read);
189 } else {
190 Dart_SetReturnValue(args, DartUtils::NewDartOSError());
191 }
192 }
193
194
195 void FUNCTION_NAME(SynchronousSocket_Available)(Dart_NativeArguments args) {
196 SynchronousSocket* socket = SynchronousSocket::GetSocketIdNativeField(
197 Dart_GetNativeArgument(args, 0));
198 intptr_t available = SocketBase::Available(socket->fd());
199 if (available >= 0) {
200 Dart_SetIntegerReturnValue(args, available);
201 } else {
202 Dart_SetReturnValue(args, DartUtils::NewDartOSError());
203 }
204 }
205
206
207 void FUNCTION_NAME(SynchronousSocket_CloseSync)(Dart_NativeArguments args) {
208 SynchronousSocket* socket = SynchronousSocket::GetSocketIdNativeField(
209 Dart_GetNativeArgument(args, 0));
210 SocketBase::Close(socket->fd());
211 socket->SetClosedFd();
212 }
213
214
215 void FUNCTION_NAME(SynchronousSocket_Read)(Dart_NativeArguments args) {
216 SynchronousSocket* socket = SynchronousSocket::GetSocketIdNativeField(
217 Dart_GetNativeArgument(args, 0));
218 int64_t length = 0;
219 if (!DartUtils::GetInt64Value(Dart_GetNativeArgument(args, 1), &length)) {
220 Dart_SetReturnValue(args, DartUtils::NewDartArgumentError(
221 "First parameter must be an integer."));
222 return;
223 }
224 uint8_t* buffer = NULL;
225 Dart_Handle result = IOBuffer::Allocate(length, &buffer);
226 if (Dart_IsError(result)) {
zra 2017/04/11 16:19:40 IOBuffer::Allocate can't return an error handle. (
bkonyi 2017/04/11 18:11:17 Acknowledged.
227 if (buffer != NULL) {
228 IOBuffer::Free(buffer);
229 }
230 Dart_SetReturnValue(args, result);
231 return;
232 }
233 ASSERT(buffer != NULL);
234 intptr_t bytes_read = SocketBase::Read(socket->fd(), buffer, length);
235 if (bytes_read == length) {
236 Dart_SetReturnValue(args, result);
237 } else if (bytes_read > 0) {
238 uint8_t* new_buffer = NULL;
239 Dart_Handle new_result = IOBuffer::Allocate(bytes_read, &new_buffer);
240 if (Dart_IsError(new_result)) {
zra 2017/04/11 16:19:41 IOBuffer::Allocate can't return an error handle.
bkonyi 2017/04/11 18:11:18 Acknowledged.
241 IOBuffer::Free(buffer);
242 if (new_buffer != NULL) {
243 IOBuffer::Free(new_buffer);
244 }
245 return;
zra 2017/04/11 16:19:40 Return the error that's in new_result
bkonyi 2017/04/11 18:11:18 Done.
246 }
247 ASSERT(new_buffer != NULL);
248 memmove(new_buffer, buffer, bytes_read);
249 IOBuffer::Free(buffer);
zra 2017/04/11 16:19:40 IOBuffer attaches 'buffer' to a WeakPersistentHand
bkonyi 2017/04/11 18:11:18 Acknowledged.
250 Dart_SetReturnValue(args, new_result);
251 } else if (bytes_read == -1) {
252 Dart_SetReturnValue(args, DartUtils::NewDartOSError());
253 }
254 }
255
256
257 void FUNCTION_NAME(SynchronousSocket_ShutdownRead)(Dart_NativeArguments args) {
258 SynchronousSocket* socket = SynchronousSocket::GetSocketIdNativeField(
259 Dart_GetNativeArgument(args, 0));
260 SynchronousSocket::ShutdownRead(socket->fd());
261 }
262
263
264 void FUNCTION_NAME(SynchronousSocket_ShutdownWrite)(Dart_NativeArguments args) {
265 SynchronousSocket* socket = SynchronousSocket::GetSocketIdNativeField(
266 Dart_GetNativeArgument(args, 0));
267 SynchronousSocket::ShutdownWrite(socket->fd());
268 }
269
270
271 void FUNCTION_NAME(SynchronousSocket_GetPort)(Dart_NativeArguments args) {
272 SynchronousSocket* socket = SynchronousSocket::GetSocketIdNativeField(
273 Dart_GetNativeArgument(args, 0));
274 intptr_t port = SocketBase::GetPort(socket->fd());
275 if (port > 0) {
276 Dart_SetReturnValue(args, Dart_NewInteger(port));
277 } else {
278 Dart_SetReturnValue(args, DartUtils::NewDartOSError());
279 }
280 }
281
282
283 void FUNCTION_NAME(SynchronousSocket_GetRemotePeer)(Dart_NativeArguments args) {
284 SynchronousSocket* socket = SynchronousSocket::GetSocketIdNativeField(
285 Dart_GetNativeArgument(args, 0));
286 intptr_t port = 0;
287 SocketAddress* addr = SocketBase::GetRemotePeer(socket->fd(), &port);
288 if (addr == NULL) {
289 Dart_SetReturnValue(args, DartUtils::NewDartOSError());
290 return;
291 }
292 Dart_Handle list = Dart_NewList(2);
293 DART_CHECK_ERROR_AND_CLEANUP(list, addr);
294
295 Dart_Handle entry = Dart_NewList(3);
296 DART_CHECK_ERROR_AND_CLEANUP(entry, addr);
297
298 Dart_Handle error =
299 Dart_ListSetAt(entry, 0, Dart_NewInteger(addr->GetType()));
300 DART_CHECK_ERROR_AND_CLEANUP(error, addr);
301 error =
302 Dart_ListSetAt(entry, 1, Dart_NewStringFromCString(addr->as_string()));
303 DART_CHECK_ERROR_AND_CLEANUP(error, addr);
304
305 RawAddr raw = addr->addr();
306 error = Dart_ListSetAt(entry, 2, SocketAddress::ToTypedData(raw));
307 DART_CHECK_ERROR_AND_CLEANUP(error, addr);
308
309 error = Dart_ListSetAt(list, 0, entry);
310 DART_CHECK_ERROR_AND_CLEANUP(error, addr);
311 error = Dart_ListSetAt(list, 1, Dart_NewInteger(port));
312 DART_CHECK_ERROR_AND_CLEANUP(error, addr);
313 Dart_SetReturnValue(args, list);
314 delete addr;
315 }
316
317 #undef DART_CHECK_ERROR_AND_CLEANUP
zra 2017/04/11 16:19:40 This is probably unnecessary.
bkonyi 2017/04/11 18:11:17 Yes, this was left over from some rough stuff I wa
318
319 static void SynchronousSocketFinalizer(void* isolate_data,
320 Dart_WeakPersistentHandle handle,
321 void* data) {
322 SynchronousSocket* socket = reinterpret_cast<SynchronousSocket*>(data);
323 if (socket->fd() >= 0) {
324 SocketBase::Close(socket->fd());
325 socket->SetClosedFd();
326 }
327 delete socket;
328 }
329
330
331 void SynchronousSocket::SetSocketIdNativeField(Dart_Handle handle,
332 intptr_t id) {
333 SynchronousSocket* socket = new SynchronousSocket(id);
334 Dart_SetNativeInstanceField(handle, kSocketIdNativeField,
335 reinterpret_cast<intptr_t>(socket));
336 Dart_NewWeakPersistentHandle(handle, reinterpret_cast<void*>(socket),
337 sizeof(SynchronousSocket),
338 SynchronousSocketFinalizer);
339 }
340
341
342 SynchronousSocket* SynchronousSocket::GetSocketIdNativeField(
343 Dart_Handle socket_obj) {
344 intptr_t id;
345 Dart_GetNativeInstanceField(socket_obj, kSocketIdNativeField, &id);
346 SynchronousSocket* socket = reinterpret_cast<SynchronousSocket*>(id);
347 return socket;
348 }
349
350 } // namespace bin
351 } // namespace dart
352
353 #endif // !defined(DART_IO_DISABLED)
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698