1 /*
2 * Copyright (c) 2021-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 "battery_stats_proxy.h"
17
18 #include "battery_stats_ipc_interface_code.h"
19 #include <message_parcel.h>
20 #include "errors.h"
21 #include "message_option.h"
22 #include "stats_common.h"
23 #include "stats_log.h"
24 #include "stats_utils.h"
25
26 namespace OHOS {
27 namespace PowerMgr {
28 constexpr int32_t PARAM_MAX_NUM = 2000;
GetBatteryStats()29 BatteryStatsInfoList BatteryStatsProxy::GetBatteryStats()
30 {
31 STATS_HILOGD(COMP_FWK, "Enter");
32 BatteryStatsInfoList infoList;
33 sptr<IRemoteObject> remote = Remote();
34 STATS_RETURN_IF_WITH_RET(remote == nullptr, infoList);
35
36 MessageParcel data;
37 MessageParcel reply;
38 MessageOption option;
39
40 if (!data.WriteInterfaceToken(BatteryStatsProxy::GetDescriptor())) {
41 STATS_HILOGE(COMP_FWK, "Write descriptor failed");
42 return infoList;
43 }
44
45 int ret = remote->SendRequest(
46 static_cast<uint32_t>(PowerMgr::BatteryStatsInterfaceCode::BATTERY_STATS_GET),
47 data, reply, option);
48 if (ret != ERR_OK) {
49 STATS_HILOGE(COMP_FWK, "Send request is failed, error code: %{public}d", ret);
50 return infoList;
51 }
52 int32_t size = reply.ReadInt32();
53 if (size < 0 || size > PARAM_MAX_NUM) {
54 STATS_HILOGE(COMP_FWK, "size exceed limit, size=%{public}d", size);
55 return infoList;
56 }
57 for (int32_t i = 0; i < size; ++i) {
58 std::shared_ptr<BatteryStatsInfo> info = std::make_shared<BatteryStatsInfo>();
59 info->ReadFromParcel(reply);
60 infoList.emplace_back(info);
61 }
62 int32_t error;
63 STATS_RETURN_IF_READ_PARCEL_FAILED_WITH_RET(COMP_FWK, reply, Int32, error, infoList);
64 lastError_ = static_cast<StatsError>(error);
65 return infoList;
66 }
67
GetTotalTimeSecond(const StatsUtils::StatsType & statsType,const int32_t & uid)68 uint64_t BatteryStatsProxy::GetTotalTimeSecond(const StatsUtils::StatsType& statsType, const int32_t& uid)
69 {
70 STATS_HILOGD(COMP_FWK, "Enter");
71 sptr<IRemoteObject> remote = Remote();
72 STATS_RETURN_IF_WITH_RET(remote == nullptr, StatsUtils::DEFAULT_VALUE);
73
74 MessageParcel data;
75 MessageParcel reply;
76 MessageOption option;
77
78 if (!data.WriteInterfaceToken(BatteryStatsProxy::GetDescriptor())) {
79 STATS_HILOGE(COMP_FWK, "Write descriptor failed");
80 return StatsUtils::DEFAULT_VALUE;
81 }
82
83 uint64_t time = StatsUtils::DEFAULT_VALUE;
84 STATS_RETURN_IF_WRITE_PARCEL_FAILED_WITH_RET(COMP_FWK, data, Int32, static_cast<int32_t>(statsType),
85 StatsUtils::DEFAULT_VALUE);
86 STATS_RETURN_IF_WRITE_PARCEL_FAILED_WITH_RET(COMP_FWK, data, Int32, uid, StatsUtils::DEFAULT_VALUE);
87
88 int ret = remote->SendRequest(
89 static_cast<uint32_t>(PowerMgr::BatteryStatsInterfaceCode::BATTERY_STATS_GETTIME),
90 data, reply, option);
91 if (ret != ERR_OK) {
92 STATS_HILOGE(COMP_FWK, "Transact is failed, error code: %{public}d", ret);
93 }
94
95 STATS_RETURN_IF_READ_PARCEL_FAILED_WITH_RET(COMP_FWK, reply, Uint64, time, StatsUtils::DEFAULT_VALUE);
96 return time;
97 }
98
GetTotalDataBytes(const StatsUtils::StatsType & statsType,const int32_t & uid)99 uint64_t BatteryStatsProxy::GetTotalDataBytes(const StatsUtils::StatsType& statsType, const int32_t& uid)
100 {
101 STATS_HILOGD(COMP_FWK, "Enter");
102 sptr<IRemoteObject> remote = Remote();
103 STATS_RETURN_IF_WITH_RET(remote == nullptr, StatsUtils::DEFAULT_VALUE);
104
105 MessageParcel data;
106 MessageParcel reply;
107 MessageOption option;
108
109 if (!data.WriteInterfaceToken(BatteryStatsProxy::GetDescriptor())) {
110 STATS_HILOGE(COMP_FWK, "Write descriptor failed");
111 return StatsUtils::DEFAULT_VALUE;
112 }
113
114 uint64_t count = StatsUtils::DEFAULT_VALUE;
115 STATS_RETURN_IF_WRITE_PARCEL_FAILED_WITH_RET(COMP_FWK, data, Int32, static_cast<int32_t>(statsType),
116 StatsUtils::DEFAULT_VALUE);
117 STATS_RETURN_IF_WRITE_PARCEL_FAILED_WITH_RET(COMP_FWK, data, Int32, uid, StatsUtils::DEFAULT_VALUE);
118
119 int ret = remote->SendRequest(
120 static_cast<uint32_t>(PowerMgr::BatteryStatsInterfaceCode::BATTERY_STATS_GETDATA),
121 data, reply, option);
122 if (ret != ERR_OK) {
123 STATS_HILOGE(COMP_FWK, "Transact is failed, error code: %{public}d", ret);
124 }
125
126 STATS_RETURN_IF_READ_PARCEL_FAILED_WITH_RET(COMP_FWK, reply, Uint64, count, StatsUtils::DEFAULT_VALUE);
127 return count;
128 }
129
GetAppStatsMah(const int32_t & uid)130 double BatteryStatsProxy::GetAppStatsMah(const int32_t& uid)
131 {
132 STATS_HILOGD(COMP_FWK, "Enter");
133 sptr<IRemoteObject> remote = Remote();
134 STATS_RETURN_IF_WITH_RET(remote == nullptr, StatsUtils::DEFAULT_VALUE);
135
136 MessageParcel data;
137 MessageParcel reply;
138 MessageOption option;
139
140 if (!data.WriteInterfaceToken(BatteryStatsProxy::GetDescriptor())) {
141 STATS_HILOGE(COMP_FWK, "Write descriptor failed");
142 return StatsUtils::DEFAULT_VALUE;
143 }
144
145 data.WriteInt32(uid);
146
147 int ret = remote->SendRequest(
148 static_cast<uint32_t>(PowerMgr::BatteryStatsInterfaceCode::BATTERY_STATS_GETAPPMAH),
149 data, reply, option);
150 if (ret != ERR_OK) {
151 STATS_HILOGE(COMP_FWK, "Transact is failed, error code: %{public}d", ret);
152 }
153
154 double appStatsMah = reply.ReadDouble();
155 STATS_HILOGD(COMP_FWK, "Get stats mah: %{public}lf for uid: %{public}d", appStatsMah, uid);
156 int32_t error;
157 STATS_RETURN_IF_READ_PARCEL_FAILED_WITH_RET(COMP_FWK, reply, Int32, error, appStatsMah);
158 lastError_ = static_cast<StatsError>(error);
159 return appStatsMah;
160 }
161
SetOnBattery(bool isOnBattery)162 void BatteryStatsProxy::SetOnBattery(bool isOnBattery)
163 {
164 STATS_HILOGD(COMP_FWK, "Enter");
165 sptr<IRemoteObject> remote = Remote();
166 STATS_RETURN_IF(remote == nullptr);
167
168 MessageParcel data;
169 MessageParcel reply;
170 MessageOption option;
171
172 if (!data.WriteInterfaceToken(BatteryStatsProxy::GetDescriptor())) {
173 STATS_HILOGE(COMP_FWK, "Write descriptor failed");
174 return;
175 }
176
177 data.WriteBool(isOnBattery);
178
179 int ret = remote->SendRequest(
180 static_cast<uint32_t>(PowerMgr::BatteryStatsInterfaceCode::BATTERY_STATS_SETONBATT),
181 data, reply, option);
182 if (ret != ERR_OK) {
183 STATS_HILOGE(COMP_FWK, "Transact is failed, error code: %{public}d", ret);
184 }
185 }
186
GetAppStatsPercent(const int32_t & uid)187 double BatteryStatsProxy::GetAppStatsPercent(const int32_t& uid)
188 {
189 STATS_HILOGD(COMP_FWK, "Enter");
190 sptr<IRemoteObject> remote = Remote();
191 STATS_RETURN_IF_WITH_RET(remote == nullptr, StatsUtils::DEFAULT_VALUE);
192
193 MessageParcel data;
194 MessageParcel reply;
195 MessageOption option;
196
197 if (!data.WriteInterfaceToken(BatteryStatsProxy::GetDescriptor())) {
198 STATS_HILOGE(COMP_FWK, "Write descriptor failed");
199 return StatsUtils::DEFAULT_VALUE;
200 }
201
202 data.WriteInt32(uid);
203
204 int ret = remote->SendRequest(
205 static_cast<uint32_t>(PowerMgr::BatteryStatsInterfaceCode::BATTERY_STATS_GETAPPPER),
206 data, reply, option);
207 if (ret != ERR_OK) {
208 STATS_HILOGE(COMP_FWK, "Transact is failed, error code: %{public}d", ret);
209 }
210
211 double appStatsPercent = reply.ReadDouble();
212 STATS_HILOGD(COMP_FWK, "Get stats percent: %{public}lf for uid: %{public}d", appStatsPercent, uid);
213 int32_t error;
214 STATS_RETURN_IF_READ_PARCEL_FAILED_WITH_RET(COMP_FWK, reply, Int32, error, appStatsPercent);
215 lastError_ = static_cast<StatsError>(error);
216 return appStatsPercent;
217 }
218
GetPartStatsMah(const BatteryStatsInfo::ConsumptionType & type)219 double BatteryStatsProxy::GetPartStatsMah(const BatteryStatsInfo::ConsumptionType& type)
220 {
221 STATS_HILOGD(COMP_FWK, "Enter");
222 sptr<IRemoteObject> remote = Remote();
223 STATS_RETURN_IF_WITH_RET(remote == nullptr, StatsUtils::DEFAULT_VALUE);
224
225 MessageParcel data;
226 MessageParcel reply;
227 MessageOption option;
228
229 if (!data.WriteInterfaceToken(BatteryStatsProxy::GetDescriptor())) {
230 STATS_HILOGE(COMP_FWK, "Write descriptor failed");
231 return StatsUtils::DEFAULT_VALUE;
232 }
233
234 data.WriteInt32(type);
235
236 int ret = remote->SendRequest(
237 static_cast<uint32_t>(PowerMgr::BatteryStatsInterfaceCode::BATTERY_STATS_GETPARTMAH),
238 data, reply, option);
239 if (ret != ERR_OK) {
240 STATS_HILOGE(COMP_FWK, "Transact is failed, error code: %{public}d", ret);
241 }
242
243 double partStatsMah = reply.ReadDouble();
244 STATS_HILOGD(COMP_FWK, "Get stats mah: %{public}lf for type: %{public}d", partStatsMah, type);
245 int32_t error;
246 STATS_RETURN_IF_READ_PARCEL_FAILED_WITH_RET(COMP_FWK, reply, Int32, error, partStatsMah);
247 lastError_ = static_cast<StatsError>(error);
248 return partStatsMah;
249 }
250
GetPartStatsPercent(const BatteryStatsInfo::ConsumptionType & type)251 double BatteryStatsProxy::GetPartStatsPercent(const BatteryStatsInfo::ConsumptionType& type)
252 {
253 STATS_HILOGD(COMP_FWK, "Enter");
254 sptr<IRemoteObject> remote = Remote();
255 STATS_RETURN_IF_WITH_RET(remote == nullptr, StatsUtils::DEFAULT_VALUE);
256
257 MessageParcel data;
258 MessageParcel reply;
259 MessageOption option;
260
261 if (!data.WriteInterfaceToken(BatteryStatsProxy::GetDescriptor())) {
262 STATS_HILOGE(COMP_FWK, "Write descriptor failed");
263 return StatsUtils::DEFAULT_VALUE;
264 }
265
266 data.WriteInt32(type);
267
268 int ret = remote->SendRequest(
269 static_cast<uint32_t>(PowerMgr::BatteryStatsInterfaceCode::BATTERY_STATS_GETPARTPER),
270 data, reply, option);
271 if (ret != ERR_OK) {
272 STATS_HILOGE(COMP_FWK, "Transact is failed, error code: %{public}d", ret);
273 }
274
275 double partStatsPercent = reply.ReadDouble();
276 STATS_HILOGD(COMP_FWK, "Get stats percent: %{public}lf for type: %{public}d", partStatsPercent, type);
277 int32_t error;
278 STATS_RETURN_IF_READ_PARCEL_FAILED_WITH_RET(COMP_FWK, reply, Int32, error, partStatsPercent);
279 lastError_ = static_cast<StatsError>(error);
280 return partStatsPercent;
281 }
282
Reset()283 void BatteryStatsProxy::Reset()
284 {
285 STATS_HILOGD(COMP_FWK, "Enter");
286 sptr<IRemoteObject> remote = Remote();
287 STATS_RETURN_IF(remote == nullptr);
288
289 MessageParcel data;
290 MessageParcel reply;
291 MessageOption option;
292
293 if (!data.WriteInterfaceToken(BatteryStatsProxy::GetDescriptor())) {
294 STATS_HILOGE(COMP_FWK, "Write descriptor failed");
295 return;
296 }
297
298 int ret = remote->SendRequest(
299 static_cast<uint32_t>(PowerMgr::BatteryStatsInterfaceCode::BATTERY_STATS_RESET),
300 data, reply, option);
301 if (ret != ERR_OK) {
302 STATS_HILOGE(COMP_FWK, "Transact is failed, error code: %{public}d", ret);
303 }
304 }
305
ShellDump(const std::vector<std::string> & args,uint32_t argc)306 std::string BatteryStatsProxy::ShellDump(const std::vector<std::string>& args, uint32_t argc)
307 {
308 sptr<IRemoteObject> remote = Remote();
309 std::string result = "remote error";
310 STATS_RETURN_IF_WITH_RET(remote == nullptr, result);
311
312 MessageParcel data;
313 MessageParcel reply;
314 MessageOption option;
315
316 if (!data.WriteInterfaceToken(BatteryStatsProxy::GetDescriptor())) {
317 STATS_HILOGE(COMP_FWK, "Write descriptor failed");
318 return result;
319 }
320 if (argc > args.size()) {
321 STATS_HILOGE(COMP_FWK, "argc is greater than args size!");
322 return result;
323 }
324
325 data.WriteUint32(argc);
326 for (uint32_t i = 0; i < argc; i++) {
327 data.WriteString(args[i]);
328 }
329 int ret = remote->SendRequest(
330 static_cast<uint32_t>(PowerMgr::BatteryStatsInterfaceCode::BATTERY_STATS_DUMP),
331 data, reply, option);
332 if (ret != ERR_OK) {
333 STATS_HILOGE(COMP_FWK, "SendRequest is failed, error code: %{public}d", ret);
334 return result;
335 }
336 result = reply.ReadString();
337
338 return result;
339 }
340
GetLastError()341 StatsError BatteryStatsProxy::GetLastError()
342 {
343 StatsError tmpError = lastError_;
344 lastError_ = StatsError::ERR_OK;
345 return tmpError;
346 }
347 } // namespace PowerMgr
348 } // namespace OHOS
349