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 "admin_manager_fuzzer.h"
17
18 #include "common_fuzzer.h"
19 #include "edm_ipc_interface_code.h"
20 #include "func_code.h"
21 #include "get_data_template.h"
22 #include "message_parcel.h"
23 #define private public
24 #include "admin_manager.h"
25 #include "admin_container.h"
26 #undef private
27 #include "ienterprise_device_mgr.h"
28
29
30 namespace OHOS {
31 namespace EDM {
32 constexpr size_t MIN_SIZE = 64;
33
InitAdminParam(Admin & admin,std::string fuzzString,EntInfo entInfo,ManagedEvent event)34 void InitAdminParam(Admin &admin, std::string fuzzString, EntInfo entInfo, ManagedEvent event)
35 {
36 AdminInfo fuzzAdminInfo;
37 fuzzAdminInfo.packageName_ = fuzzString;
38 fuzzAdminInfo.className_ = fuzzString;
39 fuzzAdminInfo.entInfo_ = entInfo;
40 fuzzAdminInfo.permission_ = { fuzzString };
41 fuzzAdminInfo.managedEvents_ = { event };
42 fuzzAdminInfo.parentAdminName_ = fuzzString;
43 admin.adminInfo_ = fuzzAdminInfo;
44 }
45
DoSomethingInterestingWithMyAPI(const uint8_t * data,size_t size)46 void DoSomethingInterestingWithMyAPI(const uint8_t* data, size_t size)
47 {
48 std::shared_ptr<AdminManager> adminManager = AdminManager::GetInstance();
49 adminManager->Init();
50 std::string fuzzString(reinterpret_cast<const char*>(data), size);
51 std::vector<std::string> permissions = { fuzzString };
52 ManagedEvent event = GetData<ManagedEvent>();
53 Admin admin;
54 EntInfo entInfo;
55 entInfo.enterpriseName = fuzzString;
56 entInfo.description = fuzzString;
57 InitAdminParam(admin, fuzzString, entInfo, event);
58 std::shared_ptr<Admin> adminPtr = std::make_shared<Admin>(admin);
59 std::vector<std::shared_ptr<Admin>> adminPtrVec = { adminPtr };
60 int32_t eventId = CommonFuzzer::GetU32Data(data);
61 std::unordered_map<int32_t, std::vector<std::shared_ptr<Admin>>> subscribeAdmins;
62 subscribeAdmins[eventId] = adminPtrVec;
63 int32_t userId = CommonFuzzer::GetU32Data(data);
64 std::string bundleName(reinterpret_cast<const char*>(data), size);
65 AdminType role = GetData<AdminType>();
66 std::string packageName(reinterpret_cast<const char*>(data), size);
67 std::vector<std::string> packageNameList = { packageName };
68 std::string subAdminName(reinterpret_cast<const char*>(data), size);
69 std::shared_ptr<Admin> subOrSuperAdmin = std::make_shared<Admin>(admin);
70 std::string parentName(reinterpret_cast<const char*>(data), size);
71 std::vector<std::string> subAdmins = { fuzzString };
72 uint32_t fuzzEvent = GetData<uint32_t>();
73 std::vector<uint32_t> events = { fuzzEvent };
74 adminManager->GetAdminBySubscribeEvent(event, subscribeAdmins);
75 adminManager->SetAdminValue(userId, admin);
76 adminManager->GetAdminByPkgName(packageName, userId);
77 adminManager->DeleteAdmin(packageName, userId);
78 adminManager->IsSuperAdminExist();
79 adminManager->IsByodAdminExist();
80 adminManager->IsSuperAdmin(bundleName);
81 adminManager->IsByodAdmin(bundleName, userId);
82 adminManager->IsSuperOrSubSuperAdmin(bundleName);
83 adminManager->GetEnabledAdmin(role, packageNameList, userId);
84 adminManager->GetSubOrSuperOrByodAdminByPkgName(subAdminName, subOrSuperAdmin);
85 adminManager->GetSubSuperAdminsByParentName(parentName, subAdmins);
86 adminManager->GetEntInfo(packageName, entInfo, userId);
87 adminManager->SetEntInfo(packageName, entInfo, userId);
88 adminManager->SaveSubscribeEvents(events, bundleName, userId);
89 adminManager->RemoveSubscribeEvents(events, bundleName, userId);
90 adminManager->GetSuperAdmin();
91
92 AppExecFwk::ExtensionAbilityInfo abilityInfo;
93 Admin adminItem = admin;
94 Admin adminItem1(abilityInfo, role, entInfo, permissions, true);
95 adminItem1 = admin;
96 Admin adminItem2(bundleName, role, permissions);
97 adminItem1.SetParentAdminName(fuzzString);
98 adminItem1.GetParentAdminName();
99 adminItem2.CheckPermission(fuzzString);
100 std::shared_ptr<Admin> getAdmin = std::make_shared<Admin>(admin);
101 std::string policyName(reinterpret_cast<const char*>(data), size);
102 adminManager->UpdateAdmin(getAdmin, userId, adminItem);
103 adminManager->ReplaceSuperAdminByPackageName(parentName, adminItem);
104 adminManager->GetPoliciesByVirtualAdmin(bundleName, parentName, subAdmins);
105 adminManager->GetVirtualAdminsByPolicy(policyName, parentName, packageNameList);
106 adminManager->HasPermissionToHandlePolicy(getAdmin, policyName);
107 adminManager->Dump();
108 adminManager->ClearAdmins();
109 adminManager->InsertAdmins(userId, adminPtrVec);
110 adminManager->IsAdminExist();
111 adminManager->GetAdmins(adminPtrVec, userId);
112 adminManager->SetAdminValue(0, adminItem1);
113 adminManager->SetAdminValue(0, adminItem2);
114 AdminContainer::GetInstance()->UpdateAdmin(0, bundleName, 0x1FF, adminItem2);
115 }
116
117 // Fuzzer entry point.
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)118 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
119 {
120 if (data == nullptr) {
121 return 0;
122 }
123 if (size < MIN_SIZE) {
124 return 0;
125 }
126 g_data = data;
127 g_size = size;
128 g_pos = 0;
129
130 DoSomethingInterestingWithMyAPI(data, size);
131 return 0;
132 }
133 } // namespace EDM
134 } // namespace OHOS