feat(wsl): support detection of current system has virtualization ena…

…bled (#75)

Signed-off-by: Kevin Cui <[email protected]>

pkg/ipc/event/event.go

@@ -30,11 +30,12 @@ type prepare string
 const (
 	SystemNotSupport prepare = "SystemNotSupport"
 
-	NeedEnableFeature    prepare = "NeedEnableFeature"
-	EnableFeaturing      prepare = "EnableFeaturing"
-	EnableFeatureFailed  prepare = "EnableFeatureFailed"
-	EnableFeatureSuccess prepare = "EnableFeatureSuccess"
-	NeedReboot           prepare = "NeedReboot"
+	NotSupportVirtualization prepare = "NotSupportVirtualization"
+	NeedEnableFeature        prepare = "NeedEnableFeature"
+	EnableFeaturing          prepare = "EnableFeaturing"
+	EnableFeatureFailed      prepare = "EnableFeatureFailed"
+	EnableFeatureSuccess     prepare = "EnableFeatureSuccess"
+	NeedReboot               prepare = "NeedReboot"
 
 	NeedUpdateWSL    prepare = "NeedUpdateWSL"
 	UpdatingWSL      prepare = "UpdatingWSL"

pkg/winapi/kernel32.go

@@ -25,3 +25,8 @@ func AttachConsole(pid uintptr) error {
 
 	return nil
 }
+
+func IsProcessorFeaturePresent(feature int) bool {
+	ret, _, _ := isProcessorFeaturePresent.Call(uintptr(feature))
+	return ret != 0
+}

pkg/winapi/sys/processthreadsapi.go

@@ -0,0 +1,19 @@
+package sys
+
+import (
+	"github.com/oomol-lab/ovm-win/pkg/winapi"
+)
+
+// https://learn.microsoft.com/en-us/windows/win32/api/processthreadsapi/nf-processthreadsapi-isprocessorfeaturepresent
+const (
+	PF_SECOND_LEVEL_ADDRESS_TRANSLATION = 20
+	PF_VIRT_FIRMWARE_ENABLED            = 21
+)
+
+// IsSupportedVirtualization checks if the system supports virtualization
+func IsSupportedVirtualization() (vf, slat bool) {
+	vf = winapi.IsProcessorFeaturePresent(PF_VIRT_FIRMWARE_ENABLED)
+	slat = winapi.IsProcessorFeaturePresent(PF_SECOND_LEVEL_ADDRESS_TRANSLATION)
+
+	return
+}

pkg/winapi/winapi.go

@@ -15,10 +15,11 @@ var (
 	kernel32 *windows.LazyDLL
 	mpr      *windows.LazyDLL
 
-	shellExecuteEx       *windows.LazyProc
-	freeConsole          *windows.LazyProc
-	attachConsole        *windows.LazyProc
-	wNetGetUniversalName *windows.LazyProc
+	shellExecuteEx            *windows.LazyProc
+	freeConsole               *windows.LazyProc
+	attachConsole             *windows.LazyProc
+	isProcessorFeaturePresent *windows.LazyProc
+	wNetGetUniversalName      *windows.LazyProc
 )
 
 func init() {
@@ -31,6 +32,7 @@ func init() {
 	shellExecuteEx = shell32.NewProc("ShellExecuteExW")
 	freeConsole = kernel32.NewProc("FreeConsole")
 	attachConsole = kernel32.NewProc("AttachConsole")
+	isProcessorFeaturePresent = kernel32.NewProc("IsProcessorFeaturePresent")
 	wNetGetUniversalName = mpr.NewProc("WNetGetUniversalNameW")
 }
 

pkg/wsl/check.go

@@ -15,6 +15,7 @@ import (
 	"github.com/oomol-lab/ovm-win/pkg/logger"
 	"github.com/oomol-lab/ovm-win/pkg/types"
 	"github.com/oomol-lab/ovm-win/pkg/util"
+	"github.com/oomol-lab/ovm-win/pkg/winapi/sys"
 )
 
 var (
@@ -24,6 +25,13 @@ var (
 
 func Check(opt *types.PrepareOpt) {
 	log := opt.Logger
+
+	if !isSupportedVirtualization(log) {
+		log.Info("Virtualization is not supported")
+		event.NotifyPrepare(event.NotSupportVirtualization)
+		return
+	}
+
 	if isEnabled := isFeatureEnabled(log); !isEnabled {
 		log.Info("WSL2 feature is not enabled")
 		event.NotifyPrepare(event.NeedEnableFeature)
@@ -51,6 +59,21 @@ func Check(opt *types.PrepareOpt) {
 	return
 }
 
+func isSupportedVirtualization(log *logger.Context) bool {
+	vf, slat := sys.IsSupportedVirtualization()
+	// If the CPU does not support SLAT, WSL2 cannot be started (but WSL1 can be started).
+	// In modern CPUs, almost all CPUs support SLAT.
+	// It is not possible to strictly determine this through `vf && slat`, because in VMware, SLAT is always false (even if "Virtualize Intel VT-x/EPT or AMD-V/RVI" is checked).
+	// See:
+	// 		https://github.com/microsoft/WSL/issues/4709
+	// 		https://www.reddit.com/r/bashonubuntuonwindows/comments/izf4qp/cpus_without_slat_capability_cant_run_wsl_2/
+	if !slat {
+		log.Warn("SLAT is not supported")
+	}
+
+	return vf
+}
+
 func existsKernel() bool {
 	// from `MSI` or `Windows Update`
 	if system32, ok := util.System32Root(); ok {