mirror of
				https://github.com/microsoft/playwright.git
				synced 2025-06-26 21:40:17 +00:00 
			
		
		
		
	 a3f34fb4b7
			
		
	
	
		a3f34fb4b7
		
			
		
	
	
	
	
		
			
			This leaves our firefox diff to gecko instrumentation changes only. Drive-by: rename webkit "src" folder into "embedder".
		
			
				
	
	
		
			270 lines
		
	
	
		
			8.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			270 lines
		
	
	
		
			8.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /*
 | |
|  * Copyright (C) 2006, 2008, 2013-2015 Apple Inc.  All rights reserved.
 | |
|  * Copyright (C) 2009, 2011 Brent Fulgham.  All rights reserved.
 | |
|  * Copyright (C) 2009, 2010, 2011 Appcelerator, Inc. All rights reserved.
 | |
|  * Copyright (C) 2013 Alex Christensen. All rights reserved.
 | |
|  *
 | |
|  * Redistribution and use in source and binary forms, with or without
 | |
|  * modification, are permitted provided that the following conditions
 | |
|  * are met:
 | |
|  * 1. Redistributions of source code must retain the above copyright
 | |
|  *    notice, this list of conditions and the following disclaimer.
 | |
|  * 2. Redistributions in binary form must reproduce the above copyright
 | |
|  *    notice, this list of conditions and the following disclaimer in the
 | |
|  *    documentation and/or other materials provided with the distribution.
 | |
|  *
 | |
|  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
 | |
|  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 | |
|  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 | |
|  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
 | |
|  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 | |
|  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 | |
|  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 | |
|  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
 | |
|  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 | |
|  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 | |
|  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
 | |
|  */
 | |
| 
 | |
| #include "stdafx.h"
 | |
| #include "Common.h"
 | |
| 
 | |
| #include "DialogHelper.h"
 | |
| #include "PlaywrightLibResource.h"
 | |
| #include "PlaywrightReplace.h"
 | |
| #include <dbghelp.h>
 | |
| #include <shlobj.h>
 | |
| #include <wtf/Optional.h>
 | |
| #include <wtf/StdLibExtras.h>
 | |
| #include <vector>
 | |
| 
 | |
| // Global Variables:
 | |
| HINSTANCE hInst;
 | |
| 
 | |
| // Support moving the transparent window
 | |
| POINT s_windowPosition = { 100, 100 };
 | |
| SIZE s_windowSize = { 500, 200 };
 | |
| 
 | |
| namespace WebCore {
 | |
| float deviceScaleFactorForWindow(HWND);
 | |
| }
 | |
| 
 | |
| void computeFullDesktopFrame()
 | |
| {
 | |
|     RECT desktop;
 | |
|     if (!::SystemParametersInfo(SPI_GETWORKAREA, 0, static_cast<void*>(&desktop), 0))
 | |
|         return;
 | |
| 
 | |
|     float scaleFactor = WebCore::deviceScaleFactorForWindow(nullptr);
 | |
| 
 | |
|     s_windowPosition.x = 0;
 | |
|     s_windowPosition.y = 0;
 | |
|     s_windowSize.cx = scaleFactor * (desktop.right - desktop.left);
 | |
|     s_windowSize.cy = scaleFactor * (desktop.bottom - desktop.top);
 | |
| }
 | |
| 
 | |
| BOOL WINAPI DllMain(HINSTANCE dllInstance, DWORD reason, LPVOID)
 | |
| {
 | |
|     if (reason == DLL_PROCESS_ATTACH)
 | |
|         hInst = dllInstance;
 | |
| 
 | |
|     return TRUE;
 | |
| }
 | |
| 
 | |
| bool getAppDataFolder(_bstr_t& directory)
 | |
| {
 | |
|     wchar_t appDataDirectory[MAX_PATH];
 | |
|     if (FAILED(SHGetFolderPathW(0, CSIDL_LOCAL_APPDATA | CSIDL_FLAG_CREATE, 0, 0, appDataDirectory)))
 | |
|         return false;
 | |
| 
 | |
|     wchar_t executablePath[MAX_PATH];
 | |
|     if (!::GetModuleFileNameW(0, executablePath, MAX_PATH))
 | |
|         return false;
 | |
| 
 | |
|     ::PathRemoveExtensionW(executablePath);
 | |
| 
 | |
|     directory = _bstr_t(appDataDirectory) + L"\\" + ::PathFindFileNameW(executablePath);
 | |
| 
 | |
|     return true;
 | |
| }
 | |
