util/cpt_upgraders/mempool-sections.py

# This upgrader moves memory pool attributes from a vector of pointers to the
# next free page and a vector of the limit of each pool to sections which each
# represent one pool and which hold the page shift amount, the next free page,
# and the total pages in the pool.
def upgrader(cpt):
    systems = {}
    for sec in cpt.sections():
        ptrs = cpt.get(sec, "ptrs", fallback=None)
        limits = cpt.get(sec, "limits", fallback=None)

        if ptrs and limits:
            systems[sec] = ptrs, limits

    for sec, (ptrs, limits) in systems.items():

        ptrs = list(map(int, ptrs.split()))
        limits = list(map(int, limits.split()))

        if len(ptrs) != len(limits):
            print(
                f"'{sec}ptrs' and '{limits}limits' were not the same length!"
            )

        cpt.set(sec, "num_mem_pools", str(len(ptrs)))

        cpt.remove_option(sec, "ptrs")
        cpt.remove_option(sec, "limits")

        # Assume the page shift is 12, for a 4KiB page.
        page_shift = 12

        for idx, (ptr, limit) in enumerate(zip(ptrs, limits)):
            new_sec = f"{sec}.memPool{idx}"
            cpt.add_section(new_sec)
            cpt.set(new_sec, "page_shift", str(page_shift))
            # Since there's no way to tell where the pool actually started,
            # just assume it started wherever it is right now.
            cpt.set(new_sec, "start_page", str(ptr >> page_shift))
            cpt.set(new_sec, "free_page_num", str(ptr >> page_shift))
            cpt.set(new_sec, "total_pages", str((limit - ptr) >> page_shift))