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 "unsubscribesystemprocess_fuzzer.h"
17
18#include "fuzztest_utils.h"
19#include "samgr_ipc_interface_code.h"
20namespace OHOS {
21namespace Samgr {
22namespace {
23constexpr size_t THRESHOLD = 4;
24const std::u16string SAMGR_INTERFACE_TOKEN = u"ohos.samgr.accessToken";
25}
26
27void FuzzSubscribeSystemProcess(const uint8_t *data, size_t size)
28{
29    sptr<MockSystemProcessStatusChange> listener = new(std::nothrow) MockSystemProcessStatusChange();
30    if (listener == nullptr) {
31        return;
32    }
33    MessageParcel parcelData;
34    parcelData.WriteInterfaceToken(SAMGR_INTERFACE_TOKEN);
35    parcelData.WriteRemoteObject(listener);
36    FuzzTestUtils::FuzzTestRemoteRequest(parcelData,
37        static_cast<uint32_t>(SamgrInterfaceCode::SUBSCRIBE_SYSTEM_PROCESS_TRANSACTION));
38}
39
40void FuzzUnSubscribeSystemProcess(const uint8_t *data, size_t size)
41{
42    sptr<MockSystemProcessStatusChange> listener = new(std::nothrow) MockSystemProcessStatusChange();
43    if (listener == nullptr) {
44        return;
45    }
46    MessageParcel parcelData;
47    parcelData.WriteInterfaceToken(SAMGR_INTERFACE_TOKEN);
48    parcelData.WriteRemoteObject(listener);
49    FuzzTestUtils::FuzzTestRemoteRequest(parcelData,
50        static_cast<uint32_t>(SamgrInterfaceCode::UNSUBSCRIBE_SYSTEM_PROCESS_TRANSACTION));
51}
52}
53}
54/* Fuzzer entry point */
55extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
56{
57    if (size < OHOS::Samgr::THRESHOLD) {
58        return 0;
59    }
60    OHOS::Samgr::FuzzSubscribeSystemProcess(data, size);
61    OHOS::Samgr::FuzzUnSubscribeSystemProcess(data, size);
62    return 0;
63}