summaryrefslogtreecommitdiff
path: root/src/dl/dlfcn.go
blob: 3ab5abbdfe4619132b49c61ac08cef7e9f27bba6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
// Copyright 2015 Luke Shumaker <lukeshu@sbcglobal.net>.
//
// This is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License as
// published by the Free Software Foundation; either version 2 of
// the License, or (at your option) any later version.
//
// The GNU General Public License's references to "object code" and
// "executables" are to be interpreted to also include the output of
// any document formatting or typesetting system, including
// intermediate and printed output.
//
// This software is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public
// License along with this manual; if not, see
// <http://www.gnu.org/licenses/>.

// Package dl provides an interface to the POSIX runtime linker.
package dl

import (
	"errors"
	"unsafe"
)

//#cgo LDFLAGS: -ldl
//#include <stdlib.h>
//#include <dlfcn.h>
import "C"

type Flag int

// POSIX specifies these four flags to Open().
const (
	// Relocations are performed at an implementation-defined
	// time.
	RTLD_LAZY Flag = C.RTLD_LAZY

	// Relocations are performed when the object is loaded.
	RTLD_NOW Flag = C.RTLD_NOW

	// All symbols are available for relocation processing of
	// other modules.
	RTLD_GLOBAL Flag = C.RTLD_GLOBAL

	// All symbols are not made available for relocation
	// processing by other modules.
	RTLD_LOCAL Flag = C.RTLD_LOCAL
)

type Handle struct {
	c unsafe.Pointer
}

// Open a shared object file, returning a Handle to it, or an error.
// If name is an empty string, then the returned handle is the global
// symbol table for the current process; if the name contains a slash,
// then it is interpretted as a pathname; otherwise, it is
// interpretted in an implementation-defined manner.
func Open(name string, flags Flag) (Handle, error) {
	nameC := C.CString(name)
	defer C.free(unsafe.Pointer(nameC))
	if name == "" {
		nameC = nil
	}

	dlerror()
	ptr := C.dlopen(nameC, C.int(flags))
	if ptr == nil {
		return Handle{}, dlerror()
	}
	return Handle{ptr}, nil
}

// Look up a symbol, and return a pointer to it.
//
// This returns uintptr instead of unsafe.Pointer so that code using
// dl cannot obtain unsafe.Pointers without importing the unsafe
// package explicitly.
func (h Handle) Sym(symbol string) (uintptr, error) {
	symbolC := C.CString(symbol)
	defer C.free(unsafe.Pointer(symbolC))

	dlerror()
	ptr := C.dlsym(h.c, symbolC)
	if ptr == nil {
		return 0, dlerror()
	}
	return uintptr(ptr), nil
}

// Close this handle on a shared object; decrementint the reference
// count; if the reference count drops below 0, then the object is
// unloaded.
func (h Handle) Close() error {
	dlerror()
	r := C.dlclose(h.c)
	if r != 0 {
		return dlerror()
	}
	return nil
}

func dlerror() error {
	strC := C.dlerror()
	if strC == nil {
		return nil
	}
	return errors.New(C.GoString(strC))
}