1 /*
2  * Copyright (c) 2021-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 "sensor_data_channel.h"
17 
18 #include "errors.h"
19 
20 #include "fd_listener.h"
21 #include "sensor_errors.h"
22 #include "sensor_file_descriptor_listener.h"
23 
24 #undef LOG_TAG
25 #define LOG_TAG "SensorDataChannel"
26 namespace OHOS {
27 namespace Sensors {
28 using namespace OHOS::HiviewDFX;
29 using namespace OHOS::AppExecFwk;
30 
CreateSensorDataChannel(DataChannelCB callBack,void * data)31 int32_t SensorDataChannel::CreateSensorDataChannel(DataChannelCB callBack, void *data)
32 {
33     SEN_HILOGI("In");
34     CHKPR(callBack, SENSOR_NATIVE_REGSITER_CB_ERR);
35     dataCB_ = callBack;
36     privateData_ = data;
37     return InnerSensorDataChannel();
38 }
39 
RestoreSensorDataChannel()40 int32_t SensorDataChannel::RestoreSensorDataChannel()
41 {
42     CHKPR(dataCB_, SENSOR_NATIVE_REGSITER_CB_ERR);
43     if (GetReceiveDataFd() != -1) {
44         SEN_HILOGW("Restore sensor data channel failed, please destroy sensor data channel first");
45         return SENSOR_CHANNEL_RESTORE_FD_ERR;
46     }
47     return InnerSensorDataChannel();
48 }
49 
InnerSensorDataChannel()50 int32_t SensorDataChannel::InnerSensorDataChannel()
51 {
52     SEN_HILOGI("In");
53     std::lock_guard<std::mutex> eventRunnerLock(eventRunnerMutex_);
54     // create basic data channel
55     int32_t ret = CreateSensorBasicChannel();
56     if (ret != ERR_OK) {
57         SEN_HILOGE("Create basic channel failed, ret:%{public}d", ret);
58         return ret;
59     }
60     auto listener = std::make_shared<SensorFileDescriptorListener>();
61     listener->SetChannel(this);
62     if (eventHandler_ == nullptr) {
63         auto myRunner = AppExecFwk::EventRunner::Create(true, AppExecFwk::ThreadMode::FFRT);
64         CHKPR(myRunner, ERROR);
65         eventHandler_ = std::make_shared<SensorEventHandler>(myRunner);
66     }
67     int32_t receiveFd = GetReceiveDataFd();
68     auto inResult = eventHandler_->AddFileDescriptorListener(receiveFd,
69         AppExecFwk::FILE_DESCRIPTOR_INPUT_EVENT, listener, "SensorTask");
70     if (inResult != 0) {
71         SEN_HILOGE("AddFileDescriptorListener fail");
72         return ERROR;
73     }
74     auto pairRet = listenedFdSet_.insert(receiveFd);
75     if (!pairRet.second) {
76         SEN_HILOGE("ListenedFdSet insert fd fail, fd:%{public}d", receiveFd);
77         return ERROR;
78     }
79     SEN_HILOGI("Done");
80     return ERR_OK;
81 }
82 
DestroySensorDataChannel()83 int32_t SensorDataChannel::DestroySensorDataChannel()
84 {
85     DelFdListener(GetReceiveDataFd());
86     return DestroySensorBasicChannel();
87 }
88 
~SensorDataChannel()89 SensorDataChannel::~SensorDataChannel()
90 {
91     DestroySensorDataChannel();
92 }
93 
AddFdListener(int32_t fd,ReceiveMessageFun receiveMessage,DisconnectFun disconnect)94 int32_t SensorDataChannel::AddFdListener(int32_t fd, ReceiveMessageFun receiveMessage, DisconnectFun disconnect)
95 {
96     receiveMessage_ = receiveMessage;
97     disconnect_ = disconnect;
98     std::lock_guard<std::mutex> eventRunnerLock(eventRunnerMutex_);
99     if (eventHandler_ == nullptr) {
100         auto myRunner = AppExecFwk::EventRunner::Create(true, AppExecFwk::ThreadMode::FFRT);
101         CHKPR(myRunner, ERROR);
102         eventHandler_ = std::make_shared<SensorEventHandler>(myRunner);
103     }
104     auto listener = std::make_shared<FdListener>();
105     listener->SetChannel(this);
106     auto errCode = eventHandler_->AddFileDescriptorListener(fd, AppExecFwk::FILE_DESCRIPTOR_INPUT_EVENT, listener,
107         "SensorTask");
108     if (errCode != ERR_OK) {
109         SEN_HILOGE("Add fd listener failed, fd:%{public}d, errCode:%{public}u", fd, errCode);
110         return ERROR;
111     }
112     auto pairRet = listenedFdSet_.insert(fd);
113     if (!pairRet.second) {
114         SEN_HILOGE("ListenedFdSet insert fd fail, fd:%{public}d", fd);
115         return ERROR;
116     }
117     return ERR_OK;
118 }
119 
DelFdListener(int32_t fd)120 int32_t SensorDataChannel::DelFdListener(int32_t fd)
121 {
122     std::lock_guard<std::mutex> eventRunnerLock(eventRunnerMutex_);
123     CHKPR(eventHandler_, ERROR);
124     eventHandler_->RemoveFileDescriptorListener(fd);
125     auto it = listenedFdSet_.find(fd);
126     if (it == listenedFdSet_.end()) {
127         SEN_HILOGE("ListenedFdSet not find fd, fd:%{public}d", fd);
128         return ERROR;
129     }
130     listenedFdSet_.erase(it);
131     if (listenedFdSet_.empty() && eventHandler_ != nullptr) {
132         eventHandler_ = nullptr;
133         SEN_HILOGD("Set eventHandler_ nullptr");
134     }
135     return ERR_OK;
136 }
137 
GetReceiveMessageFun() const138 ReceiveMessageFun SensorDataChannel::GetReceiveMessageFun() const
139 {
140     return receiveMessage_;
141 }
142 
GetDisconnectFun() const143 DisconnectFun SensorDataChannel::GetDisconnectFun() const
144 {
145     return disconnect_;
146 }
147 }  // namespace Sensors
148 }  // namespace OHOS
149