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 "scenesessionmgrstubreg_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 
DoSomethingInterestingWithMyAPI(const uint8_t * data,size_t size)38 bool DoSomethingInterestingWithMyAPI(const uint8_t* data, size_t size)
39 {
40     if (data == nullptr || size < DATA_MIN_SIZE) {
41         return false;
42     }
43 
44     MessageParcel parcel;
45     MessageParcel reply;
46     MessageOption option;
47 
48     parcel.WriteInterfaceToken(SceneSessionManagerStub::GetDescriptor());
49     parcel.WriteBuffer(data, size);
50 
51     parcel.RewindRead(0);
52     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
53         SceneSessionManagerMessage::TRANS_ID_REGISTER_WINDOW_MANAGER_AGENT), parcel, reply, option);
54     parcel.RewindRead(0);
55     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
56         SceneSessionManagerMessage::TRANS_ID_UNREGISTER_WINDOW_MANAGER_AGENT), parcel, reply, option);
57     parcel.RewindRead(0);
58     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
59         SceneSessionManagerMessage::TRANS_ID_REGISTER_SESSION_LISTENER), parcel, reply, option);
60     parcel.RewindRead(0);
61     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
62         SceneSessionManagerMessage::TRANS_ID_UNREGISTER_SESSION_LISTENER), parcel, reply, option);
63     parcel.RewindRead(0);
64     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
65         SceneSessionManagerMessage::TRANS_ID_REGISTER_COLLABORATOR), parcel, reply, option);
66     parcel.RewindRead(0);
67     SceneSessionManager::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManager::
68         SceneSessionManagerMessage::TRANS_ID_UNREGISTER_COLLABORATOR), parcel, reply, option);
69     return true;
70 }
71 } // namespace OHOS
72 
73 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)74 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
75 {
76     /* Run your code on data */
77     OHOS::DoSomethingInterestingWithMyAPI(data, size);
78     return 0;
79 }