2021-02-14 15:02:02 +03:00
|
|
|
/*
|
2021-02-14 15:10:39 +03:00
|
|
|
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
|
2021-02-14 15:02:02 +03:00
|
|
|
* Copyright (c) 2021, Sergey Bugaev <bugaevc@serenityos.org>
|
|
|
|
*
|
2021-04-22 11:24:48 +03:00
|
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
2021-02-14 15:02:02 +03:00
|
|
|
*/
|
|
|
|
|
2022-04-15 01:58:40 +03:00
|
|
|
#include "Name.h"
|
2021-05-14 18:17:26 +03:00
|
|
|
#include <AK/Random.h>
|
2023-02-09 05:11:50 +03:00
|
|
|
#include <AK/Stream.h>
|
2021-11-10 13:05:21 +03:00
|
|
|
#include <AK/Vector.h>
|
2021-02-14 16:25:48 +03:00
|
|
|
#include <ctype.h>
|
2021-02-14 15:02:02 +03:00
|
|
|
|
2022-04-13 08:25:07 +03:00
|
|
|
namespace DNS {
|
2021-02-14 15:02:02 +03:00
|
|
|
|
2022-12-04 21:02:33 +03:00
|
|
|
Name::Name(DeprecatedString const& name)
|
2021-02-14 15:02:02 +03:00
|
|
|
{
|
|
|
|
if (name.ends_with('.'))
|
|
|
|
m_name = name.substring(0, name.length() - 1);
|
|
|
|
else
|
|
|
|
m_name = name;
|
|
|
|
}
|
|
|
|
|
2022-04-15 01:58:40 +03:00
|
|
|
Name Name::parse(u8 const* data, size_t& offset, size_t max_offset, size_t recursion_level)
|
2021-02-14 15:10:39 +03:00
|
|
|
{
|
|
|
|
if (recursion_level > 4)
|
2022-04-13 08:38:39 +03:00
|
|
|
return {};
|
2021-02-14 15:10:39 +03:00
|
|
|
|
|
|
|
StringBuilder builder;
|
|
|
|
while (true) {
|
|
|
|
if (offset >= max_offset)
|
2022-04-13 08:38:39 +03:00
|
|
|
return {};
|
2021-02-14 15:10:39 +03:00
|
|
|
u8 b = data[offset++];
|
|
|
|
if (b == '\0') {
|
|
|
|
// This terminates the name.
|
2022-12-06 04:12:49 +03:00
|
|
|
return builder.to_deprecated_string();
|
2021-02-14 15:10:39 +03:00
|
|
|
} else if ((b & 0xc0) == 0xc0) {
|
|
|
|
// The two bytes tell us the offset when to continue from.
|
|
|
|
if (offset >= max_offset)
|
2022-04-13 08:38:39 +03:00
|
|
|
return {};
|
2021-02-14 15:10:39 +03:00
|
|
|
size_t dummy = (b & 0x3f) << 8 | data[offset++];
|
|
|
|
auto rest_of_name = parse(data, dummy, max_offset, recursion_level + 1);
|
|
|
|
builder.append(rest_of_name.as_string());
|
2022-12-06 04:12:49 +03:00
|
|
|
return builder.to_deprecated_string();
|
2021-02-14 15:10:39 +03:00
|
|
|
} else {
|
|
|
|
// This is the length of a part.
|
|
|
|
if (offset + b >= max_offset)
|
2022-04-13 08:38:39 +03:00
|
|
|
return {};
|
2022-04-01 20:58:27 +03:00
|
|
|
builder.append((char const*)&data[offset], (size_t)b);
|
2021-02-14 15:10:39 +03:00
|
|
|
builder.append('.');
|
|
|
|
offset += b;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-04-15 01:58:40 +03:00
|
|
|
size_t Name::serialized_size() const
|
2021-02-14 15:17:27 +03:00
|
|
|
{
|
|
|
|
if (m_name.is_empty())
|
|
|
|
return 1;
|
|
|
|
return m_name.length() + 2;
|
|
|
|
}
|
|
|
|
|
2022-04-15 01:58:40 +03:00
|
|
|
void Name::randomize_case()
|
2021-02-14 16:25:48 +03:00
|
|
|
{
|
|
|
|
StringBuilder builder;
|
|
|
|
for (char c : m_name) {
|
|
|
|
// Randomize the 0x20 bit in every ASCII character.
|
|
|
|
if (isalpha(c)) {
|
2021-05-14 18:17:26 +03:00
|
|
|
if (get_random_uniform(2))
|
2021-02-14 16:25:48 +03:00
|
|
|
c |= 0x20;
|
|
|
|
else
|
|
|
|
c &= ~0x20;
|
|
|
|
}
|
|
|
|
builder.append(c);
|
|
|
|
}
|
2022-12-06 04:12:49 +03:00
|
|
|
m_name = builder.to_deprecated_string();
|
2021-02-14 16:25:48 +03:00
|
|
|
}
|
|
|
|
|
2023-02-10 03:00:18 +03:00
|
|
|
ErrorOr<void> Name::write_to_stream(Stream& stream) const
|
2021-02-14 15:16:20 +03:00
|
|
|
{
|
2023-01-09 20:57:41 +03:00
|
|
|
auto parts = as_string().split_view('.');
|
2021-02-14 15:16:20 +03:00
|
|
|
for (auto& part : parts) {
|
2023-01-14 23:13:29 +03:00
|
|
|
TRY(stream.write_value<u8>(part.length()));
|
2023-01-09 20:57:41 +03:00
|
|
|
TRY(stream.write_entire_buffer(part.bytes()));
|
2021-02-14 15:16:20 +03:00
|
|
|
}
|
2023-01-14 23:13:29 +03:00
|
|
|
TRY(stream.write_value('\0'));
|
2023-01-09 20:57:41 +03:00
|
|
|
return {};
|
2021-02-14 15:16:20 +03:00
|
|
|
}
|
|
|
|
|
2022-04-15 01:58:40 +03:00
|
|
|
unsigned Name::Traits::hash(Name const& name)
|
2021-02-14 16:33:16 +03:00
|
|
|
{
|
|
|
|
return CaseInsensitiveStringTraits::hash(name.as_string());
|
|
|
|
}
|
|
|
|
|
2022-04-15 01:58:40 +03:00
|
|
|
bool Name::Traits::equals(Name const& a, Name const& b)
|
2021-02-14 16:33:16 +03:00
|
|
|
{
|
|
|
|
return CaseInsensitiveStringTraits::equals(a.as_string(), b.as_string());
|
|
|
|
}
|
|
|
|
|
2021-02-14 15:02:02 +03:00
|
|
|
}
|