Morph MORPH
SIGN IN SIGN UP
SeleniumHQ / selenium UNCLAIMED

A browser automation framework and ecosystem.

34179 0 143 Java
Normal View History Raw
Copyright uniformity for C++ projects Fixes copyright notices for files in the iedriver, iedriverserver, and webdriver-server, in accordance with issue #401.
2015-04-13 12:09:26 -04:00
// Licensed to the Software Freedom Conservancy (SFC) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The SFC licenses this file
// to you under the Apache License, Version 2.0 (the "License");
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "DocumentHost.h"
Moving implementation code of IE driver command handlers from headers The initial implementation of the IE driver's command handlers put the implementation for each command directly in the header files. While this works, it's not standard practice, and has led to a spaghetti-code of include statements. This takes the first step toward fixing that by moving the implementation of each command into a .cpp file, leaving the header to only contain the class definition.
2017-02-14 09:48:56 -08:00
Improving recognition of Protected Mode boundary crossing in IE In particular, when navigating, properly detect URLs beginning with "about:blank", not just that being the entire URL.
2019-02-22 08:13:23 -08:00
#include <IEPMapi.h>
Slightly modified setting IE window to have focus when sending input This code only affects the browser when the requireWindowFocus capability is set at session start.
2019-07-09 14:13:42 -07:00
#include <UIAutomation.h>
Improving recognition of Protected Mode boundary crossing in IE In particular, when navigating, properly detect URLs beginning with "about:blank", not just that being the entire URL.
2019-02-22 08:13:23 -08:00
Moving implementation code of IE driver command handlers from headers The initial implementation of the IE driver's command handlers put the implementation for each command directly in the header files. While this works, it's not standard practice, and has led to a spaghetti-code of include statements. This takes the first step toward fixing that by moving the implementation of each command into a .cpp file, leaving the header to only contain the class definition.
2017-02-14 09:48:56 -08:00
#include "errorcodes.h"
#include "logging.h"
Rewriting cookie handling for IE. The IE driver should now be able to return full information for most cookies, including expiration date and secure status. Additionally, the driver will now return HTTP-only cookies from IE. Note that some specific cookie information will still be incomplete. In particular, session cookies that may have been set with domain, path, or secure tokens in the cookie string will not have those tokens set in the returned cookie, as that information is not available via any API. Also note that this change introduces additional use of Windows APIs that are not supported when run from a service process, making that more completely unsupported. Fixes issue #391.
2015-06-23 18:18:18 -04:00
#include "BrowserCookie.h"
Cleaning up #includes in C++ code Also updating Visual Studio solution file to build correct projects in proper platform/configurations.
2014-08-19 21:11:00 +00:00
#include "BrowserFactory.h"
Rewriting cookie handling for IE. The IE driver should now be able to return full information for most cookies, including expiration date and secure status. Additionally, the driver will now return HTTP-only cookies from IE. Note that some specific cookie information will still be incomplete. In particular, session cookies that may have been set with domain, path, or secure tokens in the cookie string will not have those tokens set in the returned cookie, as that information is not available via any API. Also note that this change introduces additional use of Windows APIs that are not supported when run from a service process, making that more completely unsupported. Fixes issue #391.
2015-06-23 18:18:18 -04:00
#include "CookieManager.h"
Improving recognition of Protected Mode boundary crossing in IE In particular, when navigating, properly detect URLs beginning with "about:blank", not just that being the entire URL.
2019-02-22 08:13:23 -08:00
#include "HookProcessor.h"
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
#include "messages.h"
Refactored IE driver registry access methods to RegistryUtilities class. This eliminates the need for an unnecessary extra instance of the BrowserFactory class.
2013-07-10 15:53:29 -04:00
#include "RegistryUtilities.h"
Cleaning up #includes in C++ code Also updating Visual Studio solution file to build correct projects in proper platform/configurations.
2014-08-19 21:11:00 +00:00
#include "Script.h"
Moving implementation code of IE driver command handlers from headers The initial implementation of the IE driver's command handlers put the implementation for each command directly in the header files. While this works, it's not standard practice, and has led to a spaghetti-code of include statements. This takes the first step toward fixing that by moving the implementation of each command into a .cpp file, leaving the header to only contain the class definition.
2017-02-14 09:48:56 -08:00
#include "StringUtilities.h"
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
namespace webdriver {
DocumentHost::DocumentHost(HWND hwnd, HWND executor_handle) {
LOG(TRACE) << "Entering DocumentHost::DocumentHost";
// NOTE: COM should be initialized on this thread, so we
// could use CoCreateGuid() and StringFromGUID2() instead.
UUID guid;
RPC_WSTR guid_string = NULL;
RPC_STATUS status = ::UuidCreate(&guid);
Fixing issues found via static code analysis for IE driver. No functional changes
2013-02-25 11:32:30 -05:00
if (status != RPC_S_OK) {
// If we encounter an error, not bloody much we can do about it.
// Just log it and continue.
LOG(WARN) << "UuidCreate returned a status other then RPC_S_OK: " << status;
}
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
status = ::UuidToString(&guid, &guid_string);
Fixing issues found via static code analysis for IE driver. No functional changes
2013-02-25 11:32:30 -05:00
if (status != RPC_S_OK) {
// If we encounter an error, not bloody much we can do about it.
// Just log it and continue.
LOG(WARN) << "UuidToString returned a status other then RPC_S_OK: " << status;
}
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
// RPC_WSTR is currently typedef'd in RpcDce.h (pulled in by rpc.h)
// as unsigned short*. It needs to be typedef'd as wchar_t*
wchar_t* cast_guid_string = reinterpret_cast<wchar_t*>(guid_string);
Replacing CA2W and CW2A macros in IE driver with more transparent StringUtilities class
2013-03-21 13:58:47 -04:00
this->browser_id_ = StringUtilities::ToString(cast_guid_string);
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
::RpcStringFree(&guid_string);
this->window_handle_ = hwnd;
this->executor_handle_ = executor_handle;
Refactor IE user-initiated JavaScript execution to use JSON objects This IE driver refactor is internal, but should not change behavior of the driver. It is intended to let the driver's internal script executor use the direct internal JSON objects as arguments for script execution, deferring their conversion into variants until directly before execution.
2018-02-01 16:08:30 -08:00
this->script_executor_handle_ = NULL;
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
this->is_closing_ = false;
this->wait_required_ = false;
Improving IE driver use with invalid Protected Mode settings Now, when the user does not set the Protected Mode settings of the browser and sends the capability to bypass the checks for those settings, the driver will attempt to predict when a Protected Mode boundary will be crossed, and set in motion a process to reattach itself to the newly created browser. This process is far from perfect. It is subject to really challenging race conditions that are truly impossible to eliminate entirely, because of the architecture of the browser itself. Nevertheless, even in its flawed state, this is still a better outcome than it was previously for users. Please note that the advice and support policy of the IE driver will continue to be that the user must set the Protected Mode settings of the browser properly before using the driver. Any "issues" that arise by not having the settings set, but that disappear when the settings are corrected, are not considered by the project to be valid issues. This will include, but not be limited to, issues like abandoned browser instances not being closed, use of multiple instances of the driver where the wrong browser window is connected to and automated, and issues where the driver appears to hang upon navigation to a new page. If the problem disappears when the browser is properly configured, any issue reports will be immediately closed with a note to properly configure the browser and remove the capability. The following situations should be at least partially mitigated by the change: * Navigation to a new page * Clicking on a link (specifically an <a> tag) that will lead to a navigation to a new page * Clicking a link that opens a new window Other cases, like navigating backward and forward through the browser history, clicking an element that submits a form, and so on, may not be handled. In those cases, issue reports will be summarily closed, unless a specific pull request fixing the issue is also provided. Additionally, use of things like proxies to capture traffic between the browser and web server may miss some traffic because of the race conditions inherent in the mechanism used to reattach to a newly created browser. Again, these race conditions are unavoidable, and issue reports that are based on them will be immediately closed with a note indicating that the browser must have its settings properly set. These strict guidelines are not intended to be harsh, and are not put in place with the intent to avoid investigating and fixing issues; rather, they must be enforced because the underlying architecture of the browser makes them unavoidable.
2019-02-17 09:29:01 -08:00
this->is_awaiting_new_process_ = false;
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
this->focused_frame_window_ = NULL;
Rewriting cookie handling for IE. The IE driver should now be able to return full information for most cookies, including expiration date and secure status. Additionally, the driver will now return HTTP-only cookies from IE. Note that some specific cookie information will still be incomplete. In particular, session cookies that may have been set with domain, path, or secure tokens in the cookie string will not have those tokens set in the returned cookie, as that information is not available via any API. Also note that this change introduces additional use of Windows APIs that are not supported when run from a service process, making that more completely unsupported. Fixes issue #391.
2015-06-23 18:18:18 -04:00
this->cookie_manager_ = new CookieManager();
Updating to manually initialize cookie manager on new window open for IE When a new window is requested by the browser, the driver creates one, but does not set the window handle of the window to a proper value. This is normally not an issue, but in the edge case where the user wants to manipulate cookies in the new window, the cookie manager does not have a window handle set to facilitate the transfer of cookie information from the driver code to the browser. This change rectifies that problem.
2018-09-10 16:23:01 -07:00
if (this->window_handle_ != NULL) {
this->cookie_manager_->Initialize(this->window_handle_);
}
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
}
DocumentHost::~DocumentHost(void) {
Rewriting cookie handling for IE. The IE driver should now be able to return full information for most cookies, including expiration date and secure status. Additionally, the driver will now return HTTP-only cookies from IE. Note that some specific cookie information will still be incomplete. In particular, session cookies that may have been set with domain, path, or secure tokens in the cookie string will not have those tokens set in the returned cookie, as that information is not available via any API. Also note that this change introduces additional use of Windows APIs that are not supported when run from a service process, making that more completely unsupported. Fixes issue #391.
2015-06-23 18:18:18 -04:00
delete this->cookie_manager_;
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
}
std::string DocumentHost::GetCurrentUrl() {
LOG(TRACE) << "Entering DocumentHost::GetCurrentUrl";
CComPtr<IHTMLDocument2> doc;
this->GetDocument(&doc);
if (!doc) {
Adding ability to get current URL from IWebBrowser2 if IHTMLDocument2 fails. Under normal circumstances, the IE driver gets the current URL from the document loaded in the browser. When IE browses to a non-HTML page, such as a PDF file with the Adobe Reader plugin installed, getting the document fails, since there is no "document" in the browser. In that case, fall back to getting the LocationURL property of the IWebBrowser2 interface. It's arguable that this is the wrong thing to do, since the driver is useless without an actual document, but we'll let this work anyway. Fixes issue #5361.
2013-03-19 13:03:28 -04:00
LOG(WARN) << "Unable to get document object, DocumentHost::GetDocument returned NULL. "
<< "Attempting to get URL from IWebBrowser2 object";
return this->GetBrowserUrl();
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
}
CComBSTR url;
HRESULT hr = doc->get_URL(&url);
if (FAILED(hr)) {
LOGHR(WARN, hr) << "Unable to get current URL, call to IHTMLDocument2::get_URL failed";
return "";
}
Updating IE to return top-level URL when focused on a frame. This makes the driver now consistent with other implementations.
2015-10-08 13:31:24 -07:00
std::wstring converted_url(url, ::SysStringLen(url));
Replacing CA2W and CW2A macros in IE driver with more transparent StringUtilities class
2013-03-21 13:58:47 -04:00
std::string current_url = StringUtilities::ToString(converted_url);
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
return current_url;
}
std::string DocumentHost::GetPageSource() {
LOG(TRACE) << "Entering DocumentHost::GetPageSource";
CComPtr<IHTMLDocument2> doc;
this->GetDocument(&doc);
Added error checking in DocumentHost::GetPageSource. Fixes issue #5967.
2013-07-30 18:36:18 -04:00
if (!doc) {
LOG(WARN) << "Unable to get document object, DocumentHost::GetDocument did not return a valid IHTMLDocument2 pointer";
return "";
}
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
CComPtr<IHTMLDocument3> doc3;
Removing use of CComQIPtr from IE driver native code. The use of CComQIPtr is not needed, as we are substituting a call to the QueryInterface method instead. This is what CComQIPtr does under the covers anyway.
2013-03-07 17:06:31 -05:00
HRESULT hr = doc->QueryInterface<IHTMLDocument3>(&doc3);
if (FAILED(hr) || !doc3) {
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
LOG(WARN) << "Unable to get document object, QueryInterface to IHTMLDocument3 failed";
return "";
}
CComPtr<IHTMLElement> document_element;
Removing use of CComQIPtr from IE driver native code. The use of CComQIPtr is not needed, as we are substituting a call to the QueryInterface method instead. This is what CComQIPtr does under the covers anyway.
2013-03-07 17:06:31 -05:00
hr = doc3->get_documentElement(&document_element);
if (FAILED(hr) || !document_element) {
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
LOGHR(WARN, hr) << "Unable to get document element from page, call to IHTMLDocument3::get_documentElement failed";
return "";
}
CComBSTR html;
hr = document_element->get_outerHTML(&html);
if (FAILED(hr)) {
LOGHR(WARN, hr) << "Have document element but cannot read source, call to IHTMLElement::get_outerHTML failed";
return "";
}
Replacing CA2W and CW2A macros in IE driver with more transparent StringUtilities class
2013-03-21 13:58:47 -04:00
std::wstring converted_html = html;
std::string page_source = StringUtilities::ToString(converted_html);
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
return page_source;
}
Updated IE driver to better support W3C WebDriver Specification This is a large update that includes implementation of the following things that are part of the WebDriver spec: * Implement the "get element property" command * Implement the "get named cookie" command * Implement the "minimize window" command * Implement the "fullscreen window" command * Update "set window rect" command to correctly recognize and restore from minimized or full screen state * Fix the "release actions" command to properly roll back all pending actions * Update the "perform actions" command to be more robust for pointer actions * Fix alert handling to handle dialogs with the "Do not show any more dialogs" check box * Added handling for the "dismiss and notify" and "accept and notify" user prompt handler states
2017-08-31 09:41:44 -07:00
void DocumentHost::Restore(void) {
if (this->IsFullScreen()) {
this->SetFullScreen(false);
}
HWND window_handle = this->GetTopLevelWindowHandle();
if (::IsZoomed(window_handle) || ::IsIconic(window_handle)) {
::ShowWindow(window_handle, SW_RESTORE);
}
}
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
int DocumentHost::SetFocusedFrameByElement(IHTMLElement* frame_element) {
LOG(TRACE) << "Entering DocumentHost::SetFocusedFrameByElement";
HRESULT hr = S_OK;
if (!frame_element) {
this->focused_frame_window_ = NULL;
Updating IEDriver native code to use the new WD_SUCCESS #define instead of just SUCCESS
2013-01-16 14:37:30 -05:00
return WD_SUCCESS;
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
}
Removing use of CComQIPtr from IE driver native code. The use of CComQIPtr is not needed, as we are substituting a call to the QueryInterface method instead. This is what CComQIPtr does under the covers anyway.
2013-03-07 17:06:31 -05:00
CComPtr<IHTMLWindow2> interim_result;
Treat object tag with embedded document as frame Signed-off-by: Jim Evans <james.h.evans.jr@gmail.com>
2017-02-27 10:03:44 -05:00
CComPtr<IHTMLObjectElement4> object_element;
hr = frame_element->QueryInterface<IHTMLObjectElement4>(&object_element);
if (SUCCEEDED(hr) && object_element) {
CComPtr<IDispatch> object_disp;
object_element->get_contentDocument(&object_disp);
if (!object_disp) {
LOG(WARN) << "Cannot get IDispatch interface from IHTMLObjectElement4 element";
return ENOSUCHFRAME;
}
CComPtr<IHTMLDocument2> object_doc;
object_disp->QueryInterface<IHTMLDocument2>(&object_doc);
if (!object_doc) {
LOG(WARN) << "Cannot get IHTMLDocument2 document from IDispatch reference";
return ENOSUCHFRAME;
}
hr = object_doc->get_parentWindow(&interim_result);
if (FAILED(hr)) {
LOGHR(WARN, hr) << "Cannot get parentWindow from IHTMLDocument2, call to IHTMLDocument2::get_parentWindow failed";
return ENOSUCHFRAME;
}
} else {
CComPtr<IHTMLFrameBase2> frame_base;
frame_element->QueryInterface<IHTMLFrameBase2>(&frame_base);
if (!frame_base) {
LOG(WARN) << "IHTMLElement is not a FRAME or IFRAME element";
return ENOSUCHFRAME;
}
hr = frame_base->get_contentWindow(&interim_result);
if (FAILED(hr)) {
LOGHR(WARN, hr) << "Cannot get contentWindow from IHTMLFrameBase2, call to IHTMLFrameBase2::get_contentWindow failed";
return ENOSUCHFRAME;
}
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
}
this->focused_frame_window_ = interim_result;
Updating IEDriver native code to use the new WD_SUCCESS #define instead of just SUCCESS
2013-01-16 14:37:30 -05:00
return WD_SUCCESS;
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
}
int DocumentHost::SetFocusedFrameByName(const std::string& frame_name) {
LOG(TRACE) << "Entering DocumentHost::SetFocusedFrameByName";
Simplifying object handling code in IE driver. COM objects are now correctly referenced everywhere via smart pointers, and use of some aliased types is now replaced by the C++ primitives.
2013-03-25 13:34:23 -04:00
CComVariant frame_identifier = StringUtilities::ToWString(frame_name).c_str();
return this->SetFocusedFrameByIdentifier(frame_identifier);
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
}
int DocumentHost::SetFocusedFrameByIndex(const int frame_index) {
LOG(TRACE) << "Entering DocumentHost::SetFocusedFrameByIndex";
Simplifying object handling code in IE driver. COM objects are now correctly referenced everywhere via smart pointers, and use of some aliased types is now replaced by the C++ primitives.
2013-03-25 13:34:23 -04:00
CComVariant frame_identifier;
frame_identifier.vt = VT_I4;
frame_identifier.lVal = frame_index;
return this->SetFocusedFrameByIdentifier(frame_identifier);
}
Implementing switchToParentFrame command for IE driver
2014-06-05 11:56:33 -04:00
void DocumentHost::SetFocusedFrameToParent() {
LOG(TRACE) << "Entering DocumentHost::SetFocusedFrameToParent";
// Three possible outcomes.
// Outcome 1: Already at top-level browsing context. No-op.
if (this->focused_frame_window_ != NULL) {
CComPtr<IHTMLWindow2> parent_window;
HRESULT hr = this->focused_frame_window_->get_parent(&parent_window);
if (FAILED(hr)) {
LOGHR(WARN, hr) << "IHTMLWindow2::get_parent call failed.";
}
CComPtr<IHTMLWindow2> top_window;
hr = this->focused_frame_window_->get_top(&top_window);
if (FAILED(hr)) {
LOGHR(WARN, hr) << "IHTMLWindow2::get_top call failed.";
}
if (top_window.IsEqualObject(parent_window)) {
// Outcome 2: Focus is on a frame one level deep, making the
// parent the top-level browsing context. Set focused frame
// pointer to NULL.
this->focused_frame_window_ = NULL;
} else {
// Outcome 3: Focus is on a frame more than one level deep.
// Set focused frame pointer to parent frame.
this->focused_frame_window_ = parent_window;
}
}
}
Simplifying object handling code in IE driver. COM objects are now correctly referenced everywhere via smart pointers, and use of some aliased types is now replaced by the C++ primitives.
2013-03-25 13:34:23 -04:00
int DocumentHost::SetFocusedFrameByIdentifier(VARIANT frame_identifier) {
LOG(TRACE) << "Entering DocumentHost::SetFocusedFrameByIdentifier";
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
CComPtr<IHTMLDocument2> doc;
this->GetDocument(&doc);
Removing use of CComQIPtr from IE driver native code. The use of CComQIPtr is not needed, as we are substituting a call to the QueryInterface method instead. This is what CComQIPtr does under the covers anyway.
2013-03-07 17:06:31 -05:00
CComPtr<IHTMLFramesCollection2> frames;
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
HRESULT hr = doc->get_frames(&frames);
Removing use of CComQIPtr from IE driver native code. The use of CComQIPtr is not needed, as we are substituting a call to the QueryInterface method instead. This is what CComQIPtr does under the covers anyway.
2013-03-07 17:06:31 -05:00
if (!frames) {
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
LOG(WARN) << "No frames in document are set, IHTMLDocument2::get_frames returned NULL";
return ENOSUCHFRAME;
}
long length = 0;
frames->get_length(&length);
if (!length) {
LOG(WARN) << "No frames in document are found IHTMLFramesCollection2::get_length returned 0";
return ENOSUCHFRAME;
}
// Find the frame
CComVariant frame_holder;
Simplifying object handling code in IE driver. COM objects are now correctly referenced everywhere via smart pointers, and use of some aliased types is now replaced by the C++ primitives.
2013-03-25 13:34:23 -04:00
hr = frames->item(&frame_identifier, &frame_holder);
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
if (FAILED(hr)) {
LOGHR(WARN, hr) << "Error retrieving frame holder, call to IHTMLFramesCollection2::item failed";
return ENOSUCHFRAME;
}
Removing use of CComQIPtr from IE driver native code. The use of CComQIPtr is not needed, as we are substituting a call to the QueryInterface method instead. This is what CComQIPtr does under the covers anyway.
2013-03-07 17:06:31 -05:00
CComPtr<IHTMLWindow2> interim_result;
frame_holder.pdispVal->QueryInterface<IHTMLWindow2>(&interim_result);
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
if (!interim_result) {
LOG(WARN) << "Error retrieving frame, IDispatch cannot be cast to IHTMLWindow2";
return ENOSUCHFRAME;
}
this->focused_frame_window_ = interim_result;
Updating IEDriver native code to use the new WD_SUCCESS #define instead of just SUCCESS
2013-01-16 14:37:30 -05:00
return WD_SUCCESS;
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
}
void DocumentHost::PostQuitMessage() {
LOG(TRACE) << "Entering DocumentHost::PostQuitMessage";
LPSTR message_payload = new CHAR[this->browser_id_.size() + 1];
strcpy_s(message_payload, this->browser_id_.size() + 1, this->browser_id_.c_str());
::PostMessage(this->executor_handle(),
WD_BROWSER_QUIT,
NULL,
reinterpret_cast<LPARAM>(message_payload));
}
HWND DocumentHost::FindContentWindowHandle(HWND top_level_window_handle) {
LOG(TRACE) << "Entering DocumentHost::FindContentWindowHandle";
ProcessWindowInfo process_window_info;
process_window_info.pBrowser = NULL;
process_window_info.hwndBrowser = NULL;
DWORD process_id;
::GetWindowThreadProcessId(top_level_window_handle, &process_id);
process_window_info.dwProcessId = process_id;
::EnumChildWindows(top_level_window_handle,
&BrowserFactory::FindChildWindowForProcess,
reinterpret_cast<LPARAM>(&process_window_info));
return process_window_info.hwndBrowser;
}
int DocumentHost::GetDocumentMode(IHTMLDocument2* doc) {
LOG(TRACE) << "Entering DocumentHost::GetDocumentMode";
Removing use of CComQIPtr from IE driver native code. The use of CComQIPtr is not needed, as we are substituting a call to the QueryInterface method instead. This is what CComQIPtr does under the covers anyway.
2013-03-07 17:06:31 -05:00
CComPtr<IHTMLDocument6> mode_doc;
doc->QueryInterface<IHTMLDocument6>(&mode_doc);
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
if (!mode_doc) {
LOG(DEBUG) << "QueryInterface for IHTMLDocument6 fails, so document mode must be 7 or less.";
return 5;
}
CComVariant mode;
HRESULT hr = mode_doc->get_documentMode(&mode);
if (FAILED(hr)) {
LOGHR(WARN, hr) << "get_documentMode failed.";
return 5;
}
Corrected logic in getting location of element after scrolling into view. For IE 6 and 7, an additional 2-pixel offset must be added for every frame on the page when calculating element location. This commit also temporarily reverts the version number resource bump.
2013-07-03 19:18:46 -04:00
int document_mode = static_cast<int>(mode.fltVal);
return document_mode;
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
}
bool DocumentHost::IsStandardsMode(IHTMLDocument2* doc) {
LOG(TRACE) << "Entering DocumentHost::IsStandardsMode";
Removing use of CComQIPtr from IE driver native code. The use of CComQIPtr is not needed, as we are substituting a call to the QueryInterface method instead. This is what CComQIPtr does under the covers anyway.
2013-03-07 17:06:31 -05:00
CComPtr<IHTMLDocument5> compatibility_mode_doc;
doc->QueryInterface<IHTMLDocument5>(&compatibility_mode_doc);
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
if (!compatibility_mode_doc) {
Remove wstring address usages in log messages. It fixes messages in log like: ... ElementFinder.cpp(49) 55097F38xpath = //*[@id='aaa']
2013-04-29 23:03:15 +04:00
LOG(WARN) << "Unable to cast document to IHTMLDocument5. IE6 or greater is required.";
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
return false;
}
CComBSTR compatibility_mode;
HRESULT hr = compatibility_mode_doc->get_compatMode(&compatibility_mode);
if (FAILED(hr)) {
Remove wstring address usages in log messages. It fixes messages in log like: ... ElementFinder.cpp(49) 55097F38xpath = //*[@id='aaa']
2013-04-29 23:03:15 +04:00
LOGHR(WARN, hr) << "Failed calling get_compatMode.";
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
return false;
}
// Compatibility mode should be "BackCompat" for quirks mode, and
// "CSS1Compat" for standards mode. Check for "BackCompat" because
// that's less likely to change.
return compatibility_mode != L"BackCompat";
}
bool DocumentHost::GetDocumentDimensions(IHTMLDocument2* doc, LocationInfo* info) {
LOG(TRACE) << "Entering DocumentHost::GetDocumentDimensions";
CComVariant document_height;
CComVariant document_width;
// In non-standards-compliant mode, the BODY element represents the canvas.
// In standards-compliant mode, the HTML element represents the canvas.
CComPtr<IHTMLElement> canvas_element;
if (!IsStandardsMode(doc)) {
doc->get_body(&canvas_element);
if (!canvas_element) {
LOG(WARN) << "Unable to get canvas element from document in compatibility mode";
return false;
}
} else {
Removing use of CComQIPtr from IE driver native code. The use of CComQIPtr is not needed, as we are substituting a call to the QueryInterface method instead. This is what CComQIPtr does under the covers anyway.
2013-03-07 17:06:31 -05:00
CComPtr<IHTMLDocument3> document_element_doc;
doc->QueryInterface<IHTMLDocument3>(&document_element_doc);
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
if (!document_element_doc) {
Remove wstring address usages in log messages. It fixes messages in log like: ... ElementFinder.cpp(49) 55097F38xpath = //*[@id='aaa']
2013-04-29 23:03:15 +04:00
LOG(WARN) << "Unable to get IHTMLDocument3 handle from document.";
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
return false;
}
// The root node should be the HTML element.
document_element_doc->get_documentElement(&canvas_element);
if (!canvas_element) {
Remove wstring address usages in log messages. It fixes messages in log like: ... ElementFinder.cpp(49) 55097F38xpath = //*[@id='aaa']
2013-04-29 23:03:15 +04:00
LOG(WARN) << "Could not retrieve document element.";
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
return false;
}
Removing use of CComQIPtr from IE driver native code. The use of CComQIPtr is not needed, as we are substituting a call to the QueryInterface method instead. This is what CComQIPtr does under the covers anyway.
2013-03-07 17:06:31 -05:00
CComPtr<IHTMLHtmlElement> html_element;
canvas_element->QueryInterface<IHTMLHtmlElement>(&html_element);
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
if (!html_element) {
Remove wstring address usages in log messages. It fixes messages in log like: ... ElementFinder.cpp(49) 55097F38xpath = //*[@id='aaa']
2013-04-29 23:03:15 +04:00
LOG(WARN) << "Document element is not the HTML element.";
KristianRosenvold: Removed windows carriage returns
2013-01-11 22:18:32 +01:00
return false;
}
}
canvas_element->getAttribute(CComBSTR("scrollHeight"), 0, &document_height);
canvas_element->getAttribute(CComBSTR("scrollWidth"), 0, &document_width);
info->height = document_height.lVal;
info->width = document_width.lVal;
return true;
}
Improving recognition of Protected Mode boundary crossing in IE In particular, when navigating, properly detect URLs beginning with "about:blank", not just that being the entire URL.
2019-02-22 08:13:23 -08:00
bool DocumentHost::IsCrossZoneUrl(std::string url) {
LOG(TRACE) << "Entering Browser::IsCrossZoneUrl";
std::wstring target_url = StringUtilities::ToWString(url);
CComPtr<IUri> parsed_url;
HRESULT hr = ::CreateUri(target_url.c_str(),
Uri_CREATE_IE_SETTINGS,
0,
&parsed_url);
if (FAILED(hr)) {
// If we can't parse the URL, assume that it's invalid, and
// therefore won't cross a Protected Mode boundary.
return false;
}
bool is_protected_mode_browser = this->IsProtectedMode();
bool is_protected_mode_url = is_protected_mode_browser;
if (url.find("about:blank") != 0) {
// If the URL starts with "about:blank", it won't cross the Protected
// Mode boundary, so skip checking if it's a Protected Mode URL.
is_protected_mode_url = ::IEIsProtectedModeURL(target_url.c_str()) == S_OK;
}
bool is_cross_zone = is_protected_mode_browser != is_protected_mode_url;
if (is_cross_zone) {
LOG(DEBUG) << "Navigation across Protected Mode zone detected. URL: "
<< url
<< ", is URL Protected Mode: "
<< (is_protected_mode_url ? "true" : "false")
<< ", is IE in Protected Mode: "
<< (is_protected_mode_browser ? "true" : "false");
}
return is_cross_zone;
}
bool DocumentHost::IsProtectedMode() {
LOG(TRACE) << "Entering DocumentHost::IsProtectedMode";
HWND window_handle = this->GetBrowserWindowHandle();
HookSettings hook_settings;
hook_settings.hook_procedure_name = "ProtectedModeWndProc";
hook_settings.hook_procedure_type = WH_CALLWNDPROC;
hook_settings.window_handle = window_handle;
hook_settings.communication_type = OneWay;
HookProcessor hook;
if (!hook.CanSetWindowsHook(window_handle)) {
LOG(WARN) << "Cannot check Protected Mode because driver and browser are "
<< "not the same bit-ness.";
return false;
}
hook.Initialize(hook_settings);
HookProcessor::ResetFlag();
::SendMessage(window_handle, WD_IS_BROWSER_PROTECTED_MODE, NULL, NULL);
bool is_protected_mode = HookProcessor::GetFlagValue();
return is_protected_mode;
}
Slightly modified setting IE window to have focus when sending input This code only affects the browser when the requireWindowFocus capability is set at session start.
2019-07-09 14:13:42 -07:00
bool DocumentHost::SetFocusToBrowser() {
LOG(TRACE) << "Entering DocumentHost::SetFocusToBrowser";
HWND top_level_window_handle = this->GetTopLevelWindowHandle();
HWND foreground_window = ::GetAncestor(::GetForegroundWindow(), GA_ROOT);
if (foreground_window != top_level_window_handle) {
LOG(TRACE) << "Top-level IE window is " << top_level_window_handle
<< " foreground window is " << foreground_window;
CComPtr<IUIAutomation> ui_automation;
HRESULT hr = ::CoCreateInstance(CLSID_CUIAutomation,
NULL,
CLSCTX_INPROC_SERVER,
IID_IUIAutomation,
reinterpret_cast<void**>(&ui_automation));
if (SUCCEEDED(hr)) {
LOG(TRACE) << "Using UI Automation to set window focus";
CComPtr<IUIAutomationElement> parent_window;
hr = ui_automation->ElementFromHandle(top_level_window_handle,
&parent_window);
if (SUCCEEDED(hr)) {
CComPtr<IUIAutomationWindowPattern> window_pattern;
hr = parent_window->GetCurrentPatternAs(UIA_WindowPatternId,
IID_PPV_ARGS(&window_pattern));
[iedriver] Prevent null pointer access in UI Automation for setting browser focus
2021-11-03 12:08:25 -04:00
if (SUCCEEDED(hr) && window_pattern != nullptr) {
Slightly modified setting IE window to have focus when sending input This code only affects the browser when the requireWindowFocus capability is set at session start.
2019-07-09 14:13:42 -07:00
BOOL is_topmost;
hr = window_pattern->get_CurrentIsTopmost(&is_topmost);
WindowVisualState visual_state;
hr = window_pattern->get_CurrentWindowVisualState(&visual_state);
if (visual_state == WindowVisualState::WindowVisualState_Maximized ||
visual_state == WindowVisualState::WindowVisualState_Normal) {
parent_window->SetFocus();
window_pattern->SetWindowVisualState(visual_state);
}
}
}
}
}
foreground_window = ::GetAncestor(::GetForegroundWindow(), GA_ROOT);
if (foreground_window != top_level_window_handle) {
HWND content_window_handle = this->GetContentWindowHandle();
LOG(TRACE) << "Top-level IE window is " << top_level_window_handle
<< " foreground window is " << foreground_window;
LOG(TRACE) << "Window still not in foreground; "
<< "attempting to use SetForegroundWindow API";
UINT_PTR lock_timeout = 0;
DWORD process_id = 0;
DWORD thread_id = ::GetWindowThreadProcessId(top_level_window_handle,
&process_id);
DWORD current_thread_id = ::GetCurrentThreadId();
DWORD current_process_id = ::GetCurrentProcessId();
if (current_thread_id != thread_id) {
::AttachThreadInput(current_thread_id, thread_id, TRUE);
::SystemParametersInfo(SPI_GETFOREGROUNDLOCKTIMEOUT,
0,
&lock_timeout,
0);
::SystemParametersInfo(SPI_SETFOREGROUNDLOCKTIMEOUT,
0,
0,
SPIF_SENDWININICHANGE | SPIF_UPDATEINIFILE);
HookSettings hook_settings;
hook_settings.hook_procedure_name = "AllowSetForegroundProc";
hook_settings.hook_procedure_type = WH_CALLWNDPROC;
hook_settings.window_handle = content_window_handle;
hook_settings.communication_type = OneWay;
HookProcessor hook;
if (!hook.CanSetWindowsHook(content_window_handle)) {
LOG(WARN) << "Setting window focus may fail because driver and browser "
<< "are not the same bit-ness.";
return false;
}
hook.Initialize(hook_settings);
::SendMessage(content_window_handle,
WD_ALLOW_SET_FOREGROUND,
NULL,
NULL);
hook.Dispose();
}
::SetForegroundWindow(top_level_window_handle);
::Sleep(100);
if (current_thread_id != thread_id) {
::SystemParametersInfo(SPI_SETFOREGROUNDLOCKTIMEOUT,
0,
reinterpret_cast<void*>(lock_timeout),
SPIF_SENDWININICHANGE | SPIF_UPDATEINIFILE);
::AttachThreadInput(current_thread_id, thread_id, FALSE);
}
}
foreground_window = ::GetAncestor(::GetForegroundWindow(), GA_ROOT);
return foreground_window == top_level_window_handle;
}
Updated IE driver to better support W3C WebDriver Specification This is a large update that includes implementation of the following things that are part of the WebDriver spec: * Implement the "get element property" command * Implement the "get named cookie" command * Implement the "minimize window" command * Implement the "fullscreen window" command * Update "set window rect" command to correctly recognize and restore from minimized or full screen state * Fix the "release actions" command to properly roll back all pending actions * Update the "perform actions" command to be more robust for pointer actions * Fix alert handling to handle dialogs with the "Do not show any more dialogs" check box * Added handling for the "dismiss and notify" and "accept and notify" user prompt handler states
2017-08-31 09:41:44 -07:00
} // namespace webdriver
Improving recognition of Protected Mode boundary crossing in IE In particular, when navigating, properly detect URLs beginning with "about:blank", not just that being the entire URL.
2019-02-22 08:13:23 -08:00
#ifdef __cplusplus
extern "C" {
#endif
LRESULT CALLBACK ProtectedModeWndProc(int nCode, WPARAM wParam, LPARAM lParam) {
CWPSTRUCT* call_window_proc_struct = reinterpret_cast<CWPSTRUCT*>(lParam);
if (WD_IS_BROWSER_PROTECTED_MODE == call_window_proc_struct->message) {
BOOL is_protected_mode = FALSE;
HRESULT hr = ::IEIsProtectedModeProcess(&is_protected_mode);
webdriver::HookProcessor::SetFlagValue(is_protected_mode == TRUE);
}
return ::CallNextHookEx(NULL, nCode, wParam, lParam);
}
Slightly modified setting IE window to have focus when sending input This code only affects the browser when the requireWindowFocus capability is set at session start.
2019-07-09 14:13:42 -07:00
LRESULT CALLBACK AllowSetForegroundProc(int nCode, WPARAM wParam, LPARAM lParam) {
if ((nCode == HC_ACTION) && (wParam == PM_REMOVE)) {
MSG* msg = reinterpret_cast<MSG*>(lParam);
if (msg->message == WD_ALLOW_SET_FOREGROUND) {
::AllowSetForegroundWindow(ASFW_ANY);
}
}
return CallNextHookEx(NULL, nCode, wParam, lParam);
}
Improving recognition of Protected Mode boundary crossing in IE In particular, when navigating, properly detect URLs beginning with "about:blank", not just that being the entire URL.
2019-02-22 08:13:23 -08:00
#ifdef __cplusplus
}
#endif
Reference in New Issue Copy Permalink