/*++

Copyright (c) 1996 Microsoft Corporation

Module Name:

    scedll.cpp

Abstract:

    SCE Client dll initialization

Author:

    Jin Huang (jinhuang) 23-Jan-1998 created

--*/
#include "headers.h"
#include "sceutil.h"
#include <winsafer.h>
#include <winsaferp.h>
#include <alloca.h>
#include <objbase.h>
#include <initguid.h>
#include <wbemidl.h>
#include <wbemprov.h>
#include <atlbase.h>

CComModule _Module;

#include <atlcom.h>

extern HINSTANCE MyModuleHandle;
extern CRITICAL_SECTION DiagnosisPolicypropSync;
extern CRITICAL_SECTION PolicyNotificationSync;
extern LIST_ENTRY ScepNotifyList;

BOOL
UninitializeChangeNotify();

#define GPT_SCEDLL_PATH   TEXT("Software\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon\\GPExtensions\\SceCli")

#define SCE_REGVALUE_DEFAULT_FILE  TEXT("sceregvl.inf")

#define SAM_FILTER_PATH   TEXT("System\\CurrentControlSet\\Control\\Lsa")
#define SAM_FILTER_VALUE  TEXT("Notification Packages")

DWORD
ScepQuerySamFilterValue(
    IN HKEY hKey,
    IN BOOL bAdd,
    OUT PWSTR *pmszValue,
    OUT DWORD *pcbSize,
    OUT BOOL *pbChanged
    );

VOID
ScepInitClientData(
    void
    );

VOID
ScepUnInitClientData(
    void
    );


DWORD
DllpModifySamFilterRegistration(
    IN BOOL bAdd
    )
{
    DWORD lResult;
    HKEY hKey=NULL;

    if(( lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
                              SAM_FILTER_PATH,
                              0,
                              KEY_READ | KEY_WRITE,
                              &hKey
                             )) == ERROR_SUCCESS ) {

        DWORD dSize=0;
        PWSTR mszValue=NULL;
        BOOL bChanged=FALSE;

        //
        // find out if "scecli" is registered already
        //

        lResult = ScepQuerySamFilterValue(hKey,
                                          bAdd,
                                          &mszValue,
                                          &dSize,
                                          &bChanged
                                         );

        if ( lResult == ERROR_SUCCESS &&
             bChanged && mszValue ) {
            //
            // set the value
            //
            lResult = RegSetValueEx (
                            hKey,
                            SAM_FILTER_VALUE,
                            0,
                            REG_MULTI_SZ,
                            (BYTE *)mszValue,
                            dSize
                            );
        }

        RegCloseKey(hKey);

    }

    return lResult;
}

DWORD
ScepQuerySamFilterValue(
    IN HKEY hKey,
    IN BOOL bAdd,
    OUT PWSTR *pmszValue,
    OUT DWORD *pcbSize,
    OUT BOOL *pbChanged
    )
