jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2021, Sergey Bugaev <bugaevc@serenityos.org>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#include "ConnectionFromClient.h"
8#include "LookupServer.h"
9#include <AK/IPv4Address.h>
10#include <LibDNS/Packet.h>
11
12namespace LookupServer {
13
14using namespace DNS;
15
16static HashMap<int, RefPtr<ConnectionFromClient>> s_connections;
17
18ConnectionFromClient::ConnectionFromClient(NonnullOwnPtr<Core::LocalSocket> socket, int client_id)
19 : IPC::ConnectionFromClient<LookupClientEndpoint, LookupServerEndpoint>(*this, move(socket), client_id)
20{
21 s_connections.set(client_id, *this);
22}
23
24void ConnectionFromClient::die()
25{
26 s_connections.remove(client_id());
27}
28
29Messages::LookupServer::LookupNameResponse ConnectionFromClient::lookup_name(DeprecatedString const& name)
30{
31 auto maybe_answers = LookupServer::the().lookup(name, RecordType::A);
32 if (maybe_answers.is_error()) {
33 dbgln("LookupServer: Failed to lookup A record: {}", maybe_answers.error());
34 return { 1, {} };
35 }
36
37 auto answers = maybe_answers.release_value();
38 Vector<DeprecatedString> addresses;
39 for (auto& answer : answers) {
40 addresses.append(answer.record_data());
41 }
42 return { 0, move(addresses) };
43}
44
45Messages::LookupServer::LookupAddressResponse ConnectionFromClient::lookup_address(DeprecatedString const& address)
46{
47 if (address.length() != 4)
48 return { 1, DeprecatedString() };
49 IPv4Address ip_address { (u8 const*)address.characters() };
50 auto name = DeprecatedString::formatted("{}.{}.{}.{}.in-addr.arpa",
51 ip_address[3],
52 ip_address[2],
53 ip_address[1],
54 ip_address[0]);
55
56 auto maybe_answers = LookupServer::the().lookup(name, RecordType::PTR);
57 if (maybe_answers.is_error()) {
58 dbgln("LookupServer: Failed to lookup PTR record: {}", maybe_answers.error());
59 return { 1, DeprecatedString() };
60 }
61
62 auto answers = maybe_answers.release_value();
63 if (answers.is_empty())
64 return { 1, DeprecatedString() };
65 return { 0, answers[0].record_data() };
66}
67}