123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109 |
- #include <flutter/dart_project.h>
- #include <flutter/flutter_view_controller.h>
- #include <windows.h>
- #include "flutter_window.h"
- #include "utils.h"
- #include <protocol_handler/protocol_handler_plugin.h>
- BOOL isRunAsAdmin() {
- BOOL isRunAsAdmin = FALSE;
- HANDLE hToken = NULL;
- if (!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
- {
- return FALSE;
- }
- TOKEN_ELEVATION tokenEle;
- DWORD dwRetLen = 0;
- if (GetTokenInformation(hToken, TokenElevation, &tokenEle, sizeof(tokenEle), &dwRetLen))
- {
- if (dwRetLen == sizeof(tokenEle))
- {
- isRunAsAdmin = tokenEle.TokenIsElevated;
- }
- }
- CloseHandle(hToken);
- return isRunAsAdmin;
- }
- void runAsAdmin() {
- WCHAR czFileName[1024] = { 0 };
- GetModuleFileName(NULL, czFileName, _countof(czFileName) - 1);
- SHELLEXECUTEINFO EI;
- memset(&EI, 0, sizeof(EI));
- EI.cbSize = sizeof(SHELLEXECUTEINFO);
- EI.lpVerb = TEXT("runas");
- EI.fMask = 0x00000040;
- EI.lpFile = czFileName;
- EI.nShow = SW_SHOW;
- ShellExecuteEx(&EI);
- }
- int APIENTRY wWinMain(_In_ HINSTANCE instance, _In_opt_ HINSTANCE prev,
- _In_ wchar_t *command_line, _In_ int show_command) {
- if(!::IsDebuggerPresent()){
- if (!isRunAsAdmin()) {
- runAsAdmin();
- exit(0);
- }
- if (wcsstr(command_line, L"--runas-admin") != NULL) {
- }
- else {
- HWND hwnd = ::FindWindow(L"FLUTTER_RUNNER_WIN32_WINDOW", L"naiyouwl");
- if (hwnd != NULL) {
- DispatchToProtocolHandler(hwnd);
- ::ShowWindow(hwnd, SW_NORMAL);
- ::SetForegroundWindow(hwnd);
- return EXIT_FAILURE;
- }
- }
- }
- /*
- *
- else {
- AllocConsole();
- ShowWindow(GetConsoleWindow(), SW_HIDE);
- }
- * */
- // https://github.com/flutter/flutter/issues/47891#issuecomment-708850435
- // https://github.com/flutter/flutter/issues/47891#issuecomment-869729956
- // https://github.com/dart-lang/sdk/issues/39945#issuecomment-870428151
- if (!::AttachConsole(ATTACH_PARENT_PROCESS) && ::IsDebuggerPresent()) {
- CreateAndAttachConsole();
- }
- else {
- AllocConsole();
- ShowWindow(GetConsoleWindow(), SW_HIDE);
- }
- // Initialize COM, so that it is available for use in the library and/or
- // plugins.
- ::CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED);
- flutter::DartProject project(L"data");
- std::vector<std::string> command_line_arguments =
- GetCommandLineArguments();
- project.set_dart_entrypoint_arguments(std::move(command_line_arguments));
- FlutterWindow window(project);
- Win32Window::Point origin(10, 10);
- Win32Window::Size size(1280, 720);
- if (!window.Create(L"naiyouwl", origin, size)) {
- return EXIT_FAILURE;
- }
- window.SetQuitOnClose(true);
- ::MSG msg;
- while (::GetMessage(&msg, nullptr, 0, 0)) {
- ::TranslateMessage(&msg);
- ::DispatchMessage(&msg);
- }
- ::CoUninitialize();
- return EXIT_SUCCESS;
- }
|