mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-11-22 15:40:19 +00:00
e0465b8939
This reverts commit 8bb610b97a.
Linking wolfSSL seems to cause more legal trouble than it's worth due to
it being GPLv2, so let's undo this for now.
90 lines
2.3 KiB
C++
90 lines
2.3 KiB
C++
/*
|
|
* Copyright (c) 2020, Ali Mohammad Pur <mpfard@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/ByteBuffer.h>
|
|
#include <AK/ByteReader.h>
|
|
#include <AK/Endian.h>
|
|
#include <AK/Types.h>
|
|
#include <LibTLS/Extensions.h>
|
|
|
|
namespace TLS {
|
|
|
|
class PacketBuilder {
|
|
public:
|
|
PacketBuilder(ContentType type, u16 version, size_t size_hint = 0xfdf)
|
|
: PacketBuilder(type, (ProtocolVersion)version, size_hint)
|
|
{
|
|
}
|
|
|
|
PacketBuilder(ContentType type, ProtocolVersion version, size_t size_hint = 0xfdf)
|
|
{
|
|
// FIXME: Handle possible OOM situation.
|
|
m_packet_data = ByteBuffer::create_uninitialized(size_hint + 16).release_value_but_fixme_should_propagate_errors();
|
|
m_current_length = 5;
|
|
m_packet_data[0] = (u8)type;
|
|
ByteReader::store(m_packet_data.offset_pointer(1), AK::convert_between_host_and_network_endian((u16)version));
|
|
}
|
|
|
|
inline void append(u16 value)
|
|
{
|
|
value = AK::convert_between_host_and_network_endian(value);
|
|
append((u8 const*)&value, sizeof(value));
|
|
}
|
|
inline void append(u8 value)
|
|
{
|
|
append((u8 const*)&value, sizeof(value));
|
|
}
|
|
inline void append(ReadonlyBytes data)
|
|
{
|
|
append(data.data(), data.size());
|
|
}
|
|
inline void append_u24(u32 value)
|
|
{
|
|
u8 buf[3];
|
|
buf[0] = value / 0x10000;
|
|
value %= 0x10000;
|
|
buf[1] = value / 0x100;
|
|
value %= 0x100;
|
|
buf[2] = value;
|
|
|
|
append(buf, 3);
|
|
}
|
|
inline void append(u8 const* data, size_t bytes)
|
|
{
|
|
if (bytes == 0)
|
|
return;
|
|
|
|
auto old_length = m_current_length;
|
|
m_current_length += bytes;
|
|
|
|
if (m_packet_data.size() < m_current_length) {
|
|
m_packet_data.resize(m_current_length);
|
|
}
|
|
|
|
m_packet_data.overwrite(old_length, data, bytes);
|
|
}
|
|
inline ByteBuffer build()
|
|
{
|
|
auto length = m_current_length;
|
|
m_current_length = 0;
|
|
// FIXME: Propagate errors.
|
|
return MUST(m_packet_data.slice(0, length));
|
|
}
|
|
inline void set(size_t offset, u8 value)
|
|
{
|
|
VERIFY(offset < m_current_length);
|
|
m_packet_data[offset] = value;
|
|
}
|
|
size_t length() const { return m_current_length; }
|
|
|
|
private:
|
|
ByteBuffer m_packet_data;
|
|
size_t m_current_length;
|
|
};
|
|
|
|
}
|