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 "scenesessionmgrlstubset_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_lite.h"
28 #include "scene_session_manager_lite_stub.h"
29 #include "scene_session_manager_lite_interface.h"
30 
31 using namespace OHOS::Rosen;
32 
33 namespace OHOS {
34 namespace {
35 constexpr size_t DATA_MIN_SIZE = 2;
36 }
37 
SceneSessionMgrLsSetTestCode(MessageParcel & parcel)38 void SceneSessionMgrLsSetTestCode(MessageParcel& parcel)
39 {
40     MessageParcel reply;
41     MessageOption option;
42     parcel.RewindRead(0);
43     SceneSessionManagerLite::GetInstance().OnRemoteRequest(
44         static_cast<uint32_t>(ISceneSessionManagerLite::SceneSessionManagerLiteMessage::TRANS_ID_SET_SESSION_ICON),
45         parcel, reply, option);
46     parcel.RewindRead(0);
47     SceneSessionManagerLite::GetInstance().OnRemoteRequest(
48         static_cast<uint32_t>(ISceneSessionManagerLite::SceneSessionManagerLiteMessage::TRANS_ID_SET_SESSION_LABEL),
49         parcel, reply, option);
50     parcel.RewindRead(0);
51     SceneSessionManagerLite::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManagerLite::
52             SceneSessionManagerLiteMessage::TRANS_ID_SET_SESSION_CONTINUE_STATE),
53         parcel, reply, option);
54     parcel.RewindRead(0);
55     SceneSessionManagerLite::GetInstance().OnRemoteRequest(
56         static_cast<uint32_t>(ISceneSessionManagerLite::SceneSessionManagerLiteMessage::TRANS_ID_LOCK_SESSION),
57         parcel, reply, option);
58     parcel.RewindRead(0);
59     SceneSessionManagerLite::GetInstance().OnRemoteRequest(
60         static_cast<uint32_t>(ISceneSessionManagerLite::SceneSessionManagerLiteMessage::TRANS_ID_UNLOCK_SESSION),
61         parcel, reply, option);
62     parcel.RewindRead(0);
63     SceneSessionManagerLite::GetInstance().OnRemoteRequest(
64         static_cast<uint32_t>(ISceneSessionManagerLite::SceneSessionManagerLiteMessage::TRANS_ID_CLEAR_SESSION),
65         parcel, reply, option);
66     parcel.RewindRead(0);
67     SceneSessionManagerLite::GetInstance().OnRemoteRequest(
68         static_cast<uint32_t>(ISceneSessionManagerLite::SceneSessionManagerLiteMessage::TRANS_ID_CLEAR_ALL_SESSIONS),
69         parcel, reply, option);
70     parcel.RewindRead(0);
71     SceneSessionManagerLite::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManagerLite::
72             SceneSessionManagerLiteMessage::TRANS_ID_PENDING_SESSION_TO_FOREGROUND),
73         parcel, reply, option);
74     parcel.RewindRead(0);
75     SceneSessionManagerLite::GetInstance().OnRemoteRequest(static_cast<uint32_t>(ISceneSessionManagerLite::
76             SceneSessionManagerLiteMessage::TRANS_ID_PENDING_SESSION_TO_BACKGROUND_FOR_DELEGATOR),
77         parcel, reply, option);
78     parcel.RewindRead(0);
79     SceneSessionManagerLite::GetInstance().OnRemoteRequest(
80         static_cast<uint32_t>(ISceneSessionManagerLite::SceneSessionManagerLiteMessage::TRANS_ID_TERMINATE_SESSION_NEW),
81         parcel, reply, option);
82 }
83 
DoSomethingInterestingWithMyAPI(const uint8_t * data,size_t size)84 bool DoSomethingInterestingWithMyAPI(const uint8_t* data, size_t size)
85 {
86     if (data == nullptr || size < DATA_MIN_SIZE) {
87         return false;
88     }
89 
90     MessageParcel parcel;
91 
92     parcel.WriteInterfaceToken(SceneSessionManagerLiteStub::GetDescriptor());
93     parcel.WriteBuffer(data, size);
94 
95     SceneSessionMgrLsSetTestCode(parcel);
96 
97     return true;
98 }
99 } // namespace OHOS
100 
101 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)102 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
103 {
104     /* Run your code on data */
105     OHOS::DoSomethingInterestingWithMyAPI(data, size);
106     return 0;
107 }