filemoniker.c 44.7 KB
Newer Older
1 2
/*
 * FileMonikers implementation
3
 *
4
 * Copyright 1999  Noomen Hamza
5
 * Copyright 2007  Robert Shearman
6 7 8 9 10 11 12 13 14 15 16 17 18
 *
 * 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
19
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20
 */
21 22

#include <assert.h>
23
#include <stdarg.h>
Andreas Mohr's avatar
Andreas Mohr committed
24
#include <string.h>
25

26
#define COBJMACROS
27 28
#define NONAMELESSUNION
#define NONAMELESSSTRUCT
29

30
#include "windef.h"
31
#include "winbase.h"
32
#include "winerror.h"
33
#include "winnls.h"
34
#include "wine/unicode.h"
35
#include "wine/debug.h"
Noomen Hamza's avatar
Noomen Hamza committed
36
#include "objbase.h"
37
#include "moniker.h"
Noomen Hamza's avatar
Noomen Hamza committed
38

39 40
#include "compobj_private.h"

41
WINE_DEFAULT_DEBUG_CHANNEL(ole);
42

Noomen Hamza's avatar
Noomen Hamza committed
43
/* filemoniker data structure */
44
typedef struct FileMonikerImpl{
45 46 47
    IMoniker IMoniker_iface;
    IROTData IROTData_iface;
    LONG ref;
Noomen Hamza's avatar
Noomen Hamza committed
48
    LPOLESTR filePathName; /* path string identified by this filemoniker */
49
    IUnknown *pMarshal; /* custom marshaler */
50 51
} FileMonikerImpl;

52
static inline FileMonikerImpl *impl_from_IMoniker(IMoniker *iface)
53
{
54 55 56 57 58 59
    return CONTAINING_RECORD(iface, FileMonikerImpl, IMoniker_iface);
}

static inline FileMonikerImpl *impl_from_IROTData(IROTData *iface)
{
    return CONTAINING_RECORD(iface, FileMonikerImpl, IROTData_iface);
60 61
}

Noomen Hamza's avatar
Noomen Hamza committed
62
/* Local function used by filemoniker implementation */
63 64
static HRESULT FileMonikerImpl_Construct(FileMonikerImpl* iface, LPCOLESTR lpszPathName);
static HRESULT FileMonikerImpl_Destroy(FileMonikerImpl* iface);
Noomen Hamza's avatar
Noomen Hamza committed
65

66 67
/*******************************************************************************
 *        FileMoniker_QueryInterface
68 69 70
 */
static HRESULT WINAPI
FileMonikerImpl_QueryInterface(IMoniker* iface,REFIID riid,void** ppvObject)
71
{
72
    FileMonikerImpl *This = impl_from_IMoniker(iface);
73

74
    TRACE("(%p,%s,%p)\n",This,debugstr_guid(riid),ppvObject);
75

Noomen Hamza's avatar
Noomen Hamza committed
76
    /* Perform a sanity check on the parameters.*/
77
    if ( ppvObject==0 )
Noomen Hamza's avatar
Noomen Hamza committed
78
	return E_INVALIDARG;
79

Noomen Hamza's avatar
Noomen Hamza committed
80
    /* Initialize the return parameter */
81
    *ppvObject = 0;
82

Noomen Hamza's avatar
Noomen Hamza committed
83 84 85 86 87 88 89
    /* Compare the riid with the interface IDs implemented by this object.*/
    if (IsEqualIID(&IID_IUnknown, riid)      ||
        IsEqualIID(&IID_IPersist, riid)      ||
        IsEqualIID(&IID_IPersistStream,riid) ||
        IsEqualIID(&IID_IMoniker, riid)
       )
        *ppvObject = iface;
90

Noomen Hamza's avatar
Noomen Hamza committed
91
    else if (IsEqualIID(&IID_IROTData, riid))
92
        *ppvObject = &This->IROTData_iface;
93 94 95 96 97 98 99 100 101
    else if (IsEqualIID(&IID_IMarshal, riid))
    {
        HRESULT hr = S_OK;
        if (!This->pMarshal)
            hr = MonikerMarshal_Create(iface, &This->pMarshal);
        if (hr != S_OK)
            return hr;
        return IUnknown_QueryInterface(This->pMarshal, riid, ppvObject);
    }
Noomen Hamza's avatar
Noomen Hamza committed
102 103 104 105

    /* Check that we obtained an interface.*/
    if ((*ppvObject)==0)
        return E_NOINTERFACE;
106

Noomen Hamza's avatar
Noomen Hamza committed
107
    /* Query Interface always increases the reference count by one when it is successful */
108
    IMoniker_AddRef(iface);
109

Noomen Hamza's avatar
Noomen Hamza committed
110
    return S_OK;
111 112 113 114
}

/******************************************************************************
 *        FileMoniker_AddRef
115 116 117
 */
static ULONG WINAPI
FileMonikerImpl_AddRef(IMoniker* iface)
118
{
119
    FileMonikerImpl *This = impl_from_IMoniker(iface);
120

121
    TRACE("(%p)\n",iface);
122

123
    return InterlockedIncrement(&This->ref);
124 125 126 127
}

/******************************************************************************
 *        FileMoniker_Release
128 129 130
 */
static ULONG WINAPI
FileMonikerImpl_Release(IMoniker* iface)
131
{
132
    FileMonikerImpl *This = impl_from_IMoniker(iface);
133
    ULONG ref;
134

135
    TRACE("(%p)\n",iface);
136

137
    ref = InterlockedDecrement(&This->ref);
138

139
    /* destroy the object if there are no more references to it */
140
    if (ref == 0) FileMonikerImpl_Destroy(This);
Noomen Hamza's avatar
Noomen Hamza committed
141

142
    return ref;
143 144 145 146
}

/******************************************************************************
 *        FileMoniker_GetClassID
147 148 149
 */
static HRESULT WINAPI
FileMonikerImpl_GetClassID(IMoniker* iface, CLSID *pClassID)
150
{
Robert Shearman's avatar
Robert Shearman committed
151
    TRACE("(%p,%p)\n",iface,pClassID);
152

Noomen Hamza's avatar
Noomen Hamza committed
153 154
    if (pClassID==NULL)
        return E_POINTER;
155

Noomen Hamza's avatar
Noomen Hamza committed
156
    *pClassID = CLSID_FileMoniker;
157

Noomen Hamza's avatar
Noomen Hamza committed
158
    return S_OK;
159 160 161 162
}

/******************************************************************************
 *        FileMoniker_IsDirty
163 164 165 166 167 168 169
 *
 * Note that the OLE-provided implementations of the IPersistStream::IsDirty
 * method in the OLE-provided moniker interfaces always return S_FALSE because
 * their internal state never changes.
 */
static HRESULT WINAPI
FileMonikerImpl_IsDirty(IMoniker* iface)
170
{
171

172
    TRACE("(%p)\n",iface);
173

Noomen Hamza's avatar
Noomen Hamza committed
174
    return S_FALSE;
175 176 177 178
}

/******************************************************************************
 *        FileMoniker_Load
179 180 181 182 183
 *
 * this function locates and reads from the stream the filePath string
 * written by FileMonikerImpl_Save
 */
