1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: 2023 The Ebitengine Authors
3
4//go:build darwin || freebsd || linux
5
6package purego_test
7
8import (
9 "errors"
10 "fmt"
11 "os"
12 "os/exec"
13 "path/filepath"
14 "runtime"
15 "strings"
16 "testing"
17 "unsafe"
18
19 "github.com/ebitengine/purego"
20)
21
22func TestSimpleDlsym(t *testing.T) {
23 if _, err := purego.Dlsym(purego.RTLD_DEFAULT, "dlsym"); err != nil {
24 t.Errorf("Dlsym with RTLD_DEFAULT failed: %v", err)
25 }
26}
27
28func TestNestedDlopenCall(t *testing.T) {
29 libFileName := filepath.Join(t.TempDir(), "libdlnested.so")
30 t.Logf("Build %v", libFileName)
31
32 if err := buildSharedLib("CXX", libFileName, filepath.Join("testdata", "libdlnested", "nested_test.cpp")); err != nil {
33 t.Fatal(err)
34 }
35 defer os.Remove(libFileName)
36
37 lib, err := purego.Dlopen(libFileName, purego.RTLD_NOW|purego.RTLD_GLOBAL)
38 if err != nil {
39 t.Fatalf("Dlopen(%q) failed: %v", libFileName, err)
40 }
41
42 purego.Dlclose(lib)
43}
44
45func buildSharedLib(compilerEnv, libFile string, sources ...string) error {
46 out, err := exec.Command("go", "env", compilerEnv).Output()
47 if err != nil {
48 return fmt.Errorf("go env %s error: %w", compilerEnv, err)
49 }
50
51 compiler := strings.TrimSpace(string(out))
52 if compiler == "" {
53 return errors.New("compiler not found")
54 }
55
56 var args []string
57 if runtime.GOOS == "freebsd" {
58 args = []string{"-shared", "-Wall", "-Werror", "-fPIC", "-o", libFile}
59 } else {
60 args = []string{"-shared", "-Wall", "-Werror", "-o", libFile}
61 }
62 if runtime.GOARCH == "386" {
63 args = append(args, "-m32")
64 }
65 // macOS arm64 can run amd64 tests through Rossetta.
66 // Build the shared library based on the GOARCH and not
67 // the default behavior of the compiler.
68 if runtime.GOOS == "darwin" {
69 var arch string
70 switch runtime.GOARCH {
71 case "arm64":
72 arch = "arm64"
73 case "amd64":
74 arch = "x86_64"
75 default:
76 return fmt.Errorf("unknown macOS architecture %s", runtime.GOARCH)
77 }
78 args = append(args, "-arch", arch)
79 }
80 cmd := exec.Command(compiler, append(args, sources...)...)
81 if out, err := cmd.CombinedOutput(); err != nil {
82 return fmt.Errorf("compile lib: %w\n%q\n%s", err, cmd, string(out))
83 }
84
85 return nil
86}
87
88func TestSyscallN(t *testing.T) {
89 var dlsym uintptr
90 var err error
91 if dlsym, err = purego.Dlsym(purego.RTLD_DEFAULT, "dlsym"); err != nil {
92 t.Errorf("Dlsym with RTLD_DEFAULT failed: %v", err)
93 }
94 r1, _, err2 := purego.SyscallN(dlsym, purego.RTLD_DEFAULT, uintptr(unsafe.Pointer(&[]byte("dlsym\x00")[0])))
95 if dlsym != r1 {
96 t.Fatalf("SyscallN didn't return the same result as purego.Dlsym: %d", err2)
97 }
98}