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