使用golang写一个简单的agent

package main

import (
    "flag"
    "fmt"
    "log"
    "net/http"
    "os/exec"
    "strings"
    "time"
)

const expectedParams = 4

func handler(password, scriptPath string) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        log.Println("Received HTTP request!")
        query := r.URL.RawQuery
        parts := strings.Split(query, "?")
        if len(parts) != expectedParams+1 { // 1 for password, 4 for parameters
            http.Error(w, fmt.Sprintf("Expected exactly %d parameters", expectedParams), http.StatusBadRequest)
            return
        }

        passwd := parts[0]
        cmdParts := parts[1:]


        if passwd != password {
            http.Error(w, "Incorrect password!", http.StatusUnauthorized)
            return
        }

        cmd := exec.Command("bash", "-c", fmt.Sprintf("cd %s && ./%s %s %s %s", scriptPath, cmdParts[0], cmdParts[1], cmdParts[2], cmdParts[3]))

        out, err := cmd.CombinedOutput()
        if err != nil {
            http.Error(w, fmt.Sprintf("Command execution error: %v", err), http.StatusInternalServerError)
            return
        }

        w.WriteHeader(http.StatusOK)
        w.Write(out)
    }
}

func main() {
    // 定义命令行参数
    port := flag.String("port", "8888", "The port to listen on")
    pass := flag.String("password", "", "The password for authentication")
    scriptPath := flag.String("scriptPath", "/opt/dba_scripts/", "The path to the script directory")
    flag.Parse()

    if *pass == "" {
        log.Fatal("Password must be provided")
    }

    http.HandleFunc("/", handler(*pass, *scriptPath))
    serverAddr := fmt.Sprintf("0.0.0.0:%s", *port)
    s := &http.Server{
        Addr:           serverAddr,
        ReadTimeout:    10 * time.Second,
        WriteTimeout:   10 * time.Second,
        MaxHeaderBytes: 1 << 20,
    }

    log.Println(time.Now().Format(time.RFC1123), "Server Starts -", serverAddr)
    err := s.ListenAndServe()
    if err != nil {
        log.Fatal(err)
    }
    log.Println(time.Now().Format(time.RFC1123), "Server Stops -", serverAddr)
}

go build 
./main -port=8888 -password=123456 -scriptPath=/opt/

1、编写测试python脚本
    # 注意：
    # 如果是python脚本的话，需要在第一行定义python的版本。
    # 如果是shell脚本，需要在第一行定义#!/bin/bash字样。
    
    cat aaa.py   内容如下：
    #!/usr/bin/python3
    print("aaaaaaaaaaaaa")


2、需要先确保文件是可执行权限的
     chmod +x aaa.py 


3、 调用
    curl '127.0.0.1:8888/?123456?aaa.py?arg1?arg2?arg3'
    直接在屏幕上输出了 aaaaaaaaaaaaa


其它例子:
    带参数的例子
        curl '127.0.0.1:8888/?123456?test2.sh?db1?tb1?size'
        curl '127.0.0.1:8888/?123456?test2.sh?db1?tb1?' 
    
    不带参数的例子（注意确保参数是4个即可，不足的补齐）：
        curl '127.0.0.1:8888/?123456?test2.sh???'