1 /*
2 * Copyright (c) 2023-2024 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 "start_ability_sandbox_savefile.h"
17
18 #include "hitrace_meter.h"
19 #include "ability_util.h"
20 #include "ability_manager_service.h"
21 #include "display_manager.h"
22 #include "display_util.h"
23
24 namespace OHOS {
25 namespace AAFwk {
26 namespace {
27 #ifdef WITH_DLP
28 const std::string DLP_BUNDLE_NAME = "com.ohos.dlpmanager";
29 const std::string DLP_ABILITY_NAME = "SaveAsAbility";
30 #endif // WITH_DLP
31
32 class EmptyConnection : public IRemoteStub<IAbilityConnection> {
33 public:
OnAbilityConnectDone(const AppExecFwk::ElementName & element,const sptr<IRemoteObject> & remoteObject,int resultCode)34 void OnAbilityConnectDone(const AppExecFwk::ElementName &element,
35 const sptr<IRemoteObject> &remoteObject, int resultCode) override
36 {
37 TAG_LOGD(AAFwkTag::ABILITYMGR, "OnAbilityConnectDone");
38 }
OnAbilityDisconnectDone(const AppExecFwk::ElementName & element,int resultCode)39 void OnAbilityDisconnectDone(const AppExecFwk::ElementName &element, int resultCode) override
40 {
41 TAG_LOGD(AAFwkTag::ABILITYMGR, "OnAbilityDisconnectDone");
42 }
43 };
44 }
45 const std::string StartAbilitySandboxSavefile::handlerName_ = "start_ability_snadbox_savefile";
46
MatchStartRequest(StartAbilityParams & params)47 bool StartAbilitySandboxSavefile::MatchStartRequest(StartAbilityParams ¶ms)
48 {
49 if (params.IsCallerSandboxApp() && params.want.GetAction() == "ohos.want.action.CREATE_FILE" &&
50 params.want.GetStringParam("startMode") == "save") {
51 return true;
52 }
53
54 auto element = params.want.GetElement();
55 #ifdef WITH_DLP
56 if (element.GetBundleName() == DLP_BUNDLE_NAME && element.GetAbilityName() == DLP_ABILITY_NAME &&
57 !ContainRecord(params.requestCode)) {
58 return true;
59 }
60 #endif // WITH_DLP
61 return false;
62 }
63
HandleStartRequest(StartAbilityParams & params)64 int StartAbilitySandboxSavefile::HandleStartRequest(StartAbilityParams ¶ms)
65 {
66 HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
67 TAG_LOGD(AAFwkTag::ABILITYMGR, "called");
68 auto callerRecord = params.GetCallerRecord();
69 if (!callerRecord) {
70 TAG_LOGE(AAFwkTag::ABILITYMGR, "null caller shouldn't happen");
71 return CHECK_PERMISSION_FAILED;
72 }
73
74 #ifdef WITH_DLP
75 if (!params.SandboxExternalAuth()) {
76 TAG_LOGW(AAFwkTag::ABILITYMGR, "sandbox external auth failed");
77 return CHECK_PERMISSION_FAILED;
78 }
79 #endif // WITH_DLP
80
81 auto reqCode = PushRecord(params.requestCode, callerRecord);
82 auto &want = params.want;
83 #ifdef WITH_DLP
84 want.SetElementName(DLP_BUNDLE_NAME, DLP_ABILITY_NAME);
85 #endif // WITH_DLP
86 want.SetParam("requestCode", reqCode);
87 want.SetParam("startMode", std::string("save_redirect"));
88
89 return StartAbility(params, reqCode);
90 }
91
StartAbility(StartAbilityParams & params,int requestCode)92 int StartAbilitySandboxSavefile::StartAbility(StartAbilityParams ¶ms, int requestCode)
93 {
94 HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
95 AbilityRequest abilityRequest;
96 abilityRequest.callType = AbilityCallType::CALL_REQUEST_TYPE;
97 abilityRequest.callerUid = IPCSkeleton::GetCallingUid();
98 abilityRequest.callerToken = params.callerToken;
99 abilityRequest.startSetting = nullptr;
100 abilityRequest.want = params.want;
101 abilityRequest.connect = sptr<IAbilityConnection>(new EmptyConnection());
102
103 auto abilityMs = DelayedSingleton<AbilityManagerService>::GetInstance();
104 auto ret = abilityMs->GenerateAbilityRequest(params.want, requestCode,
105 abilityRequest, params.callerToken, params.GetValidUserId());
106 if (ret != ERR_OK) {
107 TAG_LOGE(AAFwkTag::ABILITYMGR, "generate ability request error");
108 return ret;
109 }
110
111 if (params.startOptions) {
112 if (params.startOptions->GetDisplayID() == 0) {
113 abilityRequest.want.SetParam(Want::PARAM_RESV_DISPLAY_ID,
114 DisplayUtil::GetDefaultDisplayId());
115 } else {
116 abilityRequest.want.SetParam(Want::PARAM_RESV_DISPLAY_ID, params.startOptions->GetDisplayID());
117 }
118 abilityRequest.want.SetParam(Want::PARAM_RESV_WINDOW_MODE, params.startOptions->GetWindowMode());
119 }
120
121 return abilityMs->StartAbilityJust(abilityRequest, params.GetValidUserId());
122 }
123
GetHandlerName()124 std::string StartAbilitySandboxSavefile::GetHandlerName()
125 {
126 return StartAbilitySandboxSavefile::handlerName_;
127 }
128
PushRecord(int reqCode,const std::shared_ptr<AbilityRecord> & caller)129 int StartAbilitySandboxSavefile::PushRecord(int reqCode, const std::shared_ptr<AbilityRecord> &caller)
130 {
131 std::lock_guard guard(recordsMutex_);
132 requestCode_++;
133 if (requestCode_ >= INT_MAX) {
134 requestCode_ = 0;
135 }
136
137 auto it = fileSavingRecords_.find(requestCode_);
138 if (it != fileSavingRecords_.end()) {
139 TAG_LOGE(AAFwkTag::ABILITYMGR, "repeated request code");
140 fileSavingRecords_.erase(it);
141 }
142
143 SaveFileRecord record{reqCode, caller};
144 fileSavingRecords_.emplace(requestCode_, record);
145 return requestCode_;
146 }
147
ContainRecord(int reqCode)148 bool StartAbilitySandboxSavefile::ContainRecord(int reqCode)
149 {
150 std::lock_guard guard(recordsMutex_);
151 return fileSavingRecords_.count(reqCode) > 0;
152 }
153
HandleResult(const Want & want,int resultCode,int requestCode)154 void StartAbilitySandboxSavefile::HandleResult(const Want &want, int resultCode, int requestCode)
155 {
156 std::shared_ptr<AbilityRecord> callerRecord;
157 int originReqCode = -1;
158 {
159 std::lock_guard guard(recordsMutex_);
160 auto it = fileSavingRecords_.find(requestCode);
161 if (it != fileSavingRecords_.end()) {
162 callerRecord = it->second.caller.lock();
163 originReqCode = it->second.originReqCode;
164 fileSavingRecords_.erase(it);
165 }
166 }
167 if (!callerRecord) {
168 TAG_LOGE(AAFwkTag::ABILITYMGR, "request code not found: %{public}d.", requestCode);
169 return;
170 }
171 callerRecord->SendSandboxSavefileResult(want, resultCode, originReqCode);
172 }
173 } // namespace AAFwk
174 } // namespace OHOS