src/devices/bus/drivers/platform/test/test-board.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 <assert.h>
#include <stdint.h>
#include <stdlib.h>
#include <unistd.h>
#include <zircon/assert.h>
#include <zircon/process.h>
#include <zircon/syscalls.h>

#include <iterator>
#include <memory>

#include <ddk/debug.h>
#include <ddk/device.h>
#include <ddk/driver.h>
#include <ddk/metadata.h>
#include <ddk/platform-defs.h>
#include <ddk/protocol/platform/bus.h>
#include <fbl/algorithm.h>

#include "src/devices/bus/drivers/platform/test/test-board-bind.h"
#include "src/devices/bus/drivers/platform/test/test-metadata.h"
#include "src/devices/bus/drivers/platform/test/test-resources.h"
#include "src/devices/bus/drivers/platform/test/test.h"

namespace board_test {

void TestBoard::DdkRelease() { delete this; }

int TestBoard::Thread() {
  zx_status_t status;

  status = GoldfishAddressSpaceInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: GpioInit failed: %d", __func__, status);
  }

  status = GoldfishPipeInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: GpioInit failed: %d", __func__, status);
  }

  status = GpioInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: GpioInit failed: %d", __func__, status);
  }

  status = I2cInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: I2cInit failed: %d", __func__, status);
  }

  status = SpiInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: SpiInit failed: %d", __func__, status);
  }

  status = ClockInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: ClockInit failed: %d", __func__, status);
  }

  status = PowerInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: PowerInit failed: %d", __func__, status);
  }

  status = TestInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: TestInit failed: %d", __func__, status);
  }

  status = PwmInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: PwmInit failed: %d", __func__, status);
  }

  status = RpmbInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: RpmbInit failed: %d", __func__, status);
  }

  status = VregInit();
  if (status != ZX_OK) {
    zxlogf(ERROR, "%s: VregInit failed: %d", __func__, status);
  }

  return 0;
}

zx_status_t TestBoard::Start() {
  int rc = thrd_create_with_name(
      &thread_, [](void* arg) -> int { return reinterpret_cast<TestBoard*>(arg)->Thread(); }, this,
      "test-board-start-thread");
  if (rc != thrd_success) {
    return ZX_ERR_INTERNAL;
  }
  return ZX_OK;
}

