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 AddCallFuncMap();
31 EDMLOGD("EnterpriseAdminStub()");
32 }
33
~EnterpriseAdminStub()34 EnterpriseAdminStub::~EnterpriseAdminStub()
35 {
36 EDMLOGD("~EnterpriseAdminStub()");
37 }
38
AddCallFuncMap()39 void EnterpriseAdminStub::AddCallFuncMap()
40 {
41 memberFuncMap_[COMMAND_ON_ADMIN_ENABLED] = &EnterpriseAdminStub::OnAdminEnabledInner;
42 memberFuncMap_[COMMAND_ON_ADMIN_DISABLED] = &EnterpriseAdminStub::OnAdminDisabledInner;
43 memberFuncMap_[COMMAND_ON_BUNDLE_ADDED] = &EnterpriseAdminStub::OnBundleAddedInner;
44 memberFuncMap_[COMMAND_ON_BUNDLE_REMOVED] = &EnterpriseAdminStub::OnBundleRemovedInner;
45 memberFuncMap_[COMMAND_ON_APP_START] = &EnterpriseAdminStub::OnAppStartInner;
46 memberFuncMap_[COMMAND_ON_APP_STOP] = &EnterpriseAdminStub::OnAppStopInner;
47 }
48
OnAdminEnabledInner(MessageParcel & data,MessageParcel & reply)49 void EnterpriseAdminStub::OnAdminEnabledInner(MessageParcel& data, MessageParcel& reply)
50 {
51 EDMLOGI("EnterpriseAdminStub::OnAdminEnabled");
52 OnAdminEnabled();
53 }
54
OnAdminDisabledInner(MessageParcel & data,MessageParcel & reply)55 void EnterpriseAdminStub::OnAdminDisabledInner(MessageParcel& data, MessageParcel& reply)
56 {
57 EDMLOGI("EnterpriseAdminStub::OnAdminDisabled");
58 OnAdminDisabled();
59 }
60
OnBundleAddedInner(MessageParcel & data,MessageParcel & reply)61 void EnterpriseAdminStub::OnBundleAddedInner(MessageParcel& data, MessageParcel& reply)
62 {
63 EDMLOGI("EnterpriseAdminStub::OnBundleAdded");
64 std::string bundleName = data.ReadString();
65 OnBundleAdded(bundleName);
66 }
67
OnBundleRemovedInner(MessageParcel & data,MessageParcel & reply)68 void EnterpriseAdminStub::OnBundleRemovedInner(MessageParcel& data, MessageParcel& reply)
69 {
70 EDMLOGI("EnterpriseAdminStub::OnBundleRemoved");
71 std::string bundleName = data.ReadString();
72 OnBundleRemoved(bundleName);
73 }
74
OnAppStartInner(MessageParcel & data,MessageParcel & reply)75 void EnterpriseAdminStub::OnAppStartInner(MessageParcel& data, MessageParcel& reply)
76 {
77 EDMLOGI("EnterpriseAdminStub::OnAppStart");
78 std::string bundleName = data.ReadString();
79 OnAppStart(bundleName);
80 }
81
OnAppStopInner(MessageParcel & data,MessageParcel & reply)82 void EnterpriseAdminStub::OnAppStopInner(MessageParcel& data, MessageParcel& reply)
83 {
84 EDMLOGI("EnterpriseAdminStub::OnAppStop");
85 std::string bundleName = data.ReadString();
86 OnAppStop(bundleName);
87 }
88
OnRemoteRequest(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)89 int32_t EnterpriseAdminStub::OnRemoteRequest(uint32_t code, MessageParcel& data, MessageParcel& reply,
90 MessageOption &option)
91 {
92 auto remoteDescriptor = data.ReadInterfaceToken();
93 EDMLOGI("EnterpriseAdminStub code %{public}u", code);
94 if (GetDescriptor() != remoteDescriptor) {
95 EDMLOGE("EnterpriseAdminStub::OnRemoteRequest failed, descriptor is not matched!");
96 return ERR_INVALID_STATE;
97 }
98
99 Security::AccessToken::AccessTokenID callerToken = IPCSkeleton::GetCallingTokenID();
100 Security::AccessToken::ATokenTypeEnum tokenType =
101 Security::AccessToken::AccessTokenKit::GetTokenTypeFlag(callerToken);
102 if (tokenType != Security::AccessToken::ATokenTypeEnum::TOKEN_NATIVE) {
103 EDMLOGE("EnterpriseAdminStub::OnRemoteRequest failed, caller tokenType is not matched");
104 return ERR_INVALID_STATE;
105 }
106 int uid = GetCallingUid();
107 if (uid != EDM_UID) {
108 EDMLOGE("EnterpriseAdminStub::OnRemoteRequest failed, caller uid is not matched");
109 return ERR_INVALID_STATE;
110 }
111
112 auto func = memberFuncMap_.find(code);
113 if (func != memberFuncMap_.end()) {
114 auto memberFunc = func->second;
115 if (memberFunc != nullptr) {
116 (this->*memberFunc)(data, reply);
117 return ERR_NONE;
118 }
119 return IPCObjectStub::OnRemoteRequest(code, data, reply, option);
120 }
121
122 return ERR_TRANSACTION_FAILED;
123 }
124 } // namespace EDM
125 } // namespace OHOS
126