net/tools/quic/quic_client_test.cc

   1 // Copyright (c) 2014 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_client.h"
   6
   7 #include <dirent.h>
   8 #include <stdio.h>
   9
  10 #include "base/basictypes.h"
  11 #include "base/strings/string_util.h"
  12 #include "net/quic/test_tools/quic_test_utils.h"
  13 #include "net/tools/epoll_server/epoll_server.h"
  14 #include "testing/gtest/include/gtest/gtest.h"
  15
  16 using net::EpollServer;
  17
  18 namespace net {
  19 namespace tools {
  20 namespace test {
  21 namespace {
  22
  23 int NumOpenFDs() {
  24   int number_of_open_fds = 0;
  25   char buf[256];
  26   struct dirent* dp;
  27
  28   base::snprintf(buf, arraysize(buf), "/proc/%i/fd/", getpid());
  29   DIR* dir = opendir(buf);
  30   while ((dp = readdir(dir)) != NULL)
  31     number_of_open_fds++;
  32   closedir(dir);
  33
  34   return number_of_open_fds;
  35 }
  36
  37 // Creates a new QuicClient and Initializes it. Caller is responsible for
  38 // deletion.
  39 QuicClient* CreateAndInitializeQuicClient(EpollServer* eps, uint16 port) {
  40   IPEndPoint server_address(IPEndPoint(net::test::Loopback4(), port));
  41   QuicServerId server_id("hostname", server_address.port(), false,
  42                          PRIVACY_MODE_DISABLED);
  43   QuicVersionVector versions = QuicSupportedVersions();
  44   QuicClient* client = new QuicClient(server_address, server_id, versions, eps);
  45   EXPECT_TRUE(client->Initialize());
  46   return client;
  47 }
  48
  49 TEST(QuicClientTest, DoNotLeakFDs) {
  50   // Make sure that the QuicClient doesn't leak FDs. Doing so could cause port
  51   // exhaustion in long running processes which repeatedly create clients.
  52
  53   // Record initial number of FDs, after creation of EpollServer.
  54   EpollServer eps;
  55   int number_of_open_fds = NumOpenFDs();
  56
  57   // Create a number of clients, initialize them, and verify this has resulted
  58   // in additional FDs being opened.
  59   const int kNumClients = 5;
  60   for (int i = 0; i < kNumClients; ++i) {
  61     scoped_ptr<QuicClient> client(
  62         CreateAndInitializeQuicClient(&eps, net::test::kTestPort + i));
  63
  64     // Initializing the client will create a new FD.
  65     EXPECT_LT(number_of_open_fds, NumOpenFDs());
  66   }
  67
  68   // The FDs created by the QuicClients should now be closed.
  69   EXPECT_EQ(number_of_open_fds, NumOpenFDs());
  70 }
  71
  72 }  // namespace
  73 }  // namespace test
  74 }  // namespace tools
  75 }  // namespace net