Initial publish

1 files changed, 271 insertions, 0 deletions

diff --git a/general/filehistory/exe/fhsetup.cpp b/general/filehistory/exe/fhsetup.cpp
new file mode 100644
index 00000000..41edfd02
--- /dev/null
+++ b/general/filehistory/exe/fhsetup.cpp
@@ -0,0 +1,271 @@
+//
+// File History Sample Setup Tool
+// Copyright (c) Microsoft Corporation.  All Rights Reserved.
+//
+
+#include <fhsetup.h>
+
+HRESULT ScheduleBackups()
+/*++
+
+Routine Description:
+
+    This function starts the File History service if it is stopped
+    and schedules regular backups.
+
+Arguments:
+
+    None
+
+Return Value:
+
+    S_OK if successful
+    HRESULT from underlying functions
+
+--*/
+{
+    HRESULT backupHr = S_OK;
+    HRESULT pipeHr = S_OK;
+    FH_SERVICE_PIPE_HANDLE pipe = NULL;
+
+    pipeHr = FhServiceOpenPipe(TRUE, &pipe);
+    if (SUCCEEDED(pipeHr))
+    {
+        backupHr = FhServiceReloadConfiguration(pipe);
+        pipeHr = FhServiceClosePipe(pipe);
+    }
+
+    // The HRESULT from the backup operation is more important than
+    // the HRESULT from pipe operations
+    return FAILED(backupHr) ? backupHr : pipeHr;
+}
+
+HRESULT ConfigureFileHistory(
+    _In_ PWSTR TargetPath
+    )
+/*++
+
+Routine Description:
+
+    This function configures a target for File History.
+    It will only succeed if the user has never configured File History
+    before and there is no File History data on the target.
+
+Arguments:
+
+    TargetPath -
+        The path of the File History target
+
+Return Value:
+
+    S_OK if successful
+    E_INVALIDARG if TargetPath is NULL
+    E_FAIL if configuration failed because File History is disabled by
+           group policy or the target is not valid
+    HRESULT from underlying functions
+
+--*/
+{
+    HRESULT hr = S_OK;
+    FH_BACKUP_STATUS backupStatus;
+    FH_DEVICE_VALIDATION_RESULT validationResult;
+    CComPtr<IFhConfigMgr> configMgr;
+    CComBSTR targetPath;
+    CComBSTR targetName;
+
+    // TargetPath must not be NULL
+    if (TargetPath == NULL)
+    {
+        hr = E_INVALIDARG;
+        goto Cleanup;
+    }
+    
+    // Copy the target path into a local variable and set the target name
+    // to an empty string to allow the config manager to set a name
+    _ATLTRY
+    {
+        targetPath = TargetPath;
+        targetName = L"";
+    }
+    _ATLCATCH(e)
+    {
+        hr = e;
+        goto Cleanup;
+    }
+
+    // The configuration manager is used to create and load configuration 
+    // files, get/set the backup status, validate a target, etc
+    hr = configMgr.CoCreateInstance(CLSID_FhConfigMgr);
+    if (FAILED(hr))
+    {
+        wprintf(L"Error: CoCreateInstance failed (0x%X)\n", hr);
+        goto Cleanup;
+    }
+    
+    // Create a new default configuration file - do not overwrite if one
+    // already exists
+    wprintf(L"Creating default configuration\n");
+    hr = configMgr->CreateDefaultConfiguration(FALSE);
+    if (FAILED(hr))
+    {
+        if (hr == FHCFG_E_CONFIG_ALREADY_EXISTS)
+        {
+            wprintf(L"Error: File History has previously been configured\n");
+        }
+        else
+        {
+            wprintf(L"Error: CreateDefaultConfiguration failed (0x%X)\n", hr);
+        }
+        goto Cleanup;
+    }
+
+    // Check the backup status
+    // If File History is disabled by group policy, quit
+    wprintf(L"Getting backup status\n");
+    hr = configMgr->GetBackupStatus(&backupStatus);
+    if (FAILED(hr))
+    {
+        wprintf(L"Error: GetBackupStatus failed (0x%X)\n", hr);
+        goto Cleanup;
+    }
+    if (backupStatus == FH_STATUS_DISABLED_BY_GP)
+    {
+        wprintf(L"Error: File History is disabled by group policy\n");
+        hr = E_FAIL;
+        goto Cleanup;
+    }
+
+    // Make sure the target is valid to be used for File History
+    wprintf(L"Validating target\n");
+    hr = configMgr->ValidateTarget(targetPath, &validationResult);
+    if (FAILED(hr))
+    {
+        wprintf(L"Error: ValidateTarget failed (0x%X)\n", hr);
+        goto Cleanup;
+    }
+    if (validationResult != FH_VALID_TARGET)
+    {
+        // If the target is inaccessible, read-only, an invalid drive type
+        // (such as a CD), already being used for File History, or part of
+        // the protected namespace - don't enable File History
+        wprintf(L"Error: %ws is not a valid target\n", targetPath.m_str);
+        hr = E_FAIL;
+        goto Cleanup;
+    }
+
+    // Provision the target to be used for File History and set
+    // it as the default target
+    wprintf(L"Provisioning and setting target\n");
+    hr = configMgr->ProvisionAndSetNewTarget(targetPath, targetName);
+    if (FAILED(hr))
+    {
+        wprintf(L"Error: ProvisionAndSetNewTarget failed (0x%X)\n", hr);
+        goto Cleanup;
+    }
+    
+    // Enable File History
+    wprintf(L"Enabling File History\n");
+    hr = configMgr->SetBackupStatus(FH_STATUS_ENABLED);
+    if (FAILED(hr))
+    {
+        wprintf(L"Error: SetBackupStatus failed (0x%X)\n", hr);
+        goto Cleanup;
+    }
+
+    // Save the configuration to disk
+    wprintf(L"Saving configuration\n");
+    hr = configMgr->SaveConfiguration();
+    if (FAILED(hr))
+    {
+        wprintf(L"Error: SaveConfiguration failed (0x%X)\n", hr);
+        goto Cleanup;
+    }
+
+    // Tell the File History service to schedule backups
+    wprintf(L"Scheduling regular backups\n");
+    hr = ScheduleBackups();
+    if (FAILED(hr))
+    {
+        wprintf(L"Error: ScheduleBackups failed (0x%X)\n", hr);
+        goto Cleanup;
+    }
+
+    // Recommend the target to other Homegroup members
+    wprintf(L"Recommending target to Homegroup\n");
+    HRESULT hrRecommend = configMgr->ChangeDefaultTargetRecommendation(TRUE);
+    if (FAILED(hrRecommend))
+    {
+        wprintf(L"Warning: Failed to recommend target to Homegroup (0x%X)\n", hrRecommend);
+    }
+
+    wprintf(L"Success! File History is now enabled\n");
+
+Cleanup:
+    return hr;
+}
+
+int __cdecl wmain(
+    _In_ int Argc,
+    _In_reads_(Argc) PWSTR Argv[]
+    )
+/*++
+
+Routine Description:
+
+    This is the main entry point of the console application.
+
+Arguments:
+
+    Argc - the number of command line arguments
+    Argv - command line arguments
+
+Return Value:
+
+    exit code
+
+--*/
+{
+    HRESULT hr = S_OK;
+    BOOL comInitialized = FALSE;
+
+    wprintf(L"\nFile History Sample Setup Tool\n");
+    wprintf(L"Copyright (C) Microsoft Corporation. All rights reserved.\n\n");
+
+    // If there are fewer than 2 command-line arguments, print the correct
+    // usage and exit
+    if (Argc < 2)
+    {
+        wprintf(L"Usage: fhsetup <path>\n\n");
+        wprintf(L"Examples:\n");
+        wprintf(L"    fhsetup D:\\\n");
+        wprintf(L"    fhsetup \\\\server\\share\\\n\n");
+        goto Cleanup;
+    }
+
+    // COM is needed to use the Config Manager
+    wprintf(L"Initializing COM...\n");
+    hr = CoInitialize(NULL);
+    if (FAILED(hr))
+    {
+        wprintf(L"Error: CoInitialize failed (0x%X)\n", hr);
+        goto Cleanup;
+    }
+    comInitialized = TRUE;
+
+    hr = ConfigureFileHistory(Argv[1]);
+    if (FAILED(hr))
+    {
+        wprintf(L"File History configuration failed (0x%X)\n", hr);
+        goto Cleanup;
+    }
+
+Cleanup:
+    // If COM was initialized, make sure it is uninitialized
+    if (comInitialized)
+    {
+        CoUninitialize();
+        comInitialized = FALSE;
+    }
+
+    return 0;
+}