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
|
// Copyright 2025 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/40285824): Remove this and convert code to safer constructs.
#pragma allow_unsafe_buffers
#endif
#include "chrome/browser/actor/aggregated_journal_in_memory_serializer.h"
#include "third_party/perfetto/include/perfetto/ext/tracing/core/trace_packet.h"
namespace actor {
AggregatedJournalInMemorySerializer::AggregatedJournalInMemorySerializer(
AggregatedJournal& journal)
: AggregatedJournalSerializer(journal) {}
AggregatedJournalInMemorySerializer::~AggregatedJournalInMemorySerializer() =
default;
void AggregatedJournalInMemorySerializer::Init() {
InitImpl();
}
void AggregatedJournalInMemorySerializer::WriteTracePacket(
std::vector<uint8_t> message) {
buffer_list_.push_back(std::move(message));
}
size_t AggregatedJournalInMemorySerializer::ApproximateSnapshotSize() {
size_t total_size = 0;
for (const auto& buffer : buffer_list_) {
total_size += buffer.size();
total_size += perfetto::TracePacket::kMaxPreambleBytes;
}
return total_size;
}
std::vector<uint8_t> AggregatedJournalInMemorySerializer::Snapshot(
size_t max_bytes) {
size_t total_size = 0;
std::vector<uint8_t> result_buffer;
result_buffer.reserve(std::min(ApproximateSnapshotSize(), max_bytes));
for (const auto& buffer : buffer_list_) {
if (total_size + buffer.size() + perfetto::TracePacket::kMaxPreambleBytes >
max_bytes) {
break;
}
perfetto::TracePacket packet;
packet.AddSlice(buffer.data(), buffer.size());
auto [preamble, preamble_size] = packet.GetProtoPreamble();
auto preamble_span =
base::span(reinterpret_cast<const uint8_t*>(preamble), preamble_size);
result_buffer.insert(result_buffer.end(), preamble_span.begin(),
preamble_span.end());
for (const perfetto::Slice& slice : packet.slices()) {
auto data_span =
base::span(static_cast<const uint8_t*>(slice.start), slice.size);
result_buffer.insert(result_buffer.end(), data_span.begin(),
data_span.end());
}
}
return result_buffer;
}
} // namespace actor
|
