mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-11-25 17:10:23 +00:00
8e79bde2b7
KBufferBuilder::build() now returns an OwnPtr<KBuffer> and can fail. Clients of the API have been updated to handle that situation.
126 lines
3.4 KiB
C++
126 lines
3.4 KiB
C++
/*
|
|
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright notice, this
|
|
* list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
* and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
|
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
|
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
|
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include <AK/PrintfImplementation.h>
|
|
#include <AK/StdLibExtras.h>
|
|
#include <Kernel/KBufferBuilder.h>
|
|
#include <stdarg.h>
|
|
|
|
namespace Kernel {
|
|
|
|
inline bool KBufferBuilder::can_append(size_t size) const
|
|
{
|
|
if (!m_buffer)
|
|
return false;
|
|
bool has_space = ((m_size + size) < m_buffer->size());
|
|
ASSERT(has_space);
|
|
return has_space;
|
|
}
|
|
|
|
OwnPtr<KBuffer> KBufferBuilder::build()
|
|
{
|
|
m_buffer->set_size(m_size);
|
|
return m_buffer.release_nonnull();
|
|
}
|
|
|
|
KBufferBuilder::KBufferBuilder()
|
|
: m_buffer(KBuffer::try_create_with_size(4 * MiB, Region::Access::Read | Region::Access::Write))
|
|
{
|
|
}
|
|
|
|
void KBufferBuilder::append(const StringView& str)
|
|
{
|
|
if (str.is_empty())
|
|
return;
|
|
if (!can_append(str.length()))
|
|
return;
|
|
memcpy(insertion_ptr(), str.characters_without_null_termination(), str.length());
|
|
m_size += str.length();
|
|
}
|
|
|
|
void KBufferBuilder::append(const char* characters, int length)
|
|
{
|
|
if (!length)
|
|
return;
|
|
if (!can_append(length))
|
|
return;
|
|
memcpy(insertion_ptr(), characters, length);
|
|
m_size += length;
|
|
}
|
|
|
|
void KBufferBuilder::append(char ch)
|
|
{
|
|
if (!can_append(1))
|
|
return;
|
|
insertion_ptr()[0] = ch;
|
|
m_size += 1;
|
|
}
|
|
|
|
void KBufferBuilder::appendvf(const char* fmt, va_list ap)
|
|
{
|
|
printf_internal([this](char*&, char ch) {
|
|
append(ch);
|
|
},
|
|
nullptr, fmt, ap);
|
|
}
|
|
|
|
void KBufferBuilder::appendf(const char* fmt, ...)
|
|
{
|
|
va_list ap;
|
|
va_start(ap, fmt);
|
|
appendvf(fmt, ap);
|
|
va_end(ap);
|
|
}
|
|
|
|
void KBufferBuilder::append_escaped_for_json(const StringView& string)
|
|
{
|
|
for (auto ch : string) {
|
|
switch (ch) {
|
|
case '\e':
|
|
append("\\u001B");
|
|
break;
|
|
case '\b':
|
|
append("\\b");
|
|
break;
|
|
case '\n':
|
|
append("\\n");
|
|
break;
|
|
case '\t':
|
|
append("\\t");
|
|
break;
|
|
case '\"':
|
|
append("\\\"");
|
|
break;
|
|
case '\\':
|
|
append("\\\\");
|
|
break;
|
|
default:
|
|
append(ch);
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|