system/ulib/ddktl/include/ddktl/protocol/iommu.h

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

// WARNING: This file is machine generated by fidlc.

#pragma once

#include <ddk/protocol/iommu.h>
#include <ddktl/device-internal.h>
#include <zircon/assert.h>
#include <zircon/compiler.h>
#include <zircon/types.h>

#include "iommu-internal.h"

// DDK iommu-protocol support
//
// :: Proxies ::
//
// ddk::IommuProtocolProxy is a simple wrapper around
// iommu_protocol_t. It does not own the pointers passed to it
//
// :: Mixins ::
//
// ddk::IommuProtocol is a mixin class that simplifies writing DDK drivers
// that implement the iommu protocol. It doesn't set the base protocol.
//
// :: Examples ::
//
// // A driver that implements a ZX_PROTOCOL_IOMMU device.
// class IommuDevice {
// using IommuDeviceType = ddk::Device<IommuDevice, /* ddk mixins */>;
//
// class IommuDevice : public IommuDeviceType,
//                     public ddk::IommuProtocol<IommuDevice> {
//   public:
//     IommuDevice(zx_device_t* parent)
//         : IommuDeviceType("my-iommu-protocol-device", parent) {}
//
//     zx_status_t IommuGetBti(uint32_t iommu_index, uint32_t bti_id, zx_handle_t* out_handle);
//
//     ...
// };

namespace ddk {

template <typename D>
class IommuProtocol : public internal::base_mixin {
public:
    IommuProtocol() {
        internal::CheckIommuProtocolSubclass<D>();
        iommu_protocol_ops_.get_bti = IommuGetBti;
    }

protected:
    iommu_protocol_ops_t iommu_protocol_ops_ = {};

private:
    static zx_status_t IommuGetBti(void* ctx, uint32_t iommu_index, uint32_t bti_id,
                                   zx_handle_t* out_handle) {
        return static_cast<D*>(ctx)->IommuGetBti(iommu_index, bti_id, out_handle);
    }
};

class IommuProtocolProxy {
public:
    IommuProtocolProxy() : ops_(nullptr), ctx_(nullptr) {}
    IommuProtocolProxy(const iommu_protocol_t* proto) : ops_(proto->ops), ctx_(proto->ctx) {}

    void GetProto(iommu_protocol_t* proto) {
        proto->ctx = ctx_;
        proto->ops = ops_;
    }
    bool is_valid() { return ops_ != nullptr; }
    void clear() {
        ctx_ = nullptr;
        ops_ = nullptr;
    }
    zx_status_t GetBti(uint32_t iommu_index, uint32_t bti_id, zx_handle_t* out_handle) {
        return ops_->get_bti(ctx_, iommu_index, bti_id, out_handle);
    }

private:
    iommu_protocol_ops_t* ops_;
    void* ctx_;
};

} // namespace ddk