mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-11-28 01:20:58 +01:00
124 lines
3.9 KiB
C++
124 lines
3.9 KiB
C++
/*
|
|
* 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();
|
|
}
|
|
|
|
} |