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 default:
65 return IPCObjectStub::OnRemoteRequest(code, data, reply, option);
66 }
67 }
68
OnAdminEnabledInner(MessageParcel & data,MessageParcel & reply)69 void EnterpriseAdminStub::OnAdminEnabledInner(MessageParcel& data, MessageParcel& reply)
70 {
71 EDMLOGI("EnterpriseAdminStub::OnAdminEnabled");
72 OnAdminEnabled();
73 }
74
OnAdminDisabledInner(MessageParcel & data,MessageParcel & reply)75 void EnterpriseAdminStub::OnAdminDisabledInner(MessageParcel& data, MessageParcel& reply)
76 {
77 EDMLOGI("EnterpriseAdminStub::OnAdminDisabled");
78 OnAdminDisabled();
79 }
80
OnBundleAddedInner(MessageParcel & data,MessageParcel & reply)81 void EnterpriseAdminStub::OnBundleAddedInner(MessageParcel& data, MessageParcel& reply)
82 {
83 EDMLOGI("EnterpriseAdminStub::OnBundleAdded");
84 std::string bundleName = data.ReadString();
85 int32_t accountId = data.ReadInt32();
86 OnBundleAdded(bundleName, accountId);
87 }
88
OnBundleRemovedInner(MessageParcel & data,MessageParcel & reply)89 void EnterpriseAdminStub::OnBundleRemovedInner(MessageParcel& data, MessageParcel& reply)
90 {
91 EDMLOGI("EnterpriseAdminStub::OnBundleRemoved");
92 std::string bundleName = data.ReadString();
93 int32_t accountId = data.ReadInt32();
94 OnBundleRemoved(bundleName, accountId);
95 }
96
OnAppStartInner(MessageParcel & data,MessageParcel & reply)97 void EnterpriseAdminStub::OnAppStartInner(MessageParcel& data, MessageParcel& reply)
98 {
99 EDMLOGI("EnterpriseAdminStub::OnAppStart");
100 std::string bundleName = data.ReadString();
101 OnAppStart(bundleName);
102 }
103
OnAppStopInner(MessageParcel & data,MessageParcel & reply)104 void EnterpriseAdminStub::OnAppStopInner(MessageParcel& data, MessageParcel& reply)
105 {
106 EDMLOGI("EnterpriseAdminStub::OnAppStop");
107 std::string bundleName = data.ReadString();
108 OnAppStop(bundleName);
109 }
110
OnSystemUpdateInner(MessageParcel & data,MessageParcel & reply)111 void EnterpriseAdminStub::OnSystemUpdateInner(MessageParcel& data, MessageParcel& reply)
112 {
113 EDMLOGI("EnterpriseAdminStub::OnSystemUpdate");
114 UpdateInfo updateInfo;
115 updateInfo.version = data.ReadString();
116 updateInfo.firstReceivedTime = data.ReadInt64();
117 updateInfo.packageType = data.ReadString();
118 OnSystemUpdate(updateInfo);
119 }
120
OnRemoteRequest(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)121 int32_t EnterpriseAdminStub::OnRemoteRequest(uint32_t code, MessageParcel& data, MessageParcel& reply,
122 MessageOption &option)
123 {
124 auto remoteDescriptor = data.ReadInterfaceToken();
125 EDMLOGI("EnterpriseAdminStub code %{public}u", code);
126 if (GetDescriptor() != remoteDescriptor) {
127 EDMLOGE("EnterpriseAdminStub::OnRemoteRequest failed, descriptor is not matched!");
128 return ERR_INVALID_STATE;
129 }
130
131 Security::AccessToken::AccessTokenID callerToken = IPCSkeleton::GetCallingTokenID();
132 Security::AccessToken::ATokenTypeEnum tokenType =
133 Security::AccessToken::AccessTokenKit::GetTokenTypeFlag(callerToken);
134 if (tokenType != Security::AccessToken::ATokenTypeEnum::TOKEN_NATIVE) {
135 EDMLOGE("EnterpriseAdminStub::OnRemoteRequest failed, caller tokenType is not matched");
136 return ERR_INVALID_STATE;
137 }
138 int uid = GetCallingUid();
139 if (uid != EDM_UID) {
140 EDMLOGE("EnterpriseAdminStub::OnRemoteRequest failed, caller uid is not matched");
141 return ERR_INVALID_STATE;
142 }
143 return CallFuncByCode(code, data, reply, option);
144 }
145 } // namespace EDM
146 } // namespace OHOS
147