| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "net/websockets/websocket_frame.h" |
| |
| #include <algorithm> |
| #include <vector> |
| |
| #include "base/macros.h" |
| #include "base/test/perf_time_logger.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace net { |
| |
| namespace { |
| |
| const int kIterations = 100000; |
| const int kLongPayloadSize = 1 << 16; |
| const char kMaskingKey[] = "\xFE\xED\xBE\xEF"; |
| |
| static_assert(arraysize(kMaskingKey) == |
| WebSocketFrameHeader::kMaskingKeyLength + 1, |
| "incorrect masking key size"); |
| |
| class WebSocketFrameTestMaskBenchmark : public ::testing::Test { |
| protected: |
| void Benchmark(const char* const name, |
| const char* const payload, |
| size_t size) { |
| std::vector<char> scratch(payload, payload + size); |
| WebSocketMaskingKey masking_key; |
| std::copy(kMaskingKey, |
| kMaskingKey + WebSocketFrameHeader::kMaskingKeyLength, |
| masking_key.key); |
| base::PerfTimeLogger timer(name); |
| for (int x = 0; x < kIterations; ++x) { |
| MaskWebSocketFramePayload( |
| masking_key, x % size, &scratch.front(), scratch.size()); |
| } |
| timer.Done(); |
| } |
| }; |
| |
| TEST_F(WebSocketFrameTestMaskBenchmark, BenchmarkMaskShortPayload) { |
| static const char kShortPayload[] = "Short Payload"; |
| Benchmark( |
| "Frame_mask_short_payload", kShortPayload, arraysize(kShortPayload)); |
| } |
| |
| TEST_F(WebSocketFrameTestMaskBenchmark, BenchmarkMaskLongPayload) { |
| std::vector<char> payload(kLongPayloadSize, 'a'); |
| Benchmark("Frame_mask_long_payload", &payload.front(), payload.size()); |
| } |
| |
| // A 31-byte payload is guaranteed to do 7 byte mask operations and 3 vector |
| // mask operations with an 8-byte vector. With a 16-byte vector it will fall |
| // back to the byte-only code path and do 31 byte mask operations. |
| TEST_F(WebSocketFrameTestMaskBenchmark, Benchmark31BytePayload) { |
| std::vector<char> payload(31, 'a'); |
| Benchmark("Frame_mask_31_payload", &payload.front(), payload.size()); |
| } |
| |
| } // namespace |
| |
| } // namespace net |