taigrr/wtf
1
0
Fork 0
mirror of https://github.com/taigrr/wtf synced 2025-01-18 04:03:14 -08:00
Code Issues Projects Releases Wiki Activity

ExecuteCommand() moved into the utils dumping ground

Browse Source
This commit is contained in:
Chris Cummer
2018-04-02 08:36:43 -07:00
committed by Chris Cummer
parent de7480b221
commit 0b8a063487
3 changed files with 12 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
security/firewall.go
View File

@@ -1,9 +1,10 @@
package security
import (
"io/ioutil"
"os/exec"
"strings"
"github.com/senorprogrammer/wtf/wtf"
)
const osxFirewallCmd = "/usr/libexec/ApplicationFirewall/socketfilterfw"
@@ -12,39 +13,21 @@ const osxFirewallCmd = "/usr/libexec/ApplicationFirewall/socketfilterfw"
func FirewallState() string {
cmd := exec.Command(osxFirewallCmd, "--getglobalstate")
str := executeCommand(cmd)
str := wtf.ExecuteCommand(cmd)
return str
return status(str)
}
func FirewallStealthState() string {
cmd := exec.Command(osxFirewallCmd, "--getstealthmode")
str := executeCommand(cmd)
str := wtf.ExecuteCommand(cmd)
return str
return status(str)
}
/* -------------------- Unexported Functions -------------------- */
func executeCommand(cmd *exec.Cmd) string {
stdout, err := cmd.StdoutPipe()
if err != nil {
return firewallStr("err")
}
if err := cmd.Start(); err != nil {
return firewallStr("err")
}
var str string
if b, err := ioutil.ReadAll(stdout); err == nil {
str += string(b)
}
return firewallStr(str)
}
func firewallStr(str string) string {
func status(str string) string {
icon := "[red]off[white]"
if strings.Contains(str, "enabled") {