X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/d17c4984cc6f7ed55cd3f89b060144570f7cbbe9..50ac3e136606baf487eedb6d42729c4e46f55386:/src/xbt/memory_map.cpp diff --git a/src/xbt/memory_map.cpp b/src/xbt/memory_map.cpp index d9092a25db..b2185d7266 100644 --- a/src/xbt/memory_map.cpp +++ b/src/xbt/memory_map.cpp @@ -1,5 +1,4 @@ -/* Copyright (c) 2008-2015. The SimGrid Team. - * All rights reserved. */ +/* Copyright (c) 2008-2017. The SimGrid Team. All rights reserved. */ /* This program is free software; you can redistribute it and/or modify it * under the terms of the license (GNU LGPL) which comes with this package. */ @@ -9,8 +8,39 @@ #include #include -#ifdef __linux__ + +#if defined __APPLE__ +# include +# include +# include +# include +# include +# include +# include +# if __MAC_OS_X_VERSION_MIN_REQUIRED < 1050 +# define mach_vm_address_t vm_address_t +# define mach_vm_size_t vm_size_t +# if defined __ppc64__ || defined __x86_64__ +# define mach_vm_region vm_region64 +# else +# define mach_vm_region vm_region +# endif +# endif +#endif + +#if defined __linux__ +# include +#endif + +#if defined __FreeBSD__ +# include # include +# include +# include +# include +# include +# include +# include #endif #include @@ -35,7 +65,107 @@ namespace xbt { */ XBT_PRIVATE std::vector get_memory_map(pid_t pid) { -#ifdef __linux__ + std::vector ret; +#if defined __APPLE__ + vm_map_t map; + + /* Request authorization to read mappings */ + if (task_for_pid(mach_task_self(), pid, &map) != KERN_SUCCESS) { + std::perror("task_for_pid failed"); + xbt_die("Cannot request authorization for kernel information access"); + } + + /* + * Darwin do not give us the number of mappings, so we read entries until + * we get an KERN_INVALID_ADDRESS return. + */ + mach_vm_address_t address = VM_MIN_ADDRESS; + while (true) { + kern_return_t kr; + memory_object_name_t object; + mach_vm_size_t size; +#if defined __ppc64__ || defined __x86_64__ + vm_region_flavor_t flavor = VM_REGION_BASIC_INFO_64; + struct vm_region_basic_info_64 info; + mach_msg_type_number_t info_count = VM_REGION_BASIC_INFO_COUNT_64; +#else + vm_region_flavor_t flavor = VM_REGION_BASIC_INFO; + struct vm_region_basic_info info; + mach_msg_type_number_t info_count = VM_REGION_BASIC_INFO_COUNT; +#endif + + kr = + mach_vm_region( + map, + &address, + &size, + flavor, + (vm_region_info_t)&info, + &info_count, + &object); + if (kr == KERN_INVALID_ADDRESS) { + break; + } + else if (kr != KERN_SUCCESS) { + std::perror("mach_vm_region failed"); + xbt_die("Cannot request authorization for kernel information access"); + } + + VmMap memreg; + + /* Addresses */ + memreg.start_addr = address; + memreg.end_addr = address + size; + + /* Permissions */ + memreg.prot = PROT_NONE; + if (info.protection & VM_PROT_READ) + memreg.prot |= PROT_READ; + if (info.protection & VM_PROT_WRITE) + memreg.prot |= PROT_WRITE; + if (info.protection & VM_PROT_EXECUTE) + memreg.prot |= PROT_EXEC; + + /* Private (copy-on-write) or shared? */ + memreg.flags = 0; + if (info.shared) + memreg.flags |= MAP_SHARED; + else + memreg.flags |= MAP_PRIVATE; + + /* Offset */ + memreg.offset = info.offset; + + /* Device : not sure this can be mapped to something outside of Linux? */ + memreg.dev_major = 0; + memreg.dev_minor = 0; + + /* Inode */ + memreg.inode = 0; + + /* Path */ + char path[MAXPATHLEN]; + int pathlen; + pathlen = proc_regionfilename(pid, address, path, sizeof(path)); + path[pathlen] = '\0'; + memreg.pathname = path; + +#if 0 /* Display mappings for debug */ + fprintf(stderr, + "%#014llx - %#014llx | %c%c%c | %s\n", + memreg.start_addr, memreg.end_addr, + (memreg.prot & PROT_READ) ? 'r' : '-', + (memreg.prot & PROT_WRITE) ? 'w' : '-', + (memreg.prot & PROT_EXEC) ? 'x' : '-', + memreg.pathname.c_str()); +#endif + + ret.push_back(std::move(memreg)); + address += size; + } + + mach_port_deallocate(mach_task_self(), map); +#elif defined __linux__ /* Open the actual process's proc maps file and create the memory_map_t */ /* to be returned. */ char* path = bprintf("/proc/%i/maps", (int) pid); @@ -47,8 +177,6 @@ XBT_PRIVATE std::vector get_memory_map(pid_t pid) free(path); setbuf(fp, nullptr); - std::vector ret; - /* Read one line at the time, parse it and add it to the memory map to be returned */ ssize_t read; /* Number of bytes readed */ char* line = nullptr; @@ -104,7 +232,6 @@ XBT_PRIVATE std::vector get_memory_map(pid_t pid) xbt_abort(); memreg.prot = 0; - for (i = 0; i < 3; i++){ switch(lfields[1][i]){ case 'r': @@ -123,10 +250,16 @@ XBT_PRIVATE std::vector get_memory_map(pid_t pid) if (memreg.prot == 0) memreg.prot |= PROT_NONE; - if (lfields[1][4] == 'p') + memreg.flags = 0; + if (lfields[1][3] == 'p') { memreg.flags |= MAP_PRIVATE; - else if (lfields[1][4] == 's') + } else { memreg.flags |= MAP_SHARED; + if (lfields[1][3] != 's') + XBT_WARN("The protection is neither 'p' (private) nor 's' (shared) but '%s'. Let's assume shared, as on b0rken " + "win-ubuntu systems.\nFull line: %s\n", + lfields[1], line); + } /* Get the offset value */ memreg.offset = std::strtoull(lfields[2], &endptr, 16); @@ -164,18 +297,96 @@ XBT_PRIVATE std::vector get_memory_map(pid_t pid) /* Create space for a new map region in the region's array and copy the */ /* parsed stuff from the temporal memreg variable */ - XBT_DEBUG("Found region for %s", !memreg.pathname.empty() ? memreg.pathname.c_str() : "(null)"); + XBT_DEBUG("Found region for %s", not memreg.pathname.empty() ? memreg.pathname.c_str() : "(null)"); ret.push_back(std::move(memreg)); } std::free(line); std::fclose(fp); - return ret; +#elif defined __FreeBSD__ + struct procstat *prstat; + struct kinfo_proc *proc; + struct kinfo_vmentry *vmentries; + unsigned int cnt; + + if ((prstat = procstat_open_sysctl()) == NULL) { + std::perror("procstat_open_sysctl failed"); + xbt_die("Cannot access kernel state information"); + } + if ((proc = procstat_getprocs(prstat, KERN_PROC_PID, pid, &cnt)) == NULL) { + std::perror("procstat_open_sysctl failed"); + xbt_die("Cannot access process information"); + } + if ((vmentries = procstat_getvmmap(prstat, proc, &cnt)) == NULL) { + std::perror("procstat_getvmmap failed"); + xbt_die("Cannot access process memory mappings"); + } + for (unsigned int i = 0; i < cnt; i++) { + VmMap memreg; + + /* Addresses */ + memreg.start_addr = vmentries[i].kve_start; + memreg.end_addr = vmentries[i].kve_end; + + /* Permissions */ + memreg.prot = PROT_NONE; + if (vmentries[i].kve_protection & KVME_PROT_READ) + memreg.prot |= PROT_READ; + if (vmentries[i].kve_protection & KVME_PROT_WRITE) + memreg.prot |= PROT_WRITE; + if (vmentries[i].kve_protection & KVME_PROT_EXEC) + memreg.prot |= PROT_EXEC; + + /* Private (copy-on-write) or shared? */ + memreg.flags = 0; + if (vmentries[i].kve_flags & KVME_FLAG_COW) + memreg.flags |= MAP_PRIVATE; + else + memreg.flags |= MAP_SHARED; + + /* Offset */ + memreg.offset = vmentries[i].kve_offset; + + /* Device : not sure this can be mapped to something outside of Linux? */ + memreg.dev_major = 0; + memreg.dev_minor = 0; + + /* Inode */ + memreg.inode = vmentries[i].kve_vn_fileid; + + /* + * Path. Linuxize result by giving an anonymous mapping a path from + * the previous mapping, provided previous is vnode and has a path, + * and mark the stack. + */ + if (vmentries[i].kve_path[0] != '\0') + memreg.pathname = vmentries[i].kve_path; + else if (vmentries[i].kve_type == KVME_TYPE_DEFAULT && vmentries[i - 1].kve_type == KVME_TYPE_VNODE && + vmentries[i - 1].kve_path[0] != '\0') + memreg.pathname = vmentries[i-1].kve_path; + else if (vmentries[i].kve_type == KVME_TYPE_DEFAULT + && vmentries[i].kve_flags & KVME_FLAG_GROWS_DOWN) + memreg.pathname = "[stack]"; + + /* + * One last dirty modification: remove write permission from shared + * libraries private clean pages. This is necessary because simgrid + * later identifies mappings based on the permissions that are expected + * when running the Linux kernel. + */ + if (vmentries[i].kve_type == KVME_TYPE_VNODE && not(vmentries[i].kve_flags & KVME_FLAG_NEEDS_COPY)) + memreg.prot &= ~PROT_WRITE; + + ret.push_back(std::move(memreg)); + } + procstat_freevmmap(prstat, vmentries); + procstat_freeprocs(prstat, proc); + procstat_close(prstat); #else - /* On FreeBSD, kinfo_getvmmap() could be used but mmap() support is disabled anyway. */ xbt_die("Could not get memory map from process %lli", (long long int) pid); #endif + return ret; } }