static HRESULT WINAPI
184
FileMonikerImpl_Load(IMoniker* iface, IStream* pStm)
185
{
186
    FileMonikerImpl *This = impl_from_IMoniker(iface);
Noomen Hamza's avatar
Noomen Hamza committed
187
    HRESULT res;
188 189
    CHAR* filePathA = NULL;
    WCHAR* filePathW = NULL;
Noomen Hamza's avatar
Noomen Hamza committed
190 191
    ULONG bread;
    WORD  wbuffer;
192 193
    DWORD dwbuffer, bytesA, bytesW, len;
    int i;
194

195

196
    TRACE("(%p,%p)\n",iface,pStm);
Noomen Hamza's avatar
Noomen Hamza committed
197

198
    /* first WORD */
Noomen Hamza's avatar
Noomen Hamza committed
199
    res=IStream_Read(pStm,&wbuffer,sizeof(WORD),&bread);
200
    if (bread!=sizeof(WORD))
201
    {
202 203
        WARN("Couldn't read 0 word\n");
        goto fail;
204
    }
205

Noomen Hamza's avatar
Noomen Hamza committed
206
    /* read filePath string length (plus one) */
207
    res=IStream_Read(pStm,&bytesA,sizeof(DWORD),&bread);
Noomen Hamza's avatar
Noomen Hamza committed
208
    if (bread != sizeof(DWORD))
209
    {
210 211
        WARN("Couldn't read file string length\n");
        goto fail;
212
    }
Noomen Hamza's avatar
Noomen Hamza committed
213 214

    /* read filePath string */
215 216 217 218 219 220 221 222 223
    filePathA=HeapAlloc(GetProcessHeap(),0,bytesA);
    if (!filePathA)
    {
        res = E_OUTOFMEMORY;
        goto fail;
    }

    res=IStream_Read(pStm,filePathA,bytesA,&bread);
    if (bread != bytesA)
224
    {
225 226
        WARN("Couldn't read file path string\n");
        goto fail;
227
    }
Noomen Hamza's avatar
Noomen Hamza committed
228

229 230 231
    /* read the unknown value */
    IStream_Read(pStm,&wbuffer,sizeof(WORD),&bread);
    if (bread != sizeof(WORD))
232
    {
233 234 235 236 237 238 239 240 241
        WARN("Couldn't read unknown value\n");
        goto fail;
    }

    /* read the DEAD constant */
    IStream_Read(pStm,&wbuffer,sizeof(WORD),&bread);
    if (bread != sizeof(WORD))
    {
        WARN("Couldn't read DEAD constant\n");
242
        goto fail;
243
    }
244

245 246 247
    for(i=0;i<5;i++)
    {
        res=IStream_Read(pStm,&dwbuffer,sizeof(DWORD),&bread);
248
        if (bread!=sizeof(DWORD))
249
        {
250 251
            WARN("Couldn't read 0 padding\n");
            goto fail;
252
        }
Noomen Hamza's avatar
Noomen Hamza committed
253
    }
254

Noomen Hamza's avatar
Noomen Hamza committed
255
    res=IStream_Read(pStm,&dwbuffer,sizeof(DWORD),&bread);
256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274
    if (bread!=sizeof(DWORD))
        goto fail;

    if (!dwbuffer) /* No W-string */
    {        
        bytesA--;
        len=MultiByteToWideChar(CP_ACP, MB_ERR_INVALID_CHARS, filePathA, bytesA, NULL, 0);
        if (!len)
            goto fail;

        filePathW=HeapAlloc(GetProcessHeap(),0,(len+1)*sizeof(WCHAR));
        if (!filePathW)
        {
            res = E_OUTOFMEMORY;
            goto fail;
        }
        MultiByteToWideChar(CP_ACP, MB_ERR_INVALID_CHARS, filePathA, -1, filePathW, len+1);
        goto succeed;
    }
Noomen Hamza's avatar
Noomen Hamza committed
275

276 277 278 279
    if (dwbuffer < 6)
        goto fail;

    bytesW=dwbuffer - 6;
280

Noomen Hamza's avatar
Noomen Hamza committed
281
    res=IStream_Read(pStm,&dwbuffer,sizeof(DWORD),&bread);
282 283
    if (bread!=sizeof(DWORD) || dwbuffer!=bytesW)
        goto fail;
Noomen Hamza's avatar
Noomen Hamza committed
284 285 286

    res=IStream_Read(pStm,&wbuffer,sizeof(WORD),&bread);
    if (bread!=sizeof(WORD) || wbuffer!=0x3)
287 288 289 290 291 292 293 294
        goto fail;

    len=bytesW/sizeof(WCHAR);
    filePathW=HeapAlloc(GetProcessHeap(),0,(len+1)*sizeof(WCHAR));
    if(!filePathW)
    {
         res = E_OUTOFMEMORY;
         goto fail;
295
    }
296 297 298
    res=IStream_Read(pStm,filePathW,bytesW,&bread);
    if (bread!=bytesW)
         goto fail;
Noomen Hamza's avatar
Noomen Hamza committed
299

300
    filePathW[len]=0;
Noomen Hamza's avatar
Noomen Hamza committed
301

302 303 304
 succeed:
    HeapFree(GetProcessHeap(),0,filePathA);
    HeapFree(GetProcessHeap(),0,This->filePathName);
Noomen Hamza's avatar
Noomen Hamza committed
305 306
    This->filePathName=filePathW;

307 308 309 310 311 312 313 314
    return S_OK;

 fail:
    HeapFree(GetProcessHeap(), 0, filePathA);
    HeapFree(GetProcessHeap(), 0, filePathW);

    if (SUCCEEDED(res))
         res = E_FAIL;
Noomen Hamza's avatar
Noomen Hamza committed
315
    return res;
316 317 318
}

/******************************************************************************
319
 *        FileMoniker_Save
320
 *
321
 * This function saves data of this object. In the beginning I thought
322
 * that I have just to write the filePath string on Stream. But, when I
323
 * tested this function with windows program samples, I noticed that it
324 325 326 327
 * was not the case. This implementation is based on XP SP2. Other versions
 * of Windows have minor variations.
 *
 * Data which must be written on stream is:
328
 * 1) WORD constant: zero (not validated by Windows)
329 330
 * 2) length of the path string ("\0" included)
 * 3) path string type A
331 332 333 334 335
 * 4) Unknown WORD value: Frequently 0xFFFF, but not always. If set very large,
 *     Windows returns E_OUTOFMEMORY
 * 5) WORD Constant: 0xDEAD (not validated by Windows)
 * 6) five DWORD constant: zero (not validated by Windows)
 * 7) If we're only writing the multibyte version,
336 337
 *     write a zero DWORD and finish.
 *
338
 * 8) DWORD: double-length of the path string type W ("\0" not
339
 *    included)
340 341
 * 9) WORD constant: 0x3
 * 10) filePath unicode string.
342
 *
343 344 345
 */
static HRESULT WINAPI
FileMonikerImpl_Save(IMoniker* iface, IStream* pStm, BOOL fClearDirty)
346
{
347
    FileMonikerImpl *This = impl_from_IMoniker(iface);
Noomen Hamza's avatar
Noomen Hamza committed
348 349
    HRESULT res;
    LPOLESTR filePathW=This->filePathName;
350 351
    CHAR*    filePathA;
    DWORD bytesA, bytesW, len;
Noomen Hamza's avatar
Noomen Hamza committed
352

353 354
    static const WORD FFFF = 0xFFFF; /* Constants */
    static const WORD DEAD = 0xDEAD;
355 356
    static const DWORD ZERO     = 0;
    static const WORD  THREE    = 0x3;
Noomen Hamza's avatar
Noomen Hamza committed
357

358 359
    int i;
    BOOL bUsedDefault, bWriteWide;
Noomen Hamza's avatar
Noomen Hamza committed
360

361
    TRACE("(%p,%p,%d)\n",iface,pStm,fClearDirty);
Noomen Hamza's avatar
Noomen Hamza committed
362 363 364 365

    if (pStm==NULL)
        return E_POINTER;

366 367
    /* write a 0 WORD */
    res=IStream_Write(pStm,&ZERO,sizeof(WORD),NULL);
368
    if (FAILED(res)) return res;
369 370 371 372

    /* write length of filePath string ( 0 included )*/
    bytesA = WideCharToMultiByte( CP_ACP, 0, filePathW, -1, NULL, 0, NULL, NULL );
    res=IStream_Write(pStm,&bytesA,sizeof(DWORD),NULL);
373
    if (FAILED(res)) return res;
374 375 376 377 378 379 380

    /* write A string (with '\0') */
    filePathA=HeapAlloc(GetProcessHeap(),0,bytesA);
    if (!filePathA)
        return E_OUTOFMEMORY;
    WideCharToMultiByte( CP_ACP, 0, filePathW, -1, filePathA, bytesA, NULL, &bUsedDefault);
    res=IStream_Write(pStm,filePathA,bytesA,NULL);
Noomen Hamza's avatar
Noomen Hamza committed
381
    HeapFree(GetProcessHeap(),0,filePathA);
382
    if (FAILED(res)) return res;
Noomen Hamza's avatar
Noomen Hamza committed
383

384 385 386 387 388 389
    /* write a WORD 0xFFFF */
    res=IStream_Write(pStm,&FFFF,sizeof(WORD),NULL);
    if (FAILED(res)) return res;

    /* write a WORD 0xDEAD */
    res=IStream_Write(pStm,&DEAD,sizeof(WORD),NULL);
390
    if (FAILED(res)) return res;
391

392 393 394 395
    /* write 5 zero DWORDs */
    for(i=0;i<5;i++)
    {
        res=IStream_Write(pStm,&ZERO,sizeof(DWORD),NULL);
396
        if (FAILED(res)) return res;
397
    }
Noomen Hamza's avatar
Noomen Hamza committed
398

399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417
    /* Write the wide version if:
     *    + couldn't convert to CP_ACP, 
     * or + it's a directory, 
     * or + there's a character > 0xFF 
     */
    len = lstrlenW(filePathW);
    bWriteWide = (bUsedDefault || (len > 0 && filePathW[len-1]=='\\' ));
    if (!bWriteWide)
    {
        WCHAR* pch;
        for(pch=filePathW;*pch;++pch) 
        {
            if (*pch > 0xFF)
            {
                bWriteWide = TRUE;
                break;
            }
        }
    }
Noomen Hamza's avatar
Noomen Hamza committed
418

419
    if (!bWriteWide)
420
        return IStream_Write(pStm,&ZERO,sizeof(DWORD),NULL);
421 422 423 424

    /* write bytes needed for the filepathW (without 0) + 6 */
    bytesW = len*sizeof(WCHAR) + 6;
    res=IStream_Write(pStm,&bytesW,sizeof(DWORD),NULL);
425
    if (FAILED(res)) return res;
Noomen Hamza's avatar
Noomen Hamza committed
426

427 428 429
    /* try again, without the extra 6 */
    bytesW -= 6;
    res=IStream_Write(pStm,&bytesW,sizeof(DWORD),NULL);
430
    if (FAILED(res)) return res;
Noomen Hamza's avatar
Noomen Hamza committed
431

432 433
    /* write a WORD 3 */
    res=IStream_Write(pStm,&THREE,sizeof(WORD),NULL);
434
    if (FAILED(res)) return res;
Noomen Hamza's avatar
Noomen Hamza committed
435

436
    /* write W string (no 0) */
437
    return IStream_Write(pStm,filePathW,bytesW,NULL);
438 439 440 441
}

