OLD | NEW |
1 // Copyright (c) 2013, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2013, the Dart project authors. Please see the AUTHORS file |
2 // for details. All rights reserved. Use of this source code is governed by a | 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. | 3 // BSD-style license that can be found in the LICENSE file. |
4 // | 4 // |
5 // VMOptions= | 5 // VMOptions= |
6 // VMOptions=--short_socket_read | 6 // VMOptions=--short_socket_read |
7 // VMOptions=--short_socket_write | 7 // VMOptions=--short_socket_write |
8 // VMOptions=--short_socket_read --short_socket_write | 8 // VMOptions=--short_socket_read --short_socket_write |
9 | 9 |
10 import "package:expect/expect.dart"; | 10 import "package:expect/expect.dart"; |
(...skipping 137 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
148 // between client and server before the connection is secured. This argument | 148 // between client and server before the connection is secured. This argument |
149 // only makes sense when both listenSecure and connectSecure are false. | 149 // only makes sense when both listenSecure and connectSecure are false. |
150 // | 150 // |
151 // postponeSecure | 151 // postponeSecure |
152 // When this argument is false the securing of the server end will | 152 // When this argument is false the securing of the server end will |
153 // happen as soon as the last byte of the handshake before securing | 153 // happen as soon as the last byte of the handshake before securing |
154 // has been written. When this argument is true the securing of the | 154 // has been written. When this argument is true the securing of the |
155 // server will not happen until the first TLS handshake data has been | 155 // server will not happen until the first TLS handshake data has been |
156 // received from the client. This argument only takes effect when | 156 // received from the client. This argument only takes effect when |
157 // handshakeBeforeSecure is true. | 157 // handshakeBeforeSecure is true. |
158 void testSimpleReadWrite(bool listenSecure, | 158 void testSimpleReadWrite({bool listenSecure, |
159 bool connectSecure, | 159 bool connectSecure, |
160 bool handshakeBeforeSecure, | 160 bool handshakeBeforeSecure, |
161 [bool postponeSecure = false]) { | 161 bool postponeSecure, |
| 162 bool dropReads}) { |
| 163 int clientReads = 0; |
| 164 int serverReads = 0; |
162 if (handshakeBeforeSecure == true && | 165 if (handshakeBeforeSecure == true && |
163 (listenSecure == true || connectSecure == true)) { | 166 (listenSecure == true || connectSecure == true)) { |
164 Expect.fail("Invalid arguments to testSimpleReadWrite"); | 167 Expect.fail("Invalid arguments to testSimpleReadWrite"); |
165 } | 168 } |
166 | 169 |
167 ReceivePort port = new ReceivePort(); | 170 ReceivePort port = new ReceivePort(); |
168 | 171 |
169 const messageSize = 1000; | 172 const messageSize = 1000; |
170 const handshakeMessageSize = 100; | 173 const handshakeMessageSize = 100; |
171 | 174 |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
204 Future runServer(RawSocket client) { | 207 Future runServer(RawSocket client) { |
205 var completer = new Completer(); | 208 var completer = new Completer(); |
206 int bytesRead = 0; | 209 int bytesRead = 0; |
207 int bytesWritten = 0; | 210 int bytesWritten = 0; |
208 List<int> data = new List<int>(messageSize); | 211 List<int> data = new List<int>(messageSize); |
209 client.writeEventsEnabled = false; | 212 client.writeEventsEnabled = false; |
210 var subscription; | 213 var subscription; |
211 subscription = client.listen((event) { | 214 subscription = client.listen((event) { |
212 switch (event) { | 215 switch (event) { |
213 case RawSocketEvent.READ: | 216 case RawSocketEvent.READ: |
| 217 if (dropReads) { |
| 218 if (serverReads != 10) { |
| 219 ++serverReads; |
| 220 break; |
| 221 } else { |
| 222 serverReads = 0; |
| 223 } |
| 224 } |
214 Expect.isTrue(bytesWritten == 0); | 225 Expect.isTrue(bytesWritten == 0); |
215 Expect.isTrue(client.available() > 0); | 226 Expect.isTrue(client.available() > 0); |
216 var buffer = client.read(); | 227 var buffer = client.read(); |
217 if (buffer != null) { | 228 if (buffer != null) { |
218 data.setRange(bytesRead, bytesRead + buffer.length, buffer); | 229 data.setRange(bytesRead, bytesRead + buffer.length, buffer); |
219 bytesRead += buffer.length; | 230 bytesRead += buffer.length; |
220 for (var value in buffer) { | 231 for (var value in buffer) { |
221 Expect.isTrue(value is int); | 232 Expect.isTrue(value is int); |
222 Expect.isTrue(value < 256 && value >= 0); | 233 Expect.isTrue(value < 256 && value >= 0); |
223 } | 234 } |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
255 Future<RawSocket> runClient(RawSocket socket) { | 266 Future<RawSocket> runClient(RawSocket socket) { |
256 var completer = new Completer(); | 267 var completer = new Completer(); |
257 int bytesRead = 0; | 268 int bytesRead = 0; |
258 int bytesWritten = 0; | 269 int bytesWritten = 0; |
259 List<int> dataSent = createTestData(); | 270 List<int> dataSent = createTestData(); |
260 List<int> dataReceived = new List<int>(dataSent.length); | 271 List<int> dataReceived = new List<int>(dataSent.length); |
261 socket.listen((event) { | 272 socket.listen((event) { |
262 switch (event) { | 273 switch (event) { |
263 case RawSocketEvent.READ: | 274 case RawSocketEvent.READ: |
264 Expect.isTrue(socket.available() > 0); | 275 Expect.isTrue(socket.available() > 0); |
| 276 if (dropReads) { |
| 277 if (clientReads != 10) { |
| 278 ++clientReads; |
| 279 break; |
| 280 } else { |
| 281 clientReads = 0; |
| 282 } |
| 283 } |
265 var buffer = socket.read(); | 284 var buffer = socket.read(); |
266 if (buffer != null) { | 285 if (buffer != null) { |
267 dataReceived.setRange(bytesRead, bytesRead + buffer.length, buffer); | 286 dataReceived.setRange(bytesRead, bytesRead + buffer.length, buffer); |
268 bytesRead += buffer.length; | 287 bytesRead += buffer.length; |
269 } | 288 } |
270 break; | 289 break; |
271 case RawSocketEvent.WRITE: | 290 case RawSocketEvent.WRITE: |
272 Expect.isTrue(bytesRead == 0); | 291 Expect.isTrue(bytesRead == 0); |
273 Expect.isFalse(socket.writeEventsEnabled); | 292 Expect.isFalse(socket.writeEventsEnabled); |
274 bytesWritten += socket.write( | 293 bytesWritten += socket.write( |
(...skipping 19 matching lines...) Expand all Loading... |
294 List<int> data = new List<int>(handshakeMessageSize); | 313 List<int> data = new List<int>(handshakeMessageSize); |
295 client.writeEventsEnabled = false; | 314 client.writeEventsEnabled = false; |
296 var subscription; | 315 var subscription; |
297 subscription = client.listen((event) { | 316 subscription = client.listen((event) { |
298 switch (event) { | 317 switch (event) { |
299 case RawSocketEvent.READ: | 318 case RawSocketEvent.READ: |
300 if (bytesRead < data.length) { | 319 if (bytesRead < data.length) { |
301 Expect.isTrue(bytesWritten == 0); | 320 Expect.isTrue(bytesWritten == 0); |
302 } | 321 } |
303 Expect.isTrue(client.available() > 0); | 322 Expect.isTrue(client.available() > 0); |
| 323 if (dropReads) { |
| 324 if (serverReads != 10) { |
| 325 ++serverReads; |
| 326 break; |
| 327 } else { |
| 328 serverReads = 0; |
| 329 } |
| 330 } |
304 var buffer = client.read(); | 331 var buffer = client.read(); |
305 if (buffer != null) { | 332 if (buffer != null) { |
306 if (bytesRead == data.length) { | 333 if (bytesRead == data.length) { |
307 // Read first part of TLS handshake from client. | 334 // Read first part of TLS handshake from client. |
308 Expect.isTrue(postponeSecure); | 335 Expect.isTrue(postponeSecure); |
309 completer.complete([subscription, buffer]); | 336 completer.complete([subscription, buffer]); |
310 client.readEventsEnabled = false; | 337 client.readEventsEnabled = false; |
311 return; | 338 return; |
312 } | 339 } |
313 data.setRange(bytesRead, bytesRead + buffer.length, buffer); | 340 data.setRange(bytesRead, bytesRead + buffer.length, buffer); |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
352 Future<RawSocket> runClientHandshake(RawSocket socket) { | 379 Future<RawSocket> runClientHandshake(RawSocket socket) { |
353 var completer = new Completer(); | 380 var completer = new Completer(); |
354 int bytesRead = 0; | 381 int bytesRead = 0; |
355 int bytesWritten = 0; | 382 int bytesWritten = 0; |
356 List<int> dataSent = createHandshakeTestData(); | 383 List<int> dataSent = createHandshakeTestData(); |
357 List<int> dataReceived = new List<int>(dataSent.length); | 384 List<int> dataReceived = new List<int>(dataSent.length); |
358 var subscription; | 385 var subscription; |
359 subscription = socket.listen((event) { | 386 subscription = socket.listen((event) { |
360 switch (event) { | 387 switch (event) { |
361 case RawSocketEvent.READ: | 388 case RawSocketEvent.READ: |
| 389 if (dropReads) { |
| 390 if (clientReads != 10) { |
| 391 ++clientReads; |
| 392 break; |
| 393 } else { |
| 394 clientReads = 0; |
| 395 } |
| 396 } |
362 Expect.isTrue(socket.available() > 0); | 397 Expect.isTrue(socket.available() > 0); |
363 var buffer = socket.read(); | 398 var buffer = socket.read(); |
364 if (buffer != null) { | 399 if (buffer != null) { |
365 dataReceived.setRange(bytesRead, bytesRead + buffer.length, buffer); | 400 dataReceived.setRange(bytesRead, bytesRead + buffer.length, buffer); |
366 bytesRead += buffer.length; | 401 bytesRead += buffer.length; |
367 if (bytesRead == dataSent.length) { | 402 if (bytesRead == dataSent.length) { |
368 verifyHandshakeTestData(dataReceived); | 403 verifyHandshakeTestData(dataReceived); |
369 completer.complete(subscription); | 404 completer.complete(subscription); |
370 } | 405 } |
371 } | 406 } |
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
411 } else if (!handshakeBeforeSecure) { | 446 } else if (!handshakeBeforeSecure) { |
412 RawSecureSocket.secureServer(client, CERTIFICATE).then((client) { | 447 RawSecureSocket.secureServer(client, CERTIFICATE).then((client) { |
413 runServer(client).then((_) => server.close()); | 448 runServer(client).then((_) => server.close()); |
414 }); | 449 }); |
415 } else { | 450 } else { |
416 runServerHandshake(client).then((secure) { | 451 runServerHandshake(client).then((secure) { |
417 RawSecureSocket.secureServer( | 452 RawSecureSocket.secureServer( |
418 client, | 453 client, |
419 CERTIFICATE, | 454 CERTIFICATE, |
420 subscription: secure[0], | 455 subscription: secure[0], |
421 carryOverData: secure[1]).then((client) { | 456 bufferedData: secure[1]).then((client) { |
422 runServer(client).then((_) => server.close()); | 457 runServer(client).then((_) => server.close()); |
423 }); | 458 }); |
424 }); | 459 }); |
425 } | 460 } |
426 }); | 461 }); |
427 | 462 |
428 connectClient(server.port).then(runClient).then((socket) { | 463 connectClient(server.port).then(runClient).then((socket) { |
429 socket.close(); | 464 socket.close(); |
430 port.close(); | 465 port.close(); |
431 }); | 466 }); |
(...skipping 15 matching lines...) Expand all Loading... |
447 useBuiltinRoots: false); | 482 useBuiltinRoots: false); |
448 testSimpleBind(); | 483 testSimpleBind(); |
449 testInvalidBind(); | 484 testInvalidBind(); |
450 testSimpleConnect(CERTIFICATE); | 485 testSimpleConnect(CERTIFICATE); |
451 testSimpleConnect("CN=localhost"); | 486 testSimpleConnect("CN=localhost"); |
452 testSimpleConnectFail("not_a_nickname", false); | 487 testSimpleConnectFail("not_a_nickname", false); |
453 testSimpleConnectFail("CN=notARealDistinguishedName", false); | 488 testSimpleConnectFail("CN=notARealDistinguishedName", false); |
454 testSimpleConnectFail("not_a_nickname", true); | 489 testSimpleConnectFail("not_a_nickname", true); |
455 testSimpleConnectFail("CN=notARealDistinguishedName", true); | 490 testSimpleConnectFail("CN=notARealDistinguishedName", true); |
456 testServerListenAfterConnect(); | 491 testServerListenAfterConnect(); |
457 testSimpleReadWrite(true, true, false); | 492 testSimpleReadWrite(listenSecure: true, |
458 testSimpleReadWrite(true, false, false); | 493 connectSecure: true, |
459 testSimpleReadWrite(false, true, false); | 494 handshakeBeforeSecure: false, |
460 testSimpleReadWrite(false, false, false); | 495 postponeSecure: false, |
461 testSimpleReadWrite(false, false, true, true); | 496 dropReads: false); |
462 testSimpleReadWrite(false, false, true, false); | 497 testSimpleReadWrite(listenSecure: true, |
| 498 connectSecure: false, |
| 499 handshakeBeforeSecure: false, |
| 500 postponeSecure: false, |
| 501 dropReads: false); |
| 502 testSimpleReadWrite(listenSecure: false, |
| 503 connectSecure: true, |
| 504 handshakeBeforeSecure: false, |
| 505 postponeSecure: false, |
| 506 dropReads: false); |
| 507 testSimpleReadWrite(listenSecure: false, |
| 508 connectSecure: false, |
| 509 handshakeBeforeSecure: false, |
| 510 postponeSecure: false, |
| 511 dropReads: false); |
| 512 testSimpleReadWrite(listenSecure: false, |
| 513 connectSecure: false, |
| 514 handshakeBeforeSecure: true, |
| 515 postponeSecure: true, |
| 516 dropReads: false); |
| 517 testSimpleReadWrite(listenSecure: false, |
| 518 connectSecure: false, |
| 519 handshakeBeforeSecure: true, |
| 520 postponeSecure: false, |
| 521 dropReads: false); |
| 522 testSimpleReadWrite(listenSecure: true, |
| 523 connectSecure: true, |
| 524 handshakeBeforeSecure: false, |
| 525 postponeSecure: false, |
| 526 dropReads: true); |
| 527 testSimpleReadWrite(listenSecure: false, |
| 528 connectSecure: false, |
| 529 handshakeBeforeSecure: true, |
| 530 postponeSecure: true, |
| 531 dropReads: true); |
463 } | 532 } |
OLD | NEW |