使用 shell2http 在 Golang 中通过 HTTP 执行 Shell 命令

shell2http 是一个轻量级的库，允许你将 shell 命令作为 HTTP 服务暴露出来。它非常适合创建简单的 API 端点来执行系统管理任务或自动化脚本。

安装 shell2http

go get github.com/msoap/shell2http

基本使用方法

下面是一个简单的示例，展示如何使用 shell2http 创建一个 HTTP 服务器，通过不同的端点执行不同的 shell 命令：

package main

import (
	"log"
	"net/http"
	
	"github.com/msoap/shell2http"
)

func main() {
	// 创建一个新的 shell2http 实例
	handler := shell2http.NewHandler()
	
	// 添加命令端点
	handler.AddCommand("/date", "date")  // 执行 date 命令
	handler.AddCommand("/disk", "df -h") // 执行 df -h 命令
	handler.AddCommand("/mem", "free -m") // 执行 free -m 命令
	
	// 设置 HTTP 服务器
	http.Handle("/", handler)
	
	log.Println("Starting server on :8080...")
	log.Fatal(http.ListenAndServe(":8080", nil))
}

高级功能

1. 带参数的命令

handler.AddCommand("/greet", "echo Hello, ${1}!")

访问 /greet?name=John 会输出 “Hello, John!”

2. 自定义响应格式

handler.AddCommand("/custom", "echo '{\"status\":\"ok\", \"data\":\"some data\"}'")

3. 限制访问

handler.AddCommandWithAuth("/admin", "sudo some_admin_command", "admin", "secretpassword")

4. 异步执行长时间运行的命令

handler.AddAsyncCommand("/longtask", "sleep 10; echo Done")

完整示例

下面是一个更完整的示例，包含错误处理和日志记录：

package main

import (
	"log"
	"net/http"
	"os"
	
	"github.com/msoap/shell2http"
)

func main() {
	// 创建日志文件
	logFile, err := os.OpenFile("shell2http.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
	if err != nil {
		log.Fatal(err)
	}
	defer logFile.Close()
	
	// 设置日志输出
	logger := log.New(logFile, "shell2http: ", log.LstdFlags)
	
	// 创建 handler 并设置日志
	handler := shell2http.NewHandler()
	handler.SetLogger(logger)
	
	// 添加命令端点
	handler.AddCommand("/date", "date")
	handler.AddCommand("/disk", "df -h")
	handler.AddCommand("/mem", "free -m")
	
	// 带参数的示例
	handler.AddCommand("/greet", "echo Hello, ${1}!")
	
	// 异步命令示例
	handler.AddAsyncCommand("/backup", "tar -czf /tmp/backup.tar.gz /var/www")
	
	// 安全端点
	handler.AddCommandWithAuth("/admin", "sudo systemctl status", "admin", "secret123")
	
	// 自定义 404 处理
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path != "/" {
			http.NotFound(w, r)
			return
		}
		handler.ServeHTTP(w, r)
	})
	
	// 启动服务器
	logger.Println("Starting server on :8080")
	if err := http.ListenAndServe(":8080", nil); err != nil {
		logger.Fatal("Server error: ", err)
	}
}

安全注意事项

1. 输入验证：永远不要直接将用户输入传递给 shell 命令，这可能导致命令注入漏洞
2. 权限控制：确保命令以最小必要权限运行
3. 认证：对敏感操作实施认证
4. 日志记录：记录所有执行的命令和请求来源

替代方案

如果你需要更复杂的控制，可以考虑：

1. 直接使用 os/exec 包
2. 使用 gorilla/mux 等路由库构建自定义处理程序
3. 对于生产环境，考虑使用更完整的框架如 Gin 或 Echo

shell2http 最适合简单的管理任务和原型开发，对于生产环境中的复杂需求，可能需要更完整的解决方案。