[init] first commit

Author: siriusa51

.gitignore

@@ -0,0 +1,31 @@
+# If you prefer the allow list template instead of the deny list, see community template:
+# https://github.com/github/gitignore/blob/main/community/Golang/Go.AllowList.gitignore
+#
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/
+
+# Go workspace file
+go.work
+go.work.sum
+
+# env file
+.env
+
+build/
+.idea
+.vscode
+
+.DS_Store

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Siri
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -0,0 +1,100 @@
+# SSHProxy
+An SSH proxy command proxy program implemented using WebSocket.
+
+```mermaid
+graph LR
+    classDef startend fill:#F5EBFF,stroke:#BE8FED,stroke-width:2px
+    classDef process fill:#E5F6FF,stroke:#73A6FF,stroke-width:2px
+    
+    A([Client]):::startend -->|ssh proxycommand by sshproxy| B(sshproxy - server):::process
+    B --> C([SSH Server]):::startend
+```
+
+## Usage
+
+1. Navigate to the release page to download the executable file corresponding to your system and then extract it.
+
+2. Launch the sshproxy proxy server.
+
+   ```shell
+   sshproxy server
+   ```
+
+3. Use SSH to access the target SSH server via the proxy server.
+
+4. ```shell
+   ssh -o ProxyCommand='sshproxy client --addr=%h:%p --ws=ws://localhost:8080/ --token=token' <user>@<host>:<port>
+   ```
+
+   Global commands:
+
+   ```shell
+   $ sshproxy -h
+   
+   sshproxy is a WebSocket proxy for SSH.
+   
+   Usage:
+     sshproxy [flags]
+     sshproxy [command]
+   
+   Available Commands:
+     client      Run the sshproxy client.
+     completion  Generate the autocompletion script for the specified shell.
+     help        Get help about any command.
+     server      Run the sshproxy server.
+   
+   Flags:
+     -h, --help   Get help for sshproxy.
+   
+   Use "sshproxy [command] --help" for more information about a command.
+   ```
+
+   Client commands:
+
+   ```shell
+   $ sshproxy client -h
+   
+   The sshproxy client is a WebSocket-based client for sshproxy. It connects to a WebSocket server and forwards data to the SSH server. 
+   For example: ssh -o ProxyCommand='sshproxy --addr=%h:%p --ws=ws://localhost:8080/' user@localhost:22
+   
+   Usage:
+     sshproxy client [flags]
+   
+   Flags:
+         --addr string    The target SSH address, e.g., --addr=%h:%p
+     -h, --help           Get help for the client.
+         --sid string     The session ID. If not provided, a random one will be generated.
+         --token string   The token for authorization.
+         --ws string      The WebSocket URL (e.g., ws://localhost:8080/).
+   ```
+
+   Server commands:
+
+   ```shell
+   $ sshproxy server -h
+   
+   The sshproxy server is a WebSocket-based server for sshproxy. It listens on a port and forwards data to the SSH server.
+   
+   Usage:
+     sshproxy server [flags]
+   
+   Flags:
+     -h, --help            Get help for the server.
+         --host string     The host to listen on (default is "localhost").
+         --port int        The port to listen on (default is 8080).
+         --prefix string   The prefix path, e.g., /ssh.
+         --token string    The token for authorization.
+   ```
+
+## Building
+The framework employed in the building process is: https://taskfile.dev/
+
+The following command is used to build all binary packages:
+
+```shell
+task build
+```
+
+## Troubleshooting
+
+If you find that the SSH connection fails, you can add `-v` after the SSH command to view detailed logs.

agent/agent.go

@@ -0,0 +1,36 @@
+package agent
+
+import (
+	"fmt"
+	"log/slog"
+	"net"
+)
+
+type Agent struct {
+	conn net.Conn
+	log  *slog.Logger
+}
+
+func New(addr string, log *slog.Logger) (*Agent, error) {
+	conn, err := net.Dial("tcp", addr)
+	if err != nil {
+		return nil, fmt.Errorf("failed to connect to SSH server: %v", err)
+	}
+
+	return &Agent{
+		conn: conn,
+		log:  log.With("module", "sshproxy/agent"),
+	}, nil
+}
+
+func (c *Agent) Read(p []byte) (int, error) {
+	return c.conn.Read(p)
+}
+
+func (c *Agent) Write(p []byte) (int, error) {
+	return c.conn.Write(p)
+}
+
+func (c *Agent) Close() error {
+	return c.conn.Close()
+}