platform_bionic/tests/utils.h
Alex Cheung 6fcd7a7d28 Revert "Fix CFI initialization crash on x86."
This reverts commit be46d3871c.

Reverting due to broken build:

https://android-build.googleplex.com/builds/branch-dashboard/aosp-master?build_id=3687823

Change-Id: I30968398edd9a094fce878839ba0a002ed487a77
2017-01-31 04:19:50 +00:00

145 lines
3.4 KiB
C++

/*
* Copyright (C) 2012 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.
*/
#ifndef __TEST_UTILS_H
#define __TEST_UTILS_H
#include <inttypes.h>
#include <sys/mman.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>
#include <atomic>
#include <string>
#include <regex>
#include <android-base/file.h>
#include <android-base/stringprintf.h>
#include "private/ScopeGuard.h"
#if defined(__BIONIC__)
#define KNOWN_FAILURE_ON_BIONIC(x) xfail_ ## x
#else
#define KNOWN_FAILURE_ON_BIONIC(x) x
#endif
#if defined(__linux__)
struct map_record {
uintptr_t addr_start;
uintptr_t addr_end;
int perms;
size_t offset;
dev_t device;
ino_t inode;
std::string pathname;
};
class Maps {
public:
static bool parse_maps(std::vector<map_record>* maps) {
FILE* fp = fopen("/proc/self/maps", "re");
if (fp == nullptr) {
return false;
}
auto fp_guard = make_scope_guard([&]() {
fclose(fp);
});
char line[BUFSIZ];
while (fgets(line, sizeof(line), fp) != nullptr) {
map_record record;
uint32_t dev_major, dev_minor;
int path_offset;
char prot[5]; // sizeof("rwxp")
if (sscanf(line, "%" SCNxPTR "-%" SCNxPTR " %4s %" SCNxPTR " %x:%x %lu %n",
&record.addr_start, &record.addr_end, prot, &record.offset,
&dev_major, &dev_minor, &record.inode, &path_offset) == 7) {
record.perms = 0;
if (prot[0] == 'r') {
record.perms |= PROT_READ;
}
if (prot[1] == 'w') {
record.perms |= PROT_WRITE;
}
if (prot[2] == 'x') {
record.perms |= PROT_EXEC;
}
// TODO: parse shared/private?
record.device = makedev(dev_major, dev_minor);
record.pathname = line + path_offset;
if (!record.pathname.empty() && record.pathname.back() == '\n') {
record.pathname.pop_back();
}
maps->push_back(record);
}
}
return true;
}
};
extern "C" pid_t gettid();
#endif
static inline void WaitUntilThreadSleep(std::atomic<pid_t>& tid) {
while (tid == 0) {
usleep(1000);
}
std::string filename = android::base::StringPrintf("/proc/%d/stat", tid.load());
std::regex regex {R"(\s+S\s+)"};
while (true) {
std::string content;
ASSERT_TRUE(android::base::ReadFileToString(filename, &content));
if (std::regex_search(content, regex)) {
break;
}
usleep(1000);
}
}
static inline void AssertChildExited(int pid, int expected_exit_status) {
int status;
ASSERT_EQ(pid, waitpid(pid, &status, 0));
ASSERT_TRUE(WIFEXITED(status));
ASSERT_EQ(expected_exit_status, WEXITSTATUS(status));
}
// The absolute path to the executable
const std::string& get_executable_path();
// Get realpath
bool get_realpath(const std::string& path, std::string* realpath);
// Get dirname
std::string get_dirname(const char* path);
// Access to argc/argv/envp
int get_argc();
char** get_argv();
char** get_envp();
#endif