diff --git a/sshsvc/sshsvc.go.1 b/sshsvc/sshsvc.go.1 deleted file mode 100644 index f606fd8c..00000000 --- a/sshsvc/sshsvc.go.1 +++ /dev/null @@ -1,251 +0,0 @@ -package main - -import ( - "fmt" - "io" - "log" - "net" - "os" - "os/exec" - "strings" - - "golang.org/x/crypto/ssh" - "golang.org/x/crypto/ssh/terminal" -) - -func main() { - // 生成SSH密钥对 - privateKeyBytes, err := os.ReadFile("../.ssh/id_rsa") - if err != nil { - log.Fatal("无法读取私钥文件：", err) - } - - privateKey, err := ssh.ParsePrivateKey(privateKeyBytes) - if err != nil { - log.Fatal("无法解析私钥：", err) - } - - // 配置SSH服务器 - config := &ssh.ServerConfig{ - PasswordCallback: func(conn ssh.ConnMetadata, password []byte) (*ssh.Permissions, error) { - // 这里可以进行密码验证逻辑，例如检查用户名和密码是否匹配 - if conn.User() == "admin" && string(password) == "123456" { - return nil, nil - } - return nil, fmt.Errorf("密码错误") - }, - PublicKeyCallback: func(conn ssh.ConnMetadata, key ssh.PublicKey) (*ssh.Permissions, error) { - // 这里可以进行公钥验证逻辑，例如检查用户的公钥是否在允许的公钥列表中 - return nil, fmt.Errorf("公钥验证失败") - }, - } - - config.AddHostKey(privateKey) - - // 启动SSH服务器 - listener, err := net.Listen("tcp", "192.168.2.119:2222") - if err != nil { - log.Fatal("无法监听端口：", err) - } - - log.Println("SSH服务器已启动，监听端口：2222") - - for { - conn, err := listener.Accept() - if err != nil { - log.Fatal("无法接受连接：", err) - } - - go handleSSHConnection(conn, config) - } -} - -func handleSSHConnection(conn net.Conn, config *ssh.ServerConfig) { - // SSH握手 - sshConn, chans, reqs, err := ssh.NewServerConn(conn, config) - if err != nil { - log.Println("SSH握手失败：", err) - return - } - - log.Printf("SSH连接已建立，客户端版本：%s，用户：%s", sshConn.ClientVersion(), sshConn.User()) - - // 处理SSH请求 - go ssh.DiscardRequests(reqs) - - // 处理SSH通道 - for newChannel := range chans { - if newChannel.ChannelType() != "session" { - newChannel.Reject(ssh.UnknownChannelType, "不支持的通道类型") - continue - } - - channel, requests, err := newChannel.Accept() - if err != nil { - log.Println("无法接受通道：", err) - continue - } - - go handleSSHChannel(channel, requests) - } -} - -func handleSSHChannel(channel ssh.Channel, requests <-chan *ssh.Request) { - for req := range requests { - switch req.Type { - case "exec": - // 执行远程命令 - cmd := exec.Command("cmd", "/c", strings.TrimSpace(string(req.Payload))) - cmd.Stdin = channel - cmd.Stdout = channel - cmd.Stderr = channel.Stderr() - log.Println("cmd:", strings.TrimSpace(string(req.Payload))) - - err := cmd.Run() - if err != nil { - log.Println("无法执行命令：", err) - } - - channel.SendRequest("exit-status", false, []byte{0, 0, 0, 0}) - channel.Close() - case "shell": - // // 处理交互式shell会话 - // // 在这里添加您的处理逻辑，例如启动一个shell进程并将其连接到channel - // // 请注意，处理交互式shell会话需要更复杂的逻辑，您可能需要使用类似于pty包来处理终端相关的操作 - // channel.Write([]byte("交互式shell会话已启动\n")) - // channel.Close() - handleSSHShell(channel) - default: - req.Reply(false, nil) - } - } -} - -func handleSSHShell(channel ssh.Channel) { - defer channel.Close() - - // 检查通道是否支持终端 - term := terminal.NewTerminal(channel, "> ") - - // 启动交互式shell会话 - for { - line, err := term.ReadLine() - if err != nil { - if err == io.EOF { - break - } - log.Println("Failed to read line: ", err) - break - } - - // 在这里处理输入的命令 - // 您可以根据需要添加更多的逻辑 - if strings.TrimSpace(line) == "exit" { - break - } - - response := fmt.Sprintf("You entered: %s\n", line) - term.Write([]byte(response)) - } -} - -// func main() { -// // 读取 SSH 私钥文件 -// privateKeyBytes, err := os.ReadFile("../.ssh/private_key.pem") -// if err != nil { -// log.Fatalf("Failed to load private key: %v", err) -// } - -// // 解析私钥 -// privateKey, err := ssh.ParsePrivateKey(privateKeyBytes) -// if err != nil { -// log.Fatalf("Failed to parse private key: %v", err) -// } - -// // 配置 SSH 服务器 -// config := &ssh.ServerConfig{ -// PasswordCallback: func(conn ssh.ConnMetadata, password []byte) (*ssh.Permissions, error) { -// // 在此处验证密码 -// if string(password) == "123456" { -// return nil, nil -// } -// return nil, fmt.Errorf("password rejected for %s", conn.User()) -// }, -// } - -// // 添加私钥到配置 -// config.AddHostKey(privateKey) - -// // 启动 SSH 服务器 -// listener, err := net.Listen("tcp", "0.0.0.0:2222") -// if err != nil { -// log.Fatalf("Failed to listen on 2222: %v", err) -// } - -// log.Println("SSH server listening on 0.0.0.0:2222") - -// for { -// conn, err := listener.Accept() -// if err != nil { -// log.Fatalf("Failed to accept incoming connection: %v", err) -// } - -// // 处理 SSH 连接 -// go func() { -// sshConn, chans, reqs, err := ssh.NewServerConn(conn, config) -// if err != nil { -// log.Printf("Failed to handshake: %v", err) -// return -// } - -// log.Printf("SSH connection established from %s", sshConn.RemoteAddr()) - -// // 处理 SSH 请求 -// go ssh.DiscardRequests(reqs) - -// // 处理 SSH 通道 -// for newChannel := range chans { -// if newChannel.ChannelType() != "session" { -// newChannel.Reject(ssh.UnknownChannelType, "unknown channel type") -// continue -// } - -// channel, requests, err := newChannel.Accept() -// if err != nil { -// log.Printf("Failed to accept channel: %v", err) -// continue -// } - -// // 处理 SSH 会话 -// go func(in <-chan *ssh.Request) { -// for req := range in { -// log.Printf("Received out-of-band request: %v", string(req)) -// } -// }(requests) - -// // 示例：执行远程命令 -// go func() { -// defer channel.Close() - -// command := "echo 'Hello, SSH!'" -// output, err := execCommand(command) -// if err != nil { -// log.Printf("Failed to execute command: %v", err) -// return -// } - -// _, err = channel.Write([]byte(output)) -// if err != nil { -// log.Printf("Failed to write output: %v", err) -// return -// } -// }() -// } -// }() -// } -// } - -// func execCommand(command string) (string, error) { -// // 在此处执行远程命令并返回结果 -// return "hello ssh", nil -// } diff --git a/sshsvc/sshsvc.go.ok b/sshsvc/sshsvc.go.ok deleted file mode 100644 index b0b3f038..00000000 --- a/sshsvc/sshsvc.go.ok +++ /dev/null @@ -1,77 +0,0 @@ -package main - -import ( - "log" - "strings" - - "github.com/gliderlabs/ssh" - "golang.org/x/crypto/ssh/terminal" -) - -// func main() { -// ssh.Handle(func(s ssh.Session) { -// io.WriteString(s, "Hello world\n") -// }) - -// log.Fatal(ssh.ListenAndServe(":2222", nil, ssh.HostKeyFile("../.ssh/id_rsa"))) -// } - -// func main() { -// ssh.ListenAndServe(":2222", nil, ssh.HostKeyFile("../.ssh/id_rsa"), -// ssh.PasswordAuth(func(ctx ssh.Context, pass string) bool { -// return pass == "123456" -// }), -// ) -// } - -func main() { - ssh.Handle(func(s ssh.Session) { - - term := terminal.NewTerminal(s, "> ") - for { - line, err := term.ReadLine() - if err != nil { - break - } - // 在这里处理输入的命令 - // 您可以根据需要添加更多的逻辑 - if strings.TrimSpace(line) == "exit" { - break - } - - log.Println(line) - if line != "" { - term.Write(append([]byte(line), '\n')) - } - } - log.Println("terminal closed") - }) - - log.Println("starting ssh server on port 2222...") - ssh.ListenAndServe(":2222", nil, ssh.HostKeyFile("../.ssh/id_rsa"), - ssh.PasswordAuth(func(ctx ssh.Context, pass string) bool { - return pass == "123456" - }), - ) -} - -func Handle(sess ssh.Session) { - term := terminal.NewTerminal(sess, "> ") - for { - line, err := term.ReadLine() - if err != nil { - break - } - // 在这里处理输入的命令 - // 您可以根据需要添加更多的逻辑 - if strings.TrimSpace(line) == "exit" { - break - } - - log.Println(line) - if line != "" { - term.Write(append([]byte(line), '\n')) - } - } - log.Println("terminal closed") -}