157 lines
4.0 KiB
Go
157 lines
4.0 KiB
Go
package core
|
|
|
|
/*
|
|
* @Description DNS解析处理入口
|
|
* @Author www.odboy.cn
|
|
* @Date 20241107
|
|
*/
|
|
import (
|
|
"fmt"
|
|
"github.com/miekg/dns"
|
|
"kenaito-dns/constant"
|
|
"kenaito-dns/dao"
|
|
"net"
|
|
)
|
|
|
|
func HandleDNSRequest(w dns.ResponseWriter, r *dns.Msg) {
|
|
msg := new(dns.Msg)
|
|
msg.SetReply(r)
|
|
// 将 DNS 响应标记为权威应答
|
|
msg.Authoritative = true
|
|
// 将 DNS 响应标记为递归可用
|
|
msg.RecursionAvailable = true
|
|
// 遍历请求中的问题部分，生成相应的回答
|
|
for _, question := range r.Question {
|
|
switch question.Qtype {
|
|
case dns.TypeA:
|
|
handleARecord(question, msg)
|
|
case dns.TypeAAAA:
|
|
handleAAAARecord(question, msg)
|
|
case dns.TypeCNAME:
|
|
handleCNAMERecord(question, msg)
|
|
case dns.TypeMX:
|
|
handleMXRecord(question, msg)
|
|
case dns.TypeTXT:
|
|
handleTXTRecord(question, msg)
|
|
}
|
|
}
|
|
// 发送响应
|
|
err := w.WriteMsg(msg)
|
|
if err != nil {
|
|
fmt.Printf("=== Handle Failed === %v \n", err)
|
|
}
|
|
}
|
|
|
|
// 构建 A 记录 IPV4
|
|
func handleARecord(q dns.Question, msg *dns.Msg) {
|
|
name := q.Name
|
|
queryName := name[0 : len(name)-1]
|
|
records := dao.FindResolveRecordByNameType(queryName, constant.R_A)
|
|
if len(records) > 0 {
|
|
for _, record := range records {
|
|
fmt.Printf("=== A记录 === 请求解析的域名：%s,解析的目标IP地址:%s\n", name, record.Value)
|
|
ip := net.ParseIP(record.Value)
|
|
rr := &dns.A{
|
|
Hdr: dns.RR_Header{
|
|
Name: name,
|
|
Rrtype: dns.TypeA,
|
|
Class: dns.ClassINET,
|
|
Ttl: uint32(record.Ttl) * 60,
|
|
},
|
|
A: ip,
|
|
}
|
|
msg.Answer = append(msg.Answer, rr)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 构建 AAAA 记录 IPV6
|
|
func handleAAAARecord(q dns.Question, msg *dns.Msg) {
|
|
name := q.Name
|
|
queryName := name[0 : len(name)-1]
|
|
records := dao.FindResolveRecordByNameType(queryName, constant.R_AAAA)
|
|
if len(records) > 0 {
|
|
for _, record := range records {
|
|
fmt.Printf("=== AAAA记录 === 请求解析的域名：%s,解析的目标IP地址:%s\n", name, record.Value)
|
|
ip := net.ParseIP(record.Value)
|
|
rr := &dns.AAAA{
|
|
Hdr: dns.RR_Header{
|
|
Name: name,
|
|
Rrtype: dns.TypeAAAA,
|
|
Class: dns.ClassINET,
|
|
Ttl: uint32(record.Ttl) * 60,
|
|
},
|
|
AAAA: ip,
|
|
}
|
|
msg.Answer = append(msg.Answer, rr)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 构建 CNAME 记录
|
|
func handleCNAMERecord(q dns.Question, msg *dns.Msg) {
|
|
name := q.Name
|
|
queryName := name[0 : len(name)-1]
|
|
records := dao.FindResolveRecordByNameType(queryName, constant.R_CNAME)
|
|
if len(records) > 0 {
|
|
for _, record := range records {
|
|
fmt.Printf("=== CNAME记录 === 请求解析的域名：%s,解析的目标域名:%s\n", name, record.Value)
|
|
rr := &dns.CNAME{
|
|
Hdr: dns.RR_Header{
|
|
Name: name,
|
|
Rrtype: dns.TypeCNAME,
|
|
Class: dns.ClassINET,
|
|
Ttl: uint32(record.Ttl) * 60,
|
|
},
|
|
Target: record.Value + ".",
|
|
}
|
|
msg.Answer = append(msg.Answer, rr)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 构建 MX 记录
|
|
func handleMXRecord(q dns.Question, msg *dns.Msg) {
|
|
name := q.Name
|
|
queryName := name[0 : len(name)-1]
|
|
records := dao.FindResolveRecordByNameType(queryName, constant.R_MX)
|
|
if len(records) > 0 {
|
|
for _, record := range records {
|
|
fmt.Printf("=== MX记录 === 请求解析的域名：%s,解析的目标域名:%s, MX优先级: 10\n", name, record.Value)
|
|
rr := &dns.MX{
|
|
Hdr: dns.RR_Header{
|
|
Name: name,
|
|
Rrtype: dns.TypeMX,
|
|
Class: dns.ClassINET,
|
|
Ttl: uint32(record.Ttl) * 60,
|
|
},
|
|
Preference: 10,
|
|
Mx: record.Value + ".",
|
|
}
|
|
msg.Answer = append(msg.Answer, rr)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 构建 TXT 记录
|
|
func handleTXTRecord(q dns.Question, msg *dns.Msg) {
|
|
name := q.Name
|
|
queryName := name[0 : len(name)-1]
|
|
records := dao.FindResolveRecordByNameType(queryName, constant.R_TXT)
|
|
if len(records) > 0 {
|
|
for _, record := range records {
|
|
fmt.Printf("=== TXT记录 === 请求解析的域名：%s,解析的目标值:%s\n", name, record.Value)
|
|
rr := &dns.TXT{
|
|
Hdr: dns.RR_Header{
|
|
Name: name,
|
|
Rrtype: dns.TypeTXT,
|
|
Class: dns.ClassINET,
|
|
Ttl: uint32(record.Ttl) * 60,
|
|
},
|
|
Txt: []string{record.Value},
|
|
}
|
|
msg.Answer = append(msg.Answer, rr)
|
|
}
|
|
}
|
|
}
|