ole2.c 83.8 KB
Newer Older
Alexandre Julliard's avatar
Alexandre Julliard committed
1 2 3
/*
 *	OLE2 library
 *
4 5 6 7
 * Copyright 1995 Martin von Loewis
 * Copyright 1999 Francis Beaudet
 * Copyright 1999 Noel Borthwick
 * Copyright 1999, 2000 Marcus Meissner
8
 * Copyright 2005 Juan Lang
9 10 11 12 13 14 15 16 17 18 19 20 21
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
22
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
Alexandre Julliard's avatar
Alexandre Julliard committed
23 24
 */

25 26
#include "config.h"

27
#include <assert.h>
28
#include <stdlib.h>
29
#include <stdarg.h>
30
#include <stdio.h>
Patrik Stridvall's avatar
Patrik Stridvall committed
31
#include <string.h>
32

33
#define COBJMACROS
34 35
#define NONAMELESSUNION
#define NONAMELESSSTRUCT
36

37
#include "windef.h"
38
#include "winbase.h"
39
#include "winerror.h"
40
#include "wingdi.h"
41
#include "winuser.h"
42
#include "winnls.h"
43
#include "winreg.h"
44 45
#include "ole2.h"
#include "ole2ver.h"
46

47
#include "wine/unicode.h"
48
#include "compobj_private.h"
49
#include "wine/list.h"
Alexandre Julliard's avatar
Alexandre Julliard committed
50

51
#include "wine/debug.h"
52

53 54
WINE_DEFAULT_DEBUG_CHANNEL(ole);
WINE_DECLARE_DEBUG_CHANNEL(accel);
55

56
/******************************************************************************
57
 * These are static/global variables and internal data structures that the
58
 * OLE module uses to maintain it's state.
59
 */
60 61 62 63 64 65
typedef struct tagTrackerWindowInfo
{
  IDataObject* dataObject;
  IDropSource* dropSource;
  DWORD        dwOKEffect;
  DWORD*       pdwEffect;
66
  BOOL       trackingDone;
67 68
  HRESULT      returnValue;

69
  BOOL       escPressed;
70 71
  HWND       curTargetHWND;	/* window the mouse is hovering over */
  HWND       curDragTargetHWND; /* might be a ancestor of curTargetHWND */
72
  IDropTarget* curDragTarget;
73 74
  POINTL     curMousePos;       /* current position of the mouse in screen coordinates */
  DWORD      dwKeyState;        /* current state of the shift and ctrl keys and the mouse buttons */
75
} TrackerWindowInfo;
76

77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
typedef struct tagOleMenuDescriptor  /* OleMenuDescriptor */
{
  HWND               hwndFrame;         /* The containers frame window */
  HWND               hwndActiveObject;  /* The active objects window */
  OLEMENUGROUPWIDTHS mgw;               /* OLE menu group widths for the shared menu */
  HMENU              hmenuCombined;     /* The combined menu */
  BOOL               bIsServerItem;     /* True if the currently open popup belongs to the server */
} OleMenuDescriptor;

typedef struct tagOleMenuHookItem   /* OleMenu hook item in per thread hook list */
{
  DWORD tid;                /* Thread Id  */
  HANDLE hHeap;             /* Heap this is allocated from */
  HHOOK GetMsg_hHook;       /* message hook for WH_GETMESSAGE */
  HHOOK CallWndProc_hHook;  /* message hook for WH_CALLWNDPROC */
92
  struct tagOleMenuHookItem *next;
93 94
} OleMenuHookItem;

95
static OleMenuHookItem *hook_list;
96

97 98 99 100
/*
 * This is the lock count on the OLE library. It is controlled by the
 * OLEInitialize/OLEUninitialize methods.
 */
101
static LONG OLE_moduleLockCount = 0;
102 103 104 105

/*
 * Name of our registered window class.
 */
106 107 108 109 110 111 112 113
static const WCHAR OLEDD_DRAGTRACKERCLASS[] =
  {'W','i','n','e','D','r','a','g','D','r','o','p','T','r','a','c','k','e','r','3','2',0};

/*
 * Name of menu descriptor property.
 */
static const WCHAR prop_olemenuW[] =
  {'P','R','O','P','_','O','L','E','M','e','n','u','D','e','s','c','r','i','p','t','o','r',0};
114

115 116 117 118
/* property to store IDropTarget pointer */
static const WCHAR prop_oledroptarget[] =
  {'O','l','e','D','r','o','p','T','a','r','g','e','t','I','n','t','e','r','f','a','c','e',0};

119 120 121 122
/* property to store Marshalled IDropTarget pointer */
static const WCHAR prop_marshalleddroptarget[] =
  {'W','i','n','e','M','a','r','s','h','a','l','l','e','d','D','r','o','p','T','a','r','g','e','t',0};

123 124 125 126 127 128 129
static const WCHAR clsidfmtW[] =
  {'C','L','S','I','D','\\','{','%','0','8','x','-','%','0','4','x','-','%','0','4','x','-',
   '%','0','2','x','%','0','2','x','-','%','0','2','x','%','0','2','x','%','0','2','x','%','0','2','x',
    '%','0','2','x','%','0','2','x','}','\\',0};

static const WCHAR emptyW[] = { 0 };

130
/******************************************************************************
131
 * These are the prototypes of miscellaneous utility methods
132 133 134
 */
static void OLEUTL_ReadRegistryDWORDValue(HKEY regKey, DWORD* pdwValue);

135 136 137
/******************************************************************************
 * These are the prototypes of the utility methods used to manage a shared menu
 */
138 139
static void OLEMenu_Initialize(void);
static void OLEMenu_UnInitialize(void);
140 141 142
static BOOL OLEMenu_InstallHooks( DWORD tid );
static BOOL OLEMenu_UnInstallHooks( DWORD tid );
static OleMenuHookItem * OLEMenu_IsHookInstalled( DWORD tid );
143
static BOOL OLEMenu_FindMainMenuIndex( HMENU hMainMenu, HMENU hPopupMenu, UINT *pnPos );
144 145 146
static BOOL OLEMenu_SetIsServerMenu( HMENU hmenu, OleMenuDescriptor *pOleMenuDescriptor );
static LRESULT CALLBACK OLEMenu_CallWndProc(INT code, WPARAM wParam, LPARAM lParam);
static LRESULT CALLBACK OLEMenu_GetMsgProc(INT code, WPARAM wParam, LPARAM lParam);
147

148 149 150
/******************************************************************************
 * These are the prototypes of the OLE Clipboard initialization methods (in clipboard.c)
 */
151 152
extern void OLEClipbrd_UnInitialize(void);
extern void OLEClipbrd_Initialize(void);
153

154 155 156
/******************************************************************************
 * These are the prototypes of the utility methods used for OLE Drag n Drop
 */
157 158 159 160
static void OLEDD_Initialize(void);
static LRESULT WINAPI  OLEDD_DragTrackerWindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
static void OLEDD_TrackMouseMove(TrackerWindowInfo* trackerInfo);
static void OLEDD_TrackStateChange(TrackerWindowInfo* trackerInfo);
161
static DWORD OLEDD_GetButtonState(void);
162

163
/******************************************************************************
164
 *		OleBuildVersion [OLE32.@]
Alexandre Julliard's avatar
Alexandre Julliard committed
165
 */
166
DWORD WINAPI OleBuildVersion(void)
Alexandre Julliard's avatar
Alexandre Julliard committed
167
{
168
    TRACE("Returning version %d, build %d.\n", rmm, rup);
Alexandre Julliard's avatar
Alexandre Julliard committed
169
    return (rmm<<16)+rup;
Alexandre Julliard's avatar
Alexandre Julliard committed
170 171 172
}

/***********************************************************************
173
 *           OleInitialize       (OLE32.@)
Alexandre Julliard's avatar
Alexandre Julliard committed
174
 */
Alexandre Julliard's avatar
Alexandre Julliard committed
175
HRESULT WINAPI OleInitialize(LPVOID reserved)
Alexandre Julliard's avatar
Alexandre Julliard committed
176
{
177 178
  HRESULT hr;

179
  TRACE("(%p)\n", reserved);
180 181 182 183

  /*
   * The first duty of the OleInitialize is to initialize the COM libraries.
   */
184
  hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
185 186

  /*
187
   * If the CoInitializeEx call failed, the OLE libraries can't be
188 189 190
   * initialized.
   */
  if (FAILED(hr))
191
    return hr;
192

193 194 195
  if (!COM_CurrentInfo()->ole_inits)
    hr = S_OK;

196 197 198 199 200 201 202 203
  /*
   * Then, it has to initialize the OLE specific modules.
   * This includes:
   *     Clipboard
   *     Drag and Drop
   *     Object linking and Embedding
   *     In-place activation
   */
204 205
  if (!COM_CurrentInfo()->ole_inits++ &&
      InterlockedIncrement(&OLE_moduleLockCount) == 1)
206
  {
207
    /*
208 209
     * Initialize the libraries.
     */
210
    TRACE("() - Initializing the OLE libraries\n");
211

212 213 214 215 216
    /*
     * OLE Clipboard
     */
    OLEClipbrd_Initialize();

217 218 219 220
    /*
     * Drag and Drop
     */
    OLEDD_Initialize();
221 222 223 224 225

    /*
     * OLE shared menu
     */
    OLEMenu_Initialize();
226
  }
227 228

  return hr;
Alexandre Julliard's avatar
Alexandre Julliard committed
229 230
}

231
/******************************************************************************
232
 *		OleUninitialize	[OLE32.@]
Alexandre Julliard's avatar
Alexandre Julliard committed
233
 */
Alexandre Julliard's avatar
Alexandre Julliard committed
234
void WINAPI OleUninitialize(void)
Alexandre Julliard's avatar
Alexandre Julliard committed
235
{
236
  TRACE("()\n");
237 238 239 240

  /*
   * If we hit the bottom of the lock stack, free the libraries.
   */
241
  if (!--COM_CurrentInfo()->ole_inits && !InterlockedDecrement(&OLE_moduleLockCount))
242 243 244 245
  {
    /*
     * Actually free the libraries.
     */
246
    TRACE("() - Freeing the last reference count\n");
247

248 249 250 251 252
    /*
     * OLE Clipboard
     */
    OLEClipbrd_UnInitialize();

253 254 255 256
    /*
     * OLE shared menu
     */
    OLEMenu_UnInitialize();
257
  }
258

259 260 261
  /*
   * Then, uninitialize the COM libraries.
   */
262
  CoUninitialize();
Alexandre Julliard's avatar
Alexandre Julliard committed
263
}
Alexandre Julliard's avatar
Alexandre Julliard committed
264

265
/******************************************************************************
266
 *		OleInitializeWOW	[OLE32.@]
Alexandre Julliard's avatar
Alexandre Julliard committed
267
 */
268
HRESULT WINAPI OleInitializeWOW(DWORD x, DWORD y) {
269
        FIXME("(0x%08x, 0x%08x),stub!\n",x, y);
Alexandre Julliard's avatar
Alexandre Julliard committed
270 271
        return 0;
}
Alexandre Julliard's avatar
Alexandre Julliard committed
272

273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381
/*************************************************************
 *           get_droptarget_handle
 *
 * Retrieve a handle to the map containing the marshalled IDropTarget.
 * This handle belongs to the process that called RegisterDragDrop.
 * See get_droptarget_local_handle().
 */
static inline HANDLE get_droptarget_handle(HWND hwnd)
{
    return GetPropW(hwnd, prop_marshalleddroptarget);
}

/*************************************************************
 *           is_droptarget
 *
 * Is the window a droptarget.
 */
static inline BOOL is_droptarget(HWND hwnd)
{
    return get_droptarget_handle(hwnd) ? TRUE : FALSE;
}

/*************************************************************
 *           get_droptarget_local_handle
 *
 * Retrieve a handle to the map containing the marshalled IDropTarget.
 * The handle should be closed when finished with.
 */
static HANDLE get_droptarget_local_handle(HWND hwnd)
{
    HANDLE handle, local_handle = 0;

    handle = get_droptarget_handle(hwnd);

    if(handle)
    {
        DWORD pid;
        HANDLE process;

        GetWindowThreadProcessId(hwnd, &pid);
        process = OpenProcess(PROCESS_DUP_HANDLE, FALSE, pid);
        if(process)
        {
            DuplicateHandle(process, handle, GetCurrentProcess(), &local_handle, 0, FALSE, DUPLICATE_SAME_ACCESS);
            CloseHandle(process);
        }
    }
    return local_handle;
}

/***********************************************************************
 *     create_map_from_stream
 *
 * Helper for RegisterDragDrop.  Creates a file mapping object
 * with the contents of the provided stream.  The stream must
 * be a global memory backed stream.
 */
static HRESULT create_map_from_stream(IStream *stream, HANDLE *map)
{
    HGLOBAL hmem;
    DWORD size;
    HRESULT hr;
    void *data;

    hr = GetHGlobalFromStream(stream, &hmem);
    if(FAILED(hr)) return hr;

    size = GlobalSize(hmem);
    *map = CreateFileMappingW(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, size, NULL);
    if(!*map) return E_OUTOFMEMORY;

    data = MapViewOfFile(*map, FILE_MAP_WRITE, 0, 0, size);
    memcpy(data, GlobalLock(hmem), size);
    GlobalUnlock(hmem);
    UnmapViewOfFile(data);
    return S_OK;
}

/***********************************************************************
 *     create_stream_from_map
 *
 * Creates a stream from the provided map.
 */
static HRESULT create_stream_from_map(HANDLE map, IStream **stream)
{
    HRESULT hr = E_OUTOFMEMORY;
    HGLOBAL hmem;
    void *data;
    MEMORY_BASIC_INFORMATION info;

    data = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0);
    if(!data) return hr;

    VirtualQuery(data, &info, sizeof(info));
    TRACE("size %d\n", (int)info.RegionSize);

    hmem = GlobalAlloc(GMEM_MOVEABLE, info.RegionSize);
    if(hmem)
    {
        memcpy(GlobalLock(hmem), data, info.RegionSize);
        GlobalUnlock(hmem);
        hr = CreateStreamOnHGlobal(hmem, TRUE, stream);
    }
    UnmapViewOfFile(data);
    return hr;
}

/***********************************************************************
 *     get_droptarget_pointer
382
 *
383
 * Retrieves the marshalled IDropTarget from the window.
384
 */
