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
63 // macOS arm64 can run amd64 tests through Rossetta.
64 // Build the shared library based on the GOARCH and not
65 // the default behavior of the compiler.
66 if runtime.GOOS == "darwin" {
67 var arch string
68 switch runtime.GOARCH {
69 case "arm64":
70 arch = "arm64"
71 case "amd64":
72 arch = "x86_64"
73 default:
74 return fmt.Errorf("unknown macOS architecture %s", runtime.GOARCH)
75 }
76 args = append(args, "-arch", arch)
77 }
78 cmd := exec.Command(compiler, append(args, sources...)...)
79 if out, err := cmd.CombinedOutput(); err != nil {
80 return fmt.Errorf("compile lib: %w\n%q\n%s", err, cmd, string(out))
81 }
82
83 return nil
84}
85
86func TestSyscallN(t *testing.T) {
87 var dlsym uintptr
88 var err error
89 if dlsym, err = purego.Dlsym(purego.RTLD_DEFAULT, "dlsym"); err != nil {
90 t.Errorf("Dlsym with RTLD_DEFAULT failed: %v", err)
91 }
92 r1, _, err2 := purego.SyscallN(dlsym, purego.RTLD_DEFAULT, uintptr(unsafe.Pointer(&[]byte("dlsym\x00")[0])))
93 if dlsym != r1 {
94 t.Fatalf("SyscallN didn't return the same result as purego.Dlsym: %d", err2)
95 }
96}