1 
2 /*
3 * Copyright (c) 2024 Huawei Device Co., Ltd.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *     http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16 
17 #include "media_permission_manager.h"
18 
19 #include <sys/types.h>
20 #include <vector>
21 
22 #include "ability_manager_errors.h"
23 #include "hilog_tag_wrapper.h"
24 #include "hitrace_meter.h"
25 #include "in_process_call_wrapper.h"
26 #include "iservice_registry.h"
27 #include "system_ability_definition.h"
28 #include "uri_permission_utils.h"
29 #include "want.h"
30 
31 namespace OHOS {
32 namespace AAFwk {
33 
GetInstance()34 MediaPermissionManager& MediaPermissionManager::GetInstance()
35 {
36     static MediaPermissionManager mediaPermissionManager;
37     return mediaPermissionManager;
38 }
39 
MediaPermissionManager()40 MediaPermissionManager::MediaPermissionManager()
41 {
42     TAG_LOGI(AAFwkTag::URIPERMMGR, "MediaPermissionManager init.");
43     mediaLibraryManager_ = GetMediaLibraryManager();
44 }
45 
GetMediaLibraryManager()46 Media::MediaLibraryManager *MediaPermissionManager::GetMediaLibraryManager()
47 {
48     TAG_LOGI(AAFwkTag::URIPERMMGR, "GetMediaLibraryManager.");
49     std::lock_guard<std::mutex> lock(mutex_);
50     if (mediaLibraryManager_) {
51         return mediaLibraryManager_;
52     }
53     mediaLibraryManager_ = Media::MediaLibraryManager::GetMediaLibraryManager();
54     if (mediaLibraryManager_ == nullptr) {
55         TAG_LOGE(AAFwkTag::URIPERMMGR, "GetMediaLibraryManager failed.");
56         return mediaLibraryManager_;
57     }
58     mediaLibraryManager_->InitMediaLibraryManager();
59     TAG_LOGI(AAFwkTag::URIPERMMGR, "InitMediaLibraryManager success!");
60     return mediaLibraryManager_;
61 }
62 
CheckUriPermission(const std::vector<Uri> & uriVec,uint32_t callerTokenId,uint32_t flag)63 std::vector<bool> MediaPermissionManager::CheckUriPermission(const std::vector<Uri> &uriVec,
64     uint32_t callerTokenId, uint32_t flag)
65 {
66     HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
67     std::vector<std::string> uriStrVec;
68     std::vector<bool> results = std::vector<bool>(uriVec.size(), false);
69     for (auto &uri: uriVec) {
70         uriStrVec.emplace_back(uri.ToString());
71     }
72     flag &= (Want::FLAG_AUTH_READ_URI_PERMISSION | Want::FLAG_AUTH_WRITE_URI_PERMISSION);
73     std::string bundleName = "";
74     if (!UPMSUtils::GetBundleNameByTokenId(callerTokenId, bundleName)) {
75         TAG_LOGW(AAFwkTag::URIPERMMGR, "Get bundle name failed.");
76         return results;
77     }
78     std::string appId = "";
79     if (UPMSUtils::GetAppIdByBundleName(bundleName, appId) != ERR_OK) {
80         TAG_LOGW(AAFwkTag::URIPERMMGR, "Get appId by bundle failed.");
81         return results;
82     }
83     auto mediaLibraryManager = GetMediaLibraryManager();
84     if (mediaLibraryManager == nullptr) {
85         TAG_LOGE(AAFwkTag::URIPERMMGR, "GetMediaLibraryManager failed.");
86         return results;
87     }
88     TAG_LOGI(AAFwkTag::URIPERMMGR, "CheckPhotoUriPermission start.");
89     auto ret = IN_PROCESS_CALL(mediaLibraryManager->CheckPhotoUriPermission(callerTokenId, appId, uriStrVec,
90         results, flag));
91     TAG_LOGI(AAFwkTag::URIPERMMGR, "CheckPhotoUriPermission finished.");
92     if (ret != ERR_OK) {
93         TAG_LOGE(AAFwkTag::URIPERMMGR, "Check photo uri permission failed, ret is %{pubic}d", ret);
94         results = std::vector<bool>(uriStrVec.size(), false);
95         return results;
96     }
97     if (results.size() != uriStrVec.size()) {
98         TAG_LOGE(AAFwkTag::URIPERMMGR, "size of results is unexpected: %{public}zu", results.size());
99         results = std::vector<bool>(uriStrVec.size(), false);
100         return results;
101     }
102     return results;
103 }
104 
105 } // OHOS
106 } // AAFwk