/******************************************************************************
 *        FileMoniker_GetSizeMax
442 443 444
 */
static HRESULT WINAPI
FileMonikerImpl_GetSizeMax(IMoniker* iface, ULARGE_INTEGER* pcbSize)
445
{
446
    FileMonikerImpl *This = impl_from_IMoniker(iface);
447

448
    TRACE("(%p,%p)\n",iface,pcbSize);
449

450
    if (!pcbSize)
Noomen Hamza's avatar
Noomen Hamza committed
451 452
        return E_POINTER;

453 454 455 456 457
    /* We could calculate exactly (see ...::Save()) but instead
     * we'll make a quick over-estimate, like Windows (NT4, XP) does.
     */
    pcbSize->u.LowPart  = 0x38 + 4 * lstrlenW(This->filePathName);
    pcbSize->u.HighPart = 0;
Noomen Hamza's avatar
Noomen Hamza committed
458 459

    return S_OK;
460 461 462
}

/******************************************************************************
Noomen Hamza's avatar
Noomen Hamza committed
463
 *        FileMoniker_Destroy (local function)
464
 *******************************************************************************/
465
HRESULT FileMonikerImpl_Destroy(FileMonikerImpl* This)
Noomen Hamza's avatar
Noomen Hamza committed
466
{
467
    TRACE("(%p)\n",This);
468

469
    if (This->pMarshal) IUnknown_Release(This->pMarshal);
470
    HeapFree(GetProcessHeap(),0,This->filePathName);
Noomen Hamza's avatar
Noomen Hamza committed
471
    HeapFree(GetProcessHeap(),0,This);
472 473 474 475 476 477

    return S_OK;
}

/******************************************************************************
 *                  FileMoniker_BindToObject
478 479 480 481
 */
static HRESULT WINAPI
FileMonikerImpl_BindToObject(IMoniker* iface, IBindCtx* pbc, IMoniker* pmkToLeft,
                             REFIID riid, VOID** ppvResult)
482
{
483
    FileMonikerImpl *This = impl_from_IMoniker(iface);
Noomen Hamza's avatar
Noomen Hamza committed
484 485 486 487 488 489 490
    HRESULT   res=E_FAIL;
    CLSID     clsID;
    IUnknown* pObj=0;
    IRunningObjectTable *prot=0;
    IPersistFile  *ppf=0;
    IClassFactory *pcf=0;
    IClassActivator *pca=0;
491

Noomen Hamza's avatar
Noomen Hamza committed
492 493
    *ppvResult=0;

Robert Shearman's avatar
Robert Shearman committed
494
    TRACE("(%p,%p,%p,%s,%p)\n",iface,pbc,pmkToLeft,debugstr_guid(riid),ppvResult);
Noomen Hamza's avatar
Noomen Hamza committed
495 496

    if(pmkToLeft==NULL){
497

Noomen Hamza's avatar
Noomen Hamza committed
498 499 500
        res=IBindCtx_GetRunningObjectTable(pbc,&prot);

        if (SUCCEEDED(res)){
501
            /* if the requested class was loaded before ! we don't need to reload it */
Noomen Hamza's avatar
Noomen Hamza committed
502 503 504 505 506 507 508
            res = IRunningObjectTable_GetObject(prot,iface,&pObj);

            if (res==S_FALSE){
                /* first activation of this class */
                res=GetClassFile(This->filePathName,&clsID);
                if (SUCCEEDED(res)){

509
                    res=CoCreateInstance(&clsID,NULL,CLSCTX_ALL,&IID_IPersistFile,(void**)&ppf);
Noomen Hamza's avatar
Noomen Hamza committed
510 511 512 513 514 515 516 517 518 519 520 521 522 523
                    if (SUCCEEDED(res)){

                        res=IPersistFile_Load(ppf,This->filePathName,STGM_READ);
                        if (SUCCEEDED(res)){

                            pObj=(IUnknown*)ppf;
                            IUnknown_AddRef(pObj);
                        }
                    }
                }
            }
        }
    }
    else{
524
        res=IMoniker_BindToObject(pmkToLeft,pbc,NULL,&IID_IClassFactory,(void**)&pcf);
Noomen Hamza's avatar
Noomen Hamza committed
525 526 527

        if (res==E_NOINTERFACE){

528
            res=IMoniker_BindToObject(pmkToLeft,pbc,NULL,&IID_IClassActivator,(void**)&pca);
529

Noomen Hamza's avatar
Noomen Hamza committed
530 531 532 533 534
            if (res==E_NOINTERFACE)
                return MK_E_INTERMEDIATEINTERFACENOTSUPPORTED;
        }
        if (pcf!=NULL){

535
            IClassFactory_CreateInstance(pcf,NULL,&IID_IPersistFile,(void**)&ppf);
Noomen Hamza's avatar
Noomen Hamza committed
536 537 538 539 540 541 542 543 544 545 546

            res=IPersistFile_Load(ppf,This->filePathName,STGM_READ);

            if (SUCCEEDED(res)){

                pObj=(IUnknown*)ppf;
                IUnknown_AddRef(pObj);
            }
        }
        if (pca!=NULL){

547
            FIXME("()\n");
548

Noomen Hamza's avatar
Noomen Hamza committed
549 550 551 552 553 554 555 556 557 558 559 560 561
            /*res=GetClassFile(This->filePathName,&clsID);

            if (SUCCEEDED(res)){

                res=IClassActivator_GetClassObject(pca,&clsID,CLSCTX_ALL,0,&IID_IPersistFile,(void**)&ppf);

                if (SUCCEEDED(res)){

                    pObj=(IUnknown*)ppf;
                    IUnknown_AddRef(pObj);
                }
            }*/
        }
562
    }
Noomen Hamza's avatar
Noomen Hamza committed
563 564 565 566 567

    if (pObj!=NULL){
        /* get the requested interface from the loaded class */
        res= IUnknown_QueryInterface(pObj,riid,ppvResult);

568
        IBindCtx_RegisterObjectBound(pbc,*ppvResult);
Noomen Hamza's avatar
Noomen Hamza committed
569 570 571 572 573 574 575 576 577 578 579 580 581 582 583

        IUnknown_Release(pObj);
    }

    if (prot!=NULL)
        IRunningObjectTable_Release(prot);

    if (ppf!=NULL)
        IPersistFile_Release(ppf);

    if (pca!=NULL)
        IClassActivator_Release(pca);

    if (pcf!=NULL)
        IClassFactory_Release(pcf);
584

Noomen Hamza's avatar
Noomen Hamza committed
585
    return res;
586 587 588 589
}

/******************************************************************************
 *        FileMoniker_BindToStorage
590 591 592 593
 */
static HRESULT WINAPI
FileMonikerImpl_BindToStorage(IMoniker* iface, IBindCtx* pbc, IMoniker* pmkToLeft,
                              REFIID riid, VOID** ppvObject)
594
{
Noomen Hamza's avatar
Noomen Hamza committed
595 596 597 598
    LPOLESTR filePath=0;
    IStorage *pstg=0;
    HRESULT res;

Robert Shearman's avatar
Robert Shearman committed
599
    TRACE("(%p,%p,%p,%s,%p)\n",iface,pbc,pmkToLeft,debugstr_guid(riid),ppvObject);
Noomen Hamza's avatar
Noomen Hamza committed
600 601 602 603 604 605

    if (pmkToLeft==NULL){

        if (IsEqualIID(&IID_IStorage, riid)){

            /* get the file name */
606
            IMoniker_GetDisplayName(iface,pbc,pmkToLeft,&filePath);
Noomen Hamza's avatar
Noomen Hamza committed
607

Austin English's avatar
Austin English committed
608
            /* verify if the file contains a storage object */
Noomen Hamza's avatar
Noomen Hamza committed
609 610 611 612 613 614 615
            res=StgIsStorageFile(filePath);

            if(res==S_OK){

                res=StgOpenStorage(filePath,NULL,STGM_READWRITE|STGM_SHARE_DENY_WRITE,NULL,0,&pstg);

                if (SUCCEEDED(res)){
616

Noomen Hamza's avatar
Noomen Hamza committed
617 618 619 620 621 622 623 624 625 626 627
                    *ppvObject=pstg;

                    IStorage_AddRef(pstg);

                    return res;
                }
            }
            CoTaskMemFree(filePath);
        }
        else
            if ( (IsEqualIID(&IID_IStream, riid)) || (IsEqualIID(&IID_ILockBytes, riid)) )
628
                return E_FAIL;
Noomen Hamza's avatar
Noomen Hamza committed
629 630 631 632 633
            else
                return E_NOINTERFACE;
    }
    else {

Robert Shearman's avatar
Robert Shearman committed
634
        FIXME("(%p,%p,%p,%s,%p)\n",iface,pbc,pmkToLeft,debugstr_guid(riid),ppvObject);
635

636 637
        return E_NOTIMPL;
    }
Noomen Hamza's avatar
Noomen Hamza committed
638 639
    return res;
}
640 641 642 643

