SetCustomDriverPath need check arg format

HyperDbg · Jul 3, 2024 · d6fadcd · d6fadcd
1 parent 862731b
commit d6fadcd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/libhyperdbg_test.go b/libhyperdbg_test.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"github.com/ddkwork/golibrary/stream"
 	"os"
 	"path/filepath"
 	"strings"
@@ -27,8 +28,8 @@ func stringToBytePointer(s string) *byte {
 // run now will bsod
 // go test -run ^\QTestSdk\E$
 func TestSdk(t *testing.T) {
-	AddCurrentDirToPath()
-	SetCustomDriverPath(stringToBytePointer(mylog.Check2(filepath.Abs("."))), stringToBytePointer("hyperkd.sys"))
+	absPath := mylog.Check2(filepath.Abs("hyperkd.sys"))
+	SetCustomDriverPath(stringToBytePointer(absPath), stringToBytePointer(stream.BaseName(absPath)))
 	if isGithubCI() {
 		mylog.Info("github ci windows not support vt-x nested virtualization,skip test")
 		return