dll/win32/mshtml/ifacewrap.c

   1 /*
   2  * Copyright 2011 Jacek Caban for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <config.h>
  20
  21 #include <stdarg.h>
  22
  23 #define WIN32_NO_STATUS
  24 #define _INC_WINDOWS
  25
  26 #define COBJMACROS
  27
  28 #include <windef.h>
  29 #include <winbase.h>
  30 //#include "winuser.h"
  31 #include <ole2.h>
  32
  33 #include <wine/debug.h>
  34
  35 #include "mshtml_private.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  38
  39 /*
  40  * This object wraps any unrecognized interface overriding its IUnknown methods, allowing
  41  * us to return external interface from our QI implementation preserving COM rules.
  42  * This can't be done right and it seems to be broken by design.
  43  */
  44 typedef struct {
  45     IUnknown IUnknown_iface;
  46     IUnknown *iface;
  47     IUnknown *ref_unk;
  48     LONG ref;
  49 } iface_wrapper_t;
  50
  51 static inline iface_wrapper_t *impl_from_IUnknown(IUnknown *iface)
  52 {
  53     return CONTAINING_RECORD(iface, iface_wrapper_t, IUnknown_iface);
  54 }
  55
  56 static HRESULT WINAPI wrapper_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
  57 {
  58     iface_wrapper_t *This = impl_from_IUnknown(iface);
  59
  60     TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
  61
  62     return IUnknown_QueryInterface(This->ref_unk, riid, ppv);
  63 }
  64
  65 static HRESULT WINAPI wrapper_AddRef(IUnknown *iface)
  66 {
  67     iface_wrapper_t *This = impl_from_IUnknown(iface);
  68     LONG ref = InterlockedIncrement(&This->ref);
  69
  70     TRACE("(%p) ref=%d\n", This, ref);
  71
  72     return ref;
  73 }
  74
  75 static HRESULT WINAPI wrapper_Release(IUnknown *iface)
  76 {
  77     iface_wrapper_t *This = impl_from_IUnknown(iface);
  78     LONG ref = InterlockedDecrement(&This->ref);
  79
  80     TRACE("(%p) ref=%d\n", This, ref);
  81
  82     if(!ref) {
  83         IUnknown_Release(This->iface);
  84         IUnknown_Release(This->ref_unk);
  85         heap_free(This);
  86     }
  87
  88     return ref;
  89 }
  90
  91 #ifdef __i386__
  92
  93 #ifdef _MSC_VER
  94 #define DEFINE_WRAPPER_FUNC(n, off, x) HRESULT wrapper_func_##n(IUnknown*);
  95 #else
  96 #define DEFINE_WRAPPER_FUNC(n, off, x)          \
  97     HRESULT wrapper_func_##n(IUnknown*);        \
  98     __ASM_GLOBAL_FUNC(wrapper_func_##n,         \
  99         "movl 4(%esp), %eax\n\t"                \
 100         "movl 4(%eax), %eax\n\t"                \
 101         "movl %eax, 4(%esp)\n\t"                \
 102         "movl 0(%eax), %eax\n\t"                \
 103         "jmp *" #off "(%eax)\n\t")
 104 #endif
 105
 106 #elif defined(__x86_64__)
 107
 108 #define DEFINE_WRAPPER_FUNC(n, x, off)          \
 109     HRESULT WINAPI wrapper_func_##n(IUnknown*); \
 110     __ASM_GLOBAL_FUNC(wrapper_func_##n,         \
 111         "movq 8(%rcx), %rcx\n\t"                \
 112         "movq 0(%rcx), %rax\n\t"                \
 113         "jmp *" #off "(%rax)\n\t")
 114
 115 #else
 116
 117 #define DEFINE_WRAPPER_FUNC(n, x, off)                           \
 118     static HRESULT WINAPI wrapper_func_##n(IUnknown *iface) {    \
 119         ERR("Not implemented for this architecture\n");          \
 120         return E_NOTIMPL;                                        \
 121     }
 122
 123 #endif
 124
 125 /* DEFINE_WRAPPER_FUNC takes 3 arguments: index in vtbl, 32-bit offset in vtbl and 64-bit offset in vtbl */
 126 DEFINE_WRAPPER_FUNC(3, 12, 24)
 127 DEFINE_WRAPPER_FUNC(4, 16, 32)
 128 DEFINE_WRAPPER_FUNC(5, 20, 40)
 129 DEFINE_WRAPPER_FUNC(6, 24, 48)
 130 DEFINE_WRAPPER_FUNC(7, 28, 56)
 131 DEFINE_WRAPPER_FUNC(8, 32, 64)
 132 DEFINE_WRAPPER_FUNC(9, 36, 72)
 133 DEFINE_WRAPPER_FUNC(10, 40, 80)
 134 DEFINE_WRAPPER_FUNC(11, 44, 88)
 135 DEFINE_WRAPPER_FUNC(12, 48, 96)
 136 DEFINE_WRAPPER_FUNC(13, 52, 104)
 137 DEFINE_WRAPPER_FUNC(14, 56, 112)
 138 DEFINE_WRAPPER_FUNC(15, 60, 120)
 139 DEFINE_WRAPPER_FUNC(16, 64, 128)
 140 DEFINE_WRAPPER_FUNC(17, 68, 136)
 141 DEFINE_WRAPPER_FUNC(18, 72, 144)
 142 DEFINE_WRAPPER_FUNC(19, 76, 152)
 143 DEFINE_WRAPPER_FUNC(20, 80, 160)
 144 DEFINE_WRAPPER_FUNC(21, 84, 168)
 145 DEFINE_WRAPPER_FUNC(22, 88, 176)
 146 DEFINE_WRAPPER_FUNC(23, 92, 184)
 147 DEFINE_WRAPPER_FUNC(24, 96, 192)
 148 DEFINE_WRAPPER_FUNC(25, 100, 200)
 149 DEFINE_WRAPPER_FUNC(26, 104, 208)
 150 DEFINE_WRAPPER_FUNC(27, 108, 216)
 151 DEFINE_WRAPPER_FUNC(28, 112, 224)
 152 DEFINE_WRAPPER_FUNC(29, 116, 232)
 153 DEFINE_WRAPPER_FUNC(30, 120, 240)
 154 DEFINE_WRAPPER_FUNC(31, 124, 248)
 155 DEFINE_WRAPPER_FUNC(32, 128, 256)
 156 DEFINE_WRAPPER_FUNC(33, 132, 264)
 157 DEFINE_WRAPPER_FUNC(34, 136, 272)
 158 DEFINE_WRAPPER_FUNC(35, 140, 280)
 159 DEFINE_WRAPPER_FUNC(36, 144, 288)
 160 DEFINE_WRAPPER_FUNC(37, 148, 296)
 161 DEFINE_WRAPPER_FUNC(38, 152, 304)
 162 DEFINE_WRAPPER_FUNC(39, 156, 312)
 163 DEFINE_WRAPPER_FUNC(40, 160, 320)
 164 DEFINE_WRAPPER_FUNC(41, 164, 328)
 165 DEFINE_WRAPPER_FUNC(42, 168, 336)
 166 DEFINE_WRAPPER_FUNC(43, 172, 344)
 167 DEFINE_WRAPPER_FUNC(44, 176, 352)
 168 DEFINE_WRAPPER_FUNC(45, 180, 360)
 169 DEFINE_WRAPPER_FUNC(46, 184, 368)
 170 DEFINE_WRAPPER_FUNC(47, 188, 376)
 171 DEFINE_WRAPPER_FUNC(48, 192, 384)
 172 DEFINE_WRAPPER_FUNC(49, 196, 392)
 173 DEFINE_WRAPPER_FUNC(50, 200, 400)
 174 DEFINE_WRAPPER_FUNC(51, 204, 408)
 175 DEFINE_WRAPPER_FUNC(52, 208, 416)
 176 DEFINE_WRAPPER_FUNC(53, 212, 424)
 177 DEFINE_WRAPPER_FUNC(54, 216, 432)
 178 DEFINE_WRAPPER_FUNC(55, 220, 440)
 179 DEFINE_WRAPPER_FUNC(56, 224, 448)
 180 DEFINE_WRAPPER_FUNC(57, 228, 456)
 181 DEFINE_WRAPPER_FUNC(58, 232, 464)
 182 DEFINE_WRAPPER_FUNC(59, 236, 472)
 183 DEFINE_WRAPPER_FUNC(60, 240, 480)
 184 DEFINE_WRAPPER_FUNC(61, 244, 488)
 185 DEFINE_WRAPPER_FUNC(62, 248, 496)
 186 DEFINE_WRAPPER_FUNC(63, 252, 504)
 187 DEFINE_WRAPPER_FUNC(64, 256, 512)
 188 DEFINE_WRAPPER_FUNC(65, 260, 520)
 189 DEFINE_WRAPPER_FUNC(66, 264, 528)
 190 DEFINE_WRAPPER_FUNC(67, 268, 536)
 191 DEFINE_WRAPPER_FUNC(68, 272, 544)
 192 DEFINE_WRAPPER_FUNC(69, 276, 552)
 193 DEFINE_WRAPPER_FUNC(70, 280, 560)
 194 DEFINE_WRAPPER_FUNC(71, 284, 568)
 195 DEFINE_WRAPPER_FUNC(72, 288, 576)
 196 DEFINE_WRAPPER_FUNC(73, 292, 584)
 197 DEFINE_WRAPPER_FUNC(74, 296, 592)
 198 DEFINE_WRAPPER_FUNC(75, 300, 600)
 199 DEFINE_WRAPPER_FUNC(76, 304, 608)
 200 DEFINE_WRAPPER_FUNC(77, 308, 616)
 201 DEFINE_WRAPPER_FUNC(78, 312, 624)
 202 DEFINE_WRAPPER_FUNC(79, 316, 632)
 203 DEFINE_WRAPPER_FUNC(80, 320, 640)
 204 DEFINE_WRAPPER_FUNC(81, 324, 648)
 205 DEFINE_WRAPPER_FUNC(82, 328, 656)
 206 DEFINE_WRAPPER_FUNC(83, 332, 664)
 207 DEFINE_WRAPPER_FUNC(84, 336, 672)
 208 DEFINE_WRAPPER_FUNC(85, 340, 680)
 209 DEFINE_WRAPPER_FUNC(86, 344, 688)
 210 DEFINE_WRAPPER_FUNC(87, 348, 696)
 211 DEFINE_WRAPPER_FUNC(88, 352, 704)
 212 DEFINE_WRAPPER_FUNC(89, 356, 712)
 213 DEFINE_WRAPPER_FUNC(90, 360, 720)
 214 DEFINE_WRAPPER_FUNC(91, 364, 728)
 215 DEFINE_WRAPPER_FUNC(92, 368, 736)
 216 DEFINE_WRAPPER_FUNC(93, 372, 744)
 217 DEFINE_WRAPPER_FUNC(94, 376, 752)
 218 DEFINE_WRAPPER_FUNC(95, 380, 760)
 219 DEFINE_WRAPPER_FUNC(96, 384, 768)
 220 DEFINE_WRAPPER_FUNC(97, 388, 776)
 221 DEFINE_WRAPPER_FUNC(98, 392, 784)
 222 DEFINE_WRAPPER_FUNC(99, 396, 792)
 223
 224 /* The size was found by testing when calls start crashing. It looks like MS wraps up to 100 functions. */
 225 static const void *wrapper_vtbl[] = {
 226     wrapper_QueryInterface,
 227     wrapper_AddRef,
 228     wrapper_Release,
 229     wrapper_func_3,
 230     wrapper_func_4,
 231     wrapper_func_5,
 232     wrapper_func_6,
 233     wrapper_func_7,
 234     wrapper_func_8,
 235     wrapper_func_9,
 236     wrapper_func_10,
 237     wrapper_func_11,
 238     wrapper_func_12,
 239     wrapper_func_13,
 240     wrapper_func_14,
 241     wrapper_func_15,
 242     wrapper_func_16,
 243     wrapper_func_17,
 244     wrapper_func_18,
 245     wrapper_func_19,
 246     wrapper_func_20,
 247     wrapper_func_21,
 248     wrapper_func_22,
 249     wrapper_func_23,
 250     wrapper_func_24,
 251     wrapper_func_25,
 252     wrapper_func_26,
 253     wrapper_func_27,
 254     wrapper_func_28,
 255     wrapper_func_29,
 256     wrapper_func_30,
 257     wrapper_func_31,
 258     wrapper_func_32,
 259     wrapper_func_33,
 260     wrapper_func_34,
 261     wrapper_func_35,
 262     wrapper_func_36,
 263     wrapper_func_37,
 264     wrapper_func_38,
 265     wrapper_func_39,
 266     wrapper_func_40,
 267     wrapper_func_41,
 268     wrapper_func_42,
 269     wrapper_func_43,
 270     wrapper_func_44,
 271     wrapper_func_45,
 272     wrapper_func_46,
 273     wrapper_func_47,
 274     wrapper_func_48,
 275     wrapper_func_49,
 276     wrapper_func_50,
 277     wrapper_func_51,
 278     wrapper_func_52,
 279     wrapper_func_53,
 280     wrapper_func_54,
 281     wrapper_func_55,
 282     wrapper_func_56,
 283     wrapper_func_57,
 284     wrapper_func_58,
 285     wrapper_func_59,
 286     wrapper_func_60,
 287     wrapper_func_61,
 288     wrapper_func_62,
 289     wrapper_func_63,
 290     wrapper_func_64,
 291     wrapper_func_65,
 292     wrapper_func_66,
 293     wrapper_func_67,
 294     wrapper_func_68,
 295     wrapper_func_69,
 296     wrapper_func_70,
 297     wrapper_func_71,
 298     wrapper_func_72,
 299     wrapper_func_73,
 300     wrapper_func_74,
 301     wrapper_func_75,
 302     wrapper_func_76,
 303     wrapper_func_77,
 304     wrapper_func_78,
 305     wrapper_func_79,
 306     wrapper_func_80,
 307     wrapper_func_81,
 308     wrapper_func_82,
 309     wrapper_func_83,
 310     wrapper_func_84,
 311     wrapper_func_85,
 312     wrapper_func_86,
 313     wrapper_func_87,
 314     wrapper_func_88,
 315     wrapper_func_89,
 316     wrapper_func_90,
 317     wrapper_func_91,
 318     wrapper_func_92,
 319     wrapper_func_93,
 320     wrapper_func_94,
 321     wrapper_func_95,
 322     wrapper_func_96,
 323     wrapper_func_97,
 324     wrapper_func_98,
 325     wrapper_func_99
 326 };
 327
 328 HRESULT wrap_iface(IUnknown *iface, IUnknown *ref_unk, IUnknown **ret)
 329 {
 330     iface_wrapper_t *wrapper;
 331
 332     wrapper = heap_alloc(sizeof(*wrapper));
 333     if(!wrapper)
 334         return E_OUTOFMEMORY;
 335
 336     wrapper->IUnknown_iface.lpVtbl = (const IUnknownVtbl*)wrapper_vtbl;
 337     wrapper->ref = 1;
 338
 339     IUnknown_AddRef(iface);
 340     wrapper->iface = iface;
 341
 342     IUnknown_AddRef(ref_unk);
 343     wrapper->ref_unk = ref_unk;
 344
 345     *ret = &wrapper->IUnknown_iface;
 346     return S_OK;
 347 }