/*
Routine Description:

    query the existing notification packages. Add or remove "scecli" to the
    packages depending on the flag "bAdd". The packages are in MULTI_SZ
    format.

Arguments:

   hKey  - the base key handle off where the packages are saved

   bAdd  - if TRUE, add "scecli" to the packages if it's not there
           if FALSE, remove "scecli" from the packages

   pmszValue   - the modified packages value (in MULTI_SZ format)

   pcbSize     - the size of the packages (in bytes)

   pbChanged   - TRUE if anything is changed

Return Value:


   Win32 error code

*/
{
    DWORD lResult;
    DWORD RegType=REG_MULTI_SZ;
    DWORD dSize=0;
    PWSTR msz=NULL;

    if(( lResult = RegQueryValueEx(hKey,
                                 SAM_FILTER_VALUE,
                                 0,
                                 &RegType,
                                 NULL,
                                 &dSize
                                )) == ERROR_SUCCESS ) {
        //
        // query existing registered packages
        //
        if ( RegType == REG_MULTI_SZ ) {

            msz = (PWSTR)LocalAlloc( LMEM_ZEROINIT, dSize+2*sizeof(TCHAR));

            if ( msz ) {

                lResult = RegQueryValueEx(hKey,
                                         SAM_FILTER_VALUE,
                                         0,
                                         &RegType,
                                         (UCHAR *)msz,
                                         &dSize
                                        );
            } else {
                lResult = ERROR_NOT_ENOUGH_MEMORY;
            }
        } else {

            lResult = ERROR_FILE_NOT_FOUND;
        }

    }

    if ( lResult == ERROR_FILE_NOT_FOUND ||
         lResult == ERROR_PATH_NOT_FOUND ) {
        //
        // if doesn't find the value, ignore the error
        //
        lResult = ERROR_SUCCESS;
    }

    *pbChanged = FALSE;
    *pcbSize = 0;
    *pmszValue = NULL;

    if ( lResult == ERROR_SUCCESS &&
         msz == NULL &&
         bAdd ) {

        //
        // add scecli to the multi-sz value
        // note, since msz is NULL, no need to remove scecli for unregister
        //
        *pmszValue = (PWSTR)LocalAlloc(0, 16);
        if ( *pmszValue ) {

            wcscpy(*pmszValue, L"scecli");
            (*pmszValue)[6] = L'\0';
            (*pmszValue)[7] = L'\0';
            *pcbSize = 16;

            *pbChanged = TRUE;

        } else {
            lResult = ERROR_NOT_ENOUGH_MEMORY;
        }
    }

    if ( ERROR_SUCCESS == lResult &&
         msz != NULL ) {

        //
        // there are existing values in the field
        //

        PWSTR pStart=msz;
        DWORD Len=0, Len2;
        BOOL bFound = FALSE;

        while ( pStart && pStart[0] != L'\0' &&
                Len <= dSize/2 ) {

            if ( _wcsicmp(pStart, L"scecli") == 0 ) {
                bFound = TRUE;
                break;
            } else {
                Len2 = wcslen(pStart)+1;
                pStart = pStart + Len2;
                Len += Len2;
            }
        }

        //
        // add/remove scecli
        //
        if ( bFound ) {

            //
            // pStart is pointing to scecli, remove it
            //
            if ( !bAdd ) {

                Len = (DWORD)(pStart - msz);
                Len2 = wcslen(pStart);

                *pmszValue = (PWSTR)LocalAlloc(LPTR, dSize + 2*sizeof(TCHAR) - (Len2 + 1)*sizeof(TCHAR) );

                if ( *pmszValue ) {

                    memcpy(*pmszValue, msz, Len*sizeof(TCHAR));
                    memcpy(*pmszValue+Len, pStart+Len2+1, dSize - (Len+Len2+1)*sizeof(TCHAR));

                    *pcbSize = dSize - (Len2+1)*sizeof(TCHAR);

                    *pbChanged = TRUE;
                } else {

                    lResult = ERROR_NOT_ENOUGH_MEMORY;
                }
            }

        } else {

            //
            // not found, add scecli
            //
            if ( bAdd ) {

               *pmszValue = (PWSTR)LocalAlloc(LPTR, dSize + 2*sizeof(TCHAR) + 7*sizeof(TCHAR) );

               if ( *pmszValue ) {

                   memcpy(*pmszValue, msz, dSize);

                   Len2 = 1;
                   while ( msz[dSize/2-Len2] == L'\0' ) {
                      Len2++;
                   }

                   wcscpy(*pmszValue+dSize/2-Len2+2, L"scecli");

                   *pcbSize = dSize + (2-Len2+wcslen(TEXT("scecli"))+2)*sizeof(TCHAR);

                   *pbChanged = TRUE;

               } else {

                   lResult = ERROR_NOT_ENOUGH_MEMORY;
               }
            }
        }
    }

    if ( msz ) {
       LocalFree(msz);
    }


    return lResult;
}

/*=============================================================================
**  Procedure Name:     DllMain
**
**  Arguments:
**
**
**
**  Returns:    0 = SUCCESS
**             !0 = ERROR
**
**  Abstract:
**
**  Notes:
**
**===========================================================================*/
BOOL WINAPI DllMain(
    IN HANDLE DllHandle,
    IN ULONG ulReason,
    IN LPVOID Reserved )
{

    switch(ulReason) {

    case DLL_PROCESS_ATTACH:

        MyModuleHandle = (HINSTANCE)DllHandle;
        ScepInitNameTable();
        (VOID) ScepInitClientData();

        //
        // initialize dynamic stack allocation
        //

        SafeAllocaInitialize(SAFEALLOCA_USE_DEFAULT,
                             SAFEALLOCA_USE_DEFAULT,
                             NULL,
                             NULL
                            );
        //
        // Fall through to process first thread
        //

#if DBG
        DebugInitialize();
#endif

    case DLL_THREAD_ATTACH:

        break;

    case DLL_PROCESS_DETACH:

        UninitializeChangeNotify();
        (VOID) ScepUnInitClientData();

#if DBG
        DebugUninit();
#endif
        break;

    case DLL_THREAD_DETACH:

        break;
    }

    return TRUE;
}

