mirror of
				https://github.com/Atmosphere-NX/Atmosphere.git
				synced 2025-11-04 04:51:16 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			75 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			75 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
 * Copyright (c) 2018-2020 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 log file. */
 | 
						|
                mitm::fs::CreateAtmosphereSdFile("dns_log.txt", 0, ams::fs::CreateOption_None);
 | 
						|
 | 
						|
                /* Open the log file. */
 | 
						|
                R_ABORT_UNLESS(mitm::fs::OpenAtmosphereSdFile(std::addressof(g_log_file), "dns_log.txt", 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;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
}
 |