ladybird/AK/UUID.cpp
Idan Horowitz 3219ce3d61 AK: Return KString instead of String from encode_hex in the Kernel
This let's us propagate allocation errors from this API.
2022-02-16 22:21:37 +01:00

130 lines
4.7 KiB
C++

/*
* Copyright (c) 2020, Liav A. <liavalb@hotmail.co.il>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <AK/AllOf.h>
#include <AK/Hex.h>
#include <AK/StringBuilder.h>
#include <AK/UUID.h>
namespace AK {
UUID::UUID(Array<u8, 16> uuid_buffer)
{
uuid_buffer.span().copy_to(m_uuid_buffer);
}
void UUID::convert_string_view_to_little_endian_uuid(StringView uuid_string_view)
{
VERIFY(uuid_string_view.length() == 36);
auto first_unit = MUST(decode_hex(uuid_string_view.substring_view(0, 8)));
auto second_unit = MUST(decode_hex(uuid_string_view.substring_view(9, 4)));
auto third_unit = MUST(decode_hex(uuid_string_view.substring_view(14, 4)));
auto fourth_unit = MUST(decode_hex(uuid_string_view.substring_view(19, 4)));
auto fifth_unit = MUST(decode_hex(uuid_string_view.substring_view(24, 12)));
VERIFY(first_unit.size() == 4 && second_unit.size() == 2
&& third_unit.size() == 2 && fourth_unit.size() == 2
&& fifth_unit.size() == 6);
m_uuid_buffer.span().overwrite(0, first_unit.data(), first_unit.size());
m_uuid_buffer.span().overwrite(4, second_unit.data(), second_unit.size());
m_uuid_buffer.span().overwrite(6, third_unit.data(), third_unit.size());
m_uuid_buffer.span().overwrite(8, fourth_unit.data(), fourth_unit.size());
m_uuid_buffer.span().overwrite(10, fifth_unit.data(), fifth_unit.size());
}
void UUID::convert_string_view_to_mixed_endian_uuid(StringView uuid_string_view)
{
VERIFY(uuid_string_view.length() == 36);
auto first_unit = MUST(decode_hex(uuid_string_view.substring_view(0, 8)));
auto second_unit = MUST(decode_hex(uuid_string_view.substring_view(9, 4)));
auto third_unit = MUST(decode_hex(uuid_string_view.substring_view(14, 4)));
auto fourth_unit = MUST(decode_hex(uuid_string_view.substring_view(19, 4)));
auto fifth_unit = MUST(decode_hex(uuid_string_view.substring_view(24, 12)));
VERIFY(first_unit.size() == 4 && second_unit.size() == 2
&& third_unit.size() == 2 && fourth_unit.size() == 2
&& fifth_unit.size() == 6);
// Revert endianness for first 4 bytes
for (size_t index = 0; index < 4; index++) {
m_uuid_buffer[3 - index] = first_unit[index];
}
// Revert endianness for second 2 bytes and again for 2 bytes
for (size_t index = 0; index < 2; index++) {
m_uuid_buffer[3 + 2 - index] = second_unit[index];
m_uuid_buffer[5 + 2 - index] = third_unit[index];
}
m_uuid_buffer.span().overwrite(8, fourth_unit.data(), fourth_unit.size());
m_uuid_buffer.span().overwrite(10, fifth_unit.data(), fifth_unit.size());
}
UUID::UUID(StringView uuid_string_view, Endianness endianness)
{
if (endianness == Endianness::Little) {
convert_string_view_to_little_endian_uuid(uuid_string_view);
return;
} else if (endianness == Endianness::Mixed) {
convert_string_view_to_mixed_endian_uuid(uuid_string_view);
return;
}
VERIFY_NOT_REACHED();
}
#ifdef KERNEL
ErrorOr<NonnullOwnPtr<Kernel::KString>> UUID::to_string() const
{
StringBuilder builder(36);
auto nibble0 = TRY(encode_hex(m_uuid_buffer.span().trim(4)));
TRY(builder.try_append(nibble0->view()));
TRY(builder.try_append('-'));
auto nibble1 = TRY(encode_hex(m_uuid_buffer.span().slice(4).trim(2)));
TRY(builder.try_append(nibble1->view()));
TRY(builder.try_append('-'));
auto nibble2 = TRY(encode_hex(m_uuid_buffer.span().slice(6).trim(2)));
TRY(builder.try_append(nibble2->view()));
TRY(builder.try_append('-'));
auto nibble3 = TRY(encode_hex(m_uuid_buffer.span().slice(8).trim(2)));
TRY(builder.try_append(nibble3->view()));
TRY(builder.try_append('-'));
auto nibble4 = TRY(encode_hex(m_uuid_buffer.span().slice(10).trim(6)));
TRY(builder.try_append(nibble4->view()));
return Kernel::KString::try_create(builder.string_view());
}
#else
String UUID::to_string() const
{
StringBuilder builder(36);
builder.append(encode_hex(m_uuid_buffer.span().trim(4)).view());
builder.append('-');
builder.append(encode_hex(m_uuid_buffer.span().slice(4).trim(2)).view());
builder.append('-');
builder.append(encode_hex(m_uuid_buffer.span().slice(6).trim(2)).view());
builder.append('-');
builder.append(encode_hex(m_uuid_buffer.span().slice(8).trim(2)).view());
builder.append('-');
builder.append(encode_hex(m_uuid_buffer.span().slice(10).trim(6)).view());
return builder.to_string();
}
#endif
bool UUID::operator==(const UUID& other) const
{
for (size_t index = 0; index < 16; index++) {
if (m_uuid_buffer[index] != other.m_uuid_buffer[index])
return false;
}
return true;
}
bool UUID::is_zero() const
{
return all_of(m_uuid_buffer, [](const auto octet) { return octet == 0; });
}
}