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 <gtest/gtest.h> 25 #include <unistd.h> 26 27 int randomUid(); 28 29 std::vector<std::string> runCommand(const std::string& command); 30 31 std::vector<std::string> listIpRules(const char* ipVersion); 32 33 std::vector<std::string> listIptablesRule(const char* binary, const char* chainName); 34 35 int iptablesRuleLineLength(const char* binary, const char* chainName); 36 37 bool iptablesRuleExists(const char* binary, const char* chainName, const std::string& expectedRule); 38 39 std::vector<std::string> listIpRoutes(const char* ipVersion, const char* table); 40 41 bool ipRouteExists(const char* ipVersion, const char* table, 42 const std::vector<std::string>& ipRouteSubstrings); 43 44 // Require root (for seteuid). 45 class ScopedUidChange { 46 public: ScopedUidChange(uid_t uid)47 explicit ScopedUidChange(uid_t uid) : mInputUid(uid) { 48 mStoredUid = geteuid(); 49 if (mInputUid == mStoredUid) return; 50 EXPECT_TRUE(seteuid(uid) == 0); 51 } ~ScopedUidChange()52 ~ScopedUidChange() { 53 if (mInputUid == mStoredUid) return; 54 EXPECT_TRUE(seteuid(mStoredUid) == 0); 55 } 56 57 private: 58 uid_t mInputUid; 59 uid_t mStoredUid; 60 }; 61