Skip to content

Commit

Permalink
add files
Browse files Browse the repository at this point in the history
  • Loading branch information
yuk7 committed Dec 1, 2021
1 parent 01dfdcc commit 9ff1d6e
Show file tree
Hide file tree
Showing 5 changed files with 202 additions and 0 deletions.
5 changes: 5 additions & 0 deletions go.mod
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
module github.com/yuk7/wsllib-go

go 1.16

require golang.org/x/sys v0.0.0-20211124211545-fe61309f8881
2 changes: 2 additions & 0 deletions go.sum
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
golang.org/x/sys v0.0.0-20211124211545-fe61309f8881 h1:TyHqChC80pFkXWraUUf6RuB5IqFdQieMLwwCJokV2pc=
golang.org/x/sys v0.0.0-20211124211545-fe61309f8881/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
3 changes: 3 additions & 0 deletions mksyscall.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
package wsllib

//go:generate go run golang.org/x/sys/windows/mkwinsyscall -output zsyscall_windows.go wsllib.go
93 changes: 93 additions & 0 deletions wsllib.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,93 @@
package wsllib

import (
"os"
"syscall"
)

const (
//FlagEnableInterop is flag of interop feature
FlagEnableInterop = 1
//FlagAppendNTPath is flag of appending windows path
FlagAppendNTPath = 2
//FlagEnableDriveMounting is flag of mounting windows drive
FlagEnableDriveMounting = 4
//FlagEnableWsl2 is flag of enabled wsl2, read only
FlagEnableWsl2 = 8
)

//sys _WslGetDistributionConfiguration(distributionName *uint16, distributionVersion *uint32, defaultUID *uint64, wslDistributionFlags *uint32, defaultEnv ***uint16, defaultEnvCnt *uint64) (res error) = wslapi.WslGetDistributionConfiguration

// WslGetDistributionConfiguration gets distribution configuration
func WslGetDistributionConfiguration(distributionName string) (distributionVersion uint32, defaultUID uint64, flags uint32, err error) {
pDistributionName, _ := syscall.UTF16PtrFromString(distributionName)

var pEnv **uint16
var envCnt uint64

err = _WslGetDistributionConfiguration(pDistributionName, &distributionVersion, &defaultUID, &flags, &pEnv, &envCnt)

return
}

//sys _WslConfigureDistribution(distributionName *uint16, defaultUID uint64, wslDistributionFlags uint32) (res error) = wslapi.WslConfigureDistribution

// WslConfigureDistribution configures distribution configuration
func WslConfigureDistribution(distributionName string, defaultUID uint64, wslDistributionFlags uint32) (err error) {
pDistributionName, _ := syscall.UTF16PtrFromString(distributionName)

return _WslConfigureDistribution(pDistributionName, defaultUID, wslDistributionFlags)
}

//sys _WslIsDistributionRegistered(distributionName *uint16) (res bool) = wslapi.WslIsDistributionRegistered

// WslIsDistributionRegistered determines if a distribution is already registered
func WslIsDistributionRegistered(distributionName string) bool {
pDistributionName, _ := syscall.UTF16PtrFromString(distributionName)
return _WslIsDistributionRegistered(pDistributionName)
}

//sys _WslRegisterDistribution(distributionName *uint16, tarGzFilename *uint16) (res error) = wslapi.WslRegisterDistribution

// WslRegisterDistribution registers a new distribution
func WslRegisterDistribution(distributionName, tarGzFilename string) error {
pDistributionName, _ := syscall.UTF16PtrFromString(distributionName)
pTarGzFilename, _ := syscall.UTF16PtrFromString(tarGzFilename)
return _WslRegisterDistribution(pDistributionName, pTarGzFilename)
}

//sys _WslLaunch(distributionName *uint16, command *uint16, useCurrentWorkingDirectory bool, stdIn syscall.Handle, stdOut syscall.Handle, stdErr syscall.Handle, process *syscall.Handle) (err error) = wslapi.WslLaunch

// WslLaunch launches the distribution with handle
func WslLaunch(distributionName string, command string, useCurrentWorkingDirectory bool, stdIn syscall.Handle, stdOut syscall.Handle, stdErr syscall.Handle) (process syscall.Handle, err error) {
pDistributionName, _ := syscall.UTF16PtrFromString(distributionName)
pCommand, _ := syscall.UTF16PtrFromString(command)

_WslLaunch(pDistributionName, pCommand, useCurrentWorkingDirectory, stdIn, stdOut, stdErr, &process)
return
}

// WslLaunchInteractive launches the distribution with interactive shell
func WslLaunchInteractive(distributionName string, command string, useCurrentWorkingDirectory bool) (exitCode uint32, err error) {
p, _ := syscall.GetCurrentProcess()
stdin := syscall.Handle(0)
stdout := syscall.Handle(0)
stderr := syscall.Handle(0)

syscall.DuplicateHandle(p, syscall.Handle(os.Stdin.Fd()), p, &stdin, 0, true, syscall.DUPLICATE_SAME_ACCESS)
syscall.DuplicateHandle(p, syscall.Handle(os.Stdout.Fd()), p, &stdout, 0, true, syscall.DUPLICATE_SAME_ACCESS)
syscall.DuplicateHandle(p, syscall.Handle(os.Stderr.Fd()), p, &stderr, 0, true, syscall.DUPLICATE_SAME_ACCESS)

handle, err := WslLaunch(distributionName, command, useCurrentWorkingDirectory, stdin, stdout, stderr)
syscall.WaitForSingleObject(handle, syscall.INFINITE)
syscall.GetExitCodeProcess(handle, &exitCode)
return
}

//sys _WslUnregisterDistribution(distributionName *uint16) (res error) = wslapi.WslUnregisterDistribution

// WslUnregisterDistribution unregisters the specified distribution
func WslUnregisterDistribution(distributionName string) error {
pDistributionName, _ := syscall.UTF16PtrFromString(distributionName)
return _WslUnregisterDistribution(pDistributionName)
}
99 changes: 99 additions & 0 deletions zsyscall_windows.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit 9ff1d6e

Please sign in to comment.