/******************************************************************************
 *        FileMoniker_Reduce
 ******************************************************************************/
644 645 646
static HRESULT WINAPI
FileMonikerImpl_Reduce(IMoniker* iface, IBindCtx* pbc, DWORD dwReduceHowFar,
                       IMoniker** ppmkToLeft, IMoniker** ppmkReduced)
647
{
648
    TRACE("(%p,%p,%d,%p,%p)\n",iface,pbc,dwReduceHowFar,ppmkToLeft,ppmkReduced);
649

Noomen Hamza's avatar
Noomen Hamza committed
650 651
    if (ppmkReduced==NULL)
        return E_POINTER;
652

653
    IMoniker_AddRef(iface);
654

Noomen Hamza's avatar
Noomen Hamza committed
655 656 657 658
    *ppmkReduced=iface;

    return MK_S_REDUCED_TO_SELF;
}
659

660 661 662 663 664 665 666 667 668
static void free_stringtable(LPOLESTR *stringTable)
{
    int i;

    for (i=0; stringTable[i]!=NULL; i++)
        CoTaskMemFree(stringTable[i]);
    CoTaskMemFree(stringTable);
}

669 670
/******************************************************************************
 *        FileMoniker_ComposeWith
671 672 673 674
 */
static HRESULT WINAPI
FileMonikerImpl_ComposeWith(IMoniker* iface, IMoniker* pmkRight,
                            BOOL fOnlyIfNotGeneric, IMoniker** ppmkComposite)
675
{
Noomen Hamza's avatar
Noomen Hamza committed
676 677
    HRESULT res;
    LPOLESTR str1=0,str2=0,*strDec1=0,*strDec2=0,newStr=0;
678 679
    static const WCHAR twoPoint[]={'.','.',0};
    static const WCHAR bkSlash[]={'\\',0};
Noomen Hamza's avatar
Noomen Hamza committed
680 681 682
    IBindCtx *bind=0;
    int i=0,j=0,lastIdx1=0,lastIdx2=0;
    DWORD mkSys;
683

684
    TRACE("(%p,%p,%d,%p)\n",iface,pmkRight,fOnlyIfNotGeneric,ppmkComposite);
685

Noomen Hamza's avatar
Noomen Hamza committed
686 687 688 689 690 691 692
    if (ppmkComposite==NULL)
        return E_POINTER;

    if (pmkRight==NULL)
	return E_INVALIDARG;

    *ppmkComposite=0;
693

Noomen Hamza's avatar
Noomen Hamza committed
694 695
    IMoniker_IsSystemMoniker(pmkRight,&mkSys);

Austin English's avatar
Austin English committed
696
    /* check if we have two FileMonikers to compose or not */
Noomen Hamza's avatar
Noomen Hamza committed
697 698 699 700
    if(mkSys==MKSYS_FILEMONIKER){

        CreateBindCtx(0,&bind);

701
        IMoniker_GetDisplayName(iface,bind,NULL,&str1);
Noomen Hamza's avatar
Noomen Hamza committed
702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721
        IMoniker_GetDisplayName(pmkRight,bind,NULL,&str2);

        /* decompose pathnames of the two monikers : (to prepare the path merge operation ) */
        lastIdx1=FileMonikerImpl_DecomposePath(str1,&strDec1)-1;
        lastIdx2=FileMonikerImpl_DecomposePath(str2,&strDec2)-1;

        if ((lastIdx1==-1 && lastIdx2>-1)||(lastIdx1==1 && lstrcmpW(strDec1[0],twoPoint)==0))
            return MK_E_SYNTAX;

        if(lstrcmpW(strDec1[lastIdx1],bkSlash)==0)
            lastIdx1--;

        /* for etch "..\" in the left of str2 remove the right element from str1 */
        for(i=0; ( (lastIdx1>=0) && (strDec2[i]!=NULL) && (lstrcmpW(strDec2[i],twoPoint)==0) ) ;i+=2){

            lastIdx1-=2;
        }

        /* the length of the composed path string  is raised by the sum of the two paths lengths  */
        newStr=HeapAlloc(GetProcessHeap(),0,sizeof(WCHAR)*(lstrlenW(str1)+lstrlenW(str2)+1));
722

723 724 725 726 727
        if (newStr)
        {
            /* new path is the concatenation of the rest of str1 and str2 */
            for(*newStr=0,j=0;j<=lastIdx1;j++)
                strcatW(newStr,strDec1[j]);
Noomen Hamza's avatar
Noomen Hamza committed
728

729 730
            if ((strDec2[i]==NULL && lastIdx1>-1 && lastIdx2>-1) || lstrcmpW(strDec2[i],bkSlash)!=0)
                strcatW(newStr,bkSlash);
731

732 733
            for(j=i;j<=lastIdx2;j++)
                strcatW(newStr,strDec2[j]);
734

735 736
            /* create a new moniker with the new string */
            res=CreateFileMoniker(newStr,ppmkComposite);
Noomen Hamza's avatar
Noomen Hamza committed
737

738 739 740 741
            /* free all strings space memory used by this function */
            HeapFree(GetProcessHeap(),0,newStr);
        }
        else res = E_OUTOFMEMORY;
Noomen Hamza's avatar
Noomen Hamza committed
742

743 744
        free_stringtable(strDec1);
        free_stringtable(strDec2);
Noomen Hamza's avatar
Noomen Hamza committed
745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763

        CoTaskMemFree(str1);
        CoTaskMemFree(str2);

        return res;
    }
    else if(mkSys==MKSYS_ANTIMONIKER){

        *ppmkComposite=NULL;
        return S_OK;
    }
    else if (fOnlyIfNotGeneric){

        *ppmkComposite=NULL;
        return MK_E_NEEDGENERIC;
    }
    else

        return CreateGenericComposite(iface,pmkRight,ppmkComposite);
764 765 766 767
}

/******************************************************************************
 *        FileMoniker_Enum
768 769 770
 */
static HRESULT WINAPI
FileMonikerImpl_Enum(IMoniker* iface,BOOL fForward, IEnumMoniker** ppenumMoniker)
771
{
772
    TRACE("(%p,%d,%p)\n",iface,fForward,ppenumMoniker);
773

Noomen Hamza's avatar
Noomen Hamza committed
774 775
    if (ppenumMoniker == NULL)
        return E_POINTER;
776

Noomen Hamza's avatar
Noomen Hamza committed
777
    *ppenumMoniker = NULL;
778

Noomen Hamza's avatar
Noomen Hamza committed
779
    return S_OK;
780 781 782 783
}

/******************************************************************************
 *        FileMoniker_IsEqual
784 785 786
 */
static HRESULT WINAPI
FileMonikerImpl_IsEqual(IMoniker* iface,IMoniker* pmkOtherMoniker)
787
{
788
    FileMonikerImpl *This = impl_from_IMoniker(iface);
Noomen Hamza's avatar
Noomen Hamza committed
789 790 791 792
    CLSID clsid;
    LPOLESTR filePath;
    IBindCtx* bind;
    HRESULT res;
793

794
    TRACE("(%p,%p)\n",iface,pmkOtherMoniker);
795

Noomen Hamza's avatar
Noomen Hamza committed
796 797 798 799 800 801 802 803
    if (pmkOtherMoniker==NULL)
        return S_FALSE;

    IMoniker_GetClassID(pmkOtherMoniker,&clsid);

    if (!IsEqualCLSID(&clsid,&CLSID_FileMoniker))
        return S_FALSE;

804 805
    res = CreateBindCtx(0,&bind);
    if (FAILED(res)) return res;
Noomen Hamza's avatar
Noomen Hamza committed
806

807
    res = S_FALSE;
808
    if (SUCCEEDED(IMoniker_GetDisplayName(pmkOtherMoniker,bind,NULL,&filePath))) {
809 810
	if (!lstrcmpiW(filePath, This->filePathName))
            res = S_OK;
811 812
	CoTaskMemFree(filePath);
    }
813

814 815
    IBindCtx_Release(bind);
    return res;
816 817 818 819
}

