1 /*
2  * Copyright (c) 2021-2024 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 #ifndef OHOS_DISTRIBUTED_SCHED_STUB_H
17 #define OHOS_DISTRIBUTED_SCHED_STUB_H
18 
19 #include "distributed_sched_interface.h"
20 #include "iremote_stub.h"
21 #ifdef SUPPORT_DISTRIBUTED_MISSION_MANAGER
22 #include "mission/snapshot.h"
23 #endif
24 
25 namespace OHOS {
26 namespace DistributedSchedule {
27 class DistributedSchedStub : public IRemoteStub<IDistributedSched> {
28 public:
29     DistributedSchedStub();
30     ~DistributedSchedStub();
31     int32_t OnRemoteRequest(uint32_t code, MessageParcel& data,
32         MessageParcel& reply, MessageOption& option) override;
33 
34 private:
35     int32_t StartRemoteAbilityInner(MessageParcel& data, MessageParcel& reply);
36     int32_t StartAbilityFromRemoteInner(MessageParcel& data, MessageParcel& reply);
37     int32_t SendResultFromRemoteInner(MessageParcel& data, MessageParcel& reply);
38     bool IsRemoteInstall(const std::string &networkId, const std::string &bundleName);
39     int32_t ContinueMissionInner(MessageParcel& data, MessageParcel& reply);
40     int32_t ContinueMissionOfBundleNameInner(MessageParcel& data, MessageParcel& reply);
41     int32_t StartContinuationInner(MessageParcel& data, MessageParcel& reply);
42     int32_t NotifyCompleteContinuationInner(MessageParcel& data, MessageParcel& reply);
43     int32_t NotifyContinuationResultFromRemoteInner(MessageParcel& data, MessageParcel& reply);
44     int32_t NotifyDSchedEventResultFromRemoteInner(MessageParcel& data, MessageParcel& reply);
45     int32_t ConnectRemoteAbilityInner(MessageParcel& data, MessageParcel& reply);
46     int32_t DisconnectRemoteAbilityInner(MessageParcel& data, MessageParcel& reply);
47     int32_t ConnectAbilityFromRemoteInner(MessageParcel& data, MessageParcel& reply);
48     int32_t DisconnectAbilityFromRemoteInner(MessageParcel& data, MessageParcel& reply);
49     int32_t NotifyProcessDiedFromRemoteInner(MessageParcel& data, MessageParcel& reply);
50 #ifdef SUPPORT_DISTRIBUTED_MISSION_MANAGER
51     int32_t GetMissionInfosInner(MessageParcel& data, MessageParcel& reply);
52     int32_t GetRemoteMissionSnapshotInfoInner(MessageParcel& data, MessageParcel& reply);
53     int32_t RegisterMissionListenerInner(MessageParcel& data, MessageParcel& reply);
54     int32_t RegisterDSchedEventListenerInner(MessageParcel& data, MessageParcel& reply);
55     int32_t UnRegisterDSchedEventListenerInner(MessageParcel& data, MessageParcel& reply);
56     int32_t GetContinueInfoInner(MessageParcel& data, MessageParcel& reply);
57     int32_t GetDSchedEventInfoInner(MessageParcel& data, MessageParcel& reply);
58     int32_t RegisterOnListenerInner(MessageParcel& data, MessageParcel& reply);
59     int32_t RegisterOffListenerInner(MessageParcel& data, MessageParcel& reply);
60     int32_t UnRegisterMissionListenerInner(MessageParcel& data, MessageParcel& reply);
61     int32_t StartSyncRemoteMissionsInner(MessageParcel& data, MessageParcel& reply);
62     int32_t StartSyncMissionsFromRemoteInner(MessageParcel& data, MessageParcel& reply);
63     int32_t StopSyncRemoteMissionsInner(MessageParcel& data, MessageParcel& reply);
64     int32_t StopSyncMissionsFromRemoteInner(MessageParcel& data, MessageParcel& reply);
65     int32_t NotifyMissionsChangedFromRemoteInner(MessageParcel& data, MessageParcel& reply);
66     int32_t SetMissionContinueStateInner(MessageParcel& data, MessageParcel& reply);
67 #endif
68     int32_t StartRemoteAbilityByCallInner(MessageParcel& data, MessageParcel& reply);
69     int32_t ReleaseRemoteAbilityInner(MessageParcel& data, MessageParcel& reply);
70     int32_t StartAbilityByCallFromRemoteInner(MessageParcel& data, MessageParcel& reply);
71     int32_t ReleaseAbilityFromRemoteInner(MessageParcel& data, MessageParcel& reply);
72     int32_t StartRemoteFreeInstallInner(MessageParcel& data, MessageParcel& reply);
73     int32_t CreateJsonObject(std::string& extraInfo, CallerInfo& callerInfo, AccountInfo& accountInfo);
74     int32_t StartFreeInstallFromRemoteInner(MessageParcel& data, MessageParcel& reply);
75     int32_t ReadDataForFreeInstall(MessageParcel& data,
76         CallerInfo &callerInfo, AccountInfo &accountInfo, int64_t &taskId);
77     int32_t NotifyCompleteFreeInstallFromRemoteInner(MessageParcel& data, MessageParcel& reply);
78 #ifdef SUPPORT_DISTRIBUTED_FORM_SHARE
79     int32_t StartRemoteShareFormInner(MessageParcel& data, MessageParcel& reply);
80     int32_t StartShareFormFromRemoteInner(MessageParcel& data, MessageParcel& reply);
81 #endif
82     int32_t NotifyStateChangedFromRemoteInner(MessageParcel& data, MessageParcel& reply);
83     int32_t GetDistributedComponentListInner(MessageParcel& data, MessageParcel& reply);
84     void ReportEvent(const OHOS::AAFwk::Want& want, const std::string& eventName, int32_t result, int32_t callerUid);
85     bool CheckDmsRequestPermission();
86     bool CheckCallingUid();
87     bool EnforceInterfaceToken(MessageParcel& data);
88     bool CallerInfoUnmarshalling(CallerInfo& callerInfo, MessageParcel& data);
89     bool IsUsingQos(const std::string& remoteDeviceId);
90     void SaveExtraInfo(const nlohmann::json& extraInfoJson, CallerInfo& callerInfo, AccountInfo& accountInfo);
91     void SaveSendResultExtraInfo(const nlohmann::json& extraInfoJson, CallerInfo& callerInfo,
92         AccountInfo& accountInfo);
93     void InitExtendedLocalFuncsInner();
94     void InitLocalFuncsInner();
95     void InitLocalMissionManagerInner();
96     void InitRemoteFuncsInner();
97 
98     int32_t StopRemoteExtensionAbilityInner(MessageParcel& data, MessageParcel& reply);
99     int32_t StopExtensionAbilityFromRemoteInner(MessageParcel& data, MessageParcel& reply);
100 
101     std::shared_ptr<AAFwk::Want> ReadDistributedWant(MessageParcel& data);
102     int32_t GetEncodeDSchedEventNotify(const EventNotify& event, MessageParcel& reply);
103     int32_t ReadDataForConnect(MessageParcel& data, CallerInfo& callerInfo, AccountInfo& accountInfo);
104     int32_t GetStartAbilityFromRemoteExParam(MessageParcel& data, OHOS::AppExecFwk::AbilityInfo& abilityInfo,
105         int32_t& requestCode, CallerInfo& callerInfo, AccountInfo& accountInfo);
106     int32_t GetConnectAbilityFromRemoteExParam(MessageParcel& data, AppExecFwk::AbilityInfo& abilityInfo,
107         sptr<IRemoteObject>& connect, CallerInfo& callerInfo, AccountInfo& accountInfo);
108 
109 private:
110     using DistributedSchedFunc = int32_t(DistributedSchedStub::*)(MessageParcel& data, MessageParcel& reply);
111     std::map<uint32_t, DistributedSchedFunc> remoteFuncsMap_;
112     std::map<uint32_t, DistributedSchedFunc> localFuncsMap_;
113 };
114 } // namespace DistributedSchedule
115 } // namespace OHOS
116 #endif // OHOS_DISTRIBUTED_SCHEDULE_STUB_H
117