maride/pancap
1
0
Fork 0
mirror of https://github.com/maride/pancap.git synced 2024-10-18 12:06:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Avoid printing non-empty, but vacuous blocks

Browse Source
This commit is contained in:
maride 2020-12-14 14:27:49 +01:00
parent ad42bc22ed
commit 1c23ca3859
2 changed files with 33 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
protocol/dhcpv4/network.go
View File

@ -24,14 +24,21 @@ var (
// Generates the summary of relevant DHCP options // Generates the summary of relevant DHCP options
func (p *Protocol) generateNetworkSummary() string { func (p *Protocol) generateNetworkSummary() string {
// It's also possible to use strings.Builder here, but it produces code which is longer than this solution :shrug: subnetMask, subnetAvail := formatIP(p.networkSetup[layers.DHCPOptSubnetMask])
summary := fmt.Sprintf("Subnet Mask: %s\n", formatIP(p.networkSetup[layers.DHCPOptSubnetMask])) broadcastAddr, broadcastAvail := formatIP(p.networkSetup[layers.DHCPOptBroadcastAddr])
summary = fmt.Sprintf("%sBroadcast: %s\n", summary, formatIP(p.networkSetup[layers.DHCPOptBroadcastAddr])) routerAddr, routerAvail := formatIP(p.networkSetup[layers.DHCPOptRouter])
summary = fmt.Sprintf("%sRouter: %s\n", summary, formatIP(p.networkSetup[layers.DHCPOptRouter])) dnsAddr, dnsAvail := formatIP(p.networkSetup[layers.DHCPOptDNS])
summary = fmt.Sprintf("%sDNS Server: %s\n", summary, formatIP(p.networkSetup[layers.DHCPOptDNS])) ntpAddr, ntpAvail := formatIP(p.networkSetup[layers.DHCPOptNTPServers])
summary = fmt.Sprintf("%sNTP Server: %s\n", summary, formatIP(p.networkSetup[layers.DHCPOptNTPServers])) leaseTime, leaseAvail := formatDate(p.networkSetup[layers.DHCPOptLeaseTime])
summary = fmt.Sprintf( "%sLease Time: %s\n", summary, formatDate(p.networkSetup[layers.DHCPOptLeaseTime])) renewalTime, renewalAvail := formatDate(p.networkSetup[layers.DHCPOptT1])
summary = fmt.Sprintf("%sRenewal Time: %s\n", summary, formatDate(p.networkSetup[layers.DHCPOptT1]))
// Check if there even are any values
if !subnetAvail && !broadcastAvail && !routerAvail && !dnsAvail && !ntpAvail && !leaseAvail && !renewalAvail {
// No, do not return any summary. This will lead to a collapsed section.
return ""
}
summary := fmt.Sprintf("Subnet Mask: %s\nBroadcast: %s\nRouter: %s\nDNS Server: %s\nNTP Server: %s\nLease Time: %s\nRenewal Time: %s", subnetMask, broadcastAddr, routerAddr, dnsAddr, ntpAddr, leaseTime, renewalTime)
return summary return summary
} }
@ -92,24 +99,27 @@ func isRelevantOption(opt layers.DHCPOption) bool {
} }
// Formats the given byte array as string representing the IP address, or returns an error (as string) // Formats the given byte array as string representing the IP address, or returns an error (as string)
func formatIP(rawIP []byte) string { // The returned bool value states whether the IP address could be formatted or not (e.g. "not found")
func formatIP(rawIP []byte) (string, bool) {
// Check if we even have an IP // Check if we even have an IP
if rawIP == nil { if rawIP == nil {
// We don't have an IP, construct an error message (as string) // We don't have an IP, construct an error message (as string)
error := color.New(color.FgRed) error := color.New(color.FgRed)
return error.Sprint("(not found)") return error.Sprint("(not found)"), false
} }
// Return formatted IP // Return formatted IP
return net.IP(rawIP).String() return net.IP(rawIP).String(), true
} }
func formatDate(rawDate []byte) string { // Formats the given byte array as string representing the date, or returns an error (as string)
// The returned bool value states whether the date could be formatted or not (e.g. "not found")
func formatDate(rawDate []byte) (string, bool) {
// Check if we even have a date // Check if we even have a date
if rawDate == nil { if rawDate == nil {
// We don't have a date, construct an error message (as string) // We don't have a date, construct an error message (as string)
error := color.New(color.FgRed) error := color.New(color.FgRed)
return error.Sprint("(not found)") return error.Sprint("(not found)"), false
} }
// Actually format date // Actually format date
@ -149,5 +159,5 @@ func formatDate(rawDate []byte) string {
} }
} }
return formattedDate return formattedDate, true
} }

10
protocol/dhcpv4/request.go
View File

@ -13,5 +13,13 @@ func (p *Protocol) processRequestPacket(dhcppacket layers.DHCPv4) {
// Generates the summary of all DHCP request packets // Generates the summary of all DHCP request packets
func (p *Protocol) generateRequestSummary() string { func (p *Protocol) generateRequestSummary() string {
return fmt.Sprintf("%d unique DHCP requests\n%s", len(p.requestMAC), common.GenerateTree(p.requestMAC)) reqAmount := len(p.requestMAC)
// Check if there were requests
if reqAmount == 0 {
// No, don't print a summary then.
return ""
}
return fmt.Sprintf("%d unique DHCP requests\n%s", reqAmount, common.GenerateTree(p.requestMAC))
} }