!871 addusermanagerfuzztest

Merge pull request !871 from lixiang/master
This commit is contained in:
openharmony_ci 2024-03-29 06:04:05 +00:00 committed by Gitee
commit 5443da3f3d
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F

View File

@ -23,6 +23,7 @@
#include "storage_daemon_stub.h"
#include "storage_daemon.h"
#include "securec.h"
#include "user_manager.h"
using namespace OHOS::StorageDaemon;
@ -33,7 +34,8 @@ constexpr size_t U32_AT_SIZE = 4;
std::shared_ptr<StorageDaemon::StorageDaemon> storageDaemon =
std::make_shared<StorageDaemon::StorageDaemon>();
std::shared_ptr<StorageDaemon::UserManager> userManager =
StorageDaemon::UserManager::GetInstance();
uint32_t GetU32Data(const char* ptr)
{
// 将第0个数字左移24位将第1个数字左移16位将第2个数字左移8位第3个数字不左移
@ -58,6 +60,23 @@ bool StorageDaemonFuzzTest(std::unique_ptr<char[]> data, size_t size)
return true;
}
bool UserManagerFuzzTest(const uint8_t *data, size_t size)
{
if ((data == nullptr) || (size < sizeof(int32_t))) {
return false;
}
int32_t userId = *(reinterpret_cast<const int32_t *>(data));
uint32_t flag = *(reinterpret_cast<const uint32_t *>(data));
userManager->PrepareUserDirs(userId, flag);
userManager->DestroyUserDirs(userId, flag);
userManager->StartUser(userId);
userManager->StopUser(userId);
userManager->CreateBundleDataDir(flag);
return true;
}
} // namespace OHOS
/* Fuzzer entry point */
@ -79,5 +98,7 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
return 0;
}
OHOS::StorageDaemonFuzzTest(move(str), size);
OHOS::UserManagerFuzzTest(data, size);
return 0;
}