dlls/hlink/extserv.c

   1 /*
   2  * Copyright 2007 Jacek Caban for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include "hlink_private.h"
  20
  21 #include "wine/debug.h"
  22
  23 WINE_DEFAULT_DEBUG_CHANNEL(hlink);
  24
  25 typedef struct {
  26     IUnknown           IUnknown_inner;
  27     IAuthenticate      IAuthenticate_iface;
  28     IHttpNegotiate     IHttpNegotiate_iface;
  29     IExtensionServices IExtensionServices_iface;
  30
  31     IUnknown *outer_unk;
  32     LONG ref;
  33
  34     HWND hwnd;
  35     LPWSTR username;
  36     LPWSTR password;
  37     LPWSTR headers;
  38 } ExtensionService;
  39
  40 static inline ExtensionService *impl_from_IUnknown(IUnknown *iface)
  41 {
  42     return CONTAINING_RECORD(iface, ExtensionService, IUnknown_inner);
  43 }
  44
  45 static HRESULT WINAPI ExtServUnk_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
  46 {
  47     ExtensionService *This = impl_from_IUnknown(iface);
  48
  49     *ppv = NULL;
  50
  51     if(IsEqualGUID(&IID_IUnknown, riid)) {
  52         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
  53         *ppv = &This->IUnknown_inner;
  54     }else if(IsEqualGUID(&IID_IAuthenticate, riid)) {
  55         TRACE("(%p)->(IID_IAuthenticate %p)\n", This, ppv);
  56         *ppv = &This->IAuthenticate_iface;
  57     }else if(IsEqualGUID(&IID_IHttpNegotiate, riid)) {
  58         TRACE("(%p)->(IID_IHttpNegotiate %p)\n", This, ppv);
  59         *ppv = &This->IHttpNegotiate_iface;
  60     }else if(IsEqualGUID(&IID_IExtensionServices, riid)) {
  61         TRACE("(%p)->(IID_IExtensionServices %p)\n", This, ppv);
  62         *ppv = &This->IExtensionServices_iface;
  63     }
  64
  65     if(*ppv) {
  66         IUnknown_AddRef((IUnknown*)*ppv);
  67         return S_OK;
  68     }
  69
  70     FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
  71     return E_NOINTERFACE;
  72 }
  73
  74 static ULONG WINAPI ExtServUnk_AddRef(IUnknown *iface)
  75 {
  76     ExtensionService *This = impl_from_IUnknown(iface);
  77     LONG ref = InterlockedIncrement(&This->ref);
  78
  79     TRACE("(%p) ref=%d\n", This, ref);
  80
  81     return ref;
  82 }
  83
  84 static ULONG WINAPI ExtServUnk_Release(IUnknown *iface)
  85 {
  86     ExtensionService *This = impl_from_IUnknown(iface);
  87     LONG ref = InterlockedDecrement(&This->ref);
  88
  89     TRACE("(%p) ref=%d\n", This, ref);
  90
  91     if(!ref) {
  92         heap_free(This->username);
  93         heap_free(This->password);
  94         heap_free(This->headers);
  95         heap_free(This);
  96     }
  97
  98     return ref;
  99 }
 100
 101 static const IUnknownVtbl ExtServUnkVtbl = {
 102     ExtServUnk_QueryInterface,
 103     ExtServUnk_AddRef,
 104     ExtServUnk_Release
 105 };
 106
 107 static inline ExtensionService *impl_from_IAuthenticate(IAuthenticate *iface)
 108 {
 109     return CONTAINING_RECORD(iface, ExtensionService, IAuthenticate_iface);
 110 }
 111
 112 static HRESULT WINAPI Authenticate_QueryInterface(IAuthenticate *iface, REFIID riid, void **ppv)
 113 {
 114     ExtensionService *This = impl_from_IAuthenticate(iface);
 115     return IUnknown_QueryInterface(This->outer_unk, riid, ppv);
 116 }
 117
 118 static ULONG WINAPI Authenticate_AddRef(IAuthenticate *iface)
 119 {
 120     ExtensionService *This = impl_from_IAuthenticate(iface);
 121     return IUnknown_AddRef(This->outer_unk);
 122 }
 123
 124 static ULONG WINAPI Authenticate_Release(IAuthenticate *iface)
 125 {
 126     ExtensionService *This = impl_from_IAuthenticate(iface);
 127     return IUnknown_Release(This->outer_unk);
 128 }
 129
 130 static HRESULT WINAPI Authenticate_Authenticate(IAuthenticate *iface,
 131         HWND *phwnd, LPWSTR *pszUsername, LPWSTR *pszPassword)
 132 {
 133     ExtensionService *This = impl_from_IAuthenticate(iface);
 134
 135     TRACE("(%p)->(%p %p %p)\n", This, phwnd, pszUsername, pszPassword);
 136
 137     if(!phwnd || !pszUsername || !pszPassword)
 138         return E_INVALIDARG;
 139
 140     *phwnd = This->hwnd;
 141     *pszUsername = hlink_co_strdupW(This->username);
 142     *pszPassword = hlink_co_strdupW(This->password);
 143
 144     return S_OK;
 145 }
 146
 147 static const IAuthenticateVtbl AuthenticateVtbl = {
 148     Authenticate_QueryInterface,
 149     Authenticate_AddRef,
 150     Authenticate_Release,
 151     Authenticate_Authenticate
 152 };
 153
 154 static inline ExtensionService *impl_from_IHttpNegotiate(IHttpNegotiate *iface)
 155 {
 156     return CONTAINING_RECORD(iface, ExtensionService, IHttpNegotiate_iface);
 157 }
 158
 159 static HRESULT WINAPI HttpNegotiate_QueryInterface(IHttpNegotiate *iface, REFIID riid, void **ppv)
 160 {
 161     ExtensionService *This = impl_from_IHttpNegotiate(iface);
 162     return IUnknown_QueryInterface(This->outer_unk, riid, ppv);
 163 }
 164
 165 static ULONG WINAPI HttpNegotiate_AddRef(IHttpNegotiate *iface)
 166 {
 167     ExtensionService *This = impl_from_IHttpNegotiate(iface);
 168     return IUnknown_AddRef(This->outer_unk);
 169 }
 170
 171 static ULONG WINAPI HttpNegotiate_Release(IHttpNegotiate *iface)
 172 {
 173     ExtensionService *This = impl_from_IHttpNegotiate(iface);
 174     return IUnknown_Release(This->outer_unk);
 175 }
 176
 177 static HRESULT WINAPI HttpNegotiate_BeginningTransaction(IHttpNegotiate *iface,
 178         LPCWSTR szURL, LPCWSTR szHeaders, DWORD dwReserved, LPWSTR *pszAdditionalHeaders)
 179 {
 180     ExtensionService *This = impl_from_IHttpNegotiate(iface);
 181
 182     TRACE("(%p)->(%s %s %x %p)\n", This, debugstr_w(szURL), debugstr_w(szHeaders), dwReserved,
 183           pszAdditionalHeaders);
 184
 185     if(!pszAdditionalHeaders)
 186         return E_INVALIDARG;
 187
 188     *pszAdditionalHeaders = hlink_co_strdupW(This->headers);
 189     return S_OK;
 190 }
 191
 192 static HRESULT WINAPI HttpNegotiate_OnResponse(IHttpNegotiate *iface, DWORD dwResponseCode,
 193         LPCWSTR szResponseHeaders, LPCWSTR szRequestHeaders, LPWSTR *pszAdditionalRequestHeaders)
 194 {
 195     ExtensionService *This = impl_from_IHttpNegotiate(iface);
 196
 197     TRACE("(%p)->(%d %s %s %p)\n", This, dwResponseCode, debugstr_w(szResponseHeaders),
 198           debugstr_w(szRequestHeaders), pszAdditionalRequestHeaders);
 199
 200     *pszAdditionalRequestHeaders = NULL;
 201     return S_OK;
 202 }
 203
 204 static const IHttpNegotiateVtbl HttpNegotiateVtbl = {
 205     HttpNegotiate_QueryInterface,
 206     HttpNegotiate_AddRef,
 207     HttpNegotiate_Release,
 208     HttpNegotiate_BeginningTransaction,
 209     HttpNegotiate_OnResponse
 210 };
 211
 212 static inline ExtensionService *impl_from_IExtensionServices(IExtensionServices *iface)
 213 {
 214     return CONTAINING_RECORD(iface, ExtensionService, IExtensionServices_iface);
 215 }
 216
 217 static HRESULT WINAPI ExtServ_QueryInterface(IExtensionServices *iface, REFIID riid, void **ppv)
 218 {
 219     ExtensionService *This = impl_from_IExtensionServices(iface);
 220     return IUnknown_QueryInterface(This->outer_unk, riid, ppv);
 221 }
 222
 223 static ULONG WINAPI ExtServ_AddRef(IExtensionServices *iface)
 224 {
 225     ExtensionService *This = impl_from_IExtensionServices(iface);
 226     return IUnknown_AddRef(This->outer_unk);
 227 }
 228
 229 static ULONG WINAPI ExtServ_Release(IExtensionServices *iface)
 230 {
 231     ExtensionService *This = impl_from_IExtensionServices(iface);
 232     return IUnknown_Release(This->outer_unk);
 233 }
 234
 235 static HRESULT ExtServ_ImplSetAdditionalHeaders(ExtensionService* This, LPCWSTR pwzAdditionalHeaders)
 236 {
 237     int len;
 238
 239     heap_free(This->headers);
 240     This->headers = NULL;
 241
 242     if (!pwzAdditionalHeaders)
 243         return S_OK;
 244
 245     len = lstrlenW(pwzAdditionalHeaders);
 246
 247     if(len && pwzAdditionalHeaders[len-1] != '\n' && pwzAdditionalHeaders[len-1] != '\r') {
 248         This->headers = heap_alloc(len*sizeof(WCHAR) + sizeof(L"\r\n"));
 249         memcpy(This->headers, pwzAdditionalHeaders, len*sizeof(WCHAR));
 250         memcpy(This->headers+len, L"\r\n", sizeof(L"\r\n"));
 251     }else {
 252         This->headers = hlink_strdupW(pwzAdditionalHeaders);
 253     }
 254
 255     return S_OK;
 256 }
 257
 258 static HRESULT WINAPI ExtServ_SetAdditionalHeaders(IExtensionServices* iface, LPCWSTR pwzAdditionalHeaders)
 259 {
 260     ExtensionService *This = impl_from_IExtensionServices(iface);
 261
 262     TRACE("(%p)->(%s)\n", This, debugstr_w(pwzAdditionalHeaders));
 263
 264     return ExtServ_ImplSetAdditionalHeaders(This,pwzAdditionalHeaders);
 265 }
 266
 267 static HRESULT ExtServ_ImplSetAuthenticateData(ExtensionService* This, HWND phwnd, LPCWSTR pwzUsername, LPCWSTR pwzPassword)
 268 {
 269     heap_free(This->username);
 270     heap_free(This->password);
 271
 272     This->hwnd = phwnd;
 273     This->username = hlink_strdupW(pwzUsername);
 274     This->password = hlink_strdupW(pwzPassword);
 275
 276     return S_OK;
 277 }
 278
 279 static HRESULT WINAPI ExtServ_SetAuthenticateData(IExtensionServices* iface, HWND phwnd, LPCWSTR pwzUsername, LPCWSTR pwzPassword)
 280 {
 281     ExtensionService *This = impl_from_IExtensionServices(iface);
 282
 283     TRACE("(%p)->(%p %s %s)\n", This, phwnd, debugstr_w(pwzUsername), debugstr_w(pwzPassword));
 284
 285     return ExtServ_ImplSetAuthenticateData(This, phwnd, pwzUsername, pwzPassword);
 286 }
 287
 288 static const IExtensionServicesVtbl ExtServVtbl = {
 289     ExtServ_QueryInterface,
 290     ExtServ_AddRef,
 291     ExtServ_Release,
 292     ExtServ_SetAdditionalHeaders,
 293     ExtServ_SetAuthenticateData
 294 };
 295
 296 /***********************************************************************
 297  *             HlinkCreateExtensionServices (HLINK.@)
 298  */
 299 HRESULT WINAPI HlinkCreateExtensionServices(LPCWSTR pwzAdditionalHeaders,
 300         HWND phwnd, LPCWSTR pszUsername, LPCWSTR pszPassword,
 301         IUnknown *punkOuter, REFIID riid, void** ppv)
 302 {
 303     ExtensionService *ret;
 304     HRESULT hres = S_OK;
 305
 306     TRACE("%s %p %s %s %p %s %p\n",debugstr_w(pwzAdditionalHeaders),
 307             phwnd, debugstr_w(pszUsername), debugstr_w(pszPassword),
 308             punkOuter, debugstr_guid(riid), ppv);
 309
 310     ret = heap_alloc(sizeof(*ret));
 311
 312     ret->IUnknown_inner.lpVtbl = &ExtServUnkVtbl;
 313     ret->IAuthenticate_iface.lpVtbl = &AuthenticateVtbl;
 314     ret->IHttpNegotiate_iface.lpVtbl = &HttpNegotiateVtbl;
 315     ret->IExtensionServices_iface.lpVtbl = &ExtServVtbl;
 316     ret->ref = 1;
 317     ret->headers = NULL;
 318     ret->hwnd = NULL;
 319     ret->username = NULL;
 320     ret->password = NULL;
 321
 322     ExtServ_ImplSetAuthenticateData(ret, phwnd, pszUsername, pszPassword);
 323     ExtServ_ImplSetAdditionalHeaders(ret, pwzAdditionalHeaders);
 324
 325     if(!punkOuter) {
 326         ret->outer_unk = &ret->IUnknown_inner;
 327         hres = IUnknown_QueryInterface(&ret->IUnknown_inner, riid, ppv);
 328         IUnknown_Release(&ret->IUnknown_inner);
 329     }else if(IsEqualGUID(&IID_IUnknown, riid)) {
 330         ret->outer_unk = punkOuter;
 331         *ppv = &ret->IUnknown_inner;
 332     }else {
 333         IUnknown_Release(&ret->IUnknown_inner);
 334         hres = E_INVALIDARG;
 335     }
 336
 337     return hres;
 338 }