e782db30e9
This introduces PostUp and PostDown in dsnet. PostUp and PostDown allow
the user to run arbitrary commands after the device is up or down. These
are typically used to change the firewall rules via iptables. A working
example would be
...
"PostUp" : "iptables -A FORWARD -i dsnet -j ACCEPT; iptables -A FORWARD -o dsnet -j ACCEPT; iptables -t nat -A POSTROUTING -o ens2 -j MASQUERADE ",
"PostDown" : "iptables -D FORWARD -i dsnet -j ACCEPT; iptables -D FORWARD -o dsnet -j ACCEPT; iptables -t nat -D POSTROUTING -o ens2 -j MASQUERADE ",
...
All commands are executed by `/bin/sh` and no filtering or sandboxing is
applied. Users of this should know what they are doing.
Fixes https://github.com/naggie/dsnet/issues/16
80 lines
1.5 KiB
Go
80 lines
1.5 KiB
Go
package dsnet
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
func check(e error, optMsg ...string) {
|
|
if e != nil {
|
|
if len(optMsg) > 0 {
|
|
ExitFail("%s - %s", e, strings.Join(optMsg, " "))
|
|
}
|
|
ExitFail("%s", e)
|
|
}
|
|
}
|
|
|
|
func MustPromptString(prompt string, required bool) string {
|
|
reader := bufio.NewReader(os.Stdin)
|
|
var text string
|
|
var err error
|
|
|
|
for text == "" {
|
|
fmt.Fprintf(os.Stderr, "%s: ", prompt)
|
|
text, err = reader.ReadString('\n')
|
|
check(err)
|
|
text = strings.TrimSpace(text)
|
|
}
|
|
return text
|
|
}
|
|
|
|
func ExitFail(format string, a ...interface{}) {
|
|
fmt.Fprintf(os.Stderr, "\033[31m"+format+"\033[0m\n", a...)
|
|
os.Exit(1)
|
|
}
|
|
|
|
func ShellOut(command string, name string) {
|
|
if command != "" {
|
|
fmt.Printf("Running %s commands:\n %s", name, command)
|
|
shell := exec.Command("/bin/sh", "-c", command)
|
|
err := shell.Run()
|
|
if err != nil {
|
|
ExitFail("%s '%s' failed", name, command, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func ConfirmOrAbort(format string, a ...interface{}) {
|
|
fmt.Fprintf(os.Stderr, format+" [y/n] ", a...)
|
|
|
|
reader := bufio.NewReader(os.Stdin)
|
|
|
|
input, err := reader.ReadString('\n')
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if input == "y\n" {
|
|
return
|
|
} else {
|
|
ExitFail("Aborted.")
|
|
}
|
|
}
|
|
|
|
func BytesToSI(b uint64) string {
|
|
const unit = 1000
|
|
if b < unit {
|
|
return fmt.Sprintf("%d B", b)
|
|
}
|
|
div, exp := int64(unit), 0
|
|
for n := b / unit; n >= unit; n /= unit {
|
|
div *= unit
|
|
exp++
|
|
}
|
|
return fmt.Sprintf("%.1f %cB",
|
|
float64(b)/float64(div), "kMGTPE"[exp])
|
|
}
|