Hopper Disassembler script to assign names to CET-enabled PLT stubs

Rename .plt.sec stubs.py

doc = Document.getCurrentDocument()

for i in range(0, doc.getSegmentCount()):
	segment = doc.getSegment(i)
	for j in range(0, segment.getSectionCount()):
		section = segment.getSection(j)
		if section.getName() == ".plt.sec":
			address = section.getStartingAddress()
			length = section.getLength()
			off = 0
			while off < length:
				ins = segment.getInstructionAtAddress(address + off)
				name = segment.getNameAtAddress(address + off)
				if name is not None and ins.getInstructionString() == "endbr64":
					next_ins = address + off + ins.getInstructionLength()
					if segment.getInstructionAtAddress(next_ins).getInstructionString() == "bnd jmp":
						refs = segment.getReferencesFromAddress(next_ins)
						if len(refs) == 1:
							new_name = doc.getNameAtAddress(refs[0])
							new_name = new_name.rsplit("@GOT", 1)[0] + "@plt"
							segment.setNameAtAddress(address + off, new_name)
				off += ins.getInstructionLength()