Atmosphere/stratosphere/ams_mitm/source/dns_mitm/dnsmitm_debug.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 "../amsmitm_fs_utils.hpp"
#include "dnsmitm_debug.hpp"

namespace ams::mitm::socket::resolver {

    namespace {

        constinit os::SdkMutex g_log_mutex;
        constinit bool     g_log_enabled;

        constinit ::FsFile g_log_file;
        constinit s64      g_log_ofs;


        constinit char g_log_buf[0x400];

    }

    void InitializeDebug(bool enable_log) {
        {
            std::scoped_lock lk(g_log_mutex);

            g_log_enabled = enable_log;

            if (g_log_enabled) {
                /* Create the logs directory. */
                mitm::fs::CreateAtmosphereSdDirectory("/logs");

                /* Create the log file. */
                mitm::fs::CreateAtmosphereSdFile("/logs/dns_mitm_debug.log", 0, ams::fs::CreateOption_None);

                /* Open the log file. */
                R_ABORT_UNLESS(mitm::fs::OpenAtmosphereSdFile(std::addressof(g_log_file), "/logs/dns_mitm_debug.log", ams::fs::OpenMode_ReadWrite | ams::fs::OpenMode_AllowAppend));

                /* Get the current log offset. */
                R_ABORT_UNLESS(::fsFileGetSize(std::addressof(g_log_file), std::addressof(g_log_ofs)));
            }
        }

        /* Start a new log. */
        LogDebug("\n---\n");
    }

    void LogDebug(const char *fmt, ...) {
        std::scoped_lock lk(g_log_mutex);

        if (g_log_enabled) {
            int len = 0;
            {
                std::va_list vl;
                va_start(vl, fmt);
                len = util::VSNPrintf(g_log_buf, sizeof(g_log_buf), fmt, vl);
                va_end(vl);
            }

            R_ABORT_UNLESS(::fsFileWrite(std::addressof(g_log_file), g_log_ofs, g_log_buf, len, FsWriteOption_Flush));
            g_log_ofs += len;
        }
    }

}