1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/40284755): Remove this and spanify to fix the errors.
#pragma allow_unsafe_buffers
#endif
#include "net/websockets/websocket_frame.h"
#include <stddef.h>
#include <algorithm>
#include <iterator>
#include <string>
#include <string_view>
#include <vector>
#include "base/time/time.h"
#include "base/timer/elapsed_timer.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/perf/perf_result_reporter.h"
namespace net {
namespace {
constexpr int kIterations = 100000;
constexpr int kLongPayloadSize = 1 << 16;
constexpr std::string_view kMaskingKey = "\xFE\xED\xBE\xEF";
static constexpr char kMetricPrefixWebSocketFrame[] = "WebSocketFrameMask.";
static constexpr char kMetricMaskTimeMs[] = "mask_time";
perf_test::PerfResultReporter SetUpWebSocketFrameMaskReporter(
const std::string& story) {
perf_test::PerfResultReporter reporter(kMetricPrefixWebSocketFrame, story);
reporter.RegisterImportantMetric(kMetricMaskTimeMs, "ms");
return reporter;
}
static_assert(kMaskingKey.size() == WebSocketFrameHeader::kMaskingKeyLength,
"incorrect masking key size");
class WebSocketFrameTestMaskBenchmark : public ::testing::Test {
protected:
void Benchmark(const char* const story,
const char* const payload,
size_t size) {
std::vector<char> scratch(payload, payload + size);
WebSocketMaskingKey masking_key;
base::as_writable_byte_span(masking_key.key)
.copy_from(base::as_byte_span(kMaskingKey));
auto reporter = SetUpWebSocketFrameMaskReporter(story);
base::ElapsedTimer timer;
for (int x = 0; x < kIterations; ++x) {
MaskWebSocketFramePayload(masking_key, x % size,
base::as_writable_byte_span(scratch));
}
reporter.AddResult(kMetricMaskTimeMs, timer.Elapsed().InMillisecondsF());
}
};
TEST_F(WebSocketFrameTestMaskBenchmark, BenchmarkMaskShortPayload) {
static constexpr char kShortPayload[] = "Short Payload";
Benchmark("short_payload", kShortPayload, std::size(kShortPayload));
}
TEST_F(WebSocketFrameTestMaskBenchmark, BenchmarkMaskLongPayload) {
std::vector<char> payload(kLongPayloadSize, 'a');
Benchmark("long_payload", payload.data(), 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("31_payload", payload.data(), payload.size());
}
} // namespace
} // namespace net
|
