LyScript 实现对内存堆栈扫描

LyScript插件中提供了三种基本的堆栈操作方法，其中push_stack用于入栈，pop_stack用于出栈，而最有用的是peek_stack函数，该函数可用于检查指定堆栈位置处的内存参数，利用这个特性就可以实现，对堆栈地址的检测，或对堆栈的扫描等。

LyScript项目地址：https://github.com/lyshark/LyScript

peek_stack命令传入的是堆栈下标位置默认从0开始，并输出一个十进制有符号长整数，首先实现有符号与无符号数之间的转换操作，为后续堆栈扫描做准备。

from LyScript32 import MyDebug
 
# 有符号整数转无符号数
def long_to_ulong(inter,is_64 = False):
    if is_64 == False:
        return inter & ((1 << 32) - 1)
    else:
        return inter & ((1 << 64) - 1)
 
# 无符号整数转有符号数
def ulong_to_long(inter,is_64 = False):
    if is_64 == False:
        return (inter & ((1 << 31) - 1)) - (inter & (1 << 31))
    else:
        return (inter & ((1 << 63) - 1)) - (inter & (1 << 63))
 
if __name__ == "__main__":
    dbg = MyDebug()
 
    connect_flag = dbg.connect()
    print("连接状态: {}".format(connect_flag))
 
    for index in range(0,10):
 
        # 默认返回有符号数
        stack_address = dbg.peek_stack(index)
 
        # 使用转换
        print("默认有符号数: {:15} --> 转为无符号数: {:15} --> 转为有符号数: {:15}".
              format(stack_address, long_to_ulong(stack_address),ulong_to_long(long_to_ulong(stack_address))))
 
    dbg.close()

通过上述封装函数，即可实现对有符号和无符号数的转换。

继续完善该功能，我们使用get_disasm_one_code()函数，扫描堆栈地址并得到该地址处的反汇编代码。

from LyScript32 import MyDebug
 
# 有符号整数转无符号数
def long_to_ulong(inter,is_64 = False):
    if is_64 == False:
        return inter & ((1 << 32) - 1)
    else:
        return inter & ((1 << 64) - 1)
 
# 无符号整数转有符号数
def ulong_to_long(inter,is_64 = False):
    if is_64 == False:
        return (inter & ((1 << 31) - 1)) - (inter & (1 << 31))
    else:
        return (inter & ((1 << 63) - 1)) - (inter & (1 << 63))
 
if __name__ == "__main__":
    dbg = MyDebug()
 
    connect_flag = dbg.connect()
    print("连接状态: {}".format(connect_flag))
 
    for index in range(0,10):
 
        # 默认返回有符号数
        stack_address = dbg.peek_stack(index)
 
        # 反汇编一行
        dasm = dbg.get_disasm_one_code(stack_address)
 
        # 根据地址得到模块基址
        if stack_address <= 0:
            mod_base = 0
        else:
            mod_base = dbg.get_base_from_address(long_to_ulong(stack_address))
 
        print("stack => [{}] addr = {:10} base = {:10} dasm = {}".format(index, hex(long_to_ulong(stack_address)),hex(mod_base), dasm))
 
    dbg.close()

得到的堆栈参数如下：

由此我们可以得到堆栈处的反汇编参数，但如果我们需要检索堆栈特定区域内是否存在返回到模块的地址，该如何实现呢？

其实很简单，首先我们需要得到程序全局状态下的所有加载模块的基地址，然后得到当前堆栈内存地址内的实际地址，并通过实际内存地址得到模块基地址，对比全局表即可拿到当前模块是返回到了哪里。

from LyScript32 import MyDebug
 
# 有符号整数转无符号数
def long_to_ulong(inter,is_64 = False):
    if is_64 == False:
        return inter & ((1 << 32) - 1)
    else:
        return inter & ((1 << 64) - 1)
 
# 无符号整数转有符号数
def ulong_to_long(inter,is_64 = False):
    if is_64 == False:
        return (inter & ((1 << 31) - 1)) - (inter & (1 << 31))
    else:
        return (inter & ((1 << 63) - 1)) - (inter & (1 << 63))
 
if __name__ == "__main__":
    dbg = MyDebug()
 
    connect_flag = dbg.connect()
    print("连接状态: {}".format(connect_flag))
 
    # 得到程序加载过的所有模块信息
    module_list = dbg.get_all_module()
 
    # 向下扫描堆栈
    for index in range(0,10):
 
        # 默认返回有符号数
        stack_address = dbg.peek_stack(index)
 
        # 反汇编一行
        dasm = dbg.get_disasm_one_code(stack_address)
 
        # 根据地址得到模块基址
        if stack_address <= 0:
            mod_base = 0
        else:
            mod_base = dbg.get_base_from_address(long_to_ulong(stack_address))
 
        # print("stack => [{}] addr = {:10} base = {:10} dasm = {}".format(index, hex(long_to_ulong(stack_address)),hex(mod_base), dasm))
        if mod_base > 0:
            for x in module_list:
                if mod_base == x.get("base"):
                    print("stack => [{}] addr = {:10} base = {:10} dasm = {:15} return = {:10}"
                          .format(index,hex(long_to_ulong(stack_address)),hex(mod_base), dasm,
                                  x.get("name")))
 
    dbg.close()

运行后，即可扫描到堆栈内的所有返回模块的位置。