diff options
| author | jet2tlf <jet2tlf@gmail.com> | 2024-06-03 06:03:00 +0000 |
|---|---|---|
| committer | jet2tlf <jet2tlf@gmail.com> | 2024-06-03 06:03:00 +0000 |
| commit | 8659adcd1500f8e199ccef1c255130327a7597f2 (patch) | |
| tree | 353f3f014335fda196427ea9f458f7048d535061 /app/main.go | |
| parent | b6b5b6c0b80798147dd8b27120aa888880998ccb (diff) | |
| download | dns-server-go-8659adcd1500f8e199ccef1c255130327a7597f2.tar.gz dns-server-go-8659adcd1500f8e199ccef1c255130327a7597f2.zip | |
codecrafters submit [skip ci]
Diffstat (limited to 'app/main.go')
| -rw-r--r-- | app/main.go | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/app/main.go b/app/main.go index 69aefae..b02972d 100644 --- a/app/main.go +++ b/app/main.go @@ -31,7 +31,7 @@ func main() { receivedData := string(buf[:size]) fmt.Printf("Received %d bytes from %s: %s\n", size, source, receivedData) requestHeader := ParseHeader(buf[:size]) - question := ParseQuestion(buf[:size]) + questions := ParseQuestions(buf[:size], requestHeader.QDCOUNT) rcode := uint8(4) if requestHeader.OPCODE == 0 { @@ -48,17 +48,19 @@ func main() { RA: 0, Z: 0, RCODE: rcode, - QDCOUNT: 1, + QDCOUNT: 0, ANCOUNT: 0, - NSCOUNT: 0, - ARCOUNT: 0, + NSCOUNT: requestHeader.NSCOUNT, + ARCOUNT: requestHeader.ARCOUNT, } response := MakeMessage(header) - response.AddQuestion(question) - answer := MakeAnswer(question.Name, []byte("\x08\x08\x08\x08")) - response.AddAnswer(answer) + for _, question := range questions { + response.AddQuestion(question) + answer := MakeAnswer(question.Name, []byte("\x08\x08\x08\x08")) + response.AddAnswer(answer) + } _, err = udpConn.WriteToUDP(response.Bytes(), source) if err != nil { |