88 lines
1.7 KiB
Go
88 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"dns"
|
|
"fmt"
|
|
)
|
|
|
|
func handleXfrOut(d *dns.Conn, i *dns.Msg) os.Error {
|
|
if i.IsAxfr() {
|
|
fmt.Printf("Axfr request seen\n")
|
|
if i.Question[0].Name == Zone.name {
|
|
fmt.Printf("Matching current zone\n")
|
|
if !Zone.correct {
|
|
fmt.Printf("Zone was not deemed correct\n")
|
|
// Deny it
|
|
d.Close()
|
|
return nil
|
|
}
|
|
|
|
m := make(chan *dns.Xfr)
|
|
e := make(chan os.Error)
|
|
defer close(m)
|
|
defer close(e)
|
|
go d.XfrWrite(i, m, e)
|
|
for j := 0; j < Zone.size; j++ {
|
|
select {
|
|
case m <- &dns.Xfr{Add: true, RR: Zone.rrs[j]}: //
|
|
case err := <-e:
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func handleNotify(d *dns.Conn, i *dns.Msg) os.Error {
|
|
if i.IsNotify() {
|
|
fmt.Printf("Notify seen\n")
|
|
q := new(dns.Msg)
|
|
q.SetReply(i)
|
|
err := d.WriteMsg(q)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
handleXfrIn(i)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func handleXfrIn(i *dns.Msg) ([]dns.RR, os.Error) {
|
|
q := new(dns.Msg)
|
|
q.SetAxfr(i.Question[0].Name)
|
|
|
|
m := make(chan *dns.Xfr)
|
|
fmt.Printf("Preparing Xfr for %s\n", i.Question[0].Name)
|
|
|
|
d := new(dns.Conn)
|
|
d.RemoteAddr = "127.0.0.1:53"
|
|
err := d.Dial("tcp")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer d.Close()
|
|
|
|
fmt.Printf("Calling 127.0.0.1 successful\n")
|
|
go d.XfrRead(q, m)
|
|
|
|
Zone.name = i.Question[0].Name
|
|
j := 0
|
|
for x := range m {
|
|
fmt.Printf("%v %v\n", x.Add, x.RR)
|
|
Zone.rrs[j] = x.RR
|
|
j++
|
|
}
|
|
Zone.size = j
|
|
return nil, nil
|
|
}
|
|
|
|
func sendNotify(addr, zone string) {
|
|
d := new(dns.Conn)
|
|
d.RemoteAddr = addr
|
|
m := new(dns.Msg)
|
|
m.SetNotify(zone)
|
|
dns.QueryRequest <- &dns.Query{Conn: d, Query: m}
|
|
}
|