9a3a19671761071e9571d20d8fa4a1eab777531a
[reactos.git] / dll / win32 / jsproxy / main.c
1 /*
2 * Copyright 2014 Hans Leidekker 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 <wine/config.h>
20
21 #define WIN32_NO_STATUS
22 #define _INC_WINDOWS
23 #define COM_NO_WINDOWS_H
24
25 #define COBJMACROS
26
27 #include <windef.h>
28 #include <winbase.h>
29 #include <objbase.h>
30 #include <oleauto.h>
31 #include <dispex.h>
32 #include <activscp.h>
33 #include <wininet.h>
34
35 #include <wine/debug.h>
36 #include <wine/unicode.h>
37
38 #ifdef HAVE_SYS_SOCKET_H
39 # include <sys/socket.h>
40 #endif
41 #ifdef HAVE_NETINET_IN_H
42 # include <netinet/in.h>
43 #endif
44 #ifdef HAVE_NETDB_H
45 # include <netdb.h>
46 #endif
47 #if defined(__MINGW32__) || defined (_MSC_VER)
48 # include <ws2tcpip.h>
49 #else
50 # define closesocket close
51 # define ioctlsocket ioctl
52 #endif
53
54 #ifndef __MINGW32__
55 #define USE_WS_PREFIX
56 #endif
57
58 static HINSTANCE instance;
59
60 WINE_DEFAULT_DEBUG_CHANNEL(jsproxy);
61
62 static CRITICAL_SECTION cs_jsproxy;
63 static CRITICAL_SECTION_DEBUG critsect_debug =
64 {
65 0, 0, &cs_jsproxy,
66 { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
67 0, 0, { (DWORD_PTR)(__FILE__ ": cs_jsproxy") }
68 };
69 static CRITICAL_SECTION cs_jsproxy = { &critsect_debug, -1, 0, 0, 0, 0 };
70
71 static const WCHAR global_funcsW[] = {'g','l','o','b','a','l','_','f','u','n','c','s',0};
72 static const WCHAR dns_resolveW[] = {'d','n','s','_','r','e','s','o','l','v','e',0};
73
74 /******************************************************************
75 * DllMain (jsproxy.@)
76 */
77 BOOL WINAPI DllMain( HINSTANCE hinst, DWORD reason, LPVOID reserved )
78 {
79 switch (reason)
80 {
81 case DLL_PROCESS_ATTACH:
82 instance = hinst;
83 DisableThreadLibraryCalls( hinst );
84 break;
85
86 case DLL_PROCESS_DETACH:
87 break;
88 }
89 return TRUE;
90 }
91
92 static inline void* __WINE_ALLOC_SIZE(1) heap_alloc(size_t size)
93 {
94 return HeapAlloc(GetProcessHeap(), 0, size);
95 }
96
97 static inline BOOL heap_free(void *mem)
98 {
99 return HeapFree(GetProcessHeap(), 0, mem);
100 }
101
102 static inline WCHAR *strdupAW( const char *src, int len )
103 {
104 WCHAR *dst = NULL;
105 if (src)
106 {
107 int dst_len = MultiByteToWideChar( CP_ACP, 0, src, len, NULL, 0 );
108 if ((dst = heap_alloc( (dst_len + 1) * sizeof(WCHAR) )))
109 {
110 len = MultiByteToWideChar( CP_ACP, 0, src, len, dst, dst_len );
111 dst[dst_len] = 0;
112 }
113 }
114 return dst;
115 }
116
117 static inline char *strdupWA( const WCHAR *src )
118 {
119 char *dst = NULL;
120 if (src)
121 {
122 int len = WideCharToMultiByte( CP_ACP, 0, src, -1, NULL, 0, NULL, NULL );
123 if ((dst = heap_alloc( len ))) WideCharToMultiByte( CP_ACP, 0, src, -1, dst, len, NULL, NULL );
124 }
125 return dst;
126 }
127
128 static struct pac_script
129 {
130 WCHAR *text;
131 } pac_script;
132 static struct pac_script *global_script = &pac_script;
133
134 /******************************************************************
135 * InternetDeInitializeAutoProxyDll (jsproxy.@)
136 */
137 BOOL WINAPI InternetDeInitializeAutoProxyDll( LPSTR mime, DWORD reserved )
138 {
139 TRACE( "%s, %u\n", debugstr_a(mime), reserved );
140
141 EnterCriticalSection( &cs_jsproxy );
142
143 heap_free( global_script->text );
144 global_script->text = NULL;
145
146 LeaveCriticalSection( &cs_jsproxy );
147 return TRUE;
148 }
149
150 static WCHAR *load_script( const char *filename )
151 {
152 HANDLE handle;
153 DWORD size, bytes_read;
154 char *buffer;
155 int len;
156 WCHAR *script = NULL;
157
158 handle = CreateFileA( filename, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, 0 );
159 if (handle == INVALID_HANDLE_VALUE) return NULL;
160
161 size = GetFileSize( handle, NULL );
162 if (!(buffer = heap_alloc( size ))) goto done;
163 if (!ReadFile( handle, buffer, size, &bytes_read, NULL ) || bytes_read != size) goto done;
164
165 len = MultiByteToWideChar( CP_ACP, 0, buffer, size, NULL, 0 );
166 if (!(script = heap_alloc( (len + 1) * sizeof(WCHAR) ))) goto done;
167 MultiByteToWideChar( CP_ACP, 0, buffer, size, script, len );
168 script[len] = 0;
169
170 done:
171 CloseHandle( handle );
172 heap_free( buffer );
173 return script;
174 }
175
176 /******************************************************************
177 * InternetInitializeAutoProxyDll (jsproxy.@)
178 */
179 BOOL WINAPI JSPROXY_InternetInitializeAutoProxyDll( DWORD version, LPSTR tmpfile, LPSTR mime,
180 AutoProxyHelperFunctions *callbacks,
181 LPAUTO_PROXY_SCRIPT_BUFFER buffer )
182 {
183 BOOL ret = FALSE;
184
185 TRACE( "%u, %s, %s, %p, %p\n", version, debugstr_a(tmpfile), debugstr_a(mime), callbacks, buffer );
186
187 if (callbacks) FIXME( "callbacks not supported\n" );
188
189 EnterCriticalSection( &cs_jsproxy );
190
191 if (buffer && buffer->dwStructSize == sizeof(*buffer) && buffer->lpszScriptBuffer)
192 {
193 DWORD i, len = 0;
194 for (i = 0; i < buffer->dwScriptBufferSize; i++)
195 {
196 if (!buffer->lpszScriptBuffer[i]) break;
197 len++;
198 }
199 if (len == buffer->dwScriptBufferSize)
200 {
201 SetLastError( ERROR_INVALID_PARAMETER );
202 LeaveCriticalSection( &cs_jsproxy );
203 return FALSE;
204 }
205 heap_free( global_script->text );
206 if ((global_script->text = strdupAW( buffer->lpszScriptBuffer, len ))) ret = TRUE;
207 }
208 else
209 {
210 heap_free( global_script->text );
211 if ((global_script->text = load_script( tmpfile ))) ret = TRUE;
212 }
213
214 LeaveCriticalSection( &cs_jsproxy );
215 return ret;
216 }
217
218 static HRESULT WINAPI dispex_QueryInterface(
219 IDispatchEx *iface, REFIID riid, void **ppv )
220 {
221 *ppv = NULL;
222
223 if (IsEqualGUID( riid, &IID_IUnknown ) ||
224 IsEqualGUID( riid, &IID_IDispatch ) ||
225 IsEqualGUID( riid, &IID_IDispatchEx ))
226 *ppv = iface;
227 else
228 return E_NOINTERFACE;
229
230 return S_OK;
231 }
232
233 static ULONG WINAPI dispex_AddRef(
234 IDispatchEx *iface )
235 {
236 return 2;
237 }
238
239 static ULONG WINAPI dispex_Release(
240 IDispatchEx *iface )
241 {
242 return 1;
243 }
244
245 static HRESULT WINAPI dispex_GetTypeInfoCount(
246 IDispatchEx *iface, UINT *info )
247 {
248 return E_NOTIMPL;
249 }
250
251 static HRESULT WINAPI dispex_GetTypeInfo(
252 IDispatchEx *iface, UINT info, LCID lcid, ITypeInfo **type_info )
253 {
254 return E_NOTIMPL;
255 }
256
257 static HRESULT WINAPI dispex_GetIDsOfNames(
258 IDispatchEx *iface, REFIID riid, LPOLESTR *names, UINT count, LCID lcid, DISPID *id )
259 {
260 return E_NOTIMPL;
261 }
262
263 static HRESULT WINAPI dispex_Invoke(
264 IDispatchEx *iface, DISPID member, REFIID riid, LCID lcid, WORD flags,
265 DISPPARAMS *params, VARIANT *result, EXCEPINFO *excep, UINT *err )
266 {
267 return E_NOTIMPL;
268 }
269
270 static HRESULT WINAPI dispex_DeleteMemberByName(
271 IDispatchEx *iface, BSTR name, DWORD flags )
272 {
273 return E_NOTIMPL;
274 }
275
276 static HRESULT WINAPI dispex_DeleteMemberByDispID(
277 IDispatchEx *iface, DISPID id )
278 {
279 return E_NOTIMPL;
280 }
281
282 static HRESULT WINAPI dispex_GetMemberProperties(
283 IDispatchEx *iface, DISPID id, DWORD flags_fetch, DWORD *flags )
284 {
285 return E_NOTIMPL;
286 }
287
288 static HRESULT WINAPI dispex_GetMemberName(
289 IDispatchEx *iface, DISPID id, BSTR *name )
290 {
291 return E_NOTIMPL;
292 }
293
294 static HRESULT WINAPI dispex_GetNextDispID(
295 IDispatchEx *iface, DWORD flags, DISPID id, DISPID *next )
296 {
297 return E_NOTIMPL;
298 }
299
300 static HRESULT WINAPI dispex_GetNameSpaceParent(
301 IDispatchEx *iface, IUnknown **unk )
302 {
303 return E_NOTIMPL;
304 }
305
306 #define DISPID_GLOBAL_DNSRESOLVE 0x1000
307
308 static HRESULT WINAPI dispex_GetDispID(
309 IDispatchEx *iface, BSTR name, DWORD flags, DISPID *id )
310 {
311 if (!strcmpW( name, dns_resolveW ))
312 {
313 *id = DISPID_GLOBAL_DNSRESOLVE;
314 return S_OK;
315 }
316 return DISP_E_UNKNOWNNAME;
317 }
318
319 static char *get_computer_name( COMPUTER_NAME_FORMAT format )
320 {
321 char *ret;
322 DWORD size = 0;
323
324 GetComputerNameExA( format, NULL, &size );
325 if (GetLastError() != ERROR_MORE_DATA) return NULL;
326 if (!(ret = heap_alloc( size ))) return NULL;
327 if (!GetComputerNameExA( format, ret, &size ))
328 {
329 heap_free( ret );
330 return NULL;
331 }
332 return ret;
333 }
334
335 static void printf_addr( const WCHAR *fmt, WCHAR *buf, struct sockaddr_in *addr )
336 {
337 sprintfW( buf, fmt,
338 (unsigned int)(ntohl( addr->sin_addr.s_addr ) >> 24 & 0xff),
339 (unsigned int)(ntohl( addr->sin_addr.s_addr ) >> 16 & 0xff),
340 (unsigned int)(ntohl( addr->sin_addr.s_addr ) >> 8 & 0xff),
341 (unsigned int)(ntohl( addr->sin_addr.s_addr ) & 0xff) );
342 }
343
344 static HRESULT dns_resolve( const WCHAR *hostname, VARIANT *result )
345 {
346 #ifdef HAVE_GETADDRINFO
347 static const WCHAR fmtW[] = {'%','u','.','%','u','.','%','u','.','%','u',0};
348 WCHAR addr[16];
349 struct addrinfo *ai, *elem;
350 char *hostnameA;
351 int res;
352
353 if (hostname[0])
354 hostnameA = strdupWA( hostname );
355 else
356 hostnameA = get_computer_name( ComputerNamePhysicalDnsFullyQualified );
357
358 if (!hostnameA) return E_OUTOFMEMORY;
359 res = getaddrinfo( hostnameA, NULL, NULL, &ai );
360 heap_free( hostnameA );
361 if (res) return S_FALSE;
362
363 elem = ai;
364 while (elem && elem->ai_family != AF_INET) elem = elem->ai_next;
365 if (!elem)
366 {
367 freeaddrinfo( ai );
368 return S_FALSE;
369 }
370 printf_addr( fmtW, addr, (struct sockaddr_in *)elem->ai_addr );
371 freeaddrinfo( ai );
372 V_VT( result ) = VT_BSTR;
373 V_BSTR( result ) = SysAllocString( addr );
374 return S_OK;
375 #else
376 FIXME("getaddrinfo not found at build time\n");
377 return S_FALSE;
378 #endif
379 }
380
381 static HRESULT WINAPI dispex_InvokeEx(
382 IDispatchEx *iface, DISPID id, LCID lcid, WORD flags, DISPPARAMS *params,
383 VARIANT *result, EXCEPINFO *exep, IServiceProvider *caller )
384 {
385 if (id == DISPID_GLOBAL_DNSRESOLVE)
386 {
387 if (params->cArgs != 1) return DISP_E_BADPARAMCOUNT;
388 if (V_VT(&params->rgvarg[0]) != VT_BSTR) return DISP_E_BADVARTYPE;
389 return dns_resolve( V_BSTR(&params->rgvarg[0]), result );
390 }
391 return DISP_E_MEMBERNOTFOUND;
392 }
393
394 static const IDispatchExVtbl dispex_vtbl =
395 {
396 dispex_QueryInterface,
397 dispex_AddRef,
398 dispex_Release,
399 dispex_GetTypeInfoCount,
400 dispex_GetTypeInfo,
401 dispex_GetIDsOfNames,
402 dispex_Invoke,
403 dispex_GetDispID,
404 dispex_InvokeEx,
405 dispex_DeleteMemberByName,
406 dispex_DeleteMemberByDispID,
407 dispex_GetMemberProperties,
408 dispex_GetMemberName,
409 dispex_GetNextDispID,
410 dispex_GetNameSpaceParent
411 };
412
413 static IDispatchEx global_dispex = { &dispex_vtbl };
414
415 static HRESULT WINAPI site_QueryInterface(
416 IActiveScriptSite *iface, REFIID riid, void **ppv )
417 {
418 *ppv = NULL;
419
420 if (IsEqualGUID( &IID_IUnknown, riid ))
421 *ppv = iface;
422 else if (IsEqualGUID( &IID_IActiveScriptSite, riid ))
423 *ppv = iface;
424 else
425 return E_NOINTERFACE;
426
427 IUnknown_AddRef( (IUnknown *)*ppv );
428 return S_OK;
429 }
430
431 static ULONG WINAPI site_AddRef(
432 IActiveScriptSite *iface )
433 {
434 return 2;
435 }
436
437 static ULONG WINAPI site_Release(
438 IActiveScriptSite *iface )
439 {
440 return 1;
441 }
442
443 static HRESULT WINAPI site_GetLCID(
444 IActiveScriptSite *iface, LCID *lcid )
445 {
446 return E_NOTIMPL;
447 }
448
449 static HRESULT WINAPI site_GetItemInfo(
450 IActiveScriptSite *iface, LPCOLESTR name, DWORD mask,
451 IUnknown **item, ITypeInfo **type_info )
452 {
453 if (!strcmpW( name, global_funcsW ) && mask == SCRIPTINFO_IUNKNOWN)
454 {
455 *item = (IUnknown *)&global_dispex;
456 return S_OK;
457 }
458 return E_NOTIMPL;
459 }
460
461 static HRESULT WINAPI site_GetDocVersionString(
462 IActiveScriptSite *iface, BSTR *version )
463 {
464 return E_NOTIMPL;
465 }
466
467 static HRESULT WINAPI site_OnScriptTerminate(
468 IActiveScriptSite *iface, const VARIANT *result, const EXCEPINFO *info )
469 {
470 return E_NOTIMPL;
471 }
472
473 static HRESULT WINAPI site_OnStateChange(
474 IActiveScriptSite *iface, SCRIPTSTATE state )
475 {
476 return E_NOTIMPL;
477 }
478
479 static HRESULT WINAPI site_OnScriptError(
480 IActiveScriptSite *iface, IActiveScriptError *error )
481 {
482 return E_NOTIMPL;
483 }
484
485 static HRESULT WINAPI site_OnEnterScript(
486 IActiveScriptSite *iface )
487 {
488 return E_NOTIMPL;
489 }
490
491 static HRESULT WINAPI site_OnLeaveScript(
492 IActiveScriptSite *iface )
493 {
494 return E_NOTIMPL;
495 }
496
497 static const IActiveScriptSiteVtbl site_vtbl =
498 {
499 site_QueryInterface,
500 site_AddRef,
501 site_Release,
502 site_GetLCID,
503 site_GetItemInfo,
504 site_GetDocVersionString,
505 site_OnScriptTerminate,
506 site_OnStateChange,
507 site_OnScriptError,
508 site_OnEnterScript,
509 site_OnLeaveScript
510 };
511
512 static IActiveScriptSite script_site = { &site_vtbl };
513
514 static BSTR include_pac_utils( const WCHAR *script )
515 {
516 static const WCHAR pacjsW[] = {'p','a','c','.','j','s',0};
517 HMODULE hmod = GetModuleHandleA( "jsproxy.dll" );
518 HRSRC rsrc;
519 DWORD size;
520 const char *data;
521 BSTR ret;
522 int len;
523
524 if (!(rsrc = FindResourceW( hmod, pacjsW, (LPCWSTR)40 ))) return NULL;
525 size = SizeofResource( hmod, rsrc );
526 data = LoadResource( hmod, rsrc );
527
528 len = MultiByteToWideChar( CP_ACP, 0, data, size, NULL, 0 );
529 if (!(ret = SysAllocStringLen( NULL, len + strlenW( script ) + 1 ))) return NULL;
530 MultiByteToWideChar( CP_ACP, 0, data, size, ret, len );
531 strcpyW( ret + len, script );
532 return ret;
533 }
534
535 #ifdef _WIN64
536 #define IActiveScriptParse_Release IActiveScriptParse64_Release
537 #define IActiveScriptParse_InitNew IActiveScriptParse64_InitNew
538 #define IActiveScriptParse_ParseScriptText IActiveScriptParse64_ParseScriptText
539 #else
540 #define IActiveScriptParse_Release IActiveScriptParse32_Release
541 #define IActiveScriptParse_InitNew IActiveScriptParse32_InitNew
542 #define IActiveScriptParse_ParseScriptText IActiveScriptParse32_ParseScriptText
543 #endif
544
545 static BOOL run_script( const WCHAR *script, const WCHAR *url, const WCHAR *hostname, char **result_str, DWORD *result_len )
546 {
547 static const WCHAR jscriptW[] = {'J','S','c','r','i','p','t',0};
548 static const WCHAR findproxyW[] = {'F','i','n','d','P','r','o','x','y','F','o','r','U','R','L',0};
549 IActiveScriptParse *parser = NULL;
550 IActiveScript *engine = NULL;
551 IDispatch *dispatch = NULL;
552 BOOL ret = FALSE;
553 CLSID clsid;
554 DISPID dispid;
555 BSTR func = NULL, full_script = NULL;
556 VARIANT args[2], retval;
557 DISPPARAMS params;
558 HRESULT hr, init;
559
560 init = CoInitialize( NULL );
561 hr = CLSIDFromProgID( jscriptW, &clsid );
562 if (hr != S_OK) goto done;
563
564 hr = CoCreateInstance( &clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
565 &IID_IActiveScript, (void **)&engine );
566 if (hr != S_OK) goto done;
567
568 hr = IActiveScript_QueryInterface( engine, &IID_IActiveScriptParse, (void **)&parser );
569 if (hr != S_OK) goto done;
570
571 hr = IActiveScriptParse_InitNew( parser );
572 if (hr != S_OK) goto done;
573
574 hr = IActiveScript_SetScriptSite( engine, &script_site );
575 if (hr != S_OK) goto done;
576
577 hr = IActiveScript_AddNamedItem( engine, global_funcsW, SCRIPTITEM_GLOBALMEMBERS );
578 if (hr != S_OK) goto done;
579
580 if (!(full_script = include_pac_utils( script ))) goto done;
581
582 hr = IActiveScriptParse_ParseScriptText( parser, full_script, NULL, NULL, NULL, 0, 0, 0, NULL, NULL );
583 if (hr != S_OK) goto done;
584
585 hr = IActiveScript_SetScriptState( engine, SCRIPTSTATE_STARTED );
586 if (hr != S_OK) goto done;
587
588 hr = IActiveScript_GetScriptDispatch( engine, NULL, &dispatch );
589 if (hr != S_OK) goto done;
590
591 if (!(func = SysAllocString( findproxyW ))) goto done;
592 hr = IDispatch_GetIDsOfNames( dispatch, &IID_NULL, &func, 1, LOCALE_SYSTEM_DEFAULT, &dispid );
593 if (hr != S_OK) goto done;
594
595 V_VT( &args[0] ) = VT_BSTR;
596 V_BSTR( &args[0] ) = SysAllocString( hostname );
597 V_VT( &args[1] ) = VT_BSTR;
598 V_BSTR( &args[1] ) = SysAllocString( url );
599
600 params.rgvarg = args;
601 params.rgdispidNamedArgs = NULL;
602 params.cArgs = 2;
603 params.cNamedArgs = 0;
604 hr = IDispatch_Invoke( dispatch, dispid, &IID_NULL, LOCALE_SYSTEM_DEFAULT, DISPATCH_METHOD,
605 &params, &retval, NULL, NULL );
606 VariantClear( &args[0] );
607 VariantClear( &args[1] );
608 if (hr != S_OK)
609 {
610 WARN("script failed 0x%08x\n", hr);
611 goto done;
612 }
613 if ((*result_str = strdupWA( V_BSTR( &retval ) )))
614 {
615 TRACE( "result: %s\n", debugstr_a(*result_str) );
616 *result_len = strlen( *result_str ) + 1;
617 ret = TRUE;
618 }
619 VariantClear( &retval );
620
621 done:
622 SysFreeString( full_script );
623 SysFreeString( func );
624 if (dispatch) IDispatch_Release( dispatch );
625 if (parser) IActiveScriptParse_Release( parser );
626 if (engine) IActiveScript_Release( engine );
627 if (SUCCEEDED( init )) CoUninitialize();
628 return ret;
629 }
630
631 /******************************************************************
632 * InternetGetProxyInfo (jsproxy.@)
633 */
634 BOOL WINAPI InternetGetProxyInfo( LPCSTR url, DWORD len_url, LPCSTR hostname, DWORD len_hostname, LPSTR *proxy,
635 LPDWORD len_proxy )
636 {
637 WCHAR *urlW = NULL, *hostnameW = NULL;
638 BOOL ret = FALSE;
639
640 TRACE( "%s, %u, %s, %u, %p, %p\n", debugstr_a(url), len_url, hostname, len_hostname, proxy, len_proxy );
641
642 EnterCriticalSection( &cs_jsproxy );
643
644 if (!global_script->text)
645 {
646 SetLastError( ERROR_CAN_NOT_COMPLETE );
647 goto done;
648 }
649 if (hostname && len_hostname < strlen( hostname ))
650 {
651 SetLastError( ERROR_INSUFFICIENT_BUFFER );
652 goto done;
653 }
654 if (!(urlW = strdupAW( url, -1 ))) goto done;
655 if (hostname && !(hostnameW = strdupAW( hostname, -1 ))) goto done;
656
657 TRACE( "%s\n", debugstr_w(global_script->text) );
658 ret = run_script( global_script->text, urlW, hostnameW, proxy, len_proxy );
659
660 done:
661 heap_free( hostnameW );
662 heap_free( urlW );
663 LeaveCriticalSection( &cs_jsproxy );
664 return ret;
665 }