1/*
2 * Copyright (c) 2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#include "v1_0/usb_ddk_stub.h"
17#include <hdf_base.h>
18#include <hdf_device_desc.h>
19#include <hdf_log.h>
20#include <hdf_sbuf_ipc.h>
21#include "usbd_wrapper.h"
22
23#define HDF_LOG_TAG usb_ddk_driver
24
25using namespace OHOS::HDI::Usb::Ddk::V1_0;
26
27struct HdfUsbDdkHost {
28    struct IDeviceIoService ioService;
29    OHOS::sptr<OHOS::IRemoteObject> stub;
30};
31
32static int32_t UsbDdkDriverDispatch(
33    struct HdfDeviceIoClient *client, int cmdId, struct HdfSBuf *data, struct HdfSBuf *reply)
34{
35    auto *hdfUsbDdkHost = CONTAINER_OF(client->device->service, struct HdfUsbDdkHost, ioService);
36
37    OHOS::MessageParcel *dataParcel = nullptr;
38    OHOS::MessageParcel *replyParcel = nullptr;
39    OHOS::MessageOption option;
40
41    if (SbufToParcel(data, &dataParcel) != HDF_SUCCESS) {
42        HDF_LOGE("%{public}s: invalid data sbuf object to dispatch", __func__);
43        return HDF_ERR_INVALID_PARAM;
44    }
45    if (SbufToParcel(reply, &replyParcel) != HDF_SUCCESS) {
46        HDF_LOGE("%{public}s: invalid reply sbuf object to dispatch", __func__);
47        return HDF_ERR_INVALID_PARAM;
48    }
49
50    return hdfUsbDdkHost->stub->SendRequest(cmdId, *dataParcel, *replyParcel, option);
51}
52
53static int HdfUsbDdkDriverInit(struct HdfDeviceObject *deviceObject)
54{
55    HDF_LOGI("%{public}s: driver init start", __func__);
56    return HDF_SUCCESS;
57}
58
59static int HdfUsbDdkDriverBind(struct HdfDeviceObject *deviceObject)
60{
61    HDF_LOGI("%{public}s: driver bind start", __func__);
62    auto *hdfUsbDdkHost = new (std::nothrow) HdfUsbDdkHost;
63    if (hdfUsbDdkHost == nullptr) {
64        HDF_LOGE("%{public}s: failed to create create HdfUsbDdkHost object", __func__);
65        return HDF_FAILURE;
66    }
67
68    hdfUsbDdkHost->ioService.Dispatch = UsbDdkDriverDispatch;
69    hdfUsbDdkHost->ioService.Open = NULL;
70    hdfUsbDdkHost->ioService.Release = NULL;
71
72    auto serviceImpl = OHOS::HDI::Usb::Ddk::V1_0::IUsbDdk::Get(true);
73    if (serviceImpl == nullptr) {
74        HDF_LOGE("%{public}s: failed to get of implement service", __func__);
75        delete hdfUsbDdkHost;
76        return HDF_FAILURE;
77    }
78
79    hdfUsbDdkHost->stub = OHOS::HDI::ObjectCollector::GetInstance().GetOrNewObject(
80        serviceImpl, OHOS::HDI::Usb::Ddk::V1_0::IUsbDdk::GetDescriptor());
81    if (hdfUsbDdkHost->stub == nullptr) {
82        HDF_LOGE("%{public}s: failed to get stub object", __func__);
83        delete hdfUsbDdkHost;
84        return HDF_FAILURE;
85    }
86
87    deviceObject->service = &hdfUsbDdkHost->ioService;
88    return HDF_SUCCESS;
89}
90
91static void HdfUsbDdkDriverRelease(struct HdfDeviceObject *deviceObject)
92{
93    HDF_LOGI("%{public}s: driver release start", __func__);
94    if (deviceObject->service == nullptr) {
95        return;
96    }
97
98    auto *hdfUsbDdkHost = CONTAINER_OF(deviceObject->service, struct HdfUsbDdkHost, ioService);
99    if (hdfUsbDdkHost != nullptr) {
100        delete hdfUsbDdkHost;
101    }
102}
103
104static struct HdfDriverEntry g_usbddkDriverEntry = {
105    .moduleVersion = 1,
106    .moduleName = "",
107    .Bind = HdfUsbDdkDriverBind,
108    .Init = HdfUsbDdkDriverInit,
109    .Release = HdfUsbDdkDriverRelease,
110};
111#ifdef __cplusplus
112extern "C" {
113#endif /* __cplusplus */
114HDF_INIT(g_usbddkDriverEntry);
115#ifdef __cplusplus
116}
117#endif /* __cplusplus */
118