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