| 
 | |
| void createCrashReport(EXCEPTION_POINTERS* exceptionPointers)
 | |
| {
 | |
|     _bstr_t directory;
 | |
| 
 | |
|     if (!getAppDataFolder(directory))
 | |
|         return;
 | |
| 
 | |
|     if (::SHCreateDirectoryEx(0, directory, 0) != ERROR_SUCCESS
 | |
|         && ::GetLastError() != ERROR_FILE_EXISTS
 | |
|         && ::GetLastError() != ERROR_ALREADY_EXISTS)
 | |
|         return;
 | |
| 
 | |
|     std::wstring fileName = std::wstring(static_cast<const wchar_t*>(directory)) + L"\\CrashReport.dmp";
 | |
|     HANDLE miniDumpFile = ::CreateFile(fileName.c_str(), GENERIC_WRITE, 0, 0, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
 | |
| 
 | |
|     if (miniDumpFile && miniDumpFile != INVALID_HANDLE_VALUE) {
 | |
| 
 | |
|         MINIDUMP_EXCEPTION_INFORMATION mdei;
 | |
|         mdei.ThreadId = ::GetCurrentThreadId();
 | |
|         mdei.ExceptionPointers  = exceptionPointers;
 | |
|         mdei.ClientPointers = 0;
 | |
| 
 | |
| #ifdef _DEBUG
 | |
|         MINIDUMP_TYPE dumpType = MiniDumpWithFullMemory;
 | |
| #else
 | |
|         MINIDUMP_TYPE dumpType = MiniDumpNormal;
 | |
| #endif
 | |
| 
 | |
|         ::MiniDumpWriteDump(::GetCurrentProcess(), ::GetCurrentProcessId(), miniDumpFile, dumpType, &mdei, 0, 0);
 | |
|         ::CloseHandle(miniDumpFile);
 | |
|         processCrashReport(fileName.c_str());
 | |
|     }
 | |
| }
 | |
| 
 | |
| Optional<Credential> askCredential(HWND hwnd, const std::wstring& realm)
 | |
| {
 | |
|     struct AuthDialog : public Dialog {
 | |
|         std::wstring realm;
 | |
|         Credential credential;
 | |
| 
 | |
|     protected:
 | |
|         void setup()
 | |
|         {
 | |
|             setText(IDC_REALM_TEXT, realm);
 | |
|         }
 | |
| 
 | |
|         void ok() final
 | |
|         {
 | |
|             credential.username = getText(IDC_AUTH_USER);
 | |
|             credential.password = getText(IDC_AUTH_PASSWORD);
 | |
|         }
 | |
|     };
 | |
| 
 | |
|     AuthDialog dialog;
 | |
|     dialog.realm = realm;
 | |
| 
 | |
|     if (dialog.run(hInst, hwnd, IDD_AUTH))
 | |
|         return dialog.credential;
 | |
|     return WTF::nullopt;
 | |
| }
 | |
| 
 | |
| bool askServerTrustEvaluation(HWND hwnd, const std::wstring& text)
 | |
| {
 | |
|     class ServerTrustEvaluationDialog : public Dialog {
 | |
|     public:
 | |
|         ServerTrustEvaluationDialog(const std::wstring& text)
 | |
|             : m_text { text }
 | |
|         {
 | |
|             SendMessage(GetDlgItem(this->hDlg(), IDC_SERVER_TRUST_TEXT), WM_SETFONT, (WPARAM)GetStockObject(ANSI_FIXED_FONT), TRUE);
 | |
|         }
 | |
| 
 | |
|     protected:
 | |
|         std::wstring m_text;
 | |
| 
 | |
|         void setup()
 | |
|         {
 | |
|             setText(IDC_SERVER_TRUST_TEXT, m_text);
 | |
|         }
 | |
| 
 | |
|         void ok() final
 | |
|         {
 | |
| 
 | |
|         }
 | |
|     };
 | |
| 
 | |
|     ServerTrustEvaluationDialog dialog { text };
 | |
|     return dialog.run(hInst, hwnd, IDD_SERVER_TRUST);
 | |
| }
 | |
| 
 | |
| CommandLineOptions parseCommandLine()
 | |
| {
 | |
|     CommandLineOptions options;
 | |
| 
 | |
|     int argc = 0;
 | |
|     WCHAR** argv = CommandLineToArgvW(GetCommandLineW(), &argc);
 | |
|     for (int i = 1; i < argc; ++i) {
 | |
|         if (!wcsicmp(argv[i], L"--desktop"))
 | |
|             options.useFullDesktop = true;
 | |
|         else if (!wcsicmp(argv[i], L"--inspector-pipe"))
 | |
|             options.inspectorPipe = true;
 | |
|         else if (!wcsncmp(argv[i], L"--user-data-dir=", 16))
 | |
|             options.userDataDir = argv[i] + 16;
 | |
|         else if (!wcsncmp(argv[i], L"--curl-proxy=", 13))
 | |
|             options.curloptProxy = argv[i] + 13;
 | |
|         else if (!wcsncmp(argv[i], L"--curl-noproxy=", 15))
 | |
|             options.curloptNoproxy = argv[i] + 15;
 | |
|         else if (!wcsicmp(argv[i], L"--headless"))
 | |
|             options.headless = true;
 | |
|         else if (!wcsicmp(argv[i], L"--no-startup-window"))
 | |
|             options.noStartupWindow = true;
 | |
|         else if (!options.requestedURL)
 | |
|             options.requestedURL = argv[i];
 | |
|     }
 | |
| 
 | |
|     return options;
 | |
| }
 | |
| 
 | |
| std::wstring replaceString(std::wstring src, const std::wstring& oldValue, const std::wstring& newValue)
 | |
| {
 | |
|     if (src.empty() || oldValue.empty())
 | |
|         return src;
 | |
| 
 | |
|     size_t pos = 0;
 | |
|     while ((pos = src.find(oldValue, pos)) != src.npos) {
 | |
|         src.replace(pos, oldValue.length(), newValue);
 | |
|         pos += newValue.length();
 | |
|     }
 | |
| 
 | |
|     return src;
 | |
| }
 | |
| 
 | |
| std::wstring createString(WKStringRef wkString)
 | |
| {
 | |
|     size_t maxSize = WKStringGetLength(wkString);
 | |
| 
 | |
|     std::vector<WKChar> wkCharBuffer(maxSize);
 | |
|     size_t actualLength = WKStringGetCharacters(wkString, wkCharBuffer.data(), maxSize);
 | |
|     return std::wstring(wkCharBuffer.data(), actualLength);
 | |
| }
 | |
| 
 | |
| std::wstring createString(WKURLRef wkURL)
 | |
| {
 | |
|     if (!wkURL)
 | |
|         return { };
 | |
|     WKRetainPtr<WKStringRef> url = adoptWK(WKURLCopyString(wkURL));
 | |
|     return createString(url.get());
 | |
| }
 | |
| 
 | |
| std::string createUTF8String(const wchar_t* src, size_t srcLength)
 | |
| {
 | |
|     int length = WideCharToMultiByte(CP_UTF8, 0, src, srcLength, 0, 0, nullptr, nullptr);
 | |
|     std::vector<char> buffer(length);
 | |
|     size_t actualLength = WideCharToMultiByte(CP_UTF8, 0, src, srcLength, buffer.data(), length, nullptr, nullptr);
 | |
|     return { buffer.data(), actualLength };
 | |
| }
 | |
| 
 | |
| WKRetainPtr<WKStringRef> createWKString(_bstr_t str)
 | |
| {
 | |
|     auto utf8 = createUTF8String(str, str.length());
 | |
|     return adoptWK(WKStringCreateWithUTF8CString(utf8.data()));
 | |
| }
 | |
| 
 | |
| WKRetainPtr<WKStringRef> createWKString(const std::wstring& str)
 | |
| {
 | |
|     auto utf8 = createUTF8String(str.c_str(), str.length());
 | |
|     return adoptWK(WKStringCreateWithUTF8CString(utf8.data()));
 | |
| }
 | |
| 
 | |
| WKRetainPtr<WKURLRef> createWKURL(_bstr_t str)
 | |
| {
 | |
|     auto utf8 = createUTF8String(str, str.length());
 | |
|     return adoptWK(WKURLCreateWithUTF8CString(utf8.data()));
 | |
| }
 | |
| 
 | |
| WKRetainPtr<WKURLRef> createWKURL(const std::wstring& str)
 | |
| {
 | |
|     auto utf8 = createUTF8String(str.c_str(), str.length());
 | |
|     return adoptWK(WKURLCreateWithUTF8CString(utf8.data()));
 | |
| }
 |