Wine Cross Reference
wine/dlls/urlmon/urlmon_main.h

   /*
    * Copyright 2002 Huw D M Davies for CodeWeavers
    * Copyright 2009 Jacek Caban for CodeWeavers
    *
    * 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
   * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
   */
  
  #ifndef __WINE_URLMON_MAIN_H
  #define __WINE_URLMON_MAIN_H
  
  #include <stdarg.h>
  
  #define COBJMACROS
  #define NONAMELESSUNION
  #define NONAMELESSSTRUCT
  
  #include "windef.h"
  #include "winbase.h"
  #include "winuser.h"
  #include "ole2.h"
  #include "urlmon.h"
  #include "wininet.h"
  
  #include "wine/unicode.h"
  #include "wine/list.h"
  
  extern HINSTANCE URLMON_hInstance;
  extern HRESULT SecManagerImpl_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
  extern HRESULT ZoneMgrImpl_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
  extern HRESULT StdURLMoniker_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
  extern HRESULT FileProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
  extern HRESULT HttpProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
  extern HRESULT HttpSProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
  extern HRESULT FtpProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
  extern HRESULT GopherProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
  extern HRESULT MkProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
  extern HRESULT MimeFilter_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
  
  /**********************************************************************
   * Dll lifetime tracking declaration for urlmon.dll
   */
  extern LONG URLMON_refCount;
  static inline void URLMON_LockModule(void) { InterlockedIncrement( &URLMON_refCount ); }
  static inline void URLMON_UnlockModule(void) { InterlockedDecrement( &URLMON_refCount ); }
  
  #define DEFINE_THIS2(cls,ifc,iface) ((cls*)((BYTE*)(iface)-offsetof(cls,ifc)))
  #define DEFINE_THIS(cls,ifc,iface) DEFINE_THIS2(cls,lp ## ifc ## Vtbl,iface)
  
  IInternetProtocolInfo *get_protocol_info(LPCWSTR);
  HRESULT get_protocol_handler(LPCWSTR,CLSID*,BOOL*,IClassFactory**);
  IInternetProtocol *get_mime_filter(LPCWSTR);
  BOOL is_registered_protocol(LPCWSTR);
  void register_urlmon_namespace(IClassFactory*,REFIID,LPCWSTR,BOOL);
  HINTERNET get_internet_session(IInternetBindInfo*);
  LPWSTR get_useragent(void);
  void free_session(void);
  
  HRESULT bind_to_storage(LPCWSTR url, IBindCtx *pbc, REFIID riid, void **ppv);
  HRESULT bind_to_object(IMoniker *mon, LPCWSTR url, IBindCtx *pbc, REFIID riid, void **ppv);
  
  HRESULT create_binding_protocol(LPCWSTR url, BOOL from_urlmon, IInternetProtocol **protocol);
  void set_binding_sink(IInternetProtocol *bind_protocol, IInternetProtocolSink *sink);
  IWinInetInfo *get_wininet_info(IInternetProtocol*);
  
  typedef struct ProtocolVtbl ProtocolVtbl;
  
  typedef struct {
      const ProtocolVtbl *vtbl;
  
      IInternetProtocol *protocol;
      IInternetProtocolSink *protocol_sink;
  
      DWORD bindf;
      BINDINFO bind_info;
  
      HINTERNET request;
      HINTERNET connection;
      DWORD flags;
      HANDLE lock;
  
      ULONG current_position;
      ULONG content_length;
      ULONG available_bytes;
  
      LONG priority;
  } Protocol;
  
 struct ProtocolVtbl {
     HRESULT (*open_request)(Protocol*,LPCWSTR,DWORD,HINTERNET,IInternetBindInfo*);
     HRESULT (*start_downloading)(Protocol*);
     void (*close_connection)(Protocol*);
 };
 
 HRESULT protocol_start(Protocol*,IInternetProtocol*,LPCWSTR,IInternetProtocolSink*,IInternetBindInfo*);
 HRESULT protocol_continue(Protocol*,PROTOCOLDATA*);
 HRESULT protocol_read(Protocol*,void*,ULONG,ULONG*);
 HRESULT protocol_lock_request(Protocol*);
 HRESULT protocol_unlock_request(Protocol*);
 void protocol_close_connection(Protocol*);
 
 typedef struct {
     const IInternetProtocolVtbl      *lpIInternetProtocolVtbl;
     const IInternetProtocolSinkVtbl  *lpIInternetProtocolSinkVtbl;
 
     LONG ref;
 
     IInternetProtocolSink *protocol_sink;
     IInternetProtocol *protocol;
 } ProtocolProxy;
 
 #define PROTOCOL(x)  ((IInternetProtocol*)       &(x)->lpIInternetProtocolVtbl)
 #define PROTSINK(x)  ((IInternetProtocolSink*)   &(x)->lpIInternetProtocolSinkVtbl)
 
 HRESULT create_protocol_proxy(IInternetProtocol*,IInternetProtocolSink*,ProtocolProxy**);
 
 typedef struct {
     HWND notif_hwnd;
     DWORD notif_hwnd_cnt;
 
     struct list entry;
 } tls_data_t;
 
 tls_data_t *get_tls_data(void);
 
 HWND get_notif_hwnd(void);
 void release_notif_hwnd(HWND);
 
 static inline void *heap_alloc(size_t len)
 {
     return HeapAlloc(GetProcessHeap(), 0, len);
 }
 
 static inline void *heap_alloc_zero(size_t len)
 {
     return HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len);
 }
 
 static inline void *heap_realloc(void *mem, size_t len)
 {
     return HeapReAlloc(GetProcessHeap(), 0, mem, len);
 }
 
 static inline void *heap_realloc_zero(void *mem, size_t len)
 {
     return HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, mem, len);
 }
 
 static inline BOOL heap_free(void *mem)
 {
     return HeapFree(GetProcessHeap(), 0, mem);
 }
 
 static inline LPWSTR heap_strdupW(LPCWSTR str)
 {
     LPWSTR ret = NULL;
 
     if(str) {
         DWORD size;
 
         size = (strlenW(str)+1)*sizeof(WCHAR);
         ret = heap_alloc(size);
         memcpy(ret, str, size);
     }
 
     return ret;
 }
 
 static inline LPWSTR heap_strndupW(LPCWSTR str, int len)
 {
     LPWSTR ret = NULL;
 
     if(str) {
         ret = heap_alloc((len+1)*sizeof(WCHAR));
         if(ret) {
             memcpy(ret, str, len*sizeof(WCHAR));
             ret[len] = 0;
         }
     }
 
     return ret;
 }
 
 static inline LPWSTR heap_strdupAtoW(const char *str)
 {
     LPWSTR ret = NULL;
 
     if(str) {
         DWORD len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
         ret = heap_alloc(len*sizeof(WCHAR));
         MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
     }
 
     return ret;
 }
 
 #endif /* __WINE_URLMON_MAIN_H */