Atmosphere/libraries/libstratosphere/source/os/impl/os_initialize.os.windows.cpp

107 lines
2.8 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 "os_resource_manager.hpp"
namespace ams {
void Main();
namespace init {
void InitializeDefaultAllocator();
}
/* TODO: This should probably instead be a custom init::Initialize*()? */
namespace fs {
void InitializeForHostTool();
}
}
namespace ams::os {
void SetHostArgc(int argc);
void SetHostArgv(char **argv);
namespace {
void SetupWindowsConsole(DWORD which) {
/* Get handle to standard device. */
const auto handle = ::GetStdHandle(which);
if (handle == INVALID_HANDLE_VALUE) {
return;
}
/* Get the console mode. */
DWORD mode;
if (!::GetConsoleMode(handle, std::addressof(mode))) {
return;
}
/* Enable printing with ANSI escape codes. */
mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING;
/* Set console mode. */
::SetConsoleMode(handle, mode);
}
}
void Initialize() {
/* Only allow os::Initialize to be called once. */
AMS_FUNCTION_LOCAL_STATIC_CONSTINIT(bool, s_initialized, false);
if (s_initialized) {
return;
}
s_initialized = true;
/* Initialize the global os resource manager. */
os::impl::ResourceManagerHolder::InitializeResourceManagerInstance();
/* Initialize virtual address memory. */
os::InitializeVirtualAddressMemory();
/* Ensure that the init library's allocator has been setup. */
init::InitializeDefaultAllocator();
/* Try to set up the windows console. */
SetupWindowsConsole(STD_OUTPUT_HANDLE);
SetupWindowsConsole(STD_ERROR_HANDLE);
}
}
extern "C" int WINAPI WinMain(HINSTANCE, HINSTANCE, LPSTR, int) {
/* Ensure os library is initialized. */
::ams::os::Initialize();
/* Set argc/argv. */
::ams::os::SetHostArgc(__argc);
::ams::os::SetHostArgv(__argv);
/* Ensure filesystem library is initialized. */
::ams::fs::InitializeForHostTool();
/* Call main. */
::ams::Main();
/* TODO: Should we try to implement a custom exit here? */
return 0;
}