/******************************************************************************
 *        FileMoniker_Hash
820 821 822
 */
static HRESULT WINAPI
FileMonikerImpl_Hash(IMoniker* iface,DWORD* pdwHash)
823
{
824
    FileMonikerImpl *This = impl_from_IMoniker(iface);
Noomen Hamza's avatar
Noomen Hamza committed
825 826 827
    int  h = 0,i,skip,len;
    int  off = 0;
    LPOLESTR val;
828

Noomen Hamza's avatar
Noomen Hamza committed
829 830
    if (pdwHash==NULL)
        return E_POINTER;
831

Noomen Hamza's avatar
Noomen Hamza committed
832 833 834 835 836 837 838 839 840 841 842 843 844
    val =  This->filePathName;
    len = lstrlenW(val);

    if (len < 16) {
        for (i = len ; i > 0; i--) {
            h = (h * 37) + val[off++];
 	}
    } else {
        /* only sample some characters */
 	skip = len / 8;
 	for (i = len ; i > 0; i -= skip, off += skip) {
            h = (h * 39) + val[off];
 	}
845
    }
Noomen Hamza's avatar
Noomen Hamza committed
846 847 848 849

    *pdwHash=h;

    return S_OK;
850 851 852 853
}

/******************************************************************************
 *        FileMoniker_IsRunning
854 855 856 857
 */
static HRESULT WINAPI
FileMonikerImpl_IsRunning(IMoniker* iface, IBindCtx* pbc, IMoniker* pmkToLeft,
                          IMoniker* pmkNewlyRunning)
858
{
Noomen Hamza's avatar
Noomen Hamza committed
859 860
    IRunningObjectTable* rot;
    HRESULT res;
861

862
    TRACE("(%p,%p,%p,%p)\n",iface,pbc,pmkToLeft,pmkNewlyRunning);
863

Noomen Hamza's avatar
Noomen Hamza committed
864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879
    if ( (pmkNewlyRunning!=NULL) && (IMoniker_IsEqual(pmkNewlyRunning,iface)==S_OK) )
        return S_OK;

    if (pbc==NULL)
        return E_POINTER;

    res=IBindCtx_GetRunningObjectTable(pbc,&rot);

    if (FAILED(res))
        return res;

    res = IRunningObjectTable_IsRunning(rot,iface);

    IRunningObjectTable_Release(rot);

    return res;
880 881 882 883 884
}

/******************************************************************************
 *        FileMoniker_GetTimeOfLastChange
 ******************************************************************************/
885 886 887
static HRESULT WINAPI
FileMonikerImpl_GetTimeOfLastChange(IMoniker* iface, IBindCtx* pbc,
                                    IMoniker* pmkToLeft, FILETIME* pFileTime)
888
{
889
    FileMonikerImpl *This = impl_from_IMoniker(iface);
Noomen Hamza's avatar
Noomen Hamza committed
890 891 892
    IRunningObjectTable* rot;
    HRESULT res;
    WIN32_FILE_ATTRIBUTE_DATA info;
893

894
    TRACE("(%p,%p,%p,%p)\n",iface,pbc,pmkToLeft,pFileTime);
895

Noomen Hamza's avatar
Noomen Hamza committed
896 897 898 899 900 901 902 903 904 905 906 907 908
    if (pFileTime==NULL)
        return E_POINTER;

    if (pmkToLeft!=NULL)
        return E_INVALIDARG;

    res=IBindCtx_GetRunningObjectTable(pbc,&rot);

    if (FAILED(res))
        return res;

    res= IRunningObjectTable_GetTimeOfLastChange(rot,iface,pFileTime);

909
    if (FAILED(res)){ /* the moniker is not registered */
Noomen Hamza's avatar
Noomen Hamza committed
910 911 912

        if (!GetFileAttributesExW(This->filePathName,GetFileExInfoStandard,&info))
            return MK_E_NOOBJECT;
913

Noomen Hamza's avatar
Noomen Hamza committed
914
        *pFileTime=info.ftLastWriteTime;
915
    }
Noomen Hamza's avatar
Noomen Hamza committed
916 917

    return S_OK;
918 919 920 921
}

/******************************************************************************
 *        FileMoniker_Inverse
922 923 924
 */
static HRESULT WINAPI
FileMonikerImpl_Inverse(IMoniker* iface,IMoniker** ppmk)
925
{
926
    TRACE("(%p,%p)\n",iface,ppmk);
927

Noomen Hamza's avatar
Noomen Hamza committed
928
    return CreateAntiMoniker(ppmk);
929 930 931 932
}

/******************************************************************************
 *        FileMoniker_CommonPrefixWith
933 934 935
 */
static HRESULT WINAPI
FileMonikerImpl_CommonPrefixWith(IMoniker* iface,IMoniker* pmkOther,IMoniker** ppmkPrefix)
936
{
937

Noomen Hamza's avatar
Noomen Hamza committed
938 939 940 941
    LPOLESTR pathThis,pathOther,*stringTable1,*stringTable2,commonPath;
    IBindCtx *pbind;
    DWORD mkSys;
    ULONG nb1,nb2,i,sameIdx;
942
    BOOL machineNameCase = FALSE;
943

Noomen Hamza's avatar
Noomen Hamza committed
944 945 946 947 948
    if (ppmkPrefix==NULL)
        return E_POINTER;

    if (pmkOther==NULL)
        return E_INVALIDARG;
949

Noomen Hamza's avatar
Noomen Hamza committed
950
    *ppmkPrefix=0;
951

Noomen Hamza's avatar
Noomen Hamza committed
952 953 954 955
    /* check if we have the same type of moniker */
    IMoniker_IsSystemMoniker(pmkOther,&mkSys);

    if(mkSys==MKSYS_FILEMONIKER){
956
        HRESULT ret;
Noomen Hamza's avatar
Noomen Hamza committed
957

958 959 960
        ret = CreateBindCtx(0,&pbind);
        if (FAILED(ret))
            return ret;
Noomen Hamza's avatar
Noomen Hamza committed
961 962 963

        /* create a string based on common part of the two paths */

964 965 966 967 968 969
        ret = IMoniker_GetDisplayName(iface,pbind,NULL,&pathThis);
        if (FAILED(ret))
            return ret;
        ret = IMoniker_GetDisplayName(pmkOther,pbind,NULL,&pathOther);
        if (FAILED(ret))
            return ret;
Noomen Hamza's avatar
Noomen Hamza committed
970 971

        nb1=FileMonikerImpl_DecomposePath(pathThis,&stringTable1);
972 973
        if (FAILED(nb1))
            return nb1;
Noomen Hamza's avatar
Noomen Hamza committed
974
        nb2=FileMonikerImpl_DecomposePath(pathOther,&stringTable2);
975
        if (FAILED(nb2))
976 977
        {
            free_stringtable(stringTable1);
978
            return nb2;
979
        }
Noomen Hamza's avatar
Noomen Hamza committed
980 981

        if (nb1==0 || nb2==0)
982 983 984
        {
            free_stringtable(stringTable1);
            free_stringtable(stringTable2);
Noomen Hamza's avatar
Noomen Hamza committed
985
            return MK_E_NOPREFIX;
986
        }
Noomen Hamza's avatar
Noomen Hamza committed
987 988

        commonPath=HeapAlloc(GetProcessHeap(),0,sizeof(WCHAR)*(min(lstrlenW(pathThis),lstrlenW(pathOther))+1));
989 990
        if (!commonPath)
            return E_OUTOFMEMORY;
Noomen Hamza's avatar
Noomen Hamza committed
991 992

        *commonPath=0;
993

Noomen Hamza's avatar
Noomen Hamza committed
994 995 996 997 998 999
        for(sameIdx=0; ( (stringTable1[sameIdx]!=NULL) &&
                         (stringTable2[sameIdx]!=NULL) &&
                         (lstrcmpiW(stringTable1[sameIdx],stringTable2[sameIdx])==0)); sameIdx++);

        if (sameIdx > 1 && *stringTable1[0]=='\\' && *stringTable2[1]=='\\'){

1000
            machineNameCase = TRUE;
Noomen Hamza's avatar
Noomen Hamza committed
1001 1002 1003 1004

            for(i=2;i<sameIdx;i++)

                if( (*stringTable1[i]=='\\') && (i+1 < sameIdx) && (*stringTable1[i+1]=='\\') ){
1005
                    machineNameCase = FALSE;
Noomen Hamza's avatar
Noomen Hamza committed
1006
                    break;
1007
            }
Noomen Hamza's avatar
Noomen Hamza committed
1008 1009
        }

1010
        if (machineNameCase && *stringTable1[sameIdx-1]=='\\')
Noomen Hamza's avatar
Noomen Hamza committed
1011
            sameIdx--;
1012

1013
        if (machineNameCase && (sameIdx<=3) && (nb1 > 3 || nb2 > 3) )
1014 1015 1016 1017 1018
            ret = MK_E_NOPREFIX;
        else
        {
            for(i=0;i<sameIdx;i++)
                strcatW(commonPath,stringTable1[i]);
1019 1020 1021

            free_stringtable(stringTable1);
            free_stringtable(stringTable2);
1022 1023
            ret = CreateFileMoniker(commonPath,ppmkPrefix);
        }
Noomen Hamza's avatar
Noomen Hamza committed
1024
        HeapFree(GetProcessHeap(),0,commonPath);
1025
        return ret;
Noomen Hamza's avatar
Noomen Hamza committed
1026 1027 1028 1029 1030 1031 1032
    }
    else
        return MonikerCommonPrefixWith(iface,pmkOther,ppmkPrefix);
}

