Error in Free_Bp_handler when getting argument

[Yuuoniy]

Pwngdb/angelheap/angelheap.py

Lines 165 to 169 in cd736c1

	getarch()
	if arch == "x86-64":
	reg = "$rsi"
	result = int(gdb.execute("info register " + reg,to_string=True).split()[1].strip(),16) + 0x10
	else :

According to x86-64 function calling convention, The first six integer or pointer arguments are passed in registers RDI, RSI, RDX, RCX, R8, R9. so RDI stores the first argument of the free function. which denotes the pointer to free. Using RSI would get the wrong value.
Same for Malloc_Bp_handler function, but I found $RDI = $RSI in malloc when debugging. So malloc trace seems normal. But Free_Bp_handler failed to get the value of the pointer to be freed.