385
static IDropTarget* get_droptarget_pointer(HWND hwnd)
386
{
387 388 389 390 391 392 393 394 395 396 397 398 399 400
    IDropTarget *droptarget = NULL;
    HANDLE map;
    IStream *stream;

    map = get_droptarget_local_handle(hwnd);
    if(!map) return NULL;

    if(SUCCEEDED(create_stream_from_map(map, &stream)))
    {
        CoUnmarshalInterface(stream, &IID_IDropTarget, (void**)&droptarget);
        IStream_Release(stream);
    }
    CloseHandle(map);
    return droptarget;
401 402
}

Alexandre Julliard's avatar
Alexandre Julliard committed
403
/***********************************************************************
404
 *           RegisterDragDrop (OLE32.@)
Alexandre Julliard's avatar
Alexandre Julliard committed
405
 */
406
HRESULT WINAPI RegisterDragDrop(HWND hwnd, LPDROPTARGET pDropTarget)
407
{
408
  DWORD pid = 0;
409 410 411
  HRESULT hr;
  IStream *stream;
  HANDLE map;
412
  IUnknown *unk;
413

414
  TRACE("(%p,%p)\n", hwnd, pDropTarget);
415

416 417 418
  if (!COM_CurrentApt())
  {
    ERR("COM not initialized\n");
419
    return E_OUTOFMEMORY;
420 421
  }

422 423
  if (!pDropTarget)
    return E_INVALIDARG;
424 425

  if (!IsWindow(hwnd))
426 427
  {
    ERR("invalid hwnd %p\n", hwnd);
428
    return DRAGDROP_E_INVALIDHWND;
429
  }
430

431 432 433 434 435 436 437
  /* block register for other processes windows */
  GetWindowThreadProcessId(hwnd, &pid);
  if (pid != GetCurrentProcessId())
  {
    FIXME("register for another process windows is disabled\n");
    return DRAGDROP_E_INVALIDHWND;
  }
438

439
  /* check if the window is already registered */
440
  if (is_droptarget(hwnd))
441 442
    return DRAGDROP_E_ALREADYREGISTERED;

443 444 445 446 447 448
  /*
   * Marshal the drop target pointer into a shared memory map and
   * store the map's handle in a Wine specific window prop.  We also
   * store the drop target pointer itself in the
   * "OleDropTargetInterface" prop for compatibility with Windows.
   */
449

450 451 452
  hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
  if(FAILED(hr)) return hr;

453
  unk = NULL;
454
  hr = IDropTarget_QueryInterface(pDropTarget, &IID_IUnknown, (void**)&unk);
455
  if (SUCCEEDED(hr) && !unk) hr = E_NOINTERFACE;
456 457 458 459 460 461 462 463
  if(FAILED(hr))
  {
      IStream_Release(stream);
      return hr;
  }
  hr = CoMarshalInterface(stream, &IID_IDropTarget, unk, MSHCTX_LOCAL, NULL, MSHLFLAGS_TABLESTRONG);
  IUnknown_Release(unk);

464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483
  if(SUCCEEDED(hr))
  {
    hr = create_map_from_stream(stream, &map);
    if(SUCCEEDED(hr))
    {
      IDropTarget_AddRef(pDropTarget);
      SetPropW(hwnd, prop_oledroptarget, pDropTarget);
      SetPropW(hwnd, prop_marshalleddroptarget, map);
    }
    else
    {
      LARGE_INTEGER zero;
      zero.QuadPart = 0;
      IStream_Seek(stream, zero, STREAM_SEEK_SET, NULL);
      CoReleaseMarshalData(stream);
    }
  }
  IStream_Release(stream);

  return hr;
Alexandre Julliard's avatar
Alexandre Julliard committed
484
}
Alexandre Julliard's avatar
Alexandre Julliard committed
485 486

/***********************************************************************
487
 *           RevokeDragDrop (OLE32.@)
Alexandre Julliard's avatar
Alexandre Julliard committed
488
 */
489
HRESULT WINAPI RevokeDragDrop(HWND hwnd)
490
{
491 492 493 494
  HANDLE map;
  IStream *stream;
  IDropTarget *drop_target;
  HRESULT hr;
495

496
  TRACE("(%p)\n", hwnd);
497

498 499 500 501 502 503
  if (!IsWindow(hwnd))
  {
    ERR("invalid hwnd %p\n", hwnd);
    return DRAGDROP_E_INVALIDHWND;
  }

504
  /* no registration data */
505
  if (!(map = get_droptarget_handle(hwnd)))
506 507
    return DRAGDROP_E_NOTREGISTERED;

508 509 510
  drop_target = GetPropW(hwnd, prop_oledroptarget);
  if(drop_target) IDropTarget_Release(drop_target);

511
  RemovePropW(hwnd, prop_oledroptarget);
512
  RemovePropW(hwnd, prop_marshalleddroptarget);
513

514 515 516 517 518 519 520 521 522
  hr = create_stream_from_map(map, &stream);
  if(SUCCEEDED(hr))
  {
      CoReleaseMarshalData(stream);
      IStream_Release(stream);
  }
  CloseHandle(map);

  return hr;
Alexandre Julliard's avatar
Alexandre Julliard committed
523
}
524

525
/***********************************************************************
526
 *           OleRegGetUserType (OLE32.@)
527 528 529 530 531
 *
 * This implementation of OleRegGetUserType ignores the dwFormOfType
 * parameter and always returns the full name of the object. This is
 * not too bad since this is the case for many objects because of the
 * way they are registered.
532
 */
533 534
HRESULT WINAPI OleRegGetUserType(
	REFCLSID clsid,
535
	DWORD dwFormOfType,
536
	LPOLESTR* pszUserType)
537
{
538
  WCHAR   keyName[60];
539 540 541 542
  DWORD   dwKeyType;
  DWORD   cbData;
  HKEY    clsidKey;
  LONG    hres;
543

544 545 546 547 548 549 550 551
  /*
   * Initialize the out parameter.
   */
  *pszUserType = NULL;

  /*
   * Build the key name we're looking for
   */
552 553 554 555
  sprintfW( keyName, clsidfmtW,
            clsid->Data1, clsid->Data2, clsid->Data3,
            clsid->Data4[0], clsid->Data4[1], clsid->Data4[2], clsid->Data4[3],
            clsid->Data4[4], clsid->Data4[5], clsid->Data4[6], clsid->Data4[7] );
556

557
  TRACE("(%s, %d, %p)\n", debugstr_w(keyName), dwFormOfType, pszUserType);
558 559 560 561

  /*
   * Open the class id Key
   */
562
  hres = RegOpenKeyW(HKEY_CLASSES_ROOT,
563 564 565 566 567 568 569 570 571 572 573
		     keyName,
		     &clsidKey);

  if (hres != ERROR_SUCCESS)
    return REGDB_E_CLASSNOTREG;

  /*
   * Retrieve the size of the name string.
   */
  cbData = 0;

574 575
  hres = RegQueryValueExW(clsidKey,
			  emptyW,
576 577 578 579 580 581 582 583 584 585 586 587 588 589
			  NULL,
			  &dwKeyType,
			  NULL,
			  &cbData);

  if (hres!=ERROR_SUCCESS)
  {
    RegCloseKey(clsidKey);
    return REGDB_E_READREGDB;
  }

  /*
   * Allocate a buffer for the registry value.
   */
590
  *pszUserType = CoTaskMemAlloc(cbData);
591 592 593 594 595 596 597

  if (*pszUserType==NULL)
  {
    RegCloseKey(clsidKey);
    return E_OUTOFMEMORY;
  }

598 599
  hres = RegQueryValueExW(clsidKey,
			  emptyW,
600 601
			  NULL,
			  &dwKeyType,
602
			  (LPBYTE) *pszUserType,
603 604 605
			  &cbData);

  RegCloseKey(clsidKey);
606

607
  if (hres != ERROR_SUCCESS)
608 609
  {
    CoTaskMemFree(*pszUserType);
610
    *pszUserType = NULL;
611

612
    return REGDB_E_READREGDB;
613 614
  }

615
  return S_OK;
616 617
}

618
/***********************************************************************
619
 * DoDragDrop [OLE32.@]
620
 */
621
HRESULT WINAPI DoDragDrop (
622 623 624 625
  IDataObject *pDataObject,  /* [in] ptr to the data obj           */
  IDropSource* pDropSource,  /* [in] ptr to the source obj         */
  DWORD       dwOKEffect,    /* [in] effects allowed by the source */
  DWORD       *pdwEffect)    /* [out] ptr to effects of the source */
626
{
627
  static const WCHAR trackerW[] = {'T','r','a','c','k','e','r','W','i','n','d','o','w',0};
628
  TrackerWindowInfo trackerInfo;
629 630
  HWND            hwndTrackWindow;
  MSG             msg;
631

632 633 634 635
  TRACE("(%p, %p, %d, %p)\n", pDataObject, pDropSource, dwOKEffect, pdwEffect);

  if (!pDataObject || !pDropSource || !pdwEffect)
      return E_INVALIDARG;
636 637 638 639

  /*
   * Setup the drag n drop tracking window.
   */
640

641 642 643 644 645 646 647
  trackerInfo.dataObject        = pDataObject;
  trackerInfo.dropSource        = pDropSource;
  trackerInfo.dwOKEffect        = dwOKEffect;
  trackerInfo.pdwEffect         = pdwEffect;
  trackerInfo.trackingDone      = FALSE;
  trackerInfo.escPressed        = FALSE;
  trackerInfo.curDragTargetHWND = 0;
648
  trackerInfo.curTargetHWND     = 0;
649 650
  trackerInfo.curDragTarget     = 0;

651
  hwndTrackWindow = CreateWindowW(OLEDD_DRAGTRACKERCLASS, trackerW,
652 653 654
                                  WS_POPUP, CW_USEDEFAULT, CW_USEDEFAULT,
                                  CW_USEDEFAULT, CW_USEDEFAULT, 0, 0, 0,
                                  &trackerInfo);
655

656
  if (hwndTrackWindow)
657 658 659 660
  {
    /*
     * Capture the mouse input
     */
661
    SetCapture(hwndTrackWindow);
662

663 664
    msg.message = 0;

665
    /*
666
     * Pump messages. All mouse input should go to the capture window.
667
     */
668
    while (!trackerInfo.trackingDone && GetMessageW(&msg, 0, 0, 0) )
669
    {
670 671 672 673
      trackerInfo.curMousePos.x = msg.pt.x;
      trackerInfo.curMousePos.y = msg.pt.y;
      trackerInfo.dwKeyState = OLEDD_GetButtonState();
	    
674
      if ( (msg.message >= WM_KEYFIRST) &&
675
	   (msg.message <= WM_KEYLAST) )
676 677 678 679 680 681 682 683 684 685 686 687 688 689 690
      {
	/*
	 * When keyboard messages are sent to windows on this thread, we
	 * want to ignore notify the drop source that the state changed.
	 * in the case of the Escape key, we also notify the drop source
	 * we give it a special meaning.
	 */
	if ( (msg.message==WM_KEYDOWN) &&
	     (msg.wParam==VK_ESCAPE) )
	{
	  trackerInfo.escPressed = TRUE;
	}

	/*
	 * Notify the drop source.
691
	 */
692
        OLEDD_TrackStateChange(&trackerInfo);
693 694 695 696 697 698
      }
      else
      {
	/*
	 * Dispatch the messages only when it's not a keyboard message.
	 */
699
	DispatchMessageW(&msg);
700 701 702
      }
    }

703 704 705
    /* re-post the quit message to outer message loop */
    if (msg.message == WM_QUIT)
        PostQuitMessage(msg.wParam);
706 707 708
    /*
     * Destroy the temporary window.
     */
709
    DestroyWindow(hwndTrackWindow);
710 711 712 713 714 715 716 717

    return trackerInfo.returnValue;
  }

  return E_FAIL;
}

/***********************************************************************
718
 * OleQueryLinkFromData [OLE32.@]
719
 */
720
HRESULT WINAPI OleQueryLinkFromData(
721 722
  IDataObject* pSrcDataObject)
{
723
  FIXME("(%p),stub!\n", pSrcDataObject);
724
  return S_FALSE;
725 726 727
}

/***********************************************************************
728
 * OleRegGetMiscStatus [OLE32.@]
729
 */
730
HRESULT WINAPI OleRegGetMiscStatus(
731 732 733 734
  REFCLSID clsid,
  DWORD    dwAspect,
  DWORD*   pdwStatus)
{
735 736 737
  static const WCHAR miscstatusW[] = {'M','i','s','c','S','t','a','t','u','s',0};
  static const WCHAR dfmtW[] = {'%','d',0};
  WCHAR   keyName[60];
738 739 740 741 742 743 744 745 746 747 748 749 750
  HKEY    clsidKey;
  HKEY    miscStatusKey;
  HKEY    aspectKey;
  LONG    result;

  /*
   * Initialize the out parameter.
   */
  *pdwStatus = 0;

  /*
   * Build the key name we're looking for
   */
751 752 753 754
  sprintfW( keyName, clsidfmtW,
            clsid->Data1, clsid->Data2, clsid->Data3,
            clsid->Data4[0], clsid->Data4[1], clsid->Data4[2], clsid->Data4[3],
            clsid->Data4[4], clsid->Data4[5], clsid->Data4[6], clsid->Data4[7] );
755

756
  TRACE("(%s, %d, %p)\n", debugstr_w(keyName), dwAspect, pdwStatus);
757 758 759 760

  /*
   * Open the class id Key
   */
761
  result = RegOpenKeyW(HKEY_CLASSES_ROOT,
762 763 764 765 766 767 768 769 770
		       keyName,
		       &clsidKey);

  if (result != ERROR_SUCCESS)
    return REGDB_E_CLASSNOTREG;

  /*
   * Get the MiscStatus
   */
771 772
  result = RegOpenKeyW(clsidKey,
		       miscstatusW,
773 774
		       &miscStatusKey);

775

776 777 778 779 780 781 782 783 784 785 786 787 788 789
  if (result != ERROR_SUCCESS)
  {
    RegCloseKey(clsidKey);
    return REGDB_E_READREGDB;
  }

  /*
   * Read the default value
   */
  OLEUTL_ReadRegistryDWORDValue(miscStatusKey, pdwStatus);

  /*
   * Open the key specific to the requested aspect.
   */
790
  sprintfW(keyName, dfmtW, dwAspect);
791

792
  result = RegOpenKeyW(miscStatusKey,
793 794
		       keyName,
		       &aspectKey);
795

796 797 798 799 800 801 802 803 804 805 806 807 808 809 810
  if (result == ERROR_SUCCESS)
  {
    OLEUTL_ReadRegistryDWORDValue(aspectKey, pdwStatus);
    RegCloseKey(aspectKey);
  }

  /*
   * Cleanup
   */
  RegCloseKey(miscStatusKey);
  RegCloseKey(clsidKey);

  return S_OK;
}