/******************************************************************************
 *        DecomposePath (local function)
1033 1034
 */
int FileMonikerImpl_DecomposePath(LPCOLESTR str, LPOLESTR** stringTable)
Noomen Hamza's avatar
Noomen Hamza committed
1035
{
1036
    static const WCHAR bSlash[] = {'\\',0};
1037 1038
    LPOLESTR word;
    int i=0,j,tabIndex=0, ret=0;
Noomen Hamza's avatar
Noomen Hamza committed
1039 1040 1041 1042
    LPOLESTR *strgtable ;

    int len=lstrlenW(str);

1043 1044
    TRACE("%s, %p\n", debugstr_w(str), *stringTable);

1045
    strgtable = CoTaskMemAlloc((len + 1)*sizeof(*strgtable));
1046

Noomen Hamza's avatar
Noomen Hamza committed
1047 1048
    if (strgtable==NULL)
	return E_OUTOFMEMORY;
1049

1050 1051 1052 1053 1054 1055 1056 1057
    word = CoTaskMemAlloc((len + 1)*sizeof(WCHAR));

    if (word==NULL)
    {
        ret = E_OUTOFMEMORY;
        goto lend;
    }

Noomen Hamza's avatar
Noomen Hamza committed
1058 1059 1060 1061 1062 1063 1064
    while(str[i]!=0){

        if(str[i]==bSlash[0]){

            strgtable[tabIndex]=CoTaskMemAlloc(2*sizeof(WCHAR));

            if (strgtable[tabIndex]==NULL)
1065 1066 1067 1068
            {
                ret = E_OUTOFMEMORY;
                goto lend;
            }
Noomen Hamza's avatar
Noomen Hamza committed
1069

1070
            strcpyW(strgtable[tabIndex++],bSlash);
Noomen Hamza's avatar
Noomen Hamza committed
1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084

            i++;

        }
        else {

            for(j=0; str[i]!=0 && str[i]!=bSlash[0] ; i++,j++)
                word[j]=str[i];

            word[j]=0;

            strgtable[tabIndex]=CoTaskMemAlloc(sizeof(WCHAR)*(j+1));

            if (strgtable[tabIndex]==NULL)
1085 1086 1087 1088
            {
                ret = E_OUTOFMEMORY;
                goto lend;
            }
Noomen Hamza's avatar
Noomen Hamza committed
1089

1090
            strcpyW(strgtable[tabIndex++],word);
Noomen Hamza's avatar
Noomen Hamza committed
1091 1092 1093
        }
    }
    strgtable[tabIndex]=NULL;
1094

Noomen Hamza's avatar
Noomen Hamza committed
1095
    *stringTable=strgtable;
1096

1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111
    ret = tabIndex;

lend:
    if (ret < 0)
    {
        for (i = 0; i < tabIndex; i++)
            CoTaskMemFree(strgtable[i]);

        CoTaskMemFree(strgtable);
    }

    if (word)
        CoTaskMemFree(word);

    return ret;
1112 1113 1114 1115
}

/******************************************************************************
 *        FileMoniker_RelativePathTo
1116 1117 1118
 */
static HRESULT WINAPI
FileMonikerImpl_RelativePathTo(IMoniker* iface,IMoniker* pmOther, IMoniker** ppmkRelPath)
1119
{
Noomen Hamza's avatar
Noomen Hamza committed
1120 1121 1122 1123
    IBindCtx *bind;
    HRESULT res;
    LPOLESTR str1=0,str2=0,*tabStr1=0,*tabStr2=0,relPath=0;
    DWORD len1=0,len2=0,sameIdx=0,j=0;
1124
    static const WCHAR back[] ={'.','.','\\',0};
1125

1126
    TRACE("(%p,%p,%p)\n",iface,pmOther,ppmkRelPath);
1127

Noomen Hamza's avatar
Noomen Hamza committed
1128 1129
    if (ppmkRelPath==NULL)
        return E_POINTER;
1130

Noomen Hamza's avatar
Noomen Hamza committed
1131 1132
    if (pmOther==NULL)
        return E_INVALIDARG;
1133

Noomen Hamza's avatar
Noomen Hamza committed
1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145
    res=CreateBindCtx(0,&bind);
    if (FAILED(res))
	return res;

    res=IMoniker_GetDisplayName(iface,bind,NULL,&str1);
    if (FAILED(res))
	return res;
    res=IMoniker_GetDisplayName(pmOther,bind,NULL,&str2);
    if (FAILED(res))
	return res;

    len1=FileMonikerImpl_DecomposePath(str1,&tabStr1);
1146 1147
    if (FAILED(len1))
        return E_OUTOFMEMORY;
Noomen Hamza's avatar
Noomen Hamza committed
1148 1149
    len2=FileMonikerImpl_DecomposePath(str2,&tabStr2);

1150 1151 1152 1153 1154
    if (FAILED(len2))
    {
        free_stringtable(tabStr1);
        return E_OUTOFMEMORY;
    }
1155

Noomen Hamza's avatar
Noomen Hamza committed
1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170
    /* count the number of similar items from the begin of the two paths */
    for(sameIdx=0; ( (tabStr1[sameIdx]!=NULL) &&
		   (tabStr2[sameIdx]!=NULL) &&
               (lstrcmpiW(tabStr1[sameIdx],tabStr2[sameIdx])==0)); sameIdx++);

    /* begin the construction of relativePath */
    /* if the two paths have a consecutive similar item from the begin ! the relativePath will be composed */
    /* by "..\\" in the begin */
    relPath=HeapAlloc(GetProcessHeap(),0,sizeof(WCHAR)*(1+lstrlenW(str1)+lstrlenW(str2)));

    *relPath=0;

    if (len2>0 && !(len1==1 && len2==1 && sameIdx==0))
        for(j=sameIdx;(tabStr1[j] != NULL); j++)
            if (*tabStr1[j]!='\\')
1171
                strcatW(relPath,back);
Noomen Hamza's avatar
Noomen Hamza committed
1172 1173 1174

    /* add items of the second path (similar items with the first path are not included) to the relativePath */
    for(j=sameIdx;tabStr2[j]!=NULL;j++)
1175
        strcatW(relPath,tabStr2[j]);
1176

Noomen Hamza's avatar
Noomen Hamza committed
1177
    res=CreateFileMoniker(relPath,ppmkRelPath);
1178

1179 1180
    free_stringtable(tabStr1);
    free_stringtable(tabStr2);
Noomen Hamza's avatar
Noomen Hamza committed
1181 1182 1183 1184 1185 1186 1187 1188
    CoTaskMemFree(str1);
    CoTaskMemFree(str2);
    HeapFree(GetProcessHeap(),0,relPath);

    if (len1==0 || len2==0 || (len1==1 && len2==1 && sameIdx==0))
        return MK_S_HIM;

    return res;
1189 1190 1191 1192
}

/******************************************************************************
 *        FileMoniker_GetDisplayName
1193 1194 1195 1196
 */
static HRESULT WINAPI
FileMonikerImpl_GetDisplayName(IMoniker* iface, IBindCtx* pbc,
                               IMoniker* pmkToLeft, LPOLESTR *ppszDisplayName)
1197
{
1198
    FileMonikerImpl *This = impl_from_IMoniker(iface);
Noomen Hamza's avatar
Noomen Hamza committed
1199
    int len=lstrlenW(This->filePathName);
1200

1201
    TRACE("(%p,%p,%p,%p)\n",iface,pbc,pmkToLeft,ppszDisplayName);
Noomen Hamza's avatar
Noomen Hamza committed
1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212

    if (ppszDisplayName==NULL)
        return E_POINTER;

    if (pmkToLeft!=NULL)
        return E_INVALIDARG;

    *ppszDisplayName=CoTaskMemAlloc(sizeof(WCHAR)*(len+1));
    if (*ppszDisplayName==NULL)
        return E_OUTOFMEMORY;

1213
    strcpyW(*ppszDisplayName,This->filePathName);
1214

Robert Shearman's avatar
Robert Shearman committed
1215 1216
    TRACE("-- %s\n", debugstr_w(*ppszDisplayName));
    
Noomen Hamza's avatar
Noomen Hamza committed
1217
    return S_OK;
1218 1219 1220 1221
}

