Skip to content

Commit

Permalink
Skip DNS length field (only included over TCP)
Browse files Browse the repository at this point in the history
  • Loading branch information
Alfonso Acosta committed Sep 23, 2016
1 parent 6f95d05 commit 326208c
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion probe/endpoint/dns_snooper_linux_amd64.go
Original file line number Diff line number Diff line change
Expand Up @@ -112,6 +112,7 @@ func (s *DNSSnooper) Stop() {
}

// Gopacket doesn't provide direct support for DNS over TCP, see https://github.com/google/gopacket/issues/236
// TODO: deal with TCP fragmentation and out-of-order segments
type tcpWithDNSSupport struct {
tcp layers.TCP
}
Expand All @@ -126,7 +127,15 @@ func (m *tcpWithDNSSupport) NextLayerType() gopacket.LayerType {
}
return m.tcp.NextLayerType()
}
func (m *tcpWithDNSSupport) LayerPayload() []byte { return m.tcp.LayerPayload() }
func (m *tcpWithDNSSupport) LayerPayload() []byte {
payload := m.tcp.LayerPayload()
// Omit the DNS length field, only included
// in TCP, in order to reuse the DNS UDP parser
if len(payload) > 1 && (m.tcp.SrcPort == 53 || m.tcp.DstPort == 53) {
payload = payload[2:]
}
return payload
}

func (s *DNSSnooper) run() {
var (
Expand Down

0 comments on commit 326208c

Please sign in to comment.