811 812 813 814
static HRESULT EnumOLEVERB_Construct(HKEY hkeyVerb, ULONG index, IEnumOLEVERB **ppenum);

typedef struct
{
815
    IEnumOLEVERB IEnumOLEVERB_iface;
816 817 818 819 820 821
    LONG ref;

    HKEY hkeyVerb;
    ULONG index;
} EnumOLEVERB;

822 823 824 825 826
static inline EnumOLEVERB *impl_from_IEnumOLEVERB(IEnumOLEVERB *iface)
{
    return CONTAINING_RECORD(iface, EnumOLEVERB, IEnumOLEVERB_iface);
}

827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843
static HRESULT WINAPI EnumOLEVERB_QueryInterface(
    IEnumOLEVERB *iface, REFIID riid, void **ppv)
{
    TRACE("(%s, %p)\n", debugstr_guid(riid), ppv);
    if (IsEqualIID(riid, &IID_IUnknown) ||
        IsEqualIID(riid, &IID_IEnumOLEVERB))
    {
        IUnknown_AddRef(iface);
        *ppv = iface;
        return S_OK;
    }
    return E_NOINTERFACE;
}

static ULONG WINAPI EnumOLEVERB_AddRef(
    IEnumOLEVERB *iface)
{
844
    EnumOLEVERB *This = impl_from_IEnumOLEVERB(iface);
845 846 847 848 849 850 851
    TRACE("()\n");
    return InterlockedIncrement(&This->ref);
}

static ULONG WINAPI EnumOLEVERB_Release(
    IEnumOLEVERB *iface)
{
852
    EnumOLEVERB *This = impl_from_IEnumOLEVERB(iface);
853 854 855 856 857 858 859 860 861 862 863 864 865 866
    LONG refs = InterlockedDecrement(&This->ref);
    TRACE("()\n");
    if (!refs)
    {
        RegCloseKey(This->hkeyVerb);
        HeapFree(GetProcessHeap(), 0, This);
    }
    return refs;
}

static HRESULT WINAPI EnumOLEVERB_Next(
    IEnumOLEVERB *iface, ULONG celt, LPOLEVERB rgelt,
    ULONG *pceltFetched)
{
867
    EnumOLEVERB *This = impl_from_IEnumOLEVERB(iface);
868 869
    HRESULT hr = S_OK;

870
    TRACE("(%d, %p, %p)\n", celt, rgelt, pceltFetched);
871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889

    if (pceltFetched)
        *pceltFetched = 0;

    for (; celt; celt--, rgelt++)
    {
        WCHAR wszSubKey[20];
        LONG cbData;
        LPWSTR pwszOLEVERB;
        LPWSTR pwszMenuFlags;
        LPWSTR pwszAttribs;
        LONG res = RegEnumKeyW(This->hkeyVerb, This->index, wszSubKey, sizeof(wszSubKey)/sizeof(wszSubKey[0]));
        if (res == ERROR_NO_MORE_ITEMS)
        {
            hr = S_FALSE;
            break;
        }
        else if (res != ERROR_SUCCESS)
        {
890
            ERR("RegEnumKeyW failed with error %d\n", res);
891 892 893 894 895 896
            hr = REGDB_E_READREGDB;
            break;
        }
        res = RegQueryValueW(This->hkeyVerb, wszSubKey, NULL, &cbData);
        if (res != ERROR_SUCCESS)
        {
897
            ERR("RegQueryValueW failed with error %d\n", res);
898 899 900 901 902 903 904 905 906 907 908 909
            hr = REGDB_E_READREGDB;
            break;
        }
        pwszOLEVERB = CoTaskMemAlloc(cbData);
        if (!pwszOLEVERB)
        {
            hr = E_OUTOFMEMORY;
            break;
        }
        res = RegQueryValueW(This->hkeyVerb, wszSubKey, pwszOLEVERB, &cbData);
        if (res != ERROR_SUCCESS)
        {
910
            ERR("RegQueryValueW failed with error %d\n", res);
911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944
            hr = REGDB_E_READREGDB;
            CoTaskMemFree(pwszOLEVERB);
            break;
        }

        TRACE("verb string: %s\n", debugstr_w(pwszOLEVERB));
        pwszMenuFlags = strchrW(pwszOLEVERB, ',');
        if (!pwszMenuFlags)
        {
            hr = OLEOBJ_E_INVALIDVERB;
            CoTaskMemFree(pwszOLEVERB);
            break;
        }
        /* nul terminate the name string and advance to first character */
        *pwszMenuFlags = '\0';
        pwszMenuFlags++;
        pwszAttribs = strchrW(pwszMenuFlags, ',');
        if (!pwszAttribs)
        {
            hr = OLEOBJ_E_INVALIDVERB;
            CoTaskMemFree(pwszOLEVERB);
            break;
        }
        /* nul terminate the menu string and advance to first character */
        *pwszAttribs = '\0';
        pwszAttribs++;

        /* fill out structure for this verb */
        rgelt->lVerb = atolW(wszSubKey);
        rgelt->lpszVerbName = pwszOLEVERB; /* user should free */
        rgelt->fuFlags = atolW(pwszMenuFlags);
        rgelt->grfAttribs = atolW(pwszAttribs);

        if (pceltFetched)
945
            (*pceltFetched)++;
946 947 948 949 950 951 952 953
        This->index++;
    }
    return hr;
}

static HRESULT WINAPI EnumOLEVERB_Skip(
    IEnumOLEVERB *iface, ULONG celt)
{
954
    EnumOLEVERB *This = impl_from_IEnumOLEVERB(iface);
955

956
    TRACE("(%d)\n", celt);
957 958 959 960 961 962 963 964

    This->index += celt;
    return S_OK;
}

static HRESULT WINAPI EnumOLEVERB_Reset(
    IEnumOLEVERB *iface)
{
965
    EnumOLEVERB *This = impl_from_IEnumOLEVERB(iface);
966 967 968 969 970 971 972 973 974 975 976

    TRACE("()\n");

    This->index = 0;
    return S_OK;
}

static HRESULT WINAPI EnumOLEVERB_Clone(
    IEnumOLEVERB *iface,
    IEnumOLEVERB **ppenum)
{
977
    EnumOLEVERB *This = impl_from_IEnumOLEVERB(iface);
978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003
    HKEY hkeyVerb;
    TRACE("(%p)\n", ppenum);
    if (!DuplicateHandle(GetCurrentProcess(), This->hkeyVerb, GetCurrentProcess(), (HANDLE *)&hkeyVerb, 0, FALSE, DUPLICATE_SAME_ACCESS))
        return HRESULT_FROM_WIN32(GetLastError());
    return EnumOLEVERB_Construct(hkeyVerb, This->index, ppenum);
}

static const IEnumOLEVERBVtbl EnumOLEVERB_VTable =
{
    EnumOLEVERB_QueryInterface,
    EnumOLEVERB_AddRef,
    EnumOLEVERB_Release,
    EnumOLEVERB_Next,
    EnumOLEVERB_Skip,
    EnumOLEVERB_Reset,
    EnumOLEVERB_Clone
};

static HRESULT EnumOLEVERB_Construct(HKEY hkeyVerb, ULONG index, IEnumOLEVERB **ppenum)
{
    EnumOLEVERB *This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
    if (!This)
    {
        RegCloseKey(hkeyVerb);
        return E_OUTOFMEMORY;
    }
1004
    This->IEnumOLEVERB_iface.lpVtbl = &EnumOLEVERB_VTable;
1005 1006 1007
    This->ref = 1;
    This->index = index;
    This->hkeyVerb = hkeyVerb;
1008 1009
    *ppenum = &This->IEnumOLEVERB_iface;
    return S_OK;
1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036
}

/***********************************************************************
 *           OleRegEnumVerbs    [OLE32.@]
 *
 * Enumerates verbs associated with a class stored in the registry.
 *
 * PARAMS
 *  clsid  [I] Class ID to enumerate the verbs for.
 *  ppenum [O] Enumerator.
 *
 * RETURNS
 *  S_OK: Success.
 *  REGDB_E_CLASSNOTREG: The specified class does not have a key in the registry.
 *  REGDB_E_READREGDB: The class key could not be opened for some other reason.
 *  OLE_E_REGDB_KEY: The Verb subkey for the class is not present.
 *  OLEOBJ_E_NOVERBS: The Verb subkey for the class is empty.
 */
HRESULT WINAPI OleRegEnumVerbs (REFCLSID clsid, LPENUMOLEVERB* ppenum)
{
    LONG res;
    HKEY hkeyVerb;
    DWORD dwSubKeys;
    static const WCHAR wszVerb[] = {'V','e','r','b',0};

    TRACE("(%s, %p)\n", debugstr_guid(clsid), ppenum);

1037 1038
    res = COM_OpenKeyForCLSID(clsid, wszVerb, KEY_READ, &hkeyVerb);
    if (FAILED(res))
1039
    {
1040 1041 1042 1043 1044
        if (res == REGDB_E_CLASSNOTREG)
            ERR("CLSID %s not registered\n", debugstr_guid(clsid));
        else if (res == REGDB_E_KEYMISSING)
            ERR("no Verbs key for class %s\n", debugstr_guid(clsid));
        else
1045
            ERR("failed to open Verbs key for CLSID %s with error %d\n",
1046 1047
                debugstr_guid(clsid), res);
        return res;
1048 1049 1050 1051 1052 1053
    }

    res = RegQueryInfoKeyW(hkeyVerb, NULL, NULL, NULL, &dwSubKeys, NULL,
                          NULL, NULL, NULL, NULL, NULL, NULL);
    if (res != ERROR_SUCCESS)
    {
1054
        ERR("failed to get subkey count with error %d\n", GetLastError());
1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067
        return REGDB_E_READREGDB;
    }

    if (!dwSubKeys)
    {
        WARN("class %s has no verbs\n", debugstr_guid(clsid));
        RegCloseKey(hkeyVerb);
        return OLEOBJ_E_NOVERBS;
    }

    return EnumOLEVERB_Construct(hkeyVerb, 0, ppenum);
}

1068
/******************************************************************************
1069
 *              OleSetContainedObject        [OLE32.@]
1070 1071
 */
HRESULT WINAPI OleSetContainedObject(
1072
  LPUNKNOWN pUnknown,
1073 1074 1075 1076 1077
  BOOL      fContained)
{
  IRunnableObject* runnable = NULL;
  HRESULT          hres;

1078
  TRACE("(%p,%x)\n", pUnknown, fContained);
1079 1080 1081

  hres = IUnknown_QueryInterface(pUnknown,
				 &IID_IRunnableObject,
1082
				 (void**)&runnable);
1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095

  if (SUCCEEDED(hres))
  {
    hres = IRunnableObject_SetContainedObject(runnable, fContained);

    IRunnableObject_Release(runnable);

    return hres;
  }

  return S_OK;
}

1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123
/******************************************************************************
 *              OleRun        [OLE32.@]
 *
 * Set the OLE object to the running state.
 *
 * PARAMS
 *  pUnknown [I] OLE object to run.
 *
 * RETURNS
 *  Success: S_OK.
 *  Failure: Any HRESULT code.
 */
HRESULT WINAPI OleRun(LPUNKNOWN pUnknown)
{
    IRunnableObject *runable;
    HRESULT hres;

    TRACE("(%p)\n", pUnknown);

    hres = IUnknown_QueryInterface(pUnknown, &IID_IRunnableObject, (void**)&runable);
    if (FAILED(hres))
        return S_OK; /* Appears to return no error. */

    hres = IRunnableObject_Run(runable, NULL);
    IRunnableObject_Release(runable);
    return hres;
}

1124
/******************************************************************************
1125
 *              OleLoad        [OLE32.@]
1126 1127
 */
HRESULT WINAPI OleLoad(
1128 1129 1130
  LPSTORAGE       pStg,
  REFIID          riid,
  LPOLECLIENTSITE pClientSite,
1131 1132 1133
  LPVOID*         ppvObj)
{
  IPersistStorage* persistStorage = NULL;
1134 1135
  IUnknown*        pUnk;
  IOleObject*      pOleObject      = NULL;
1136 1137 1138
  STATSTG          storageInfo;
  HRESULT          hres;

1139
  TRACE("(%p, %s, %p, %p)\n", pStg, debugstr_guid(riid), pClientSite, ppvObj);
1140

1141 1142
  *ppvObj = NULL;

1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156
  /*
   * TODO, Conversion ... OleDoAutoConvert
   */

  /*
   * Get the class ID for the object.
   */
  hres = IStorage_Stat(pStg, &storageInfo, STATFLAG_NONAME);

  /*
   * Now, try and create the handler for the object
   */
  hres = CoCreateInstance(&storageInfo.clsid,
			  NULL,
1157
			  CLSCTX_INPROC_HANDLER|CLSCTX_INPROC_SERVER,
1158 1159
			  riid,
			  (void**)&pUnk);
1160 1161 1162 1163 1164 1165 1166 1167 1168

  /*
   * If that fails, as it will most times, load the default
   * OLE handler.
   */
  if (FAILED(hres))
  {
    hres = OleCreateDefaultHandler(&storageInfo.clsid,
				   NULL,
1169 1170
				   riid,
				   (void**)&pUnk);
1171 1172 1173 1174 1175 1176 1177 1178
  }

  /*
   * If we couldn't find a handler... this is bad. Abort the whole thing.
   */
  if (FAILED(hres))
    return hres;

1179 1180 1181 1182 1183 1184 1185 1186 1187
  if (pClientSite)
  {
    hres = IUnknown_QueryInterface(pUnk, &IID_IOleObject, (void **)&pOleObject);
    if (SUCCEEDED(hres))
    {
        DWORD dwStatus;
        hres = IOleObject_GetMiscStatus(pOleObject, DVASPECT_CONTENT, &dwStatus);
    }
  }
1188

1189 1190 1191 1192 1193 1194
  /*
   * Initialize the object with its IPersistStorage interface.
   */
  hres = IOleObject_QueryInterface(pUnk,
                                   &IID_IPersistStorage,
                                   (void**)&persistStorage);
1195

1196
  if (SUCCEEDED(hres))
1197
  {
1198
    hres = IPersistStorage_Load(persistStorage, pStg);
1199 1200 1201 1202 1203

    IPersistStorage_Release(persistStorage);
    persistStorage = NULL;
  }

1204 1205 1206 1207 1208
  if (SUCCEEDED(hres) && pClientSite)
    /*
     * Inform the new object of it's client site.
     */
    hres = IOleObject_SetClientSite(pOleObject, pClientSite);
1209 1210 1211 1212

  /*
   * Cleanup interfaces used internally
   */
1213 1214 1215
  if (pOleObject)
    IOleObject_Release(pOleObject);

1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227
  if (SUCCEEDED(hres))
  {
    IOleLink *pOleLink;
    HRESULT hres1;
    hres1 = IUnknown_QueryInterface(pUnk, &IID_IOleLink, (void **)&pOleLink);
    if (SUCCEEDED(hres1))
    {
      FIXME("handle OLE link\n");
      IOleLink_Release(pOleLink);
    }
  }

1228 1229 1230 1231 1232 1233 1234
  if (FAILED(hres))
  {
    IUnknown_Release(pUnk);
    pUnk = NULL;
  }

  *ppvObj = pUnk;
1235 1236

  return hres;
1237 1238
}

