blob: fd5ca05e91dae2f37bbf9fe45ef4438556311277 [file] [log] [blame]
/*
* Copyright (C) 2018 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "src/tracing/test/fake_packet.h"
#include <ostream>
#include "perfetto/base/logging.h"
#include "perfetto/protozero/proto_utils.h"
#include "perfetto/tracing/core/shared_memory_abi.h"
#include "src/tracing/core/trace_buffer.h"
using protozero::proto_utils::ParseVarInt;
using protozero::proto_utils::WriteVarInt;
namespace perfetto {
FakePacketFragment::FakePacketFragment(size_t size, char prefix) {
// |size| has to be at least == 2, because one byte will be taken just by the
// varint header.
PERFETTO_CHECK(size > 1);
// Finding the |payload_size| from |size| is quite tricky:
// A packet with 127 bytes of payload requires:
// 1B header (to represent the number 127 in varint) + 127 = 128 bytes.
// A packet with 128 bytes payload requires:
// 2B header (to represent the number 128 in varint) + 128 = 130 bytes.
// So the only way to generate a packet of 129 bytes in total we need to
// generate a redundant varint header of 2 bytes + 127 bytes of payload.
if (size <= 128) {
header_[0] = static_cast<uint8_t>(size - 1);
header_size_ = 1;
payload_.resize(size - 1);
} else if (size == 129) {
header_[0] = 0x80 | 127;
header_size_ = 2;
payload_.resize(127);
} else {
WriteVarInt(size - 2, &header_[0]);
header_size_ = 2;
payload_.resize(size - 2);
}
// Fills the payload as follow: X00-X01-X02 (where X == |prefix|);
for (size_t i = 0; i < payload_.size(); i++) {
switch (i % 4) {
case 0:
payload_[i] = prefix;
break;
case 1:
payload_[i] = '0' + ((i / 4 / 10) % 10);
break;
case 2:
payload_[i] = '0' + ((i / 4) % 10);
break;
case 3:
payload_[i] = '-';
break;
}
}
}
FakePacketFragment::FakePacketFragment(const void* payload,
size_t payload_size) {
PERFETTO_CHECK(payload_size <= 4096 - 2);
payload_.assign(reinterpret_cast<const char*>(payload), payload_size);
uint8_t* end = WriteVarInt(payload_.size(), &header_[0]);
header_size_ = end - &header_[0];
}
void FakePacketFragment::CopyInto(std::vector<uint8_t>* data) const {
data->insert(data->end(), &header_[0], &header_[0] + header_size_);
data->insert(data->end(), payload_.begin(), payload_.end());
}
size_t FakePacketFragment::GetSizeHeader() const {
uint64_t size = 0;
ParseVarInt(&header_[0], &header_[0] + header_size_, &size);
return static_cast<size_t>(size);
}
bool FakePacketFragment::operator==(const FakePacketFragment& o) const {
if (payload_ != o.payload_)
return false;
PERFETTO_CHECK(GetSizeHeader() == o.GetSizeHeader());
return true;
}
std::ostream& operator<<(std::ostream& os, const FakePacketFragment& packet) {
return os << "{len:" << packet.payload().size() << ", payload:\""
<< packet.payload() << "\"}";
}
FakeChunk::FakeChunk(TraceBuffez* t, ProducerID p, WriterID w, ChunkID c)
: trace_buffer_{t}, producer_id{p}, writer_id{w}, chunk_id{c} {}
FakeChunk& FakeChunk::AddPacket(size_t size, char seed, uint8_t packet_flag) {
PERFETTO_DCHECK(size <= 4096);
PERFETTO_CHECK(
!(packet_flag &
SharedMemoryABI::ChunkHeader::kFirstPacketContinuesFromPrevChunk) ||
num_packets == 0);
PERFETTO_CHECK(
!(flags & SharedMemoryABI::ChunkHeader::kLastPacketContinuesOnNextChunk));
flags |= packet_flag;
FakePacketFragment(size, seed).CopyInto(&data);
num_packets++;
return *this;
}
FakeChunk& FakeChunk::AddPacket(std::initializer_list<uint8_t> raw) {
data.insert(data.end(), raw.begin(), raw.end());
num_packets++;
return *this;
}
FakeChunk& FakeChunk::ClearBytes(size_t offset, size_t len) {
PERFETTO_DCHECK(offset + len <= data.size());
memset(data.data() + offset, 0, len);
return *this;
}
FakeChunk& FakeChunk::SetUID(uid_t u) {
uid = u;
return *this;
}
size_t FakeChunk::CopyIntoTraceBuffer() {
trace_buffer_->CopyChunkUntrusted(producer_id, uid, writer_id, chunk_id,
num_packets, flags, data.data(),
data.size());
return data.size() + TraceBuffez::InlineChunkHeaderSize;
}
} // namespace perfetto