shell32: Reject NULL path in IShellLink_SetPath.
[wine] / dlls / shell32 / shlfsbind.c
1 /*
2  * File System Bind Data object to use as parameter for the bind context to
3  * IShellFolder_ParseDisplayName
4  *
5  * Copyright 2003 Rolf Kalbermatter
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * 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 "config.h"
23 #include "wine/port.h"
24
25 #include <stdarg.h>
26
27 #define COBJMACROS
28
29 #include "windef.h"
30 #include "winbase.h"
31 #include "winuser.h"
32 #include "shlobj.h"
33 #include "shell32_main.h"
34
35 #include "wine/debug.h"
36
37 WINE_DEFAULT_DEBUG_CHANNEL(pidl);
38
39 /***********************************************************************
40  * IFileSystemBindData implementation
41  */
42 typedef struct
43 {
44     const IFileSystemBindDataVtbl *lpVtbl;
45     LONG              ref;
46     WIN32_FIND_DATAW findFile;
47 } IFileSystemBindDataImpl;
48
49 static HRESULT WINAPI IFileSystemBindData_fnQueryInterface(IFileSystemBindData *, REFIID, LPVOID*);
50 static ULONG WINAPI IFileSystemBindData_fnAddRef(IFileSystemBindData *);
51 static ULONG WINAPI IFileSystemBindData_fnRelease(IFileSystemBindData *);
52 static HRESULT WINAPI IFileSystemBindData_fnGetFindData(IFileSystemBindData *, WIN32_FIND_DATAW *);
53 static HRESULT WINAPI IFileSystemBindData_fnSetFindData(IFileSystemBindData *, const WIN32_FIND_DATAW *);
54
55 static const IFileSystemBindDataVtbl sbvt =
56 {
57     IFileSystemBindData_fnQueryInterface,
58     IFileSystemBindData_fnAddRef,
59     IFileSystemBindData_fnRelease,
60     IFileSystemBindData_fnSetFindData,
61     IFileSystemBindData_fnGetFindData,
62 };
63
64 static const WCHAR wFileSystemBindData[] = {
65     'F','i','l','e',' ','S','y','s','t','e','m',' ','B','i','n','d','D','a','t','a',0};
66
67 HRESULT WINAPI IFileSystemBindData_Constructor(const WIN32_FIND_DATAW *pfd, LPBC *ppV)
68 {
69     IFileSystemBindDataImpl *sb;
70     HRESULT ret = E_OUTOFMEMORY;
71
72     TRACE("%p, %p\n", pfd, ppV);
73
74     if (!ppV)
75        return E_INVALIDARG;
76
77     *ppV = NULL;
78
79     sb = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IFileSystemBindDataImpl));
80     if (!sb)
81         return ret;
82
83     sb->lpVtbl = &sbvt;
84     sb->ref = 1;
85     IFileSystemBindData_fnSetFindData((IFileSystemBindData*)sb, pfd);
86
87     ret = CreateBindCtx(0, ppV);
88     if (SUCCEEDED(ret))
89     {
90         BIND_OPTS bindOpts;
91
92         bindOpts.cbStruct = sizeof(BIND_OPTS);
93         bindOpts.grfFlags = 0;
94         bindOpts.grfMode = STGM_CREATE;
95         bindOpts.dwTickCountDeadline = 0;
96         IBindCtx_SetBindOptions(*ppV, &bindOpts);
97         IBindCtx_RegisterObjectParam(*ppV, (LPOLESTR)wFileSystemBindData, (LPUNKNOWN)sb);
98
99         IFileSystemBindData_Release((IFileSystemBindData*)sb);
100     }
101     else
102         HeapFree(GetProcessHeap(), 0, sb);
103     return ret;
104 }
105
106 static HRESULT WINAPI IFileSystemBindData_fnQueryInterface(
107                 IFileSystemBindData *iface, REFIID riid, LPVOID *ppV)
108 {
109     IFileSystemBindDataImpl *This = (IFileSystemBindDataImpl *)iface;
110     TRACE("(%p)->(\n\tIID:\t%s, %p)\n", This, debugstr_guid(riid), ppV);
111
112     *ppV = NULL;
113
114     if (IsEqualIID(riid, &IID_IUnknown))
115         *ppV = This;
116     else if (IsEqualIID(riid, &IID_IFileSystemBindData))
117         *ppV = This;
118
119     if (*ppV)
120     {
121         IUnknown_AddRef((IUnknown*)(*ppV));
122         TRACE("-- Interface: (%p)->(%p)\n", ppV, *ppV);
123         return S_OK;
124     }
125     TRACE("-- Interface: E_NOINTERFACE\n");
126     return E_NOINTERFACE;
127 }
128
129 static ULONG WINAPI IFileSystemBindData_fnAddRef(IFileSystemBindData *iface)
130 {
131     IFileSystemBindDataImpl *This = (IFileSystemBindDataImpl *)iface;
132     ULONG refCount = InterlockedIncrement(&This->ref);
133
134     TRACE("(%p)->(count=%i)\n", This, refCount - 1);
135
136     return refCount;
137 }
138
139 static ULONG WINAPI IFileSystemBindData_fnRelease(IFileSystemBindData *iface)
140 {
141     IFileSystemBindDataImpl *This = (IFileSystemBindDataImpl *)iface;
142     ULONG refCount = InterlockedDecrement(&This->ref);
143     
144     TRACE("(%p)->(count=%i)\n", This, refCount + 1);
145
146     if (!refCount)
147     {
148         TRACE(" destroying ISFBindPidl(%p)\n",This);
149         HeapFree(GetProcessHeap(), 0, This);
150     }
151     return refCount;
152 }
153
154 static HRESULT WINAPI IFileSystemBindData_fnGetFindData(
155                IFileSystemBindData *iface, WIN32_FIND_DATAW *pfd)
156 {
157     IFileSystemBindDataImpl *This = (IFileSystemBindDataImpl *)iface;
158     TRACE("(%p), %p\n", This, pfd);
159
160     if (!pfd)
161         return E_INVALIDARG;
162
163     *pfd = This->findFile;
164     return NOERROR;
165 }
166
167 static HRESULT WINAPI IFileSystemBindData_fnSetFindData(
168                IFileSystemBindData *iface, const WIN32_FIND_DATAW *pfd)
169 {
170     IFileSystemBindDataImpl *This = (IFileSystemBindDataImpl *)iface;
171     TRACE("(%p), %p\n", This, pfd);
172
173     if (pfd)
174         This->findFile = *pfd;
175     else
176         memset(&This->findFile, 0, sizeof(WIN32_FIND_DATAW));
177     return NOERROR;
178 }