Golang exec.command
Jump to navigation
Jump to search
Windows and Linux
package main import ( "fmt" "os/exec" "os" "strings" "runtime" "regexp" //match, _ := regexp.MatchString("p([a-z]+)ch", "peach") ) func cmd_filter(name string, match_list []string){ allow := false for k, v := range match_list { fmt.Println("At index", k, "value is", v) match, _ := regexp.MatchString(v, name) if match == true { allow = true break } } if allow != true { fmt.Printf("E: Command %v is %v in shellcmd not allowed\n", name, allow) os.Exit(1) } } func shell_cmd(name string, args []string) { cmd := exec.Command(name, args...) out, err := cmd.Output() if err != nil { fmt.Println(err) } fmt.Printf(string(out)) } func shellcmd() { system_os := runtime.GOOS if system_os == "linux" { name := os.Args[2] args := os.Args[3:] match_list := []string{"Get", "Fet"} cmd_filter(name, match_list) shell_cmd(name, args) }else if system_os == "windows" { if len(os.Args) < 3 { fmt.Printf("Usage: %v <cmd> <args>\n", os.Args[0]) fmt.Printf("Example: %v ls -lhat\n", os.Args[0]) os.Exit(3) } var name string = "powershell.exe" match_list := []string{"Get", "Fet"} // match_list := []string{".*", "Get", "Fet"} args := os.Args[2:] s := os.Args[2] fmt.Println(string(s)) cmd_filter(s, match_list) shell_cmd(name, args) }else{ fmt.Printf("OS %v not supported.\n", system_os) } } func main() { if len(os.Args) < 2 { fmt.Printf("Usage: %v <fuction>\n", os.Args[0]) fmt.Printf("Example: %v shellcmd ls -lhat\n", os.Args[0]) os.Exit(3) } function := os.Args[1] if function == "shellcmd" { shellcmd() }else{ fmt.Printf("E: Unsupported function %v\n", function) } }
Windows
package main import( "fmt" "os/exec" "os" "log" ) func pscmd(s string){ args := []string{s} // cmd := exec.Command("powershell.exe", "ls", "-Depth", "1") cmd := exec.Command("powershell.exe", args...) cmd.Stdout = os.Stdout cmd.Stderr = os.Stderr err := cmd.Run() if err != nil { log.Fatalf("cmd.Run() failed with %s\n", err) } fmt.Println("======") } func main(){ pscmd("ls -Depth 1") pscmd("echo 'hi there'") }