MagiTrickle/main.go

71 lines
1.7 KiB
Go
Raw Normal View History

package main
import (
"fmt"
2024-08-24 17:46:34 +03:00
dnsProxy "kvas2-go/dns-proxy"
2024-08-24 19:47:10 +03:00
ruleComposer "kvas2-go/rule-composer"
"log"
2024-08-24 21:26:49 +03:00
"os"
"os/signal"
"syscall"
)
var (
2024-08-24 17:46:34 +03:00
ListenPort = uint16(7548)
UsableDNSServerAddress = "127.0.0.1"
2024-08-24 17:46:34 +03:00
UsableDNSServerPort = uint16(53)
)
func main() {
2024-08-24 19:47:10 +03:00
records := ruleComposer.NewRecords()
2024-08-24 17:46:34 +03:00
proxy := dnsProxy.New("", ListenPort, UsableDNSServerAddress, UsableDNSServerPort)
proxy.MsgHandler = func(msg *dnsProxy.Message) {
for _, q := range msg.QD {
fmt.Printf("%x: <- Request name: %s\n", msg.ID, q.QName.String())
}
2024-08-24 17:46:34 +03:00
for _, a := range msg.AN {
switch v := a.(type) {
case dnsProxy.Address:
fmt.Printf("%x: -> A: Name: %s; Address: %s; TTL: %d\n", msg.ID, v.Name, v.Address.String(), v.TTL)
2024-08-24 19:47:10 +03:00
records.PutIPv4Address(v.Name.String(), v.Address, int64(v.TTL))
2024-08-24 17:46:34 +03:00
case dnsProxy.CName:
fmt.Printf("%x: -> CNAME: Name: %s; CName: %s\n", msg.ID, v.Name, v.CName)
2024-08-24 19:47:10 +03:00
records.PutCName(v.Name.String(), v.CName.String(), int64(v.TTL))
2024-08-24 17:46:34 +03:00
default:
fmt.Printf("%x: -> Unknown: %x\n", msg.ID, v.EncodeResource())
}
}
2024-08-24 17:46:34 +03:00
for _, a := range msg.NS {
fmt.Printf("%x: -> NS: %x\n", msg.ID, a.EncodeResource())
}
2024-08-24 17:46:34 +03:00
for _, a := range msg.AR {
fmt.Printf("%x: -> NS: %x\n", msg.ID, a.EncodeResource())
}
2024-08-24 19:47:10 +03:00
for _, q := range msg.QD {
fmt.Printf("%x: DBG Known addresses for: %s\n", msg.ID, q.QName.String())
for idx, addr := range records.GetIPv4Addresses(q.QName.String()) {
fmt.Printf("%x: #%d: %s\n", msg.ID, idx, addr.String())
}
}
}
2024-08-24 21:26:49 +03:00
go func() {
err := proxy.Listen()
if err != nil {
log.Fatal(err)
}
}()
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
for {
select {
case <-c:
proxy.Close()
return
}
}
}