net/websockets/websocket_frame_perftest.cc

   1 // Copyright 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/websockets/websocket_frame.h"
   6
   7 #include <algorithm>
   8 #include <vector>
   9
  10 #include "base/macros.h"
  11 #include "base/test/perf_time_logger.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 namespace net {
  15
  16 namespace {
  17
  18 const int kIterations = 100000;
  19 const int kLongPayloadSize = 1 << 16;
  20 const char kMaskingKey[] = "\xFE\xED\xBE\xEF";
  21
  22 static_assert(arraysize(kMaskingKey) ==
  23                   WebSocketFrameHeader::kMaskingKeyLength + 1,
  24               "incorrect masking key size");
  25
  26 class WebSocketFrameTestMaskBenchmark : public ::testing::Test {
  27  protected:
  28   void Benchmark(const char* const name,
  29                  const char* const payload,
  30                  size_t size) {
  31     std::vector<char> scratch(payload, payload + size);
  32     WebSocketMaskingKey masking_key;
  33     std::copy(kMaskingKey,
  34               kMaskingKey + WebSocketFrameHeader::kMaskingKeyLength,
  35               masking_key.key);
  36     base::PerfTimeLogger timer(name);
  37     for (int x = 0; x < kIterations; ++x) {
  38       MaskWebSocketFramePayload(
  39           masking_key, x % size, &scratch.front(), scratch.size());
  40     }
  41     timer.Done();
  42   }
  43 };
  44
  45 TEST_F(WebSocketFrameTestMaskBenchmark, BenchmarkMaskShortPayload) {
  46   static const char kShortPayload[] = "Short Payload";
  47   Benchmark(
  48       "Frame_mask_short_payload", kShortPayload, arraysize(kShortPayload));
  49 }
  50
  51 TEST_F(WebSocketFrameTestMaskBenchmark, BenchmarkMaskLongPayload) {
  52   std::vector<char> payload(kLongPayloadSize, 'a');
  53   Benchmark("Frame_mask_long_payload", &payload.front(), payload.size());
  54 }
  55
  56 }  // namespace
  57
  58 }  // namespace net