VOID
ScepInitClientData()
/*
Routine Description:

    Initialize global data for the client

Arguments:

    None

Return Value:

    None
*/
{
    /*
     initialize critical section that protects global rsop pointers (namespace, status, logfilename)
     by serializing multiple diagnostic modes/policy prop (planning mode uses no globals and is synch)
     this is necessary because the asynch thread that calls back the client and needs the above
     globals to be preserved (simple thread variables will not do since the asynch thread doesn't know
     that it is calling back to the same client thread that it got spawned by)

     logic for the acquizition/release of the crit sec is as follows for two cases that arise:


     case (a) background thread (no asynch thread spawned)

        Exported Policy Function (grab cs) ---> all GPO processing is synch ---> client returns (release cs)

    case (b) foreground thread (asynch thread is spawned for slow config areas)

        Exported Policy Function (grab cs) ---> try to spawn asynch thread ---> IF asynch thread spawned succ-
                        essfully it (releases cs) ELSE the synch thread (releases cs)
     */

    InitializeCriticalSection(&DiagnosisPolicypropSync);

    //
    // Initialize critical section used by policy notification from LSA/SAM.
    // The critical section protects a global counter for the number of
    // notifications sent to SCE. If the count is 0, there is no pending
    // notification that has not been added to the queue in SCE server;
    // otherwise, some notifications have been sent but has not returned yet.
    //
    // The global count is used to control if policy propagation should be
    // allowed.
    //
    InitializeCriticalSection(&PolicyNotificationSync);

    InitializeListHead( &ScepNotifyList );
}

VOID
ScepUnInitClientData()
/*
Routine Description:

    Uninitialize global data for the client

Arguments:

    None

Return Value:

    None
*/
{

    DeleteCriticalSection(&DiagnosisPolicypropSync);

    DeleteCriticalSection(&PolicyNotificationSync);
}

/////////////////////////////////////////////////////////////////////////////
// DllRegisterServer - Adds entries to the system registry

