1 /*
2 * Copyright (c) 2022 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 <hdf_base.h>
17 #include <hdf_device_desc.h>
18 #include <hdf_log.h>
19 #include <hdf_sbuf_ipc.h>
20 #include "v1_0/nnrt_device_stub.h"
21
22 using namespace OHOS::HDI::Nnrt::V1_0;
23
24 struct HdfNnrtDeviceHost {
25 struct IDeviceIoService ioService;
26 OHOS::sptr<OHOS::IRemoteObject> stub;
27 };
28
NnrtDeviceDriverDispatch(struct HdfDeviceIoClient * client,int cmdId,struct HdfSBuf * data,struct HdfSBuf * reply)29 static int32_t NnrtDeviceDriverDispatch(struct HdfDeviceIoClient *client, int cmdId, struct HdfSBuf *data,
30 struct HdfSBuf *reply)
31 {
32 auto *hdfNnrtDeviceHost = CONTAINER_OF(client->device->service, struct HdfNnrtDeviceHost, ioService);
33
34 OHOS::MessageParcel *dataParcel = nullptr;
35 OHOS::MessageParcel *replyParcel = nullptr;
36 OHOS::MessageOption option;
37
38 if (SbufToParcel(data, &dataParcel) != HDF_SUCCESS) {
39 HDF_LOGE("%{public}s:invalid data sbuf object to dispatch", __func__);
40 return HDF_ERR_INVALID_PARAM;
41 }
42 if (SbufToParcel(reply, &replyParcel) != HDF_SUCCESS) {
43 HDF_LOGE("%{public}s:invalid reply sbuf object to dispatch", __func__);
44 return HDF_ERR_INVALID_PARAM;
45 }
46
47 return hdfNnrtDeviceHost->stub->SendRequest(cmdId, *dataParcel, *replyParcel, option);
48 }
49
HdfNnrtDeviceDriverInit(struct HdfDeviceObject * deviceObject)50 static int HdfNnrtDeviceDriverInit(struct HdfDeviceObject *deviceObject)
51 {
52 HDF_LOGI("HdfNnrtDeviceDriverInit enter");
53 return HDF_SUCCESS;
54 }
55
HdfNnrtDeviceDriverBind(struct HdfDeviceObject * deviceObject)56 static int HdfNnrtDeviceDriverBind(struct HdfDeviceObject *deviceObject)
57 {
58 HDF_LOGI("HdfNnrtDeviceDriverBind enter");
59
60 auto *hdfNnrtDeviceHost = new (std::nothrow) HdfNnrtDeviceHost;
61 if (hdfNnrtDeviceHost == nullptr) {
62 HDF_LOGE("%{public}s: failed to create create HdfNnrtDeviceHost object", __func__);
63 return HDF_FAILURE;
64 }
65
66 hdfNnrtDeviceHost->ioService.Dispatch = NnrtDeviceDriverDispatch;
67 hdfNnrtDeviceHost->ioService.Open = NULL;
68 hdfNnrtDeviceHost->ioService.Release = NULL;
69
70 auto serviceImpl = INnrtDevice::Get(true);
71 if (serviceImpl == nullptr) {
72 HDF_LOGE("%{public}s: failed to get of implement service", __func__);
73 delete hdfNnrtDeviceHost;
74 return HDF_FAILURE;
75 }
76
77 hdfNnrtDeviceHost->stub = OHOS::HDI::ObjectCollector::GetInstance().GetOrNewObject(serviceImpl,
78 INnrtDevice::GetDescriptor());
79 if (hdfNnrtDeviceHost->stub == nullptr) {
80 HDF_LOGE("%{public}s: failed to get stub object", __func__);
81 delete hdfNnrtDeviceHost;
82 return HDF_FAILURE;
83 }
84
85 deviceObject->service = &hdfNnrtDeviceHost->ioService;
86 return HDF_SUCCESS;
87 }
88
HdfNnrtDeviceDriverRelease(struct HdfDeviceObject * deviceObject)89 static void HdfNnrtDeviceDriverRelease(struct HdfDeviceObject *deviceObject)
90 {
91 HDF_LOGI("HdfNnrtDeviceDriverRelease enter");
92 if (deviceObject->service == nullptr) {
93 HDF_LOGE("HdfNnrtDeviceDriverRelease not initted");
94 return;
95 }
96
97 auto *hdfNnrtDeviceHost = CONTAINER_OF(deviceObject->service, struct HdfNnrtDeviceHost, ioService);
98 delete hdfNnrtDeviceHost;
99 }
100
101 struct HdfDriverEntry g_nnrtdeviceDriverEntry = {
102 .moduleVersion = 1,
103 .moduleName = "nnrt",
104 .Bind = HdfNnrtDeviceDriverBind,
105 .Init = HdfNnrtDeviceDriverInit,
106 .Release = HdfNnrtDeviceDriverRelease,
107 };
108
109 #ifdef __cplusplus
110 extern "C" {
111 #endif /* __cplusplus */
112 HDF_INIT(g_nnrtdeviceDriverEntry);
113 #ifdef __cplusplus
114 }
115 #endif /* __cplusplus */