OLD | NEW |
1 // Copyright (c) 2012, 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 // Echo server test program to test socket streams. | 5 // Echo server test program to test socket streams. |
6 // | 6 // |
7 // VMOptions= | 7 // VMOptions= |
8 // VMOptions=--short_socket_read | 8 // VMOptions=--short_socket_read |
9 // VMOptions=--short_socket_write | 9 // VMOptions=--short_socket_write |
10 // VMOptions=--short_socket_read --short_socket_write | 10 // VMOptions=--short_socket_read --short_socket_write |
11 | 11 |
(...skipping 15 matching lines...) Expand all Loading... |
27 _buffer = new List<int>.fixedLength(MSGSIZE), | 27 _buffer = new List<int>.fixedLength(MSGSIZE), |
28 _messages = 0 { | 28 _messages = 0 { |
29 for (int i = 0; i < MSGSIZE; i++) { | 29 for (int i = 0; i < MSGSIZE; i++) { |
30 _buffer[i] = FIRSTCHAR + i; | 30 _buffer[i] = FIRSTCHAR + i; |
31 } | 31 } |
32 _sendPort = spawnFunction(startEchoServer); | 32 _sendPort = spawnFunction(startEchoServer); |
33 initialize(); | 33 initialize(); |
34 } | 34 } |
35 | 35 |
36 void sendData() { | 36 void sendData() { |
| 37 int offset = 0; |
| 38 List<int> data; |
37 | 39 |
38 void errorHandler(Exception e) { | 40 void onData(List<int> data) { |
| 41 int bytesRead = data.length; |
| 42 for (int i = 0; i < data.length; i++) { |
| 43 Expect.equals(FIRSTCHAR + i + offset, data[i]); |
| 44 } |
| 45 offset += bytesRead; |
| 46 } |
| 47 |
| 48 void onClosed() { |
| 49 Expect.equals(MSGSIZE, offset); |
| 50 _messages++; |
| 51 if (_messages < MESSAGES) { |
| 52 sendData(); |
| 53 } else { |
| 54 shutdown(); |
| 55 } |
| 56 } |
| 57 |
| 58 void errorHandler(e) { |
39 Expect.fail("Socket error $e"); | 59 Expect.fail("Socket error $e"); |
40 } | 60 } |
41 | 61 |
42 void connectHandler() { | 62 void connectHandler() { |
43 | 63 _socket.listen(onData, |
44 OutputStream stream = _socket.outputStream; | 64 onError: errorHandler, |
45 | 65 onDone: onClosed); |
46 void dataSent() { | 66 _socket.add(_buffer); |
47 InputStream inputStream = _socket.inputStream; | 67 _socket.close(); |
48 int offset = 0; | 68 data = new List<int>.fixedLength(MSGSIZE); |
49 List<int> data; | |
50 | |
51 void onClosed() { | |
52 Expect.equals(MSGSIZE, offset); | |
53 _messages++; | |
54 if (_messages < MESSAGES) { | |
55 sendData(); | |
56 } else { | |
57 shutdown(); | |
58 } | |
59 } | |
60 | |
61 void onData() { | |
62 // Test both read and readInto. | |
63 int bytesRead = 0; | |
64 if (_messages % 2 == 0) { | |
65 bytesRead = inputStream.readInto(data, offset, MSGSIZE - offset); | |
66 for (int i = 0; i < offset + bytesRead; i++) { | |
67 Expect.equals(FIRSTCHAR + i, data[i]); | |
68 } | |
69 } else { | |
70 data = inputStream.read(); | |
71 bytesRead = data.length; | |
72 for (int i = 0; i < data.length; i++) { | |
73 Expect.equals(FIRSTCHAR + i + offset, data[i]); | |
74 } | |
75 } | |
76 | |
77 offset += bytesRead; | |
78 } | |
79 | |
80 if (_messages % 2 == 0) data = new List<int>.fixedLength(MSGSIZE); | |
81 inputStream.onData = onData; | |
82 inputStream.onClosed = onClosed; | |
83 } | |
84 | |
85 _socket.onError = errorHandler; | |
86 | |
87 // Test both write and writeFrom in different forms. | |
88 switch (_messages % 4) { | |
89 case 0: | |
90 stream.write(_buffer); | |
91 break; | |
92 case 1: | |
93 stream.write(_buffer, false); | |
94 break; | |
95 case 2: | |
96 stream.writeFrom(_buffer); | |
97 break; | |
98 case 3: | |
99 Expect.equals(0, _buffer.length % 2); | |
100 stream.writeFrom(_buffer, 0, _buffer.length ~/ 2); | |
101 stream.writeFrom(_buffer, _buffer.length ~/ 2); | |
102 break; | |
103 } | |
104 stream.close(); | |
105 dataSent(); | |
106 } | 69 } |
107 | 70 |
108 _socket = new Socket(TestingServer.HOST, _port); | 71 Socket.connect(TestingServer.HOST, _port).then((s) { |
109 if (_socket != null) { | 72 _socket = s; |
110 _socket.onConnect = connectHandler; | 73 connectHandler(); |
111 } else { | 74 }); |
112 Expect.fail("socket creation failed"); | |
113 } | |
114 } | 75 } |
115 | 76 |
116 void initialize() { | 77 void initialize() { |
117 _receivePort.receive((var message, SendPort replyTo) { | 78 _receivePort.receive((var message, SendPort replyTo) { |
118 _port = message; | 79 _port = message; |
119 sendData(); | 80 sendData(); |
120 }); | 81 }); |
121 _sendPort.send(TestingServer.INIT, _receivePort.toSendPort()); | 82 _sendPort.send(TestingServer.INIT, _receivePort.toSendPort()); |
122 } | 83 } |
123 | 84 |
(...skipping 15 matching lines...) Expand all Loading... |
139 var server = new EchoServer(); | 100 var server = new EchoServer(); |
140 port.receive(server.dispatch); | 101 port.receive(server.dispatch); |
141 } | 102 } |
142 | 103 |
143 | 104 |
144 class EchoServer extends TestingServer { | 105 class EchoServer extends TestingServer { |
145 | 106 |
146 static const int MSGSIZE = EchoServerGame.MSGSIZE; | 107 static const int MSGSIZE = EchoServerGame.MSGSIZE; |
147 | 108 |
148 void onConnection(Socket connection) { | 109 void onConnection(Socket connection) { |
149 InputStream inputStream; | |
150 List<int> buffer = new List<int>.fixedLength(MSGSIZE); | 110 List<int> buffer = new List<int>.fixedLength(MSGSIZE); |
151 int offset = 0; | 111 int offset = 0; |
152 | 112 |
153 void dataReceived() { | 113 void dataReceived(List<int> data) { |
154 int bytesRead; | 114 int bytesRead; |
155 OutputStream outputStream = connection.outputStream; | 115 bytesRead = data.length; |
156 bytesRead = inputStream.readInto(buffer, offset, MSGSIZE - offset); | |
157 if (bytesRead > 0) { | 116 if (bytesRead > 0) { |
| 117 buffer.setRange(offset, data.length, data); |
158 offset += bytesRead; | 118 offset += bytesRead; |
159 for (int i = 0; i < offset; i++) { | 119 for (int i = 0; i < offset; i++) { |
160 Expect.equals(EchoServerGame.FIRSTCHAR + i, buffer[i]); | 120 Expect.equals(EchoServerGame.FIRSTCHAR + i, buffer[i]); |
161 } | 121 } |
162 if (offset == MSGSIZE) { | 122 if (offset == MSGSIZE) { |
163 outputStream.write(buffer); | 123 connection.add(buffer); |
164 outputStream.close(); | 124 connection.close(); |
165 } | 125 } |
166 } | 126 } |
167 } | 127 } |
168 | 128 |
169 void errorHandler(Exception e) { | 129 void errorHandler(e) { |
170 Expect.fail("Socket error $e"); | 130 Expect.fail("Socket error $e"); |
171 } | 131 } |
172 | 132 |
173 inputStream = connection.inputStream; | 133 connection.listen(dataReceived, onError: errorHandler); |
174 inputStream.onData = dataReceived; | |
175 connection.onError = errorHandler; | |
176 } | 134 } |
177 } | 135 } |
178 | 136 |
179 main() { | 137 main() { |
180 EchoServerGame echoServerGame = new EchoServerGame.start(); | 138 EchoServerGame echoServerGame = new EchoServerGame.start(); |
181 } | 139 } |
OLD | NEW |