| /* |
| * 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. |
| */ |
| |
| #ifndef SRC_TRACING_TEST_FAKE_PACKET_H_ |
| #define SRC_TRACING_TEST_FAKE_PACKET_H_ |
| |
| #include <stdint.h> |
| |
| #include <array> |
| #include <initializer_list> |
| #include <string> |
| #include <vector> |
| |
| #include "perfetto/tracing/core/basic_types.h" |
| |
| namespace perfetto { |
| |
| class TraceBuffez; |
| |
| class FakePacketFragment { |
| public: |
| // Generates a packet of given |size| with pseudo random payload. The |size| |
| // argument includes the size of the varint header. |
| FakePacketFragment(size_t size, char prefix); |
| FakePacketFragment(const void* payload, size_t payload_size); |
| void CopyInto(std::vector<uint8_t>* data) const; |
| size_t GetSizeHeader() const; |
| bool operator==(const FakePacketFragment& other) const; |
| const std::string& payload() const { return payload_; } |
| |
| private: |
| std::array<uint8_t, 2> header_{}; |
| size_t header_size_ = 0; |
| std::string payload_; |
| }; |
| |
| std::ostream& operator<<(std::ostream&, const FakePacketFragment&); |
| |
| class FakeChunk { |
| public: |
| FakeChunk(TraceBuffez* t, ProducerID p, WriterID w, ChunkID c); |
| |
| // Appends a packet of exactly |size| bytes (including the varint header |
| // that states the size of the packet itself. The payload of the packet is |
| // NOT a valid proto and is just filled with pseudo-random bytes generated |
| // from |seed|. |
| FakeChunk& AddPacket(size_t size, char seed, uint8_t packet_flag = 0); |
| |
| // Appends a packet with the given raw content (including varint header). |
| FakeChunk& AddPacket(std::initializer_list<uint8_t>); |
| |
| FakeChunk& ClearBytes(size_t offset, size_t len); |
| |
| FakeChunk& SetUID(uid_t); |
| |
| // Returns the full size of the chunk including the ChunkRecord header. |
| size_t CopyIntoTraceBuffer(); |
| |
| private: |
| TraceBuffez* trace_buffer_; |
| ProducerID producer_id; |
| WriterID writer_id; |
| ChunkID chunk_id; |
| uint8_t flags = 0; |
| uint16_t num_packets = 0; |
| uid_t uid = -1; |
| std::vector<uint8_t> data; |
| }; |
| |
| } // namespace perfetto |
| |
| #endif // SRC_TRACING_TEST_FAKE_PACKET_H_ |