1239
/***********************************************************************
1240
 *           OleSave     [OLE32.@]
1241 1242 1243 1244 1245 1246 1247 1248 1249
 */
HRESULT WINAPI OleSave(
  LPPERSISTSTORAGE pPS,
  LPSTORAGE        pStg,
  BOOL             fSameAsLoad)
{
  HRESULT hres;
  CLSID   objectClass;

1250
  TRACE("(%p,%p,%x)\n", pPS, pStg, fSameAsLoad);
1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272

  /*
   * First, we transfer the class ID (if available)
   */
  hres = IPersistStorage_GetClassID(pPS, &objectClass);

  if (SUCCEEDED(hres))
  {
    WriteClassStg(pStg, &objectClass);
  }

  /*
   * Then, we ask the object to save itself to the
   * storage. If it is successful, we commit the storage.
   */
  hres = IPersistStorage_Save(pPS, pStg, fSameAsLoad);

  if (SUCCEEDED(hres))
  {
    IStorage_Commit(pStg,
		    STGC_DEFAULT);
  }
1273

1274 1275 1276 1277
  return hres;
}


1278
/******************************************************************************
1279
 *              OleLockRunning        [OLE32.@]
1280
 */
1281
HRESULT WINAPI OleLockRunning(LPUNKNOWN pUnknown, BOOL fLock, BOOL fLastUnlockCloses)
1282 1283 1284 1285 1286 1287 1288 1289
{
  IRunnableObject* runnable = NULL;
  HRESULT          hres;

  TRACE("(%p,%x,%x)\n", pUnknown, fLock, fLastUnlockCloses);

  hres = IUnknown_QueryInterface(pUnknown,
				 &IID_IRunnableObject,
1290
				 (void**)&runnable);
1291 1292 1293 1294 1295 1296 1297 1298 1299

  if (SUCCEEDED(hres))
  {
    hres = IRunnableObject_LockRunning(runnable, fLock, fLastUnlockCloses);

    IRunnableObject_Release(runnable);

    return hres;
  }
1300 1301

  return S_OK;
1302 1303 1304
}


1305 1306 1307 1308 1309 1310 1311 1312 1313 1314
/**************************************************************************
 * Internal methods to manage the shared OLE menu in response to the
 * OLE***MenuDescriptor API
 */

/***
 * OLEMenu_Initialize()
 *
 * Initializes the OLEMENU data structures.
 */
1315
static void OLEMenu_Initialize(void)
1316 1317 1318 1319 1320 1321 1322 1323
{
}

/***
 * OLEMenu_UnInitialize()
 *
 * Releases the OLEMENU data structures.
 */
1324
static void OLEMenu_UnInitialize(void)
1325 1326 1327 1328 1329 1330 1331
{
}

/*************************************************************************
 * OLEMenu_InstallHooks
 * Install thread scope message hooks for WH_GETMESSAGE and WH_CALLWNDPROC
 *
1332
 * RETURNS: TRUE if message hooks were successfully installed
1333 1334
 *          FALSE on failure
 */
1335
static BOOL OLEMenu_InstallHooks( DWORD tid )
1336
{
1337
  OleMenuHookItem *pHookItem;
1338 1339 1340 1341 1342 1343 1344 1345

  /* Create an entry for the hook table */
  if ( !(pHookItem = HeapAlloc(GetProcessHeap(), 0,
                               sizeof(OleMenuHookItem)) ) )
    return FALSE;

  pHookItem->tid = tid;
  pHookItem->hHeap = GetProcessHeap();
1346
  pHookItem->CallWndProc_hHook = NULL;
1347

1348
  /* Install a thread scope message hook for WH_GETMESSAGE */
1349
  pHookItem->GetMsg_hHook = SetWindowsHookExW( WH_GETMESSAGE, OLEMenu_GetMsgProc,
1350 1351 1352 1353 1354
                                               0, GetCurrentThreadId() );
  if ( !pHookItem->GetMsg_hHook )
    goto CLEANUP;

  /* Install a thread scope message hook for WH_CALLWNDPROC */
1355
  pHookItem->CallWndProc_hHook = SetWindowsHookExW( WH_CALLWNDPROC, OLEMenu_CallWndProc,
1356 1357 1358 1359 1360
                                                    0, GetCurrentThreadId() );
  if ( !pHookItem->CallWndProc_hHook )
    goto CLEANUP;

  /* Insert the hook table entry */
1361 1362
  pHookItem->next = hook_list;
  hook_list = pHookItem;
1363

1364
  return TRUE;
1365

1366 1367 1368 1369 1370 1371 1372 1373
CLEANUP:
  /* Unhook any hooks */
  if ( pHookItem->GetMsg_hHook )
    UnhookWindowsHookEx( pHookItem->GetMsg_hHook );
  if ( pHookItem->CallWndProc_hHook )
    UnhookWindowsHookEx( pHookItem->CallWndProc_hHook );
  /* Release the hook table entry */
  HeapFree(pHookItem->hHeap, 0, pHookItem );
1374

1375 1376 1377 1378 1379 1380 1381
  return FALSE;
}

/*************************************************************************
 * OLEMenu_UnInstallHooks
 * UnInstall thread scope message hooks for WH_GETMESSAGE and WH_CALLWNDPROC
 *
1382
 * RETURNS: TRUE if message hooks were successfully installed
1383 1384
 *          FALSE on failure
 */
1385
static BOOL OLEMenu_UnInstallHooks( DWORD tid )
1386 1387
{
  OleMenuHookItem *pHookItem = NULL;
1388
  OleMenuHookItem **ppHook = &hook_list;
1389

1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400
  while (*ppHook)
  {
      if ((*ppHook)->tid == tid)
      {
          pHookItem = *ppHook;
          *ppHook = pHookItem->next;
          break;
      }
      ppHook = &(*ppHook)->next;
  }
  if (!pHookItem) return FALSE;
1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414

  /* Uninstall the hooks installed for this thread */
  if ( !UnhookWindowsHookEx( pHookItem->GetMsg_hHook ) )
    goto CLEANUP;
  if ( !UnhookWindowsHookEx( pHookItem->CallWndProc_hHook ) )
    goto CLEANUP;

  /* Release the hook table entry */
  HeapFree(pHookItem->hHeap, 0, pHookItem );

  return TRUE;

CLEANUP:
  /* Release the hook table entry */
1415
  HeapFree(pHookItem->hHeap, 0, pHookItem );
1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426

  return FALSE;
}

/*************************************************************************
 * OLEMenu_IsHookInstalled
 * Tests if OLEMenu hooks have been installed for a thread
 *
 * RETURNS: The pointer and index of the hook table entry for the tid
 *          NULL and -1 for the index if no hooks were installed for this thread
 */
1427
static OleMenuHookItem * OLEMenu_IsHookInstalled( DWORD tid )
1428
{
1429
  OleMenuHookItem *pHookItem;
1430 1431 1432

  /* Do a simple linear search for an entry whose tid matches ours.
   * We really need a map but efficiency is not a concern here. */
1433
  for (pHookItem = hook_list; pHookItem; pHookItem = pHookItem->next)
1434 1435 1436 1437
  {
    if ( tid == pHookItem->tid )
      return pHookItem;
  }
1438

1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451
  return NULL;
}

/***********************************************************************
 *           OLEMenu_FindMainMenuIndex
 *
 * Used by OLEMenu API to find the top level group a menu item belongs to.
 * On success pnPos contains the index of the item in the top level menu group
 *
 * RETURNS: TRUE if the ID was found, FALSE on failure
 */
static BOOL OLEMenu_FindMainMenuIndex( HMENU hMainMenu, HMENU hPopupMenu, UINT *pnPos )
{
1452
  INT i, nItems;
1453 1454 1455 1456 1457 1458

  nItems = GetMenuItemCount( hMainMenu );

  for (i = 0; i < nItems; i++)
  {
    HMENU hsubmenu;
1459

1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490
    /*  Is the current item a submenu? */
    if ( (hsubmenu = GetSubMenu(hMainMenu, i)) )
    {
      /* If the handle is the same we're done */
      if ( hsubmenu == hPopupMenu )
      {
        if (pnPos)
          *pnPos = i;
        return TRUE;
      }
      /* Recursively search without updating pnPos */
      else if ( OLEMenu_FindMainMenuIndex( hsubmenu, hPopupMenu, NULL ) )
      {
        if (pnPos)
          *pnPos = i;
        return TRUE;
      }
    }
  }

  return FALSE;
}

/***********************************************************************
 *           OLEMenu_SetIsServerMenu
 *
 * Checks whether a popup menu belongs to a shared menu group which is
 * owned by the server, and sets the menu descriptor state accordingly.
 * All menu messages from these groups should be routed to the server.
 *
 * RETURNS: TRUE if the popup menu is part of a server owned group
1491
 *          FALSE if the popup menu is part of a container owned group
1492
 */
1493
static BOOL OLEMenu_SetIsServerMenu( HMENU hmenu, OleMenuDescriptor *pOleMenuDescriptor )
1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505
{
  UINT nPos = 0, nWidth, i;

  pOleMenuDescriptor->bIsServerItem = FALSE;

  /* Don't bother searching if the popup is the combined menu itself */
  if ( hmenu == pOleMenuDescriptor->hmenuCombined )
    return FALSE;

  /* Find the menu item index in the shared OLE menu that this item belongs to */
  if ( !OLEMenu_FindMainMenuIndex( pOleMenuDescriptor->hmenuCombined, hmenu,  &nPos ) )
    return FALSE;
1506

1507 1508 1509 1510
  /* The group widths array has counts for the number of elements
   * in the groups File, Edit, Container, Object, Window, Help.
   * The Edit, Object & Help groups belong to the server object
   * and the other three belong to the container.
1511
   * Loop through the group widths and locate the group we are a member of.
1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531
   */
  for ( i = 0, nWidth = 0; i < 6; i++ )
  {
    nWidth += pOleMenuDescriptor->mgw.width[i];
    if ( nPos < nWidth )
    {
      /* Odd elements are server menu widths */
      pOleMenuDescriptor->bIsServerItem = (i%2) ? TRUE : FALSE;
      break;
    }
  }

  return pOleMenuDescriptor->bIsServerItem;
}

/*************************************************************************
 * OLEMenu_CallWndProc
 * Thread scope WH_CALLWNDPROC hook proc filter function (callback)
 * This is invoked from a message hook installed in OleSetMenuDescriptor.
 */
1532
static LRESULT CALLBACK OLEMenu_CallWndProc(INT code, WPARAM wParam, LPARAM lParam)
1533
{
1534
  LPCWPSTRUCT pMsg;
1535 1536 1537 1538
  HOLEMENU hOleMenu = 0;
  OleMenuDescriptor *pOleMenuDescriptor = NULL;
  OleMenuHookItem *pHookItem = NULL;
  WORD fuFlags;
1539

1540
  TRACE("%i, %04lx, %08lx\n", code, wParam, lParam );
1541 1542 1543 1544

  /* Check if we're being asked to process the message */
  if ( HC_ACTION != code )
    goto NEXTHOOK;
1545

1546 1547 1548 1549 1550 1551 1552
  /* Retrieve the current message being dispatched from lParam */
  pMsg = (LPCWPSTRUCT)lParam;

  /* Check if the message is destined for a window we are interested in:
   * If the window has an OLEMenu property we may need to dispatch
   * the menu message to its active objects window instead. */

1553
  hOleMenu = GetPropW( pMsg->hwnd, prop_olemenuW );
1554 1555 1556 1557
  if ( !hOleMenu )
    goto NEXTHOOK;

  /* Get the menu descriptor */
1558
  pOleMenuDescriptor = GlobalLock( hOleMenu );
1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570
  if ( !pOleMenuDescriptor ) /* Bad descriptor! */
    goto NEXTHOOK;

  /* Process menu messages */
  switch( pMsg->message )
  {
    case WM_INITMENU:
    {
      /* Reset the menu descriptor state */
      pOleMenuDescriptor->bIsServerItem = FALSE;

      /* Send this message to the server as well */
1571
      SendMessageW( pOleMenuDescriptor->hwndActiveObject,
1572 1573 1574
                  pMsg->message, pMsg->wParam, pMsg->lParam );
      goto NEXTHOOK;
    }
1575

1576 1577 1578 1579 1580 1581
    case WM_INITMENUPOPUP:
    {
      /* Save the state for whether this is a server owned menu */
      OLEMenu_SetIsServerMenu( (HMENU)pMsg->wParam, pOleMenuDescriptor );
      break;
    }
1582

1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594
    case WM_MENUSELECT:
    {
      fuFlags = HIWORD(pMsg->wParam);  /* Get flags */
      if ( fuFlags & MF_SYSMENU )
         goto NEXTHOOK;

      /* Save the state for whether this is a server owned popup menu */
      else if ( fuFlags & MF_POPUP )
        OLEMenu_SetIsServerMenu( (HMENU)pMsg->lParam, pOleMenuDescriptor );

      break;
    }
1595

1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611
    case WM_DRAWITEM:
    {
      LPDRAWITEMSTRUCT lpdis = (LPDRAWITEMSTRUCT) pMsg->lParam;
      if ( pMsg->wParam != 0 || lpdis->CtlType != ODT_MENU )
        goto NEXTHOOK;  /* Not a menu message */

      break;
    }

    default:
      goto NEXTHOOK;
  }

  /* If the message was for the server dispatch it accordingly */
  if ( pOleMenuDescriptor->bIsServerItem )
  {
1612
    SendMessageW( pOleMenuDescriptor->hwndActiveObject,
1613 1614
                  pMsg->message, pMsg->wParam, pMsg->lParam );
  }
1615

1616 1617 1618
NEXTHOOK:
  if ( pOleMenuDescriptor )
    GlobalUnlock( hOleMenu );
1619

1620
  /* Lookup the hook item for the current thread */
1621
  if ( !( pHookItem = OLEMenu_IsHookInstalled( GetCurrentThreadId() ) ) )
1622 1623
  {
    /* This should never fail!! */
1624
    WARN("could not retrieve hHook for current thread!\n" );
1625 1626
    return 0;
  }
1627

1628 1629 1630 1631 1632 1633 1634 1635 1636
  /* Pass on the message to the next hooker */
  return CallNextHookEx( pHookItem->CallWndProc_hHook, code, wParam, lParam );
}

