1 /*
2  * Copyright (c) 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 #include "scenesessionmgrstubinfo_fuzzer.h"
17 
18 #include <cstddef>
19 #include <cstdint>
20 #include <parcel.h>
21 #include <securec.h>
22 
23 #include <iremote_stub.h>
24 #include "message_option.h"
25 #include "message_parcel.h"
26 #include "marshalling_helper.h"
27 #include "scene_session_manager.h"
28 #include "scene_session_manager_stub.h"
29 #include "scene_session_manager_interface.h"
30 
31 using namespace OHOS::Rosen;
32 
33 namespace OHOS {
34 namespace {
35 constexpr size_t DATA_MIN_SIZE = 2;
36 }
37 
SceneSessionMgrGetTestCode(MessageParcel& parcel)38 void SceneSessionMgrGetTestCode(MessageParcel& parcel)
39 {
40     MessageParcel reply;
41     MessageOption option;
42     parcel.RewindRead(0);
43     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
44         SceneSessionManagerMessage::TRANS_ID_GET_FOCUS_SESSION_INFO), parcel, reply, option);
45     parcel.RewindRead(0);
46     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
47         SceneSessionManagerMessage::TRANS_ID_GET_WINDOW_INFO), parcel, reply, option);
48     parcel.RewindRead(0);
49     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
50         SceneSessionManagerMessage::TRANS_ID_GET_SESSION_SNAPSHOT), parcel, reply, option);
51     parcel.RewindRead(0);
52     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
53         SceneSessionManagerMessage::TRANS_ID_GET_SESSION_DUMP_INFO), parcel, reply, option);
54     parcel.RewindRead(0);
55     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
56         SceneSessionManagerMessage::TRANS_ID_GET_SESSION_INFO_BY_CONTINUE_SESSION_ID), parcel, reply, option);
57     parcel.RewindRead(0);
58     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
59         SceneSessionManagerMessage::TRANS_ID_GET_FOCUS_SESSION_TOKEN), parcel, reply, option);
60     parcel.RewindRead(0);
61     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
62         SceneSessionManagerMessage::TRANS_ID_GET_FOCUS_SESSION_ELEMENT), parcel, reply, option);
63     parcel.RewindRead(0);
64     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
65         SceneSessionManagerMessage::TRANS_ID_GET_MISSION_INFOS), parcel, reply, option);
66     parcel.RewindRead(0);
67     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
68         SceneSessionManagerMessage::TRANS_ID_GET_MISSION_INFO_BY_ID), parcel, reply, option);
69     parcel.RewindRead(0);
70     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
71         SceneSessionManagerMessage::TRANS_ID_GET_TOP_WINDOW_ID), parcel, reply, option);
72     parcel.RewindRead(0);
73     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
74         SceneSessionManagerMessage::TRANS_ID_GET_VISIBILITY_WINDOW_INFO_ID), parcel, reply, option);
75     parcel.RewindRead(0);
76     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
77         SceneSessionManagerMessage::TRANS_ID_GET_HOST_WINDOW_RECT), parcel, reply, option);
78     parcel.RewindRead(0);
79     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
80         SceneSessionManagerMessage::TRANS_ID_GET_WINDOW_STATUS), parcel, reply, option);
81     parcel.RewindRead(0);
82     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
83         SceneSessionManagerMessage::TRANS_ID_GET_WINDOW_RECT), parcel, reply, option);
84     parcel.RewindRead(0);
85     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
86         SceneSessionManagerMessage::TRANS_ID_GET_WINDOW_MODE_TYPE), parcel, reply, option);
87 }
88 
SceneSessionMgrDumpTestCode(MessageParcel& parcel)89 void SceneSessionMgrDumpTestCode(MessageParcel& parcel)
90 {
91     MessageParcel reply;
92     MessageOption option;
93     parcel.RewindRead(0);
94     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
95         SceneSessionManagerMessage::TRANS_ID_DUMP_SESSION_ALL), parcel, reply, option);
96     parcel.RewindRead(0);
97     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
98         SceneSessionManagerMessage::TRANS_ID_DUMP_SESSION_WITH_ID), parcel, reply, option);
99     parcel.RewindRead(0);
100     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
101         SceneSessionManagerMessage::TRANS_ID_IS_VALID_SESSION_IDS), parcel, reply, option);
102     parcel.RewindRead(0);
103     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
104         SceneSessionManagerMessage::TRANS_ID_CHECK_WINDOW_ID), parcel, reply, option);
105     parcel.RewindRead(0);
106     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
107         SceneSessionManagerMessage::TRANS_ID_NOTIFY_DUMP_INFO_RESULT), parcel, reply, option);
108 }
109 
DoSomethingInterestingWithMyAPI(const uint8_t* data, size_t size)110 bool DoSomethingInterestingWithMyAPI(const uint8_t* data, size_t size)
111 {
112     if (data == nullptr || size < DATA_MIN_SIZE) {
113         return false;
114     }
115 
116     MessageParcel parcel;
117 
118     parcel.WriteInterfaceToken(SceneSessionManagerStub::GetDescriptor());
119     parcel.WriteBuffer(data, size);
120 
121     SceneSessionMgrGetTestCode(parcel);
122     SceneSessionMgrDumpTestCode(parcel);
123     return true;
124 }
125 } // namespace OHOS
126 
127 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)128 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
129 {
130     /* Run your code on data */
131     OHOS::DoSomethingInterestingWithMyAPI(data, size);
132     return 0;
133 }