gdb-linux-kaslr.py

import subprocess

def vmlinux_kaslr_base():
    out = subprocess.run("awk '/ _text$/ { print($1); exit(0); }' /proc/kallsyms",
                         shell=True, capture_output=True)
    if out.returncode != 0:
        raise Exception("Unable to determine the running kernel's base address")

    base = int(out.stdout.decode(), 16)
    if base == 0:
        raise Exception("Unable to determine the running kernel's base address")

    return base

def vmlinux_link_base(path):
    out = subprocess.run("readelf -WS %s | awk '/ .text/ { print($5); exit(0); }'" % path,
                         shell=True, capture_output=True)
    if out.returncode != 0:
        raise Exception("Unable to determine the kernel's base link address")

    return int(out.stdout.decode(), 16)

class AddVmlinux (gdb.Command):
    """Load vmlinux, compensating for KASLR"""

    def __init__ (self):
        super (AddVmlinux, self).__init__ ("add-vmlinux", gdb.COMMAND_USER)

    def invoke (self, arg, from_tty):
        kaslr_offset = vmlinux_kaslr_base() - vmlinux_link_base(arg)

        gdb.execute("add-symbol-file %s -o %#x" % (arg, kaslr_offset), False)

AddVmlinux ()