1 /*
2 * Copyright 2002 Huw D M Davies for CodeWeavers
3 * Copyright 2009 Jacek Caban for CodeWeavers
4 *
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2.1 of the License, or (at your option) any later version.
9 *
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
18 */
19
20 #ifndef __WINE_URLMON_MAIN_H
21 #define __WINE_URLMON_MAIN_H
22
23 #include <stdarg.h>
24
25 #define COBJMACROS
26 #define NONAMELESSUNION
27 #define NONAMELESSSTRUCT
28
29 #include "windef.h"
30 #include "winbase.h"
31 #include "winuser.h"
32 #include "ole2.h"
33 #include "urlmon.h"
34 #include "wininet.h"
35
36 #include "wine/unicode.h"
37
38 extern HINSTANCE URLMON_hInstance;
39 extern HRESULT SecManagerImpl_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
40 extern HRESULT ZoneMgrImpl_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
41 extern HRESULT StdURLMoniker_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
42 extern HRESULT FileProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
43 extern HRESULT HttpProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
44 extern HRESULT HttpSProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
45 extern HRESULT FtpProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
46 extern HRESULT GopherProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
47 extern HRESULT MkProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
48 extern HRESULT MimeFilter_Construct(IUnknown *pUnkOuter, LPVOID *ppobj);
49
50 /**********************************************************************
51 * Dll lifetime tracking declaration for urlmon.dll
52 */
53 extern LONG URLMON_refCount;
54 static inline void URLMON_LockModule(void) { InterlockedIncrement( &URLMON_refCount ); }
55 static inline void URLMON_UnlockModule(void) { InterlockedDecrement( &URLMON_refCount ); }
56
57 #define DEFINE_THIS2(cls,ifc,iface) ((cls*)((BYTE*)(iface)-offsetof(cls,ifc)))
58 #define DEFINE_THIS(cls,ifc,iface) DEFINE_THIS2(cls,lp ## ifc ## Vtbl,iface)
59
60 IInternetProtocolInfo *get_protocol_info(LPCWSTR);
61 HRESULT get_protocol_handler(LPCWSTR,CLSID*,BOOL*,IClassFactory**);
62 BOOL is_registered_protocol(LPCWSTR);
63 void register_urlmon_namespace(IClassFactory*,REFIID,LPCWSTR,BOOL);
64
65 HRESULT bind_to_storage(LPCWSTR url, IBindCtx *pbc, REFIID riid, void **ppv);
66 HRESULT bind_to_object(IMoniker *mon, LPCWSTR url, IBindCtx *pbc, REFIID riid, void **ppv);
67
68 HRESULT create_binding_protocol(LPCWSTR url, BOOL from_urlmon, IInternetProtocol **protocol);
69 void set_binding_sink(IInternetProtocol *bind_protocol, IInternetProtocolSink *sink);
70 IWinInetInfo *get_wininet_info(IInternetProtocol*);
71
72 typedef struct ProtocolVtbl ProtocolVtbl;
73
74 typedef struct {
75 const ProtocolVtbl *vtbl;
76
77 IInternetProtocol *protocol;
78 IInternetProtocolSink *protocol_sink;
79
80 DWORD bindf;
81 BINDINFO bind_info;
82
83 HINTERNET internet;
84 HINTERNET request;
85 HINTERNET connection;
86 DWORD flags;
87 HANDLE lock;
88
89 ULONG current_position;
90 ULONG content_length;
91 ULONG available_bytes;
92
93 LONG priority;
94 } Protocol;
95
96 struct ProtocolVtbl {
97 HRESULT (*open_request)(Protocol*,LPCWSTR,DWORD,IInternetBindInfo*);
98 HRESULT (*start_downloading)(Protocol*);
99 void (*close_connection)(Protocol*);
100 };
101
102 HRESULT protocol_start(Protocol*,IInternetProtocol*,LPCWSTR,IInternetProtocolSink*,IInternetBindInfo*);
103 HRESULT protocol_continue(Protocol*,PROTOCOLDATA*);
104 HRESULT protocol_read(Protocol*,void*,ULONG,ULONG*);
105 HRESULT protocol_lock_request(Protocol*);
106 HRESULT protocol_unlock_request(Protocol*);
107 void protocol_close_connection(Protocol*);
108
109 static inline void *heap_alloc(size_t len)
110 {
111 return HeapAlloc(GetProcessHeap(), 0, len);
112 }
113
114 static inline void *heap_alloc_zero(size_t len)
115 {
116 return HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len);
117 }
118
119 static inline void *heap_realloc(void *mem, size_t len)
120 {
121 return HeapReAlloc(GetProcessHeap(), 0, mem, len);
122 }
123
124 static inline BOOL heap_free(void *mem)
125 {
126 return HeapFree(GetProcessHeap(), 0, mem);
127 }
128
129 static inline LPWSTR heap_strdupW(LPCWSTR str)
130 {
131 LPWSTR ret = NULL;
132
133 if(str) {
134 DWORD size;
135
136 size = (strlenW(str)+1)*sizeof(WCHAR);
137 ret = heap_alloc(size);
138 memcpy(ret, str, size);
139 }
140
141 return ret;
142 }
143
144 static inline LPWSTR heap_strndupW(LPCWSTR str, int len)
145 {
146 LPWSTR ret = NULL;
147
148 if(str) {
149 ret = heap_alloc((len+1)*sizeof(WCHAR));
150 if(ret) {
151 memcpy(ret, str, len*sizeof(WCHAR));
152 ret[len] = 0;
153 }
154 }
155
156 return ret;
157 }
158
159 static inline LPWSTR heap_strdupAtoW(const char *str)
160 {
161 LPWSTR ret = NULL;
162
163 if(str) {
164 DWORD len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
165 ret = heap_alloc(len*sizeof(WCHAR));
166 MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
167 }
168
169 return ret;
170 }
171
172 #endif /* __WINE_URLMON_MAIN_H */
173
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.