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