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