1 /*
2  * Copyright (c) 2022-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 "utils.h"
17 #include "parameters.h"
18 
19 namespace OHOS {
20 namespace EDM {
21 namespace TEST {
22 uint64_t Utils::selfTokenId_ = 0;
23 const std::string SET_EDM_SERVICE = "persist.edm.edm_enable";
24 const std::string EDM_MANAGE_DATETIME_PERMISSION = "ohos.permission.SET_TIME";
25 const std::string EDM_FACTORY_RESET_PERMISSION = "ohos.permission.FACTORY_RESET";
26 const std::string EDM_MANAGE_LOCAL_ACCOUNTS_PERMISSION = "ohos.permission.MANAGE_LOCAL_ACCOUNTS";
27 const std::string EDM_MANAGE_SECURE_SETTINGS_PERMISSION = "ohos.permission.MANAGE_SECURE_SETTINGS";
28 const std::string PERMISSION_ENTERPRISE_SUBSCRIBE_MANAGED_EVENT = "ohos.permission.ENTERPRISE_SUBSCRIBE_MANAGED_EVENT";
29 const std::string PERMISSION_MANAGE_ENTERPRISE_DEVICE_ADMIN = "ohos.permission.MANAGE_ENTERPRISE_DEVICE_ADMIN";
30 const std::string PERMISSION_SET_ENTERPRISE_INFO = "ohos.permission.SET_ENTERPRISE_INFO";
31 const std::string PERMISSION_ACCESS_CERT_MANAGER_INTERNAL = "ohos.permission.ACCESS_CERT_MANAGER_INTERNAL";
32 const std::string PERMISSION_ACCESS_CERT_MANAGER = "ohos.permission.ACCESS_CERT_MANAGER";
33 const std::string PERMISSION_ACCESS_BLUETOOTH = "ohos.permission.ACCESS_BLUETOOTH";
34 const std::string PERMISSION_MANAGE_BLUETOOTH = "ohos.permission.MANAGE_BLUETOOTH";
35 const std::string PERMISSION_NETSYS_INTERNAL = "ohos.permission.NETSYS_INTERNAL";
36 const std::string PERMISSION_INSTALL_ENTERPRISE_NORMAL_BUNDLE = "ohos.permission.INSTALL_ENTERPRISE_NORMAL_BUNDLE";
37 const std::string PERMISSION_INSTALL_ENTERPRISE_MDM_BUNDLE = "ohos.permission.INSTALL_ENTERPRISE_MDM_BUNDLE";
38 const std::string PERMISSION_INSTALL_SELF_BUNDLE = "ohos.permission.INSTALL_SELF_BUNDLE";
39 const std::string PERMISSION_UNINSTALL_BUNDLE = "ohos.permission.UNINSTALL_BUNDLE";
40 const std::string PERMISSION_RUNNING_STATE_OBSERVER = "ohos.permission.RUNNING_STATE_OBSERVER";
41 const std::string PERMISSION_ENTERPRISE_MANAGE_SECURITY = "ohos.permission.ENTERPRISE_MANAGE_SECURITY";
42 const std::string PERMISSION_ENTERPRISE_MANAGE_BLUETOOTH = "ohos.permission.ENTERPRISE_MANAGE_BLUETOOTH";
43 const std::string PERMISSION_ENTERPRISE_MANAGE_LOCATION = "ohos.permission.ENTERPRISE_MANAGE_LOCATION";
44 const std::string PERMISSION_ENTERPRISE_ACCESS_SCREEN_LOCK_INNER = "ohos.permission.ACCESS_SCREEN_LOCK_INNER";
45 const std::string PERMISSION_REBOOT = "ohos.permission.REBOOT";
46 const std::string PERMISSION_MANAGE_APP_BOOT_INTERNAL = "ohos.permission.MANAGE_APP_BOOT_INTERNAL";
47 const std::string PERMISSION_CONNECTIVITY_INTERNAL = "ohos.permission.CONNECTIVITY_INTERNAL";
48 const std::string PERMISSION_MANAGE_USB_CONFIG = "ohos.permission.MANAGE_USB_CONFIG";
49 const std::string PERMISSION_STORAGE_MANAGER = "ohos.permission.STORAGE_MANAGER";
50 const std::string PERMISSION_MOUNT_UNMOUNT_MANAGER = "ohos.permission.MOUNT_UNMOUNT_MANAGER";
51 const std::string PERMISSION_GET_BUNDLE_INFO = "ohos.permission.GET_BUNDLE_INFO";
52 const std::string PERSIST_BLUETOOTH_CONTROL = "persist.edm.prohibit_bluetooth";
53 const std::string PERMISSION_ACCESS_USER_TRUSTED_CERT = "ohos.permission.ACCESS_USER_TRUSTED_CERT";
54 const std::string PERMISSION_ACCESS_USER_AUTH_INTERNAL = "ohos.permission.ACCESS_USER_AUTH_INTERNAL";
55 const std::string PERMISSION_SET_WIFI_INFO = "ohos.permission.SET_WIFI_INFO";
56 const std::string PERMISSION_GET_WIFI_INFO = "ohos.permission.GET_WIFI_INFO";
57 const std::string PERMISSION_MANAGE_WIFI_CONNECTION = "ohos.permission.MANAGE_WIFI_CONNECTION";
58 const std::string PERMISSION_MANAGE_SETTINGS = "ohos.permission.MANAGE_SETTINGS";
59 const std::string PERMISSION_MICROPHONE_CONTROL = "ohos.permission.MICROPHONE_CONTROL";
60 const std::string PERMISSION_CAMERA = "ohos.permission.CAMERA";
61 const std::string PERMISSION_CAMERA_CONTROL = "ohos.permission.CAMERA_CONTROL";
ExecCmdSync(const std::string & cmd)62 void Utils::ExecCmdSync(const std::string &cmd)
63 {
64     FILE *fp = popen(cmd.c_str(), "r");
65     if (fp != nullptr) {
66         pclose(fp);
67         fp = nullptr;
68     }
69 }
70 
SetNativeTokenTypeAndPermissions(const char * permissions[],int size)71 void Utils::SetNativeTokenTypeAndPermissions(const char* permissions[], int size)
72 {
73     uint64_t tokenId;
74     NativeTokenInfoParams infoInstance = {
75         .dcapsNum = 0,
76         .permsNum = size,
77         .aclsNum = 0,
78         .dcaps = nullptr,
79         .perms = permissions,
80         .acls = nullptr,
81         .aplStr = "system_basic",
82     };
83 
84     infoInstance.processName = "EdmServicesUnitTest";
85     tokenId = GetAccessTokenId(&infoInstance);
86     SetSelfTokenID(tokenId);
87     seteuid(ROOT_UID);
88     OHOS::Security::AccessToken::AccessTokenKit::ReloadNativeTokenInfo();
89     seteuid(EDM_UID);
90 }
91 
SetEdmInitialEnv()92 void Utils::SetEdmInitialEnv()
93 {
94     selfTokenId_ = GetSelfTokenID();
95     const char* permissions[] = {
96         EDM_MANAGE_DATETIME_PERMISSION.c_str(),
97         EDM_FACTORY_RESET_PERMISSION.c_str(),
98         EDM_MANAGE_LOCAL_ACCOUNTS_PERMISSION.c_str(),
99         EDM_MANAGE_SECURE_SETTINGS_PERMISSION.c_str(),
100         PERMISSION_ENTERPRISE_SUBSCRIBE_MANAGED_EVENT.c_str(),
101         PERMISSION_MANAGE_ENTERPRISE_DEVICE_ADMIN.c_str(),
102         PERMISSION_SET_ENTERPRISE_INFO.c_str(),
103         PERMISSION_ACCESS_BLUETOOTH.c_str(),
104         PERMISSION_MANAGE_BLUETOOTH.c_str(),
105         PERMISSION_ACCESS_CERT_MANAGER_INTERNAL.c_str(),
106         PERMISSION_ACCESS_CERT_MANAGER.c_str(),
107         PERMISSION_NETSYS_INTERNAL.c_str(),
108         PERMISSION_INSTALL_ENTERPRISE_MDM_BUNDLE.c_str(),
109         PERMISSION_INSTALL_ENTERPRISE_NORMAL_BUNDLE.c_str(),
110         PERMISSION_INSTALL_SELF_BUNDLE.c_str(),
111         PERMISSION_UNINSTALL_BUNDLE.c_str(),
112         PERMISSION_RUNNING_STATE_OBSERVER.c_str(),
113         PERMISSION_ENTERPRISE_MANAGE_SECURITY.c_str(),
114         PERMISSION_ENTERPRISE_MANAGE_BLUETOOTH.c_str(),
115         PERMISSION_ENTERPRISE_MANAGE_LOCATION.c_str(),
116         PERMISSION_ENTERPRISE_ACCESS_SCREEN_LOCK_INNER.c_str(),
117         PERMISSION_REBOOT.c_str(),
118         PERMISSION_MANAGE_APP_BOOT_INTERNAL.c_str(),
119         PERMISSION_CONNECTIVITY_INTERNAL.c_str(),
120         PERMISSION_MANAGE_USB_CONFIG.c_str(),
121         PERMISSION_STORAGE_MANAGER.c_str(),
122         PERMISSION_MOUNT_UNMOUNT_MANAGER.c_str(),
123         PERMISSION_GET_BUNDLE_INFO.c_str(),
124         PERMISSION_ACCESS_USER_TRUSTED_CERT.c_str(),
125         PERMISSION_ACCESS_USER_AUTH_INTERNAL.c_str(),
126         PERMISSION_GET_WIFI_INFO.c_str(),
127         PERMISSION_SET_WIFI_INFO.c_str(),
128         PERMISSION_MANAGE_WIFI_CONNECTION.c_str(),
129         PERMISSION_MANAGE_SETTINGS.c_str(),
130         PERMISSION_MICROPHONE_CONTROL.c_str(),
131         PERMISSION_CAMERA.c_str(),
132         PERMISSION_CAMERA_CONTROL.c_str(),
133     };
134     Utils::SetNativeTokenTypeAndPermissions(permissions, sizeof(permissions) / sizeof(permissions[0]));
135     seteuid(Utils::EDM_UID);
136 }
137 
ResetTokenTypeAndUid()138 void Utils::ResetTokenTypeAndUid()
139 {
140     seteuid(Utils::ROOT_UID);
141     SetSelfTokenID(selfTokenId_);
142 }
143 
IsOriginalUTEnv()144 bool Utils::IsOriginalUTEnv()
145 {
146     return Utils::ROOT_UID == geteuid() && selfTokenId_ == GetSelfTokenID();
147 }
148 
GetEdmServiceState()149 bool Utils::GetEdmServiceState()
150 {
151     std::string edmParaValue = system::GetParameter("persist.edm.edm_enable", "false");
152     return edmParaValue == "true";
153 }
154 
SetEdmServiceEnable()155 void Utils::SetEdmServiceEnable()
156 {
157     system::SetParameter(SET_EDM_SERVICE, "true");
158 }
159 
SetEdmServiceDisable()160 void Utils::SetEdmServiceDisable()
161 {
162     system::SetParameter(SET_EDM_SERVICE, "false");
163 }
164 
SetBluetoothEnable()165 void Utils::SetBluetoothEnable()
166 {
167     system::SetParameter(PERSIST_BLUETOOTH_CONTROL, "false");
168 }
169 
SetBluetoothDisable()170 void Utils::SetBluetoothDisable()
171 {
172     system::SetParameter(PERSIST_BLUETOOTH_CONTROL, "true");
173 }
174 } // namespace TEST
175 } // namespace EDM
176 } // namespace OHOS
177