Abstract xfr nicely away
This commit is contained in:
parent
90a1d813d9
commit
d67c427953
|
@ -5,7 +5,6 @@
|
||||||
// DNS resolver client: see RFC 1035.
|
// DNS resolver client: see RFC 1035.
|
||||||
|
|
||||||
package dns
|
package dns
|
||||||
// TODO: refacter this
|
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"os"
|
"os"
|
||||||
|
@ -270,6 +269,7 @@ Server:
|
||||||
}
|
}
|
||||||
|
|
||||||
if !first {
|
if !first {
|
||||||
|
d.Tsig.TimersOnly = true
|
||||||
if !checkXfrSOA(in, false) {
|
if !checkXfrSOA(in, false) {
|
||||||
// Soa record not the last one
|
// Soa record not the last one
|
||||||
sendMsg(in, m, false)
|
sendMsg(in, m, false)
|
||||||
|
@ -278,7 +278,6 @@ Server:
|
||||||
sendMsg(in, m, true)
|
sendMsg(in, m, true)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
d.Tsig.TimersOnly = true //
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
panic("not reached")
|
panic("not reached")
|
||||||
|
|
180
xfr.go
180
xfr.go
|
@ -1,37 +1,177 @@
|
||||||
package dns
|
package dns
|
||||||
|
|
||||||
import (
|
|
||||||
"net"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Outgoing AXFR and IXFR implementations
|
// Outgoing AXFR and IXFR implementations
|
||||||
|
// error handling??
|
||||||
|
|
||||||
// Read from m until it is closed. Group the RRs until
|
// Msg tells use what to do
|
||||||
// no space is left and send the messages.
|
func (d *Conn) XfrRead(q *Msg, m chan Xfr) {
|
||||||
// How do I know the message still fits in 64 K???
|
switch q.Question[0].Qtype {
|
||||||
func AxfrTCP(c *net.TCPConn, a net.Addr, m chan Xfr) {
|
case TypeAXFR:
|
||||||
msg := new(Msg)
|
d.axfrRead(q, m)
|
||||||
msg.Answer = make([]RR, 1000)
|
case TypeIXFR:
|
||||||
|
d.ixfrRead(q, m)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Conn) XfrWrite(q *Msg, m chan Xfr) {
|
||||||
|
switch q.Question[0].Qtype {
|
||||||
|
case TypeAXFR:
|
||||||
|
d.axfrWrite(q, m)
|
||||||
|
case TypeIXFR:
|
||||||
|
// d.ixfrWrite(q, m)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Conn) axfrRead(q *Msg, m chan Xfr) {
|
||||||
|
defer close(m)
|
||||||
|
first := true
|
||||||
|
in := new(Msg)
|
||||||
|
for {
|
||||||
|
inb := make([]byte, MaxMsgSize)
|
||||||
|
n, err := d.Read(inb)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
inb = inb[:n]
|
||||||
|
|
||||||
|
if !in.Unpack(inb) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if in.Id != q.Id {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if first {
|
||||||
|
if !checkXfrSOA(in, true) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
first = !first
|
||||||
|
}
|
||||||
|
|
||||||
|
if !first {
|
||||||
|
if !checkXfrSOA(in, false) {
|
||||||
|
// Soa record not the last one
|
||||||
|
sendMsg(in, m, false)
|
||||||
|
continue
|
||||||
|
} else {
|
||||||
|
sendMsg(in, m, true)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
d.Tsig.TimersOnly = true // Subsequent envelopes use this
|
||||||
|
}
|
||||||
|
}
|
||||||
|
panic("not reached")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Just send the zone
|
||||||
|
func (d *Conn) axfrWrite(q *Msg, m chan Xfr) {
|
||||||
|
out := new(Msg)
|
||||||
|
out.Id = q.Id
|
||||||
|
out.Question = q.Question
|
||||||
|
out.Answer = make([]RR, 1000)
|
||||||
|
var soa *RR_SOA;
|
||||||
i := 0
|
i := 0
|
||||||
var soa *RR_SOA
|
for r := range m {
|
||||||
for r := range m {
|
out.Answer[i] = r.RR
|
||||||
msg.Answer[i] = r.RR
|
|
||||||
if soa == nil {
|
if soa == nil {
|
||||||
if r.RR.Header().Rrtype != TypeSOA {
|
if r.RR.Header().Rrtype != TypeSOA {
|
||||||
// helegaar geen SOA
|
return
|
||||||
} else {
|
} else {
|
||||||
soa = r.RR.(*RR_SOA)
|
soa = r.RR.(*RR_SOA)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
i++
|
i++
|
||||||
if i > 1000 {
|
if i > 1000 {
|
||||||
// send it
|
// Send it
|
||||||
msg.Answer = msg.Answer[:0]
|
send, _ := out.Pack()
|
||||||
|
_, err := d.Write(send)
|
||||||
|
if err != nil {
|
||||||
|
/* ... */
|
||||||
|
}
|
||||||
i = 0
|
i = 0
|
||||||
|
out.Answer = out.Answer[:0]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// Last one, what if was 1000 send lonely soa?? No matter
|
// Everything is send, only the closing soa is left.
|
||||||
// what, add the SOA and send the msg
|
out.Answer[i] = soa
|
||||||
msg.Answer[i] = soa
|
send, _ := out.Pack()
|
||||||
// send it
|
_, err := d.Write(send)
|
||||||
|
if err != nil {
|
||||||
|
/* ... */
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Conn) ixfrRead(q *Msg, m chan Xfr) {
|
||||||
|
defer close(m)
|
||||||
|
var serial uint32 // The first serial seen is the current server serial
|
||||||
|
var x Xfr
|
||||||
|
first := true
|
||||||
|
in := new(Msg)
|
||||||
|
for {
|
||||||
|
var inb []byte
|
||||||
|
if d.TCP != nil {
|
||||||
|
inb = make([]byte, MaxMsgSize)
|
||||||
|
} else {
|
||||||
|
inb = make([]byte, DefaultMsgSize)
|
||||||
|
}
|
||||||
|
n, err := d.Read(inb)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
inb = inb[:n]
|
||||||
|
|
||||||
|
if !in.Unpack(inb) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if in.Id != q.Id {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if first {
|
||||||
|
// A single SOA RR signals "no changes"
|
||||||
|
if len(in.Answer) == 1 && checkXfrSOA(in, true) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// But still check if the returned answer is ok
|
||||||
|
if !checkXfrSOA(in, true) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// This serial is important
|
||||||
|
serial = in.Answer[0].(*RR_SOA).Serial
|
||||||
|
first = !first
|
||||||
|
}
|
||||||
|
|
||||||
|
// Now we need to check each message for SOA records, to see what we need to do
|
||||||
|
x.Add = true
|
||||||
|
if !first {
|
||||||
|
d.Tsig.TimersOnly = true
|
||||||
|
for k, r := range in.Answer {
|
||||||
|
// If the last record in the IXFR contains the servers' SOA, we should quit
|
||||||
|
if r.Header().Rrtype == TypeSOA {
|
||||||
|
switch {
|
||||||
|
case r.(*RR_SOA).Serial == serial:
|
||||||
|
if k == len(in.Answer)-1 {
|
||||||
|
// last rr is SOA with correct serial
|
||||||
|
//m <- r dont' send it
|
||||||
|
return
|
||||||
|
}
|
||||||
|
x.Add = true
|
||||||
|
if k != 0 {
|
||||||
|
// Intermediate SOA
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
case r.(*RR_SOA).Serial != serial:
|
||||||
|
x.Add = false
|
||||||
|
continue // Don't need to see this SOA
|
||||||
|
}
|
||||||
|
}
|
||||||
|
x.RR = r
|
||||||
|
m <- x
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
panic("not reached")
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue