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'")
}