1 /*
2  * Copyright (C) 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 #ifndef MEDIALIBRARY_PERMISSION_UTILS_H
16 #define MEDIALIBRARY_PERMISSION_UTILS_H
17 
18 #include <array>
19 #include <list>
20 #include <string>
21 #include <unordered_map>
22 #include <vector>
23 #include <thread>
24 #include <mutex>
25 
26 #include "bundle_mgr_interface.h"
27 #include "userfile_manager_types.h"
28 #include "permission_used_type.h"
29 #include "privacy_kit.h"
30 #include "tokenid_kit.h"
31 
32 namespace OHOS {
33 namespace Media {
34 #define COMPILE_HIDDEN __attribute__ ((visibility ("hidden")))
35 const std::string PERMISSION_NAME_READ_MEDIA = "ohos.permission.READ_MEDIA";
36 const std::string PERMISSION_NAME_WRITE_MEDIA = "ohos.permission.WRITE_MEDIA";
37 const std::string PERMISSION_NAME_MEDIA_LOCATION = "ohos.permission.MEDIA_LOCATION";
38 const std::string PERM_READ_IMAGEVIDEO = "ohos.permission.READ_IMAGEVIDEO";
39 const std::string PERM_READ_AUDIO = "ohos.permission.READ_AUDIO";
40 const std::string PERM_READ_DOCUMENT = "ohos.permission.READ_DOCUMENT";
41 const std::string PERM_WRITE_IMAGEVIDEO = "ohos.permission.WRITE_IMAGEVIDEO";
42 const std::string PERM_WRITE_AUDIO = "ohos.permission.WRITE_AUDIO";
43 const std::string PERM_WRITE_DOCUMENT = "ohos.permission.WRITE_DOCUMENT";
44 const std::string PERM_MANAGE_PRIVATE_PHOTOS = "ohos.permission.MANAGE_PRIVATE_PHOTOS";
45 const std::string PERM_SHORT_TERM_WRITE_IMAGEVIDEO = "ohos.permission.SHORT_TERM_WRITE_IMAGEVIDEO";
46 const std::string E_POLICY = "E";
47 constexpr int THREE_HUNDERD_S = 300;
48 
49 enum SaveType {
50     SHORT_IMAGE_PERM = 0,
51 };
52 
53 const std::vector<std::string> WRITE_PERMS_V10 = {
54     PERM_WRITE_IMAGEVIDEO,
55     PERM_WRITE_AUDIO,
56     PERM_WRITE_DOCUMENT
57 };
58 
59 struct BundleInfo {
60     std::string bundleName;
61     std::string packageName;
62     std::string appId;
63 };
64 
65 class PermissionUtils {
66 public:
67     static bool CheckCallerPermission(const std::string &permission);
68     static bool CheckCallerPermission(const std::vector<std::string> &perms);
69     static bool CheckHasPermission(const std::vector<std::string> &perms);
70     static void GetClientBundle(const int uid, std::string &bundleName);
71     static void GetPackageName(const int uid, std::string &packageName);
72     static uint32_t GetTokenId();
73     static bool IsSystemApp();
74     static bool IsNativeSAApp();
75     static bool IsRootShell();
76     static bool IsHdcShell();
77     static bool CheckIsSystemAppByUid();
78     static std::string GetPackageNameByBundleName(const std::string &bundleName);
79     static std::string GetAppIdByBundleName(const std::string &bundleName);
80     static std::string GetAppIdByBundleName(const std::string &bundleName, int32_t uid);
81     static void ClearBundleInfoInCache();
82     static bool CheckPhotoCallerPermission(const std::vector<std::string> &perms);
83     static bool CheckPhotoCallerPermission(const std::string &permission);
84     static void CollectPermissionInfo(const std::string &permission, const bool permGranted,
85         const Security::AccessToken::PermissionUsedType type);
86     static bool SetEPolicy();
87 
88 private:
89     static sptr<AppExecFwk::IBundleMgr> GetSysBundleManager();
90     COMPILE_HIDDEN static sptr<AppExecFwk::IBundleMgr> bundleMgr_;
91     COMPILE_HIDDEN static std::mutex bundleMgrMutex_;
92     static void GetBundleNameFromCache(int uid, std::string &bundleName);
93     static void GetPackageNameFromCache(int uid, const std::string &bundleName, std::string &packageName);
94     static void GetAppIdFromCache(int uid, const std::string &bundleName, std::string &appId);
95     static void UpdateLatestBundleInfo(int uid, const BundleInfo &bundleInfo);
96     static void UpdateBundleNameInCache(int uid, const std::string &bundleName);
97     static void UpdatePackageNameInCache(int uid, const std::string &packageName);
98     static void UpdateAppIdInCache(int uid, const std::string &appId);
99 
100     static std::mutex uninstallMutex_;
101     static std::list<std::pair<int32_t, BundleInfo>> bundleInfoList_; // 用来快速获取使用频率最低的uid
102     static std::unordered_map<int32_t, std::list<std::pair<int32_t, BundleInfo>>::iterator> bundleInfoMap_;
103 };
104 }  // namespace Media
105 }  // namespace OHOS
106 #endif // MEDIALIBRARY_PERMISSION_UTILS_H
107