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 "scenesessionmgrstubcreate_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 
SceneSessionMgrCreateTestCode(MessageParcel & parcel)38 void SceneSessionMgrCreateTestCode(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_CREATE_AND_CONNECT_SPECIFIC_SESSION), parcel, reply, option);
45     parcel.RewindRead(0);
46     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
47         SceneSessionManagerMessage::TRANS_ID_DESTROY_AND_DISCONNECT_SPECIFIC_SESSION), parcel, reply, option);
48     parcel.RewindRead(0);
49     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
50         SceneSessionManagerMessage::TRANS_ID_REQUEST_FOCUS), parcel, reply, option);
51     parcel.RewindRead(0);
52     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
53         SceneSessionManagerMessage::TRANS_ID_TERMINATE_SESSION_NEW), parcel, reply, option);
54     parcel.RewindRead(0);
55     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
56         SceneSessionManagerMessage::TRANS_ID_RECOVER_AND_RECONNECT_SCENE_SESSION), parcel, reply, option);
57     parcel.RewindRead(0);
58     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
59         SceneSessionManagerMessage::TRANS_ID_RECOVER_AND_CONNECT_SPECIFIC_SESSION), parcel, reply, option);
60     parcel.RewindRead(0);
61     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
62         SceneSessionManagerMessage::TRANS_ID_ADD_EXTENSION_WINDOW_STAGE_TO_SCB), parcel, reply, option);
63     parcel.RewindRead(0);
64     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
65         SceneSessionManagerMessage::TRANS_ID_ADD_OR_REMOVE_SECURE_SESSION), parcel, reply, option);
66     parcel.RewindRead(0);
67     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
68         SceneSessionManagerMessage::TRANS_ID_DESTROY_AND_DISCONNECT_SPECIFIC_SESSION_WITH_DETACH_CALLBACK),
69         parcel, reply, option);
70     parcel.RewindRead(0);
71     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
72         SceneSessionManagerMessage::TRANS_ID_BIND_DIALOG_TARGET), parcel, reply, option);
73     parcel.RewindRead(0);
74     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
75         SceneSessionManagerMessage::TRANS_ID_CLEAR_SESSION), parcel, reply, option);
76     parcel.RewindRead(0);
77     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
78         SceneSessionManagerMessage::TRANS_ID_CLEAR_ALL_SESSIONS), parcel, reply, option);
79 }
80 
DoSomethingInterestingWithMyAPI(const uint8_t * data,size_t size)81 bool DoSomethingInterestingWithMyAPI(const uint8_t* data, size_t size)
82 {
83     if (data == nullptr || size < DATA_MIN_SIZE) {
84         return false;
85     }
86 
87     MessageParcel parcel;
88 
89     parcel.WriteInterfaceToken(SceneSessionManagerStub::GetDescriptor());
90     parcel.WriteBuffer(data, size);
91 
92     SceneSessionMgrCreateTestCode(parcel);
93 
94     return true;
95 }
96 } // namespace OHOS
97 
98 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)99 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
100 {
101     /* Run your code on data */
102     OHOS::DoSomethingInterestingWithMyAPI(data, size);
103     return 0;
104 }