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 "scenesessionmgrstubupdate_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 
MakeThingsPart01(MessageParcel& parcel)38 void MakeThingsPart01(MessageParcel& parcel)
39 {
40     MessageParcel reply;
41     MessageOption option;
42 
43     parcel.RewindRead(0);
44     SceneSessionManager::GetInstance().OnRemoteRequest(
45         static_cast<uint32_t>(ISceneSessionManager::
46             SceneSessionManagerMessage::TRANS_ID_SET_SESSION_LABEL),
47         parcel, reply, option);
48     parcel.RewindRead(0);
49     SceneSessionManager::GetInstance().OnRemoteRequest(
50         static_cast<uint32_t>(ISceneSessionManager::
51             SceneSessionManagerMessage::TRANS_ID_UPDATE_PROPERTY),
52         parcel, reply, option);
53     parcel.RewindRead(0);
54     SceneSessionManager::GetInstance().OnRemoteRequest(
55         static_cast<uint32_t>(ISceneSessionManager::
56             SceneSessionManagerMessage::TRANS_ID_SET_SESSION_ICON),
57         parcel, reply, option);
58     parcel.RewindRead(0);
59     SceneSessionManager::GetInstance().OnRemoteRequest(
60         static_cast<uint32_t>(ISceneSessionManager::
61             SceneSessionManagerMessage::TRANS_ID_GET_SESSION_SNAPSHOT_BY_ID),
62         parcel, reply, option);
63     parcel.RewindRead(0);
64     SceneSessionManager::GetInstance().OnRemoteRequest(
65         static_cast<uint32_t>(ISceneSessionManager::
66             SceneSessionManagerMessage::TRANS_ID_SET_GESTURE_NAVIGATION_ENABLED),
67         parcel, reply, option);
68     parcel.RewindRead(0);
69     SceneSessionManager::GetInstance().OnRemoteRequest(
70         static_cast<uint32_t>(ISceneSessionManager::
71             SceneSessionManagerMessage::TRANS_ID_UPDATE_AVOIDAREA_LISTENER),
72         parcel, reply, option);
73 }
74 
MakeThingsPart02(MessageParcel& parcel)75 void MakeThingsPart02(MessageParcel& parcel)
76 {
77     MessageParcel reply;
78     MessageOption option;
79     parcel.RewindRead(0);
80     SceneSessionManager::GetInstance().OnRemoteRequest(
81         static_cast<uint32_t>(ISceneSessionManager::
82             SceneSessionManagerMessage::TRANS_ID_SET_SESSION_CONTINUE_STATE),
83         parcel, reply, option);
84     parcel.RewindRead(0);
85     SceneSessionManager::GetInstance().OnRemoteRequest(
86         static_cast<uint32_t>(ISceneSessionManager::
87             SceneSessionManagerMessage::TRANS_ID_UPDATE_TOUCHOUTSIDE_LISTENER),
88         parcel, reply, option);
89     parcel.RewindRead(0);
90     SceneSessionManager::GetInstance().OnRemoteRequest(
91         static_cast<uint32_t>(ISceneSessionManager::
92             SceneSessionManagerMessage::TRANS_ID_UPDATE_WINDOW_VISIBILITY_LISTENER),
93         parcel, reply, option);
94     parcel.RewindRead(0);
95     SceneSessionManager::GetInstance().OnRemoteRequest(
96         static_cast<uint32_t>(ISceneSessionManager::
97             SceneSessionManagerMessage::TRANS_ID_UPDATE_EXTENSION_WINDOW_FLAGS),
98         parcel, reply, option);
99 }
100 
DoSomethingInterestingWithMyAPI(const uint8_t* data, size_t size)101 bool DoSomethingInterestingWithMyAPI(const uint8_t* data, size_t size)
102 {
103     if (data == nullptr || size < DATA_MIN_SIZE) {
104         return false;
105     }
106 
107     MessageParcel parcel;
108     parcel.WriteInterfaceToken(SceneSessionManagerStub::GetDescriptor());
109     parcel.WriteBuffer(data, size);
110 
111     MakeThingsPart01(parcel);
112     MakeThingsPart02(parcel);
113     return true;
114 }
115 } // namespace OHOS
116 
117 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)118 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
119 {
120     /* Run your code on data */
121     OHOS::DoSomethingInterestingWithMyAPI(data, size);
122     return 0;
123 }