| OLD | NEW |
| (Empty) |
| 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 | |
| OLD | NEW |