/* 
 * Send back REFUSED
 *
 * Stephane Bortzmeyer <stephane+grong@bortzmeyer.org>
 *
 * Adapted to Go DNS (i.e. completely rewritten)
 * Miek Gieben <miek@miek.nl>
 */

package main

import (
	"os"
	"net"
	"dns"
	"dns/responder"
	"os/signal"
)

type server responder.Server

func (s *server) ResponderUDP(c *net.UDPConn, a net.Addr, in []byte) {
	inmsg := new(dns.Msg)
	if !inmsg.Unpack(in) {
		// FormError
		// NXdomain 'n stuff
		println("Unpacking failed")
	}
	if inmsg.MsgHdr.Response == true {
		return // don't answer responses
	}
	m := new(dns.Msg)
	m.MsgHdr.Id = inmsg.MsgHdr.Id
	m.MsgHdr.Response = true
	m.MsgHdr.Opcode = dns.OpcodeQuery

	m.MsgHdr.Rcode = dns.RcodeRefused
	m.Question = make([]dns.Question, 1)
	m.Question[0] = inmsg.Question[0]
	out, b := m.Pack()
	if !b {
		println("Failed to pack")
	}
	responder.SendUDP(out, c, a)
}

func (s *server) ResponderTCP(c *net.TCPConn, in []byte) {
	return
}

func main() {
	s := new(responder.Server)
	s.Address = "127.0.0.1"
	s.Port = "8053"
	var srv *server
	ch := make(chan os.Error)
	go s.NewResponder(srv, ch)

forever:
	for {
		// Wait for a signal to stop
		select {
		case <-signal.Incoming:
			println("Signal received, stopping")
			ch <- nil
			<-ch
			break forever
		case e := <-cht:
			println(e.String())
		case e := <-ch:
			println(e.String())
		}
	}
}