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 <vector>
17 #include "iremote_stub.h"
18 #include "wifi_datashare_utils.h"
19 #ifdef HAS_ACCOUNT_PART
20 #include "os_account_manager.h"
21 #endif
22 #include "wifi_log.h"
23 #include "wifi_logger.h"
24 #include "system_ability_definition.h"
25 
26 DEFINE_WIFILOG_LABEL("WifiDataShareHelperUtils");
27 
28 namespace OHOS {
29 namespace Wifi {
30 namespace {
31 constexpr const char *SETTINGS_DATASHARE_URI =
32     "datashare:///com.ohos.settingsdata/entry/settingsdata/SETTINGSDATA?Proxy=true";
33 constexpr const char *SETTINGS_DATA_EXT_URI = "datashare:///com.ohos.settingsdata.DataAbility";
34 constexpr const char *SETTINGS_DATA_COLUMN_KEYWORD = "KEYWORD";
35 constexpr const char *SETTINGS_DATA_COLUMN_VALUE = "VALUE";
36 
37 // E_OK and E_DATA_SHARE_NOT_READY used to check datashare ready
38 constexpr const int32_t E_OK = 0;
39 constexpr const int32_t E_DATA_SHARE_NOT_READY = 1055;
40 }
41 
IsDataMgrServiceActive()42 bool WifiDataShareHelperUtils::IsDataMgrServiceActive()
43 {
44     sptr<ISystemAbilityManager> saMgr = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
45     if (saMgr == nullptr) {
46         WIFI_LOGE("Failed to get SystemAbilityManager!");
47         return false;
48     }
49     sptr<IRemoteObject> object = saMgr->CheckSystemAbility(DISTRIBUTED_KV_DATA_SERVICE_ABILITY_ID);
50     if (object == nullptr) {
51         WIFI_LOGE("Failed to get DataMgrService!");
52         return false;
53     }
54     return true;
55 }
56 
CheckIfSettingsDataReady()57 bool WifiDataShareHelperUtils::CheckIfSettingsDataReady()
58 {
59     if (isDataShareReady_) {
60         return true;
61     }
62 
63     if (!IsDataMgrServiceActive()) {
64         return false;
65     }
66 
67     auto remote = sptr<IWifiDataShareRemoteBroker>(new (std::nothrow) IRemoteStub<IWifiDataShareRemoteBroker>());
68     if (remote == nullptr) {
69         WIFI_LOGE("%{public}s remote is nullptr", __func__);
70         return false;
71     }
72     auto remoteObj = remote->AsObject();
73     if (remoteObj == nullptr) {
74         WIFI_LOGE("%{public}s remoteObj_ is nullptr", __func__);
75         return false;
76     }
77 
78     std::pair<int, std::shared_ptr<DataShare::DataShareHelper>> ret =
79         DataShare::DataShareHelper::Create(remoteObj, SETTINGS_DATASHARE_URI, SETTINGS_DATA_EXT_URI);
80     WIFI_LOGI("%{public}s create datashare helper, ret = %{public}d", __func__, ret.first);
81 
82     if (ret.first == E_DATA_SHARE_NOT_READY) {
83         return false;
84     }
85 
86     if (ret.first == E_OK) {
87         if (ret.second) {
88             ret.second->Release();
89         }
90         isDataShareReady_ = true;
91     }
92     return true;
93 }
94 
WifiCreateDataShareHelper(bool onlySettingsData)95 std::shared_ptr<DataShare::DataShareHelper> WifiDataShareHelperUtils::WifiCreateDataShareHelper(bool onlySettingsData)
96 {
97     if (!CheckIfSettingsDataReady()) {
98         WIFI_LOGE("%{public}s datashare not ready.", __func__);
99         return nullptr;
100     }
101 
102     auto remote = sptr<IWifiDataShareRemoteBroker>(new (std::nothrow) IRemoteStub<IWifiDataShareRemoteBroker>());
103     if (remote == nullptr) {
104         WIFI_LOGE("%{public}s remote is nullptr", __func__);
105         return nullptr;
106     }
107     auto remoteObj = remote->AsObject();
108     if (remoteObj == nullptr) {
109         WIFI_LOGE("%{public}s remoteObj_ is nullptr", __func__);
110         return nullptr;
111     }
112     if (onlySettingsData) {
113         return DataShare::DataShareHelper::Creator(remoteObj, SETTINGS_DATA_EXT_URI);
114     }
115     return DataShare::DataShareHelper::Creator(remoteObj, SETTINGS_DATASHARE_URI, SETTINGS_DATA_EXT_URI);
116 }
117 
Query(Uri & uri,const std::string & key,std::string & value,bool onlySettingsData)118 ErrCode WifiDataShareHelperUtils::Query(Uri &uri, const std::string &key, std::string &value, bool onlySettingsData)
119 {
120     std::shared_ptr<DataShare::DataShareHelper> queryHelper = WifiCreateDataShareHelper(onlySettingsData);
121     CHECK_NULL_AND_RETURN(queryHelper, WIFI_OPT_FAILED);
122 
123     DataShare::DataSharePredicates predicates;
124     std::vector<std::string> columns;
125     predicates.EqualTo(SETTINGS_DATA_COLUMN_KEYWORD, key);
126     auto result = queryHelper->Query(uri, predicates, columns);
127     if (result == nullptr) {
128         WIFI_LOGE("WifiDataShareHelper query error, result is null");
129         ClearResources(queryHelper, result);
130         return WIFI_OPT_FAILED;
131     }
132 
133     if (result->GoToFirstRow() != DataShare::E_OK) {
134         WIFI_LOGE("WifiDataShareHelper query failed,go to first row error");
135         ClearResources(queryHelper, result);
136         return WIFI_OPT_FAILED;
137     }
138 
139     int columnIndex;
140     result->GetColumnIndex(SETTINGS_DATA_COLUMN_VALUE, columnIndex);
141     result->GetString(columnIndex, value);
142     ClearResources(queryHelper, result);
143     WIFI_LOGI("WifiDataShareHelper query success,value[%{public}s]", value.c_str());
144     return WIFI_OPT_SUCCESS;
145 }
146 
Insert(Uri & uri,const std::string & key,const std::string & value)147 ErrCode WifiDataShareHelperUtils::Insert(Uri &uri, const std::string &key, const std::string &value)
148 {
149     std::shared_ptr<DataShare::DataShareHelper> insertHelper = WifiCreateDataShareHelper();
150     CHECK_NULL_AND_RETURN(insertHelper, WIFI_OPT_FAILED);
151 
152     DataShare::DataShareValuesBucket valuesBucket;
153     DataShare::DataShareValueObject keyObj(key);
154     DataShare::DataShareValueObject valueObj(value);
155     valuesBucket.Put(SETTINGS_DATA_COLUMN_KEYWORD, keyObj);
156     valuesBucket.Put(SETTINGS_DATA_COLUMN_VALUE, valueObj);
157     int result = insertHelper->Insert(uri, valuesBucket);
158     if (result <= 0) {
159         WIFI_LOGE("WifiDataShareHelper insert failed, resultCode=%{public}d", result);
160         ClearResources(insertHelper, nullptr);
161         return WIFI_OPT_FAILED;
162     }
163     insertHelper->NotifyChange(uri);
164     ClearResources(insertHelper, nullptr);
165     WIFI_LOGE("DataShareHelper insert success");
166     return WIFI_OPT_SUCCESS;
167 }
168 
Update(Uri & uri,const std::string & key,const std::string & value)169 ErrCode WifiDataShareHelperUtils::Update(Uri &uri, const std::string &key, const std::string &value)
170 {
171     std::shared_ptr<DataShare::DataShareHelper> updateHelper = WifiCreateDataShareHelper();
172     CHECK_NULL_AND_RETURN(updateHelper, WIFI_OPT_FAILED);
173 
174     DataShare::DataShareValuesBucket valuesBucket;
175     DataShare::DataShareValueObject valueObj(value);
176     valuesBucket.Put(SETTINGS_DATA_COLUMN_VALUE, valueObj);
177     DataShare::DataSharePredicates predicates;
178     predicates.EqualTo(SETTINGS_DATA_COLUMN_KEYWORD, key);
179     int result = updateHelper->Update(uri, predicates, valuesBucket);
180     if (result <= 0) {
181         WIFI_LOGE("WifiDataShareHelper update failed, resultCode=%{public}d", result);
182         ClearResources(updateHelper, nullptr);
183         return WIFI_OPT_FAILED;
184     }
185     updateHelper->NotifyChange(uri);
186     ClearResources(updateHelper, nullptr);
187     WIFI_LOGE("DataShareHelper update success");
188     return WIFI_OPT_SUCCESS;
189 }
190 
RegisterObserver(const Uri & uri,const sptr<AAFwk::IDataAbilityObserver> & observer)191 ErrCode WifiDataShareHelperUtils::RegisterObserver(const Uri &uri, const sptr<AAFwk::IDataAbilityObserver> &observer)
192 {
193     std::shared_ptr<DataShare::DataShareHelper> registerHelper = WifiCreateDataShareHelper();
194     CHECK_NULL_AND_RETURN(registerHelper, WIFI_OPT_FAILED);
195     CHECK_NULL_AND_RETURN(observer, WIFI_OPT_FAILED);
196     registerHelper->RegisterObserver(uri, observer);
197     ClearResources(registerHelper, nullptr);
198     return WIFI_OPT_SUCCESS;
199 }
200 
UnRegisterObserver(const Uri & uri,const sptr<AAFwk::IDataAbilityObserver> & observer)201 ErrCode WifiDataShareHelperUtils::UnRegisterObserver(const Uri &uri, const sptr<AAFwk::IDataAbilityObserver> &observer)
202 {
203     std::shared_ptr<DataShare::DataShareHelper> unregisterHelper = WifiCreateDataShareHelper();
204     CHECK_NULL_AND_RETURN(unregisterHelper, WIFI_OPT_FAILED);
205     CHECK_NULL_AND_RETURN(observer, WIFI_OPT_FAILED);
206     unregisterHelper->UnregisterObserver(uri, observer);
207     ClearResources(unregisterHelper, nullptr);
208     return WIFI_OPT_SUCCESS;
209 }
210 
GetLoactionDataShareUri()211 std::string WifiDataShareHelperUtils::GetLoactionDataShareUri()
212 {
213     std::vector<int> accountIds;
214 #ifdef HAS_ACCOUNT_PART
215     OHOS::AccountSA::OsAccountManager::QueryActiveOsAccountIds(accountIds);
216 #endif
217     std::string userId = "100";
218     if (!accountIds.empty()) {
219         userId = std::to_string(accountIds[0]);
220     }
221 
222     std::string uri = "datashare:///com.ohos.settingsdata/entry/settingsdata/USER_SETTINGSDATA_" +
223         userId + "?Proxy=true&key=location_enable";
224     return uri;
225 }
226 
ClearResources(std::shared_ptr<DataShare::DataShareHelper> operatrPtr,std::shared_ptr<DataShare::DataShareResultSet> result)227 void WifiDataShareHelperUtils::ClearResources(std::shared_ptr<DataShare::DataShareHelper> operatrPtr,
228     std::shared_ptr<DataShare::DataShareResultSet> result)
229 {
230     if (result != nullptr) {
231         result->Close();
232         result = nullptr;
233     }
234     if (operatrPtr != nullptr) {
235         operatrPtr->Release();
236         operatrPtr = nullptr;
237     }
238 }
239 }   // namespace Wifi
240 }   // namespace OHOS