Videre
This commit is contained in:
@@ -0,0 +1,82 @@
|
||||
from llvmlite.binding import ffi
|
||||
from ctypes import (c_bool, c_char_p, c_char, c_size_t, string_at, c_uint64,
|
||||
POINTER)
|
||||
|
||||
|
||||
class SectionIteratorRef(ffi.ObjectRef):
|
||||
def name(self):
|
||||
return ffi.lib.LLVMPY_GetSectionName(self)
|
||||
|
||||
def is_text(self):
|
||||
return ffi.lib.LLVMPY_IsSectionText(self)
|
||||
|
||||
def size(self):
|
||||
return ffi.lib.LLVMPY_GetSectionSize(self)
|
||||
|
||||
def address(self):
|
||||
return ffi.lib.LLVMPY_GetSectionAddress(self)
|
||||
|
||||
def data(self):
|
||||
return string_at(ffi.lib.LLVMPY_GetSectionContents(self), self.size())
|
||||
|
||||
def is_end(self, object_file):
|
||||
return ffi.lib.LLVMPY_IsSectionIteratorAtEnd(object_file, self)
|
||||
|
||||
def next(self):
|
||||
ffi.lib.LLVMPY_MoveToNextSection(self)
|
||||
|
||||
def _dispose(self):
|
||||
ffi.lib.LLVMPY_DisposeSectionIterator(self)
|
||||
|
||||
|
||||
class ObjectFileRef(ffi.ObjectRef):
|
||||
@classmethod
|
||||
def from_data(cls, data):
|
||||
return cls(ffi.lib.LLVMPY_CreateObjectFile(data, len(data)))
|
||||
|
||||
@classmethod
|
||||
def from_path(cls, path):
|
||||
with open(path, 'rb') as f:
|
||||
data = f.read()
|
||||
return cls(ffi.lib.LLVMPY_CreateObjectFile(data, len(data)))
|
||||
|
||||
def sections(self):
|
||||
it = SectionIteratorRef(ffi.lib.LLVMPY_GetSections(self))
|
||||
while not it.is_end(self):
|
||||
yield it
|
||||
it.next()
|
||||
|
||||
def _dispose(self):
|
||||
ffi.lib.LLVMPY_DisposeObjectFile(self)
|
||||
|
||||
|
||||
ffi.lib.LLVMPY_CreateObjectFile.argtypes = [c_char_p, c_size_t]
|
||||
ffi.lib.LLVMPY_CreateObjectFile.restype = ffi.LLVMObjectFileRef
|
||||
|
||||
ffi.lib.LLVMPY_DisposeObjectFile.argtypes = [ffi.LLVMObjectFileRef]
|
||||
|
||||
ffi.lib.LLVMPY_GetSections.argtypes = [ffi.LLVMObjectFileRef]
|
||||
ffi.lib.LLVMPY_GetSections.restype = ffi.LLVMSectionIteratorRef
|
||||
|
||||
ffi.lib.LLVMPY_DisposeSectionIterator.argtypes = [ffi.LLVMSectionIteratorRef]
|
||||
|
||||
ffi.lib.LLVMPY_MoveToNextSection.argtypes = [ffi.LLVMSectionIteratorRef]
|
||||
|
||||
ffi.lib.LLVMPY_IsSectionIteratorAtEnd.argtypes = [
|
||||
ffi.LLVMObjectFileRef, ffi.LLVMSectionIteratorRef]
|
||||
ffi.lib.LLVMPY_IsSectionIteratorAtEnd.restype = c_bool
|
||||
|
||||
ffi.lib.LLVMPY_GetSectionName.argtypes = [ffi.LLVMSectionIteratorRef]
|
||||
ffi.lib.LLVMPY_GetSectionName.restype = c_char_p
|
||||
|
||||
ffi.lib.LLVMPY_GetSectionSize.argtypes = [ffi.LLVMSectionIteratorRef]
|
||||
ffi.lib.LLVMPY_GetSectionSize.restype = c_uint64
|
||||
|
||||
ffi.lib.LLVMPY_GetSectionAddress.argtypes = [ffi.LLVMSectionIteratorRef]
|
||||
ffi.lib.LLVMPY_GetSectionAddress.restype = c_uint64
|
||||
|
||||
ffi.lib.LLVMPY_GetSectionContents.argtypes = [ffi.LLVMSectionIteratorRef]
|
||||
ffi.lib.LLVMPY_GetSectionContents.restype = POINTER(c_char)
|
||||
|
||||
ffi.lib.LLVMPY_IsSectionText.argtypes = [ffi.LLVMSectionIteratorRef]
|
||||
ffi.lib.LLVMPY_IsSectionText.restype = c_bool
|
||||
Reference in New Issue
Block a user