xref: /base/theme/screenlock_mgr/test/fuzztest/screenlockmanager_fuzzer/screenlockmanager_fuzzer.cpp

/*
 * Copyright (c) 2021-2022 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * miscservices under the License is miscservices on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "screenlockmanager_fuzzer.h"

#include <cstddef>
#include <cstdint>

#include "message_parcel.h"
#include "screenlock_app_manager.h"
#include "screenlock_callback.h"
#include "screenlock_manager.h"
#include "screenlock_manager_interface.h"
#include "screenlock_system_ability.h"
#include "screenlock_system_ability_callback.h"

using namespace OHOS::ScreenLock;

namespace OHOS {
constexpr size_t THRESHOLD = 10;
constexpr int32_t OFFSET = 4;
constexpr size_t LENGTH = 1;
constexpr size_t RANDNUM_ZERO = 0;
constexpr size_t RANDNUM_ONE = 1;
constexpr size_t RANDNUM_TWO = 2;

uint32_t ConvertToUint32(const uint8_t *ptr)
{
    if (ptr == nullptr) {
        return 0;
    }
    uint32_t bigvar = (ptr[0] << 24) | (ptr[1] << 16) | (ptr[2] << 8) | (ptr[3]);
    return bigvar;
}

bool FuzzScreenlockManager(const uint8_t *rawData, size_t size)
{
    uint32_t code = ConvertToUint32(rawData);
    EventListener eventListener;
    if (code == RANDNUM_ZERO) {
        return ScreenLockManager::GetInstance()->IsScreenLocked();
    }
    if (code == RANDNUM_ONE) {
        return ScreenLockManager::GetInstance()->GetSecure();
    }
    if (code == RANDNUM_TWO) {
        sptr<ScreenlockCallback> listener_ = new ScreenlockCallback(eventListener);
        int32_t ret = ScreenLockManager::GetInstance()->Lock(listener_);
        return ret == E_SCREENLOCK_OK;
    }
    return true;
}

bool UnlockFuzzTest(const uint8_t *rawData, size_t size)
{
    EventListener eventListener;
    sptr<ScreenlockCallback> listener_ = new ScreenlockCallback(eventListener);
    if (size < LENGTH) {
        return true;
    }
    int32_t ret = ScreenLockManager::GetInstance()->Unlock(static_cast<Action>(rawData[0] % 3), listener_);
    return ret == E_SCREENLOCK_OK;
}

bool IsLockedFuzzTest(const uint8_t *rawData, size_t size)
{
    if (size < LENGTH) {
        return true;
    }
    bool isLocked = static_cast<bool>(rawData[0] % 2);
    int32_t ret = ScreenLockManager::GetInstance()->IsLocked(isLocked);
    return ret == E_SCREENLOCK_OK;
}

bool FuzzScreenlockAppManager(const uint8_t *rawData, size_t size)
{
    uint32_t code = ConvertToUint32(rawData);
    rawData = rawData + OFFSET;
    size = size - OFFSET;
    EventListener eventListener;
    if (code == RANDNUM_ZERO) {
        sptr<ScreenlockSystemAbilityCallback> listener_ = new ScreenlockSystemAbilityCallback(eventListener);
        int32_t ret = ScreenLockAppManager::GetInstance()->OnSystemEvent(listener_);
        return ret == E_SCREENLOCK_OK;
    }
    if (code == RANDNUM_ONE) {
        int param = 0;
        std::string event(reinterpret_cast<const char *>(rawData), size);
        int32_t ret = ScreenLockAppManager::GetInstance()->SendScreenLockEvent(event, param);
        return ret == E_SCREENLOCK_OK;
    }
    return true;
}

bool FuzzScreenlockIsDisabled(const uint8_t *rawData, size_t size)
{
    if (size < LENGTH) {
        return true;
    }
    int32_t userId = 100;
    bool isDisabled = static_cast<bool>(rawData[0] % 2);
    int32_t ret = ScreenLockAppManager::GetInstance()->IsScreenLockDisabled(userId, isDisabled);
    return ret == E_SCREENLOCK_OK;
}

bool FuzzScreenlockSetDisabled(const uint8_t *rawData, size_t size)
{
    if (size < LENGTH) {
        return true;
    }
    int32_t userId = 100;
    bool isDisabled = static_cast<bool>(rawData[0] % 2);
    int32_t ret = ScreenLockAppManager::GetInstance()->SetScreenLockDisabled(isDisabled, userId);
    return ret == E_SCREENLOCK_OK;
}

bool FuzzScreenlockSetAuthState(const uint8_t *rawData, size_t size)
{
    if (size < LENGTH) {
        return true;
    }
    int32_t userId = 100;
    int32_t authState = 2;
    std::string authToken = "test";
    int32_t ret = ScreenLockAppManager::GetInstance()->SetScreenLockAuthState(authState, userId, authToken);
    return ret == E_SCREENLOCK_OK;
}

bool FuzzScreenlockGetAuthState(const uint8_t *rawData, size_t size)
{
    if (size < LENGTH) {
        return true;
    }
    int32_t userId = 100;
    int32_t authState = 2;
    int32_t ret = ScreenLockAppManager::GetInstance()->GetScreenLockAuthState(userId, authState);
    return ret == E_SCREENLOCK_OK;
}

bool FuzzScreenlockRequestStrongAuth(const uint8_t *rawData, size_t size)
{
    if (size < LENGTH) {
        return true;
    }
    int32_t userId = 100;
    int reasonFlag = 1;
    int32_t ret = ScreenLockAppManager::GetInstance()->RequestStrongAuth(reasonFlag, userId);
    return ret == E_SCREENLOCK_OK;
}

bool FuzzScreenlockGetStrongAuth(const uint8_t *rawData, size_t size)
{
    if (size < LENGTH) {
        return true;
    }
    int32_t userId = 100;
    int reasonFlag = 1;
    int32_t ret = ScreenLockAppManager::GetInstance()->GetStrongAuth(userId, reasonFlag);
    return ret == E_SCREENLOCK_OK;
}

} // namespace OHOS

/* Fuzzer entry point */
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
{
    if (size < OHOS::THRESHOLD) {
        return 0;
    }

    /* Run your code on data */
    OHOS::FuzzScreenlockManager(data, size);
    OHOS::UnlockFuzzTest(data, size);
    OHOS::IsLockedFuzzTest(data, size);
    OHOS::FuzzScreenlockAppManager(data, size);
    OHOS::FuzzScreenlockIsDisabled(data, size);
    OHOS::FuzzScreenlockSetDisabled(data, size);
    OHOS::FuzzScreenlockSetAuthState(data, size);
    OHOS::FuzzScreenlockGetAuthState(data, size);
    OHOS::FuzzScreenlockRequestStrongAuth(data, size);
    OHOS::FuzzScreenlockGetStrongAuth(data, size);
    return 0;
}