mirror of
https://github.com/sliverarmory/reflektor
synced 2026-08-25 09:14:48 +00:00
moloch--
38 lines
893 B
Go
38 lines
893 B
Go
//go:build windows && (386 || amd64 || arm64)
|
|
|
|
package reflektor_test
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
)
|
|
|
|
func TestLoadGeneratedCWindowsDLLAndCallStartW(t *testing.T) {
|
|
requireCommand(t, "zig")
|
|
|
|
outDir := t.TempDir()
|
|
dllPath := buildOneSharedLib(t, outDir, "windows", runtime.GOARCH)
|
|
markerPath := filepath.Join(t.TempDir(), "reflektor_marker.txt")
|
|
_ = os.Remove(windowsFallbackMarkerPath)
|
|
t.Cleanup(func() {
|
|
_ = os.Remove(windowsFallbackMarkerPath)
|
|
})
|
|
|
|
if err := os.Setenv("REFLEKTOR_MARKER", markerPath); err != nil {
|
|
t.Fatalf("set env REFLEKTOR_MARKER: %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
_ = os.Unsetenv("REFLEKTOR_MARKER")
|
|
})
|
|
|
|
callWindowsExportFromDLL(t, dllPath, "StartW")
|
|
|
|
got := readMarkerWithWindowsFallback(t, markerPath)
|
|
if !bytes.Equal(got, []byte("ok")) {
|
|
t.Fatalf("unexpected marker bytes: got=%q want=%q", got, []byte("ok"))
|
|
}
|
|
}
|