| 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 part of dart.io; | 5 part of dart.io; |
| 6 | 6 |
| 7 const String _webSocketGUID = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"; | 7 const String _webSocketGUID = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"; |
| 8 | 8 |
| 9 // Matches _WebSocketOpcode. | 9 // Matches _WebSocketOpcode. |
| 10 class _WebSocketMessageType { | 10 class _WebSocketMessageType { |
| (...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 62 int _remainingPayloadBytes = -1; | 62 int _remainingPayloadBytes = -1; |
| 63 int _unmaskingIndex = 0; | 63 int _unmaskingIndex = 0; |
| 64 int _currentMessageType = _WebSocketMessageType.NONE; | 64 int _currentMessageType = _WebSocketMessageType.NONE; |
| 65 int closeCode = WebSocketStatus.NO_STATUS_RECEIVED; | 65 int closeCode = WebSocketStatus.NO_STATUS_RECEIVED; |
| 66 String closeReason = ""; | 66 String closeReason = ""; |
| 67 | 67 |
| 68 EventSink _eventSink; | 68 EventSink _eventSink; |
| 69 | 69 |
| 70 final bool _serverSide; | 70 final bool _serverSide; |
| 71 final List _maskingBytes = new List(4); | 71 final List _maskingBytes = new List(4); |
| 72 final List<Uint8List> _payloads = new List<Uint8List>(); | 72 final BytesBuilder _payload = new BytesBuilder(copy: false); |
| 73 | 73 |
| 74 _WebSocketProtocolTransformer([this._serverSide = false]); | 74 _WebSocketProtocolTransformer([this._serverSide = false]); |
| 75 | 75 |
| 76 Stream bind(Stream stream) { | 76 Stream bind(Stream stream) { |
| 77 return new Stream.eventTransformed( | 77 return new Stream.eventTransformed( |
| 78 stream, | 78 stream, |
| 79 (EventSink eventSink) { | 79 (EventSink eventSink) { |
| 80 if (_eventSink != null) { | 80 if (_eventSink != null) { |
| 81 throw new StateError("WebSocket transformer already used."); | 81 throw new StateError("WebSocket transformer already used."); |
| 82 } | 82 } |
| 83 _eventSink = eventSink; | 83 _eventSink = eventSink; |
| 84 return this; | 84 return this; |
| 85 }); | 85 }); |
| 86 } | 86 } |
| 87 | 87 |
| 88 void addError(Object error, [StackTrace stackTrace]) => | 88 void addError(Object error, [StackTrace stackTrace]) => |
| 89 _eventSink.addError(error, stackTrace); | 89 _eventSink.addError(error, stackTrace); |
| 90 | 90 |
| 91 void close() => _eventSink.close(); | 91 void close() => _eventSink.close(); |
| 92 | 92 |
| 93 Uint8List _takePayload() { | |
| 94 if (_payloads.length == 0) return new Uint8List(0); | |
| 95 if (_payloads.length == 1) { | |
| 96 Uint8List result = _payloads.single; | |
| 97 _payloads.clear(); | |
| 98 return result; | |
| 99 } | |
| 100 int length = 0; | |
| 101 for (Uint8List payload in _payloads) { | |
| 102 length += payload.length; | |
| 103 } | |
| 104 Uint8List result = new Uint8List(length); | |
| 105 int offset = 0; | |
| 106 for (Uint8List payload in _payloads) { | |
| 107 result.setRange(offset, offset + payload.length, payload); | |
| 108 offset += payload.length; | |
| 109 } | |
| 110 _payloads.clear(); | |
| 111 return result; | |
| 112 } | |
| 113 | |
| 114 /** | 93 /** |
| 115 * Process data received from the underlying communication channel. | 94 * Process data received from the underlying communication channel. |
| 116 */ | 95 */ |
| 117 void add(Uint8List buffer) { | 96 void add(Uint8List buffer) { |
| 118 int count = buffer.length; | 97 int count = buffer.length; |
| 119 int index = 0; | 98 int index = 0; |
| 120 int lastIndex = count; | 99 int lastIndex = count; |
| 121 if (_state == CLOSED) { | 100 if (_state == CLOSED) { |
| 122 throw new WebSocketException("Data on closed connection"); | 101 throw new WebSocketException("Data on closed connection"); |
| 123 } | 102 } |
| (...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 190 } else { | 169 } else { |
| 191 assert(_state == PAYLOAD); | 170 assert(_state == PAYLOAD); |
| 192 // The payload is not handled one byte at a time but in blocks. | 171 // The payload is not handled one byte at a time but in blocks. |
| 193 int payloadLength = min(lastIndex - index, _remainingPayloadBytes); | 172 int payloadLength = min(lastIndex - index, _remainingPayloadBytes); |
| 194 _remainingPayloadBytes -= payloadLength; | 173 _remainingPayloadBytes -= payloadLength; |
| 195 // Unmask payload if masked. | 174 // Unmask payload if masked. |
| 196 if (_masked) { | 175 if (_masked) { |
| 197 _unmask(index, payloadLength, buffer); | 176 _unmask(index, payloadLength, buffer); |
| 198 } | 177 } |
| 199 // Control frame and data frame share _payloads. | 178 // Control frame and data frame share _payloads. |
| 200 _payloads.add( | 179 _payload.add( |
| 201 new Uint8List.view(buffer.buffer, index, payloadLength)); | 180 new Uint8List.view(buffer.buffer, index, payloadLength)); |
| 202 index += payloadLength; | 181 index += payloadLength; |
| 203 if (_isControlFrame()) { | 182 if (_isControlFrame()) { |
| 204 if (_remainingPayloadBytes == 0) _controlFrameEnd(); | 183 if (_remainingPayloadBytes == 0) _controlFrameEnd(); |
| 205 } else { | 184 } else { |
| 206 if (_currentMessageType != _WebSocketMessageType.TEXT && | 185 if (_currentMessageType != _WebSocketMessageType.TEXT && |
| 207 _currentMessageType != _WebSocketMessageType.BINARY) { | 186 _currentMessageType != _WebSocketMessageType.BINARY) { |
| 208 throw new WebSocketException("Protocol error"); | 187 throw new WebSocketException("Protocol error"); |
| 209 } | 188 } |
| 210 if (_remainingPayloadBytes == 0) _messageFrameEnd(); | 189 if (_remainingPayloadBytes == 0) _messageFrameEnd(); |
| (...skipping 89 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 300 } | 279 } |
| 301 } else { | 280 } else { |
| 302 _state = PAYLOAD; | 281 _state = PAYLOAD; |
| 303 } | 282 } |
| 304 } | 283 } |
| 305 | 284 |
| 306 void _messageFrameEnd() { | 285 void _messageFrameEnd() { |
| 307 if (_fin) { | 286 if (_fin) { |
| 308 switch (_currentMessageType) { | 287 switch (_currentMessageType) { |
| 309 case _WebSocketMessageType.TEXT: | 288 case _WebSocketMessageType.TEXT: |
| 310 _eventSink.add(UTF8.decode(_takePayload())); | 289 _eventSink.add(UTF8.decode(_payload.takeBytes())); |
| 311 break; | 290 break; |
| 312 case _WebSocketMessageType.BINARY: | 291 case _WebSocketMessageType.BINARY: |
| 313 _eventSink.add(_takePayload()); | 292 _eventSink.add(_payload.takeBytes()); |
| 314 break; | 293 break; |
| 315 } | 294 } |
| 316 _currentMessageType = _WebSocketMessageType.NONE; | 295 _currentMessageType = _WebSocketMessageType.NONE; |
| 317 } | 296 } |
| 318 _prepareForNextFrame(); | 297 _prepareForNextFrame(); |
| 319 } | 298 } |
| 320 | 299 |
| 321 void _controlFrameEnd() { | 300 void _controlFrameEnd() { |
| 322 switch (_opcode) { | 301 switch (_opcode) { |
| 323 case _WebSocketOpcode.CLOSE: | 302 case _WebSocketOpcode.CLOSE: |
| 324 closeCode = WebSocketStatus.NO_STATUS_RECEIVED; | 303 closeCode = WebSocketStatus.NO_STATUS_RECEIVED; |
| 325 var payload = _takePayload(); | 304 var payload = _payload.takeBytes(); |
| 326 if (payload.length > 0) { | 305 if (payload.length > 0) { |
| 327 if (payload.length == 1) { | 306 if (payload.length == 1) { |
| 328 throw new WebSocketException("Protocol error"); | 307 throw new WebSocketException("Protocol error"); |
| 329 } | 308 } |
| 330 closeCode = payload[0] << 8 | payload[1]; | 309 closeCode = payload[0] << 8 | payload[1]; |
| 331 if (closeCode == WebSocketStatus.NO_STATUS_RECEIVED) { | 310 if (closeCode == WebSocketStatus.NO_STATUS_RECEIVED) { |
| 332 throw new WebSocketException("Protocol error"); | 311 throw new WebSocketException("Protocol error"); |
| 333 } | 312 } |
| 334 if (payload.length > 2) { | 313 if (payload.length > 2) { |
| 335 closeReason = UTF8.decode(payload.sublist(2)); | 314 closeReason = UTF8.decode(payload.sublist(2)); |
| 336 } | 315 } |
| 337 } | 316 } |
| 338 _state = CLOSED; | 317 _state = CLOSED; |
| 339 _eventSink.close(); | 318 _eventSink.close(); |
| 340 break; | 319 break; |
| 341 | 320 |
| 342 case _WebSocketOpcode.PING: | 321 case _WebSocketOpcode.PING: |
| 343 _eventSink.add(new _WebSocketPing(_takePayload())); | 322 _eventSink.add(new _WebSocketPing(_payload.takeBytes())); |
| 344 break; | 323 break; |
| 345 | 324 |
| 346 case _WebSocketOpcode.PONG: | 325 case _WebSocketOpcode.PONG: |
| 347 _eventSink.add(new _WebSocketPong(_takePayload())); | 326 _eventSink.add(new _WebSocketPong(_payload.takeBytes())); |
| 348 break; | 327 break; |
| 349 } | 328 } |
| 350 _prepareForNextFrame(); | 329 _prepareForNextFrame(); |
| 351 } | 330 } |
| 352 | 331 |
| 353 bool _isControlFrame() { | 332 bool _isControlFrame() { |
| 354 return _opcode == _WebSocketOpcode.CLOSE || | 333 return _opcode == _WebSocketOpcode.CLOSE || |
| 355 _opcode == _WebSocketOpcode.PING || | 334 _opcode == _WebSocketOpcode.PING || |
| 356 _opcode == _WebSocketOpcode.PONG; | 335 _opcode == _WebSocketOpcode.PONG; |
| 357 } | 336 } |
| (...skipping 627 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 985 (code < WebSocketStatus.NORMAL_CLOSURE || | 964 (code < WebSocketStatus.NORMAL_CLOSURE || |
| 986 code == WebSocketStatus.RESERVED_1004 || | 965 code == WebSocketStatus.RESERVED_1004 || |
| 987 code == WebSocketStatus.NO_STATUS_RECEIVED || | 966 code == WebSocketStatus.NO_STATUS_RECEIVED || |
| 988 code == WebSocketStatus.ABNORMAL_CLOSURE || | 967 code == WebSocketStatus.ABNORMAL_CLOSURE || |
| 989 (code > WebSocketStatus.INTERNAL_SERVER_ERROR && | 968 (code > WebSocketStatus.INTERNAL_SERVER_ERROR && |
| 990 code < WebSocketStatus.RESERVED_1015) || | 969 code < WebSocketStatus.RESERVED_1015) || |
| 991 (code >= WebSocketStatus.RESERVED_1015 && | 970 (code >= WebSocketStatus.RESERVED_1015 && |
| 992 code < 3000)); | 971 code < 3000)); |
| 993 } | 972 } |
| 994 } | 973 } |
| OLD | NEW |