/******************************************************************************
 *        FileMoniker_ParseDisplayName
1222 1223 1224 1225
 */
static HRESULT WINAPI
FileMonikerImpl_ParseDisplayName(IMoniker* iface, IBindCtx* pbc, IMoniker* pmkToLeft,
                     LPOLESTR pszDisplayName, ULONG* pchEaten, IMoniker** ppmkOut)
1226
{
1227
    FIXME("(%p,%p,%p,%p,%p,%p),stub!\n",iface,pbc,pmkToLeft,pszDisplayName,pchEaten,ppmkOut);
1228 1229 1230 1231
    return E_NOTIMPL;
}

/******************************************************************************
Andreas Mohr's avatar
Andreas Mohr committed
1232
 *        FileMoniker_IsSystemMoniker
1233 1234 1235
 */
static HRESULT WINAPI
FileMonikerImpl_IsSystemMoniker(IMoniker* iface,DWORD* pwdMksys)
1236
{
1237
    TRACE("(%p,%p)\n",iface,pwdMksys);
Noomen Hamza's avatar
Noomen Hamza committed
1238 1239 1240

    if (!pwdMksys)
        return E_POINTER;
1241

Noomen Hamza's avatar
Noomen Hamza committed
1242 1243 1244 1245 1246 1247 1248
    (*pwdMksys)=MKSYS_FILEMONIKER;

    return S_OK;
}

/*******************************************************************************
 *        FileMonikerIROTData_QueryInterface
1249 1250 1251
 */
static HRESULT WINAPI
FileMonikerROTDataImpl_QueryInterface(IROTData *iface,REFIID riid,VOID** ppvObject)
Noomen Hamza's avatar
Noomen Hamza committed
1252 1253
{

1254
    FileMonikerImpl *This = impl_from_IROTData(iface);
Noomen Hamza's avatar
Noomen Hamza committed
1255

Robert Shearman's avatar
Robert Shearman committed
1256
    TRACE("(%p,%s,%p)\n",This,debugstr_guid(riid),ppvObject);
Noomen Hamza's avatar
Noomen Hamza committed
1257

1258
    return FileMonikerImpl_QueryInterface(&This->IMoniker_iface, riid, ppvObject);
Noomen Hamza's avatar
Noomen Hamza committed
1259 1260 1261 1262 1263
}

/***********************************************************************
 *        FileMonikerIROTData_AddRef
 */
1264 1265
static ULONG WINAPI
FileMonikerROTDataImpl_AddRef(IROTData *iface)
Noomen Hamza's avatar
Noomen Hamza committed
1266
{
1267
    FileMonikerImpl *This = impl_from_IROTData(iface);
1268

1269
    TRACE("(%p)\n",This);
Noomen Hamza's avatar
Noomen Hamza committed
1270

1271
    return IMoniker_AddRef(&This->IMoniker_iface);
Noomen Hamza's avatar
Noomen Hamza committed
1272 1273 1274 1275 1276
}

/***********************************************************************
 *        FileMonikerIROTData_Release
 */
1277 1278
static ULONG WINAPI
FileMonikerROTDataImpl_Release(IROTData* iface)
Noomen Hamza's avatar
Noomen Hamza committed
1279
{
1280
    FileMonikerImpl *This = impl_from_IROTData(iface);
1281

1282
    TRACE("(%p)\n",This);
Noomen Hamza's avatar
Noomen Hamza committed
1283

1284
    return FileMonikerImpl_Release(&This->IMoniker_iface);
Noomen Hamza's avatar
Noomen Hamza committed
1285
}
1286

Noomen Hamza's avatar
Noomen Hamza committed
1287
/******************************************************************************
1288
 *        FileMonikerIROTData_GetComparisonData
1289 1290
 */
static HRESULT WINAPI
1291
FileMonikerROTDataImpl_GetComparisonData(IROTData* iface, BYTE* pbData,
1292
                                          ULONG cbMax, ULONG* pcbData)
Noomen Hamza's avatar
Noomen Hamza committed
1293
{
1294 1295
    FileMonikerImpl *This = impl_from_IROTData(iface);
    int len = strlenW(This->filePathName)+1;
1296 1297 1298
    int i;
    LPWSTR pszFileName;

1299
    TRACE("(%p, %u, %p)\n", pbData, cbMax, pcbData);
1300 1301 1302 1303 1304 1305 1306 1307

    *pcbData = sizeof(CLSID) + len * sizeof(WCHAR);
    if (cbMax < *pcbData)
        return E_OUTOFMEMORY;

    memcpy(pbData, &CLSID_FileMoniker, sizeof(CLSID));
    pszFileName = (LPWSTR)(pbData+sizeof(CLSID));
    for (i = 0; i < len; i++)
1308
        pszFileName[i] = toupperW(This->filePathName[i]);
1309 1310

    return S_OK;
1311 1312
}

1313 1314 1315 1316
/*
 * Virtual function table for the FileMonikerImpl class which include IPersist,
 * IPersistStream and IMoniker functions.
 */
1317
static const IMonikerVtbl VT_FileMonikerImpl =
1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344
{
    FileMonikerImpl_QueryInterface,
    FileMonikerImpl_AddRef,
    FileMonikerImpl_Release,
    FileMonikerImpl_GetClassID,
    FileMonikerImpl_IsDirty,
    FileMonikerImpl_Load,
    FileMonikerImpl_Save,
    FileMonikerImpl_GetSizeMax,
    FileMonikerImpl_BindToObject,
    FileMonikerImpl_BindToStorage,
    FileMonikerImpl_Reduce,
    FileMonikerImpl_ComposeWith,
    FileMonikerImpl_Enum,
    FileMonikerImpl_IsEqual,
    FileMonikerImpl_Hash,
    FileMonikerImpl_IsRunning,
    FileMonikerImpl_GetTimeOfLastChange,
    FileMonikerImpl_Inverse,
    FileMonikerImpl_CommonPrefixWith,
    FileMonikerImpl_RelativePathTo,
    FileMonikerImpl_GetDisplayName,
    FileMonikerImpl_ParseDisplayName,
    FileMonikerImpl_IsSystemMoniker
};

/* Virtual function table for the IROTData class. */
1345
static const IROTDataVtbl VT_ROTDataImpl =
1346 1347 1348 1349
{
    FileMonikerROTDataImpl_QueryInterface,
    FileMonikerROTDataImpl_AddRef,
    FileMonikerROTDataImpl_Release,
1350
    FileMonikerROTDataImpl_GetComparisonData
1351 1352 1353 1354 1355
};

/******************************************************************************
 *         FileMoniker_Construct (local function)
 */
1356
static HRESULT FileMonikerImpl_Construct(FileMonikerImpl* This, LPCOLESTR lpszPathName)
1357 1358 1359 1360 1361 1362
{
    int nb=0,i;
    int sizeStr=lstrlenW(lpszPathName);
    LPOLESTR *tabStr=0;
    static const WCHAR twoPoint[]={'.','.',0};
    static const WCHAR bkSlash[]={'\\',0};
1363
    BOOL addBkSlash;
1364 1365 1366

    TRACE("(%p,%s)\n",This,debugstr_w(lpszPathName));

Austin English's avatar
Austin English committed
1367
    /* Initialize the virtual function table. */
1368 1369
    This->IMoniker_iface.lpVtbl = &VT_FileMonikerImpl;
    This->IROTData_iface.lpVtbl = &VT_ROTDataImpl;
1370
    This->ref          = 0;
1371
    This->pMarshal     = NULL;
1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383

    This->filePathName=HeapAlloc(GetProcessHeap(),0,sizeof(WCHAR)*(sizeStr+1));

    if (This->filePathName==NULL)
        return E_OUTOFMEMORY;

    strcpyW(This->filePathName,lpszPathName);

    nb=FileMonikerImpl_DecomposePath(This->filePathName,&tabStr);

    if (nb > 0 ){

1384
        addBkSlash = TRUE;
1385
        if (lstrcmpW(tabStr[0],twoPoint)!=0)
1386
            addBkSlash = FALSE;
1387 1388 1389 1390
        else
            for(i=0;i<nb;i++){

                if ( (lstrcmpW(tabStr[i],twoPoint)!=0) && (lstrcmpW(tabStr[i],bkSlash)!=0) ){
1391
                    addBkSlash = FALSE;
1392 1393 1394 1395 1396 1397 1398
                    break;
                }
                else

                    if (lstrcmpW(tabStr[i],bkSlash)==0 && i<nb-1 && lstrcmpW(tabStr[i+1],bkSlash)==0){
                        *tabStr[i]=0;
                        sizeStr--;
1399
                        addBkSlash = FALSE;
1400 1401 1402 1403 1404
                        break;
                    }
            }

        if (lstrcmpW(tabStr[nb-1],bkSlash)==0)
1405
            addBkSlash = FALSE;
1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417

        This->filePathName=HeapReAlloc(GetProcessHeap(),0,This->filePathName,(sizeStr+1)*sizeof(WCHAR));

        *This->filePathName=0;

        for(i=0;tabStr[i]!=NULL;i++)
            strcatW(This->filePathName,tabStr[i]);

        if (addBkSlash)
            strcatW(This->filePathName,bkSlash);
    }

1418
    free_stringtable(tabStr);
1419 1420 1421 1422

    return S_OK;
}

