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

Various changes.

Browse Source
This commit is contained in:
Matt Burchett 2019-01-14 12:31:46 -06:00
parent d62f45cc10
commit 43ed2ea438
3 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/model/model.go
View File

@ -1,11 +1,13 @@
package model package model
// Results is for DNS results
type Results struct { type Results struct {
IP string `json:"ip"` IP string `json:"ip"`
Hostname string `json:"hostname"` Hostname string `json:"hostname"`
TXT string `json:"TXT` TXT string `json:"TXT"`
} }
// UniqResults is for Unique DNS Results
type UniqResults struct { type UniqResults struct {
IP string `json:"ip"` IP string `json:"ip"`
Hostname string `json:"hostname"` Hostname string `json:"hostname"`

2
pkg/resolver/resolver.go
View File

@ -28,6 +28,7 @@ func lookupName(fqdn, serverAddr string) (string, error) {
return "", errors.New("no A record returned") return "", errors.New("no A record returned")
} }
// PerformZoneTransfer performs zone transfers and gathers a list from config.Domains
func PerformZoneTransfer(config config.Config) []string { func PerformZoneTransfer(config config.Config) []string {
data := make([]string, 0) data := make([]string, 0)
@ -46,7 +47,6 @@ func PerformZoneTransfer(config config.Config) []string {
var ip, hostname, txt string var ip, hostname, txt string
switch v := a.(type) { switch v := a.(type) {
case *dns.TXT: case *dns.TXT:
txt = string(v.Txt[0])
hostname = v.Hdr.Name hostname = v.Hdr.Name
cip, err := lookupName(strings.TrimRight(v.Hdr.Name, "."), server) cip, err := lookupName(strings.TrimRight(v.Hdr.Name, "."), server)
if err != nil || cip == "" { if err != nil || cip == "" {

5
pkg/shell/shell.go
View File

@ -7,6 +7,7 @@ import (
"git.linuxrocker.com/mattburchett/go_tab-magic/pkg/config" "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) {
for _, i := range data { for _, i := range data {
splitStrings := strings.Split(i, " ") splitStrings := strings.Split(i, " ")
@ -40,7 +41,7 @@ func CreateShellAliases(data []string, username string, config config.Config) {
for _, i := range txtSplit { for _, i := range txtSplit {
if strings.Contains(i, "SSH_PORT") { if strings.Contains(i, "SSH_PORT") {
port := strings.TrimLeft(i, "SSH_PORT=") port := strings.TrimPrefix(i, "SSH_PORT=")
racOpts = fmt.Sprintf("-AXt -p %v -l", port) racOpts = fmt.Sprintf("-AXt -p %v -l", port)
} else if strings.Contains(i, "OS_FAMILY") { } else if strings.Contains(i, "OS_FAMILY") {
osFamily := strings.Split(i, "=") osFamily := strings.Split(i, "=")
@ -58,7 +59,7 @@ func CreateShellAliases(data []string, username string, config config.Config) {
sudo = "" sudo = ""
} }
} else if strings.Contains(i, "REMOTE_USER") { } else if strings.Contains(i, "REMOTE_USER") {
user := strings.TrimLeft(i, "REMOTE_USER=") user := strings.TrimPrefix(i, "REMOTE_USER=")
remoteUser = user remoteUser = user
} }
} }