1 /*
2  * Copyright (c) 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 "form_data_proxy_mgr.h"
17 
18 #include "fms_log_wrapper.h"
19 #include "form_bms_helper.h"
20 #include "form_data_mgr.h"
21 #include "form_data_proxy_record.h"
22 #include "form_util.h"
23 #include "form_mgr_errors.h"
24 
25 namespace OHOS {
26 namespace AppExecFwk {
FormDataProxyMgr()27 FormDataProxyMgr::FormDataProxyMgr()
28 {}
29 
~FormDataProxyMgr()30 FormDataProxyMgr::~FormDataProxyMgr()
31 {}
32 
SubscribeFormData(int64_t formId, const std::vector<FormDataProxy> &formDataProxies, const AAFwk::Want &want)33 ErrCode FormDataProxyMgr::SubscribeFormData(int64_t formId, const std::vector<FormDataProxy> &formDataProxies,
34     const AAFwk::Want &want)
35 {
36     HILOG_DEBUG("subscribe form data. formId:%{public}s, proxy data size:%{public}zu",
37         std::to_string(formId).c_str(), formDataProxies.size());
38     if (formDataProxies.empty()) {
39         HILOG_DEBUG("empty formDataProxies");
40         return ERR_OK;
41     }
42 
43     FormRecord formRecord;
44     if (!FormDataMgr::GetInstance().GetFormRecord(formId, formRecord)) {
45         HILOG_ERROR("not exist such form:%{public}" PRId64 ".", formId);
46         return ERR_APPEXECFWK_FORM_NOT_EXIST_ID;
47     }
48 
49     if (!formRecord.isDataProxy) {
50         HILOG_DEBUG("not data proxy form, formId:%{public}" PRId64 ".", formId);
51         return ERR_OK;
52     }
53 
54     std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
55     auto search = formDataProxyRecordMap_.find(formId);
56     if (search != formDataProxyRecordMap_.end()) {
57         if (search->second != nullptr) {
58             HILOG_DEBUG("the form has already subscribed, formId:%{public}s", std::to_string(formId).c_str());
59             search->second->UnsubscribeFormData();
60         }
61     }
62 
63     ApplicationInfo appInfo;
64     if (FormBmsHelper::GetInstance().GetApplicationInfo(formRecord.bundleName, FormUtil::GetCurrentAccountId(),
65         appInfo) != ERR_OK) {
66         HILOG_ERROR("get app info failed");
67         return ERR_APPEXECFWK_FORM_COMMON_CODE;
68     }
69     std::shared_ptr<FormDataProxyRecord> formDataProxyRecord = std::make_shared<FormDataProxyRecord>(formId,
70         formRecord.bundleName, formRecord.uiSyntax, appInfo.accessTokenId, formRecord.uid);
71     formDataProxyRecord->SetWant(want);
72     auto ret = formDataProxyRecord->SubscribeFormData(formDataProxies);
73     if (ret != ERR_OK) {
74         HILOG_ERROR("SubscribeFormData failed");
75         return ret;
76     }
77     formDataProxyRecordMap_[formId] = formDataProxyRecord;
78     return ERR_OK;
79 }
80 
UnsubscribeFormData(int64_t formId)81 ErrCode FormDataProxyMgr::UnsubscribeFormData(int64_t formId)
82 {
83     HILOG_DEBUG("unsubscribe form data. formId:%{public}s", std::to_string(formId).c_str());
84     std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
85     auto search = formDataProxyRecordMap_.find(formId);
86     if (search != formDataProxyRecordMap_.end()) {
87         if (search->second != nullptr) {
88             search->second->UnsubscribeFormData();
89         }
90         formDataProxyRecordMap_.erase(formId);
91     }
92 
93     return ERR_OK;
94 }
95 
UpdateSubscribeFormData(int64_t formId, const std::vector<FormDataProxy> &formDataProxies)96 void FormDataProxyMgr::UpdateSubscribeFormData(int64_t formId, const std::vector<FormDataProxy> &formDataProxies)
97 {
98     HILOG_DEBUG("update subscribe form data. formId:%{public}s", std::to_string(formId).c_str());
99     std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
100     auto search = formDataProxyRecordMap_.find(formId);
101     if (search != formDataProxyRecordMap_.end()) {
102         if (search->second != nullptr) {
103             search->second->UpdateSubscribeFormData(formDataProxies);
104         }
105     }
106 }
107 
ProduceFormDataProxies(int64_t formId, const std::vector<FormDataProxy> &formDataProxies)108 void FormDataProxyMgr::ProduceFormDataProxies(int64_t formId, const std::vector<FormDataProxy> &formDataProxies)
109 {
110     std::lock_guard<std::mutex> lock(formDataProxiesMutex_);
111     formDataProxiesMap_[formId] = formDataProxies;
112 }
113 
ConsumeFormDataProxies(int64_t formId, std::vector<FormDataProxy> &formDataProxies)114 bool FormDataProxyMgr::ConsumeFormDataProxies(int64_t formId, std::vector<FormDataProxy> &formDataProxies)
115 {
116     std::lock_guard<std::mutex> lock(formDataProxiesMutex_);
117     auto search = formDataProxiesMap_.find(formId);
118     if (search == formDataProxiesMap_.end()) {
119         HILOG_DEBUG("no form data proxies, formId:%{public}s", std::to_string(formId).c_str());
120         return false;
121     }
122     formDataProxies = search->second;
123     formDataProxiesMap_.erase(search);
124     return true;
125 }
126 
EnableSubscribeFormData(const std::vector<int64_t> &formIds)127 void FormDataProxyMgr::EnableSubscribeFormData(const std::vector<int64_t> &formIds)
128 {
129     std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
130     for (const auto &formId : formIds) {
131         auto search = formDataProxyRecordMap_.find(formId);
132         if (search != formDataProxyRecordMap_.end()) {
133             if (search->second != nullptr) {
134                 search->second->EnableSubscribeFormData();
135             }
136         }
137     }
138 }
139 
DisableSubscribeFormData(const std::vector<int64_t> &formIds)140 void FormDataProxyMgr::DisableSubscribeFormData(const std::vector<int64_t> &formIds)
141 {
142     std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
143     for (const auto &formId : formIds) {
144         auto search = formDataProxyRecordMap_.find(formId);
145         if (search != formDataProxyRecordMap_.end()) {
146             if (search->second != nullptr) {
147                 search->second->DisableSubscribeFormData();
148                 HILOG_INFO("formId:%{public}s", std::to_string(formId).c_str());
149             }
150         }
151     }
152 }
153 
RetryFailureSubscribes()154 void FormDataProxyMgr::RetryFailureSubscribes()
155 {
156     std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
157     for (auto& record : formDataProxyRecordMap_) {
158         record.second->RetryFailureSubscribes();
159     }
160 }
161 
GetFormSubscribeInfo( const int64_t formId, std::vector<std::string> &subscribedKeys, int32_t &count)162 void FormDataProxyMgr::GetFormSubscribeInfo(
163     const int64_t formId, std::vector<std::string> &subscribedKeys, int32_t &count)
164 {
165     auto search = formDataProxyRecordMap_.find(formId);
166     if (search != formDataProxyRecordMap_.end()) {
167         if (search->second != nullptr) {
168             search->second->GetFormSubscribeInfo(subscribedKeys, count);
169         }
170     }
171 }
172 } // namespace AppExecFwk
173 } // namespace OHOS
174