src/WINNT/install/loopback/wmi.cpp

   1 /*
   2
   3 Copyright 2004 by the Massachusetts Institute of Technology
   4 Copyright 2006 by Secure Endpoints Inc.
   5
   6 All rights reserved.
   7
   8 Permission to use, copy, modify, and distribute this software and its
   9 documentation for any purpose and without fee is hereby granted,
  10 provided that the above copyright notice appear in all copies and that
  11 both that copyright notice and this permission notice appear in
  12 supporting documentation, and that the name of the Massachusetts
  13 Institute of Technology (M.I.T.) not be used in advertising or publicity
  14 pertaining to distribution of the software without specific, written
  15 prior permission.
  16
  17 M.I.T. DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING
  18 ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL
  19 M.I.T. BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR
  20 ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
  21 WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,
  22 ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
  23 SOFTWARE.
  24
  25 */
  26
  27 //**************************************************************************
  28 //
  29 // Description:
  30 //
  31 //      Call EnableStatic method of Win32_NetworkAdapterConfiguration
  32 //      for some network adapter GUID.
  33 //
  34 // Note:
  35 //
  36 //      The EnableStatic method is not supported on Win9x platforms.
  37 //
  38 //**************************************************************************
  39
  40 #define _WIN32_DCOM
  41 #include <windows.h>
  42 #include <comdef.h>
  43 #include <wbemidl.h>
  44 #include <tchar.h>
  45
  46 #include <devguid.h>
  47
  48 /* __RPC__out is not defined in the v6.0 Windows SDK */
  49 #ifndef __RPC__out
  50 #define __RPC__out
  51 #endif
  52
  53 /* These two are from the Windows DDK */
  54 #include <netcfgx.h>
  55 #include <netcfgn.h>
  56
  57 #include <shlobj.h>
  58 //#include <objbase.h>
  59
  60 //#ifndef TEST
  61 //inline void printf(char*, ...) {}
  62 //#else
  63 #include <stdio.h>
  64 //#endif
  65
  66 #include "loopbackutils.h"
  67
  68 #define CLEANUP_ON_FAILURE(hr) \
  69         do { if (!SUCCEEDED(hr)) {goto cleanup;} } while (0)
  70
  71 #define CLEANUP_ON_AND_SET(check, var, value) \
  72     do { if (check) { (var) = (value); goto cleanup; } } while (0)
  73
  74 #define ETCDIR "\\drivers\\etc"
  75
  76 #define EACCES (13)
  77 #define ENOENT (2)
  78
  79 DWORD AdjustMaxLana(DWORD dwMaxLana);
  80
  81 typedef
  82 HRESULT
  83 (*FindNetworkAdapterConfigurationInstance_t)(
  84     IN  PVOID pContext,
  85     IN  IWbemServices *pSvc,
  86     OUT BSTR* pPath
  87     );
  88
  89 HRESULT
  90 FindNetworkAdapterConfigurationInstanceByGUID(
  91     IN  PVOID pContext,
  92     IN  IWbemServices *pSvc,
  93     OUT BSTR* pPath
  94     )
  95 {
  96     HRESULT hr = 0;
  97     BOOL bFound = FALSE;
  98     BSTR Language = 0;
  99     BSTR Query = 0;
 100     IEnumWbemClassObject* pEnum = 0;
 101     IWbemClassObject* pObj = 0;
 102     VARIANT Value;
 103     VariantInit(&Value);
 104     LPCWSTR adapter_guid = (LPCWSTR)pContext;
 105
 106     // Check arguments
 107     if (!pPath || !adapter_guid || *pPath)
 108         return E_INVALIDARG;
 109
 110     *pPath = 0;
 111
 112     // Query for all network adapters
 113     Language = SysAllocString(L"WQL");
 114     Query = SysAllocString(L"select * from Win32_NetworkAdapterConfiguration");
 115
 116     // Issue the query.
 117     hr = pSvc->ExecQuery(Language,
 118                          Query,
 119                          WBEM_FLAG_FORWARD_ONLY,         // Flags
 120                          0,                              // Context
 121                          &pEnum);
 122     if (!SUCCEEDED(hr))
 123     {
 124         ReportMessage(0,"ExecQuery() error",NULL,NULL, hr);
 125         goto cleanup;
 126     }
 127
 128     // Retrieve the objects in the result set.
 129     while (!bFound)
 130     {
 131         ULONG uReturned = 0;
 132
 133         hr = pEnum->Next(0,                  // Time out
 134                          1,                  // One object
 135                          &pObj,
 136                          &uReturned);
 137         CLEANUP_ON_FAILURE(hr);
 138
 139         if (uReturned == 0)
 140             break;
 141
 142         // Use the object.
 143         hr  = pObj->Get(L"SettingID", // property name
 144                         0L,
 145                         &Value,       // output to this variant
 146                         NULL,
 147                         NULL);
 148         CLEANUP_ON_FAILURE(hr);
 149
 150         bFound = !wcscmp(adapter_guid, V_BSTR(&Value));
 151
 152         if (bFound)
 153         {
 154             ReportMessage(1,"Found adapter", NULL,V_BSTR(&Value),0);
 155             VariantClear(&Value);
 156             hr = pObj->Get(L"__RELPATH", // property name
 157                            0L,
 158                            &Value,       // output to this variant
 159                            NULL,
 160                            NULL);
 161             CLEANUP_ON_FAILURE(hr);
 162
 163             *pPath = SysAllocString(V_BSTR(&Value));
 164
 165         }
 166
 167         VariantClear(&Value);
 168
 169         // Release it.
 170         // ===========
 171         pObj->Release();    // Release objects not owned.
 172         pObj = 0;
 173     }
 174
 175
 176     // All done.
 177  cleanup:
 178     SysFreeString(Query);
 179     SysFreeString(Language);
 180     VariantClear(&Value);
 181     if (pEnum)
 182         pEnum->Release();
 183     if (pObj)
 184         pObj->Release();
 185
 186     return *pPath ? 0 : ( SUCCEEDED(hr) ? WBEM_E_NOT_FOUND : hr );
 187 }
 188
 189 HRESULT
 190 SetupStringAsSafeArray(LPCWSTR s, VARIANT* v)
 191 {
 192     HRESULT hr = 0;
 193     BSTR b = 0;
 194     SAFEARRAY* array = 0;
 195     long index[] = {0};
 196
 197     if (V_VT(v) != VT_EMPTY)
 198         return E_INVALIDARG;
 199
 200     b = SysAllocString(s);
 201     CLEANUP_ON_AND_SET(!b, hr, E_OUTOFMEMORY);
 202
 203     array = SafeArrayCreateVector(VT_BSTR, 0, 1);
 204     CLEANUP_ON_AND_SET(!array, hr, E_OUTOFMEMORY);
 205
 206     hr = SafeArrayPutElement(array, index, b);
 207     CLEANUP_ON_FAILURE(hr);
 208
 209     V_VT(v) = VT_ARRAY|VT_BSTR;
 210     V_ARRAY(v) = array;
 211
 212  cleanup:
 213     if (b)
 214         SysFreeString(b);
 215     if (!SUCCEEDED(hr))
 216     {
 217         if (array)
 218             SafeArrayDestroy(array);
 219     }
 220     return hr;
 221 }
 222
 223 static BOOL
 224 IsXP(void)
 225 {
 226     OSVERSIONINFOEX osInfoEx;
 227     memset(&osInfoEx, 0, sizeof(osInfoEx));
 228     osInfoEx.dwOSVersionInfoSize = sizeof(osInfoEx);
 229
 230     GetVersionEx((POSVERSIONINFO)&osInfoEx);
 231
 232     return(osInfoEx.dwMajorVersion == 5 && osInfoEx.dwMinorVersion == 1 && osInfoEx.wServicePackMajor < 2);
 233 }
 234
 235 static VOID
 236 FixupXPDNSRegistrations(LPCWSTR pCfgGuidString)
 237 {
 238     // As per http://support.microsoft.com/default.aspx?scid=kb%3Ben-us%3B834440
 239     // HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\Interfaces\<NetworkAdapterGUID>
 240     HKEY hkInterfaces=NULL, hkAdapter=NULL;
 241     DWORD dw = 0;
 242
 243     if (!IsXP())
 244         return;         // Nothing to do
 245
 246     RegOpenKeyEx(HKEY_LOCAL_MACHINE, TEXT("SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Parameters\\Interfaces"),
 247                  0, KEY_READ, &hkInterfaces);
 248
 249     RegOpenKeyEx(hkInterfaces, pCfgGuidString, 0, KEY_READ|KEY_WRITE, &hkAdapter);
 250
 251     RegDeleteValue(hkAdapter, TEXT("DisableDynamicUpdate"));
 252     RegDeleteValue(hkAdapter, TEXT("EnableAdapterDomainNameRegistration"));
 253     RegSetValueEx(hkAdapter, TEXT("RegistrationEnabled"), 0, REG_DWORD, (BYTE *)&dw, sizeof(DWORD));
 254     RegSetValueEx(hkAdapter, TEXT("RegisterAdapterName"), 0, REG_DWORD, (BYTE *)&dw, sizeof(DWORD));
 255
 256     if (hkInterfaces)
 257         RegCloseKey(hkInterfaces);
 258     if (hkAdapter)
 259         RegCloseKey(hkAdapter);
 260 }
 261
 262 HRESULT
 263 WMIEnableStatic(
 264     FindNetworkAdapterConfigurationInstance_t pFindInstance,
 265     PVOID pContext,
 266     LPCWSTR ip,
 267     LPCWSTR mask
 268     )
 269 {
 270     HRESULT hr = 0, hr2 = 0;
 271
 272     IWbemLocator* pLocator = NULL;
 273     IWbemServices* pNamespace = NULL;
 274     IWbemClassObject* pClass = NULL;
 275     IWbemClassObject* pOutInst = NULL;
 276     IWbemClassObject* pInClass = NULL;
 277     IWbemClassObject* pInInst = NULL;
 278
 279     BSTR NamespacePath = NULL;
 280     BSTR ClassPath = NULL;
 281     BSTR InstancePath = NULL;
 282     BSTR MethodName = NULL; // needs to be BSTR for ExecMethod()
 283
 284     BOOL comInitialized = FALSE;
 285
 286     VARIANT v_ip_list;
 287     VariantInit(&v_ip_list);
 288
 289     VARIANT v_mask_list;
 290     VariantInit(&v_mask_list);
 291
 292     VARIANT v_ret_value;
 293     VariantInit(&v_ret_value);
 294
 295     VARIANT v_reg_enabled;
 296     VariantInit(&v_reg_enabled);
 297
 298     VARIANT v_netbios;
 299     VariantInit(&v_netbios);
 300
 301     int count;
 302
 303     // end of declarations & NULL initialization
 304
 305     NamespacePath = SysAllocString(L"root\\cimv2");
 306     CLEANUP_ON_AND_SET(!NamespacePath, hr, E_OUTOFMEMORY);
 307
 308     ClassPath = SysAllocString(L"Win32_NetWorkAdapterConfiguration");
 309     CLEANUP_ON_AND_SET(!ClassPath, hr, E_OUTOFMEMORY);
 310
 311     // Initialize COM and connect up to CIMOM
 312
 313     ReportMessage(0, "Intializing COM", NULL, NULL, 0);
 314     hr = CoInitializeEx(0, COINIT_MULTITHREADED);
 315     if (hr == S_OK || hr == S_FALSE) {
 316         comInitialized = TRUE;
 317     } else {
 318         goto cleanup;
 319     }
 320
 321     /* When called from an MSI this will generally fail.  This should only be called once
 322            per process and not surprisingly MSI beats us to it.  So ignore return value and
 323            hope for the best. */
 324     hr = CoInitializeSecurity(NULL, -1, NULL, NULL,
 325                               RPC_C_AUTHN_LEVEL_CONNECT,
 326                               RPC_C_IMP_LEVEL_IMPERSONATE,
 327                               NULL, EOAC_NONE, 0);
 328
 329     /* CLEANUP_ON_FAILURE(hr); */
 330
 331     ReportMessage(0, "Creating Wbem Locator object", NULL, NULL, 0);
 332     hr = CoCreateInstance(CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER,
 333                           IID_IWbemLocator, (LPVOID *) &pLocator);
 334     CLEANUP_ON_FAILURE(hr);
 335
 336     ReportMessage(0, "Connecting to WMI", NULL, NULL, 0);
 337     hr = pLocator->ConnectServer(NamespacePath, NULL, NULL, NULL, 0,
 338                                  NULL, NULL, &pNamespace);
 339     CLEANUP_ON_FAILURE(hr);
 340
 341     ReportMessage(0,"Connected to WMI",NULL,NULL,0);
 342
 343     // Set the proxy so that impersonation of the client occurs.
 344     hr = CoSetProxyBlanket(pNamespace,
 345                            RPC_C_AUTHN_WINNT,
 346                            RPC_C_AUTHZ_NONE,
 347                            NULL,
 348                            RPC_C_AUTHN_LEVEL_CALL,
 349                            RPC_C_IMP_LEVEL_IMPERSONATE,
 350                            NULL,
 351                            EOAC_NONE);
 352     CLEANUP_ON_FAILURE(hr);
 353
 354     // Get the class object
 355     hr = pNamespace->GetObject(ClassPath, 0, NULL, &pClass, NULL);
 356     CLEANUP_ON_FAILURE(hr);
 357
 358     // Get the instance
 359     hr = pFindInstance(pContext, pNamespace, &InstancePath);
 360     CLEANUP_ON_FAILURE(hr);
 361
 362     ReportMessage(0,"Found Adapter Instance",NULL, InstancePath,0);
 363
 364 #if 0
 365     // Use the adapter instance index to set MAXLANA in the registry.
 366     {
 367         DWORD dwIndex;
 368         if (swscanf(InstancePath, L"Win32_NetworkAdapterConfiguration.Index=%u", &dwIndex)==1)
 369         {
 370             DWORD ret = 0;
 371             ReportMessage(1,"Setting MAXLANA",NULL,NULL,dwIndex+1);
 372             ret = AdjustMaxLana(dwIndex+1);
 373             if (ret) ReportMessage(0,"AdjustMaxLana returned the error code ",NULL,NULL,ret);
 374         }
 375     }
 376 #endif
 377
 378     MethodName = SysAllocString(L"EnableStatic");
 379     CLEANUP_ON_AND_SET(!MethodName, hr, E_OUTOFMEMORY);
 380
 381     // Get the input argument and set the property
 382     hr = pClass->GetMethod(MethodName, 0, &pInClass, NULL);
 383     CLEANUP_ON_FAILURE(hr);
 384
 385     hr = pInClass->SpawnInstance(0, &pInInst);
 386     CLEANUP_ON_FAILURE(hr);
 387
 388     // Set up parameters
 389     hr = SetupStringAsSafeArray(ip, &v_ip_list);
 390     CLEANUP_ON_FAILURE(hr);
 391
 392     hr = pInInst->Put(L"IPAddress", 0, &v_ip_list, 0);
 393     CLEANUP_ON_FAILURE(hr);
 394
 395     hr = SetupStringAsSafeArray(mask, &v_mask_list);
 396     CLEANUP_ON_FAILURE(hr);
 397
 398     hr = pInInst->Put(L"SubNetMask", 0, &v_mask_list, 0);
 399     CLEANUP_ON_FAILURE(hr);
 400
 401 //    printf("Skipping ExecMethod\n");
 402 //    hr = 0;
 403 //    goto cleanup;
 404
 405     // Try up to five times, sleeping 3 seconds between tries
 406     for (count=0; count<5; count++)
 407     {
 408         if (count>1) {
 409             ReportMessage(0,"Trying again in 3 seconds...",NULL,NULL,0);
 410             Sleep(3000);
 411         } else if (count>0) {
 412             ReportMessage(0,"Trying again in 20 seconds...",NULL,NULL,0);
 413             Sleep(10000);
 414             ReportMessage(0,"Trying again in 10 seconds...",NULL,NULL,0);
 415             Sleep(5000);
 416             ReportMessage(0,"Trying again in  5 seconds...",NULL,NULL,0);
 417             Sleep(2000);
 418         }
 419
 420         ReportMessage(0,"Calling ExecMethod EnableStatic NOW...          ",NULL,NULL,0);
 421         // Call the method
 422         hr = pNamespace->ExecMethod(InstancePath, MethodName, 0, NULL, pInInst,
 423                                   &pOutInst, NULL);
 424         if (!SUCCEEDED(hr))
 425         {
 426            ReportMessage(0,"ExecMethod EnableStatic failed",NULL,NULL, hr);
 427            continue;
 428         }
 429
 430         // Get the EnableStatic method return value
 431         hr = pOutInst->Get(L"ReturnValue", 0, &v_ret_value, 0, 0);
 432         if (!SUCCEEDED(hr))
 433         {
 434           ReportMessage(0,"WARNING: Could not determine return value for EnableStatic ",NULL,NULL, hr);
 435           continue;
 436         }
 437
 438         hr = V_I4(&v_ret_value);
 439         if (hr != 0)
 440             ReportMessage(0,"EnableStatic failed ", NULL,NULL,hr);
 441         else
 442         {
 443             ReportMessage(0,"EnableStatic succeeded",NULL,NULL,0);
 444             break;
 445         }
 446     }
 447
 448     /* if failure, skip SetDynamicDNSRegistration */
 449     if (hr)
 450         goto cleanup;
 451
 452     /* Cleanup and Prepare for SetDynamicDNSRegistration */
 453     if (pInClass) {
 454         pInClass->Release();
 455         pInClass = NULL;
 456     }
 457     if (pInInst) {
 458         pInInst->Release();
 459         pInInst = NULL;
 460     }
 461     SysFreeString(MethodName);
 462     VariantClear(&v_ret_value);
 463
 464     MethodName = SysAllocString(L"SetDynamicDNSRegistration");
 465     CLEANUP_ON_AND_SET(!MethodName, hr2, E_OUTOFMEMORY);
 466
 467     // Get the input argument and set the property
 468     hr2 = pClass->GetMethod(MethodName, 0, &pInClass, NULL);
 469     CLEANUP_ON_FAILURE(hr2);
 470
 471     hr2 = pInClass->SpawnInstance(0, &pInInst);
 472     CLEANUP_ON_FAILURE(hr2);
 473
 474     // Set up parameters
 475     V_VT(&v_reg_enabled) = VT_BOOL;
 476     V_BOOL(&v_reg_enabled) = VARIANT_FALSE;
 477
 478     hr2 = pInInst->Put(L"FullDNSRegistrationEnabled", 0, &v_reg_enabled, 0);
 479     CLEANUP_ON_FAILURE(hr2);
 480
 481     hr2 = pInInst->Put(L"DomainDNSRegistrationEnabled", 0, &v_reg_enabled, 0);
 482     CLEANUP_ON_FAILURE(hr2);
 483
 484     // Call the method
 485     hr2 = pNamespace->ExecMethod(InstancePath, MethodName, 0, NULL, pInInst,
 486                                  &pOutInst, NULL);
 487     if (!SUCCEEDED(hr2))        {
 488         ReportMessage(0,"ExecMethod SetDynamicDNSRegistration failed",NULL,NULL, hr2);
 489         goto cleanup;
 490     }
 491
 492     // Get the EnableStatic method return value
 493     hr2 = pOutInst->Get(L"ReturnValue", 0, &v_ret_value, 0, 0);
 494     if (!SUCCEEDED(hr2)) {
 495         ReportMessage(0,"WARNING: Could not determine return value for SetDynamicDNSRegistration ",NULL,NULL, hr2);
 496     } else {
 497         hr2 = V_I4(&v_ret_value);
 498         if (hr2 != 0)
 499             ReportMessage(0,"SetDynamicDNSRegistration failed ", NULL,NULL,hr2);
 500         else
 501             ReportMessage(0,"SetDynamicDNSRegistration succeeded",NULL,NULL,0);
 502     }
 503
 504     /* if failure, skip SetTcpipNetbios */
 505     if (hr)
 506         goto cleanup;
 507
 508     /* Cleanup and Prepare for SetTcpipNetbios */
 509     if (pInClass) {
 510         pInClass->Release();
 511         pInClass = NULL;
 512     }
 513     if (pInInst) {
 514         pInInst->Release();
 515         pInInst = NULL;
 516     }
 517     SysFreeString(MethodName);
 518     VariantClear(&v_ret_value);
 519
 520     ReportMessage(0,"Preparing for SetTcpipNetbios",NULL,NULL,0);
 521
 522     MethodName = SysAllocString(L"SetTcpipNetbios");
 523     CLEANUP_ON_AND_SET(!MethodName, hr2, E_OUTOFMEMORY);
 524
 525     // Get the input argument and set the property
 526     hr2 = pClass->GetMethod(MethodName, 0, &pInClass, NULL);
 527     CLEANUP_ON_FAILURE(hr2);
 528
 529     hr2 = pInClass->SpawnInstance(0, &pInInst);
 530     CLEANUP_ON_FAILURE(hr2);
 531
 532     // Set up parameters
 533     V_VT(&v_netbios) = VT_BSTR;
 534     V_BSTR(&v_netbios) = SysAllocString(L"1");  /* Use Netbios */
 535
 536     hr2 = pInInst->Put(L"TcpipNetbiosOptions", 0, &v_netbios, 0);
 537     CLEANUP_ON_FAILURE(hr2);
 538
 539     // Call the method
 540     hr2 = pNamespace->ExecMethod(InstancePath, MethodName, 0, NULL, pInInst,
 541                                  &pOutInst, NULL);
 542     if (!SUCCEEDED(hr2)) {
 543         ReportMessage(0,"ExecMethod SetTcpipNetbios failed",NULL,NULL, hr2);
 544         goto cleanup;
 545     }
 546
 547     // Get the EnableStatic method return value
 548     hr2 = pOutInst->Get(L"ReturnValue", 0, &v_ret_value, 0, 0);
 549     if (!SUCCEEDED(hr2)) {
 550         ReportMessage(0,"WARNING: Could not determine return value for SetTcpipNetbios ",NULL,NULL, hr2);
 551     } else {
 552         hr2 = V_I4(&v_ret_value);
 553         if (hr2 != 0)
 554             ReportMessage(0,"SetTcpipNetbios failed ", NULL,NULL,hr2);
 555         else
 556             ReportMessage(0,"SetTcpipNetbios succeeded",NULL,NULL,0);
 557     }
 558
 559  cleanup:
 560     // Free up resources
 561     VariantClear(&v_ret_value);
 562     VariantClear(&v_ip_list);
 563     VariantClear(&v_mask_list);
 564     VariantClear(&v_reg_enabled);
 565     VariantClear(&v_netbios);
 566
 567     // SysFreeString is NULL safe
 568     SysFreeString(NamespacePath);
 569     SysFreeString(ClassPath);
 570     SysFreeString(InstancePath);
 571     SysFreeString(MethodName);
 572
 573     if (pClass) pClass->Release();
 574     if (pInInst) pInInst->Release();
 575     if (pInClass) pInClass->Release();
 576     if (pOutInst) pOutInst->Release();
 577     if (pLocator) pLocator->Release();
 578     if (pNamespace) pNamespace->Release();
 579
 580     if (comInitialized)
 581         CoUninitialize();
 582
 583     return hr;
 584 }
 585
 586
 587 /**********************************************************
 588 * LoopbackBindings :  unbind all other
 589 *       protocols except TCP/IP, netbios, netbt.
 590 */
 591 extern "C" HRESULT
 592 LoopbackBindings (LPCWSTR loopback_guid)
 593 {
 594     HRESULT                     hr = 0;
 595     INetCfg                     *pCfg = NULL;
 596     INetCfgLock         *pLock = NULL;
 597     INetCfgComponent *pAdapter = NULL;
 598     IEnumNetCfgComponent *pEnumComponent = NULL;
 599     BOOL                        bLockGranted = FALSE;
 600     BOOL                        bInitialized = FALSE;
 601     BOOL                        bConfigChanged = FALSE;
 602     LPWSTR                      swName = NULL;
 603     GUID            g;
 604     wchar_t         device_guid[100];
 605
 606     ReportMessage(0,"Running LoopbackBindings()...",NULL,NULL,0);
 607
 608     hr = CoInitializeEx( NULL, COINIT_DISABLE_OLE1DDE | COINIT_APARTMENTTHREADED );
 609     CLEANUP_ON_FAILURE(hr);
 610
 611     hr = CoCreateInstance( CLSID_CNetCfg, NULL, CLSCTX_INPROC_SERVER, IID_INetCfg, (void**)&pCfg );
 612     CLEANUP_ON_FAILURE(hr);
 613
 614     hr = pCfg->QueryInterface( IID_INetCfgLock, (void**)&pLock );
 615     CLEANUP_ON_FAILURE(hr);
 616
 617     hr = pLock->AcquireWriteLock( 1000, L"AFS Configuration", NULL );
 618     CLEANUP_ON_FAILURE(hr);
 619     bLockGranted = TRUE;
 620
 621     hr = pCfg->Initialize( NULL );
 622     CLEANUP_ON_FAILURE(hr);
 623     bInitialized = TRUE;
 624
 625     hr = pCfg->EnumComponents( &GUID_DEVCLASS_NET, &pEnumComponent );
 626     CLEANUP_ON_FAILURE(hr);
 627
 628     while( pEnumComponent->Next( 1, &pAdapter, NULL ) == S_OK )
 629     {
 630         pAdapter->GetDisplayName( &swName );
 631         pAdapter->GetInstanceGuid( &g );
 632         StringFromGUID2(g, device_guid, 99);
 633
 634         if (!wcscmp( device_guid, loopback_guid )) // found loopback adapter
 635         {
 636             INetCfgComponentBindings *pBindings;
 637             INetCfgBindingPath *pPath;
 638             IEnumNetCfgBindingPath *pEnumPaths;
 639             INetCfgComponent *upper;
 640
 641             ReportMessage(0,"LoopbackBindings found", NULL, device_guid,0 );
 642
 643             hr = pAdapter->QueryInterface( IID_INetCfgComponentBindings, (void**) &pBindings);
 644             if (hr==S_OK)
 645             {
 646                 hr = pBindings->EnumBindingPaths( EBP_ABOVE, &pEnumPaths );
 647                 if (hr==S_OK)
 648                 {
 649                     while(pEnumPaths->Next( 1, &pPath, NULL ) == S_OK)
 650                     {
 651                         pPath->GetOwner( &upper );
 652
 653                         LPWSTR swId = NULL, swName = NULL;
 654
 655                         upper->GetDisplayName( &swName );
 656                         upper->GetId( &swId );
 657
 658                         ReportMessage(1,"Looking at ",NULL, swName, 0);
 659
 660                         {
 661                             ReportMessage(1,"  Moving to the end of binding order...",NULL,NULL,0);
 662                             INetCfgComponentBindings *pBindings2;
 663                             hr = upper->QueryInterface( IID_INetCfgComponentBindings, (void**) &pBindings2);
 664                             if (hr==S_OK)
 665                             {
 666                                 ReportMessage(1,"...",0,0,0);
 667                                 hr = pBindings2->MoveAfter(pPath, NULL);
 668                                 pBindings2->Release();
 669                                 bConfigChanged=TRUE;
 670                             }
 671                             if (hr==S_OK) ReportMessage(1,"success",0,0,0); else ReportMessage(0,"Binding change failed",0,0,hr);
 672
 673                         }
 674
 675                         if ( !_wcsicmp(swId, L"ms_netbios")  ||
 676                              !_wcsicmp(swId, L"ms_tcpip")    ||
 677                              !_wcsicmp(swId, L"ms_netbt")    ||
 678                              !_wcsicmp(swId, L"ms_msclient"))
 679                         {
 680                             if (pPath->IsEnabled()!=S_OK)
 681                             {
 682                                 ReportMessage(1,"  Enabling ",0,swName,0);
 683                                 hr = pPath->Enable(TRUE);
 684                                 if (hr==S_OK) ReportMessage(1,"success",0,0,0); else ReportMessage(0,"Proto failed",0,0,hr);
 685                                 bConfigChanged=TRUE;
 686                             }
 687
 688                         }
 689                         else //if (!_wcsicmp(swId, L"ms_server") || (!_wcsicmp(swId, L"ms_msclient"))
 690                         {
 691                             if (pPath->IsEnabled()==S_OK)
 692                             {
 693                                 ReportMessage(1,"  Disabling ",0,swName,0);
 694                                 hr = pPath->Enable(FALSE);
 695                                 if (hr==S_OK) ReportMessage(1,"success",0,0,0); else ReportMessage(0,"Proto failed",0,0,hr);
 696                                 bConfigChanged=TRUE;
 697                             }
 698                         }
 699
 700                         CoTaskMemFree( swName );
 701                         CoTaskMemFree( swId );
 702
 703                         pPath->Release();
 704                     }
 705                     pEnumPaths->Release();
 706                 }
 707                 pBindings->Release();
 708             } // hr==S_OK for QueryInterface IID_INetCfgComponentBindings
 709         }
 710
 711         CoTaskMemFree( swName );
 712
 713         pAdapter->Release();
 714     }
 715
 716     pEnumComponent->Release();
 717
 718     hr = 0;
 719
 720 cleanup:
 721
 722     if (bConfigChanged) pCfg->Apply();
 723
 724     if (pAdapter) pAdapter->Release();
 725
 726     if (bInitialized) pCfg->Uninitialize();
 727     if (bLockGranted) pLock->ReleaseWriteLock();
 728
 729     if (pLock) pLock->Release();
 730     if (pCfg) pCfg->Release();
 731
 732     if (hr) ReportMessage(0,"LoopbackBindings() is returning ",0,0,hr);
 733     return hr;
 734 }
 735
 736
 737 extern "C"
 738 DWORD SetIpAddress(LPCWSTR guid, LPCWSTR ip, LPCWSTR mask)
 739 {
 740     ReportMessage(0,"Running SetIpAddress()...",0,0,0);
 741     HRESULT hr = 0;
 742
 743     hr = WMIEnableStatic(FindNetworkAdapterConfigurationInstanceByGUID,
 744                         (PVOID)guid, ip, mask);
 745     if (hr == 0)
 746         FixupXPDNSRegistrations(guid);
 747     return hr;
 748 }
 749
 750 /* Set MAXLANA in the registry to the specified value, unless the existing registry value is larger */
 751 static DWORD
 752 AdjustMaxLana(DWORD dwMaxLana)
 753 {
 754
 755     LONG ret = 0;
 756     HKEY hNetBiosParamKey = NULL;
 757     DWORD dwType, dwExistingMaxLana, dwSize;
 758
 759     ReportMessage(0,"Making sure MaxLana is large enough",0,0, dwMaxLana);
 760
 761     ret = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SYSTEM\\CurrentControlSet\\Services\\NetBIOS\\Parameters"),
 762         0, KEY_ALL_ACCESS , &hNetBiosParamKey);
 763     if (ret) return ret;
 764
 765
 766
 767     dwSize = 4;
 768     ret = RegQueryValueEx(hNetBiosParamKey, _T("MaxLana"), 0, &dwType, (LPBYTE) &dwExistingMaxLana, &dwSize);
 769     if ((ret) && (ret != ERROR_MORE_DATA) && (ret != ERROR_FILE_NOT_FOUND))
 770     {
 771         RegCloseKey(hNetBiosParamKey);
 772         return ret;
 773     }
 774
 775     if ((dwType != REG_DWORD) || (ret)) dwExistingMaxLana = 0;
 776
 777     ReportMessage (1,"MaxLana is currently",0,0, dwExistingMaxLana);
 778
 779     if (dwExistingMaxLana < dwMaxLana)
 780     {
 781         ReportMessage (1,"Changing MaxLana", 0,0,dwMaxLana);
 782         ret = RegSetValueEx(hNetBiosParamKey, _T("MaxLana"), 0, REG_DWORD, (const BYTE*)&dwMaxLana, 4);
 783         if (ret)
 784         {
 785             RegCloseKey(hNetBiosParamKey);
 786             return ret;
 787         }
 788
 789     }
 790
 791     RegCloseKey(hNetBiosParamKey);
 792     return 0;
 793
 794
 795 }
 796
 797 extern "C"
 798 BOOL UpdateHostsFile( LPCWSTR swName, LPCWSTR swIp, LPCSTR szFilename, BOOL bPre )
 799 {
 800     char szIp[2048], szName[2048];
 801     char etcPath[MAX_PATH];
 802     char tempPath[MAX_PATH];
 803     char buffer[2048], temp[2048];
 804     char *str;
 805     HRESULT rv;
 806     DWORD fa,len;
 807     FILE *hFile, *hTemp;
 808     size_t nameLen;
 809
 810     _snprintf(szIp, 2047, "%S", swIp);
 811     _snprintf(szName, 2047, "%S", swName);
 812     strupr(szName);
 813     nameLen = strlen(szName);
 814     ReportMessage(0,"Starting UpdateHostsFile() on file",szFilename,0,0);
 815
 816     rv = SHGetFolderPathA( NULL, CSIDL_SYSTEM, NULL, SHGFP_TYPE_CURRENT, etcPath );
 817     if (rv != S_OK)
 818         return FALSE;
 819
 820     strcat( etcPath, ETCDIR );
 821
 822     fa = GetFileAttributesA( etcPath );
 823
 824     if (fa == INVALID_FILE_ATTRIBUTES)
 825     {
 826         // the directory doesn't exist
 827         // it should be there. non-existence implies more things are wrong
 828         ReportMessage(0, "Path does not exist ", etcPath,0,0 );
 829         return FALSE;
 830     }
 831
 832     strcpy( tempPath, etcPath );
 833     strcat( etcPath, "\\" );
 834     strcat( etcPath, szFilename );
 835
 836     fa = GetFileAttributesA( etcPath );
 837
 838     if (fa == INVALID_FILE_ATTRIBUTES)
 839     {
 840         ReportMessage(0, "File not found. Creating...", szFilename,0,0);
 841
 842         hFile = fopen( etcPath, "w" );
 843         if (!hFile)
 844         {
 845             ReportMessage(0,"FAILED : can't create file",etcPath,0,errno);
 846             return FALSE;
 847         }
 848
 849         fprintf(hFile, "%s\t%s%s\n", szIp, szName, (bPre)?"\t#PRE":"");
 850
 851         fclose( hFile );
 852
 853         ReportMessage(1,"done",0,0,0);
 854     }
 855     else // the file exists. parse and update
 856     {
 857
 858         ReportMessage(1, "Updating file ...",szFilename,0,0 );
 859
 860         hFile = fopen( etcPath, "r");
 861         if (!hFile)
 862         {
 863             ReportMessage(0,"FAILED : can't open file",etcPath,0,errno);
 864             return FALSE;
 865         }
 866
 867         strcat( tempPath, szFilename );
 868         strcat( tempPath, ".tmp" );
 869         hTemp = fopen( tempPath, "w");
 870         if (!hTemp)
 871         {
 872             ReportMessage(0,"FAILED : can't create temp file",tempPath,0,errno);
 873             fclose(hFile);
 874             return FALSE;
 875         }
 876
 877         while(fgets( buffer, 2046, hFile))
 878         {
 879             size_t len;
 880
 881             strcpy( temp, buffer );
 882             strupr( temp );
 883             len = strlen(temp);
 884
 885             if ((len < 1) || (temp[len-1] != '\n'))
 886                 strcat(temp, "\n");
 887
 888             if (!(str = strstr(temp, szName)))
 889             {
 890                 fputs( temp, hTemp );
 891             }
 892             else
 893             {
 894                 // check for FOOBAFS or AFSY
 895                 if ( (str == temp) || (!str[nameLen]) || (!isspace(*(str-1))) || (!isspace(str[nameLen])) )
 896                     fputs( temp, hTemp );
 897             }
 898         }
 899
 900         len = 2048;
 901         GetComputerNameA( buffer, &len );
 902         buffer[11] = 0;
 903         fprintf( hTemp, "%s\t%s%s\n", szIp, szName, (bPre)?"\t#PRE":"");
 904
 905         fclose( hTemp );
 906         fclose( hFile );
 907
 908         strcpy(buffer, etcPath);
 909         strcat(buffer, ".old");
 910
 911         if (!DeleteFileA(buffer)) {
 912             DWORD status;
 913             int i;
 914             char * eos;
 915
 916             status = GetLastError();
 917             if (status == ERROR_ACCESS_DENIED) {
 918                 /* try changing the file attribtues. */
 919                 if (SetFileAttributesA(buffer, FILE_ATTRIBUTE_NORMAL) &&
 920                     DeleteFileA(buffer)) {
 921                     status = 0;
 922                     ReportMessage(0,"Changed attributes and deleted back host file", buffer, 0, 0);
 923                 }
 924             }
 925             if (status && status != ERROR_FILE_NOT_FOUND) {
 926                 /* we can't delete the file.  Try to come up with
 927                    a different name that's not already taken. */
 928                 srand(GetTickCount());
 929                 eos = buffer + strlen(buffer);
 930                 for(i=0; i < 50; i++) {
 931                     itoa(rand(), eos, 16);
 932                     if (GetFileAttributesA(buffer) == INVALID_FILE_ATTRIBUTES &&
 933                         GetLastError() == ERROR_FILE_NOT_FOUND)
 934                         break;
 935                 }
 936                 /* At this point if we don't have a unique name, we just let the rename
 937                    fail.  Too bad. */
 938             }
 939         }
 940
 941         if (!MoveFileA( etcPath, buffer )) {
 942             ReportMessage(0,"FAILED: Can't rename old file", etcPath, 0, GetLastError());
 943             return FALSE;
 944         }
 945
 946         if (!MoveFileA( tempPath, etcPath ) != 0)
 947         {
 948             ReportMessage(0,"FAILED : Can't rename new file", tempPath, 0, GetLastError());
 949             return FALSE;
 950         }
 951
 952     }
 953
 954     return TRUE;
 955 }
 956
 957 #ifdef TEST
 958 #if 0
 959 int
 960 wmain(
 961     int argc,
 962     wchar_t* argv[]
 963     )
 964 {
 965     if (argc < 3)
 966     {
 967         printf("usage: %S ip mask\n"
 968                "  example: %S 10.0.0.1 255.0.0.0", argv[0], argv[0]);
 969         return 0;
 970     }
 971
 972     return WMIEnableStatic(FindNetworkAdapterConfigurationInstanceByGUID,
 973                            L"{B4981E32-551C-4164-96B6-B8874BD2E555}",
 974                            argv[1], argv[2]);
 975 }
 976 #else
 977 int
 978 wmain(
 979     int argc,
 980     wchar_t* argv[]
 981     )
 982 {
 983     if (argc < 4)
 984     {
 985         printf("usage: %S adapter_guid ip mask\n"
 986                "  example: %S {B4981E32-551C-4164-96B6-B8874BD2E555} "
 987                "10.0.0.1 255.0.0.0", argv[0], argv[0]);
 988         return 0;
 989     }
 990
 991     return WMIEnableStatic(FindNetworkAdapterConfigurationInstanceByGUID,
 992                            argv[1], argv[2], argv[3]);
 993 }
 994 #endif
 995 #endif
 996