| | | 1 | | // Copyright (c) ZeroC, Inc. |
| | | 2 | | |
| | | 3 | | using System.Diagnostics; |
| | | 4 | | using System.Net.Sockets; |
| | | 5 | | using System.Security.Cryptography; |
| | | 6 | | using System.Text; |
| | | 7 | | |
| | | 8 | | namespace Ice.Internal; |
| | | 9 | | |
| | | 10 | | internal sealed class WSTransceiver : Transceiver |
| | | 11 | | { |
| | 0 | 12 | | public Socket fd() => _delegate.fd(); |
| | | 13 | | |
| | | 14 | | public int initialize(Buffer readBuffer, Buffer writeBuffer, ref bool hasMoreData) |
| | | 15 | | { |
| | | 16 | | // |
| | | 17 | | // Delegate logs exceptions that occur during initialize(), so there's no need to trap them here. |
| | | 18 | | // |
| | 1 | 19 | | if (_state == StateInitializeDelegate) |
| | | 20 | | { |
| | 1 | 21 | | int op = _delegate.initialize(readBuffer, writeBuffer, ref hasMoreData); |
| | 1 | 22 | | if (op != 0) |
| | | 23 | | { |
| | 1 | 24 | | return op; |
| | | 25 | | } |
| | 1 | 26 | | _state = StateConnected; |
| | | 27 | | } |
| | | 28 | | |
| | | 29 | | try |
| | | 30 | | { |
| | 1 | 31 | | if (_state == StateConnected) |
| | | 32 | | { |
| | | 33 | | // |
| | | 34 | | // We don't know how much we'll need to read. |
| | | 35 | | // |
| | 1 | 36 | | _readBuffer.resize(1024, true); |
| | 1 | 37 | | _readBuffer.b.position(0); |
| | 1 | 38 | | _readBufferPos = 0; |
| | | 39 | | |
| | | 40 | | // |
| | | 41 | | // The server waits for the client's upgrade request, the |
| | | 42 | | // client sends the upgrade request. |
| | | 43 | | // |
| | 1 | 44 | | _state = StateUpgradeRequestPending; |
| | 1 | 45 | | if (!_incoming) |
| | | 46 | | { |
| | | 47 | | // |
| | | 48 | | // Compose the upgrade request. |
| | | 49 | | // |
| | 1 | 50 | | var @out = new StringBuilder(); |
| | 1 | 51 | | @out.Append("GET " + _resource + " HTTP/1.1\r\n"); |
| | 1 | 52 | | @out.Append("Host: " + _host + "\r\n"); |
| | 1 | 53 | | @out.Append("Upgrade: websocket\r\n"); |
| | 1 | 54 | | @out.Append("Connection: Upgrade\r\n"); |
| | 1 | 55 | | @out.Append("Sec-WebSocket-Protocol: " + _iceProtocol + "\r\n"); |
| | 1 | 56 | | @out.Append("Sec-WebSocket-Version: 13\r\n"); |
| | 1 | 57 | | @out.Append("Sec-WebSocket-Key: "); |
| | | 58 | | |
| | | 59 | | // |
| | | 60 | | // The value for Sec-WebSocket-Key is a 16-byte random number, |
| | | 61 | | // encoded with Base64. |
| | | 62 | | // |
| | 1 | 63 | | byte[] key = new byte[16]; |
| | 1 | 64 | | RandomNumberGenerator.Fill(key); |
| | 1 | 65 | | _key = System.Convert.ToBase64String(key); |
| | 1 | 66 | | @out.Append(_key + "\r\n\r\n"); // EOM |
| | | 67 | | |
| | 1 | 68 | | byte[] bytes = _utf8.GetBytes(@out.ToString()); |
| | 1 | 69 | | _writeBuffer.resize(bytes.Length, false); |
| | 1 | 70 | | _writeBuffer.b.position(0); |
| | 1 | 71 | | _writeBuffer.b.put(bytes); |
| | 1 | 72 | | _writeBuffer.b.flip(); |
| | | 73 | | } |
| | | 74 | | } |
| | | 75 | | |
| | | 76 | | // |
| | | 77 | | // Try to write the client's upgrade request. |
| | | 78 | | // |
| | 1 | 79 | | if (_state == StateUpgradeRequestPending && !_incoming) |
| | | 80 | | { |
| | 1 | 81 | | if (_writeBuffer.b.hasRemaining()) |
| | | 82 | | { |
| | 1 | 83 | | int s = _delegate.write(_writeBuffer); |
| | 1 | 84 | | if (s != 0) |
| | | 85 | | { |
| | 1 | 86 | | return s; |
| | | 87 | | } |
| | | 88 | | } |
| | | 89 | | Debug.Assert(!_writeBuffer.b.hasRemaining()); |
| | 1 | 90 | | _state = StateUpgradeResponsePending; |
| | | 91 | | |
| | 1 | 92 | | if (_instance.traceLevel() >= 1) |
| | | 93 | | { |
| | 1 | 94 | | _instance.logger().trace( |
| | 1 | 95 | | _instance.traceCategory(), |
| | 1 | 96 | | "sent " + protocol() + " connection HTTP upgrade request\n" + ToString()); |
| | | 97 | | } |
| | | 98 | | } |
| | | 99 | | |
| | | 100 | | while (true) |
| | | 101 | | { |
| | 1 | 102 | | if (_readBuffer.b.hasRemaining()) |
| | | 103 | | { |
| | 1 | 104 | | int s = _delegate.read(_readBuffer, ref hasMoreData); |
| | 1 | 105 | | if (s == SocketOperation.Write || _readBuffer.b.position() == 0) |
| | | 106 | | { |
| | 1 | 107 | | return s; |
| | | 108 | | } |
| | | 109 | | } |
| | | 110 | | |
| | | 111 | | // |
| | | 112 | | // Try to read the client's upgrade request or the server's response. |
| | | 113 | | // |
| | 1 | 114 | | if ((_state == StateUpgradeRequestPending && _incoming) || |
| | 1 | 115 | | (_state == StateUpgradeResponsePending && !_incoming)) |
| | | 116 | | { |
| | | 117 | | // |
| | | 118 | | // Check if we have enough data for a complete message. |
| | | 119 | | // |
| | 1 | 120 | | int p = _parser.isCompleteMessage(_readBuffer.b, 0, _readBuffer.b.position()); |
| | 1 | 121 | | if (p == -1) |
| | | 122 | | { |
| | 0 | 123 | | if (_readBuffer.b.hasRemaining()) |
| | | 124 | | { |
| | 0 | 125 | | return SocketOperation.Read; |
| | | 126 | | } |
| | | 127 | | |
| | | 128 | | // |
| | | 129 | | // Enlarge the buffer and try to read more. |
| | | 130 | | // |
| | 0 | 131 | | int oldSize = _readBuffer.b.position(); |
| | 0 | 132 | | if (oldSize + 1024 > _instance.messageSizeMax()) |
| | | 133 | | { |
| | 0 | 134 | | Ex.throwMemoryLimitException( |
| | 0 | 135 | | requested: oldSize + 1024, |
| | 0 | 136 | | maximum: _instance.messageSizeMax()); |
| | | 137 | | } |
| | 0 | 138 | | _readBuffer.resize(oldSize + 1024, true); |
| | 0 | 139 | | _readBuffer.b.position(oldSize); |
| | 0 | 140 | | continue; // Try again to read the response/request |
| | | 141 | | } |
| | | 142 | | |
| | | 143 | | // |
| | | 144 | | // Set _readBufferPos at the end of the response/request message. |
| | | 145 | | // |
| | 1 | 146 | | _readBufferPos = p; |
| | | 147 | | } |
| | | 148 | | |
| | | 149 | | // |
| | | 150 | | // We're done, the client's upgrade request or server's response is read. |
| | | 151 | | // |
| | | 152 | | break; |
| | | 153 | | } |
| | | 154 | | |
| | | 155 | | try |
| | | 156 | | { |
| | | 157 | | // |
| | | 158 | | // Parse the client's upgrade request. |
| | | 159 | | // |
| | 1 | 160 | | if (_state == StateUpgradeRequestPending && _incoming) |
| | | 161 | | { |
| | 1 | 162 | | if (_parser.parse(_readBuffer.b, 0, _readBufferPos)) |
| | | 163 | | { |
| | 1 | 164 | | handleRequest(_writeBuffer); |
| | 1 | 165 | | _state = StateUpgradeResponsePending; |
| | | 166 | | } |
| | | 167 | | else |
| | | 168 | | { |
| | 0 | 169 | | throw new Ice.ProtocolException("incomplete request message"); |
| | | 170 | | } |
| | | 171 | | } |
| | | 172 | | |
| | 1 | 173 | | if (_state == StateUpgradeResponsePending) |
| | | 174 | | { |
| | 1 | 175 | | if (_incoming) |
| | | 176 | | { |
| | 1 | 177 | | if (_writeBuffer.b.hasRemaining()) |
| | | 178 | | { |
| | 1 | 179 | | int s = _delegate.write(_writeBuffer); |
| | 1 | 180 | | if (s != 0) |
| | | 181 | | { |
| | 1 | 182 | | return s; |
| | | 183 | | } |
| | | 184 | | } |
| | | 185 | | } |
| | | 186 | | else |
| | | 187 | | { |
| | | 188 | | // |
| | | 189 | | // Parse the server's response |
| | | 190 | | // |
| | 1 | 191 | | if (_parser.parse(_readBuffer.b, 0, _readBufferPos)) |
| | | 192 | | { |
| | 1 | 193 | | handleResponse(); |
| | | 194 | | } |
| | | 195 | | else |
| | | 196 | | { |
| | 0 | 197 | | throw new Ice.ProtocolException("incomplete response message"); |
| | | 198 | | } |
| | | 199 | | } |
| | | 200 | | } |
| | 1 | 201 | | } |
| | 1 | 202 | | catch (WebSocketException ex) |
| | | 203 | | { |
| | 1 | 204 | | throw new Ice.ProtocolException(ex.Message); |
| | | 205 | | } |
| | | 206 | | |
| | 1 | 207 | | _state = StateOpened; |
| | 1 | 208 | | _nextState = StateOpened; |
| | | 209 | | |
| | 1 | 210 | | hasMoreData = _readBufferPos < _readBuffer.b.position(); |
| | 1 | 211 | | } |
| | 1 | 212 | | catch (Ice.LocalException ex) |
| | | 213 | | { |
| | 1 | 214 | | if (_instance.traceLevel() >= 2) |
| | | 215 | | { |
| | 1 | 216 | | _instance.logger().trace( |
| | 1 | 217 | | _instance.traceCategory(), |
| | 1 | 218 | | protocol() + " connection HTTP upgrade request failed\n" + ToString() + "\n" + ex); |
| | | 219 | | } |
| | 1 | 220 | | throw; |
| | | 221 | | } |
| | | 222 | | |
| | 1 | 223 | | if (_instance.traceLevel() >= 1) |
| | | 224 | | { |
| | 1 | 225 | | if (_incoming) |
| | | 226 | | { |
| | 1 | 227 | | _instance.logger().trace( |
| | 1 | 228 | | _instance.traceCategory(), |
| | 1 | 229 | | "accepted " + protocol() + " connection HTTP upgrade request\n" + ToString()); |
| | | 230 | | } |
| | | 231 | | else |
| | | 232 | | { |
| | 1 | 233 | | _instance.logger().trace( |
| | 1 | 234 | | _instance.traceCategory(), |
| | 1 | 235 | | protocol() + " connection HTTP upgrade request accepted\n" + ToString()); |
| | | 236 | | } |
| | | 237 | | } |
| | | 238 | | |
| | 1 | 239 | | return SocketOperation.None; |
| | 1 | 240 | | } |
| | | 241 | | |
| | | 242 | | public int closing(bool initiator, Ice.LocalException reason) |
| | | 243 | | { |
| | 1 | 244 | | if (_instance.traceLevel() >= 1) |
| | | 245 | | { |
| | 1 | 246 | | _instance.logger().trace( |
| | 1 | 247 | | _instance.traceCategory(), |
| | 1 | 248 | | "gracefully closing " + protocol() + " connection\n" + ToString()); |
| | | 249 | | } |
| | | 250 | | |
| | 1 | 251 | | int s = _nextState == StateOpened ? _state : _nextState; |
| | | 252 | | |
| | 1 | 253 | | if (s == StateClosingRequestPending && _closingInitiator) |
| | | 254 | | { |
| | | 255 | | // |
| | | 256 | | // If we initiated a close connection but also received a |
| | | 257 | | // close connection, we assume we didn't initiate the |
| | | 258 | | // close and we send the close frame now. This is to |
| | | 259 | | // ensure that if both peers close the connection at the same |
| | | 260 | | // time we don't hang having both peers waiting for the close |
| | | 261 | | // frame of the other. |
| | | 262 | | // |
| | | 263 | | Debug.Assert(!initiator); |
| | 1 | 264 | | _closingInitiator = false; |
| | 1 | 265 | | return SocketOperation.Write; |
| | | 266 | | } |
| | 1 | 267 | | else if (s >= StateClosingRequestPending) |
| | | 268 | | { |
| | 0 | 269 | | return SocketOperation.None; |
| | | 270 | | } |
| | | 271 | | |
| | 1 | 272 | | _closingInitiator = initiator; |
| | 1 | 273 | | if (reason is Ice.CloseConnectionException) |
| | | 274 | | { |
| | 1 | 275 | | _closingReason = CLOSURE_NORMAL; |
| | | 276 | | } |
| | 1 | 277 | | else if (reason is Ice.ObjectAdapterDeactivatedException || |
| | 1 | 278 | | reason is Ice.ObjectAdapterDestroyedException || |
| | 1 | 279 | | reason is Ice.CommunicatorDestroyedException) |
| | | 280 | | { |
| | 1 | 281 | | _closingReason = CLOSURE_SHUTDOWN; |
| | | 282 | | } |
| | 1 | 283 | | else if (reason is Ice.ProtocolException) |
| | | 284 | | { |
| | 0 | 285 | | _closingReason = CLOSURE_PROTOCOL_ERROR; |
| | | 286 | | } |
| | 1 | 287 | | if (_state == StateOpened) |
| | | 288 | | { |
| | 1 | 289 | | _state = StateClosingRequestPending; |
| | 1 | 290 | | return initiator ? SocketOperation.Read : SocketOperation.Write; |
| | | 291 | | } |
| | | 292 | | else |
| | | 293 | | { |
| | 0 | 294 | | _nextState = StateClosingRequestPending; |
| | 0 | 295 | | return SocketOperation.None; |
| | | 296 | | } |
| | | 297 | | } |
| | | 298 | | |
| | | 299 | | public void close() |
| | | 300 | | { |
| | 1 | 301 | | _delegate.close(); |
| | 1 | 302 | | _state = StateClosed; |
| | | 303 | | |
| | | 304 | | // |
| | | 305 | | // Clear the buffers now instead of waiting for destruction. |
| | | 306 | | // |
| | 1 | 307 | | if (!_readPending) |
| | | 308 | | { |
| | 1 | 309 | | _readBuffer.clear(); |
| | | 310 | | } |
| | 1 | 311 | | if (!_writePending) |
| | | 312 | | { |
| | 1 | 313 | | _writeBuffer.clear(); |
| | | 314 | | } |
| | 1 | 315 | | } |
| | | 316 | | |
| | | 317 | | public EndpointI bind() |
| | | 318 | | { |
| | | 319 | | Debug.Assert(false); |
| | 0 | 320 | | return null; |
| | | 321 | | } |
| | | 322 | | |
| | 1 | 323 | | public void destroy() => _delegate.destroy(); |
| | | 324 | | |
| | | 325 | | public int write(Buffer buf) |
| | | 326 | | { |
| | 1 | 327 | | if (_writePending) |
| | | 328 | | { |
| | 0 | 329 | | return SocketOperation.Write; |
| | | 330 | | } |
| | | 331 | | |
| | 1 | 332 | | if (_state < StateOpened) |
| | | 333 | | { |
| | 1 | 334 | | if (_state < StateConnected) |
| | | 335 | | { |
| | 1 | 336 | | return _delegate.write(buf); |
| | | 337 | | } |
| | | 338 | | else |
| | | 339 | | { |
| | 1 | 340 | | return _delegate.write(_writeBuffer); |
| | | 341 | | } |
| | | 342 | | } |
| | | 343 | | |
| | 1 | 344 | | int s = SocketOperation.None; |
| | | 345 | | do |
| | | 346 | | { |
| | 1 | 347 | | if (preWrite(buf)) |
| | | 348 | | { |
| | 1 | 349 | | if (_writeState == WriteStateFlush) |
| | | 350 | | { |
| | | 351 | | // |
| | | 352 | | // Invoke write() even though there's nothing to write. |
| | | 353 | | // |
| | | 354 | | Debug.Assert(!buf.b.hasRemaining()); |
| | 0 | 355 | | s = _delegate.write(buf); |
| | | 356 | | } |
| | | 357 | | |
| | 1 | 358 | | if (s == SocketOperation.None && _writeBuffer.b.hasRemaining()) |
| | | 359 | | { |
| | 1 | 360 | | s = _delegate.write(_writeBuffer); |
| | | 361 | | } |
| | 1 | 362 | | else if (s == SocketOperation.None && _incoming && !buf.empty() && _writeState == WriteStatePayload) |
| | | 363 | | { |
| | 1 | 364 | | s = _delegate.write(buf); |
| | | 365 | | } |
| | | 366 | | } |
| | | 367 | | } |
| | 1 | 368 | | while (postWrite(buf, s)); |
| | | 369 | | |
| | 1 | 370 | | if (s != SocketOperation.None) |
| | | 371 | | { |
| | 1 | 372 | | return s; |
| | | 373 | | } |
| | 1 | 374 | | if (_state == StateClosingResponsePending && !_closingInitiator) |
| | | 375 | | { |
| | 1 | 376 | | return SocketOperation.Read; |
| | | 377 | | } |
| | 1 | 378 | | return SocketOperation.None; |
| | | 379 | | } |
| | | 380 | | |
| | | 381 | | public int read(Buffer buf, ref bool hasMoreData) |
| | | 382 | | { |
| | 1 | 383 | | if (_readPending) |
| | | 384 | | { |
| | 0 | 385 | | return SocketOperation.Read; |
| | | 386 | | } |
| | | 387 | | |
| | 1 | 388 | | if (_state < StateOpened) |
| | | 389 | | { |
| | 1 | 390 | | if (_state < StateConnected) |
| | | 391 | | { |
| | 1 | 392 | | return _delegate.read(buf, ref hasMoreData); |
| | | 393 | | } |
| | | 394 | | else |
| | | 395 | | { |
| | 1 | 396 | | if (_delegate.read(_readBuffer, ref hasMoreData) == SocketOperation.Write) |
| | | 397 | | { |
| | 0 | 398 | | return SocketOperation.Write; |
| | | 399 | | } |
| | | 400 | | else |
| | | 401 | | { |
| | 1 | 402 | | return SocketOperation.None; |
| | | 403 | | } |
| | | 404 | | } |
| | | 405 | | } |
| | | 406 | | |
| | 1 | 407 | | if (!buf.b.hasRemaining()) |
| | | 408 | | { |
| | 1 | 409 | | hasMoreData |= _readBufferPos < _readBuffer.b.position(); |
| | 1 | 410 | | return SocketOperation.None; |
| | | 411 | | } |
| | | 412 | | |
| | | 413 | | int s; |
| | | 414 | | do |
| | | 415 | | { |
| | 1 | 416 | | if (preRead(buf)) |
| | | 417 | | { |
| | 1 | 418 | | if (_readState == ReadStatePayload) |
| | | 419 | | { |
| | | 420 | | // |
| | | 421 | | // If the payload length is smaller than what remains to be read, we read |
| | | 422 | | // no more than the payload length. The remaining of the buffer will be |
| | | 423 | | // sent over in another frame. |
| | | 424 | | // |
| | 1 | 425 | | int readSz = _readPayloadLength - (buf.b.position() - _readStart); |
| | 1 | 426 | | if (buf.b.remaining() > readSz) |
| | | 427 | | { |
| | 0 | 428 | | int size = buf.size(); |
| | 0 | 429 | | buf.resize(buf.b.position() + readSz, true); |
| | 0 | 430 | | s = _delegate.read(buf, ref hasMoreData); |
| | 0 | 431 | | buf.resize(size, true); |
| | | 432 | | } |
| | | 433 | | else |
| | | 434 | | { |
| | 1 | 435 | | s = _delegate.read(buf, ref hasMoreData); |
| | | 436 | | } |
| | | 437 | | } |
| | | 438 | | else |
| | | 439 | | { |
| | 1 | 440 | | s = _delegate.read(_readBuffer, ref hasMoreData); |
| | | 441 | | } |
| | | 442 | | |
| | 1 | 443 | | if (s == SocketOperation.Write) |
| | | 444 | | { |
| | 0 | 445 | | postRead(buf); |
| | 0 | 446 | | return s; |
| | | 447 | | } |
| | | 448 | | } |
| | | 449 | | } |
| | 1 | 450 | | while (postRead(buf)); |
| | | 451 | | |
| | 1 | 452 | | if (!buf.b.hasRemaining()) |
| | | 453 | | { |
| | 1 | 454 | | hasMoreData |= _readBufferPos < _readBuffer.b.position(); |
| | 1 | 455 | | s = SocketOperation.None; |
| | | 456 | | } |
| | | 457 | | else |
| | | 458 | | { |
| | 1 | 459 | | hasMoreData = false; |
| | 1 | 460 | | s = SocketOperation.Read; |
| | | 461 | | } |
| | | 462 | | |
| | 1 | 463 | | if (((_state == StateClosingRequestPending && !_closingInitiator) || |
| | 1 | 464 | | (_state == StateClosingResponsePending && _closingInitiator) || |
| | 1 | 465 | | _state == StatePingPending || |
| | 1 | 466 | | _state == StatePongPending) && |
| | 1 | 467 | | _writeState == WriteStateHeader) |
| | | 468 | | { |
| | | 469 | | // We have things to write, ask to be notified when writes are ready. |
| | 1 | 470 | | s |= SocketOperation.Write; |
| | | 471 | | } |
| | | 472 | | |
| | 1 | 473 | | return s; |
| | | 474 | | } |
| | | 475 | | |
| | | 476 | | public bool startRead(Buffer buf, AsyncCallback callback, object state) |
| | | 477 | | { |
| | 1 | 478 | | _readPending = true; |
| | 1 | 479 | | if (_state < StateOpened) |
| | | 480 | | { |
| | 1 | 481 | | _finishRead = true; |
| | 1 | 482 | | if (_state < StateConnected) |
| | | 483 | | { |
| | 1 | 484 | | return _delegate.startRead(buf, callback, state); |
| | | 485 | | } |
| | | 486 | | else |
| | | 487 | | { |
| | 1 | 488 | | return _delegate.startRead(_readBuffer, callback, state); |
| | | 489 | | } |
| | | 490 | | } |
| | | 491 | | |
| | 1 | 492 | | if (preRead(buf)) |
| | | 493 | | { |
| | 1 | 494 | | _finishRead = true; |
| | 1 | 495 | | if (_readState == ReadStatePayload) |
| | | 496 | | { |
| | | 497 | | // |
| | | 498 | | // If the payload length is smaller than what remains to be read, we read |
| | | 499 | | // no more than the payload length. The remaining of the buffer will be |
| | | 500 | | // sent over in another frame. |
| | | 501 | | // |
| | 1 | 502 | | int readSz = _readPayloadLength - (buf.b.position() - _readStart); |
| | 1 | 503 | | if (buf.b.remaining() > readSz) |
| | | 504 | | { |
| | 0 | 505 | | int size = buf.size(); |
| | 0 | 506 | | buf.resize(buf.b.position() + readSz, true); |
| | 0 | 507 | | bool completedSynchronously = _delegate.startRead(buf, callback, state); |
| | 0 | 508 | | buf.resize(size, true); |
| | 0 | 509 | | return completedSynchronously; |
| | | 510 | | } |
| | | 511 | | else |
| | | 512 | | { |
| | 1 | 513 | | return _delegate.startRead(buf, callback, state); |
| | | 514 | | } |
| | | 515 | | } |
| | | 516 | | else |
| | | 517 | | { |
| | 1 | 518 | | return _delegate.startRead(_readBuffer, callback, state); |
| | | 519 | | } |
| | | 520 | | } |
| | | 521 | | else |
| | | 522 | | { |
| | 1 | 523 | | return true; |
| | | 524 | | } |
| | | 525 | | } |
| | | 526 | | |
| | | 527 | | public void finishRead(Buffer buf) |
| | | 528 | | { |
| | | 529 | | Debug.Assert(_readPending); |
| | 1 | 530 | | _readPending = false; |
| | | 531 | | |
| | 1 | 532 | | if (_state < StateOpened) |
| | | 533 | | { |
| | | 534 | | Debug.Assert(_finishRead); |
| | 1 | 535 | | _finishRead = false; |
| | 1 | 536 | | if (_state < StateConnected) |
| | | 537 | | { |
| | 1 | 538 | | _delegate.finishRead(buf); |
| | | 539 | | } |
| | | 540 | | else |
| | | 541 | | { |
| | 1 | 542 | | _delegate.finishRead(_readBuffer); |
| | | 543 | | } |
| | 1 | 544 | | return; |
| | | 545 | | } |
| | | 546 | | |
| | 1 | 547 | | if (!_finishRead) |
| | | 548 | | { |
| | | 549 | | // Nothing to do. |
| | | 550 | | } |
| | 1 | 551 | | else if (_readState == ReadStatePayload) |
| | | 552 | | { |
| | | 553 | | Debug.Assert(_finishRead); |
| | 1 | 554 | | _finishRead = false; |
| | 1 | 555 | | _delegate.finishRead(buf); |
| | | 556 | | } |
| | | 557 | | else |
| | | 558 | | { |
| | | 559 | | Debug.Assert(_finishRead); |
| | 1 | 560 | | _finishRead = false; |
| | 1 | 561 | | _delegate.finishRead(_readBuffer); |
| | | 562 | | } |
| | | 563 | | |
| | 1 | 564 | | if (_state == StateClosed) |
| | | 565 | | { |
| | 0 | 566 | | _readBuffer.clear(); |
| | 0 | 567 | | return; |
| | | 568 | | } |
| | | 569 | | |
| | 1 | 570 | | postRead(buf); |
| | 1 | 571 | | } |
| | | 572 | | |
| | | 573 | | public bool startWrite(Buffer buf, AsyncCallback callback, object state, out bool messageWritten) |
| | | 574 | | { |
| | 1 | 575 | | _writePending = true; |
| | 1 | 576 | | if (_state < StateOpened) |
| | | 577 | | { |
| | 1 | 578 | | if (_state < StateConnected) |
| | | 579 | | { |
| | 1 | 580 | | return _delegate.startWrite(buf, callback, state, out messageWritten); |
| | | 581 | | } |
| | | 582 | | else |
| | | 583 | | { |
| | 1 | 584 | | return _delegate.startWrite(_writeBuffer, callback, state, out messageWritten); |
| | | 585 | | } |
| | | 586 | | } |
| | | 587 | | |
| | 1 | 588 | | if (preWrite(buf)) |
| | | 589 | | { |
| | 1 | 590 | | if (_writeBuffer.b.hasRemaining()) |
| | | 591 | | { |
| | 1 | 592 | | return _delegate.startWrite(_writeBuffer, callback, state, out messageWritten); |
| | | 593 | | } |
| | | 594 | | else |
| | | 595 | | { |
| | | 596 | | Debug.Assert(_incoming); |
| | 1 | 597 | | return _delegate.startWrite(buf, callback, state, out messageWritten); |
| | | 598 | | } |
| | | 599 | | } |
| | | 600 | | else |
| | | 601 | | { |
| | 0 | 602 | | messageWritten = true; |
| | 0 | 603 | | return false; |
| | | 604 | | } |
| | | 605 | | } |
| | | 606 | | |
| | | 607 | | public void finishWrite(Buffer buf) |
| | | 608 | | { |
| | 1 | 609 | | _writePending = false; |
| | | 610 | | |
| | 1 | 611 | | if (_state < StateOpened) |
| | | 612 | | { |
| | 1 | 613 | | if (_state < StateConnected) |
| | | 614 | | { |
| | 1 | 615 | | _delegate.finishWrite(buf); |
| | | 616 | | } |
| | | 617 | | else |
| | | 618 | | { |
| | 1 | 619 | | _delegate.finishWrite(_writeBuffer); |
| | | 620 | | } |
| | 1 | 621 | | return; |
| | | 622 | | } |
| | | 623 | | |
| | 1 | 624 | | if (_writeBuffer.b.hasRemaining()) |
| | | 625 | | { |
| | 1 | 626 | | _delegate.finishWrite(_writeBuffer); |
| | | 627 | | } |
| | 1 | 628 | | else if (!buf.empty() && buf.b.hasRemaining()) |
| | | 629 | | { |
| | | 630 | | Debug.Assert(_incoming); |
| | 1 | 631 | | _delegate.finishWrite(buf); |
| | | 632 | | } |
| | | 633 | | |
| | 1 | 634 | | if (_state == StateClosed) |
| | | 635 | | { |
| | 0 | 636 | | _writeBuffer.clear(); |
| | 0 | 637 | | return; |
| | | 638 | | } |
| | | 639 | | |
| | 1 | 640 | | postWrite(buf, SocketOperation.None); |
| | 1 | 641 | | } |
| | | 642 | | |
| | 1 | 643 | | public string protocol() => _instance.protocol(); |
| | | 644 | | |
| | | 645 | | public ConnectionInfo getInfo(bool incoming, string adapterName, string connectionId) => |
| | 1 | 646 | | new WSConnectionInfo(_delegate.getInfo(incoming, adapterName, connectionId), _parser.getHeaders()); |
| | | 647 | | |
| | 1 | 648 | | public void checkSendSize(Buffer buf) => _delegate.checkSendSize(buf); |
| | | 649 | | |
| | 1 | 650 | | public void setBufferSize(int rcvSize, int sndSize) => _delegate.setBufferSize(rcvSize, sndSize); |
| | | 651 | | |
| | 1 | 652 | | public override string ToString() => _delegate.ToString(); |
| | | 653 | | |
| | 0 | 654 | | public string toDetailedString() => _delegate.toDetailedString(); |
| | | 655 | | |
| | 1 | 656 | | internal |
| | 1 | 657 | | WSTransceiver(ProtocolInstance instance, Transceiver del, string host, string resource) |
| | | 658 | | { |
| | 1 | 659 | | init(instance, del); |
| | 1 | 660 | | _host = host; |
| | 1 | 661 | | _resource = resource; |
| | 1 | 662 | | _incoming = false; |
| | | 663 | | |
| | | 664 | | // |
| | | 665 | | // Use a 16KB write buffer size. We use 16KB for the write |
| | | 666 | | // buffer size because all the data needs to be copied to the |
| | | 667 | | // write buffer for the purpose of masking. A 16KB buffer |
| | | 668 | | // appears to be a good compromise to reduce the number of |
| | | 669 | | // socket write calls and not consume too much memory. |
| | | 670 | | // |
| | 1 | 671 | | _writeBufferSize = 16 * 1024; |
| | | 672 | | |
| | | 673 | | // |
| | | 674 | | // Write and read buffer size must be large enough to hold the frame header! |
| | | 675 | | // |
| | | 676 | | Debug.Assert(_writeBufferSize > 256); |
| | | 677 | | Debug.Assert(_readBufferSize > 256); |
| | 1 | 678 | | } |
| | | 679 | | |
| | 1 | 680 | | internal WSTransceiver(ProtocolInstance instance, Transceiver del, HashSet<string> allowedOrigins) |
| | | 681 | | { |
| | 1 | 682 | | init(instance, del); |
| | 1 | 683 | | _host = ""; |
| | 1 | 684 | | _resource = ""; |
| | 1 | 685 | | _incoming = true; |
| | 1 | 686 | | _allowedOrigins = allowedOrigins; |
| | | 687 | | |
| | | 688 | | // |
| | | 689 | | // Write and read buffer size must be large enough to hold the frame header! |
| | | 690 | | // |
| | | 691 | | Debug.Assert(_writeBufferSize > 256); |
| | | 692 | | Debug.Assert(_readBufferSize > 256); |
| | 1 | 693 | | } |
| | | 694 | | |
| | | 695 | | // Returns true if the Connection header field - a comma-separated list of tokens, already trimmed and lowercased |
| | | 696 | | // by HttpParser.getHeader - includes the "upgrade" token required by RFC 6455 section 4. |
| | | 697 | | private static bool hasUpgradeToken(string connectionField) => |
| | 1 | 698 | | connectionField.Split(',').Any(token => token.Trim().Equals("upgrade", StringComparison.Ordinal)); |
| | | 699 | | |
| | | 700 | | private void init(ProtocolInstance instance, Transceiver del) |
| | | 701 | | { |
| | 1 | 702 | | _instance = instance; |
| | 1 | 703 | | _delegate = del; |
| | 1 | 704 | | _state = StateInitializeDelegate; |
| | 1 | 705 | | _parser = new HttpParser(); |
| | 1 | 706 | | _readState = ReadStateOpcode; |
| | 1 | 707 | | _readBuffer = new Buffer(ByteBuffer.ByteOrder.BigEndian); // Network byte order |
| | 1 | 708 | | _readBufferSize = 1024; |
| | 1 | 709 | | _readLastFrame = true; |
| | 1 | 710 | | _readOpCode = 0; |
| | 1 | 711 | | _readHeaderLength = 0; |
| | 1 | 712 | | _readPayloadLength = 0; |
| | 1 | 713 | | _writeState = WriteStateHeader; |
| | 1 | 714 | | _writeBuffer = new Buffer(ByteBuffer.ByteOrder.BigEndian); // Network byte order |
| | 1 | 715 | | _writeBufferSize = 1024; |
| | 1 | 716 | | _readPending = false; |
| | 1 | 717 | | _finishRead = false; |
| | 1 | 718 | | _writePending = false; |
| | 1 | 719 | | _readMask = new byte[4]; |
| | 1 | 720 | | _writeMask = new byte[4]; |
| | 1 | 721 | | _key = ""; |
| | 1 | 722 | | _pingPayload = []; |
| | 1 | 723 | | } |
| | | 724 | | |
| | | 725 | | private void handleRequest(Buffer responseBuffer) |
| | | 726 | | { |
| | | 727 | | // |
| | | 728 | | // The opening handshake must be a GET request (RFC 6455 section 4.1). We check the message type first |
| | | 729 | | // because method() (used just below) and uri() (used later) assert the parser holds a request. |
| | | 730 | | // |
| | 1 | 731 | | if (_parser.type() != HttpParser.Type.Request) |
| | | 732 | | { |
| | 0 | 733 | | throw new WebSocketException("WebSocket handshake is not an HTTP request"); |
| | | 734 | | } |
| | 1 | 735 | | if (_parser.method() != "GET") |
| | | 736 | | { |
| | 0 | 737 | | throw new WebSocketException($"unsupported HTTP method '{_parser.method()}' for WebSocket handshake"); |
| | | 738 | | } |
| | | 739 | | |
| | | 740 | | // |
| | | 741 | | // HTTP/1.1 |
| | | 742 | | // |
| | 1 | 743 | | if (_parser.versionMajor() != 1 || _parser.versionMinor() != 1) |
| | | 744 | | { |
| | 0 | 745 | | throw new WebSocketException("unsupported HTTP version"); |
| | | 746 | | } |
| | | 747 | | |
| | | 748 | | // |
| | | 749 | | // "An |Upgrade| header field containing the value 'websocket', |
| | | 750 | | // treated as an ASCII case-insensitive value." |
| | | 751 | | // |
| | 1 | 752 | | string val = _parser.getHeader("Upgrade", true); |
| | 1 | 753 | | if (val == null) |
| | | 754 | | { |
| | 0 | 755 | | throw new WebSocketException("missing value for Upgrade field"); |
| | | 756 | | } |
| | 1 | 757 | | else if (val != "websocket") |
| | | 758 | | { |
| | 0 | 759 | | throw new WebSocketException("invalid value '" + val + "' for Upgrade field"); |
| | | 760 | | } |
| | | 761 | | |
| | | 762 | | // |
| | | 763 | | // "A |Connection| header field that includes the token 'Upgrade', |
| | | 764 | | // treated as an ASCII case-insensitive value. |
| | | 765 | | // |
| | 1 | 766 | | val = _parser.getHeader("Connection", true); |
| | 1 | 767 | | if (val == null) |
| | | 768 | | { |
| | 0 | 769 | | throw new WebSocketException("missing value for Connection field"); |
| | | 770 | | } |
| | 1 | 771 | | else if (!hasUpgradeToken(val)) |
| | | 772 | | { |
| | 1 | 773 | | throw new WebSocketException("invalid value '" + val + "' for Connection field"); |
| | | 774 | | } |
| | | 775 | | |
| | | 776 | | // |
| | | 777 | | // "A |Sec-WebSocket-Version| header field, with a value of 13." |
| | | 778 | | // |
| | 1 | 779 | | val = _parser.getHeader("Sec-WebSocket-Version", false); |
| | 1 | 780 | | if (val == null) |
| | | 781 | | { |
| | 0 | 782 | | throw new WebSocketException("missing value for WebSocket version"); |
| | | 783 | | } |
| | 1 | 784 | | else if (val != "13") |
| | | 785 | | { |
| | 0 | 786 | | throw new WebSocketException("unsupported WebSocket version '" + val + "'"); |
| | | 787 | | } |
| | | 788 | | |
| | | 789 | | // |
| | | 790 | | // "Optionally, a |Sec-WebSocket-Protocol| header field, with a list |
| | | 791 | | // of values indicating which protocols the client would like to |
| | | 792 | | // speak, ordered by preference." |
| | | 793 | | // |
| | 1 | 794 | | bool addProtocol = false; |
| | 1 | 795 | | val = _parser.getHeader("Sec-WebSocket-Protocol", true); |
| | 1 | 796 | | if (val != null) |
| | | 797 | | { |
| | 1 | 798 | | string[] protocols = Ice.UtilInternal.StringUtil.splitString(val, ",") ?? |
| | 1 | 799 | | throw new WebSocketException("invalid value '" + val + "' for WebSocket protocol"); |
| | 1 | 800 | | foreach (string p in protocols) |
| | | 801 | | { |
| | 1 | 802 | | if (!p.Trim().Equals(_iceProtocol, StringComparison.Ordinal)) |
| | | 803 | | { |
| | 0 | 804 | | throw new WebSocketException("unknown value '" + p + "' for WebSocket protocol"); |
| | | 805 | | } |
| | 1 | 806 | | addProtocol = true; |
| | | 807 | | } |
| | | 808 | | } |
| | | 809 | | |
| | | 810 | | // |
| | | 811 | | // "A |Sec-WebSocket-Key| header field with a base64-encoded |
| | | 812 | | // value that, when decoded, is 16 bytes in length." |
| | | 813 | | // |
| | 1 | 814 | | string key = _parser.getHeader("Sec-WebSocket-Key", false) ?? |
| | 1 | 815 | | throw new WebSocketException("missing value for WebSocket key"); |
| | | 816 | | try |
| | | 817 | | { |
| | 1 | 818 | | byte[] decodedKey = Convert.FromBase64String(key); |
| | 1 | 819 | | if (decodedKey.Length != 16) |
| | | 820 | | { |
| | 0 | 821 | | throw new WebSocketException("WebSocket key '" + key + "' has invalid length"); |
| | | 822 | | } |
| | 1 | 823 | | } |
| | 0 | 824 | | catch (FormatException) |
| | | 825 | | { |
| | 0 | 826 | | throw new WebSocketException("invalid base64 value '" + key + "' for WebSocket key"); |
| | | 827 | | } |
| | | 828 | | |
| | | 829 | | // |
| | | 830 | | // Optionally validate the Origin header against the adapter's allowed-origins list. |
| | | 831 | | // Browsers always send Origin; non-browser clients do not, so an absent header bypasses the check. |
| | | 832 | | // A wildcard ("*") allowlist is normalized to an empty set at parse time, so empty here means "no enforcement". |
| | | 833 | | // |
| | 1 | 834 | | if (_allowedOrigins.Count > 0) |
| | | 835 | | { |
| | 1 | 836 | | string origin = _parser.getHeader("Origin", false); |
| | 1 | 837 | | if (origin is not null) |
| | | 838 | | { |
| | | 839 | | string canonical; |
| | | 840 | | try |
| | | 841 | | { |
| | 1 | 842 | | canonical = canonicalizeOrigin(origin.Trim()); |
| | 1 | 843 | | } |
| | 1 | 844 | | catch (ArgumentException) |
| | | 845 | | { |
| | 1 | 846 | | throw new WebSocketException($"invalid Origin header '{origin}'"); |
| | | 847 | | } |
| | 1 | 848 | | if (!_allowedOrigins.Contains(canonical)) |
| | | 849 | | { |
| | 1 | 850 | | throw new WebSocketException($"origin '{origin}' is not allowed"); |
| | | 851 | | } |
| | | 852 | | } |
| | | 853 | | } |
| | | 854 | | |
| | | 855 | | // |
| | | 856 | | // Retain the target resource. |
| | | 857 | | // |
| | 1 | 858 | | _resource = _parser.uri(); |
| | | 859 | | |
| | | 860 | | // |
| | | 861 | | // Compose the response. |
| | | 862 | | // |
| | 1 | 863 | | var @out = new StringBuilder(); |
| | 1 | 864 | | @out.Append("HTTP/1.1 101 Switching Protocols\r\n"); |
| | 1 | 865 | | @out.Append("Upgrade: websocket\r\n"); |
| | 1 | 866 | | @out.Append("Connection: Upgrade\r\n"); |
| | 1 | 867 | | if (addProtocol) |
| | | 868 | | { |
| | 1 | 869 | | @out.Append("Sec-WebSocket-Protocol: " + _iceProtocol + "\r\n"); |
| | | 870 | | } |
| | | 871 | | |
| | | 872 | | // |
| | | 873 | | // The response includes: |
| | | 874 | | // |
| | | 875 | | // "A |Sec-WebSocket-Accept| header field. The value of this |
| | | 876 | | // header field is constructed by concatenating /key/, defined |
| | | 877 | | // above in step 4 in Section 4.2.2, with the string "258EAFA5- |
| | | 878 | | // E914-47DA-95CA-C5AB0DC85B11", taking the SHA-1 hash of this |
| | | 879 | | // concatenated value to obtain a 20-byte value and base64- |
| | | 880 | | // encoding (see Section 4 of [RFC4648]) this 20-byte hash. |
| | | 881 | | // |
| | 1 | 882 | | @out.Append("Sec-WebSocket-Accept: "); |
| | 1 | 883 | | string input = key + _wsUUID; |
| | | 884 | | #pragma warning disable CA5350 // SHA1 is used for compatibility with the WebSocket protocol |
| | 1 | 885 | | using var sha1 = SHA1.Create(); |
| | 1 | 886 | | byte[] hash = sha1.ComputeHash(_utf8.GetBytes(input)); |
| | | 887 | | #pragma warning restore CA5350 |
| | 1 | 888 | | @out.Append(Convert.ToBase64String(hash) + "\r\n\r\n"); // EOM |
| | | 889 | | |
| | 1 | 890 | | byte[] bytes = _utf8.GetBytes(@out.ToString()); |
| | | 891 | | Debug.Assert(bytes.Length == @out.Length); |
| | 1 | 892 | | responseBuffer.resize(bytes.Length, false); |
| | 1 | 893 | | responseBuffer.b.position(0); |
| | 1 | 894 | | responseBuffer.b.put(bytes); |
| | 1 | 895 | | responseBuffer.b.flip(); |
| | 1 | 896 | | } |
| | | 897 | | |
| | | 898 | | private void handleResponse() |
| | | 899 | | { |
| | | 900 | | string val; |
| | | 901 | | |
| | | 902 | | // |
| | | 903 | | // HTTP/1.1 |
| | | 904 | | // |
| | 1 | 905 | | if (_parser.versionMajor() != 1 || _parser.versionMinor() != 1) |
| | | 906 | | { |
| | 0 | 907 | | throw new WebSocketException("unsupported HTTP version"); |
| | | 908 | | } |
| | | 909 | | |
| | | 910 | | // |
| | | 911 | | // "If the status code received from the server is not 101, the |
| | | 912 | | // client handles the response per HTTP [RFC2616] procedures. In |
| | | 913 | | // particular, the client might perform authentication if it |
| | | 914 | | // receives a 401 status code; the server might redirect the client |
| | | 915 | | // using a 3xx status code (but clients are not required to follow |
| | | 916 | | // them), etc." |
| | | 917 | | // |
| | 1 | 918 | | if (_parser.status() != 101) |
| | | 919 | | { |
| | 0 | 920 | | var @out = new StringBuilder("unexpected status value " + _parser.status()); |
| | 0 | 921 | | if (_parser.reason().Length > 0) |
| | | 922 | | { |
| | 0 | 923 | | @out.Append(":\n" + _parser.reason()); |
| | | 924 | | } |
| | 0 | 925 | | throw new WebSocketException(@out.ToString()); |
| | | 926 | | } |
| | | 927 | | |
| | | 928 | | // |
| | | 929 | | // "If the response lacks an |Upgrade| header field or the |Upgrade| |
| | | 930 | | // header field contains a value that is not an ASCII case- |
| | | 931 | | // insensitive match for the value "websocket", the client MUST |
| | | 932 | | // _Fail the WebSocket Connection_." |
| | | 933 | | // |
| | 1 | 934 | | val = _parser.getHeader("Upgrade", true); |
| | 1 | 935 | | if (val == null) |
| | | 936 | | { |
| | 0 | 937 | | throw new WebSocketException("missing value for Upgrade field"); |
| | | 938 | | } |
| | 1 | 939 | | else if (val != "websocket") |
| | | 940 | | { |
| | 0 | 941 | | throw new WebSocketException("invalid value '" + val + "' for Upgrade field"); |
| | | 942 | | } |
| | | 943 | | |
| | | 944 | | // |
| | | 945 | | // "If the response lacks a |Connection| header field or the |
| | | 946 | | // |Connection| header field doesn't contain a token that is an |
| | | 947 | | // ASCII case-insensitive match for the value "Upgrade", the client |
| | | 948 | | // MUST _Fail the WebSocket Connection_." |
| | | 949 | | // |
| | 1 | 950 | | val = _parser.getHeader("Connection", true); |
| | 1 | 951 | | if (val == null) |
| | | 952 | | { |
| | 0 | 953 | | throw new WebSocketException("missing value for Connection field"); |
| | | 954 | | } |
| | 1 | 955 | | else if (!hasUpgradeToken(val)) |
| | | 956 | | { |
| | 0 | 957 | | throw new WebSocketException("invalid value '" + val + "' for Connection field"); |
| | | 958 | | } |
| | | 959 | | |
| | | 960 | | // |
| | | 961 | | // "If the response includes a |Sec-WebSocket-Protocol| header field |
| | | 962 | | // and this header field indicates the use of a subprotocol that was |
| | | 963 | | // not present in the client's handshake (the server has indicated a |
| | | 964 | | // subprotocol not requested by the client), the client MUST _Fail |
| | | 965 | | // the WebSocket Connection_." |
| | | 966 | | // |
| | 1 | 967 | | val = _parser.getHeader("Sec-WebSocket-Protocol", true); |
| | 1 | 968 | | if (val != null && !val.Equals(_iceProtocol, StringComparison.Ordinal)) |
| | | 969 | | { |
| | 0 | 970 | | throw new WebSocketException("invalid value '" + val + "' for WebSocket protocol"); |
| | | 971 | | } |
| | | 972 | | |
| | | 973 | | // |
| | | 974 | | // "If the response lacks a |Sec-WebSocket-Accept| header field or |
| | | 975 | | // the |Sec-WebSocket-Accept| contains a value other than the |
| | | 976 | | // base64-encoded SHA-1 of the concatenation of the |Sec-WebSocket- |
| | | 977 | | // Key| (as a string, not base64-decoded) with the string "258EAFA5- |
| | | 978 | | // E914-47DA-95CA-C5AB0DC85B11" but ignoring any leading and |
| | | 979 | | // trailing whitespace, the client MUST _Fail the WebSocket |
| | | 980 | | // Connection_." |
| | | 981 | | // |
| | 1 | 982 | | val = _parser.getHeader("Sec-WebSocket-Accept", false) ?? |
| | 1 | 983 | | throw new WebSocketException("missing value for Sec-WebSocket-Accept"); |
| | 1 | 984 | | string input = _key + _wsUUID; |
| | | 985 | | #pragma warning disable CA5350 // SHA1 is used for compatibility with the WebSocket protocol |
| | 1 | 986 | | using var sha1 = SHA1.Create(); |
| | 1 | 987 | | byte[] hash = sha1.ComputeHash(_utf8.GetBytes(input)); |
| | | 988 | | #pragma warning restore CA5350 |
| | 1 | 989 | | if (!val.Equals(Convert.ToBase64String(hash), StringComparison.Ordinal)) |
| | | 990 | | { |
| | 0 | 991 | | throw new WebSocketException("invalid value '" + val + "' for Sec-WebSocket-Accept"); |
| | | 992 | | } |
| | 1 | 993 | | } |
| | | 994 | | |
| | | 995 | | private bool preRead(Buffer buf) |
| | | 996 | | { |
| | | 997 | | while (true) |
| | | 998 | | { |
| | 1 | 999 | | if (_readState == ReadStateOpcode) |
| | | 1000 | | { |
| | | 1001 | | // |
| | | 1002 | | // Is there enough data available to read the opcode? |
| | | 1003 | | // |
| | 1 | 1004 | | if (!readBuffered(2)) |
| | | 1005 | | { |
| | 1 | 1006 | | return true; |
| | | 1007 | | } |
| | | 1008 | | |
| | | 1009 | | // |
| | | 1010 | | // Most-significant bit indicates whether this is the |
| | | 1011 | | // last frame. Least-significant four bits hold the |
| | | 1012 | | // opcode. |
| | | 1013 | | // |
| | 1 | 1014 | | int ch = _readBuffer.b.get(_readBufferPos++); |
| | 1 | 1015 | | _readOpCode = ch & 0xf; |
| | | 1016 | | |
| | | 1017 | | // |
| | | 1018 | | // No extension is negotiated, so the RSV1, RSV2, and RSV3 bits must all be 0. |
| | | 1019 | | // |
| | 1 | 1020 | | if ((ch & 0x70) != 0) |
| | | 1021 | | { |
| | 0 | 1022 | | throw new Ice.ProtocolException("invalid WebSocket frame: RSV bits must be 0"); |
| | | 1023 | | } |
| | | 1024 | | |
| | 1 | 1025 | | bool finalFrame = (ch & FLAG_FINAL) == FLAG_FINAL; |
| | | 1026 | | |
| | | 1027 | | // |
| | | 1028 | | // Remember if last frame if we're going to read a data or |
| | | 1029 | | // continuation frame, this is only for protocol |
| | | 1030 | | // correctness checking purpose. |
| | | 1031 | | // |
| | 1 | 1032 | | if (_readOpCode == OP_DATA) |
| | | 1033 | | { |
| | 1 | 1034 | | if (!_readLastFrame) |
| | | 1035 | | { |
| | 0 | 1036 | | throw new Ice.ProtocolException("invalid data frame, no FIN on previous frame"); |
| | | 1037 | | } |
| | 1 | 1038 | | _readLastFrame = finalFrame; |
| | | 1039 | | } |
| | 1 | 1040 | | else if (_readOpCode == OP_CONT) |
| | | 1041 | | { |
| | 0 | 1042 | | if (_readLastFrame) |
| | | 1043 | | { |
| | 0 | 1044 | | throw new Ice.ProtocolException("invalid continuation frame, previous frame FIN set"); |
| | | 1045 | | } |
| | 0 | 1046 | | _readLastFrame = finalFrame; |
| | | 1047 | | } |
| | | 1048 | | |
| | 1 | 1049 | | ch = _readBuffer.b.get(_readBufferPos++); |
| | | 1050 | | |
| | | 1051 | | // |
| | | 1052 | | // Check the MASK bit. Messages sent by a client must be masked; |
| | | 1053 | | // messages sent by a server must not be masked. |
| | | 1054 | | // |
| | 1 | 1055 | | bool masked = (ch & FLAG_MASKED) == FLAG_MASKED; |
| | 1 | 1056 | | if (masked != _incoming) |
| | | 1057 | | { |
| | 0 | 1058 | | throw new Ice.ProtocolException("invalid masking"); |
| | | 1059 | | } |
| | | 1060 | | |
| | | 1061 | | // |
| | | 1062 | | // Extract the payload length, which can have the following values: |
| | | 1063 | | // |
| | | 1064 | | // 0-125: The payload length |
| | | 1065 | | // 126: The subsequent two bytes contain the payload length |
| | | 1066 | | // 127: The subsequent eight bytes contain the payload length |
| | | 1067 | | // |
| | 1 | 1068 | | _readPayloadLength = ch & 0x7f; |
| | | 1069 | | |
| | | 1070 | | // |
| | | 1071 | | // RFC 6455 section 5.5: control frames (close, ping, and pong) must not be fragmented |
| | | 1072 | | // and must have a payload length of 125 bytes or less - they cannot use the 16-bit |
| | | 1073 | | // or 64-bit extended length encoding. Enforce this before allocating any payload |
| | | 1074 | | // buffer. |
| | | 1075 | | // |
| | 1 | 1076 | | if (_readOpCode == OP_CLOSE || _readOpCode == OP_PING || _readOpCode == OP_PONG) |
| | | 1077 | | { |
| | 1 | 1078 | | if (!finalFrame) |
| | | 1079 | | { |
| | 0 | 1080 | | throw new Ice.ProtocolException("invalid WebSocket control frame: the FIN bit is not set"); |
| | | 1081 | | } |
| | 1 | 1082 | | if (_readPayloadLength > 125) |
| | | 1083 | | { |
| | 0 | 1084 | | throw new Ice.ProtocolException( |
| | 0 | 1085 | | "invalid WebSocket control frame: the payload length exceeds 125 bytes"); |
| | | 1086 | | } |
| | | 1087 | | } |
| | | 1088 | | |
| | 1 | 1089 | | if (_readPayloadLength < 126) |
| | | 1090 | | { |
| | 1 | 1091 | | _readHeaderLength = 0; |
| | | 1092 | | } |
| | 1 | 1093 | | else if (_readPayloadLength == 126) |
| | | 1094 | | { |
| | 1 | 1095 | | _readHeaderLength = 2; // Need to read a 16-bit payload length. |
| | | 1096 | | } |
| | | 1097 | | else |
| | | 1098 | | { |
| | 1 | 1099 | | _readHeaderLength = 8; // Need to read a 64-bit payload length. |
| | | 1100 | | } |
| | 1 | 1101 | | if (masked) |
| | | 1102 | | { |
| | 1 | 1103 | | _readHeaderLength += 4; // Need to read a 32-bit mask. |
| | | 1104 | | } |
| | | 1105 | | |
| | 1 | 1106 | | _readState = ReadStateHeader; |
| | | 1107 | | } |
| | | 1108 | | |
| | 1 | 1109 | | if (_readState == ReadStateHeader) |
| | | 1110 | | { |
| | | 1111 | | // |
| | | 1112 | | // Is there enough data available to read the header? |
| | | 1113 | | // |
| | 1 | 1114 | | if (_readHeaderLength > 0 && !readBuffered(_readHeaderLength)) |
| | | 1115 | | { |
| | 1 | 1116 | | return true; |
| | | 1117 | | } |
| | | 1118 | | |
| | 1 | 1119 | | if (_readPayloadLength == 126) |
| | | 1120 | | { |
| | 1 | 1121 | | _readPayloadLength = _readBuffer.b.getShort(_readBufferPos); // Uses network byte order. |
| | 1 | 1122 | | if (_readPayloadLength < 0) |
| | | 1123 | | { |
| | 0 | 1124 | | _readPayloadLength += 65536; |
| | | 1125 | | } |
| | 1 | 1126 | | _readBufferPos += 2; |
| | | 1127 | | } |
| | 1 | 1128 | | else if (_readPayloadLength == 127) |
| | | 1129 | | { |
| | 1 | 1130 | | long l = _readBuffer.b.getLong(_readBufferPos); // Uses network byte order. |
| | 1 | 1131 | | _readBufferPos += 8; |
| | 1 | 1132 | | if (l < 0 || l > int.MaxValue) |
| | | 1133 | | { |
| | 0 | 1134 | | throw new Ice.ProtocolException("invalid WebSocket payload length: " + l); |
| | | 1135 | | } |
| | 1 | 1136 | | _readPayloadLength = (int)l; |
| | | 1137 | | } |
| | | 1138 | | |
| | | 1139 | | // |
| | | 1140 | | // Read the mask if this is an incoming connection. |
| | | 1141 | | // |
| | 1 | 1142 | | if (_incoming) |
| | | 1143 | | { |
| | | 1144 | | // |
| | | 1145 | | // We must have needed to read the mask. |
| | | 1146 | | // |
| | | 1147 | | Debug.Assert(_readBuffer.b.position() - _readBufferPos >= 4); |
| | 1 | 1148 | | for (int i = 0; i < 4; ++i) |
| | | 1149 | | { |
| | 1 | 1150 | | _readMask[i] = _readBuffer.b.get(_readBufferPos++); // Copy the mask. |
| | | 1151 | | } |
| | | 1152 | | } |
| | | 1153 | | |
| | 1 | 1154 | | switch (_readOpCode) |
| | | 1155 | | { |
| | | 1156 | | case OP_TEXT: // Text frame |
| | | 1157 | | { |
| | 0 | 1158 | | throw new Ice.ProtocolException("text frames not supported"); |
| | | 1159 | | } |
| | | 1160 | | case OP_DATA: // Data frame |
| | | 1161 | | case OP_CONT: // Continuation frame |
| | | 1162 | | { |
| | 1 | 1163 | | if (_instance.traceLevel() >= 2) |
| | | 1164 | | { |
| | 1 | 1165 | | _instance.logger().trace( |
| | 1 | 1166 | | _instance.traceCategory(), "received " + protocol() + |
| | 1 | 1167 | | (_readOpCode == OP_DATA ? " data" : " continuation") + |
| | 1 | 1168 | | " frame with payload length of " + _readPayloadLength + |
| | 1 | 1169 | | " bytes\n" + ToString()); |
| | | 1170 | | } |
| | | 1171 | | |
| | 1 | 1172 | | if (_readPayloadLength <= 0) |
| | | 1173 | | { |
| | 0 | 1174 | | throw new Ice.ProtocolException("payload length is 0"); |
| | | 1175 | | } |
| | 1 | 1176 | | _readState = ReadStatePayload; |
| | | 1177 | | Debug.Assert(buf.b.hasRemaining()); |
| | 1 | 1178 | | _readFrameStart = buf.b.position(); |
| | 1 | 1179 | | break; |
| | | 1180 | | } |
| | | 1181 | | case OP_CLOSE: // Connection close |
| | | 1182 | | { |
| | 1 | 1183 | | if (_instance.traceLevel() >= 2) |
| | | 1184 | | { |
| | 1 | 1185 | | _instance.logger().trace( |
| | 1 | 1186 | | _instance.traceCategory(), |
| | 1 | 1187 | | "received " + protocol() + " connection close frame\n" + ToString()); |
| | | 1188 | | } |
| | | 1189 | | |
| | 1 | 1190 | | _readState = ReadStateControlFrame; |
| | 1 | 1191 | | int s = _nextState == StateOpened ? _state : _nextState; |
| | 1 | 1192 | | if (s == StateClosingRequestPending) |
| | | 1193 | | { |
| | | 1194 | | // |
| | | 1195 | | // If we receive a close frame while we were actually |
| | | 1196 | | // waiting to send one, change the role and send a |
| | | 1197 | | // close frame response. |
| | | 1198 | | // |
| | 1 | 1199 | | if (!_closingInitiator) |
| | | 1200 | | { |
| | 0 | 1201 | | _closingInitiator = true; |
| | | 1202 | | } |
| | 1 | 1203 | | if (_state == StateClosingRequestPending) |
| | | 1204 | | { |
| | 1 | 1205 | | _state = StateClosingResponsePending; |
| | | 1206 | | } |
| | | 1207 | | else |
| | | 1208 | | { |
| | 0 | 1209 | | _nextState = StateClosingResponsePending; |
| | | 1210 | | } |
| | 1 | 1211 | | return false; // No longer interested in reading |
| | | 1212 | | } |
| | | 1213 | | else |
| | | 1214 | | { |
| | 1 | 1215 | | throw new Ice.ConnectionLostException(peerAddress: null); |
| | | 1216 | | } |
| | | 1217 | | } |
| | | 1218 | | case OP_PING: |
| | | 1219 | | { |
| | 1 | 1220 | | if (_instance.traceLevel() >= 2) |
| | | 1221 | | { |
| | 0 | 1222 | | _instance.logger().trace( |
| | 0 | 1223 | | _instance.traceCategory(), |
| | 0 | 1224 | | "received " + protocol() + " connection ping frame\n" + ToString()); |
| | | 1225 | | } |
| | 1 | 1226 | | _readState = ReadStateControlFrame; |
| | 1 | 1227 | | break; |
| | | 1228 | | } |
| | | 1229 | | case OP_PONG: // Pong |
| | | 1230 | | { |
| | 0 | 1231 | | if (_instance.traceLevel() >= 2) |
| | | 1232 | | { |
| | 0 | 1233 | | _instance.logger().trace( |
| | 0 | 1234 | | _instance.traceCategory(), |
| | 0 | 1235 | | "received " + protocol() + " connection pong frame\n" + ToString()); |
| | | 1236 | | } |
| | 0 | 1237 | | _readState = ReadStateControlFrame; |
| | 0 | 1238 | | break; |
| | | 1239 | | } |
| | | 1240 | | default: |
| | | 1241 | | { |
| | 0 | 1242 | | throw new Ice.ProtocolException("unsupported opcode: " + _readOpCode); |
| | | 1243 | | } |
| | | 1244 | | } |
| | | 1245 | | } |
| | | 1246 | | |
| | 1 | 1247 | | if (_readState == ReadStateControlFrame) |
| | | 1248 | | { |
| | 1 | 1249 | | if (_readPayloadLength > 0 && !readBuffered(_readPayloadLength)) |
| | | 1250 | | { |
| | 0 | 1251 | | return true; |
| | | 1252 | | } |
| | | 1253 | | |
| | 1 | 1254 | | if (_readPayloadLength > 0 && _readOpCode == OP_PING) |
| | | 1255 | | { |
| | 1 | 1256 | | _pingPayload = new byte[_readPayloadLength]; |
| | 1 | 1257 | | System.Buffer.BlockCopy( |
| | 1 | 1258 | | _readBuffer.b.rawBytes(), |
| | 1 | 1259 | | _readBufferPos, |
| | 1 | 1260 | | _pingPayload, |
| | 1 | 1261 | | 0, |
| | 1 | 1262 | | _readPayloadLength); |
| | 1 | 1263 | | if (_incoming) |
| | | 1264 | | { |
| | | 1265 | | // A client masks its frames (RFC 6455 section 5.3); unmask the ping payload so the echoed |
| | | 1266 | | // pong matches the ping (RFC 6455 section 5.5.3), like the data path does in postRead. |
| | 1 | 1267 | | for (int i = 0; i < _pingPayload.Length; ++i) |
| | | 1268 | | { |
| | 1 | 1269 | | _pingPayload[i] = (byte)(_pingPayload[i] ^ _readMask[i % 4]); |
| | | 1270 | | } |
| | | 1271 | | } |
| | | 1272 | | } |
| | | 1273 | | |
| | 1 | 1274 | | _readBufferPos += _readPayloadLength; |
| | 1 | 1275 | | _readPayloadLength = 0; |
| | | 1276 | | |
| | 1 | 1277 | | if (_readOpCode == OP_PING) |
| | | 1278 | | { |
| | 1 | 1279 | | if (_state == StateOpened) |
| | | 1280 | | { |
| | 1 | 1281 | | _state = StatePongPending; // Send pong frame now |
| | | 1282 | | } |
| | 0 | 1283 | | else if (_nextState < StatePongPending) |
| | | 1284 | | { |
| | 0 | 1285 | | _nextState = StatePongPending; // Send pong frame next |
| | | 1286 | | } |
| | | 1287 | | } |
| | | 1288 | | |
| | | 1289 | | // |
| | | 1290 | | // We've read the payload of the PING/PONG frame, we're ready |
| | | 1291 | | // to read a new frame. |
| | | 1292 | | // |
| | 1 | 1293 | | _readState = ReadStateOpcode; |
| | | 1294 | | } |
| | | 1295 | | |
| | 1 | 1296 | | if (_readState == ReadStatePayload) |
| | | 1297 | | { |
| | | 1298 | | // |
| | | 1299 | | // This must be assigned before the check for the buffer. If the buffer is empty |
| | | 1300 | | // or already read, postRead will return false. |
| | | 1301 | | // |
| | 1 | 1302 | | _readStart = buf.b.position(); |
| | | 1303 | | |
| | 1 | 1304 | | if (buf.empty() || !buf.b.hasRemaining()) |
| | | 1305 | | { |
| | 0 | 1306 | | return false; |
| | | 1307 | | } |
| | | 1308 | | |
| | 1 | 1309 | | int n = Math.Min(_readBuffer.b.position() - _readBufferPos, buf.b.remaining()); |
| | 1 | 1310 | | if (n > _readPayloadLength) |
| | | 1311 | | { |
| | 0 | 1312 | | n = _readPayloadLength; |
| | | 1313 | | } |
| | 1 | 1314 | | if (n > 0) |
| | | 1315 | | { |
| | 1 | 1316 | | System.Buffer.BlockCopy( |
| | 1 | 1317 | | _readBuffer.b.rawBytes(), |
| | 1 | 1318 | | _readBufferPos, |
| | 1 | 1319 | | buf.b.rawBytes(), |
| | 1 | 1320 | | buf.b.position(), |
| | 1 | 1321 | | n); |
| | 1 | 1322 | | buf.b.position(buf.b.position() + n); |
| | 1 | 1323 | | _readBufferPos += n; |
| | | 1324 | | } |
| | | 1325 | | |
| | | 1326 | | // |
| | | 1327 | | // Continue reading if we didn't read the full message, otherwise give back |
| | | 1328 | | // the control to the connection |
| | | 1329 | | // |
| | 1 | 1330 | | return buf.b.hasRemaining() && n < _readPayloadLength; |
| | | 1331 | | } |
| | | 1332 | | } |
| | | 1333 | | } |
| | | 1334 | | |
| | | 1335 | | private bool postRead(Buffer buf) |
| | | 1336 | | { |
| | 1 | 1337 | | if (_readState != ReadStatePayload) |
| | | 1338 | | { |
| | 1 | 1339 | | return _readStart < _readBuffer.b.position(); // Returns true if data was read. |
| | | 1340 | | } |
| | | 1341 | | |
| | 1 | 1342 | | if (_readStart == buf.b.position()) |
| | | 1343 | | { |
| | 1 | 1344 | | return false; // Nothing was read or nothing to read. |
| | | 1345 | | } |
| | | 1346 | | Debug.Assert(_readStart < buf.b.position()); |
| | | 1347 | | |
| | 1 | 1348 | | if (_incoming) |
| | | 1349 | | { |
| | | 1350 | | // |
| | | 1351 | | // Unmask the data we just read. |
| | | 1352 | | // |
| | 1 | 1353 | | int pos = buf.b.position(); |
| | 1 | 1354 | | byte[] arr = buf.b.rawBytes(); |
| | 1 | 1355 | | for (int n = _readStart; n < pos; ++n) |
| | | 1356 | | { |
| | 1 | 1357 | | arr[n] = (byte)(arr[n] ^ _readMask[(n - _readFrameStart) % 4]); |
| | | 1358 | | } |
| | | 1359 | | } |
| | | 1360 | | |
| | 1 | 1361 | | _readPayloadLength -= buf.b.position() - _readStart; |
| | 1 | 1362 | | _readStart = buf.b.position(); |
| | 1 | 1363 | | if (_readPayloadLength == 0) |
| | | 1364 | | { |
| | | 1365 | | // |
| | | 1366 | | // We've read the complete payload, we're ready to read a new frame. |
| | | 1367 | | // |
| | 1 | 1368 | | _readState = ReadStateOpcode; |
| | | 1369 | | } |
| | 1 | 1370 | | return buf.b.hasRemaining(); |
| | | 1371 | | } |
| | | 1372 | | |
| | | 1373 | | private bool preWrite(Buffer buf) |
| | | 1374 | | { |
| | 1 | 1375 | | if (_writeState == WriteStateHeader) |
| | | 1376 | | { |
| | 1 | 1377 | | if (_state == StateOpened) |
| | | 1378 | | { |
| | 1 | 1379 | | if (buf.empty() || !buf.b.hasRemaining()) |
| | | 1380 | | { |
| | 1 | 1381 | | return false; |
| | | 1382 | | } |
| | | 1383 | | |
| | | 1384 | | Debug.Assert(buf.b.position() == 0); |
| | 1 | 1385 | | prepareWriteHeader((byte)OP_DATA, buf.size()); |
| | | 1386 | | |
| | 1 | 1387 | | _writeState = WriteStatePayload; |
| | | 1388 | | } |
| | 1 | 1389 | | else if (_state == StatePingPending) |
| | | 1390 | | { |
| | 0 | 1391 | | prepareWriteHeader((byte)OP_PING, 0); // Don't send any payload |
| | | 1392 | | |
| | 0 | 1393 | | _writeState = WriteStateControlFrame; |
| | 0 | 1394 | | _writeBuffer.b.flip(); |
| | | 1395 | | } |
| | 1 | 1396 | | else if (_state == StatePongPending) |
| | | 1397 | | { |
| | 1 | 1398 | | prepareWriteHeader((byte)OP_PONG, _pingPayload.Length); |
| | 1 | 1399 | | if (_pingPayload.Length > _writeBuffer.b.remaining()) |
| | | 1400 | | { |
| | 0 | 1401 | | int pos = _writeBuffer.b.position(); |
| | 0 | 1402 | | _writeBuffer.resize(pos + _pingPayload.Length, false); |
| | 0 | 1403 | | _writeBuffer.b.position(pos); |
| | | 1404 | | } |
| | 1 | 1405 | | if (_incoming) |
| | | 1406 | | { |
| | | 1407 | | // Server-to-client frames are not masked (RFC 6455 section 5.1). |
| | 1 | 1408 | | _writeBuffer.b.put(_pingPayload); |
| | | 1409 | | } |
| | | 1410 | | else |
| | | 1411 | | { |
| | | 1412 | | // Client-to-server frames are masked with _writeMask, like the data and close paths; |
| | | 1413 | | // prepareWriteHeader set FLAG_MASKED, so the payload must be masked to match (RFC 6455 5.5.3). |
| | 0 | 1414 | | for (int i = 0; i < _pingPayload.Length; ++i) |
| | | 1415 | | { |
| | 0 | 1416 | | _writeBuffer.b.put((byte)(_pingPayload[i] ^ _writeMask[i % 4])); |
| | | 1417 | | } |
| | | 1418 | | } |
| | 1 | 1419 | | _pingPayload = []; |
| | | 1420 | | |
| | 1 | 1421 | | _writeState = WriteStateControlFrame; |
| | 1 | 1422 | | _writeBuffer.b.flip(); |
| | | 1423 | | } |
| | 1 | 1424 | | else if ((_state == StateClosingRequestPending && !_closingInitiator) || |
| | 1 | 1425 | | (_state == StateClosingResponsePending && _closingInitiator)) |
| | | 1426 | | { |
| | 1 | 1427 | | prepareWriteHeader((byte)OP_CLOSE, 2); |
| | | 1428 | | |
| | | 1429 | | // Write closing reason |
| | 1 | 1430 | | _writeBuffer.b.putShort((short)_closingReason); |
| | | 1431 | | |
| | 1 | 1432 | | if (!_incoming) |
| | | 1433 | | { |
| | | 1434 | | byte b; |
| | 1 | 1435 | | int pos = _writeBuffer.b.position() - 2; |
| | 1 | 1436 | | b = (byte)(_writeBuffer.b.get(pos) ^ _writeMask[0]); |
| | 1 | 1437 | | _writeBuffer.b.put(pos, b); |
| | 1 | 1438 | | pos++; |
| | 1 | 1439 | | b = (byte)(_writeBuffer.b.get(pos) ^ _writeMask[1]); |
| | 1 | 1440 | | _writeBuffer.b.put(pos, b); |
| | | 1441 | | } |
| | | 1442 | | |
| | 1 | 1443 | | _writeState = WriteStateControlFrame; |
| | 1 | 1444 | | _writeBuffer.b.flip(); |
| | | 1445 | | } |
| | | 1446 | | else |
| | | 1447 | | { |
| | | 1448 | | Debug.Assert(_state != StateClosed); |
| | 1 | 1449 | | return false; // Nothing to write in this state |
| | | 1450 | | } |
| | | 1451 | | |
| | 1 | 1452 | | _writePayloadLength = 0; |
| | | 1453 | | } |
| | | 1454 | | |
| | 1 | 1455 | | if (_writeState == WriteStatePayload) |
| | | 1456 | | { |
| | | 1457 | | // |
| | | 1458 | | // For an outgoing connection, each message must be masked with a random |
| | | 1459 | | // 32-bit value, so we copy the entire message into the internal buffer |
| | | 1460 | | // for writing. For incoming connections, we just copy the start of the |
| | | 1461 | | // message in the internal buffer after the header. If the message is |
| | | 1462 | | // larger, the reminder is sent directly from the message buffer to avoid |
| | | 1463 | | // copying. |
| | | 1464 | | // |
| | 1 | 1465 | | if (!_incoming && (_writePayloadLength == 0 || !_writeBuffer.b.hasRemaining())) |
| | | 1466 | | { |
| | 1 | 1467 | | if (!_writeBuffer.b.hasRemaining()) |
| | | 1468 | | { |
| | 1 | 1469 | | _writeBuffer.b.position(0); |
| | | 1470 | | } |
| | | 1471 | | |
| | 1 | 1472 | | int n = buf.b.position(); |
| | 1 | 1473 | | int sz = buf.size(); |
| | 1 | 1474 | | int pos = _writeBuffer.b.position(); |
| | 1 | 1475 | | int count = Math.Min(sz - n, _writeBuffer.b.remaining()); |
| | 1 | 1476 | | byte[] src = buf.b.rawBytes(); |
| | 1 | 1477 | | byte[] dest = _writeBuffer.b.rawBytes(); |
| | 1 | 1478 | | for (int i = 0; i < count; ++i, ++n, ++pos) |
| | | 1479 | | { |
| | 1 | 1480 | | dest[pos] = (byte)(src[n] ^ _writeMask[n % 4]); |
| | | 1481 | | } |
| | 1 | 1482 | | _writeBuffer.b.position(pos); |
| | 1 | 1483 | | _writePayloadLength = n; |
| | | 1484 | | |
| | 1 | 1485 | | _writeBuffer.b.flip(); |
| | | 1486 | | } |
| | 1 | 1487 | | else if (_writePayloadLength == 0) |
| | | 1488 | | { |
| | | 1489 | | Debug.Assert(_incoming); |
| | 1 | 1490 | | if (_writeBuffer.b.hasRemaining()) |
| | | 1491 | | { |
| | | 1492 | | Debug.Assert(buf.b.position() == 0); |
| | 1 | 1493 | | int n = Math.Min(_writeBuffer.b.remaining(), buf.b.remaining()); |
| | 1 | 1494 | | int pos = _writeBuffer.b.position(); |
| | 1 | 1495 | | System.Buffer.BlockCopy(buf.b.rawBytes(), 0, _writeBuffer.b.rawBytes(), pos, n); |
| | 1 | 1496 | | _writeBuffer.b.position(pos + n); |
| | 1 | 1497 | | _writePayloadLength = n; |
| | | 1498 | | } |
| | 1 | 1499 | | _writeBuffer.b.flip(); |
| | | 1500 | | } |
| | 1 | 1501 | | return true; |
| | | 1502 | | } |
| | 1 | 1503 | | else if (_writeState == WriteStateControlFrame) |
| | | 1504 | | { |
| | 1 | 1505 | | return _writeBuffer.b.hasRemaining(); |
| | | 1506 | | } |
| | | 1507 | | else |
| | | 1508 | | { |
| | | 1509 | | Debug.Assert(_writeState == WriteStateFlush); |
| | 0 | 1510 | | return true; |
| | | 1511 | | } |
| | | 1512 | | } |
| | | 1513 | | |
| | | 1514 | | private bool postWrite(Buffer buf, int status) |
| | | 1515 | | { |
| | 1 | 1516 | | if (_state > StateOpened && _writeState == WriteStateControlFrame) |
| | | 1517 | | { |
| | 1 | 1518 | | if (!_writeBuffer.b.hasRemaining()) |
| | | 1519 | | { |
| | 1 | 1520 | | if (_state == StatePingPending) |
| | | 1521 | | { |
| | 0 | 1522 | | if (_instance.traceLevel() >= 2) |
| | | 1523 | | { |
| | 0 | 1524 | | _instance.logger().trace( |
| | 0 | 1525 | | _instance.traceCategory(), |
| | 0 | 1526 | | "sent " + protocol() + " connection ping frame\n" + ToString()); |
| | | 1527 | | } |
| | | 1528 | | } |
| | 1 | 1529 | | else if (_state == StatePongPending) |
| | | 1530 | | { |
| | 1 | 1531 | | if (_instance.traceLevel() >= 2) |
| | | 1532 | | { |
| | 0 | 1533 | | _instance.logger().trace( |
| | 0 | 1534 | | _instance.traceCategory(), |
| | 0 | 1535 | | "sent " + protocol() + " connection pong frame\n" + ToString()); |
| | | 1536 | | } |
| | | 1537 | | } |
| | 1 | 1538 | | else if ((_state == StateClosingRequestPending && !_closingInitiator) || |
| | 1 | 1539 | | (_state == StateClosingResponsePending && _closingInitiator)) |
| | | 1540 | | { |
| | 1 | 1541 | | if (_instance.traceLevel() >= 2) |
| | | 1542 | | { |
| | 1 | 1543 | | _instance.logger().trace( |
| | 1 | 1544 | | _instance.traceCategory(), |
| | 1 | 1545 | | "sent " + protocol() + " connection close frame\n" + ToString()); |
| | | 1546 | | } |
| | | 1547 | | |
| | 1 | 1548 | | if (_state == StateClosingRequestPending && !_closingInitiator) |
| | | 1549 | | { |
| | 1 | 1550 | | _writeState = WriteStateHeader; |
| | 1 | 1551 | | _state = StateClosingResponsePending; |
| | 1 | 1552 | | return false; |
| | | 1553 | | } |
| | | 1554 | | else |
| | | 1555 | | { |
| | 1 | 1556 | | throw new Ice.ConnectionLostException(peerAddress: null); |
| | | 1557 | | } |
| | | 1558 | | } |
| | 0 | 1559 | | else if (_state == StateClosed) |
| | | 1560 | | { |
| | 0 | 1561 | | return false; |
| | | 1562 | | } |
| | | 1563 | | |
| | 1 | 1564 | | _state = _nextState; |
| | 1 | 1565 | | _nextState = StateOpened; |
| | 1 | 1566 | | _writeState = WriteStateHeader; |
| | | 1567 | | } |
| | | 1568 | | else |
| | | 1569 | | { |
| | 1 | 1570 | | return status == SocketOperation.None; |
| | | 1571 | | } |
| | | 1572 | | } |
| | | 1573 | | |
| | 1 | 1574 | | if ((!_incoming || buf.b.position() == 0) && _writePayloadLength > 0) |
| | | 1575 | | { |
| | 1 | 1576 | | if (!_writeBuffer.b.hasRemaining()) |
| | | 1577 | | { |
| | 1 | 1578 | | buf.b.position(_writePayloadLength); |
| | | 1579 | | } |
| | | 1580 | | } |
| | | 1581 | | |
| | 1 | 1582 | | if (status == SocketOperation.Write && !buf.b.hasRemaining() && !_writeBuffer.b.hasRemaining()) |
| | | 1583 | | { |
| | | 1584 | | // |
| | | 1585 | | // Our buffers are empty but the delegate needs another call to write(). |
| | | 1586 | | // |
| | 0 | 1587 | | _writeState = WriteStateFlush; |
| | 0 | 1588 | | return false; |
| | | 1589 | | } |
| | 1 | 1590 | | else if (!buf.b.hasRemaining()) |
| | | 1591 | | { |
| | 1 | 1592 | | _writeState = WriteStateHeader; |
| | 1 | 1593 | | if (_state == StatePingPending || |
| | 1 | 1594 | | _state == StatePongPending || |
| | 1 | 1595 | | (_state == StateClosingRequestPending && !_closingInitiator) || |
| | 1 | 1596 | | (_state == StateClosingResponsePending && _closingInitiator)) |
| | | 1597 | | { |
| | 1 | 1598 | | return true; |
| | | 1599 | | } |
| | | 1600 | | } |
| | 1 | 1601 | | else if (_state == StateOpened) |
| | | 1602 | | { |
| | 1 | 1603 | | return status == SocketOperation.None; |
| | | 1604 | | } |
| | | 1605 | | |
| | 1 | 1606 | | return false; |
| | | 1607 | | } |
| | | 1608 | | |
| | | 1609 | | private bool readBuffered(int sz) |
| | | 1610 | | { |
| | 1 | 1611 | | if (_readBufferPos == _readBuffer.b.position()) |
| | | 1612 | | { |
| | 1 | 1613 | | _readBuffer.resize(_readBufferSize, true); |
| | 1 | 1614 | | _readBufferPos = 0; |
| | 1 | 1615 | | _readBuffer.b.position(0); |
| | | 1616 | | } |
| | | 1617 | | else |
| | | 1618 | | { |
| | 1 | 1619 | | int available = _readBuffer.b.position() - _readBufferPos; |
| | 1 | 1620 | | if (available < sz) |
| | | 1621 | | { |
| | 1 | 1622 | | if (_readBufferPos > 0) |
| | | 1623 | | { |
| | 1 | 1624 | | _readBuffer.b.limit(_readBuffer.b.position()); |
| | 1 | 1625 | | _readBuffer.b.position(_readBufferPos); |
| | 1 | 1626 | | _readBuffer.b.compact(); |
| | | 1627 | | Debug.Assert(_readBuffer.b.position() == available); |
| | | 1628 | | } |
| | 1 | 1629 | | _readBuffer.resize(Math.Max(_readBufferSize, sz), true); |
| | 1 | 1630 | | _readBufferPos = 0; |
| | 1 | 1631 | | _readBuffer.b.position(available); |
| | | 1632 | | } |
| | | 1633 | | } |
| | | 1634 | | |
| | 1 | 1635 | | _readStart = _readBuffer.b.position(); |
| | 1 | 1636 | | if (_readBufferPos + sz > _readBuffer.b.position()) |
| | | 1637 | | { |
| | 1 | 1638 | | return false; // Not enough read. |
| | | 1639 | | } |
| | | 1640 | | Debug.Assert(_readBuffer.b.position() > _readBufferPos); |
| | 1 | 1641 | | return true; |
| | | 1642 | | } |
| | | 1643 | | |
| | | 1644 | | private void prepareWriteHeader(byte opCode, int payloadLength) |
| | | 1645 | | { |
| | | 1646 | | // |
| | | 1647 | | // We need to prepare the frame header. |
| | | 1648 | | // |
| | 1 | 1649 | | _writeBuffer.resize(_writeBufferSize, false); |
| | 1 | 1650 | | _writeBuffer.b.limit(_writeBufferSize); |
| | 1 | 1651 | | _writeBuffer.b.position(0); |
| | | 1652 | | |
| | | 1653 | | // |
| | | 1654 | | // Set the opcode - this is the one and only data frame. |
| | | 1655 | | // |
| | 1 | 1656 | | _writeBuffer.b.put((byte)(opCode | FLAG_FINAL)); |
| | | 1657 | | |
| | | 1658 | | // |
| | | 1659 | | // Set the payload length. |
| | | 1660 | | // |
| | 1 | 1661 | | if (payloadLength <= 125) |
| | | 1662 | | { |
| | 1 | 1663 | | _writeBuffer.b.put((byte)payloadLength); |
| | | 1664 | | } |
| | 1 | 1665 | | else if (payloadLength > 125 && payloadLength <= 65535) |
| | | 1666 | | { |
| | | 1667 | | // |
| | | 1668 | | // Use an extra 16 bits to encode the payload length. |
| | | 1669 | | // |
| | 1 | 1670 | | _writeBuffer.b.put(126); |
| | 1 | 1671 | | _writeBuffer.b.putShort((short)payloadLength); |
| | | 1672 | | } |
| | 1 | 1673 | | else if (payloadLength > 65535) |
| | | 1674 | | { |
| | | 1675 | | // |
| | | 1676 | | // Use an extra 64 bits to encode the payload length. |
| | | 1677 | | // |
| | 1 | 1678 | | _writeBuffer.b.put(127); |
| | 1 | 1679 | | _writeBuffer.b.putLong(payloadLength); |
| | | 1680 | | } |
| | | 1681 | | |
| | 1 | 1682 | | if (!_incoming) |
| | | 1683 | | { |
| | | 1684 | | // |
| | | 1685 | | // Add a random 32-bit mask to every outgoing frame, copy the payload data, |
| | | 1686 | | // and apply the mask. |
| | | 1687 | | // |
| | 1 | 1688 | | _writeBuffer.b.put(1, (byte)(_writeBuffer.b.get(1) | FLAG_MASKED)); |
| | 1 | 1689 | | RandomNumberGenerator.Fill(_writeMask); |
| | 1 | 1690 | | _writeBuffer.b.put(_writeMask); |
| | | 1691 | | } |
| | 1 | 1692 | | } |
| | | 1693 | | |
| | | 1694 | | private ProtocolInstance _instance; |
| | | 1695 | | private Transceiver _delegate; |
| | | 1696 | | private readonly string _host; |
| | | 1697 | | private string _resource; |
| | | 1698 | | private readonly bool _incoming; |
| | 1 | 1699 | | private readonly HashSet<string> _allowedOrigins = new HashSet<string>(); |
| | | 1700 | | |
| | | 1701 | | private const int StateInitializeDelegate = 0; |
| | | 1702 | | private const int StateConnected = 1; |
| | | 1703 | | private const int StateUpgradeRequestPending = 2; |
| | | 1704 | | private const int StateUpgradeResponsePending = 3; |
| | | 1705 | | private const int StateOpened = 4; |
| | | 1706 | | private const int StatePingPending = 5; |
| | | 1707 | | private const int StatePongPending = 6; |
| | | 1708 | | private const int StateClosingRequestPending = 7; |
| | | 1709 | | private const int StateClosingResponsePending = 8; |
| | | 1710 | | private const int StateClosed = 9; |
| | | 1711 | | |
| | | 1712 | | private int _state; |
| | | 1713 | | private int _nextState; |
| | | 1714 | | |
| | | 1715 | | private HttpParser _parser; |
| | | 1716 | | private string _key; |
| | | 1717 | | |
| | | 1718 | | private const int ReadStateOpcode = 0; |
| | | 1719 | | private const int ReadStateHeader = 1; |
| | | 1720 | | private const int ReadStateControlFrame = 2; |
| | | 1721 | | private const int ReadStatePayload = 3; |
| | | 1722 | | |
| | | 1723 | | private int _readState; |
| | | 1724 | | private Buffer _readBuffer; |
| | | 1725 | | private int _readBufferPos; |
| | | 1726 | | private int _readBufferSize; |
| | | 1727 | | |
| | | 1728 | | private bool _readLastFrame; |
| | | 1729 | | private int _readOpCode; |
| | | 1730 | | private int _readHeaderLength; |
| | | 1731 | | private int _readPayloadLength; |
| | | 1732 | | private int _readStart; |
| | | 1733 | | private int _readFrameStart; |
| | | 1734 | | private byte[] _readMask; |
| | | 1735 | | |
| | | 1736 | | private const int WriteStateHeader = 0; |
| | | 1737 | | private const int WriteStatePayload = 1; |
| | | 1738 | | private const int WriteStateControlFrame = 2; |
| | | 1739 | | private const int WriteStateFlush = 3; |
| | | 1740 | | |
| | | 1741 | | private int _writeState; |
| | | 1742 | | private Buffer _writeBuffer; |
| | | 1743 | | private int _writeBufferSize; |
| | | 1744 | | private byte[] _writeMask; |
| | | 1745 | | private int _writePayloadLength; |
| | | 1746 | | |
| | | 1747 | | private bool _closingInitiator; |
| | | 1748 | | private int _closingReason; |
| | | 1749 | | |
| | | 1750 | | private bool _readPending; |
| | | 1751 | | private bool _finishRead; |
| | | 1752 | | private bool _writePending; |
| | | 1753 | | |
| | | 1754 | | private byte[] _pingPayload; |
| | | 1755 | | |
| | | 1756 | | // |
| | | 1757 | | // WebSocket opcodes |
| | | 1758 | | // |
| | | 1759 | | private const int OP_CONT = 0x0; // Continuation frame |
| | | 1760 | | private const int OP_TEXT = 0x1; // Text frame |
| | | 1761 | | private const int OP_DATA = 0x2; // Data frame |
| | | 1762 | | // private const int OP_RES_0x3 = 0x3; // Reserved |
| | | 1763 | | // private const int OP_RES_0x4 = 0x4; // Reserved |
| | | 1764 | | // private const int OP_RES_0x5 = 0x5; // Reserved |
| | | 1765 | | // private const int OP_RES_0x6 = 0x6; // Reserved |
| | | 1766 | | // private const int OP_RES_0x7 = 0x7; // Reserved |
| | | 1767 | | private const int OP_CLOSE = 0x8; // Connection close |
| | | 1768 | | private const int OP_PING = 0x9; // Ping |
| | | 1769 | | private const int OP_PONG = 0xA; // Pong |
| | | 1770 | | // private const int OP_RES_0xB = 0xB; // Reserved |
| | | 1771 | | // private const int OP_RES_0xC = 0xC; // Reserved |
| | | 1772 | | // private const int OP_RES_0xD = 0xD; // Reserved |
| | | 1773 | | // private const int OP_RES_0xE = 0xE; // Reserved |
| | | 1774 | | // private const int OP_RES_0xF = 0xF; // Reserved |
| | | 1775 | | private const int FLAG_FINAL = 0x80; // Last frame |
| | | 1776 | | private const int FLAG_MASKED = 0x80; // Payload is masked |
| | | 1777 | | |
| | | 1778 | | private const int CLOSURE_NORMAL = 1000; |
| | | 1779 | | private const int CLOSURE_SHUTDOWN = 1001; |
| | | 1780 | | private const int CLOSURE_PROTOCOL_ERROR = 1002; |
| | | 1781 | | |
| | | 1782 | | private const string _iceProtocol = "ice.zeroc.com"; |
| | | 1783 | | private const string _wsUUID = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"; |
| | | 1784 | | |
| | 1 | 1785 | | private static readonly UTF8Encoding _utf8 = new UTF8Encoding(false, true); |
| | | 1786 | | |
| | | 1787 | | // Throws ArgumentException for any input that is not a serialized origin per RFC 6454. |
| | | 1788 | | internal static string canonicalizeOrigin(string origin) |
| | | 1789 | | { |
| | | 1790 | | Uri uri; |
| | | 1791 | | try |
| | | 1792 | | { |
| | 1 | 1793 | | uri = new Uri(origin, UriKind.Absolute); |
| | 1 | 1794 | | } |
| | 1 | 1795 | | catch (UriFormatException ex) |
| | | 1796 | | { |
| | 1 | 1797 | | throw new ArgumentException($"malformed origin '{origin}'", nameof(origin), ex); |
| | | 1798 | | } |
| | | 1799 | | // RFC 6454: a serialized origin is exactly "scheme://host[:port]". Reject path, query, fragment, and userinfo. |
| | | 1800 | | // System.Uri normalizes "https://x" to AbsolutePath="/", so we accept either empty or "/". |
| | 1 | 1801 | | if ((uri.AbsolutePath.Length > 0 && uri.AbsolutePath != "/") |
| | 1 | 1802 | | || uri.Query.Length > 0 |
| | 1 | 1803 | | || uri.Fragment.Length > 0 |
| | 1 | 1804 | | || uri.UserInfo.Length > 0) |
| | | 1805 | | { |
| | 1 | 1806 | | throw new ArgumentException($"malformed origin '{origin}'", nameof(origin)); |
| | | 1807 | | } |
| | 1 | 1808 | | return uri.IsDefaultPort ? $"{uri.Scheme}://{uri.Host}" : $"{uri.Scheme}://{uri.Host}:{uri.Port}"; |
| | | 1809 | | } |
| | | 1810 | | } |