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 namespace {
31 const std::string LISTENER_THREAD_NAME = "OS_SenConsumer";
32 } // namespace
33
CreateSensorDataChannel(DataChannelCB callBack,void * data)34 int32_t SensorDataChannel::CreateSensorDataChannel(DataChannelCB callBack, void *data)
35 {
36 CHKPR(callBack, SENSOR_NATIVE_REGSITER_CB_ERR);
37 dataCB_ = callBack;
38 privateData_ = data;
39 return InnerSensorDataChannel();
40 }
41
RestoreSensorDataChannel()42 int32_t SensorDataChannel::RestoreSensorDataChannel()
43 {
44 CHKPR(dataCB_, SENSOR_NATIVE_REGSITER_CB_ERR);
45 if (GetReceiveDataFd() != -1) {
46 SEN_HILOGW("Restore sensor data channel failed, please destroy sensor data channel first.");
47 return SENSOR_CHANNEL_RESTORE_FD_ERR;
48 }
49 return InnerSensorDataChannel();
50 }
51
InnerSensorDataChannel()52 int32_t SensorDataChannel::InnerSensorDataChannel()
53 {
54 std::lock_guard<std::mutex> eventRunnerLock(eventRunnerMutex_);
55 // create basic data channel
56 int32_t ret = CreateSensorBasicChannel();
57 if (ret != ERR_OK) {
58 SEN_HILOGE("Create basic channel failed, ret:%{public}d", ret);
59 return ret;
60 }
61 auto listener = std::make_shared<SensorFileDescriptorListener>();
62 listener->SetChannel(this);
63 if (eventHandler_ == nullptr) {
64 auto myRunner = AppExecFwk::EventRunner::Create(LISTENER_THREAD_NAME);
65 CHKPR(myRunner, ERROR);
66 eventHandler_ = std::make_shared<SensorEventHandler>(myRunner);
67 }
68 int32_t receiveFd = GetReceiveDataFd();
69 auto inResult = eventHandler_->AddFileDescriptorListener(receiveFd,
70 AppExecFwk::FILE_DESCRIPTOR_INPUT_EVENT, listener, "SensorTask");
71 if (inResult != 0) {
72 SEN_HILOGE("AddFileDescriptorListener fail");
73 return ERROR;
74 }
75 auto pairRet = listenedFdSet_.insert(receiveFd);
76 if (!pairRet.second) {
77 SEN_HILOGE("ListenedFdSet insert fd fail, fd:%{public}d", receiveFd);
78 return ERROR;
79 }
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(LISTENER_THREAD_NAME);
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