Atmosphere/stratosphere/creport/source/creport_scoped_file.cpp

/*
 * Copyright (c) Atmosphère-NX
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
#include <stratosphere.hpp>
#include "creport_scoped_file.hpp"

namespace ams::creport {

    namespace {

        /* Convenience definitions. */
        constexpr size_t MaximumLineLength = 0x20;

        constinit os::SdkMutex g_format_lock;
        constinit char g_format_buffer[2 * os::MemoryPageSize];

    }

    void ScopedFile::WriteString(const char *str) {
        this->Write(str, std::strlen(str));
    }

    void ScopedFile::WriteFormat(const char *fmt, ...) {
        /* Acquire exclusive access to the format buffer. */
        std::scoped_lock lk(g_format_lock);

        /* Format to the buffer. */
        {
            std::va_list vl;
            va_start(vl, fmt);
            util::VSNPrintf(g_format_buffer, sizeof(g_format_buffer), fmt, vl);
            va_end(vl);
        }

        /* Write data. */
        this->WriteString(g_format_buffer);
    }

    void ScopedFile::DumpMemory(const char *prefix, const void *data, size_t size) {
        const u8 *data_u8 = reinterpret_cast<const u8 *>(data);
        const int prefix_len = std::strlen(prefix);
        size_t data_ofs = 0;
        size_t remaining = size;
        bool first = true;

        while (remaining) {
            const size_t cur_size = std::min(MaximumLineLength, remaining);

            /* Print the line prefix. */
            if (first) {
                this->Write(prefix, prefix_len);
                first = false;
            } else {
                std::memset(g_format_buffer, ' ', prefix_len);
                this->Write(g_format_buffer, prefix_len);
            }

            /* Dump up to 0x20 of hex memory. */
            {
                char hex[MaximumLineLength * 2 + 2] = {};
                for (size_t i = 0; i < cur_size; i++) {
                    hex[i * 2 + 0] = "0123456789ABCDEF"[data_u8[data_ofs] >> 4];
                    hex[i * 2 + 1] = "0123456789ABCDEF"[data_u8[data_ofs] & 0xF];
                    ++data_ofs;
                }
                hex[cur_size * 2 + 0] = '\n';

                this->Write(hex, cur_size * 2 + 1);
            }

            /* Continue. */
            remaining -= cur_size;
        }
    }


    void ScopedFile::Write(const void *data, size_t size) {
        /* If we're not open, we can't write. */
        if (!this->IsOpen()) {
            return;
        }

        /* If we have a cache, write to it. */
        if (m_cache != nullptr) {
            /* Write into the cache, if we can. */
            if (m_cache_size - m_cache_offset >= size || R_SUCCEEDED(this->TryWriteCache())) {
                std::memcpy(m_cache + m_cache_offset, data, size);
                m_cache_offset += size;
            }
        } else {
            /* Advance, if we write successfully. */
            if (R_SUCCEEDED(fs::WriteFile(m_file, m_offset, data, size, fs::WriteOption::None))) {
                m_offset += size;
            }
        }
    }

    Result ScopedFile::TryWriteCache() {
        /* If there's no cached data, there's nothing to do. */
        R_SUCCEED_IF(m_cache_offset == 0);

        /* Try to write any cached data. */
        R_TRY(fs::WriteFile(m_file, m_offset, m_cache, m_cache_offset, fs::WriteOption::None));

        /* Update our extents. */
        m_offset += m_cache_offset;
        m_cache_offset = 0;

        R_SUCCEED();
    }

}