/* * Copyright (C) 2015 The Android Open Source Project * * 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 * distributed under the License is distributed 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 "init_parser.h" #include "init.h" #include "util.h" #include <errno.h> #include <gtest/gtest.h> TEST(init_parser, make_exec_oneshot_service_invalid_syntax) { char* argv[10]; memset(argv, 0, sizeof(argv)); // Nothing. ASSERT_EQ(nullptr, make_exec_oneshot_service(0, argv)); // No arguments to 'exec'. argv[0] = const_cast<char*>("exec"); ASSERT_EQ(nullptr, make_exec_oneshot_service(1, argv)); // No command in "exec --". argv[1] = const_cast<char*>("--"); ASSERT_EQ(nullptr, make_exec_oneshot_service(2, argv)); } TEST(init_parser, make_exec_oneshot_service_too_many_supplementary_gids) { int argc = 0; char* argv[4 + NR_SVC_SUPP_GIDS + 3]; argv[argc++] = const_cast<char*>("exec"); argv[argc++] = const_cast<char*>("seclabel"); argv[argc++] = const_cast<char*>("root"); // uid. argv[argc++] = const_cast<char*>("root"); // gid. for (int i = 0; i < NR_SVC_SUPP_GIDS; ++i) { argv[argc++] = const_cast<char*>("root"); // Supplementary gid. } argv[argc++] = const_cast<char*>("--"); argv[argc++] = const_cast<char*>("/system/bin/id"); argv[argc] = nullptr; ASSERT_EQ(nullptr, make_exec_oneshot_service(argc, argv)); } static void Test_make_exec_oneshot_service(bool dash_dash, bool seclabel, bool uid, bool gid, bool supplementary_gids) { int argc = 0; char* argv[10]; argv[argc++] = const_cast<char*>("exec"); if (seclabel) { argv[argc++] = const_cast<char*>("u:r:su:s0"); // seclabel if (uid) { argv[argc++] = const_cast<char*>("log"); // uid if (gid) { argv[argc++] = const_cast<char*>("shell"); // gid if (supplementary_gids) { argv[argc++] = const_cast<char*>("system"); // supplementary gid 0 argv[argc++] = const_cast<char*>("adb"); // supplementary gid 1 } } } } if (dash_dash) { argv[argc++] = const_cast<char*>("--"); } argv[argc++] = const_cast<char*>("/system/bin/toybox"); argv[argc++] = const_cast<char*>("id"); argv[argc] = nullptr; service* svc = make_exec_oneshot_service(argc, argv); ASSERT_NE(nullptr, svc); if (seclabel) { ASSERT_STREQ("u:r:su:s0", svc->seclabel); } else { ASSERT_EQ(nullptr, svc->seclabel); } if (uid) { ASSERT_EQ(decode_uid("log"), svc->uid); } else { ASSERT_EQ(0U, svc->uid); } if (gid) { ASSERT_EQ(decode_uid("shell"), svc->gid); } else { ASSERT_EQ(0U, svc->gid); } if (supplementary_gids) { ASSERT_EQ(2U, svc->nr_supp_gids); ASSERT_EQ(decode_uid("system"), svc->supp_gids[0]); ASSERT_EQ(decode_uid("adb"), svc->supp_gids[1]); } else { ASSERT_EQ(0U, svc->nr_supp_gids); } ASSERT_EQ(2, svc->nargs); ASSERT_EQ("/system/bin/toybox", svc->args[0]); ASSERT_EQ("id", svc->args[1]); ASSERT_EQ(nullptr, svc->args[2]); } TEST(init_parser, make_exec_oneshot_service_with_everything) { Test_make_exec_oneshot_service(true, true, true, true, true); } TEST(init_parser, make_exec_oneshot_service_with_seclabel_uid_gid) { Test_make_exec_oneshot_service(true, true, true, true, false); } TEST(init_parser, make_exec_oneshot_service_with_seclabel_uid) { Test_make_exec_oneshot_service(true, true, true, false, false); } TEST(init_parser, make_exec_oneshot_service_with_seclabel) { Test_make_exec_oneshot_service(true, true, false, false, false); } TEST(init_parser, make_exec_oneshot_service_with_just_command) { Test_make_exec_oneshot_service(true, false, false, false, false); } TEST(init_parser, make_exec_oneshot_service_with_just_command_no_dash) { Test_make_exec_oneshot_service(false, false, false, false, false); }