1423
/******************************************************************************
1424
 *        CreateFileMoniker (OLE32.@)
1425
 ******************************************************************************/
1426
HRESULT WINAPI CreateFileMoniker(LPCOLESTR lpszPathName, IMoniker **ppmk)
1427
{
Robert Shearman's avatar
Robert Shearman committed
1428 1429
    FileMonikerImpl* newFileMoniker;
    HRESULT  hr;
1430

Robert Shearman's avatar
Robert Shearman committed
1431
    TRACE("(%s,%p)\n",debugstr_w(lpszPathName),ppmk);
1432

Robert Shearman's avatar
Robert Shearman committed
1433
    if (!ppmk)
Noomen Hamza's avatar
Noomen Hamza committed
1434 1435
        return E_POINTER;

Robert Shearman's avatar
Robert Shearman committed
1436
    if(!lpszPathName)
Noomen Hamza's avatar
Noomen Hamza committed
1437
        return MK_E_SYNTAX;
1438

Robert Shearman's avatar
Robert Shearman committed
1439
    *ppmk=NULL;
1440

1441 1442
    newFileMoniker = HeapAlloc(GetProcessHeap(), 0, sizeof(FileMonikerImpl));

Robert Shearman's avatar
Robert Shearman committed
1443
    if (!newFileMoniker)
Noomen Hamza's avatar
Noomen Hamza committed
1444
        return E_OUTOFMEMORY;
1445 1446 1447

    hr = FileMonikerImpl_Construct(newFileMoniker,lpszPathName);

Noomen Hamza's avatar
Noomen Hamza committed
1448
    if (SUCCEEDED(hr))
1449
        hr = IMoniker_QueryInterface(&newFileMoniker->IMoniker_iface,&IID_IMoniker,(void**)ppmk);
Noomen Hamza's avatar
Noomen Hamza committed
1450 1451
    else
        HeapFree(GetProcessHeap(),0,newFileMoniker);
1452 1453 1454

    return hr;
}
1455

1456 1457 1458 1459 1460 1461 1462 1463 1464
/* find a character from a set in reverse without the string having to be null-terminated */
static inline WCHAR *memrpbrkW(const WCHAR *ptr, size_t n, const WCHAR *accept)
{
    const WCHAR *end, *ret = NULL;
    for (end = ptr + n; ptr < end; ptr++) if (strchrW(accept, *ptr)) ret = ptr;
    return (WCHAR *)ret;
}

HRESULT FileMoniker_CreateFromDisplayName(LPBC pbc, LPCOLESTR szDisplayName,
1465
                                          LPDWORD pchEaten, IMoniker **ppmk)
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 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554
{
    LPCWSTR end;
    static const WCHAR wszSeparators[] = {':','\\','/','!',0};

    for (end = szDisplayName + strlenW(szDisplayName);
         end && (end != szDisplayName);
         end = memrpbrkW(szDisplayName, end - szDisplayName, wszSeparators))
    {
        HRESULT hr;
        IRunningObjectTable *rot;
        IMoniker *file_moniker;
        LPWSTR file_display_name;
        LPWSTR full_path_name;
        DWORD full_path_name_len;
        int len = end - szDisplayName;

        file_display_name = HeapAlloc(GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR));
        if (!file_display_name) return E_OUTOFMEMORY;
        memcpy(file_display_name, szDisplayName, len * sizeof(WCHAR));
        file_display_name[len] = '\0';

        hr = CreateFileMoniker(file_display_name, &file_moniker);
        if (FAILED(hr))
        {
            HeapFree(GetProcessHeap(), 0, file_display_name);
            return hr;
        }

        hr = IBindCtx_GetRunningObjectTable(pbc, &rot);
        if (FAILED(hr))
        {
            HeapFree(GetProcessHeap(), 0, file_display_name);
            IMoniker_Release(file_moniker);
            return hr;
        }

        hr = IRunningObjectTable_IsRunning(rot, file_moniker);
        IRunningObjectTable_Release(rot);
        if (FAILED(hr))
        {
            HeapFree(GetProcessHeap(), 0, file_display_name);
            IMoniker_Release(file_moniker);
            return hr;
        }
        if (hr == S_OK)
        {
            TRACE("found running file moniker for %s\n", debugstr_w(file_display_name));
            *pchEaten = len;
            *ppmk = file_moniker;
            HeapFree(GetProcessHeap(), 0, file_display_name);
            return S_OK;
        }

        full_path_name_len = GetFullPathNameW(file_display_name, 0, NULL, NULL);
        if (!full_path_name_len)
        {
            HeapFree(GetProcessHeap(), 0, file_display_name);
            IMoniker_Release(file_moniker);
            return MK_E_SYNTAX;
        }
        full_path_name = HeapAlloc(GetProcessHeap(), 0, full_path_name_len * sizeof(WCHAR));
        if (!full_path_name)
        {
            HeapFree(GetProcessHeap(), 0, file_display_name);
            IMoniker_Release(file_moniker);
            return E_OUTOFMEMORY;
        }
        GetFullPathNameW(file_display_name, full_path_name_len, full_path_name, NULL);

        if (GetFileAttributesW(full_path_name) == INVALID_FILE_ATTRIBUTES)
            TRACE("couldn't open file %s\n", debugstr_w(full_path_name));
        else
        {
            TRACE("got file moniker for %s\n", debugstr_w(szDisplayName));
            *pchEaten = len;
            *ppmk = file_moniker;
            HeapFree(GetProcessHeap(), 0, file_display_name);
            HeapFree(GetProcessHeap(), 0, full_path_name);
            return S_OK;
        }
        HeapFree(GetProcessHeap(), 0, file_display_name);
        HeapFree(GetProcessHeap(), 0, full_path_name);
        IMoniker_Release(file_moniker);
    }

    return MK_E_CANTOPENFILE;
}


1555 1556 1557 1558 1559 1560 1561
static HRESULT WINAPI FileMonikerCF_QueryInterface(LPCLASSFACTORY iface,
                                                  REFIID riid, LPVOID *ppv)
{
    *ppv = NULL;
    if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IClassFactory))
    {
        *ppv = iface;
1562
        IClassFactory_AddRef(iface);
1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598
        return S_OK;
    }
    return E_NOINTERFACE;
}

static ULONG WINAPI FileMonikerCF_AddRef(LPCLASSFACTORY iface)
{
    return 2; /* non-heap based object */
}

static ULONG WINAPI FileMonikerCF_Release(LPCLASSFACTORY iface)
{
    return 1; /* non-heap based object */
}

static HRESULT WINAPI FileMonikerCF_CreateInstance(LPCLASSFACTORY iface,
    LPUNKNOWN pUnk, REFIID riid, LPVOID *ppv)
{
    FileMonikerImpl* newFileMoniker;
    HRESULT  hr;
    static const WCHAR wszEmpty[] = { 0 };

    TRACE("(%p, %s, %p)\n", pUnk, debugstr_guid(riid), ppv);

    *ppv = NULL;

    if (pUnk)
        return CLASS_E_NOAGGREGATION;

    newFileMoniker = HeapAlloc(GetProcessHeap(), 0, sizeof(FileMonikerImpl));
    if (!newFileMoniker)
        return E_OUTOFMEMORY;

    hr = FileMonikerImpl_Construct(newFileMoniker, wszEmpty);

    if (SUCCEEDED(hr))
1599
        hr = IMoniker_QueryInterface(&newFileMoniker->IMoniker_iface, riid, ppv);
1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625
    if (FAILED(hr))
        HeapFree(GetProcessHeap(),0,newFileMoniker);

    return hr;
}

static HRESULT WINAPI FileMonikerCF_LockServer(LPCLASSFACTORY iface, BOOL fLock)
{
    FIXME("(%d), stub!\n",fLock);
    return S_OK;
}

static const IClassFactoryVtbl FileMonikerCFVtbl =
{
    FileMonikerCF_QueryInterface,
    FileMonikerCF_AddRef,
    FileMonikerCF_Release,
    FileMonikerCF_CreateInstance,
    FileMonikerCF_LockServer
};
static const IClassFactoryVtbl *FileMonikerCF = &FileMonikerCFVtbl;

HRESULT FileMonikerCF_Create(REFIID riid, LPVOID *ppv)
{
    return IClassFactory_QueryInterface((IClassFactory *)&FileMonikerCF, riid, ppv);
}