mattburchett/go_tab-magic
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Adding some more debug options

Browse Source
This commit is contained in:
Matt Burchett 2019-01-14 12:39:13 -06:00
parent d98ac3058e
commit bbd1680f1e
3 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/main.go
View File

@ -34,5 +34,5 @@ func main() {
}
data := resolver.PerformZoneTransfer(cfg, debug)
shell.CreateShellAliases(data, user, cfg)
shell.CreateShellAliases(data, user, cfg, debug)
}

9
pkg/resolver/resolver.go
View File

@ -12,7 +12,10 @@ import (
)
// LookupName returns IPv4 address from A record or error.
func lookupName(fqdn, serverAddr string) (string, error) {
func lookupName(fqdn, serverAddr string, debug bool) (string, error) {
if debug {
defer utils.LogElapsedTime("Lookup Name")
}
m := &dns.Msg{}
m.SetQuestion(dns.Fqdn(fqdn), dns.TypeA)
in, err := dns.Exchange(m, serverAddr)
@ -52,7 +55,7 @@ func PerformZoneTransfer(config config.Config, debug bool) []string {
switch v := a.(type) {
case *dns.TXT:
hostname = v.Hdr.Name
cip, err := lookupName(strings.TrimRight(v.Hdr.Name, "."), server)
cip, err := lookupName(strings.TrimRight(v.Hdr.Name, "."), server, debug)
if err != nil || cip == "" {
continue
}
@ -61,7 +64,7 @@ func PerformZoneTransfer(config config.Config, debug bool) []string {
ip = v.A.String()
hostname = v.Hdr.Name
case *dns.CNAME:
cip, err := lookupName(v.Target, server)
cip, err := lookupName(v.Target, server, debug)
if err != nil || cip == "" {
continue
}

11
pkg/shell/shell.go
View File

@ -2,13 +2,17 @@ package shell
import (
"fmt"
"rfi-sower/pkg/utils"
"strings"
"git.linuxrocker.com/mattburchett/go_tab-magic/pkg/config"
)
// CreateShellAliases will create shell aliases and fmt.Println them
func CreateShellAliases(data []string, username string, config config.Config) {
func CreateShellAliases(data []string, username string, config config.Config, debug bool) {
if debug {
defer utils.LogElapsedTime("Create Shell Aliases")
}
for _, i := range data {
splitStrings := strings.Split(i, " ")
hostname := splitStrings[0]
@ -63,8 +67,9 @@ func CreateShellAliases(data []string, username string, config config.Config) {
remoteUser = user
}
}
if useJump {
if debug {
fmt.Println("Disabled Output due to Debug Mode")
} else if useJump {
fmt.Printf("alias %v=\\'%v%v%v %v@%v \"%v %v %v %v %v\"'\n", host, message, prerac, hop, username, jump, rac, racOpts, remoteUser, fqdn, sudo)
} else {
fmt.Printf("alias %v=\\'%v %v %v %v %v %v\n", host, message, rac, racOpts, remoteUser, fqdn, sudo)