147 lines
3.1 KiB
Go
Executable File
147 lines
3.1 KiB
Go
Executable File
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"cidebug/pkg/iowrappers"
|
|
"fmt"
|
|
"github.com/gorilla/websocket"
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"os/exec"
|
|
"syscall"
|
|
"unsafe"
|
|
|
|
"github.com/creack/pty"
|
|
"github.com/gliderlabs/ssh"
|
|
"github.com/hashicorp/yamux"
|
|
"github.com/pkg/sftp"
|
|
)
|
|
|
|
func SftpHandler(sess ssh.Session) {
|
|
debugStream := io.Discard
|
|
serverOptions := []sftp.ServerOption{
|
|
sftp.WithDebug(debugStream),
|
|
}
|
|
server, err := sftp.NewServer(
|
|
sess,
|
|
serverOptions...,
|
|
)
|
|
if err != nil {
|
|
log.Printf("sftp tcpserver init error: %s\n", err)
|
|
return
|
|
}
|
|
if err := server.Serve(); err == io.EOF {
|
|
server.Close()
|
|
fmt.Println("sftp client exited session.")
|
|
} else if err != nil {
|
|
fmt.Println("sftp tcpserver completed with error:", err)
|
|
}
|
|
}
|
|
|
|
func passwordAuth(ctx ssh.Context, password string) bool {
|
|
// Replace with your own logic to validate username and password
|
|
return ctx.User() == "abc" && password == "123"
|
|
}
|
|
|
|
func setWinsize(f *os.File, w, h int) {
|
|
syscall.Syscall(syscall.SYS_IOCTL, f.Fd(), uintptr(syscall.TIOCSWINSZ),
|
|
uintptr(unsafe.Pointer(&struct{ h, w, x, y uint16 }{uint16(h), uint16(w), 0, 0})))
|
|
}
|
|
|
|
func sshServer() *ssh.Server {
|
|
ssh.Handle(func(s ssh.Session) {
|
|
cmd := exec.Command("bash")
|
|
ptyReq, winCh, isPty := s.Pty()
|
|
if isPty {
|
|
cmd.Env = append(os.Environ(), fmt.Sprintf("TERM=%s", ptyReq.Term))
|
|
f, err := pty.Start(cmd)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
go func() {
|
|
for win := range winCh {
|
|
setWinsize(f, win.Width, win.Height)
|
|
}
|
|
}()
|
|
go func() {
|
|
io.Copy(f, s) // stdin
|
|
}()
|
|
io.Copy(s, f) // stdout
|
|
cmd.Wait()
|
|
} else {
|
|
io.WriteString(s, "No PTY requested.\n")
|
|
s.Exit(1)
|
|
}
|
|
})
|
|
|
|
log.Println("starting ssh server")
|
|
server := ssh.Server{
|
|
//Addr: ":2222",
|
|
PasswordHandler: passwordAuth,
|
|
SubsystemHandlers: map[string]ssh.SubsystemHandler{
|
|
"sftp": SftpHandler,
|
|
},
|
|
}
|
|
return &server
|
|
}
|
|
|
|
func echoServer(conn io.ReadWriter) {
|
|
io.Copy(conn, conn)
|
|
}
|
|
|
|
func netCatServer(conn io.ReadWriter) {
|
|
stdio := bufio.NewReadWriter(
|
|
bufio.NewReaderSize(os.Stdin, 0),
|
|
bufio.NewWriterSize(os.Stdout, 0))
|
|
iowrappers.SynchronizeStreams(conn, stdio)
|
|
}
|
|
|
|
type AgentService interface {
|
|
Run(listener net.Listener)
|
|
}
|
|
|
|
type ListenerServer func() *ssh.Server
|
|
|
|
func (server ListenerServer) Run(listener net.Listener) {
|
|
server().Serve(listener)
|
|
}
|
|
|
|
type ConnectionServer func(conn io.ReadWriter)
|
|
|
|
func (server ConnectionServer) Run(listener net.Listener) {
|
|
for {
|
|
conn, err := listener.Accept()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
go echoServer(conn)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
wsURL := os.Args[1]
|
|
conn, _, err := websocket.DefaultDialer.Dial(wsURL, nil)
|
|
if err != nil {
|
|
log.Println("WebSocket connection error:", err)
|
|
return
|
|
}
|
|
wsConn := iowrappers.NewWebSocketConn(conn)
|
|
defer wsConn.Close()
|
|
|
|
listener, err := yamux.Client(wsConn, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
log.Println("Connection established to rendez-vous server, waiting for debug sessions")
|
|
|
|
var service AgentService
|
|
service = ListenerServer(sshServer)
|
|
//service = ConnectionServer(echoServer)
|
|
//service := ConnectionServer(netCatServer)
|
|
service.Run(listener)
|
|
|
|
}
|