1 /* 2 * Copyright 2019 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 * 16 * test_utils.h - miscellaneous unit test utilities. 17 */ 18 19 #pragma once 20 21 #include <string> 22 #include <vector> 23 24 #include <binder/Status.h> 25 #include <gtest/gtest.h> 26 #include <unistd.h> 27 28 int randomUid(); 29 30 std::vector<std::string> runCommand(const std::string& command); 31 32 android::status_t runBinderCommand(const std::string serviceName, const std::string& command); 33 34 std::vector<std::string> listIpRules(const char* ipVersion); 35 36 std::vector<std::string> listIptablesRule(const char* binary, const char* chainName); 37 38 int iptablesRuleLineLength(const char* binary, const char* chainName); 39 40 bool iptablesRuleExists(const char* binary, const char* chainName, const std::string& expectedRule); 41 42 std::vector<std::string> listIpRoutes(const char* ipVersion, const char* table); 43 44 bool ipRouteExists(const char* ipVersion, const char* table, 45 const std::vector<std::string>& ipRouteSubstrings); 46 47 // Require root (for seteuid). 48 class ScopedUidChange { 49 public: ScopedUidChange(uid_t uid)50 explicit ScopedUidChange(uid_t uid) : mInputUid(uid) { 51 mStoredUid = geteuid(); 52 if (mInputUid == mStoredUid) return; 53 EXPECT_TRUE(seteuid(uid) == 0); 54 } ~ScopedUidChange()55 ~ScopedUidChange() { 56 if (mInputUid == mStoredUid) return; 57 EXPECT_TRUE(seteuid(mStoredUid) == 0); 58 } 59 60 private: 61 uid_t mInputUid; 62 uid_t mStoredUid; 63 }; 64