/*************************************************************************
 * OLEMenu_GetMsgProc
 * Thread scope WH_GETMESSAGE hook proc filter function (callback)
 * This is invoked from a message hook installed in OleSetMenuDescriptor.
 */
1637
static LRESULT CALLBACK OLEMenu_GetMsgProc(INT code, WPARAM wParam, LPARAM lParam)
1638
{
1639
  LPMSG pMsg;
1640 1641 1642 1643
  HOLEMENU hOleMenu = 0;
  OleMenuDescriptor *pOleMenuDescriptor = NULL;
  OleMenuHookItem *pHookItem = NULL;
  WORD wCode;
1644

1645
  TRACE("%i, %04lx, %08lx\n", code, wParam, lParam );
1646 1647 1648 1649

  /* Check if we're being asked to process a  messages */
  if ( HC_ACTION != code )
    goto NEXTHOOK;
1650

1651 1652 1653 1654 1655 1656 1657
  /* Retrieve the current message being dispatched from lParam */
  pMsg = (LPMSG)lParam;

  /* Check if the message is destined for a window we are interested in:
   * If the window has an OLEMenu property we may need to dispatch
   * the menu message to its active objects window instead. */

1658
  hOleMenu = GetPropW( pMsg->hwnd, prop_olemenuW );
1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676
  if ( !hOleMenu )
    goto NEXTHOOK;

  /* Process menu messages */
  switch( pMsg->message )
  {
    case WM_COMMAND:
    {
      wCode = HIWORD(pMsg->wParam);  /* Get notification code */
      if ( wCode )
        goto NEXTHOOK;  /* Not a menu message */
      break;
    }
    default:
      goto NEXTHOOK;
  }

  /* Get the menu descriptor */
1677
  pOleMenuDescriptor = GlobalLock( hOleMenu );
1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688
  if ( !pOleMenuDescriptor ) /* Bad descriptor! */
    goto NEXTHOOK;

  /* If the message was for the server dispatch it accordingly */
  if ( pOleMenuDescriptor->bIsServerItem )
  {
    /* Change the hWnd in the message to the active objects hWnd.
     * The message loop which reads this message will automatically
     * dispatch it to the embedded objects window. */
    pMsg->hwnd = pOleMenuDescriptor->hwndActiveObject;
  }
1689

1690 1691 1692
NEXTHOOK:
  if ( pOleMenuDescriptor )
    GlobalUnlock( hOleMenu );
1693

1694
  /* Lookup the hook item for the current thread */
1695
  if ( !( pHookItem = OLEMenu_IsHookInstalled( GetCurrentThreadId() ) ) )
1696 1697
  {
    /* This should never fail!! */
1698
    WARN("could not retrieve hHook for current thread!\n" );
1699 1700
    return FALSE;
  }
1701

1702 1703 1704 1705
  /* Pass on the message to the next hooker */
  return CallNextHookEx( pHookItem->GetMsg_hHook, code, wParam, lParam );
}

1706
/***********************************************************************
1707
 * OleCreateMenuDescriptor [OLE32.@]
1708 1709 1710 1711 1712 1713 1714
 * Creates an OLE menu descriptor for OLE to use when dispatching
 * menu messages and commands.
 *
 * PARAMS:
 *    hmenuCombined  -  Handle to the objects combined menu
 *    lpMenuWidths   -  Pointer to array of 6 LONG's indicating menus per group
 *
1715
 */
1716 1717 1718
HOLEMENU WINAPI OleCreateMenuDescriptor(
  HMENU                hmenuCombined,
  LPOLEMENUGROUPWIDTHS lpMenuWidths)
1719
{
1720 1721 1722 1723 1724 1725
  HOLEMENU hOleMenu;
  OleMenuDescriptor *pOleMenuDescriptor;
  int i;

  if ( !hmenuCombined || !lpMenuWidths )
    return 0;
1726

1727 1728 1729
  /* Create an OLE menu descriptor */
  if ( !(hOleMenu = GlobalAlloc(GMEM_MOVEABLE | GMEM_ZEROINIT,
                                sizeof(OleMenuDescriptor) ) ) )
1730
  return 0;
1731

1732
  pOleMenuDescriptor = GlobalLock( hOleMenu );
1733 1734 1735 1736 1737 1738
  if ( !pOleMenuDescriptor )
    return 0;

  /* Initialize menu group widths and hmenu */
  for ( i = 0; i < 6; i++ )
    pOleMenuDescriptor->mgw.width[i] = lpMenuWidths->width[i];
1739

1740 1741 1742
  pOleMenuDescriptor->hmenuCombined = hmenuCombined;
  pOleMenuDescriptor->bIsServerItem = FALSE;
  GlobalUnlock( hOleMenu );
1743

1744
  return hOleMenu;
1745 1746 1747
}

/***********************************************************************
1748
 * OleDestroyMenuDescriptor [OLE32.@]
1749
 * Destroy the shared menu descriptor
1750
 */
1751
HRESULT WINAPI OleDestroyMenuDescriptor(
1752
  HOLEMENU hmenuDescriptor)
1753
{
1754 1755
  if ( hmenuDescriptor )
    GlobalFree( hmenuDescriptor );
1756
	return S_OK;
1757 1758 1759
}

/***********************************************************************
1760
 * OleSetMenuDescriptor [OLE32.@]
1761
 * Installs or removes OLE dispatching code for the containers frame window.
1762
 *
1763
 * PARAMS
1764 1765 1766 1767 1768 1769
 *     hOleMenu         Handle to composite menu descriptor
 *     hwndFrame        Handle to containers frame window
 *     hwndActiveObject Handle to objects in-place activation window
 *     lpFrame          Pointer to IOleInPlaceFrame on containers window
 *     lpActiveObject   Pointer to IOleInPlaceActiveObject on active in-place object
 *
1770
 * RETURNS
1771 1772
 *      S_OK                               - menu installed correctly
 *      E_FAIL, E_INVALIDARG, E_UNEXPECTED - failure
1773 1774 1775 1776 1777
 *
 * FIXME
 *      The lpFrame and lpActiveObject parameters are currently ignored
 *      OLE should install context sensitive help F1 filtering for the app when
 *      these are non null.
1778 1779
 */
HRESULT WINAPI OleSetMenuDescriptor(
1780
  HOLEMENU               hOleMenu,
1781 1782
  HWND                   hwndFrame,
  HWND                   hwndActiveObject,
1783 1784 1785
  LPOLEINPLACEFRAME        lpFrame,
  LPOLEINPLACEACTIVEOBJECT lpActiveObject)
{
1786 1787 1788 1789 1790 1791 1792 1793
  OleMenuDescriptor *pOleMenuDescriptor = NULL;

  /* Check args */
  if ( !hwndFrame || (hOleMenu && !hwndActiveObject) )
    return E_INVALIDARG;

  if ( lpFrame || lpActiveObject )
  {
1794 1795
     FIXME("(%p, %p, %p, %p, %p), Context sensitive help filtering not implemented!\n",
	hOleMenu,
1796 1797 1798 1799
	hwndFrame,
	hwndActiveObject,
	lpFrame,
	lpActiveObject);
1800 1801 1802 1803 1804 1805
  }

  /* Set up a message hook to intercept the containers frame window messages.
   * The message filter is responsible for dispatching menu messages from the
   * shared menu which are intended for the object.
   */
1806

1807 1808 1809 1810 1811 1812
  if ( hOleMenu )  /* Want to install dispatching code */
  {
    /* If OLEMenu hooks are already installed for this thread, fail
     * Note: This effectively means that OleSetMenuDescriptor cannot
     * be called twice in succession on the same frame window
     * without first calling it with a null hOleMenu to uninstall */
1813
    if ( OLEMenu_IsHookInstalled( GetCurrentThreadId() ) )
1814
  return E_FAIL;
1815

1816
    /* Get the menu descriptor */
1817
    pOleMenuDescriptor = GlobalLock( hOleMenu );
1818 1819 1820 1821 1822 1823 1824 1825 1826
    if ( !pOleMenuDescriptor )
      return E_UNEXPECTED;

    /* Update the menu descriptor */
    pOleMenuDescriptor->hwndFrame = hwndFrame;
    pOleMenuDescriptor->hwndActiveObject = hwndActiveObject;

    GlobalUnlock( hOleMenu );
    pOleMenuDescriptor = NULL;
1827

1828
    /* Add a menu descriptor windows property to the frame window */
1829
    SetPropW( hwndFrame, prop_olemenuW, hOleMenu );
1830 1831 1832 1833 1834 1835 1836 1837 1838 1839

    /* Install thread scope message hooks for WH_GETMESSAGE and WH_CALLWNDPROC */
    if ( !OLEMenu_InstallHooks( GetCurrentThreadId() ) )
      return E_FAIL;
  }
  else  /* Want to uninstall dispatching code */
  {
    /* Uninstall the hooks */
    if ( !OLEMenu_UnInstallHooks( GetCurrentThreadId() ) )
      return E_FAIL;
1840

1841
    /* Remove the menu descriptor property from the frame window */
1842
    RemovePropW( hwndFrame, prop_olemenuW );
1843
  }
1844

1845
  return S_OK;
1846 1847
}

1848
/******************************************************************************
1849
 *              IsAccelerator        [OLE32.@]
1850 1851 1852 1853
 * Mostly copied from controls/menu.c TranslateAccelerator implementation
 */
BOOL WINAPI IsAccelerator(HACCEL hAccel, int cAccelEntries, LPMSG lpMsg, WORD* lpwCmd)
{
1854
    LPACCEL lpAccelTbl;
1855 1856 1857
    int i;

    if(!lpMsg) return FALSE;
1858
    if (!hAccel)
1859
    {
1860
	WARN_(accel)("NULL accel handle\n");
1861 1862 1863 1864
	return FALSE;
    }
    if((lpMsg->message != WM_KEYDOWN &&
	lpMsg->message != WM_SYSKEYDOWN &&
1865
	lpMsg->message != WM_SYSCHAR &&
1866
	lpMsg->message != WM_CHAR)) return FALSE;
1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877
    lpAccelTbl = HeapAlloc(GetProcessHeap(), 0, cAccelEntries * sizeof(ACCEL));
    if (NULL == lpAccelTbl)
    {
	return FALSE;
    }
    if (CopyAcceleratorTableW(hAccel, lpAccelTbl, cAccelEntries) != cAccelEntries)
    {
	WARN_(accel)("CopyAcceleratorTableW failed\n");
	HeapFree(GetProcessHeap(), 0, lpAccelTbl);
	return FALSE;
    }
1878

1879
    TRACE_(accel)("hAccel=%p, cAccelEntries=%d,"
1880
		"msg->hwnd=%p, msg->message=%04x, wParam=%08lx, lParam=%08lx\n",
1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891
		hAccel, cAccelEntries,
		lpMsg->hwnd, lpMsg->message, lpMsg->wParam, lpMsg->lParam);
    for(i = 0; i < cAccelEntries; i++)
    {
	if(lpAccelTbl[i].key != lpMsg->wParam)
	    continue;

	if(lpMsg->message == WM_CHAR)
	{
	    if(!(lpAccelTbl[i].fVirt & FALT) && !(lpAccelTbl[i].fVirt & FVIRTKEY))
	    {
1892
		TRACE_(accel)("found accel for WM_CHAR: ('%c')\n", LOWORD(lpMsg->wParam) & 0xff);
1893 1894 1895 1896 1897 1898 1899 1900
		goto found;
	    }
	}
	else
	{
	    if(lpAccelTbl[i].fVirt & FVIRTKEY)
	    {
		INT mask = 0;
1901
		TRACE_(accel)("found accel for virt_key %04lx (scan %04x)\n",
1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914
				lpMsg->wParam, HIWORD(lpMsg->lParam) & 0xff);
		if(GetKeyState(VK_SHIFT) & 0x8000) mask |= FSHIFT;
		if(GetKeyState(VK_CONTROL) & 0x8000) mask |= FCONTROL;
		if(GetKeyState(VK_MENU) & 0x8000) mask |= FALT;
		if(mask == (lpAccelTbl[i].fVirt & (FSHIFT | FCONTROL | FALT))) goto found;
		TRACE_(accel)("incorrect SHIFT/CTRL/ALT-state\n");
	    }
	    else
	    {
		if(!(lpMsg->lParam & 0x01000000))  /* no special_key */
		{
		    if((lpAccelTbl[i].fVirt & FALT) && (lpMsg->lParam & 0x20000000))
		    {						       /* ^^ ALT pressed */
1915
			TRACE_(accel)("found accel for Alt-%c\n", LOWORD(lpMsg->wParam) & 0xff);
1916 1917 1918 1919 1920
			goto found;
		    }
		}
	    }
	}
1921
    }
1922 1923

    WARN_(accel)("couldn't translate accelerator key\n");
1924
    HeapFree(GetProcessHeap(), 0, lpAccelTbl);
1925 1926 1927 1928
    return FALSE;

found:
    if(lpwCmd) *lpwCmd = lpAccelTbl[i].cmd;
1929
    HeapFree(GetProcessHeap(), 0, lpAccelTbl);
1930 1931 1932
    return TRUE;
}

1933
/***********************************************************************
1934
 * ReleaseStgMedium [OLE32.@]
1935 1936 1937 1938 1939 1940 1941 1942
 */
