122 lines
3 KiB
C++
122 lines
3 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"
|
|
|
|
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();
|
|
|
|
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));
|
|
}
|
|
|
|
#endif
|