• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 "enterprise_admin_stub.h"
17 
18 #include <accesstoken_kit.h>
19 #include <ipc_skeleton.h>
20 
21 #include "edm_log.h"
22 
23 namespace OHOS {
24 namespace EDM {
25 
26 constexpr const int EDM_UID = 3057;
27 
EnterpriseAdminStub()28 EnterpriseAdminStub::EnterpriseAdminStub()
29 {
30     EDMLOGD("EnterpriseAdminStub()");
31 }
32 
~EnterpriseAdminStub()33 EnterpriseAdminStub::~EnterpriseAdminStub()
34 {
35     EDMLOGD("~EnterpriseAdminStub()");
36 }
37 
38 
CallFuncByCode(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)39 int32_t EnterpriseAdminStub::CallFuncByCode(uint32_t code, MessageParcel& data, MessageParcel& reply,
40     MessageOption &option)
41 {
42     switch (code) {
43         case COMMAND_ON_ADMIN_ENABLED:
44             OnAdminEnabledInner(data, reply);
45             return ERR_NONE;
46         case COMMAND_ON_ADMIN_DISABLED:
47             OnAdminDisabledInner(data, reply);
48             return ERR_NONE;
49         case COMMAND_ON_BUNDLE_ADDED:
50             OnBundleAddedInner(data, reply);
51             return ERR_NONE;
52         case COMMAND_ON_BUNDLE_REMOVED:
53             OnBundleRemovedInner(data, reply);
54             return ERR_NONE;
55         case COMMAND_ON_APP_START:
56             OnAppStartInner(data, reply);
57             return ERR_NONE;
58         case COMMAND_ON_APP_STOP:
59             OnAppStopInner(data, reply);
60             return ERR_NONE;
61         case COMMAND_ON_SYSTEM_UPDATE:
62             OnSystemUpdateInner(data, reply);
63             return ERR_NONE;
64         case COMMAND_ON_ACCOUNT_ADDED:
65             OnAccountAddedInner(data, reply);
66             return ERR_NONE;
67         case COMMAND_ON_ACCOUNT_SWITCHED:
68             OnAccountSwitchedInner(data, reply);
69             return ERR_NONE;
70         case COMMAND_ON_ACCOUNT_REMOVED:
71             OnAccountRemovedInner(data, reply);
72             return ERR_NONE;
73         case COMMAND_ON_KIOSK_MODE_ENTERING:
74             OnKioskModeEnteringInner(data, reply);
75             return ERR_NONE;
76         case COMMAND_ON_KIOSK_MODE_EXITING:
77             OnKioskModeExitingInner(data, reply);
78             return ERR_NONE;
79         default:
80             return IPCObjectStub::OnRemoteRequest(code, data, reply, option);
81     }
82 }
83 
OnAdminEnabledInner(MessageParcel & data,MessageParcel & reply)84 void EnterpriseAdminStub::OnAdminEnabledInner(MessageParcel& data, MessageParcel& reply)
85 {
86     EDMLOGI("EnterpriseAdminStub::OnAdminEnabled");
87     OnAdminEnabled();
88 }
89 
OnAdminDisabledInner(MessageParcel & data,MessageParcel & reply)90 void EnterpriseAdminStub::OnAdminDisabledInner(MessageParcel& data, MessageParcel& reply)
91 {
92     EDMLOGI("EnterpriseAdminStub::OnAdminDisabled");
93     OnAdminDisabled();
94 }
95 
OnBundleAddedInner(MessageParcel & data,MessageParcel & reply)96 void EnterpriseAdminStub::OnBundleAddedInner(MessageParcel& data, MessageParcel& reply)
97 {
98     EDMLOGI("EnterpriseAdminStub::OnBundleAdded");
99     std::string bundleName = data.ReadString();
100     int32_t accountId = data.ReadInt32();
101     OnBundleAdded(bundleName, accountId);
102 }
103 
OnBundleRemovedInner(MessageParcel & data,MessageParcel & reply)104 void EnterpriseAdminStub::OnBundleRemovedInner(MessageParcel& data, MessageParcel& reply)
105 {
106     EDMLOGI("EnterpriseAdminStub::OnBundleRemoved");
107     std::string bundleName = data.ReadString();
108     int32_t accountId = data.ReadInt32();
109     OnBundleRemoved(bundleName, accountId);
110 }
111 
OnAppStartInner(MessageParcel & data,MessageParcel & reply)112 void EnterpriseAdminStub::OnAppStartInner(MessageParcel& data, MessageParcel& reply)
113 {
114     EDMLOGI("EnterpriseAdminStub::OnAppStart");
115     std::string bundleName = data.ReadString();
116     OnAppStart(bundleName);
117 }
118 
OnAppStopInner(MessageParcel & data,MessageParcel & reply)119 void EnterpriseAdminStub::OnAppStopInner(MessageParcel& data, MessageParcel& reply)
120 {
121     EDMLOGI("EnterpriseAdminStub::OnAppStop");
122     std::string bundleName = data.ReadString();
123     OnAppStop(bundleName);
124 }
125 
OnSystemUpdateInner(MessageParcel & data,MessageParcel & reply)126 void EnterpriseAdminStub::OnSystemUpdateInner(MessageParcel& data, MessageParcel& reply)
127 {
128     EDMLOGI("EnterpriseAdminStub::OnSystemUpdate");
129     UpdateInfo updateInfo;
130     updateInfo.version = data.ReadString();
131     updateInfo.firstReceivedTime = data.ReadInt64();
132     updateInfo.packageType = data.ReadString();
133     OnSystemUpdate(updateInfo);
134 }
135 
OnAccountAddedInner(MessageParcel & data,MessageParcel & reply)136 void EnterpriseAdminStub::OnAccountAddedInner(MessageParcel& data, MessageParcel& reply)
137 {
138     EDMLOGI("EnterpriseAdminStub::OnAccountAdded");
139     int32_t accountId = data.ReadInt32();
140     OnAccountAdded(accountId);
141 }
142 
OnAccountSwitchedInner(MessageParcel & data,MessageParcel & reply)143 void EnterpriseAdminStub::OnAccountSwitchedInner(MessageParcel& data, MessageParcel& reply)
144 {
145     EDMLOGI("EnterpriseAdminStub::OnAccountSwitched");
146     int32_t accountId = data.ReadInt32();
147     OnAccountSwitched(accountId);
148 }
149 
OnAccountRemovedInner(MessageParcel & data,MessageParcel & reply)150 void EnterpriseAdminStub::OnAccountRemovedInner(MessageParcel& data, MessageParcel& reply)
151 {
152     EDMLOGI("EnterpriseAdminStub::OnAccountRemoved");
153     int32_t accountId = data.ReadInt32();
154     OnAccountRemoved(accountId);
155 }
156 
OnKioskModeEnteringInner(MessageParcel & data,MessageParcel & reply)157 void EnterpriseAdminStub::OnKioskModeEnteringInner(MessageParcel& data, MessageParcel& reply)
158 {
159     EDMLOGI("EnterpriseAdminStub::OnKioskModeEnteringInner");
160     std::string bundleName = data.ReadString();
161     int32_t accountId = data.ReadInt32();
162     OnKioskModeEntering(bundleName, accountId);
163 }
164 
OnKioskModeExitingInner(MessageParcel & data,MessageParcel & reply)165 void EnterpriseAdminStub::OnKioskModeExitingInner(MessageParcel& data, MessageParcel& reply)
166 {
167     EDMLOGI("EnterpriseAdminStub::OnKioskModeExitingInner");
168     std::string bundleName = data.ReadString();
169     int32_t accountId = data.ReadInt32();
170     OnKioskModeExiting(bundleName, accountId);
171 }
172 
OnRemoteRequest(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)173 int32_t EnterpriseAdminStub::OnRemoteRequest(uint32_t code, MessageParcel& data, MessageParcel& reply,
174     MessageOption &option)
175 {
176     auto remoteDescriptor = data.ReadInterfaceToken();
177     EDMLOGI("EnterpriseAdminStub code %{public}u", code);
178     if (GetDescriptor() != remoteDescriptor) {
179         EDMLOGE("EnterpriseAdminStub::OnRemoteRequest failed, descriptor is not matched!");
180         return ERR_INVALID_STATE;
181     }
182 
183     Security::AccessToken::AccessTokenID callerToken = IPCSkeleton::GetCallingTokenID();
184     Security::AccessToken::ATokenTypeEnum tokenType =
185         Security::AccessToken::AccessTokenKit::GetTokenTypeFlag(callerToken);
186     if (tokenType != Security::AccessToken::ATokenTypeEnum::TOKEN_NATIVE) {
187         EDMLOGE("EnterpriseAdminStub::OnRemoteRequest failed, caller tokenType is not matched");
188         return ERR_INVALID_STATE;
189     }
190     int uid = GetCallingUid();
191     if (uid != EDM_UID) {
192         EDMLOGE("EnterpriseAdminStub::OnRemoteRequest failed, caller uid is not matched");
193         return ERR_INVALID_STATE;
194     }
195     return CallFuncByCode(code, data, reply, option);
196 }
197 } // namespace EDM
198 } // namespace OHOS
199