123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126 |
- /*
- * 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);
- }
- }
- }
- }
|