| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/websockets/websocket_handshake_handler.h" | 5 #include "net/websockets/websocket_handshake_handler.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 #include <vector> | 8 #include <vector> |
| 9 | 9 |
| 10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
| (...skipping 448 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 459 "HTTP/1.1 101 Switching Protocols\r\n" | 459 "HTTP/1.1 101 Switching Protocols\r\n" |
| 460 "Upgrade: websocket\r\n" | 460 "Upgrade: websocket\r\n" |
| 461 "Connection: Upgrade\r\n" | 461 "Connection: Upgrade\r\n" |
| 462 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n" | 462 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n" |
| 463 "Sec-WebSocket-Protocol: sample\r\n" | 463 "Sec-WebSocket-Protocol: sample\r\n" |
| 464 "\r\n"; | 464 "\r\n"; |
| 465 | 465 |
| 466 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse()); | 466 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse()); |
| 467 } | 467 } |
| 468 | 468 |
| 469 TEST(WebSocketHandshakeHandlerTest, SpdyRequestResponse) { | |
| 470 WebSocketHandshakeRequestHandler request_handler; | |
| 471 | |
| 472 static const char kHandshakeRequestMessage[] = | |
| 473 "GET /demo HTTP/1.1\r\n" | |
| 474 "Host: example.com\r\n" | |
| 475 "Upgrade: websocket\r\n" | |
| 476 "Connection: Upgrade\r\n" | |
| 477 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n" | |
| 478 "Origin: http://example.com\r\n" | |
| 479 "Sec-WebSocket-Protocol: sample\r\n" | |
| 480 "Sec-WebSocket-Extensions: foo\r\n" | |
| 481 "Sec-WebSocket-Version: 13\r\n" | |
| 482 "\r\n"; | |
| 483 | |
| 484 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage, | |
| 485 strlen(kHandshakeRequestMessage))); | |
| 486 EXPECT_EQ(13, request_handler.protocol_version()); | |
| 487 | |
| 488 GURL url("ws://example.com/demo"); | |
| 489 std::string challenge; | |
| 490 SpdyHeaderBlock headers; | |
| 491 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url, &headers, &challenge)); | |
| 492 | |
| 493 EXPECT_EQ(url.path(), headers["path"]); | |
| 494 EXPECT_TRUE(headers.find("upgrade") == headers.end()); | |
| 495 EXPECT_TRUE(headers.find("Upgrade") == headers.end()); | |
| 496 EXPECT_TRUE(headers.find("connection") == headers.end()); | |
| 497 EXPECT_TRUE(headers.find("Connection") == headers.end()); | |
| 498 EXPECT_TRUE(headers.find("Sec-WebSocket-Key") == headers.end()); | |
| 499 EXPECT_TRUE(headers.find("sec-websocket-key") == headers.end()); | |
| 500 EXPECT_TRUE(headers.find("Sec-WebSocket-Version") == headers.end()); | |
| 501 EXPECT_TRUE(headers.find("sec-webSocket-version") == headers.end()); | |
| 502 EXPECT_EQ("example.com", headers["host"]); | |
| 503 EXPECT_EQ("http://example.com", headers["origin"]); | |
| 504 EXPECT_EQ("sample", headers["sec-websocket-protocol"]); | |
| 505 EXPECT_EQ("foo", headers["sec-websocket-extensions"]); | |
| 506 EXPECT_EQ("ws", headers["scheme"]); | |
| 507 EXPECT_EQ("WebSocket/13", headers["version"]); | |
| 508 | |
| 509 const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ=="; | |
| 510 | |
| 511 EXPECT_EQ(expected_challenge, challenge); | |
| 512 | |
| 513 headers.clear(); | |
| 514 | |
| 515 headers["status"] = "101 Switching Protocols"; | |
| 516 headers["sec-websocket-protocol"] = "sample"; | |
| 517 headers["sec-websocket-extensions"] = "foo"; | |
| 518 | |
| 519 WebSocketHandshakeResponseHandler response_handler; | |
| 520 response_handler.set_protocol_version(13); | |
| 521 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock(headers, challenge)); | |
| 522 EXPECT_TRUE(response_handler.HasResponse()); | |
| 523 | |
| 524 // Note that order of sec-websocket-* is sensitive with hash_map order. | |
| 525 static const char kHandshakeResponseExpectedMessage[] = | |
| 526 "HTTP/1.1 101 Switching Protocols\r\n" | |
| 527 "Upgrade: websocket\r\n" | |
| 528 "Connection: Upgrade\r\n" | |
| 529 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n" | |
| 530 "sec-websocket-extensions: foo\r\n" | |
| 531 "sec-websocket-protocol: sample\r\n" | |
| 532 "\r\n"; | |
| 533 | |
| 534 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse()); | |
| 535 } | |
| 536 | |
| 537 TEST(WebSocketHandshakeHandlerTest, SpdyRequestResponseWithCookies) { | |
| 538 WebSocketHandshakeRequestHandler request_handler; | |
| 539 | |
| 540 // Note that websocket won't use multiple headers in request now. | |
| 541 static const char kHandshakeRequestMessage[] = | |
| 542 "GET /demo HTTP/1.1\r\n" | |
| 543 "Host: example.com\r\n" | |
| 544 "Upgrade: websocket\r\n" | |
| 545 "Connection: Upgrade\r\n" | |
| 546 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n" | |
| 547 "Origin: http://example.com\r\n" | |
| 548 "Sec-WebSocket-Protocol: sample\r\n" | |
| 549 "Sec-WebSocket-Extensions: foo\r\n" | |
| 550 "Sec-WebSocket-Version: 13\r\n" | |
| 551 "Cookie: WK-websocket-test=1; WK-websocket-test-httponly=1\r\n" | |
| 552 "\r\n"; | |
| 553 | |
| 554 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage, | |
| 555 strlen(kHandshakeRequestMessage))); | |
| 556 EXPECT_EQ(13, request_handler.protocol_version()); | |
| 557 | |
| 558 GURL url("ws://example.com/demo"); | |
| 559 std::string challenge; | |
| 560 SpdyHeaderBlock headers; | |
| 561 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url, &headers, &challenge)); | |
| 562 | |
| 563 EXPECT_EQ(url.path(), headers["path"]); | |
| 564 EXPECT_TRUE(headers.find("upgrade") == headers.end()); | |
| 565 EXPECT_TRUE(headers.find("Upgrade") == headers.end()); | |
| 566 EXPECT_TRUE(headers.find("connection") == headers.end()); | |
| 567 EXPECT_TRUE(headers.find("Connection") == headers.end()); | |
| 568 EXPECT_TRUE(headers.find("Sec-WebSocket-Key") == headers.end()); | |
| 569 EXPECT_TRUE(headers.find("sec-websocket-key") == headers.end()); | |
| 570 EXPECT_TRUE(headers.find("Sec-WebSocket-Version") == headers.end()); | |
| 571 EXPECT_TRUE(headers.find("sec-webSocket-version") == headers.end()); | |
| 572 EXPECT_EQ("example.com", headers["host"]); | |
| 573 EXPECT_EQ("http://example.com", headers["origin"]); | |
| 574 EXPECT_EQ("sample", headers["sec-websocket-protocol"]); | |
| 575 EXPECT_EQ("foo", headers["sec-websocket-extensions"]); | |
| 576 EXPECT_EQ("ws", headers["scheme"]); | |
| 577 EXPECT_EQ("WebSocket/13", headers["version"]); | |
| 578 EXPECT_EQ("WK-websocket-test=1; WK-websocket-test-httponly=1", | |
| 579 headers["cookie"]); | |
| 580 | |
| 581 const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ=="; | |
| 582 | |
| 583 EXPECT_EQ(expected_challenge, challenge); | |
| 584 | |
| 585 headers.clear(); | |
| 586 | |
| 587 headers["status"] = "101 Switching Protocols"; | |
| 588 headers["sec-websocket-protocol"] = "sample"; | |
| 589 headers["sec-websocket-extensions"] = "foo"; | |
| 590 std::string cookie = "WK-websocket-test=1"; | |
| 591 cookie.append(1, '\0'); | |
| 592 cookie += "WK-websocket-test-httponly=1; HttpOnly"; | |
| 593 headers["set-cookie"] = cookie; | |
| 594 | |
| 595 | |
| 596 WebSocketHandshakeResponseHandler response_handler; | |
| 597 response_handler.set_protocol_version(13); | |
| 598 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock(headers, challenge)); | |
| 599 EXPECT_TRUE(response_handler.HasResponse()); | |
| 600 | |
| 601 // Note that order of sec-websocket-* is sensitive with hash_map order. | |
| 602 static const char kHandshakeResponseExpectedMessage[] = | |
| 603 "HTTP/1.1 101 Switching Protocols\r\n" | |
| 604 "Upgrade: websocket\r\n" | |
| 605 "Connection: Upgrade\r\n" | |
| 606 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n" | |
| 607 "sec-websocket-extensions: foo\r\n" | |
| 608 "sec-websocket-protocol: sample\r\n" | |
| 609 "set-cookie: WK-websocket-test=1\r\n" | |
| 610 "set-cookie: WK-websocket-test-httponly=1; HttpOnly\r\n" | |
| 611 "\r\n"; | |
| 612 | |
| 613 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse()); | |
| 614 } | |
| 615 | |
| 616 } // namespace net | 469 } // namespace net |
| OLD | NEW |