[llvm.py] Make ObjectFile destructor work

Previous code had a double free in MemoryBuffer. The tests now pass.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@152422 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Gregory Szorc
2012-03-09 18:56:33 +00:00
parent 5992f67e68
commit 07c32218f4
3 changed files with 33 additions and 17 deletions

View File

@ -7,6 +7,7 @@
#
#===------------------------------------------------------------------------===#
from .common import LLVMObject
from .common import get_library
from ctypes import POINTER
@ -33,7 +34,7 @@ class MemoryBuffer(object):
if filename is None:
raise Exception("filename argument must be defined")
memory = c_void_p(None)
memory = LLVMObject()
out = c_char_p(None)
result = lib.LLVMCreateMemoryBufferWithContentsOfFile(filename,
@ -43,17 +44,23 @@ class MemoryBuffer(object):
raise Exception("Could not create memory buffer: %s" % out.value)
self._memory = memory
self._as_parameter_ = self._memory
self._owned = True
def __del__(self):
lib.LLVMDisposeMemoryBuffer(self._memory)
if self._owned:
lib.LLVMDisposeMemoryBuffer(self._memory)
def from_param(self):
return self._memory
return self._as_parameter_
def release_ownership(self):
self._owned = False
def register_library(library):
library.LLVMCreateMemoryBufferWithContentsOfFile.argtypes = [c_char_p,
POINTER(c_void_p), POINTER(c_char_p)]
POINTER(LLVMObject), POINTER(c_char_p)]
library.LLVMCreateMemoryBufferWithContentsOfFile.restype = bool
library.LLVMDisposeMemoryBuffer.argtypes = [c_void_p]