mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-11-10 13:00:29 +03:00
65 lines
1.6 KiB
C++
65 lines
1.6 KiB
C++
/*
|
|
* Copyright (c) 2021, Sergey Bugaev <bugaevc@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "DNSServer.h"
|
|
#include "DNSPacket.h"
|
|
#include "LookupServer.h"
|
|
#include <AK/IPv4Address.h>
|
|
|
|
namespace LookupServer {
|
|
|
|
DNSServer::DNSServer(Object* parent)
|
|
: Core::UDPServer(parent)
|
|
{
|
|
bind(IPv4Address(), 53);
|
|
on_ready_to_receive = [this]() {
|
|
handle_client();
|
|
};
|
|
}
|
|
|
|
void DNSServer::handle_client()
|
|
{
|
|
sockaddr_in client_address;
|
|
auto buffer = receive(1024, client_address);
|
|
auto optional_request = DNSPacket::from_raw_packet(buffer.data(), buffer.size());
|
|
if (!optional_request.has_value()) {
|
|
dbgln("Got an invalid DNS packet");
|
|
return;
|
|
}
|
|
auto& request = optional_request.value();
|
|
|
|
if (!request.is_query()) {
|
|
dbgln("It's not a request");
|
|
return;
|
|
}
|
|
|
|
LookupServer& lookup_server = LookupServer::the();
|
|
|
|
DNSPacket response;
|
|
response.set_is_response();
|
|
response.set_id(request.id());
|
|
|
|
for (auto& question : request.questions()) {
|
|
if (question.class_code() != DNSRecordClass::IN)
|
|
continue;
|
|
response.add_question(question);
|
|
auto answers = lookup_server.lookup(question.name(), question.record_type());
|
|
for (auto& answer : answers) {
|
|
response.add_answer(answer);
|
|
}
|
|
}
|
|
|
|
if (response.answer_count() == 0)
|
|
response.set_code(DNSPacket::Code::NXDOMAIN);
|
|
else
|
|
response.set_code(DNSPacket::Code::NOERROR);
|
|
|
|
buffer = response.to_byte_buffer();
|
|
sendto(fd(), buffer.data(), buffer.size(), 0, (const sockaddr*)&client_address, sizeof(client_address));
|
|
}
|
|
|
|
}
|