Difference between revisions of "Golang exec.command"

From UVOO Tech Wiki
Jump to navigation Jump to search
(Created page with "Windows ``` package main import( "fmt" "os/exec" "os" "log" ) func pscmd(s string){ args := []string{s} // cmd := exec.Command("powershell.exe", "ls", "-Depth", "1...")
 
 
(6 intermediate revisions by the same user not shown)
Line 1: Line 1:
 +
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
 
Windows
 
```
 
```

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

}