zx_status_t TestBoard::Create(zx_device_t* parent) {
  pbus_protocol_t pbus;
  if (device_get_protocol(parent, ZX_PROTOCOL_PBUS, &pbus) != ZX_OK) {
    return ZX_ERR_NOT_SUPPORTED;
  }

  auto board = std::make_unique<TestBoard>(parent, &pbus);

  zx_status_t status = board->DdkAdd("test-board", DEVICE_ADD_NON_BINDABLE);
  if (status != ZX_OK) {
    zxlogf(ERROR, "TestBoard::Create: DdkAdd failed: %d", status);
    return status;
  }

  status = board->Start();
  if (status == ZX_OK) {
    // devmgr is now in charge of the device.
    __UNUSED auto* dummy = board.release();
  }

  // Add a composite device
  const zx_bind_inst_t root_match[] = {
      BI_MATCH(),
  };
  const zx_bind_inst_t power_match[] = {
      BI_ABORT_IF(NE, BIND_PROTOCOL, ZX_PROTOCOL_POWER),
      BI_MATCH_IF(EQ, BIND_POWER_DOMAIN, 3),
  };
  device_fragment_part_t power_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(power_match), power_match},
  };
  const zx_bind_inst_t goldfish_address_space_match[] = {
      BI_MATCH_IF(EQ, BIND_PROTOCOL, ZX_PROTOCOL_GOLDFISH_ADDRESS_SPACE),
  };
  const zx_bind_inst_t goldfish_pipe_match[] = {
      BI_MATCH_IF(EQ, BIND_PROTOCOL, ZX_PROTOCOL_GOLDFISH_PIPE),
  };
  const zx_bind_inst_t gpio_match[] = {
      BI_ABORT_IF(NE, BIND_PROTOCOL, ZX_PROTOCOL_GPIO),
      BI_MATCH_IF(EQ, BIND_GPIO_PIN, 3),
  };
  const zx_bind_inst_t clock_match[] = {
      BI_ABORT_IF(NE, BIND_PROTOCOL, ZX_PROTOCOL_CLOCK),
      BI_MATCH_IF(EQ, BIND_CLOCK_ID, 1),
  };
  const zx_bind_inst_t i2c_match[] = {
      BI_ABORT_IF(NE, BIND_PROTOCOL, ZX_PROTOCOL_I2C),
      BI_ABORT_IF(NE, BIND_I2C_BUS_ID, 1),
      BI_MATCH_IF(EQ, BIND_I2C_ADDRESS, 5),
  };
  const zx_bind_inst_t child2_match[] = {
      BI_ABORT_IF(NE, BIND_PLATFORM_DEV_VID, PDEV_VID_TEST),
      BI_ABORT_IF(NE, BIND_PLATFORM_DEV_PID, PDEV_PID_PBUS_TEST),
      BI_MATCH_IF(EQ, BIND_PLATFORM_DEV_DID, PDEV_DID_TEST_CHILD_2),
  };
  const zx_bind_inst_t child4_match[] = {
      BI_ABORT_IF(NE, BIND_PLATFORM_DEV_VID, PDEV_VID_TEST),
      BI_ABORT_IF(NE, BIND_PLATFORM_DEV_PID, PDEV_PID_PBUS_TEST),
      BI_MATCH_IF(EQ, BIND_PLATFORM_DEV_DID, PDEV_DID_TEST_CHILD_4),
  };
  const zx_bind_inst_t spi_match[] = {
      BI_ABORT_IF(NE, BIND_PROTOCOL, ZX_PROTOCOL_SPI),
      BI_ABORT_IF(NE, BIND_SPI_BUS_ID, 0),
      BI_MATCH_IF(EQ, BIND_SPI_CHIP_SELECT, 0),
  };
  const zx_bind_inst_t pwm_match[] = {
      BI_ABORT_IF(NE, BIND_PROTOCOL, ZX_PROTOCOL_PWM),
      BI_MATCH_IF(EQ, BIND_PWM_ID, 0),
  };
  const zx_bind_inst_t rpmb_match[] = {
      BI_MATCH_IF(EQ, BIND_PROTOCOL, ZX_PROTOCOL_RPMB),
  };
  const zx_bind_inst_t vreg_match[] = {
      BI_MATCH_IF(EQ, BIND_PROTOCOL, ZX_PROTOCOL_VREG),
  };
  device_fragment_part_t goldfish_address_space_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(goldfish_address_space_match), goldfish_address_space_match},
  };
  device_fragment_part_t goldfish_pipe_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(goldfish_pipe_match), goldfish_pipe_match},
  };
  device_fragment_part_t gpio_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(gpio_match), gpio_match},
  };
  device_fragment_part_t clock_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(clock_match), clock_match},
  };
  device_fragment_part_t i2c_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(i2c_match), i2c_match},
  };
  device_fragment_part_t child4_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(child2_match), child2_match},
      {std::size(child4_match), child4_match},
  };
  device_fragment_part_t spi_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(spi_match), spi_match},
  };
  device_fragment_part_t pwm_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(pwm_match), pwm_match},
  };
  device_fragment_part_t rpmb_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(rpmb_match), rpmb_match},
  };
  device_fragment_part_t vreg_fragment[] = {
      {std::size(root_match), root_match},
      {std::size(vreg_match), vreg_match},
  };

  device_fragment_t composite[] = {
      {"gpio", std::size(gpio_fragment), gpio_fragment},
      {"clock", std::size(clock_fragment), clock_fragment},
      {"i2c", std::size(i2c_fragment), i2c_fragment},
      {"power", std::size(power_fragment), power_fragment},
      {"child4", std::size(child4_fragment), child4_fragment},
  };

  struct composite_test_metadata metadata_1 = {
      .composite_device_id = PDEV_DID_TEST_COMPOSITE_1,
      .metadata_value = 12345,
  };

  struct composite_test_metadata metadata_2 = {
      .composite_device_id = PDEV_DID_TEST_COMPOSITE_2,
      .metadata_value = 12345,
  };

  struct composite_test_metadata metadata_goldfish_control = {
      .composite_device_id = PDEV_DID_TEST_GOLDFISH_CONTROL_COMPOSITE,
      .metadata_value = 12345,
  };

  const pbus_metadata_t test_metadata_1[] = {{
      .type = DEVICE_METADATA_PRIVATE,
      .data_buffer = &metadata_1,
      .data_size = sizeof(composite_test_metadata),
  }};

  const pbus_metadata_t test_metadata_2[] = {{
      .type = DEVICE_METADATA_PRIVATE,
      .data_buffer = &metadata_2,
      .data_size = sizeof(composite_test_metadata),
  }};

  const pbus_metadata_t test_metadata_goldfish_control[] = {{
      .type = DEVICE_METADATA_PRIVATE,
      .data_buffer = &metadata_goldfish_control,
      .data_size = sizeof(composite_test_metadata),
  }};

  pbus_dev_t pdev = {};
  pdev.name = "composite-dev";
  pdev.vid = PDEV_VID_TEST;
  pdev.pid = PDEV_PID_PBUS_TEST;
  pdev.did = PDEV_DID_TEST_COMPOSITE_1;
  pdev.metadata_list = test_metadata_1;
  pdev.metadata_count = std::size(test_metadata_1);

  status = pbus_composite_device_add(&pbus, &pdev, composite, std::size(composite), UINT32_MAX);
  if (status != ZX_OK) {
    zxlogf(ERROR, "TestBoard::Create: pbus_composite_device_add failed: %d", status);
  }

  device_fragment_t composite2[] = {
      {"clock", std::size(clock_fragment), clock_fragment},
      {"power", std::size(power_fragment), power_fragment},
      {"child4", std::size(child4_fragment), child4_fragment},
      {"spi", std::size(spi_fragment), spi_fragment},
      {"pwm", std::size(pwm_fragment), pwm_fragment},
      {"rpmb", std::size(rpmb_fragment), rpmb_fragment},
      {"vreg", std::size(vreg_fragment), vreg_fragment},
  };

  pbus_dev_t pdev2 = {};
  pdev2.name = "composite-dev-2";
  pdev2.vid = PDEV_VID_TEST;
  pdev2.pid = PDEV_PID_PBUS_TEST;
  pdev2.did = PDEV_DID_TEST_COMPOSITE_2;
  pdev2.metadata_list = test_metadata_2;
  pdev2.metadata_count = std::size(test_metadata_2);

  status = pbus_composite_device_add(&pbus, &pdev2, composite2, std::size(composite2), UINT32_MAX);

  if (status != ZX_OK) {
    zxlogf(ERROR, "TestBoard::Create: pbus_composite_device_add failed: %d", status);
  }

  device_fragment_t goldfish_composite[] = {
      {"goldfish-address", std::size(goldfish_address_space_fragment),
       goldfish_address_space_fragment},
      {"goldfish-pipe", std::size(goldfish_pipe_fragment), goldfish_pipe_fragment},
  };

  pbus_dev_t pdev_goldfish_composite = {};
  pdev_goldfish_composite.name = "composite-dev-goldfish-control";
  pdev_goldfish_composite.vid = PDEV_VID_TEST;
  pdev_goldfish_composite.pid = PDEV_PID_PBUS_TEST;
  pdev_goldfish_composite.did = PDEV_DID_TEST_GOLDFISH_CONTROL_COMPOSITE;
  pdev_goldfish_composite.metadata_list = test_metadata_goldfish_control;
  pdev_goldfish_composite.metadata_count = std::size(test_metadata_goldfish_control);

  status = pbus_composite_device_add(&pbus, &pdev_goldfish_composite, goldfish_composite,
                                     std::size(goldfish_composite), UINT32_MAX);

  if (status != ZX_OK) {
    zxlogf(ERROR, "TestBoard::Create: pbus_composite_device_add failed: %d", status);
  }

  return status;
}

zx_status_t test_bind(void* ctx, zx_device_t* parent) { return TestBoard::Create(parent); }

static constexpr zx_driver_ops_t driver_ops = []() {
  zx_driver_ops_t ops = {};
  ops.version = DRIVER_OPS_VERSION;
  ops.bind = test_bind;
  return ops;
}();

}  // namespace board_test

ZIRCON_DRIVER(test_board, board_test::driver_ops, "zircon", "0.1");