Difference between revisions of "Golang exec.command"
Jump to navigation
Jump to search
| (2 intermediate revisions by the same user not shown) | |||
| Line 7: | Line 7: | ||
"os/exec" | "os/exec" | ||
"os" | "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) | ||
| + | } | ||
| + | } | ||
| Line 18: | Line 37: | ||
fmt.Println(err) | fmt.Println(err) | ||
} | } | ||
| − | + | fmt.Printf(string(out)) | |
} | } | ||
| Line 26: | Line 45: | ||
if system_os == "linux" { | if system_os == "linux" { | ||
name := os.Args[2] | name := os.Args[2] | ||
| − | + | args := os.Args[3:] | |
| + | match_list := []string{"Get", "Fet"} | ||
| + | cmd_filter(name, match_list) | ||
shell_cmd(name, args) | shell_cmd(name, args) | ||
}else if system_os == "windows" { | }else if system_os == "windows" { | ||
| Line 34: | Line 55: | ||
os.Exit(3) | 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) | shell_cmd(name, args) | ||
}else{ | }else{ | ||
Latest revision as of 04:49, 16 November 2020
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'")
}