zircon/system/ulib/trace-provider/provider_with_fdio.cc

// Copyright 2018 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.

#include <lib/trace-provider/fdio_connect.h>
#include <lib/trace-provider/provider.h>
#include <lib/zx/process.h>
#include <stdio.h>
#include <zircon/assert.h>
#include <zircon/status.h>

#include "export.h"

EXPORT trace_provider_t* trace_provider_create_with_name_fdio(async_dispatcher_t* dispatcher,
                                                              const char* name) {
  ZX_DEBUG_ASSERT(dispatcher);

  zx_handle_t to_service;
  auto status = trace_provider_connect_with_fdio(&to_service);
  if (status != ZX_OK) {
    fprintf(stderr, "TraceProvider: connection failed: status=%d(%s)\n", status,
            zx_status_get_string(status));
    return nullptr;
  }

  return trace_provider_create_with_name(to_service, dispatcher, name);
}

EXPORT trace_provider_t* trace_provider_create_with_fdio(async_dispatcher_t* dispatcher) {
  ZX_DEBUG_ASSERT(dispatcher);

  auto self = zx::process::self();
  char name[ZX_MAX_NAME_LEN];
  auto status = self->get_property(ZX_PROP_NAME, name, sizeof(name));
  if (status != ZX_OK) {
    fprintf(stderr, "TraceProvider: error getting process name: status=%d(%s)\n", status,
            zx_status_get_string(status));
    name[0] = '\0';
  }

  return trace_provider_create_with_name_fdio(dispatcher, name);
}

EXPORT trace_provider_t* trace_provider_create_synchronously_with_fdio(
    async_dispatcher_t* dispatcher, const char* name, bool* out_manager_is_tracing_already) {
  ZX_DEBUG_ASSERT(dispatcher);

  zx_handle_t to_service;
  auto status = trace_provider_connect_with_fdio(&to_service);
  if (status != ZX_OK) {
    fprintf(stderr, "TraceProvider: connection failed: status=%d(%s)\n", status,
            zx_status_get_string(status));
    return nullptr;
  }

  char self_name[ZX_MAX_NAME_LEN];
  if (name == nullptr) {
    auto self = zx::process::self();
    auto status = self->get_property(ZX_PROP_NAME, self_name, sizeof(self_name));
    if (status != ZX_OK) {
      fprintf(stderr, "TraceProvider: error getting process name: status=%d(%s)\n", status,
              zx_status_get_string(status));
      self_name[0] = '\0';
    }
    name = self_name;
  }

  return trace_provider_create_synchronously(to_service, dispatcher, name,
                                             out_manager_is_tracing_already);
}