net/tools/quic/quic_server_test.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "net/tools/quic/quic_server.h"
   6
   7 #include "net/quic/crypto/quic_random.h"
   8 #include "net/quic/quic_utils.h"
   9 #include "net/tools/quic/test_tools/mock_quic_dispatcher.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 using ::testing::_;
  13
  14 namespace net {
  15 namespace tools {
  16 namespace test {
  17
  18 namespace {
  19
  20 class QuicServerDispatchPacketTest : public ::testing::Test {
  21  public:
  22   QuicServerDispatchPacketTest()
  23       : crypto_config_("blah", QuicRandom::GetInstance()),
  24         dispatcher_(config_, crypto_config_, 1234, &eps_) {}
  25
  26
  27   void MaybeDispatchPacket(const QuicEncryptedPacket& packet) {
  28     IPEndPoint client_addr, server_addr;
  29     QuicServer::MaybeDispatchPacket(&dispatcher_, packet,
  30                                     client_addr, server_addr);
  31   }
  32
  33  protected:
  34   QuicConfig config_;
  35   QuicCryptoServerConfig crypto_config_;
  36   EpollServer eps_;
  37   MockQuicDispatcher dispatcher_;
  38 };
  39
  40 TEST_F(QuicServerDispatchPacketTest, DoNotDispatchPacketWithoutGUID) {
  41   // Packet too short to be considered valid.
  42   unsigned char invalid_packet[] = { 0x00 };
  43   QuicEncryptedPacket encrypted_invalid_packet(
  44       QuicUtils::AsChars(invalid_packet), arraysize(invalid_packet), false);
  45
  46   // We expect the invalid packet to be dropped, and ProcessPacket should never
  47   // be called.
  48   EXPECT_CALL(dispatcher_, ProcessPacket(_, _, _, _)).Times(0);
  49   MaybeDispatchPacket(encrypted_invalid_packet);
  50 }
  51
  52 TEST_F(QuicServerDispatchPacketTest, DispatchValidPacket) {
  53   unsigned char valid_packet[] = {
  54     // public flags (8 byte guid)
  55     0x3C,
  56     // guid
  57     0x10, 0x32, 0x54, 0x76,
  58     0x98, 0xBA, 0xDC, 0xFE,
  59     // packet sequence number
  60     0xBC, 0x9A, 0x78, 0x56,
  61     0x34, 0x12,
  62     // private flags
  63     0x00 };
  64   QuicEncryptedPacket encrypted_valid_packet(QuicUtils::AsChars(valid_packet),
  65                                              arraysize(valid_packet), false);
  66
  67   EXPECT_CALL(dispatcher_, ProcessPacket(_, _, _, _)).Times(1);
  68   MaybeDispatchPacket(encrypted_valid_packet);
  69 }
  70
  71 }  // namespace
  72 }  // namespace test
  73 }  // namespace tools
  74 }  // namespace net