void WINAPI ReleaseStgMedium(
  STGMEDIUM* pmedium)
{
  switch (pmedium->tymed)
  {
    case TYMED_HGLOBAL:
    {
1943
      if ( (pmedium->pUnkForRelease==0) &&
1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955
	   (pmedium->u.hGlobal!=0) )
	GlobalFree(pmedium->u.hGlobal);
      break;
    }
    case TYMED_FILE:
    {
      if (pmedium->u.lpszFileName!=0)
      {
	if (pmedium->pUnkForRelease==0)
	{
	  DeleteFileW(pmedium->u.lpszFileName);
	}
1956

1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978
	CoTaskMemFree(pmedium->u.lpszFileName);
      }
      break;
    }
    case TYMED_ISTREAM:
    {
      if (pmedium->u.pstm!=0)
      {
	IStream_Release(pmedium->u.pstm);
      }
      break;
    }
    case TYMED_ISTORAGE:
    {
      if (pmedium->u.pstg!=0)
      {
	IStorage_Release(pmedium->u.pstg);
      }
      break;
    }
    case TYMED_GDI:
    {
1979
      if ( (pmedium->pUnkForRelease==0) &&
1980 1981
	   (pmedium->u.hBitmap!=0) )
	DeleteObject(pmedium->u.hBitmap);
1982 1983 1984 1985
      break;
    }
    case TYMED_MFPICT:
    {
1986
      if ( (pmedium->pUnkForRelease==0) &&
1987 1988
	   (pmedium->u.hMetaFilePict!=0) )
      {
1989
	LPMETAFILEPICT pMP = GlobalLock(pmedium->u.hMetaFilePict);
1990
	DeleteMetaFile(pMP->hMF);
1991 1992
	GlobalUnlock(pmedium->u.hMetaFilePict);
	GlobalFree(pmedium->u.hMetaFilePict);
1993 1994 1995 1996 1997
      }
      break;
    }
    case TYMED_ENHMF:
    {
1998
      if ( (pmedium->pUnkForRelease==0) &&
1999 2000 2001 2002 2003 2004 2005 2006 2007 2008
	   (pmedium->u.hEnhMetaFile!=0) )
      {
	DeleteEnhMetaFile(pmedium->u.hEnhMetaFile);
      }
      break;
    }
    case TYMED_NULL:
    default:
      break;
  }
2009
  pmedium->tymed=TYMED_NULL;
2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020

  /*
   * After cleaning up, the unknown is released
   */
  if (pmedium->pUnkForRelease!=0)
  {
    IUnknown_Release(pmedium->pUnkForRelease);
    pmedium->pUnkForRelease = 0;
  }
}

2021 2022 2023 2024 2025
/***
 * OLEDD_Initialize()
 *
 * Initializes the OLE drag and drop data structures.
 */
2026
static void OLEDD_Initialize(void)
2027
{
2028
    WNDCLASSW wndClass;
2029

2030
    ZeroMemory (&wndClass, sizeof(WNDCLASSW));
2031
    wndClass.style         = CS_GLOBALCLASS;
2032
    wndClass.lpfnWndProc   = OLEDD_DragTrackerWindowProc;
2033 2034 2035 2036 2037
    wndClass.cbClsExtra    = 0;
    wndClass.cbWndExtra    = sizeof(TrackerWindowInfo*);
    wndClass.hCursor       = 0;
    wndClass.hbrBackground = 0;
    wndClass.lpszClassName = OLEDD_DRAGTRACKERCLASS;
2038

2039
    RegisterClassW (&wndClass);
2040 2041 2042 2043 2044 2045 2046 2047 2048 2049
}

/***
 * OLEDD_DragTrackerWindowProc()
 *
 * This method is the WindowProcedure of the drag n drop tracking
 * window. During a drag n Drop operation, an invisible window is created
 * to receive the user input and act upon it. This procedure is in charge
 * of this behavior.
 */
2050 2051 2052

#define DRAG_TIMER_ID 1

2053
static LRESULT WINAPI OLEDD_DragTrackerWindowProc(
2054
			 HWND   hwnd,
2055
			 UINT   uMsg,
2056
			 WPARAM wParam,
2057 2058 2059 2060 2061 2062
			 LPARAM   lParam)
{
  switch (uMsg)
  {
    case WM_CREATE:
    {
2063
      LPCREATESTRUCTA createStruct = (LPCREATESTRUCTA)lParam;
2064

2065
      SetWindowLongPtrW(hwnd, 0, (LONG_PTR)createStruct->lpCreateParams);
2066
      SetTimer(hwnd, DRAG_TIMER_ID, 50, NULL);
2067 2068 2069

      break;
    }
2070
    case WM_TIMER:
2071 2072
    case WM_MOUSEMOVE:
    {
2073
      OLEDD_TrackMouseMove((TrackerWindowInfo*)GetWindowLongPtrA(hwnd, 0));
2074 2075 2076 2077 2078 2079 2080 2081 2082
      break;
    }
    case WM_LBUTTONUP:
    case WM_MBUTTONUP:
    case WM_RBUTTONUP:
    case WM_LBUTTONDOWN:
    case WM_MBUTTONDOWN:
    case WM_RBUTTONDOWN:
    {
2083
      OLEDD_TrackStateChange((TrackerWindowInfo*)GetWindowLongPtrA(hwnd, 0));
2084 2085 2086 2087 2088
      break;
    }
    case WM_DESTROY:
    {
      KillTimer(hwnd, DRAG_TIMER_ID);
2089 2090 2091 2092 2093 2094 2095
      break;
    }
  }

  /*
   * This is a window proc after all. Let's call the default.
   */
2096
  return DefWindowProcW (hwnd, uMsg, wParam, lParam);
2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111
}

/***
 * OLEDD_TrackMouseMove()
 *
 * This method is invoked while a drag and drop operation is in effect.
 * it will generate the appropriate callbacks in the drop source
 * and drop target. It will also provide the expected feedback to
 * the user.
 *
 * params:
 *    trackerInfo - Pointer to the structure identifying the
 *                  drag & drop operation that is currently
 *                  active.
 */
2112
static void OLEDD_TrackMouseMove(TrackerWindowInfo* trackerInfo)
2113
{
2114
  HWND   hwndNewTarget = 0;
2115
  HRESULT  hr = S_OK;
2116
  POINT pt;
2117 2118 2119 2120

  /*
   * Get the handle of the window under the mouse
   */
2121 2122 2123
  pt.x = trackerInfo->curMousePos.x;
  pt.y = trackerInfo->curMousePos.y;
  hwndNewTarget = WindowFromPoint(pt);
2124

2125 2126 2127 2128 2129 2130
  /*
   * Every time, we re-initialize the effects passed to the
   * IDropTarget to the effects allowed by the source.
   */
  *trackerInfo->pdwEffect = trackerInfo->dwOKEffect;

2131 2132 2133 2134
  /*
   * If we are hovering over the same target as before, send the
   * DragOver notification
   */
2135
  if ( (trackerInfo->curDragTarget != 0) &&
2136
       (trackerInfo->curTargetHWND == hwndNewTarget) )
2137 2138
  {
    IDropTarget_DragOver(trackerInfo->curDragTarget,
2139 2140
                         trackerInfo->dwKeyState,
                         trackerInfo->curMousePos,
2141 2142 2143
			 trackerInfo->pdwEffect);
  }
  else
2144
  {
2145 2146 2147 2148
    /*
     * If we changed window, we have to notify our old target and check for
     * the new one.
     */
2149
    if (trackerInfo->curDragTarget)
2150
      IDropTarget_DragLeave(trackerInfo->curDragTarget);
2151

2152 2153 2154
    /*
     * Make sure we're hovering over a window.
     */
2155
    if (hwndNewTarget)
2156 2157 2158 2159
    {
      /*
       * Find-out if there is a drag target under the mouse
       */
2160 2161
      HWND next_target_wnd = hwndNewTarget;

2162 2163
      trackerInfo->curTargetHWND = hwndNewTarget;

2164 2165
      while (next_target_wnd && !is_droptarget(next_target_wnd))
          next_target_wnd = GetParent(next_target_wnd);
2166

2167 2168
      if (next_target_wnd) hwndNewTarget = next_target_wnd;

2169 2170 2171
      trackerInfo->curDragTargetHWND = hwndNewTarget;
      if(trackerInfo->curDragTarget) IDropTarget_Release(trackerInfo->curDragTarget);
      trackerInfo->curDragTarget     = get_droptarget_pointer(hwndNewTarget);
2172

2173 2174 2175
      /*
       * If there is, notify it that we just dragged-in
       */
2176
      if (trackerInfo->curDragTarget)
2177
      {
2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188
        hr = IDropTarget_DragEnter(trackerInfo->curDragTarget,
                                   trackerInfo->dataObject,
                                   trackerInfo->dwKeyState,
                                   trackerInfo->curMousePos,
                                   trackerInfo->pdwEffect);

        /* failed DragEnter() means invalid target */
        if (hr != S_OK)
        {
          trackerInfo->curDragTargetHWND = 0;
          trackerInfo->curTargetHWND     = 0;
2189
          IDropTarget_Release(trackerInfo->curDragTarget);
2190 2191
          trackerInfo->curDragTarget     = 0;
        }
2192 2193 2194 2195 2196 2197 2198 2199
      }
    }
    else
    {
      /*
       * The mouse is not over a window so we don't track anything.
       */
      trackerInfo->curDragTargetHWND = 0;
2200
      trackerInfo->curTargetHWND     = 0;
2201
      if(trackerInfo->curDragTarget) IDropTarget_Release(trackerInfo->curDragTarget);
2202 2203 2204 2205 2206
      trackerInfo->curDragTarget     = 0;
    }
  }

  /*
2207 2208
   * Now that we have done that, we have to tell the source to give
   * us feedback on the work being done by the target.  If we don't
2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224
   * have a target, simulate no effect.
   */
  if (trackerInfo->curDragTarget==0)
  {
    *trackerInfo->pdwEffect = DROPEFFECT_NONE;
  }

  hr = IDropSource_GiveFeedback(trackerInfo->dropSource,
  				*trackerInfo->pdwEffect);

  /*
   * When we ask for feedback from the drop source, sometimes it will
   * do all the necessary work and sometimes it will not handle it
   * when that's the case, we must display the standard drag and drop
   * cursors.
   */
2225
  if (hr == DRAGDROP_S_USEDEFAULTCURSORS)
2226
  {
2227 2228
    HCURSOR hCur;

2229
    if (*trackerInfo->pdwEffect & DROPEFFECT_MOVE)
2230
    {
2231
      hCur = LoadCursorW(hProxyDll, MAKEINTRESOURCEW(2));
2232 2233 2234
    }
    else if (*trackerInfo->pdwEffect & DROPEFFECT_COPY)
    {
2235
      hCur = LoadCursorW(hProxyDll, MAKEINTRESOURCEW(3));
2236 2237 2238
    }
    else if (*trackerInfo->pdwEffect & DROPEFFECT_LINK)
    {
2239
      hCur = LoadCursorW(hProxyDll, MAKEINTRESOURCEW(4));
2240 2241 2242
    }
    else
    {
2243
      hCur = LoadCursorW(hProxyDll, MAKEINTRESOURCEW(1));
2244
    }
2245 2246

    SetCursor(hCur);
2247
  }
2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261
}

/***
 * OLEDD_TrackStateChange()
 *
 * This method is invoked while a drag and drop operation is in effect.
 * It is used to notify the drop target/drop source callbacks when
 * the state of the keyboard or mouse button change.
 *
 * params:
 *    trackerInfo - Pointer to the structure identifying the
 *                  drag & drop operation that is currently
 *                  active.
 */
2262
static void OLEDD_TrackStateChange(TrackerWindowInfo* trackerInfo)
2263 2264 2265 2266 2267 2268
{
  /*
   * Ask the drop source what to do with the operation.
   */
  trackerInfo->returnValue = IDropSource_QueryContinueDrag(
			       trackerInfo->dropSource,
2269
			       trackerInfo->escPressed,
2270
                               trackerInfo->dwKeyState);
2271

2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283
  /*
   * All the return valued will stop the operation except the S_OK
   * return value.
   */
  if (trackerInfo->returnValue!=S_OK)
  {
    /*
     * Make sure the message loop in DoDragDrop stops
     */
    trackerInfo->trackingDone = TRUE;

    /*
2284
     * Release the mouse in case the drop target decides to show a popup
2285 2286 2287
     * or a menu or something.
     */
    ReleaseCapture();
2288

2289
    /*
2290
     * If we end-up over a target, drop the object in the target or
2291 2292
     * inform the target that the operation was cancelled.
     */
2293
    if (trackerInfo->curDragTarget)
2294 2295 2296 2297
    {
      switch (trackerInfo->returnValue)
      {
	/*
2298
	 * If the source wants us to complete the operation, we tell
2299 2300 2301
	 * the drop target that we just dropped the object in it.
	 */
        case DRAGDROP_S_DROP:
2302 2303 2304 2305 2306 2307 2308 2309 2310 2311
          if (*trackerInfo->pdwEffect != DROPEFFECT_NONE)
            IDropTarget_Drop(trackerInfo->curDragTarget,
                             trackerInfo->dataObject,
                             trackerInfo->dwKeyState,
                             trackerInfo->curMousePos,
                             trackerInfo->pdwEffect);
          else
            IDropTarget_DragLeave(trackerInfo->curDragTarget);
          break;

2312
	/*
2313
	 * If the source told us that we should cancel, fool the drop
2314
	 * target by telling it that the mouse left it's window.
2315
	 * Also set the drop effect to "NONE" in case the application
2316
	 * ignores the result of DoDragDrop.
2317 2318 2319
	 */
        case DRAGDROP_S_CANCEL:
	  IDropTarget_DragLeave(trackerInfo->curDragTarget);
2320
	  *trackerInfo->pdwEffect = DROPEFFECT_NONE;
2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333
	  break;
      }
    }
  }
}

/***
 * OLEDD_GetButtonState()
 *
 * This method will use the current state of the keyboard to build
 * a button state mask equivalent to the one passed in the
 * WM_MOUSEMOVE wParam.
 */
2334
static DWORD OLEDD_GetButtonState(void)
2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358
{
  BYTE  keyboardState[256];
  DWORD keyMask = 0;

  GetKeyboardState(keyboardState);

  if ( (keyboardState[VK_SHIFT] & 0x80) !=0)
    keyMask |= MK_SHIFT;

  if ( (keyboardState[VK_CONTROL] & 0x80) !=0)
    keyMask |= MK_CONTROL;

  if ( (keyboardState[VK_LBUTTON] & 0x80) !=0)
    keyMask |= MK_LBUTTON;

  if ( (keyboardState[VK_RBUTTON] & 0x80) !=0)
    keyMask |= MK_RBUTTON;

  if ( (keyboardState[VK_MBUTTON] & 0x80) !=0)
    keyMask |= MK_MBUTTON;

  return keyMask;
}

2359 2360 2361 2362 2363 2364 2365 2366 2367
/***
 * OLEDD_GetButtonState()
 *
 * This method will read the default value of the registry key in
 * parameter and extract a DWORD value from it. The registry key value
 * can be in a string key or a DWORD key.
 *
 * params:
 *     regKey   - Key to read the default value from
2368
 *     pdwValue - Pointer to the location where the DWORD
2369 2370 2371 2372 2373
 *                value is returned. This value is not modified
 *                if the value is not found.
 */

