1 /*
2 * Copyright (c) 2021 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 "param_stub.h"
17 #include <dirent.h>
18 #include <sys/prctl.h>
19 #include <unistd.h>
20
21 #include "begetctl.h"
22 #include "bootstage.h"
23 #include "init.h"
24 #include "init_log.h"
25 #include "init_param.h"
26 #ifndef OHOS_LITE
27 #include "init_mount.h"
28 #endif
29 #include "hookmgr.h"
30 #include "parameter.h"
31 #include "param_manager.h"
32 #include "param_security.h"
33 #include "param_utils.h"
34 #include "init_group_manager.h"
35 #include "init_module_engine.h"
36 #ifdef PARAM_LOAD_CFG_FROM_CODE
37 #include "param_cfg.h"
38 #endif
39 #include "ueventd.h"
40
41 #ifdef __cplusplus
42 #if __cplusplus
43 extern "C" {
44 #endif
45 #endif
46
47 static int g_stubResult[STUB_MAX] = { 0 };
48 static int g_testRandom = 2; // 2 is test random
49
50 static int g_testPermissionResult = DAC_RESULT_PERMISSION;
SetTestPermissionResult(int result)51 void SetTestPermissionResult(int result)
52 {
53 g_testPermissionResult = result;
54 }
55
56 static const char *selinuxLabels[][2] = {
57 {"test.permission.read", "u:object_r:test_read:s0"},
58 {"test.permission.write", "u:object_r:test_write:s0"},
59 {"test.permission.watch", "u:object_r:test_watch:s0"},
60 };
61
TestGenHashCode(const char * buff)62 static int TestGenHashCode(const char *buff)
63 {
64 int code = 0;
65 size_t buffLen = strlen(buff);
66 for (size_t i = 0; i < buffLen; i++) {
67 code += buff[i] - 'A';
68 }
69 return code;
70 }
71
TestSetSelinuxLogCallback(void)72 static void TestSetSelinuxLogCallback(void) {}
73
TestSetParamCheck(const char * paraName,const char * context,const SrcInfo * info)74 static int TestSetParamCheck(const char *paraName, const char *context, const SrcInfo *info)
75 {
76 BEGET_LOGI("TestSetParamCheck %s result %d", paraName, g_testPermissionResult);
77 return g_testPermissionResult;
78 }
79
TestGetParamLabel(const char * paraName)80 static const char *TestGetParamLabel(const char *paraName)
81 {
82 BEGET_LOGI("TestGetParamLabel %s", paraName);
83 if (paraName == nullptr) {
84 return nullptr;
85 }
86 for (size_t i = 0; i < ARRAY_LENGTH(selinuxLabels); i++) {
87 if (strncmp(selinuxLabels[i][0], paraName, strlen(selinuxLabels[i][0])) == 0) {
88 return selinuxLabels[i][1];
89 }
90 }
91 int code = TestGenHashCode(paraName);
92 code = code % (ARRAY_LENGTH(selinuxLabels));
93 return selinuxLabels[code][1];
94 }
95
TestGetSelinuxLabelIndex(const char * paraName)96 static int32_t TestGetSelinuxLabelIndex(const char *paraName)
97 {
98 for (size_t i = 0; i < ARRAY_LENGTH(selinuxLabels); i++) {
99 if (strncmp(selinuxLabels[i][0], paraName, strlen(selinuxLabels[i][0])) == 0) {
100 return i;
101 }
102 }
103 int code = TestGenHashCode(paraName);
104 code = code % (ARRAY_LENGTH(selinuxLabels));
105 return code;
106 }
107
108 static const char *g_forbidReadParamName[] = {
109 "ohos.servicectrl.",
110 // "test.permission.write",
111 };
TestReadParamCheck(const char * paraName)112 static int TestReadParamCheck(const char *paraName)
113 {
114 // forbid to read ohos.servicectrl.
115 for (size_t i = 0; i < ARRAY_LENGTH(g_forbidReadParamName); i++) {
116 if (strncmp(paraName, g_forbidReadParamName[i], strlen(g_forbidReadParamName[i])) == 0) {
117 return 1;
118 }
119 }
120 return g_testPermissionResult;
121 }
TestDestroyParamList(ParamContextsList ** list)122 static void TestDestroyParamList(ParamContextsList **list)
123 {
124 #ifdef PARAM_SUPPORT_SELINUX
125 ParamContextsList *head = *list;
126 while (head != nullptr) {
127 ParamContextsList *next = head->next;
128 free((void *)head->info.paraName);
129 free((void *)head->info.paraContext);
130 free(head);
131 head = next;
132 }
133 #endif
134 }
TestGetParamList(void)135 static ParamContextsList *TestGetParamList(void)
136 {
137 #ifdef PARAM_SUPPORT_SELINUX
138 ParamContextsList *head = (ParamContextsList *)malloc(sizeof(ParamContextsList));
139 BEGET_ERROR_CHECK(head != nullptr, return nullptr, "Failed to alloc ParamContextsList");
140 head->info.paraName = strdup(selinuxLabels[0][0]);
141 head->info.paraContext = strdup(selinuxLabels[0][1]);
142 head->info.index = 0;
143 head->next = nullptr;
144 for (size_t i = 1; i < ARRAY_LENGTH(selinuxLabels); i++) {
145 ParamContextsList *node = (ParamContextsList *)malloc(sizeof(ParamContextsList));
146 BEGET_ERROR_CHECK(node != nullptr, TestDestroyParamList(&head);
147 return nullptr, "Failed to alloc ParamContextsList");
148 node->info.paraName = strdup(selinuxLabels[i][0]);
149 node->info.paraContext = strdup(selinuxLabels[i][1]);
150 node->info.index = i;
151 node->next = head->next;
152 head->next = node;
153 }
154 // test error, no node paraName
155 ParamContextsList *node = (ParamContextsList *)malloc(sizeof(ParamContextsList));
156 BEGET_ERROR_CHECK(node != nullptr, TestDestroyParamList(&head);
157 return nullptr, "Failed to alloc ParamContextsList");
158 node->info.paraName = nullptr;
159 node->info.paraContext = strdup(selinuxLabels[0][1]);
160 node->next = head->next;
161 head->next = node;
162
163 // test error, no node paraContext
164 node = (ParamContextsList *)malloc(sizeof(ParamContextsList));
165 BEGET_ERROR_CHECK(node != nullptr, TestDestroyParamList(&head);
166 return nullptr, "Failed to alloc ParamContextsList");
167 node->info.paraName = strdup(selinuxLabels[0][0]);
168 node->info.paraContext = nullptr;
169 node->next = head->next;
170 head->next = node;
171
172 // test error, repeat
173 node = (ParamContextsList *)malloc(sizeof(ParamContextsList));
174 BEGET_ERROR_CHECK(node != nullptr, TestDestroyParamList(&head);
175 return nullptr, "Failed to alloc ParamContextsList");
176 node->info.paraName = strdup(selinuxLabels[0][0]);
177 node->info.paraContext = strdup(selinuxLabels[0][1]);
178 node->info.index = 0;
179 node->next = head->next;
180 head->next = node;
181 return head;
182 #else
183 return nullptr;
184 #endif
185 }
186
TestSetSelinuxOps(void)187 void TestSetSelinuxOps(void)
188 {
189 #ifdef PARAM_SUPPORT_SELINUX
190 SelinuxSpace *selinuxSpace = &GetParamWorkSpace()->selinuxSpace;
191 selinuxSpace->setSelinuxLogCallback = TestSetSelinuxLogCallback;
192 selinuxSpace->setParamCheck = TestSetParamCheck;
193 selinuxSpace->getParamLabel = TestGetParamLabel;
194 selinuxSpace->readParamCheck = TestReadParamCheck;
195 selinuxSpace->getParamList = TestGetParamList;
196 selinuxSpace->destroyParamList = TestDestroyParamList;
197 selinuxSpace->getParamLabelIndex = TestGetSelinuxLabelIndex;
198 #endif
199 }
200
TestSetParamCheckResult(const char * prefix,uint16_t mode,int result)201 void TestSetParamCheckResult(const char *prefix, uint16_t mode, int result)
202 {
203 ParamAuditData auditData = {};
204 auditData.name = prefix;
205 auditData.dacData.gid = 202; // 202 test dac gid
206 auditData.dacData.uid = 202; // 202 test dac uid
207 auditData.dacData.mode = mode;
208 AddSecurityLabel(&auditData);
209 SetTestPermissionResult(result);
210 }
211
CreateTestFile(const char * fileName,const char * data)212 void CreateTestFile(const char *fileName, const char *data)
213 {
214 CheckAndCreateDir(fileName);
215 PARAM_LOGV("PrepareParamTestData for %s", fileName);
216 FILE *tmpFile = fopen(fileName, "wr");
217 if (tmpFile != nullptr) {
218 fprintf(tmpFile, "%s", data);
219 (void)fflush(tmpFile);
220 fclose(tmpFile);
221 }
222 }
PrepareUeventdcfg(void)223 static void PrepareUeventdcfg(void)
224 {
225 const char *ueventdcfg = "[device]\n"
226 "/dev/test 0666 1000 1000\n"
227 "[device]\n"
228 "/dev/test1 0666 1000\n"
229 "[device]\n"
230 "/dev/test2 0666 1000 1000 1000 1000\n"
231 "[sysfs]\n"
232 "/dir/to/nothing attr_nowhere 0666 1000 1000\n"
233 "[sysfs]\n"
234 " #/dir/to/nothing attr_nowhere 0666\n"
235 "[sysfs\n"
236 "/dir/to/nothing attr_nowhere 0666\n"
237 "[firmware]\n"
238 "/etc\n"
239 "[device]\n"
240 "/dev/testbinder 0666 1000 1000 const.dev.binder\n"
241 "[device]\n"
242 "/dev/testbinder1 0666 1000 1000 const.dev.binder\n"
243 "[device]\n"
244 "/dev/testbinder2 0666 1000 1000 const.dev.binder\n"
245 "[device]\n"
246 "/dev/testbinder3 0666 1000 1000 const.dev.binder\n";
247 mkdir("/data/ueventd_ut", S_IRWXU | S_IRWXG | S_IRWXO);
248 CreateTestFile(STARTUP_INIT_UT_PATH"/ueventd_ut/valid.config", ueventdcfg);
249 }
PrepareModCfg(void)250 static void PrepareModCfg(void)
251 {
252 const char *modCfg = "testinsmod";
253 CreateTestFile(STARTUP_INIT_UT_PATH"/test_insmod", modCfg);
254 }
PrepareInnerKitsCfg()255 static void PrepareInnerKitsCfg()
256 {
257 const char *innerKitsCfg = "/dev/block/platform/soc/10100000.himci.eMMC/by-name/system /system "
258 "ext4 ro,barrier=1 wait\n"
259 "/dev/block/platform/soc/10100000.himci.eMMC/by-name/vendor /vendor "
260 "ext4 ro,barrier=1 wait\n"
261 "/dev/block/platform/soc/10100000.himci.eMMC/by-name/hos "
262 "/hos ntfs nosuid,nodev,noatime,barrier=1,data=ordered wait\n"
263 "/dev/block/platform/soc/10100000.himci.eMMC/by-name/userdata /data ext4 "
264 "nosuid,nodev,noatime,barrier=1,data=ordered,noauto_da_alloc "
265 "wait,reservedsize=104857600\n"
266 " aaaa\n"
267 "aa aa\n"
268 "aa aa aa\n"
269 "aa aa aa aa\n";
270 const char *fstabRequired = "# fstab file.\n"
271 "#<src> <mnt_point> <type> <mnt_flags and options> <fs_mgr_flags>\n"
272 "/dev/block/platform/fe310000.sdhci/by-name/testsystem /usr ext4 ro,barrier=1 wait,required,nofail\n"
273 "/dev/block/platform/fe310000.sdhci/by-name/testvendor /vendor ext4 ro,barrier=1 wait,required\n"
274 "/dev/block/platform/fe310000.sdhci/by-name/testuserdata1 /data f2fs noatime,nosuid,nodev wait,check,quota\n"
275 "/dev/block/platform/fe310000.sdhci/by-name/testuserdata2 /data ext4 noatime,fscrypt=xxx wait,check,quota\n"
276 "/dev/block/platform/fe310000.sdhci/by-name/testmisc /misc none none wait,required";
277 mkdir("/data/init_ut/mount_unitest/", S_IRWXU | S_IRWXG | S_IRWXO);
278 CreateTestFile(STARTUP_INIT_UT_PATH"/mount_unitest/ReadFstabFromFile1.fstable", innerKitsCfg);
279 CreateTestFile(STARTUP_INIT_UT_PATH"/etc/fstab.required", fstabRequired);
280 CreateTestFile(STARTUP_INIT_UT_PATH"/system/etc/fstab.required", fstabRequired);
281 }
PrepareGroupTestCfg()282 static void PrepareGroupTestCfg()
283 {
284 const char *data = "{"
285 "\"jobs\": [\"param:job1\", \"param:job2\", \"param:job4\"],"
286 "\"services\": [\"service:service1\", \"service:service3\", \"service:service2\"],"
287 "\"groups\": [\"subsystem.xxx1.group\", \"subsystem.xxx2.group\", \"subsystem.xxx4.group\"]"
288 "}";
289 const char *xxx1 = "{"
290 "\"groups\": [\"subsystem.xxx11.group\""
291 "}";
292 const char *xxx11 = "{"
293 "\"groups\": [\"subsystem.xxx12.group\""
294 "}";
295 const char *xxx12 = "{"
296 "\"groups\": [\"subsystem.xxx13.group\""
297 "}";
298 const char *xxx13 = "{"
299 "\"groups\": [\"subsystem.xxx14.group\""
300 "}";
301 const char *xxx14 = "{"
302 "\"groups\": [\"subsystem.xxx11.group\""
303 "}";
304 CreateTestFile(GROUP_DEFAULT_PATH "/device.boot.group.cfg", data);
305 CreateTestFile(GROUP_DEFAULT_PATH "/subsystem.xxx1.group.cfg", xxx1);
306 CreateTestFile(GROUP_DEFAULT_PATH "/subsystem.xxx11.group.cfg", xxx11);
307 CreateTestFile(GROUP_DEFAULT_PATH "/subsystem.xxx12.group.cfg", xxx12);
308 CreateTestFile(GROUP_DEFAULT_PATH "/subsystem.xxx13.group.cfg", xxx13);
309 CreateTestFile(GROUP_DEFAULT_PATH "/subsystem.xxx14.group.cfg", xxx14);
310 }
311
LoadParamFromCfg(void)312 static void LoadParamFromCfg(void)
313 {
314 #ifdef PARAM_LOAD_CFG_FROM_CODE
315 for (size_t i = 0; i < ARRAY_LENGTH(g_paramDefCfgNodes); i++) {
316 PARAM_LOGI("InitParamClient name %s = %s", g_paramDefCfgNodes[i].name, g_paramDefCfgNodes[i].value);
317 uint32_t dataIndex = 0;
318 int ret = WriteParam(g_paramDefCfgNodes[i].name, g_paramDefCfgNodes[i].value, &dataIndex, 0);
319 PARAM_CHECK(ret == 0, continue, "Failed to set param %d name %s %s",
320 ret, g_paramDefCfgNodes[i].name, g_paramDefCfgNodes[i].value);
321 }
322 #endif
323 }
324
325 static const char *g_triggerData = "{"
326 "\"jobs\" : [{"
327 " \"name\" : \"early-init\","
328 " \"cmds\" : ["
329 " \" write '/proc/sys/kernel/sysrq 0'\","
330 " \" load_persist_params \","
331 " \" load_persist_params \","
332 " \" # load_persist_params \","
333 " \" restorecon /postinstall\","
334 " \"mkdir /acct/uid\","
335 " \"chown root system /dev/memcg/memory.pressure_level\","
336 " \"chmod 0040 /dev/memcg/memory.pressure_level\","
337 " \"mkdir /dev/memcg/apps/ 0755 system system\","
338 " \"mkdir /dev/memcg/system 0550 system system\","
339 " \"start ueventd\","
340 " \"exec_start apexd-bootstrap\","
341 " \"setparam sys.usb.config ${persist.sys.usb.config}\""
342 " ]"
343 " },"
344 " {"
345 " \"name\" : \"param:trigger_test_1\","
346 " \"condition\" : \"test.sys.boot_from_charger_mode=5\","
347 " \"cmds\" : ["
348 " \"class_stop charger\","
349 " \"trigger late-init\""
350 " ]"
351 " },"
352 " {"
353 " \"name\" : \"param:trigger_test_2\","
354 " \"condition\" : \"test.sys.boot_from_charger_mode=1 "
355 " || test.sys.boot_from_charger_mode=2 || test.sys.boot_from_charger_mode=3\","
356 " \"cmds\" : ["
357 " \"class_stop charger\","
358 " \"trigger late-init\""
359 " ]"
360 " },"
361 " {"
362 " \"name\" : \"load_persist_params_action\","
363 " \"cmds\" : ["
364 " \"load_persist_params\","
365 " \"start logd\","
366 " \"start logd-reinit\""
367 " ]"
368 " },"
369 " {"
370 " \"name\" : \"firmware_mounts_complete\","
371 " \"cmds\" : ["
372 " \"rm /dev/.booting\""
373 " ]"
374 " }"
375 "]"
376 "}";
377
PrepareCmdLineData()378 void PrepareCmdLineData()
379 {
380 const char *cmdLine = "bootgroup=device.boot.group earlycon=uart8250,mmio32,0xfe660000 "
381 "root=PARTUUID=614e0000-0000 rw rootwait rootfstype=ext4 console=ttyFIQ0 hardware=rk3568 "
382 "BOOT_IMAGE=/kernel init=/init default_boot_device=fe310000.sdhci bootslots=2 currentslot=1 initloglevel=2 "
383 "ohos.required_mount.system="
384 "/dev/block/platform/fe310000.sdhci/by-name/system@/usr@ext4@ro,barrier=1@wait,required "
385 "ohos.required_mount.vendor="
386 "/dev/block/platform/fe310000.sdhci/by-name/vendor@/vendor@ext4@ro,barrier=1@wait,required "
387 "ohos.required_mount.misc="
388 "/dev/block/platform/fe310000.sdhci/by-name/misc@none@none@none@wait,required ohos.boot.eng_mode=on ";
389 CreateTestFile(BOOT_CMD_LINE, cmdLine);
390 }
391
PrepareAreaSizeFile(void)392 static void PrepareAreaSizeFile(void)
393 {
394 const char *ohosParamSize = "default_param=1024\n"
395 "hilog_param=2048\n"
396 "const_product_param=2048\n"
397 "startup_param=20480\n"
398 "persist_param=2048\n"
399 "const_param=20480\n"
400 "test_watch=153600\n" // 1024 * 150
401 "test_write=153600\n" // 1024 * 150
402 "const_param***=20480\n"
403 "persist_sys_param=2048\n";
404 CreateTestFile(PARAM_AREA_SIZE_CFG, ohosParamSize);
405 }
406
PrepareTestGroupFile(void)407 static void PrepareTestGroupFile(void)
408 {
409 std::string groupData = "root:x:0:\n"
410 "bin:x:2:\n"
411 "system:x:1000:\n"
412 "log:x:1007:\n"
413 "deviceinfo:x:1102:\n"
414 "samgr:x:5555:\n"
415 "hdf_devmgr:x:3044:\n\n"
416 "power_host:x:3025:\n"
417 "servicectrl:x:1050:root, shell,system, samgr, hdf_devmgr \n"
418 "powerctrl:x:1051:root, shell,system, update,power_host\r\n"
419 "bootctrl:x:1052:root,shell,system\n"
420 "deviceprivate:x:1053:root,shell,system,samgr,hdf_devmgr,deviceinfo,"
421 "dsoftbus,dms,account,useriam,access_token,device_manager,foundation,dbms,deviceauth,huks_server\n"
422 "hiview:x:1201:\n"
423 "hidumper_service:x:1212:\n"
424 "shell:x:2000:\n"
425 "cache:x:2001:\n"
426 "net_bw_stats:x:3006:\n";
427
428 CreateTestFile(STARTUP_INIT_UT_PATH "/etc/group", groupData.c_str());
429 }
430
PrepareDacData()431 static void PrepareDacData()
432 {
433 // for dac
434 std::string dacData = "ohos.servicectrl. = system:servicectrl:0775 \n";
435 dacData += "startup.service.ctl. = system:servicectrl:0775:int\n";
436 dacData += "test.permission. = root:root:0770\n";
437 dacData += "test.permission.read. = root:root:0774\n";
438 dacData += "test.permission.write.= root:root:0772\n";
439 dacData += "test.permission.watcher. = root:root:0771\n";
440 dacData += "test.test1. = system:test1:0771\n";
441 dacData += "test.test2.watcher. = test2:root:0771\n";
442 dacData += "test.type.int. = root:root:0777:int\n";
443 dacData += "test.type.bool. = root:root:0777:bool\n";
444 dacData += "test.type.string. = root:root:0777\n";
445 dacData += "test.invalid.int. = root:root:\n";
446 dacData += "test.invalid.int. = root::\n";
447 dacData += "test.invalid.int. = ::\n";
448 dacData += "test.invalid.int. = \n";
449 dacData += "test.invalid.int. \n";
450 CreateTestFile(STARTUP_INIT_UT_PATH "/system/etc/param/ohos.para.dac", dacData.c_str());
451 CreateTestFile(STARTUP_INIT_UT_PATH "/system/etc/param/ohos.para.dac_1", dacData.c_str());
452 }
453
TestHook(const HOOK_INFO * hookInfo,void * cookie)454 static int TestHook(const HOOK_INFO *hookInfo, void *cookie)
455 {
456 return 0;
457 }
458
PrepareInitUnitTestEnv(void)459 void PrepareInitUnitTestEnv(void)
460 {
461 static int evnOk = 0;
462 if (evnOk) {
463 return;
464 }
465 printf("PrepareInitUnitTestEnv \n");
466 #ifdef PARAM_SUPPORT_SELINUX
467 RegisterSecuritySelinuxOps(nullptr, 0);
468 #endif
469
470 #ifndef OHOS_LITE
471 InitAddGlobalInitHook(0, TestHook);
472 InitAddPreParamServiceHook(0, TestHook);
473 InitAddPreParamLoadHook(0, TestHook);
474 InitAddPreCfgLoadHook(0, TestHook);
475 InitAddPostCfgLoadHook(0, TestHook);
476 InitAddPostPersistParamLoadHook(0, TestHook);
477 #endif
478 // read default parameter from system
479 LoadDefaultParams("/system/etc/param/ohos_const", LOAD_PARAM_NORMAL);
480 LoadDefaultParams("/vendor/etc/param", LOAD_PARAM_NORMAL);
481 LoadDefaultParams("/system/etc/param", LOAD_PARAM_ONLY_ADD);
482
483 // read ut parameters
484 LoadDefaultParams(STARTUP_INIT_UT_PATH "/system/etc/param/ohos_const", LOAD_PARAM_NORMAL);
485 LoadDefaultParams(STARTUP_INIT_UT_PATH "/vendor/etc/param", LOAD_PARAM_NORMAL);
486 LoadDefaultParams(STARTUP_INIT_UT_PATH "/system/etc/param", LOAD_PARAM_ONLY_ADD);
487 LoadParamsFile(STARTUP_INIT_UT_PATH "/system/etc/param", LOAD_PARAM_ONLY_ADD);
488 LoadParamFromCfg();
489
490 int32_t loglevel = GetIntParameter("persist.init.debug.loglevel", INIT_ERROR);
491 SetInitLogLevel((InitLogLevel)loglevel);
492
493 // for test int get
494 SystemWriteParam("test.int.get", "-101");
495 SystemWriteParam("test.uint.get", "101");
496 SystemWriteParam("test.string.get", "101");
497 SystemWriteParam("test.bool.get.true", "true");
498 SystemWriteParam("test.bool.get.false", "false");
499
500 evnOk = 1;
501 }
502
TestCheckParamPermission(const ParamLabelIndex * labelIndex,const ParamSecurityLabel * srcLabel,const char * name,uint32_t mode)503 int TestCheckParamPermission(const ParamLabelIndex *labelIndex,
504 const ParamSecurityLabel *srcLabel, const char *name, uint32_t mode)
505 {
506 // DAC_RESULT_FORBIDED
507 return g_testPermissionResult;
508 }
509
TestFreeLocalSecurityLabel(ParamSecurityLabel * srcLabel)510 int TestFreeLocalSecurityLabel(ParamSecurityLabel *srcLabel)
511 {
512 return 0;
513 }
514
SetStubResult(STUB_TYPE type,int result)515 void SetStubResult(STUB_TYPE type, int result)
516 {
517 g_stubResult[type] = result;
518 }
519
GetStubResult(STUB_TYPE type)520 int GetStubResult(STUB_TYPE type)
521 {
522 return g_stubResult[type];
523 }
524
525 #ifndef OHOS_LITE
TestBeforeInit(void)526 static void TestBeforeInit(void)
527 {
528 ParamWorkSpace *paramSpace = GetParamWorkSpace();
529 EXPECT_NE(paramSpace, nullptr);
530 InitParamService();
531 CloseParamWorkSpace();
532 paramSpace = GetParamWorkSpace();
533 EXPECT_NE(paramSpace, nullptr);
534
535 // test read cmdline
536 Fstab *stab = LoadRequiredFstab();
537 ReleaseFstab(stab);
538 }
539 #endif
540
541 static pid_t g_currPid = 0;
ParamTestStubInit(void)542 static __attribute__((constructor(101))) void ParamTestStubInit(void)
543 {
544 g_currPid = getpid();
545 printf("Init unit test start %u \n", g_currPid);
546 EnableInitLog(INIT_INFO);
547 InitParseGroupCfg();
548 // prepare data
549 mkdir(STARTUP_INIT_UT_PATH, S_IRWXU | S_IRWXG | S_IRWXO);
550 CheckAndCreateDir(STARTUP_INIT_UT_PATH MODULE_LIB_NAME "/autorun/");
551 int cmdIndex = 0;
552 (void)GetMatchCmd("copy ", &cmdIndex);
553 DoCmdByIndex(cmdIndex, MODULE_LIB_NAME"/libbootchart.z.so "
554 STARTUP_INIT_UT_PATH MODULE_LIB_NAME "/libbootchart.z.so", nullptr);
555 DoCmdByIndex(cmdIndex, MODULE_LIB_NAME"/libbootchart.z.so "
556 STARTUP_INIT_UT_PATH MODULE_LIB_NAME "/autorun/libbootchart.z.so", nullptr);
557 PrepareUeventdcfg();
558 PrepareInnerKitsCfg();
559 PrepareModCfg();
560 PrepareGroupTestCfg();
561 PrepareDacData();
562 CreateTestFile(STARTUP_INIT_UT_PATH"/trigger_test.cfg", g_triggerData);
563 PrepareAreaSizeFile();
564 PrepareTestGroupFile();
565 PrepareCmdLineData();
566 TestSetSelinuxOps();
567
568 PARAM_LOGI("TestSetSelinuxOps \n");
569 #ifndef OHOS_LITE
570 TestBeforeInit();
571 #endif
572 #ifndef __LITEOS_A__
573 SystemInit();
574 #endif
575 PARAM_LOGI("SystemConfig \n");
576 SystemConfig(NULL);
577 PrepareInitUnitTestEnv();
578 }
579
ParamTestStubExit(void)580 __attribute__((destructor)) static void ParamTestStubExit(void)
581 {
582 PARAM_LOGI("ParamTestStubExit %u %u \n", g_currPid, getpid());
583 if (g_currPid != getpid()) {
584 return;
585 }
586 #ifndef OHOS_LITE
587 StopParamService();
588
589 HookMgrExecute(GetBootStageHookMgr(), INIT_BOOT_COMPLETE, nullptr, nullptr);
590 CloseUeventConfig();
591 CloseServiceSpace();
592 demoExit();
593 LE_CloseLoop(LE_GetDefaultLoop());
594 HookMgrDestroy(GetBootStageHookMgr());
595 #endif
596 }
597
598 #ifdef OHOS_LITE
LE_DoAsyncEvent(const LoopHandle loopHandle,const TaskHandle taskHandle)599 void __attribute__((weak))LE_DoAsyncEvent(const LoopHandle loopHandle, const TaskHandle taskHandle)
600 {
601 }
602
HalGetSerial(void)603 const char* HalGetSerial(void)
604 {
605 static const char *serial = "1234567890";
606 return serial;
607 }
608 #endif
609
SprintfStub(char * buffer,size_t size,const char * fmt,...)610 int __attribute__((weak))SprintfStub(char *buffer, size_t size, const char *fmt, ...)
611 {
612 int len = -1;
613 va_list vargs;
614 va_start(vargs, fmt);
615 len = vsnprintf_s(buffer, size, size - 1, fmt, vargs);
616 va_end(vargs);
617 return len;
618 }
619
MountStub(const char * source,const char * target,const char * filesystemtype,unsigned long mountflags,const void * data)620 int __attribute__((weak))MountStub(const char* source, const char* target,
621 const char* filesystemtype, unsigned long mountflags, const void * data)
622 {
623 return g_stubResult[STUB_MOUNT];
624 }
625
UmountStub(const char * target)626 int __attribute__((weak))UmountStub(const char *target)
627 {
628 return 0;
629 }
630
Umount2Stub(const char * target,int flags)631 int __attribute__((weak))Umount2Stub(const char *target, int flags)
632 {
633 return 0;
634 }
635
SymlinkStub(const char * oldpath,const char * newpath)636 int __attribute__((weak))SymlinkStub(const char * oldpath, const char * newpath)
637 {
638 return 0;
639 }
640
PrctlStub(int option,...)641 int PrctlStub(int option, ...)
642 {
643 if (option == PR_SET_SECUREBITS) {
644 static int count = 0;
645 count++;
646 return (count % g_testRandom == 1) ? 0 : -1;
647 }
648 if (option == PR_CAP_AMBIENT) {
649 static int count1 = 0;
650 count1++;
651 return (count1 % g_testRandom == 1) ? 0 : -1;
652 }
653 return 0;
654 }
655
ExecvStub(const char * pathname,char * const argv[])656 int ExecvStub(const char *pathname, char *const argv[])
657 {
658 printf("ExecvStub %s \n", pathname);
659 return 0;
660 }
661
LchownStub(const char * pathname,uid_t owner,gid_t group)662 int LchownStub(const char *pathname, uid_t owner, gid_t group)
663 {
664 return 0;
665 }
666
KillStub(pid_t pid,int signal)667 int KillStub(pid_t pid, int signal)
668 {
669 return 0;
670 }
671
ExecveStub(const char * pathname,char * const argv[],char * const envp[])672 int ExecveStub(const char *pathname, char *const argv[], char *const envp[])
673 {
674 printf("ExecveStub %s \n", pathname);
675 return 0;
676 }
677
LoadPolicy()678 int LoadPolicy()
679 {
680 return 0;
681 }
682
683 static int g_selinuxOptResult = 0;
setcon(const char * name)684 int setcon(const char *name)
685 {
686 g_selinuxOptResult++;
687 return g_selinuxOptResult % g_testRandom;
688 }
689
RestoreconRecurse(const char * name)690 int RestoreconRecurse(const char *name)
691 {
692 g_selinuxOptResult++;
693 return g_selinuxOptResult % g_testRandom;
694 }
695
setexeccon(const char * name)696 int setexeccon(const char *name)
697 {
698 g_selinuxOptResult++;
699 return g_selinuxOptResult % g_testRandom;
700 }
701
setsockcreatecon(const char * name)702 int setsockcreatecon(const char *name)
703 {
704 g_selinuxOptResult++;
705 return g_selinuxOptResult % g_testRandom;
706 }
707
setfilecon(const char * name,const char * content)708 int setfilecon(const char *name, const char *content)
709 {
710 g_selinuxOptResult++;
711 return g_selinuxOptResult % g_testRandom;
712 }
713
TestGetParamLabelIndex(const char * name)714 ParamLabelIndex *TestGetParamLabelIndex(const char *name)
715 {
716 static ParamLabelIndex labelIndex = {0};
717 uint32_t index = 0;
718 ParamWorkSpace *paramWorkspace = GetParamWorkSpace();
719 if (paramWorkspace == nullptr) {
720 return &labelIndex;
721 }
722 #ifdef PARAM_SUPPORT_SELINUX
723 if (paramWorkspace->selinuxSpace.getParamLabelIndex == nullptr) {
724 return &labelIndex;
725 }
726 index = (uint32_t)paramWorkspace->selinuxSpace.getParamLabelIndex(name) + WORKSPACE_INDEX_BASE;
727 if (index >= paramWorkspace->maxLabelIndex) {
728 return &labelIndex;
729 }
730 #endif
731 labelIndex.workspace = paramWorkspace->workSpace[index];
732 PARAM_CHECK(labelIndex.workspace != nullptr, return nullptr, "Invalid workSpace");
733 labelIndex.selinuxLabelIndex = labelIndex.workspace->spaceIndex;
734 (void)FindTrieNode(paramWorkspace->workSpace[0], name, strlen(name), &labelIndex.dacLabelIndex);
735 return &labelIndex;
736 }
737 #ifdef __cplusplus
738 #if __cplusplus
739 }
740 #endif
741 #endif
742