| // Copyright 2016 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef GARNET_BIN_TRACE_MANAGER_TRACEE_H_ |
| #define GARNET_BIN_TRACE_MANAGER_TRACEE_H_ |
| |
| #include <lib/async/cpp/wait.h> |
| #include <lib/fit/function.h> |
| #include <lib/zx/fifo.h> |
| #include <lib/zx/socket.h> |
| #include <lib/zx/vmo.h> |
| #include <trace-reader/reader_internal.h> |
| |
| #include <iosfwd> |
| |
| #include "garnet/bin/trace_manager/trace_provider_bundle.h" |
| #include "lib/fidl/cpp/string.h" |
| #include "lib/fidl/cpp/vector.h" |
| #include "lib/fxl/macros.h" |
| #include "lib/fxl/memory/weak_ptr.h" |
| |
| namespace tracing { |
| |
| class Tracee { |
| public: |
| enum class State { |
| // All systems go, provider hasn't been started, yet. |
| kReady, |
| // The provider was asked to start. |
| kStartPending, |
| // The provider is started and tracing. |
| kStarted, |
| // The provider is being stopped right now. |
| kStopping, |
| // The provider is stopped. |
| kStopped |
| }; |
| |
| enum class TransferStatus { |
| // The transfer is complete. |
| kComplete, |
| // The transfer is incomplete and subsequent transfers should not be |
| // executed as the underlying stream (or other part of the tracee, |
| // e.g., vmo) has been corrupted. |
| kCorrupted, |
| // The receiver of the transfer went away. |
| kReceiverDead, |
| }; |
| |
| // The size of the initialization record. |
| static constexpr size_t kInitRecordSizeBytes = 16; |
| |
| explicit Tracee(TraceProviderBundle* bundle); |
| ~Tracee(); |
| |
| bool operator==(TraceProviderBundle* bundle) const; |
| bool Start(fidl::VectorPtr<fidl::StringPtr> categories, size_t buffer_size, |
| fuchsia::tracelink::BufferingMode buffering_mode, |
| fit::closure started_callback, fit::closure stopped_callback); |
| void Stop(); |
| |
| // Called once at the end of the trace to transfer all collected records |
| // to |socket|. |
| TransferStatus TransferRecords(const zx::socket& socket) const; |
| |
| const TraceProviderBundle* bundle() const { return bundle_; } |
| State state() const { return state_; } |
| |
| private: |
| // The size of the fifo, in packets. |
| // TODO(dje): The value will need playing with. |
| static constexpr size_t kFifoSizeInPackets = 4u; |
| |
| // Given |wrapped_count|, return the corresponding buffer number. |
| static int get_buffer_number(uint32_t wrapped_count) { |
| return wrapped_count & 1; |
| } |
| |
| void TransitionToState(State new_state); |
| void OnHandleReady(async_dispatcher_t* dispatcher, async::WaitBase* wait, |
| zx_status_t status, const zx_packet_signal_t* signal); |
| void OnFifoReadable(async_dispatcher_t* dispatcher, async::WaitBase* wait); |
| void OnHandleError(zx_status_t status); |
| |
| bool VerifyBufferHeader( |
| const trace::internal::BufferHeaderReader* header) const; |
| |
| TransferStatus WriteChunk(const zx::socket& socket, size_t vmo_offset, |
| size_t size, const char* name) const; |
| |
| TransferStatus WriteProviderInfoRecord(const zx::socket& socket) const; |
| TransferStatus WriteProviderBufferOverflowEvent( |
| const zx::socket& socket) const; |
| |
| const TraceProviderBundle* const bundle_; |
| State state_ = State::kReady; |
| fuchsia::tracelink::BufferingMode buffering_mode_; |
| zx::vmo buffer_vmo_; |
| size_t buffer_vmo_size_ = 0u; |
| zx::fifo fifo_; |
| fit::closure started_callback_; |
| fit::closure stopped_callback_; |
| async_dispatcher_t* dispatcher_ = nullptr; |
| async::WaitMethod<Tracee, &Tracee::OnHandleReady> wait_; |
| |
| fxl::WeakPtrFactory<Tracee> weak_ptr_factory_; |
| FXL_DISALLOW_COPY_AND_ASSIGN(Tracee); |
| }; |
| |
| std::ostream& operator<<(std::ostream& out, Tracee::State state); |
| |
| } // namespace tracing |
| |
| #endif // GARNET_BIN_TRACE_MANAGER_TRACEE_H_ |