STDAPI DllRegisterServer(void)
{
    HKEY hKey;
    LONG lResult;
    DWORD dwDisp;
    FILETIME Time;
    
    WCHAR szBuffer[MAX_PATH];
    WCHAR szMofFile[MAX_PATH + 30];
    HRESULT hr;

/*  the old interface
    lResult = RegCreateKeyEx (HKEY_LOCAL_MACHINE, GPT_SCEDLL_PATH, 0, NULL,
                              REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL,
                              &hKey, &dwDisp);

    if (lResult != ERROR_SUCCESS)
    {
        return lResult;
    }


    RegSetValueEx (hKey, TEXT("ProcessGPO"), 0, REG_SZ, (LPBYTE)TEXT("SceWinlogonConfigureSystem"),
                   (lstrlen(TEXT("SceWinlogonConfigureSystem")) + 1) * sizeof(TCHAR));

    RegSetValueEx (hKey, TEXT("DllName"), 0, REG_EXPAND_SZ, (LPBYTE)TEXT("scecli.dll"),
                   (lstrlen(TEXT("scecli.dll")) + 1) * sizeof(TCHAR));
*/

    //
    // delete the old interface registration
    //
    RegDeleteKey ( HKEY_LOCAL_MACHINE, GPT_SCEDLL_PATH );

    //
    // register the new interface
    //
    lResult = RegCreateKeyEx (HKEY_LOCAL_MACHINE, GPT_SCEDLL_NEW_PATH, 0, NULL,
                              REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL,
                              &hKey, &dwDisp);

    if (lResult != ERROR_SUCCESS)
    {
        return lResult;
    }


    RegSetValueEx (hKey, TEXT("ProcessGroupPolicy"), 0, REG_SZ, (LPBYTE)TEXT("SceProcessSecurityPolicyGPO"),
                   (lstrlen(TEXT("SceProcessSecurityPolicyGPO")) + 1) * sizeof(TCHAR));

    // RSOP Planning mode API
    RegSetValueEx (hKey, TEXT("GenerateGroupPolicy"), 0, REG_SZ, (LPBYTE)TEXT("SceGenerateGroupPolicy"),
                   (lstrlen(TEXT("SceGenerateGroupPolicy")) + 1) * sizeof(TCHAR));

    // RSOP Planning mode logging default (planning.log) -
    // turn on logging (ignore any errors)
    dwDisp = 1;
    RegSetValueEx (hKey, TEXT("ExtensionRsopPlanningDebugLevel"), 0, REG_DWORD, (LPBYTE)&dwDisp, sizeof(DWORD));

    // RSOP Diagnosis mode API
    RegSetValueEx (hKey, TEXT("ProcessGroupPolicyEx"), 0, REG_SZ, (LPBYTE)TEXT("SceProcessSecurityPolicyGPOEx"),
                   (lstrlen(TEXT("SceProcessSecurityPolicyGPOEx")) + 1) * sizeof(TCHAR));

    // RSOP Diagnosis mode or regular Policy Propagation  logging default (diagnosis.log and/or winlogon.log)
    // turn on logging (ignore any errors)
    dwDisp = 1;
    RegSetValueEx (hKey, TEXT("ExtensionDebugLevel"), 0, REG_DWORD, (LPBYTE)&dwDisp, sizeof(DWORD));

    RegSetValueEx (hKey, TEXT("DllName"), 0, REG_EXPAND_SZ, (LPBYTE)TEXT("scecli.dll"),
                   (lstrlen(TEXT("scecli.dll")) + 1) * sizeof(TCHAR));

    RegSetValueEx (hKey, NULL, 0, REG_SZ, (LPBYTE)TEXT("Security"),
                   (lstrlen(TEXT("Security")) + 1) * sizeof(TCHAR));
    RegDeleteValue (hKey, TEXT("<No Name>"));

    dwDisp = 1;
    RegSetValueEx (hKey, TEXT("NoUserPolicy"), 0, REG_DWORD, (LPBYTE)&dwDisp, sizeof(DWORD));
    RegSetValueEx (hKey, TEXT("NoGPOListChanges"), 0, REG_DWORD, (LPBYTE)&dwDisp, sizeof(DWORD));
    RegSetValueEx (hKey, TEXT("EnableAsynchronousProcessing"), 0, REG_DWORD, (LPBYTE)&dwDisp, sizeof(DWORD));

    dwDisp = 960;
    RegSetValueEx (hKey, TEXT("MaxNoGPOListChangesInterval"), 0, REG_DWORD, (LPBYTE)&dwDisp, sizeof(DWORD));

    RegCloseKey (hKey);

    // EFS recovery policy extension
    lResult = RegCreateKeyEx (HKEY_LOCAL_MACHINE, GPT_EFS_NEW_PATH, 0, NULL,
                              REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL,
                              &hKey, &dwDisp);

    if (lResult != ERROR_SUCCESS)
    {
        return lResult;
    }


    RegSetValueEx (hKey, TEXT("ProcessGroupPolicy"), 0, REG_SZ, (LPBYTE)TEXT("SceProcessEFSRecoveryGPO"),
                   (lstrlen(TEXT("SceProcessEFSRecoveryGPO")) + 1) * sizeof(TCHAR));

    RegSetValueEx (hKey, TEXT("DllName"), 0, REG_EXPAND_SZ, (LPBYTE)TEXT("scecli.dll"),
                   (lstrlen(TEXT("scecli.dll")) + 1) * sizeof(TCHAR));

    RegSetValueEx (hKey, NULL, 0, REG_SZ, (LPBYTE)TEXT("EFS recovery"),
                   (lstrlen(TEXT("EFS recovery")) + 1) * sizeof(TCHAR));
    RegDeleteValue (hKey, TEXT("<No Name>"));

    dwDisp = 1;
    RegSetValueEx (hKey, TEXT("NoUserPolicy"), 0, REG_DWORD, (LPBYTE)&dwDisp, sizeof(DWORD));
    RegSetValueEx (hKey, TEXT("NoGPOListChanges"), 0, REG_DWORD, (LPBYTE)&dwDisp, sizeof(DWORD));

    RegDeleteValue (hKey, TEXT("RequireSuccessfulRegistry") );
    RegSetValueEx (hKey, TEXT("RequiresSuccessfulRegistry"), 0, REG_DWORD, (LPBYTE)&dwDisp, sizeof(DWORD));

    RegCloseKey (hKey);

    //
    // register default SAFER policy to disable executables embedded in Outlook
    //

    lResult = RegCreateKeyEx (HKEY_LOCAL_MACHINE,
                              SAFER_HKLM_REGBASE,
                              0,
                              NULL,
                              REG_OPTION_NON_VOLATILE,
                              KEY_WRITE,
                              NULL,
                              &hKey,
                              &dwDisp);

    if (lResult != ERROR_SUCCESS)
    {
        return lResult;
    }

    RegCloseKey (hKey);

    lResult = RegCreateKeyEx (HKEY_LOCAL_MACHINE,
                              SAFER_HKLM_REGBASE SAFER_REGKEY_SEPERATOR SAFER_CODEIDS_REGSUBKEY,
                              0,
                              NULL,
                              REG_OPTION_NON_VOLATILE,
                              KEY_WRITE,
                              NULL,
                              &hKey,
                              &dwDisp);

    if (lResult != ERROR_SUCCESS)
    {
        return lResult;
    }


    RegSetValueEx (hKey,
                   SAFER_EXETYPES_REGVALUE,
                   0,
                   REG_MULTI_SZ,
                   (LPBYTE)SAFER_DEFAULT_EXECUTABLE_FILE_TYPES,
                   sizeof(SAFER_DEFAULT_EXECUTABLE_FILE_TYPES));

    dwDisp = 0x00000001;

    RegSetValueEx (hKey,
                   SAFER_TRANSPARENTENABLED_REGVALUE,
                   0,
                   REG_DWORD,
                   (LPBYTE)&dwDisp,
                   sizeof(DWORD));

    dwDisp = 0x00040000;

    RegSetValueEx (hKey,
                   SAFER_DEFAULTOBJ_REGVALUE,
                   0,
                   REG_DWORD,
                   (LPBYTE)&dwDisp,
                   sizeof(DWORD));

    dwDisp = 0x00000000;

    RegSetValueEx (hKey,
                   SAFER_AUTHENTICODE_REGVALUE,
                   0,
                   REG_DWORD,
                   (LPBYTE)&dwDisp,
                   sizeof(DWORD));

    dwDisp = 0x00000000;

    RegSetValueEx (hKey,
                   SAFER_POLICY_SCOPE,
                   0,
                   REG_DWORD,
                   (LPBYTE)&dwDisp,
                   sizeof(DWORD));

    RegCloseKey (hKey);

    lResult = RegCreateKeyEx (HKEY_LOCAL_MACHINE,
                              SAFER_HKLM_REGBASE SAFER_REGKEY_SEPERATOR SAFER_CODEIDS_REGSUBKEY \
                              SAFER_REGKEY_SEPERATOR SAFER_LEVEL_ZERO,
                              0,
                              NULL,
                              REG_OPTION_NON_VOLATILE,
                              KEY_WRITE,
                              NULL,
                              &hKey,
                              &dwDisp);

    if (lResult != ERROR_SUCCESS)
    {
        return lResult;
    }

    RegCloseKey (hKey);

    lResult = RegCreateKeyEx (HKEY_LOCAL_MACHINE,
                              SAFER_HKLM_REGBASE SAFER_REGKEY_SEPERATOR SAFER_CODEIDS_REGSUBKEY \
                              SAFER_REGKEY_SEPERATOR SAFER_LEVEL_ZERO SAFER_REGKEY_SEPERATOR SAFER_PATHS_REGSUBKEY,
                              0,
                              NULL,
                              REG_OPTION_NON_VOLATILE,
                              KEY_WRITE,
                              NULL,
                              &hKey,
                              &dwDisp);

    if (lResult != ERROR_SUCCESS)
    {
        return lResult;
    }

    RegCloseKey (hKey);

    lResult = RegCreateKeyEx (HKEY_LOCAL_MACHINE,
                              SAFER_HKLM_REGBASE SAFER_REGKEY_SEPERATOR SAFER_CODEIDS_REGSUBKEY \
                              SAFER_REGKEY_SEPERATOR SAFER_LEVEL_ZERO SAFER_REGKEY_SEPERATOR \
                              SAFER_PATHS_REGSUBKEY SAFER_REGKEY_SEPERATOR SAFER_DEFAULT_RULE_GUID,
                              0,
                              NULL,
                              REG_OPTION_NON_VOLATILE,
                              KEY_WRITE,
                              NULL,
                              &hKey,
                              &dwDisp);

    if (lResult != ERROR_SUCCESS)
    {
        return lResult;
    }

    RegSetValueEx (hKey,
                   SAFER_IDS_DESCRIPTION_REGVALUE,
                   0,
                   REG_SZ,
                   (LPBYTE)&L"",
                   sizeof(L""));


    dwDisp = 0x00000000;

    RegSetValueEx (hKey,
                   SAFER_IDS_SAFERFLAGS_REGVALUE,
                   0,
                   REG_DWORD,
                   (LPBYTE)&dwDisp,
                   sizeof(DWORD));

    RegSetValueEx (hKey,
                   SAFER_IDS_ITEMDATA_REGVALUE,
                   0,
                   REG_EXPAND_SZ,
                   (LPBYTE)&L"%HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders\\Cache%OLK*",
                   sizeof(L"%HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders\\Cache%OLK*"));

    GetSystemTimeAsFileTime(&Time);

    RegSetValueEx (hKey,
                   SAFER_IDS_LASTMODIFIED_REGVALUE,
                   0,
                   REG_QWORD,
                   (LPBYTE)&Time,
                   sizeof(FILETIME));

    RegCloseKey (hKey);

    //
    // turn on default logging for component installs
    //
    if ( ERROR_SUCCESS == RegCreateKeyEx (HKEY_LOCAL_MACHINE, SCE_ROOT_PATH, 0, NULL,
                                  REG_OPTION_NON_VOLATILE, KEY_WRITE | KEY_QUERY_VALUE | KEY_SET_VALUE, NULL,
                                  &hKey, &dwDisp) ) {
        dwDisp = 1;
        RegSetValueEx (hKey, TEXT("SetupCompDebugLevel"), 0, REG_DWORD, (LPBYTE)&dwDisp, sizeof(DWORD));

        //
        // create value DefaultTemplate = C:\Windows\Inf\SecRecs.INF
        //

        DWORD   dwType = REG_SZ;

        if (ERROR_FILE_NOT_FOUND == RegQueryValueEx(hKey, TEXT("DefaultTemplate"), NULL, &dwType, NULL, NULL) ){

            WCHAR   szFileName[MAX_PATH*2+1];

            szFileName[0] = L'\0';

            GetSystemWindowsDirectory(szFileName, MAX_PATH);

            wcscat(szFileName, L"\\inf\\secrecs.inf");

            if ((DWORD)-1 != GetFileAttributes( szFileName ) ) {

                RegSetValueEx (hKey,
                               TEXT("DefaultTemplate"),
                               0,
                               REG_SZ,
                               (LPBYTE)&szFileName,
                               (wcslen(szFileName) + 1) * sizeof(WCHAR));

            }
        }

        RegCloseKey (hKey);
    }


    //
    // compile scersop.mof
    //

    szBuffer[0] = L'\0';
    szMofFile[0] = L'\0';

    if ( GetSystemDirectory( szBuffer, MAX_PATH ) ) {

        LPWSTR sz = szBuffer + wcslen(szBuffer);
        if ( sz != szBuffer && *(sz-1) != L'\\') {
            *sz++ = L'\\';
            *sz = L'\0';
        }


        wcscpy(szMofFile, szBuffer);
        wcscat(szMofFile, L"Wbem\\SceRsop.mof");

        if ((DWORD)-1 != GetFileAttributes(szMofFile)) {


            hr = ::CoInitialize (NULL);

            if (SUCCEEDED(hr)) {

                //
                // Get the MOF compiler interface
                //

                CComPtr<IMofCompiler> srpMof;
                hr = ::CoCreateInstance (CLSID_MofCompiler, NULL, CLSCTX_INPROC_SERVER, IID_IMofCompiler, (void **)&srpMof);

                if (SUCCEEDED(hr)) {
                    WBEM_COMPILE_STATUS_INFO  stat;

                    //
                    // compile RSOP mof
                    //

                    hr = srpMof->CompileFile( szMofFile,
                                              NULL,
                                              NULL,
                                              NULL,
                                              NULL,
                                              0,
                                              0,
                                              0, 
                                              &stat
                                            );

                }

                ::CoUninitialize();
            }
        }
    }

    //
    // register the default registry values
    //

    SceRegisterRegValues(SCE_REGVALUE_DEFAULT_FILE);

    //
    // register SAM policy filter
    //

    DllpModifySamFilterRegistration(TRUE);

    return S_OK;
}


/////////////////////////////////////////////////////////////////////////////
// DllUnregisterServer - Removes entries from the system registry

STDAPI DllUnregisterServer(void)
{
    //
    // delete old interface
    //
    RegDeleteKey (HKEY_LOCAL_MACHINE, GPT_SCEDLL_PATH);

    //
    // delete new interfaces
    //
    RegDeleteKey (HKEY_LOCAL_MACHINE, GPT_SCEDLL_NEW_PATH);
    RegDeleteKey (HKEY_LOCAL_MACHINE, GPT_EFS_NEW_PATH);

    DllpModifySamFilterRegistration(FALSE);

    return S_OK;
}