Checking in the initial version of CEEE (Chrome Extensions Execution

Environment), an optional feature of Chrome Frame that acts as an
adapter layer for a subset of the Chrome Extension APIs. This enables
extensions that stick to the supported subset of APIs to work in the
context of Chrome Frame with minimal or sometimes no changes.

See http://www.chromium.org/developers/design-documents/ceee for an
overview of the design of CEEE.

TEST=unit tests (run ceee/smoke_tests.bat as an administrator on Windows)
BUG=none



git-svn-id: svn://svn.chromium.org/chrome/trunk/src@64712 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 126 insertions, 0 deletions

diff --git a/ceee/ie/plugin/bho/dom_utils.cc b/ceee/ie/plugin/bho/dom_utils.cc
new file mode 100644
index 0000000..714e108
--- /dev/null
+++ b/ceee/ie/plugin/bho/dom_utils.cc
@@ -0,0 +1,126 @@
+// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+//
+// A collection of utility functions that interrogate or mutate the IE DOM.
+#include "ceee/ie/plugin/bho/dom_utils.h"
+
+#include <atlbase.h>
+
+#include "ceee/common/com_utils.h"
+#include "base/logging.h"
+
+HRESULT DomUtils::InjectStyleTag(IHTMLDocument2* document,
+                                 IHTMLDOMNode* head_node,
+                                 const wchar_t* code) {
+  DCHECK(document != NULL);
+  DCHECK(head_node != NULL);
+
+  CComPtr<IHTMLElement> elem;
+  HRESULT hr = document->createElement(CComBSTR(L"style"), &elem);
+  if (FAILED(hr)) {
+    LOG(ERROR) << "Could not create style element " << com::LogHr(hr);
+    return hr;
+  }
+
+  CComQIPtr<IHTMLStyleElement> style_elem(elem);
+  DCHECK(style_elem != NULL) <<
+      "Could not QueryInterface for IHTMLStyleElement";
+
+  hr = style_elem->put_type(CComBSTR(L"text/css"));
+  DCHECK(SUCCEEDED(hr)) << "Could not set type of style element" <<
+      com::LogHr(hr);
+
+  CComPtr<IHTMLStyleSheet> style_sheet;
+  hr = style_elem->get_styleSheet(&style_sheet);
+  DCHECK(SUCCEEDED(hr)) << "Could not get styleSheet of style element." <<
+      com::LogHr(hr);
+
+  hr = style_sheet->put_cssText(CComBSTR(code));
+  if (FAILED(hr)) {
+    LOG(ERROR) << "Could not set cssText of styleSheet." << com::LogHr(hr);
+    return hr;
+  }
+
+  CComQIPtr<IHTMLDOMNode> style_node(style_elem);
+  DCHECK(style_node != NULL) << "Could not query interface for IHTMLDomNode.";
+
+  CComPtr<IHTMLDOMNode> dummy;
+  hr = head_node->appendChild(style_node, &dummy);
+  if (FAILED(hr))
+    LOG(ERROR) << "Could not append style node to head node." << com::LogHr(hr);
+
+  return hr;
+}
+
+HRESULT DomUtils::GetHeadNode(IHTMLDocument* document,
+                              IHTMLDOMNode** head_node) {
+  DCHECK(document != NULL);
+  DCHECK(head_node != NULL && *head_node == NULL);
+
+  // Find the HEAD element through document.getElementsByTagName.
+  CComQIPtr<IHTMLDocument3> document3(document);
+  CComPtr<IHTMLElementCollection> head_elements;
+  DCHECK(document3 != NULL);  // Should be there on IE >= 5
+  if (document3 == NULL) {
+    LOG(ERROR) << L"Unable to retrieve IHTMLDocument3 interface";
+    return E_NOINTERFACE;
+  }
+  HRESULT hr = GetElementsByTagName(document3, CComBSTR(L"head"),
+                                    &head_elements, NULL);
+  if (FAILED(hr)) {
+    LOG(ERROR) << "Could not retrieve head elements collection "
+        << com::LogHr(hr);
+    return hr;
+  }
+
+  return GetElementFromCollection(head_elements, 0, IID_IHTMLDOMNode,
+                                  reinterpret_cast<void**>(head_node));
+}
+
+HRESULT DomUtils::GetElementsByTagName(IHTMLDocument3* document,
+                                       BSTR tag_name,
+                                       IHTMLElementCollection** elements,
+                                       long* length) {
+  DCHECK(document != NULL);
+  DCHECK(tag_name != NULL);
+  DCHECK(elements != NULL && *elements == NULL);
+
+  HRESULT hr = document->getElementsByTagName(tag_name, elements);
+  if (FAILED(hr) || *elements == NULL) {
+    hr = com::AlwaysError(hr);
+    LOG(ERROR) << "Could not retrieve elements collection " << com::LogHr(hr);
+    return hr;
+  }
+
+  if (length != NULL) {
+    hr = (*elements)->get_length(length);
+    if (FAILED(hr)) {
+      (*elements)->Release();
+      *elements = NULL;
+      LOG(ERROR) << "Could not retrieve collection length " << com::LogHr(hr);
+    }
+  }
+  return hr;
+}
+
+HRESULT DomUtils::GetElementFromCollection(IHTMLElementCollection* collection,
+                                           long index,
+                                           REFIID id,
+                                           void** element) {
+  DCHECK(collection != NULL);
+  DCHECK(element != NULL && *element == NULL);
+
+  CComPtr<IDispatch> item;
+  CComVariant index_variant(index, VT_I4);
+  HRESULT hr = collection->item(index_variant, index_variant, &item);
+  // As per http://msdn.microsoft.com/en-us/library/aa703930(VS.85).aspx
+  // item may still be NULL even if S_OK is returned.
+  if (FAILED(hr) || item == NULL) {
+    hr = com::AlwaysError(hr);
+    LOG(ERROR) << "Could not access item " << com::LogHr(hr);
+    return hr;
+  }
+
+  return item->QueryInterface(id, element);
+}