Function parse_maps_file_for_binary

src/pystack/maps.py:237–327

(
    binary_name: Path,
    all_maps_iter: Iterable[VirtualMap],
    load_point_by_module: Optional[Dict[str, int]] = None,
)

def parse_maps_file_for_binary(
    binary_name: Path,
    all_maps_iter: Iterable[VirtualMap],
    load_point_by_module: Optional[Dict[str, int]] = None,
) -> MemoryMapInformation:
    min_addr = float("inf")
    max_addr = 0
    maps_by_library: Dict[str, List[VirtualMap]] = collections.defaultdict(list)
    current_lib = ""
    all_maps = tuple(all_maps_iter)

    if load_point_by_module is None:
        load_point_by_module = collections.defaultdict(lambda: 2**64)
        for memory_range in all_maps:
            if memory_range.path is not None:
                load_point_by_module[memory_range.path.name] = min(
                    memory_range.start,
                    load_point_by_module[memory_range.path.name],
                )

    for memory_range in all_maps:
        current_lib = (
            memory_range.path.name if memory_range.path is not None else current_lib
        )
        maps_by_library[current_lib].append(memory_range)

        if memory_range.path is None or not memory_range.path.name.startswith("[v"):
            min_addr = min(min_addr, memory_range.start)
            max_addr = max(max_addr, memory_range.end)
    maps_by_library = dict(maps_by_library)

    python = libpython = bss = heap = None
    try:
        binary_maps = maps_by_library[binary_name.name]
        python = _get_base_map(binary_maps)
    except KeyError:
        LOGGER.debug("Unable to find maps for %r in %r", binary_name, maps_by_library)
        available_maps = {
            str(map.path)
            for map in all_maps
            if map.path is not None and ".so" not in map.path.name
        }
        LOGGER.debug("Available executable maps: %s", ", ".join(available_maps))
        if available_maps:
            maps_txt = ", ".join(available_maps)
            msg = f"These are the available executable memory maps: {maps_txt}"
        else:
            msg = "There are no available executable maps with known paths."
        raise MissingExecutableMaps(
            f"Unable to find maps for the executable {binary_name}. " + msg
        )
    LOGGER.info("python binary first map found: %r", python)

    libpython_binaries = [lib for lib in maps_by_library if "libpython" in lib]
    if len(libpython_binaries) > 1:
        raise PystackError(
            f"Unexpectedly found multiple libpython in process: {libpython_binaries}"
        )