KJZH001/nezhahq-agent
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Switch to ConPty on Windows 10 & support windows arm64 platform (#20)

Browse Source
This commit is contained in:
UUBulb
2024-05-21 18:40:08 +08:00
committed by GitHub
parent 060d13a759
commit f882ca3498
8 changed files with 296 additions and 146 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/pty/pty_windows.go
+107 -48
View File
@@ -1,4 +1,4 @@
//go:build windows
//go:build windows && !arm64
package pty
@@ -10,60 +10,94 @@ import (
"os"
"os/exec"
"path/filepath"
"regexp"
"runtime"
"strconv"
"github.com/UserExistsError/conpty"
"github.com/artdarek/go-unzip"
"github.com/iamacarpet/go-winpty"
"github.com/shirou/gopsutil/v3/host"
)
var isWin10 bool
type Pty struct {
tty *winpty.WinPTY
tty interface{}
}
func init() {
isWin10 = VersionCheck()
}
func VersionCheck() bool {
hi, err := host.Info()
if err != nil {
return false
}
re := regexp.MustCompile(`Build (\d+(\.\d+)?)`)
match := re.FindStringSubmatch(hi.PlatformVersion)
if len(match) > 1 {
versionStr := match[1]
version, err := strconv.ParseFloat(versionStr, 64)
if err != nil {
return false
}
return version >= 17763
} else {
return false
}
}
func DownloadDependency() {
executablePath, err := getExecutableFilePath()
if err != nil {
fmt.Println("NEZHA>> wintty 获取文件路径失败", err)
return
}
if !isWin10 {
executablePath, err := getExecutableFilePath()
if err != nil {
fmt.Println("NEZHA>> wintty 获取文件路径失败", err)
return
}
winptyAgentExe := filepath.Join(executablePath, "winpty-agent.exe")
winptyAgentDll := filepath.Join(executablePath, "winpty.dll")
winptyAgentExe := filepath.Join(executablePath, "winpty-agent.exe")
winptyAgentDll := filepath.Join(executablePath, "winpty.dll")
fe, errFe := os.Stat(winptyAgentExe)
fd, errFd := os.Stat(winptyAgentDll)
if errFe == nil && fe.Size() > 300000 && errFd == nil && fd.Size() > 300000 {
return
}
fe, errFe := os.Stat(winptyAgentExe)
fd, errFd := os.Stat(winptyAgentDll)
if errFe == nil && fe.Size() > 300000 && errFd == nil && fd.Size() > 300000 {
return
}
resp, err := http.Get("https://dn-dao-github-mirror.daocloud.io/rprichard/winpty/releases/download/0.4.3/winpty-0.4.3-msvc2015.zip")
if err != nil {
log.Println("NEZHA>> wintty 下载失败", err)
return
}
defer resp.Body.Close()
content, err := io.ReadAll(resp.Body)
if err != nil {
log.Println("NEZHA>> wintty 下载失败", err)
return
}
if err := os.WriteFile("./wintty.zip", content, os.FileMode(0777)); err != nil {
log.Println("NEZHA>> wintty 写入失败", err)
return
}
if err := unzip.New("./wintty.zip", "./wintty").Extract(); err != nil {
fmt.Println("NEZHA>> wintty 解压失败", err)
return
}
arch := "x64"
if runtime.GOARCH != "amd64" {
arch = "ia32"
}
resp, err := http.Get("https://github.com/rprichard/winpty/releases/download/0.4.3/winpty-0.4.3-msvc2015.zip")
if err != nil {
log.Println("NEZHA>> wintty 下载失败", err)
return
}
defer resp.Body.Close()
content, err := io.ReadAll(resp.Body)
if err != nil {
log.Println("NEZHA>> wintty 下载失败", err)
return
}
if err := os.WriteFile("./wintty.zip", content, os.FileMode(0777)); err != nil {
log.Println("NEZHA>> wintty 写入失败", err)
return
}
if err := unzip.New("./wintty.zip", "./wintty").Extract(); err != nil {
fmt.Println("NEZHA>> wintty 解压失败", err)
return
}
arch := "x64"
if runtime.GOARCH != "amd64" {
arch = "ia32"
}
os.Rename("./wintty/"+arch+"/bin/winpty-agent.exe", winptyAgentExe)
os.Rename("./wintty/"+arch+"/bin/winpty.dll", winptyAgentDll)
os.RemoveAll("./wintty")
os.RemoveAll("./wintty.zip")
os.Rename("./wintty/"+arch+"/bin/winpty-agent.exe", winptyAgentExe)
os.Rename("./wintty/"+arch+"/bin/winpty.dll", winptyAgentDll)
os.RemoveAll("./wintty")
os.RemoveAll("./wintty.zip")
}
}
func getExecutableFilePath() (string, error) {
@@ -75,6 +109,8 @@ func getExecutableFilePath() (string, error) {
}
func Start() (*Pty, error) {
var tty interface{}
shellPath, err := exec.LookPath("powershell.exe")
if err != nil || shellPath == "" {
shellPath = "cmd.exe"
@@ -83,24 +119,47 @@ func Start() (*Pty, error) {
if err != nil {
return nil, err
}
tty, err := winpty.OpenDefault(path, shellPath)
if !isWin10 {
tty, err = winpty.OpenDefault(path, shellPath)
} else {
tty, err = conpty.Start(shellPath, conpty.ConPtyWorkDir(path))
}
return &Pty{tty: tty}, err
}
func (pty *Pty) Write(p []byte) (n int, err error) {
return pty.tty.StdIn.Write(p)
if !isWin10 {
return pty.tty.(*winpty.WinPTY).StdIn.Write(p)
} else {
return pty.tty.(*conpty.ConPty).Write(p)
}
}
func (pty *Pty) Read(p []byte) (n int, err error) {
return pty.tty.StdOut.Read(p)
if !isWin10 {
return pty.tty.(*winpty.WinPTY).StdOut.Read(p)
} else {
return pty.tty.(*conpty.ConPty).Read(p)
}
}
func (pty *Pty) Setsize(cols, rows uint32) error {
pty.tty.SetSize(cols, rows)
return nil
if !isWin10 {
pty.tty.(*winpty.WinPTY).SetSize(cols, rows)
return nil
} else {
return pty.tty.(*conpty.ConPty).Resize(int(cols), int(rows))
}
}
func (pty *Pty) Close() error {
pty.tty.Close()
return nil
if !isWin10 {
pty.tty.(*winpty.WinPTY).Close()
return nil
} else {
if err := pty.tty.(*conpty.ConPty).Close(); err != nil {
return err
}
return nil
}
}
pkg/pty/pty_windowsarm.go
+58
View File
@@ -0,0 +1,58 @@
//go:build windows && arm64
package pty
import (
"os"
"os/exec"
"path/filepath"
"github.com/UserExistsError/conpty"
)
type Pty struct {
tty *conpty.ConPty
}
func DownloadDependency() {
}
func getExecutableFilePath() (string, error) {
ex, err := os.Executable()
if err != nil {
return "", err
}
return filepath.Dir(ex), nil
}
func Start() (*Pty, error) {
shellPath, err := exec.LookPath("powershell.exe")
if err != nil || shellPath == "" {
shellPath = "cmd.exe"
}
path, err := getExecutableFilePath()
if err != nil {
return nil, err
}
tty, err := conpty.Start(shellPath, conpty.ConPtyWorkDir(path))
return &Pty{tty: tty}, err
}
func (pty *Pty) Write(p []byte) (n int, err error) {
return pty.tty.Write(p)
}
func (pty *Pty) Read(p []byte) (n int, err error) {
return pty.tty.Read(p)
}
func (pty *Pty) Setsize(cols, rows uint32) error {
return pty.tty.Resize(int(cols), int(rows))
}
func (pty *Pty) Close() error {
if err := pty.tty.Close(); err != nil {
return err
}
return nil
}