RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-28 22:57:35 +00:00
Code Issues Activity Actions

LibCore+LookupServer: Implement and use UDPServer::send

Browse source
This commit is contained in:
sin-ack 2021-09-11 20:11:30 +00:00 • committed by Ali Mohammad Pur
parent 3da0c072f4
commit 0cca6cef95
5 changed files with 25 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
Userland/Services/LookupServer/DNSServer.cpp
View file

@ -58,7 +58,9 @@ void DNSServer::handle_client()
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));
// FIXME: We should be handling errors here.
[[maybe_unused]] auto result = send(buffer, client_address);
}
}

9
Userland/Services/LookupServer/MulticastDNS.cpp
View file

@ -100,16 +100,17 @@ void MulticastDNS::announce()
response.add_answer(answer);
}
if (emit_packet(response) < 0)
if (emit_packet(response).is_error())
perror("Failed to emit response packet");
}
ssize_t MulticastDNS::emit_packet(const DNSPacket& packet, const sockaddr_in* destination)
ErrorOr<size_t> MulticastDNS::emit_packet(const DNSPacket& packet, const sockaddr_in* destination)
{
auto buffer = packet.to_byte_buffer();
if (!destination)
destination = &mdns_addr;
return sendto(fd(), buffer.data(), buffer.size(), 0, (const sockaddr*)destination, sizeof(*destination));
return send(buffer, *destination);
}
Vector<IPv4Address> MulticastDNS::local_addresses() const
@ -148,7 +149,7 @@ Vector<DNSAnswer> MulticastDNS::lookup(const DNSName& name, DNSRecordType record
request.set_recursion_desired(false);
request.add_question({ name, record_type, DNSRecordClass::IN, false });
if (emit_packet(request) < 0) {
if (emit_packet(request).is_error()) {
perror("failed to emit request packet");
return {};
}

2
Userland/Services/LookupServer/MulticastDNS.h
View file

@ -24,7 +24,7 @@ private:
explicit MulticastDNS(Object* parent = nullptr);
void announce();
ssize_t emit_packet(const DNSPacket&, const sockaddr_in* destination = nullptr);
ErrorOr<size_t> emit_packet(const DNSPacket&, const sockaddr_in* destination = nullptr);
void handle_packet();
void handle_query(const DNSPacket&);