3a86aef546
commit 3007df0d153ade6d328321dad8f88d63869159c8
Merge: 7557851 cd14d68
Author: Carl Pearson <pearson@illinois.edu>
Date: Wed Sep 25 09:59:54 2019 -0500
Merge branch 'master' into feature/aslr
commit 7557851508f0e1c2d75244267c5f78fb3d1ca303
Author: Carl Pearson <pearson@illinois.edu>
Date: Tue Sep 24 07:50:15 2019 -0500
wish list
commit 0bbbac31f354c16ae1942dbfe4683a66ec260498
Author: Carl Pearson <pearson@illinois.edu>
Date: Tue Sep 24 07:49:36 2019 -0500
ASLR documentation
commit f1ae37e057792696a739e30ecdbd09e071b8a7d4
Author: Carl Pearson <pearson@illinois.edu>
Date: Tue Sep 24 07:45:54 2019 -0500
add ASLR interface
112 lines
2.4 KiB
C++
112 lines
2.4 KiB
C++
#pragma once
|
|
|
|
#include <cassert>
|
|
#include <cerrno>
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <cstring>
|
|
#include <iostream>
|
|
|
|
#include <sched.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#include <sys/personality.h>
|
|
|
|
#include "perfect/result.hpp"
|
|
|
|
namespace perfect {
|
|
|
|
/*! return a set of CPUs the current thread can run on
|
|
*/
|
|
std::vector<int> cpus() {
|
|
std::vector<int> result;
|
|
cpu_set_t mask;
|
|
if (sched_getaffinity(0 /*caller*/, sizeof(cpu_set_t), &mask)) {
|
|
assert(0 && "failed sched_getaffinity");
|
|
}
|
|
for (int i = 0; i < CPU_SETSIZE; ++i) {
|
|
if (CPU_ISSET(i, &mask)) {
|
|
result.push_back(i);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
Result get_governor(std::string &result, const int cpu) {
|
|
std::string path("/sys/devices/system/cpu/cpu");
|
|
path += std::to_string(cpu);
|
|
path += "/cpufreq/scaling_governor";
|
|
std::ifstream ifs(path, std::ifstream::in);
|
|
std::getline(ifs, result);
|
|
return Result::SUCCESS;
|
|
}
|
|
|
|
Result set_governor(const int cpu, const std::string &governor) {
|
|
std::string path("/sys/devices/system/cpu/cpu");
|
|
path += std::to_string(cpu);
|
|
path += "/cpufreq/scaling_governor";
|
|
std::ofstream ofs(path, std::ofstream::out);
|
|
ofs << governor;
|
|
ofs.close();
|
|
if (ofs.fail()) {
|
|
switch (errno) {
|
|
case EACCES:
|
|
return Result::NO_PERMISSION;
|
|
case ENOENT:
|
|
return Result::NOT_SUPPORTED;
|
|
default:
|
|
return Result::UNKNOWN;
|
|
}
|
|
}
|
|
return Result::SUCCESS;
|
|
}
|
|
|
|
/*! return the smallest cache line size detected on the platform.
|
|
Return 16 if the cache line size could not be detected.
|
|
*/
|
|
size_t cache_linesize() {
|
|
#ifdef __linux__
|
|
long linesize, var;
|
|
|
|
var = sysconf(_SC_LEVEL1_DCACHE_LINESIZE);
|
|
linesize = var;
|
|
|
|
var = sysconf(_SC_LEVEL2_CACHE_LINESIZE);
|
|
linesize = var ? std::min(linesize, var) : linesize;
|
|
|
|
var = sysconf(_SC_LEVEL3_CACHE_LINESIZE);
|
|
linesize = var ? std::min(linesize, var) : linesize;
|
|
|
|
var = sysconf(_SC_LEVEL4_CACHE_LINESIZE);
|
|
linesize = var ? std::min(linesize, var) : linesize;
|
|
|
|
linesize = linesize ? linesize : 16;
|
|
return linesize;
|
|
#else
|
|
#error "unsupported platform"
|
|
#endif
|
|
}
|
|
|
|
namespace detail {
|
|
Result get_personality(int &persona) {
|
|
int ret = personality(0xffffffff);
|
|
if (-1 == ret) {
|
|
return Result::UNKNOWN;
|
|
} else {
|
|
persona = ret;
|
|
}
|
|
return Result::SUCCESS;
|
|
}
|
|
|
|
Result set_personality(const int persona) {
|
|
int ret = personality(persona);
|
|
if (-1 == ret) {
|
|
return Result::UNKNOWN;
|
|
}
|
|
return Result::SUCCESS;
|
|
}
|
|
}
|
|
|
|
} // namespace perfect
|