OLD | NEW |
| (Empty) |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome_frame/com_message_event.h" | |
6 | |
7 #include "base/logging.h" | |
8 #include "base/strings/string_util.h" | |
9 #include "base/strings/stringprintf.h" | |
10 #include "base/strings/utf_string_conversions.h" | |
11 | |
12 ComMessageEvent::ComMessageEvent() { | |
13 } | |
14 | |
15 ComMessageEvent::~ComMessageEvent() { | |
16 } | |
17 | |
18 bool ComMessageEvent::Initialize(IOleContainer* container, | |
19 const std::string& message, | |
20 const std::string& origin, | |
21 const std::string& event_type) { | |
22 DLOG_IF(WARNING, !container) << __FUNCTION__ << " no container"; | |
23 message_ = message; | |
24 origin_ = origin; | |
25 type_ = event_type; | |
26 | |
27 // May remain NULL if container not IE | |
28 base::win::ScopedComPtr<IHTMLEventObj> basic_event; | |
29 base::win::ScopedComPtr<IHTMLDocument2> doc; | |
30 | |
31 // Fetching doc may fail in non-IE containers | |
32 // and container might be NULL in some applications. | |
33 if (container) | |
34 container->QueryInterface(doc.Receive()); | |
35 if (doc) { | |
36 base::win::ScopedComPtr<IHTMLDocument4> doc4; | |
37 doc4.QueryFrom(doc); | |
38 DCHECK(doc4); // supported by IE5.5 and higher | |
39 if (doc4) { | |
40 // IHTMLEventObj5 is only supported in IE8 and later, so we provide our | |
41 // own (minimalistic) implementation of it. | |
42 doc4->createEventObject(NULL, basic_event.Receive()); | |
43 DCHECK(basic_event); | |
44 } | |
45 } | |
46 | |
47 basic_event_ = basic_event; | |
48 return true; | |
49 } | |
50 | |
51 STDMETHODIMP ComMessageEvent::GetTypeInfoCount(UINT* info) { | |
52 // Don't DCHECK as python scripts might still call this function | |
53 // inadvertently. | |
54 DLOG(WARNING) << "Not implemented: " << __FUNCTION__; | |
55 return E_NOTIMPL; | |
56 } | |
57 | |
58 STDMETHODIMP ComMessageEvent::GetTypeInfo(UINT which_info, LCID lcid, | |
59 ITypeInfo** type_info) { | |
60 DLOG(WARNING) << "Not implemented: " << __FUNCTION__; | |
61 return E_NOTIMPL; | |
62 } | |
63 | |
64 STDMETHODIMP ComMessageEvent::GetIDsOfNames(REFIID iid, LPOLESTR* names, | |
65 UINT count_names, LCID lcid, | |
66 DISPID* dispids) { | |
67 HRESULT hr = S_OK; | |
68 | |
69 // Note that since we're using LowerCaseEqualsASCII for string comparison, | |
70 // the second argument _must_ be all lower case. I.e. we cannot compare | |
71 // against L"messagePort" since it has a capital 'P'. | |
72 for (UINT i = 0; SUCCEEDED(hr) && i < count_names; ++i) { | |
73 const wchar_t* name_begin = names[i]; | |
74 const wchar_t* name_end = name_begin + wcslen(name_begin); | |
75 if (LowerCaseEqualsASCII(name_begin, name_end, "data")) { | |
76 dispids[i] = DISPID_MESSAGE_EVENT_DATA; | |
77 } else if (LowerCaseEqualsASCII(name_begin, name_end, "origin")) { | |
78 dispids[i] = DISPID_MESSAGE_EVENT_ORIGIN; | |
79 } else if (LowerCaseEqualsASCII(name_begin, name_end, "lasteventid")) { | |
80 dispids[i] = DISPID_MESSAGE_EVENT_LAST_EVENT_ID; | |
81 } else if (LowerCaseEqualsASCII(name_begin, name_end, "source")) { | |
82 dispids[i] = DISPID_MESSAGE_EVENT_SOURCE; | |
83 } else if (LowerCaseEqualsASCII(name_begin, name_end, "messageport")) { | |
84 dispids[i] = DISPID_MESSAGE_EVENT_MESSAGE_PORT; | |
85 } else if (LowerCaseEqualsASCII(name_begin, name_end, "type")) { | |
86 dispids[i] = DISPID_MESSAGE_EVENT_TYPE; | |
87 } else { | |
88 if (basic_event_) { | |
89 hr = basic_event_->GetIDsOfNames(IID_IDispatch, &names[i], 1, lcid, | |
90 &dispids[i]); | |
91 } else { | |
92 hr = DISP_E_MEMBERNOTFOUND; | |
93 } | |
94 | |
95 if (FAILED(hr)) { | |
96 DLOG(WARNING) << "member not found: " << names[i] | |
97 << base::StringPrintf(L"0x%08X", hr); | |
98 } | |
99 } | |
100 } | |
101 return hr; | |
102 } | |
103 | |
104 STDMETHODIMP ComMessageEvent::Invoke(DISPID dispid, REFIID iid, LCID lcid, | |
105 WORD flags, DISPPARAMS* params, | |
106 VARIANT* result, EXCEPINFO* excepinfo, | |
107 UINT* arg_err) { | |
108 HRESULT hr = DISP_E_MEMBERNOTFOUND; | |
109 switch (dispid) { | |
110 case DISPID_MESSAGE_EVENT_DATA: | |
111 hr = GetStringProperty(flags, base::UTF8ToWide(message_).c_str(), result); | |
112 break; | |
113 | |
114 case DISPID_MESSAGE_EVENT_ORIGIN: | |
115 hr = GetStringProperty(flags, base::UTF8ToWide(origin_).c_str(), result); | |
116 break; | |
117 | |
118 case DISPID_MESSAGE_EVENT_TYPE: | |
119 hr = GetStringProperty(flags, base::UTF8ToWide(type_).c_str(), result); | |
120 break; | |
121 | |
122 case DISPID_MESSAGE_EVENT_LAST_EVENT_ID: | |
123 hr = GetStringProperty(flags, L"", result); | |
124 break; | |
125 | |
126 case DISPID_MESSAGE_EVENT_SOURCE: | |
127 case DISPID_MESSAGE_EVENT_MESSAGE_PORT: | |
128 if (flags & DISPATCH_PROPERTYGET) { | |
129 result->vt = VT_NULL; | |
130 hr = S_OK; | |
131 } else { | |
132 hr = DISP_E_TYPEMISMATCH; | |
133 } | |
134 break; | |
135 | |
136 default: | |
137 if (basic_event_) { | |
138 hr = basic_event_->Invoke(dispid, iid, lcid, flags, params, result, | |
139 excepinfo, arg_err); | |
140 } | |
141 break; | |
142 } | |
143 | |
144 return hr; | |
145 } | |
146 | |
147 HRESULT ComMessageEvent::GetStringProperty(WORD flags, const wchar_t* value, | |
148 VARIANT* result) { | |
149 if (!result) | |
150 return E_INVALIDARG; | |
151 | |
152 HRESULT hr; | |
153 if (flags & DISPATCH_PROPERTYGET) { | |
154 result->vt = VT_BSTR; | |
155 result->bstrVal = ::SysAllocString(value); | |
156 hr = S_OK; | |
157 } else { | |
158 hr = DISP_E_TYPEMISMATCH; | |
159 } | |
160 return hr; | |
161 } | |
OLD | NEW |