static void OLEUTL_ReadRegistryDWORDValue(
2374
  HKEY   regKey,
2375 2376
  DWORD* pdwValue)
{
2377
  WCHAR buffer[20];
2378
  DWORD cbData = sizeof(buffer);
2379 2380 2381
  DWORD dwKeyType;
  LONG  lres;

2382 2383
  lres = RegQueryValueExW(regKey,
			  emptyW,
2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398
			  NULL,
			  &dwKeyType,
			  (LPBYTE)buffer,
			  &cbData);

  if (lres==ERROR_SUCCESS)
  {
    switch (dwKeyType)
    {
      case REG_DWORD:
	*pdwValue = *(DWORD*)buffer;
	break;
      case REG_EXPAND_SZ:
      case REG_MULTI_SZ:
      case REG_SZ:
2399
	*pdwValue = (DWORD)strtoulW(buffer, NULL, 10);
2400 2401 2402 2403
	break;
    }
  }
}
2404

Troy Rollo's avatar
Troy Rollo committed
2405
/******************************************************************************
2406
 * OleDraw (OLE32.@)
Troy Rollo's avatar
Troy Rollo committed
2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422
 *
 * The operation of this function is documented literally in the WinAPI
 * documentation to involve a QueryInterface for the IViewObject interface,
 * followed by a call to IViewObject::Draw.
 */
HRESULT WINAPI OleDraw(
	IUnknown *pUnk,
	DWORD dwAspect,
	HDC hdcDraw,
	LPCRECT lprcBounds)
{
  HRESULT hres;
  IViewObject *viewobject;

  hres = IUnknown_QueryInterface(pUnk,
				 &IID_IViewObject,
2423
				 (void**)&viewobject);
Troy Rollo's avatar
Troy Rollo committed
2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443

  if (SUCCEEDED(hres))
  {
    RECTL rectl;

    rectl.left = lprcBounds->left;
    rectl.right = lprcBounds->right;
    rectl.top = lprcBounds->top;
    rectl.bottom = lprcBounds->bottom;
    hres = IViewObject_Draw(viewobject, dwAspect, -1, 0, 0, 0, hdcDraw, &rectl, 0, 0, 0);

    IViewObject_Release(viewobject);
    return hres;
  }
  else
  {
    return DV_E_NOIVIEWOBJECT;
  }
}

2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472
/***********************************************************************
 *             OleTranslateAccelerator [OLE32.@]
 */
HRESULT WINAPI OleTranslateAccelerator (LPOLEINPLACEFRAME lpFrame,
                   LPOLEINPLACEFRAMEINFO lpFrameInfo, LPMSG lpmsg)
{
    WORD wID;

    TRACE("(%p,%p,%p)\n", lpFrame, lpFrameInfo, lpmsg);

    if (IsAccelerator(lpFrameInfo->haccel,lpFrameInfo->cAccelEntries,lpmsg,&wID))
        return IOleInPlaceFrame_TranslateAccelerator(lpFrame,lpmsg,wID);

    return S_FALSE;
}

/******************************************************************************
 *              OleCreate        [OLE32.@]
 *
 */
HRESULT WINAPI OleCreate(
	REFCLSID rclsid,
	REFIID riid,
	DWORD renderopt,
	LPFORMATETC pFormatEtc,
	LPOLECLIENTSITE pClientSite,
	LPSTORAGE pStg,
	LPVOID* ppvObj)
{
2473
    HRESULT hres;
2474
    IUnknown * pUnk = NULL;
2475
    IOleObject *pOleObject = NULL;
2476

2477 2478
    TRACE("(%s, %s, %d, %p, %p, %p, %p)\n", debugstr_guid(rclsid),
        debugstr_guid(riid), renderopt, pFormatEtc, pClientSite, pStg, ppvObj);
2479

2480
    hres = CoCreateInstance(rclsid, 0, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER, riid, (LPVOID*)&pUnk);
2481 2482 2483 2484

    if (SUCCEEDED(hres))
        hres = IStorage_SetClass(pStg, rclsid);

2485
    if (pClientSite && SUCCEEDED(hres))
2486
    {
2487
        hres = IUnknown_QueryInterface(pUnk, &IID_IOleObject, (LPVOID*)&pOleObject);
2488 2489 2490 2491 2492 2493
        if (SUCCEEDED(hres))
        {
            DWORD dwStatus;
            hres = IOleObject_GetMiscStatus(pOleObject, DVASPECT_CONTENT, &dwStatus);
        }
    }
2494

2495
    if (SUCCEEDED(hres))
2496
    {
2497 2498
        IPersistStorage * pPS;
        if (SUCCEEDED((hres = IUnknown_QueryInterface(pUnk, &IID_IPersistStorage, (LPVOID*)&pPS))))
2499
        {
2500 2501
            TRACE("trying to set stg %p\n", pStg);
            hres = IPersistStorage_InitNew(pPS, pStg);
2502
            TRACE("-- result 0x%08x\n", hres);
2503
            IPersistStorage_Release(pPS);
2504 2505 2506
        }
    }

2507 2508 2509 2510
    if (pClientSite && SUCCEEDED(hres))
    {
        TRACE("trying to set clientsite %p\n", pClientSite);
        hres = IOleObject_SetClientSite(pOleObject, pClientSite);
2511
        TRACE("-- result 0x%08x\n", hres);
2512 2513 2514 2515
    }

    if (pOleObject)
        IOleObject_Release(pOleObject);
2516 2517 2518 2519 2520 2521

    if (((renderopt == OLERENDER_DRAW) || (renderopt == OLERENDER_FORMAT)) &&
        SUCCEEDED(hres))
    {
        IRunnableObject *pRunnable;
        IOleCache *pOleCache;
2522
        HRESULT hres2;
2523

2524 2525
        hres2 = IUnknown_QueryInterface(pUnk, &IID_IRunnableObject, (void **)&pRunnable);
        if (SUCCEEDED(hres2))
2526 2527 2528 2529 2530 2531 2532
        {
            hres = IRunnableObject_Run(pRunnable, NULL);
            IRunnableObject_Release(pRunnable);
        }

        if (SUCCEEDED(hres))
        {
2533 2534 2535 2536
            hres2 = IUnknown_QueryInterface(pUnk, &IID_IOleCache, (void **)&pOleCache);
            if (SUCCEEDED(hres2))
            {
                DWORD dwConnection;
2537
                if (renderopt == OLERENDER_DRAW && !pFormatEtc) {
2538 2539 2540 2541 2542 2543 2544 2545 2546 2547
                    FORMATETC pfe;
                    pfe.cfFormat = 0;
                    pfe.ptd = NULL;
                    pfe.dwAspect = DVASPECT_CONTENT;
                    pfe.lindex = -1;
                    pfe.tymed = TYMED_NULL;
                    hres = IOleCache_Cache(pOleCache, &pfe, ADVF_PRIMEFIRST, &dwConnection);
                }
                else
                    hres = IOleCache_Cache(pOleCache, pFormatEtc, ADVF_PRIMEFIRST, &dwConnection);
2548 2549
                IOleCache_Release(pOleCache);
            }
2550 2551 2552
        }
    }

2553
    if (FAILED(hres) && pUnk)
2554 2555 2556 2557 2558
    {
        IUnknown_Release(pUnk);
        pUnk = NULL;
    }

2559 2560
    *ppvObj = pUnk;

2561
    TRACE("-- %p\n", pUnk);
2562 2563 2564
    return hres;
}

2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591
/******************************************************************************
 *              OleGetAutoConvert        [OLE32.@]
 */
HRESULT WINAPI OleGetAutoConvert(REFCLSID clsidOld, LPCLSID pClsidNew)
{
    static const WCHAR wszAutoConvertTo[] = {'A','u','t','o','C','o','n','v','e','r','t','T','o',0};
    HKEY hkey = NULL;
    WCHAR buf[CHARS_IN_GUID];
    LONG len;
    HRESULT res = S_OK;

    res = COM_OpenKeyForCLSID(clsidOld, wszAutoConvertTo, KEY_READ, &hkey);
    if (FAILED(res))
        goto done;

    len = sizeof(buf);
    if (RegQueryValueW(hkey, NULL, buf, &len))
    {
        res = REGDB_E_KEYMISSING;
        goto done;
    }
    res = CLSIDFromString(buf, pClsidNew);
done:
    if (hkey) RegCloseKey(hkey);
    return res;
}

2592 2593 2594 2595 2596
/******************************************************************************
 *              OleSetAutoConvert        [OLE32.@]
 */
HRESULT WINAPI OleSetAutoConvert(REFCLSID clsidOld, REFCLSID clsidNew)
{
2597 2598 2599
    static const WCHAR wszAutoConvertTo[] = {'A','u','t','o','C','o','n','v','e','r','t','T','o',0};
    HKEY hkey = NULL;
    WCHAR szClsidNew[CHARS_IN_GUID];
2600 2601 2602
    HRESULT res = S_OK;

    TRACE("(%s,%s)\n", debugstr_guid(clsidOld), debugstr_guid(clsidNew));
2603
    
2604 2605 2606
    res = COM_OpenKeyForCLSID(clsidOld, NULL, KEY_READ | KEY_WRITE, &hkey);
    if (FAILED(res))
        goto done;
2607 2608
    StringFromGUID2(clsidNew, szClsidNew, CHARS_IN_GUID);
    if (RegSetValueW(hkey, wszAutoConvertTo, REG_SZ, szClsidNew, (strlenW(szClsidNew)+1) * sizeof(WCHAR)))
2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621
    {
        res = REGDB_E_WRITEREGDB;
	goto done;
    }

done:
    if (hkey) RegCloseKey(hkey);
    return res;
}

/******************************************************************************
 *              OleDoAutoConvert        [OLE32.@]
 */
2622
HRESULT WINAPI OleDoAutoConvert(LPSTORAGE pStg, LPCLSID pClsidNew)
2623 2624 2625 2626 2627
{
    FIXME("(%p,%p) : stub\n",pStg,pClsidNew);
    return E_NOTIMPL;
}

2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640
/******************************************************************************
 *              OleIsRunning        [OLE32.@]
 */
BOOL WINAPI OleIsRunning(LPOLEOBJECT pObject)
{
    IRunnableObject *pRunnable;
    HRESULT hr;
    BOOL running;

    TRACE("(%p)\n", pObject);

    hr = IOleObject_QueryInterface(pObject, &IID_IRunnableObject, (void **)&pRunnable);
    if (FAILED(hr))
2641
        return TRUE;
2642 2643 2644 2645 2646
    running = IRunnableObject_IsRunning(pRunnable);
    IRunnableObject_Release(pRunnable);
    return running;
}

2647 2648 2649 2650 2651 2652 2653 2654 2655 2656
/***********************************************************************
 *           OleNoteObjectVisible			    [OLE32.@]
 */
HRESULT WINAPI OleNoteObjectVisible(LPUNKNOWN pUnknown, BOOL bVisible)
{
    TRACE("(%p, %s)\n", pUnknown, bVisible ? "TRUE" : "FALSE");
    return CoLockObjectExternal(pUnknown, bVisible, TRUE);
}


2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670
/***********************************************************************
 *           OLE_FreeClipDataArray   [internal]
 *
 * NOTES:
 *  frees the data associated with an array of CLIPDATAs
 */
static void OLE_FreeClipDataArray(ULONG count, CLIPDATA * pClipDataArray)
{
    ULONG i;
    for (i = 0; i < count; i++)
        if (pClipDataArray[i].pClipData)
            CoTaskMemFree(pClipDataArray[i].pClipData);
}

2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686
/***********************************************************************
 *           PropSysAllocString			    [OLE32.@]
 * NOTES:
 *  Basically a copy of SysAllocStringLen.
 */
BSTR WINAPI PropSysAllocString(LPCOLESTR str)
{
    DWORD  bufferSize;
    DWORD* newBuffer;
    WCHAR* stringBuffer;
    int len;

    if (!str) return 0;

    len = lstrlenW(str);
    /*
2687
     * Find the length of the buffer passed-in, in bytes.
2688 2689 2690 2691 2692
     */
    bufferSize = len * sizeof (WCHAR);

    /*
     * Allocate a new buffer to hold the string.
2693
     * Don't forget to keep an empty spot at the beginning of the
2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715
     * buffer for the character count and an extra character at the
     * end for the NULL.
     */
    newBuffer = HeapAlloc(GetProcessHeap(), 0,
                          bufferSize + sizeof(WCHAR) + sizeof(DWORD));

    /*
     * If the memory allocation failed, return a null pointer.
     */
    if (newBuffer==0)
      return 0;

    /*
     * Copy the length of the string in the placeholder.
     */
    *newBuffer = bufferSize;

    /*
     * Skip the byte count.
     */
    newBuffer++;

2716
    memcpy(newBuffer, str, bufferSize);
2717 2718 2719 2720 2721 2722

    /*
     * Make sure that there is a nul character at the end of the
     * string.
     */
    stringBuffer = (WCHAR*)newBuffer;
2723
    stringBuffer[len] = '\0';
2724

2725
    return stringBuffer;
2726 2727 2728 2729 2730 2731 2732
}

/***********************************************************************
 *           PropSysFreeString			    [OLE32.@]
 * NOTES
 *  Copy of SysFreeString.
 */
2733
void WINAPI PropSysFreeString(LPOLESTR str)
2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754
{
    DWORD* bufferPointer;

    /* NULL is a valid parameter */
    if(!str) return;

    /*
     * We have to be careful when we free a BSTR pointer, it points to
     * the beginning of the string but it skips the byte count contained
     * before the string.
     */
    bufferPointer = (DWORD*)str;

    bufferPointer--;

    /*
     * Free the memory from its "real" origin.
     */
    HeapFree(GetProcessHeap(), 0, bufferPointer);
}

2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772
/******************************************************************************
 * Check if a PROPVARIANT's type is valid.
 */
static inline HRESULT PROPVARIANT_ValidateType(VARTYPE vt)
{
    switch (vt)
    {
    case VT_EMPTY:
    case VT_NULL:
    case VT_I2:
    case VT_I4:
    case VT_R4:
    case VT_R8:
    case VT_CY:
    case VT_DATE:
    case VT_BSTR:
    case VT_ERROR:
    case VT_BOOL:
2773
    case VT_DECIMAL:
2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805 2806 2807 2808 2809 2810 2811 2812 2813 2814 2815
    case VT_UI1:
    case VT_UI2:
    case VT_UI4:
    case VT_I8:
    case VT_UI8:
    case VT_LPSTR:
    case VT_LPWSTR:
    case VT_FILETIME:
    case VT_BLOB:
    case VT_STREAM:
    case VT_STORAGE:
    case VT_STREAMED_OBJECT:
    case VT_STORED_OBJECT:
    case VT_BLOB_OBJECT:
    case VT_CF:
    case VT_CLSID:
    case VT_I2|VT_VECTOR:
    case VT_I4|VT_VECTOR:
    case VT_R4|VT_VECTOR:
    case VT_R8|VT_VECTOR:
    case VT_CY|VT_VECTOR:
    case VT_DATE|VT_VECTOR:
    case VT_BSTR|VT_VECTOR:
    case VT_ERROR|VT_VECTOR:
    case VT_BOOL|VT_VECTOR:
    case VT_VARIANT|VT_VECTOR:
    case VT_UI1|VT_VECTOR:
    case VT_UI2|VT_VECTOR:
    case VT_UI4|VT_VECTOR:
    case VT_I8|VT_VECTOR:
    case VT_UI8|VT_VECTOR:
    case VT_LPSTR|VT_VECTOR:
    case VT_LPWSTR|VT_VECTOR:
    case VT_FILETIME|VT_VECTOR:
    case VT_CF|VT_VECTOR:
    case VT_CLSID|VT_VECTOR:
        return S_OK;
    }
    WARN("Bad type %d\n", vt);
    return STG_E_INVALIDPARAMETER;
}

