最终版本

go/go.mod

@@ -2,7 +2,10 @@ module ivampiresp.com/rCMD
 
 go 1.21.0
 
-require github.com/r3labs/sse/v2 v2.10.0
+require (
+	github.com/r3labs/sse/v2 v2.10.0
+	golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a
+)
 
 require (
 	golang.org/x/net v0.0.0-20191116160921-f9c825593386 // indirect

go/go.sum

@@ -1,15 +1,20 @@
+github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/r3labs/sse/v2 v2.10.0 h1:hFEkLLFY4LDifoHdiCN/LlGBAdVJYsANaLqNYa1l/v0=
 github.com/r3labs/sse/v2 v2.10.0/go.mod h1:Igau6Whc+F17QUgML1fYe1VPZzTV6EMCnYktEmkNJ7I=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/net v0.0.0-20191116160921-f9c825593386 h1:ktbWvQrW08Txdxno1PiDpSxPXG6ndGsfnJjRRtkM0LQ=
 golang.org/x/net v0.0.0-20191116160921-f9c825593386/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
+golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a h1:1BGLXjeY4akVXGgbC9HugT3Jv3hCI0z56oJR5vAMgBU=
 golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 gopkg.in/cenkalti/backoff.v1 v1.1.0 h1:Arh75ttbsvlpVA7WtVpH4u9h6Zl46xuptxqLxPiSo4Y=
 gopkg.in/cenkalti/backoff.v1 v1.1.0/go.mod h1:J6Vskwqd+OMVJl8C33mmtxTBs2gyzfv7UDAkHu8BrjI=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

go/start.bat

@@ -0,0 +1,7 @@
+rem set env
+set "DEVICE_ID=a"
+
+rem run win.exe at the same path
+win.exe
+
+pause

go/win.go

@@ -1,8 +1,11 @@
 package main
 
 import (
+	"encoding/json"
 	"fmt"
+	"golang.org/x/sys/windows"
 	"os"
+	"os/exec"
 
 	"github.com/r3labs/sse/v2"
 )
@@ -10,21 +13,109 @@ import (
 func main() {
 	println("Hello, World!")
 
-	device_id := os.Getenv("DEVICE_ID")
+	deviceId := os.Getenv("DEVICE_ID")
-	if device_id == "" {
+	if deviceId == "" {
 		println("DEVICE_ID env is not set")
 		os.Exit(1)
 	}
 
 	for {
-		client := sse.NewClient("https://ivampiresp.com/wp-content/rCMD/sse.php?device=" + device_id)
+		fmt.Println("开始连接。")
+		client := sse.NewClient("https://ivampiresp.com/wp-content/rCMD/sse.php?device=" + deviceId)
 
-		client.Subscribe("message", func(msg *sse.Event) {
+		err := client.Subscribe("message", func(msg *sse.Event) {
-			fmt.Println(msg.Data)
+			//fmt.Println(msg.Data)
 
-			// json
+			// byte to json
+			var data map[string]interface{}
+			err := json.Unmarshal(msg.Data, &data)
+
+			//fmt.Println(data)
+
+			if err != nil {
+				fmt.Println(err)
+				return
+			}
+
+			cmd := data["cmd"]
+			param := data["param"]
+
+			if cmd == nil {
+				fmt.Println("cmd or params is nil")
+				return
+			}
+
+			// interface to string
+			cmdStr := fmt.Sprintf("%v", cmd)
+			paramStr := fmt.Sprintf("%v", param)
+
+			fmt.Println(cmdStr)
+			fmt.Println(paramStr)
+
+			fmt.Println("参数是: " + paramStr)
+
+			// 获取当前用户目录
+			home, err := os.UserHomeDir()
+			if err != nil {
+				fmt.Println(err)
+				return
+			}
+
+			// 切换到用户目录
+			err = os.Chdir(home)
+			if err != nil {
+				fmt.Println(err)
+				return
+			}
+
+			// 快捷命令，比如 lock
+			canExec := quickCommand(cmdStr)
+			if canExec {
+				return
+			}
+
+			fmt.Println("命令: " + cmdStr + " " + paramStr)
+			cmdExec := exec.Command(cmdStr, paramStr)
+			err = cmdExec.Run()
+			if err != nil {
+				fmt.Println(err)
+				return
+			}
+			fmt.Println(cmdExec.Output())
+			//fmt.Println(cmdExec.Stdout)
+			//fmt.Println(cmdExec.Stderr)
+			//fmt.Println(cmdExec.ProcessState)
+			fmt.Println("执行完成。")
 
 		})
+		if err != nil {
+			fmt.Println(err)
+			// 下个循环重新连接
+			continue
+		}
 	}
 
 }
+
+func lockWorkStation() error {
+	user32 := windows.NewLazySystemDLL("user32.dll")
+	lockWorkStation := user32.NewProc("LockWorkStation")
+
+	ret, _, err := lockWorkStation.Call()
+	if ret == 0 {
+		return err
+	}
+	return nil
+}
+
+func quickCommand(command string) bool {
+	if command == "lock" {
+		err := lockWorkStation()
+		if err != nil {
+			fmt.Println(err)
+			return false
+		}
+		return true
+	}
+	return false
+}