kloenk/lix
0
0
Fork 0
forked from lix-project/lix
Code Pull requests Activity

switch to mmap strategy

Browse source
This commit is contained in:
Jude Taylor 2016-08-16 12:45:55 -07:00
parent 7b006122ae
commit 90516c5a7b
1 changed files with 68 additions and 81 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

149
src/resolve-system-dependencies/resolve-system-dependencies.cc
View file

@ -6,9 +6,13 @@
#include <algorithm> #include <algorithm>
#include <iostream> #include <iostream>
#include <fstream> #include <fstream>
#include <sys/mman.h>
#include <fcntl.h>
#include <mach-o/loader.h> #include <mach-o/loader.h>
#include <mach-o/swap.h> #include <mach-o/swap.h>
#define DO_SWAP(x, y) ((x) ? OSSwapInt32(y) : (y))
using namespace nix; using namespace nix;
static auto cacheDir = Path{}; static auto cacheDir = Path{};
@ -31,76 +35,59 @@ void writeCacheFile(const Path & file, std::set<string> & deps) {
fp.close(); fp.close();
} }
std::string findDylibName(FILE *obj_file, struct load_command cmd) { std::string findDylibName(bool should_swap, ptrdiff_t dylib_command_start) {
fpos_t pos; struct dylib_command *dylc = (struct dylib_command*)dylib_command_start;
fgetpos(obj_file, &pos);
struct dylib_command dylc;
dylc.cmd = cmd.cmd;
dylc.cmdsize = cmd.cmdsize;
fread(&dylc.dylib, sizeof(struct dylib), 1, obj_file);
char *dylib_name = (char*)calloc(cmd.cmdsize, sizeof(char)); return std::string((char*)(dylib_command_start + DO_SWAP(should_swap, dylc->dylib.name.offset)));
fseek(obj_file,
// offset is calculated from the beginning of the load command, which is two
// uint32_t's backwards
dylc.dylib.name.offset - (sizeof(uint32_t) * 2) + pos,
SEEK_SET);
fread(dylib_name, sizeof(char), cmd.cmdsize, obj_file);
fseek(obj_file, pos, SEEK_SET);
return std::string(dylib_name);
}
bool seekMach64Blob(FILE *obj_file, enum NXByteOrder end) {
struct fat_header head;
fread(&head, sizeof(struct fat_header), 1, obj_file);
swap_fat_header(&head, end);
for(uint32_t narches = 0; narches < head.nfat_arch; narches++) {
struct fat_arch arch;
fread(&arch, sizeof(struct fat_arch), 1, obj_file);
swap_fat_arch(&arch, 1, end);
if(arch.cputype == CPU_TYPE_X86_64) {
fseek(obj_file, arch.offset, SEEK_SET);
return true;
}
}
return false;
} }
std::set<std::string> runResolver(const Path & filename) { std::set<std::string> runResolver(const Path & filename) {
FILE *obj_file = fopen(filename.c_str(), "rb"); int fd = open(filename.c_str(), O_RDONLY);
uint32_t magic; struct stat s;
fread(&magic, sizeof(uint32_t), 1, obj_file); fstat(fd, &s);
fseek(obj_file, 0, SEEK_SET); void *obj = mmap(NULL, s.st_size, PROT_READ, MAP_SHARED, fd, 0);
enum NXByteOrder endianness;
if(magic == 0xBEBAFECA) { ptrdiff_t mach64_offset = 0;
endianness = NX_BigEndian;
if(!seekMach64Blob(obj_file, endianness)) { uint32_t magic = ((struct mach_header_64*)obj)->magic;
std::cerr << "Could not find any mach64 blobs in file " << filename << ", continuing..." << std::endl; if(magic == FAT_CIGAM || magic == FAT_MAGIC) {
bool should_swap = magic == FAT_CIGAM;
uint32_t narches = DO_SWAP(should_swap, ((struct fat_header*)obj)->nfat_arch);
for(uint32_t iter = 0; iter < narches; iter++) {
ptrdiff_t header_offset = (ptrdiff_t)obj + sizeof(struct fat_header);
struct fat_arch* arch = (struct fat_arch*)header_offset;
if(DO_SWAP(should_swap, arch->cputype) == CPU_TYPE_X86_64) {
mach64_offset = (ptrdiff_t)DO_SWAP(should_swap, arch->offset);
break;
}
}
if (mach64_offset == 0) {
printMsg(lvlError, format("Could not find any mach64 blobs in file ‘%1%’, continuing...") % filename);
return std::set<string>(); return std::set<string>();
} }
} else if (magic == MH_MAGIC_64 || magic == MH_CIGAM_64) {
mach64_offset = 0;
} }
struct mach_header_64 header;
fread(&header, sizeof(struct mach_header_64), 1, obj_file); struct mach_header_64 *m_header = (struct mach_header_64 *)((ptrdiff_t)obj + mach64_offset);
if(!(header.magic == MH_MAGIC_64 || header.magic == MH_CIGAM_64)) {
std::cerr << "Not a mach-o object file: " << filename << std::endl; bool should_swap = magic == MH_CIGAM_64;
return std::set<string>(); ptrdiff_t cmd_offset = (ptrdiff_t)m_header + sizeof(struct mach_header_64);
}
std::set<string> libs; std::set<string> libs;
for(uint32_t i = 0; i < header.ncmds; i++) { for(uint32_t i = 0; i < DO_SWAP(should_swap, m_header->ncmds); i++) {
struct load_command cmd; struct load_command *cmd = (struct load_command*)cmd_offset;
fread(&cmd.cmd, sizeof(cmd.cmd), 1, obj_file); switch(DO_SWAP(should_swap, cmd->cmd)) {
fread(&cmd.cmdsize, sizeof(cmd.cmdsize), 1, obj_file);
switch(cmd.cmd) {
case LC_LOAD_DYLIB:
case LC_LOAD_UPWARD_DYLIB: case LC_LOAD_UPWARD_DYLIB:
case LC_LOAD_DYLIB:
case LC_REEXPORT_DYLIB: case LC_REEXPORT_DYLIB:
libs.insert(findDylibName(obj_file, cmd)); libs.insert(findDylibName(should_swap, cmd_offset));
break; break;
} }
fseek(obj_file, cmd.cmdsize - (sizeof(uint32_t) * 2), SEEK_CUR); cmd_offset += DO_SWAP(should_swap, cmd->cmdsize);
} }
fclose(obj_file);
libs.erase(filename);
return libs; return libs;
} }
@ -165,40 +152,40 @@ std::set<string> getPath(const Path & path) {
int main(int argc, char ** argv) { int main(int argc, char ** argv) {
return handleExceptions(argv[0], [&]() { return handleExceptions(argv[0], [&]() {
initNix(); initNix();
struct utsname _uname; struct utsname _uname;
uname(&_uname); uname(&_uname);
auto cacheParentDir = (format("%1%/dependency-maps") % settings.nixStateDir).str(); auto cacheParentDir = (format("%1%/dependency-maps") % settings.nixStateDir).str();
cacheDir = (format("%1%/%2%-%3%-%4%") cacheDir = (format("%1%/%2%-%3%-%4%")
% cacheParentDir % cacheParentDir
% _uname.machine % _uname.machine
% _uname.sysname % _uname.sysname
% _uname.release).str(); % _uname.release).str();
mkdir(cacheParentDir.c_str(), 0755); mkdir(cacheParentDir.c_str(), 0755);
mkdir(cacheDir.c_str(), 0755); mkdir(cacheDir.c_str(), 0755);
auto store = openStore(); auto store = openStore();
auto drv = store->derivationFromPath(Path(argv[1])); auto drv = store->derivationFromPath(Path(argv[1]));
Strings impurePaths = tokenizeString<Strings>(get(drv.env, "__impureHostDeps")); Strings impurePaths = tokenizeString<Strings>(get(drv.env, "__impureHostDeps"));
std::set<string> all_paths; std::set<string> all_paths;
for (auto & path : impurePaths) { for (auto & path : impurePaths) {
for(auto & p : getPath(path)) { for(auto & p : getPath(path)) {
all_paths.insert(p); all_paths.insert(p);
}
} }
}
std::cout << "extra-chroot-dirs" << std::endl; std::cout << "extra-chroot-dirs" << std::endl;
for(auto & path : all_paths) { for(auto & path : all_paths) {
std::cout << path << std::endl; std::cout << path << std::endl;
} }
std::cout << std::endl; std::cout << std::endl;
}); });
} }