mirrors/unicorn
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

improve go binding memory leaks

Browse Source
This commit is contained in:
Ryan Hileman 2016-05-28 16:02:12 -07:00
parent 1038e639f6
commit 5fd4c8719d
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
bindings/go/unicorn/hook.go
View File

@ -18,7 +18,6 @@ type HookData struct {
type Hook uint64
var hookToUintptr = make(map[Hook]uintptr)
var hookDataMap = make(map[uintptr]*HookData)
//export hookCode
@ -105,13 +104,13 @@ func (u *uc) HookAdd(htype int, cb interface{}, begin, end uint64, extra ...int)
C.uc_hook_add_wrap(u.handle, &h2, C.uc_hook_type(htype), callback, C.uintptr_t(uptr), C.uint64_t(begin), C.uint64_t(end))
}
hookDataMap[uptr] = data
hookToUintptr[Hook(h2)] = uptr
u.hooks[Hook(h2)] = uptr
return Hook(h2), nil
}
func (u *uc) HookDel(hook Hook) error {
if uptr, ok := hookToUintptr[hook]; ok {
delete(hookToUintptr, hook)
if uptr, ok := u.hooks[hook]; ok {
delete(u.hooks, hook)
delete(hookDataMap, uptr)
}
return errReturn(C.uc_hook_del(u.handle, C.uc_hook(hook)))

7
bindings/go/unicorn/unicorn.go
View File

@ -60,6 +60,7 @@ type Unicorn interface {
type uc struct {
handle *C.uc_engine
final sync.Once
hooks map[Hook]uintptr
}
type UcOptions struct {
@ -81,7 +82,7 @@ func NewUnicorn(arch, mode int) (Unicorn, error) {
if ucerr := C.uc_open(C.uc_arch(arch), C.uc_mode(mode), &handle); ucerr != ERR_OK {
return nil, UcError(ucerr)
}
u := &uc{handle: handle}
u := &uc{handle: handle, hooks: make(map[Hook]uintptr)}
runtime.SetFinalizer(u, func(u *uc) { u.Close() })
return u, nil
}
@ -89,6 +90,10 @@ func NewUnicorn(arch, mode int) (Unicorn, error) {
func (u *uc) Close() (err error) {
u.final.Do(func() {
if u.handle != nil {
for _, uptr := range u.hooks {
delete(hookDataMap, uptr)
}
u.hooks = nil
err = errReturn(C.uc_close(u.handle))
u.handle = nil
}