1// Copyright 2018 The Fuchsia Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include <ddk/binding.h>
6#include <ddk/debug.h>
7#include <ddk/device.h>
8#include <ddk/driver.h>
9#include <ddk/protocol/platform-defs.h>
10#include <ddk/protocol/platform-device.h>
11
12#include "../qemu-virt.h"
13
14#define DRIVER_NAME "qemu-test-child-1"
15
16typedef struct {
17    zx_device_t* zxdev;
18} qemu_test_t;
19
20static void qemu_test_release(void* ctx) {
21    free(ctx);
22}
23
24static zx_protocol_device_t qemu_test_device_protocol = {
25    .version = DEVICE_OPS_VERSION,
26    .release = qemu_test_release,
27};
28
29static zx_status_t qemu_test_bti(platform_device_protocol_t* pdev) {
30    zx_status_t status;
31    zx_handle_t bti;
32
33    status = pdev_get_bti(pdev, 0, &bti);
34
35    if (status != ZX_OK) {
36        zxlogf(ERROR, "%s: failed to get bti, st = %d\n", DRIVER_NAME, status);
37    }
38
39    zx_handle_close(bti);
40
41    return status;
42}
43
44static zx_status_t qemu_test_bind(void* ctx, zx_device_t* parent) {
45    platform_device_protocol_t pdev;
46    zx_status_t status;
47
48    zxlogf(INFO, "qemu_test_bind: %s \n", DRIVER_NAME);
49
50    status = device_get_protocol(parent, ZX_PROTOCOL_PLATFORM_DEV, &pdev);
51    if (status != ZX_OK) {
52        zxlogf(ERROR, "%s: could not get ZX_PROTOCOL_PLATFORM_DEV\n", DRIVER_NAME);
53        return status;
54    }
55
56    // Make sure we can access our MMIO.
57    mmio_buffer_t mmio;
58    status = pdev_map_mmio_buffer2(&pdev, 0, ZX_CACHE_POLICY_UNCACHED_DEVICE, &mmio);
59    if (status != ZX_OK) {
60        zxlogf(ERROR, "%s: pdev_map_mmio_buffer failed\n", DRIVER_NAME);
61        return status;
62    }
63    if (mmio.size != TEST_MMIO_2_SIZE) {
64        zxlogf(ERROR, "%s: mmio.size expected %u got %zu\n", DRIVER_NAME, TEST_MMIO_2_SIZE,
65               mmio.size);
66    }
67    mmio_buffer_release(&mmio);
68
69    status = qemu_test_bti(&pdev);
70    if (status != ZX_OK) {
71        zxlogf(ERROR, "%s: bti test failed, st = %d\n", DRIVER_NAME, status);
72    }
73
74    qemu_test_t* child_2 = calloc(1, sizeof(qemu_test_t));
75    if (!child_2) {
76        return ZX_ERR_NO_MEMORY;
77    }
78
79    zx_device_prop_t child_2_props[] = {
80        { BIND_PROTOCOL, 0, ZX_PROTOCOL_PLATFORM_DEV },
81        { BIND_PLATFORM_DEV_VID, 0, PDEV_VID_QEMU },
82        { BIND_PLATFORM_DEV_PID, 0, PDEV_PID_QEMU },
83        { BIND_PLATFORM_DEV_DID, 0, PDEV_DID_QEMU_TEST_CHILD_2 },
84    };
85
86    device_add_args_t child_2_args = {
87        .version = DEVICE_ADD_ARGS_VERSION,
88        .name = "child-2",
89        .ctx = child_2,
90        .ops = &qemu_test_device_protocol,
91        .props = child_2_props,
92        .prop_count = countof(child_2_props),
93    };
94
95    status = pdev_device_add(&pdev, 0, &child_2_args, &child_2->zxdev);
96    if (status != ZX_OK) {
97        zxlogf(ERROR, "%s: pdev_device_add failed: %d\n", DRIVER_NAME, status);
98        free(child_2);
99        return status;
100    }
101
102    qemu_test_t* child_3 = calloc(1, sizeof(qemu_test_t));
103    if (!child_3) {
104        return ZX_ERR_NO_MEMORY;
105    }
106
107    zx_device_prop_t child_3_props[] = {
108        { BIND_PROTOCOL, 0, ZX_PROTOCOL_PLATFORM_DEV },
109        { BIND_PLATFORM_DEV_VID, 0, PDEV_VID_QEMU },
110        { BIND_PLATFORM_DEV_PID, 0, PDEV_PID_QEMU },
111        { BIND_PLATFORM_DEV_DID, 0, PDEV_DID_QEMU_TEST_CHILD_3 },
112    };
113
114    device_add_args_t child_3_args = {
115        .version = DEVICE_ADD_ARGS_VERSION,
116        .name = "child-3",
117        .ctx = child_3,
118        .ops = &qemu_test_device_protocol,
119        .props = child_3_props,
120        .prop_count = countof(child_3_props),
121    };
122
123    status = pdev_device_add(&pdev, 1, &child_3_args, &child_3->zxdev);
124    if (status != ZX_OK) {
125        zxlogf(ERROR, "%s: pdev_device_add failed: %d\n", DRIVER_NAME, status);
126        free(child_3);
127        return status;
128    }
129
130    return ZX_OK;
131}
132
133static zx_driver_ops_t qemu_test_driver_ops = {
134    .version = DRIVER_OPS_VERSION,
135    .bind = qemu_test_bind,
136};
137
138ZIRCON_DRIVER_BEGIN(qemu_bus, qemu_test_driver_ops, "zircon", "0.1", 4)
139    BI_ABORT_IF(NE, BIND_PROTOCOL, ZX_PROTOCOL_PLATFORM_DEV),
140    BI_ABORT_IF(NE, BIND_PLATFORM_DEV_VID, PDEV_VID_QEMU),
141    BI_ABORT_IF(NE, BIND_PLATFORM_DEV_PID, PDEV_PID_QEMU),
142    BI_MATCH_IF(EQ, BIND_PLATFORM_DEV_DID, PDEV_DID_QEMU_TEST_CHILD_1),
143ZIRCON_DRIVER_END(qemu_bus)
144