2816 2817 2818 2819 2820
/***********************************************************************
 *           PropVariantClear			    [OLE32.@]
 */
HRESULT WINAPI PropVariantClear(PROPVARIANT * pvar) /* [in/out] */
{
2821 2822
    HRESULT hr;

2823 2824 2825 2826 2827
    TRACE("(%p)\n", pvar);

    if (!pvar)
        return S_OK;

2828 2829 2830 2831
    hr = PROPVARIANT_ValidateType(pvar->vt);
    if (FAILED(hr))
        return hr;

2832 2833
    switch(pvar->vt)
    {
2834 2835 2836 2837 2838 2839 2840 2841 2842 2843
    case VT_EMPTY:
    case VT_NULL:
    case VT_I2:
    case VT_I4:
    case VT_R4:
    case VT_R8:
    case VT_CY:
    case VT_DATE:
    case VT_ERROR:
    case VT_BOOL:
2844
    case VT_DECIMAL:
2845 2846 2847 2848 2849 2850 2851
    case VT_UI1:
    case VT_UI2:
    case VT_UI4:
    case VT_I8:
    case VT_UI8:
    case VT_FILETIME:
        break;
2852 2853 2854 2855
    case VT_STREAM:
    case VT_STREAMED_OBJECT:
    case VT_STORAGE:
    case VT_STORED_OBJECT:
2856 2857
        if (pvar->u.pStream)
            IUnknown_Release(pvar->u.pStream);
2858 2859 2860 2861
        break;
    case VT_CLSID:
    case VT_LPSTR:
    case VT_LPWSTR:
2862
        /* pick an arbitrary typed pointer - we don't care about the type
2863 2864 2865 2866 2867 2868 2869 2870
         * as we are just freeing it */
        CoTaskMemFree(pvar->u.puuid);
        break;
    case VT_BLOB:
    case VT_BLOB_OBJECT:
        CoTaskMemFree(pvar->u.blob.pBlobData);
        break;
    case VT_BSTR:
2871
        if (pvar->u.bstrVal)
2872
            PropSysFreeString(pvar->u.bstrVal);
2873
        break;
2874
    case VT_CF:
2875 2876 2877 2878 2879 2880 2881
        if (pvar->u.pclipdata)
        {
            OLE_FreeClipDataArray(1, pvar->u.pclipdata);
            CoTaskMemFree(pvar->u.pclipdata);
        }
        break;
    default:
2882
        if (pvar->vt & VT_VECTOR)
2883
        {
2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908
            ULONG i;

            switch (pvar->vt & ~VT_VECTOR)
            {
            case VT_VARIANT:
                FreePropVariantArray(pvar->u.capropvar.cElems, pvar->u.capropvar.pElems);
                break;
            case VT_CF:
                OLE_FreeClipDataArray(pvar->u.caclipdata.cElems, pvar->u.caclipdata.pElems);
                break;
            case VT_BSTR:
                for (i = 0; i < pvar->u.cabstr.cElems; i++)
                    PropSysFreeString(pvar->u.cabstr.pElems[i]);
                break;
            case VT_LPSTR:
                for (i = 0; i < pvar->u.calpstr.cElems; i++)
                    CoTaskMemFree(pvar->u.calpstr.pElems[i]);
                break;
            case VT_LPWSTR:
                for (i = 0; i < pvar->u.calpwstr.cElems; i++)
                    CoTaskMemFree(pvar->u.calpwstr.pElems[i]);
                break;
            }
            if (pvar->vt & ~VT_VECTOR)
            {
2909
                /* pick an arbitrary VT_VECTOR structure - they all have the same
2910 2911 2912
                 * memory layout */
                CoTaskMemFree(pvar->u.capropvar.pElems);
            }
2913
        }
2914 2915
        else
            WARN("Invalid/unsupported type %d\n", pvar->vt);
2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929
    }

    ZeroMemory(pvar, sizeof(*pvar));

    return S_OK;
}

/***********************************************************************
 *           PropVariantCopy			    [OLE32.@]
 */
HRESULT WINAPI PropVariantCopy(PROPVARIANT *pvarDest,      /* [out] */
                               const PROPVARIANT *pvarSrc) /* [in] */
{
    ULONG len;
2930 2931
    HRESULT hr;

2932
    TRACE("(%p, %p vt %04x)\n", pvarDest, pvarSrc, pvarSrc->vt);
2933 2934 2935 2936

    hr = PROPVARIANT_ValidateType(pvarSrc->vt);
    if (FAILED(hr))
        return hr;
2937 2938

    /* this will deal with most cases */
2939
    *pvarDest = *pvarSrc;
2940 2941 2942

    switch(pvarSrc->vt)
    {
2943 2944 2945 2946 2947 2948 2949
    case VT_EMPTY:
    case VT_NULL:
    case VT_I1:
    case VT_UI1:
    case VT_I2:
    case VT_UI2:
    case VT_BOOL:
2950
    case VT_DECIMAL:
2951 2952 2953 2954 2955 2956 2957 2958 2959
    case VT_I4:
    case VT_UI4:
    case VT_R4:
    case VT_ERROR:
    case VT_I8:
    case VT_UI8:
    case VT_R8:
    case VT_CY:
    case VT_DATE:
2960 2961
    case VT_FILETIME:
        break;
2962 2963 2964 2965 2966 2967 2968 2969
    case VT_STREAM:
    case VT_STREAMED_OBJECT:
    case VT_STORAGE:
    case VT_STORED_OBJECT:
        IUnknown_AddRef((LPUNKNOWN)pvarDest->u.pStream);
        break;
    case VT_CLSID:
        pvarDest->u.puuid = CoTaskMemAlloc(sizeof(CLSID));
2970
        *pvarDest->u.puuid = *pvarSrc->u.puuid;
2971 2972 2973
        break;
    case VT_LPSTR:
        len = strlen(pvarSrc->u.pszVal);
2974 2975
        pvarDest->u.pszVal = CoTaskMemAlloc((len+1)*sizeof(CHAR));
        CopyMemory(pvarDest->u.pszVal, pvarSrc->u.pszVal, (len+1)*sizeof(CHAR));
2976 2977 2978
        break;
    case VT_LPWSTR:
        len = lstrlenW(pvarSrc->u.pwszVal);
2979 2980
        pvarDest->u.pwszVal = CoTaskMemAlloc((len+1)*sizeof(WCHAR));
        CopyMemory(pvarDest->u.pwszVal, pvarSrc->u.pwszVal, (len+1)*sizeof(WCHAR));
2981 2982 2983 2984 2985 2986 2987 2988 2989 2990 2991
        break;
    case VT_BLOB:
    case VT_BLOB_OBJECT:
        if (pvarSrc->u.blob.pBlobData)
        {
            len = pvarSrc->u.blob.cbSize;
            pvarDest->u.blob.pBlobData = CoTaskMemAlloc(len);
            CopyMemory(pvarDest->u.blob.pBlobData, pvarSrc->u.blob.pBlobData, len);
        }
        break;
    case VT_BSTR:
2992
        pvarDest->u.bstrVal = PropSysAllocString(pvarSrc->u.bstrVal);
2993 2994 2995 2996 2997
        break;
    case VT_CF:
        if (pvarSrc->u.pclipdata)
        {
            len = pvarSrc->u.pclipdata->cbSize - sizeof(pvarSrc->u.pclipdata->ulClipFmt);
2998 2999 3000
            pvarDest->u.pclipdata = CoTaskMemAlloc(sizeof (CLIPDATA));
            pvarDest->u.pclipdata->cbSize = pvarSrc->u.pclipdata->cbSize;
            pvarDest->u.pclipdata->ulClipFmt = pvarSrc->u.pclipdata->ulClipFmt;
Mike McCormack's avatar
Mike McCormack committed
3001
            pvarDest->u.pclipdata->pClipData = CoTaskMemAlloc(len);
3002 3003 3004 3005 3006 3007 3008
            CopyMemory(pvarDest->u.pclipdata->pClipData, pvarSrc->u.pclipdata->pClipData, len);
        }
        break;
    default:
        if (pvarSrc->vt & VT_VECTOR)
        {
            int elemSize;
3009 3010 3011
            ULONG i;

            switch(pvarSrc->vt & ~VT_VECTOR)
3012 3013 3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029
            {
            case VT_I1:       elemSize = sizeof(pvarSrc->u.cVal); break;
            case VT_UI1:      elemSize = sizeof(pvarSrc->u.bVal); break;
            case VT_I2:       elemSize = sizeof(pvarSrc->u.iVal); break;
            case VT_UI2:      elemSize = sizeof(pvarSrc->u.uiVal); break;
            case VT_BOOL:     elemSize = sizeof(pvarSrc->u.boolVal); break;
            case VT_I4:       elemSize = sizeof(pvarSrc->u.lVal); break;
            case VT_UI4:      elemSize = sizeof(pvarSrc->u.ulVal); break;
            case VT_R4:       elemSize = sizeof(pvarSrc->u.fltVal); break;
            case VT_R8:       elemSize = sizeof(pvarSrc->u.dblVal); break;
            case VT_ERROR:    elemSize = sizeof(pvarSrc->u.scode); break;
            case VT_I8:       elemSize = sizeof(pvarSrc->u.hVal); break;
            case VT_UI8:      elemSize = sizeof(pvarSrc->u.uhVal); break;
            case VT_CY:       elemSize = sizeof(pvarSrc->u.cyVal); break;
            case VT_DATE:     elemSize = sizeof(pvarSrc->u.date); break;
            case VT_FILETIME: elemSize = sizeof(pvarSrc->u.filetime); break;
            case VT_CLSID:    elemSize = sizeof(*pvarSrc->u.puuid); break;
            case VT_CF:       elemSize = sizeof(*pvarSrc->u.pclipdata); break;
3030 3031 3032
            case VT_BSTR:     elemSize = sizeof(pvarSrc->u.bstrVal); break;
            case VT_LPSTR:    elemSize = sizeof(pvarSrc->u.pszVal); break;
            case VT_LPWSTR:   elemSize = sizeof(pvarSrc->u.pwszVal); break;
3033
            case VT_VARIANT:  elemSize = sizeof(*pvarSrc->u.pvarVal); break;
3034 3035

            default:
3036
                FIXME("Invalid element type: %ul\n", pvarSrc->vt & ~VT_VECTOR);
3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051
                return E_INVALIDARG;
            }
            len = pvarSrc->u.capropvar.cElems;
            pvarDest->u.capropvar.pElems = CoTaskMemAlloc(len * elemSize);
            if (pvarSrc->vt == (VT_VECTOR | VT_VARIANT))
            {
                for (i = 0; i < len; i++)
                    PropVariantCopy(&pvarDest->u.capropvar.pElems[i], &pvarSrc->u.capropvar.pElems[i]);
            }
            else if (pvarSrc->vt == (VT_VECTOR | VT_CF))
            {
                FIXME("Copy clipformats\n");
            }
            else if (pvarSrc->vt == (VT_VECTOR | VT_BSTR))
            {
3052 3053
                for (i = 0; i < len; i++)
                    pvarDest->u.cabstr.pElems[i] = PropSysAllocString(pvarSrc->u.cabstr.pElems[i]);
3054 3055 3056
            }
            else if (pvarSrc->vt == (VT_VECTOR | VT_LPSTR))
            {
3057 3058 3059 3060 3061 3062 3063 3064
                size_t strLen;
                for (i = 0; i < len; i++)
                {
                    strLen = lstrlenA(pvarSrc->u.calpstr.pElems[i]) + 1;
                    pvarDest->u.calpstr.pElems[i] = CoTaskMemAlloc(strLen);
                    memcpy(pvarDest->u.calpstr.pElems[i],
                     pvarSrc->u.calpstr.pElems[i], strLen);
                }
3065
            }
3066
            else if (pvarSrc->vt == (VT_VECTOR | VT_LPWSTR))
3067
            {
3068 3069 3070 3071 3072 3073 3074 3075 3076
                size_t strLen;
                for (i = 0; i < len; i++)
                {
                    strLen = (lstrlenW(pvarSrc->u.calpwstr.pElems[i]) + 1) *
                     sizeof(WCHAR);
                    pvarDest->u.calpstr.pElems[i] = CoTaskMemAlloc(strLen);
                    memcpy(pvarDest->u.calpstr.pElems[i],
                     pvarSrc->u.calpstr.pElems[i], strLen);
                }
3077 3078 3079 3080
            }
            else
                CopyMemory(pvarDest->u.capropvar.pElems, pvarSrc->u.capropvar.pElems, len * elemSize);
        }
3081 3082
        else
            WARN("Invalid/unsupported type %d\n", pvarSrc->vt);
3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095
    }

    return S_OK;
}

/***********************************************************************
 *           FreePropVariantArray			    [OLE32.@]
 */
HRESULT WINAPI FreePropVariantArray(ULONG cVariants, /* [in] */
                                    PROPVARIANT *rgvars)    /* [in/out] */
{
    ULONG i;

3096
    TRACE("(%u, %p)\n", cVariants, rgvars);
3097

3098 3099 3100
    if (!rgvars)
        return E_INVALIDARG;

3101 3102 3103 3104 3105 3106
    for(i = 0; i < cVariants; i++)
        PropVariantClear(&rgvars[i]);

    return S_OK;
}

3107
/******************************************************************************
3108
 * DllDebugObjectRPCHook (OLE32.@)
3109 3110 3111 3112 3113 3114 3115 3116
 * turns on and off internal debugging,  pointer is only used on macintosh
 */

BOOL WINAPI DllDebugObjectRPCHook(BOOL b, void *dummy)
{
  FIXME("stub\n");
  return TRUE;
}