1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
package main
import (
"fmt"
"net"
)
func main() {
udpAddr, err := net.ResolveUDPAddr("udp", "127.0.0.1:2053")
if err != nil {
fmt.Println("Failed to resolve UDP address:", err)
return
}
udpConn, err := net.ListenUDP("udp", udpAddr)
if err != nil {
fmt.Println("Failed to bind to address:", err)
return
}
defer udpConn.Close()
buf := make([]byte, 512)
for {
size, source, err := udpConn.ReadFromUDP(buf)
if err != nil {
fmt.Println("Error receiving data:", err)
break
}
receivedData := string(buf[:size])
fmt.Printf("Received %d bytes from %s: %s\n", size, source, receivedData)
header := DNSHeader{
ID: 1234,
QR: 1,
OPCODE: 1,
AA: 0,
TC: 0,
RD: 0,
RA: 0,
Z: 0,
RCODE: 0,
QDCOUNT: 1,
ANCOUNT: 1,
NSCOUNT: 0,
ARCOUNT: 0,
}
response := MakeMessage(header)
response.AddQuestion([]byte("\x0ccodecrafters\x02io\x00"))
_, err = udpConn.WriteToUDP(response.bytes(), source)
if err != nil {
fmt.Println("Failed to send response:", err)
}
}
}
|
