/*++

AppDrv_Register -- modified from Microsoft WDK 6001.18002 sample (src\setup\DIFxAPI\AppDrv\AppDrvDepend.c)


Description:

This sample demonstrates how to use the application-to-driver dependency 
feature of the Driver Install Frameworks (DIFx) API. The DriverPackageInstall
and the DriverPackageUninstall APIs allow the calling application to register
and remove dependencies on a driver package. In order to register or remove
a dependency on a driver package, the calling application needs to populate
the fields of an INSTALLERINFO structure with information about itself. It 
then needs to pass in this structure as a parameter to the DIFx APIs, as 
shown in this sample.

--*/
#include <windows.h>
#include <stdio.h>
#include <difxapi.h>

int __cdecl	
wmain( int argc, WCHAR* argv[] )
{
    WCHAR *InfPath = NULL;
    INSTALLERINFO AppInfo;
    BOOL Result;
    BOOL NeedReboot;
//
	WCHAR FullInfPath[MAX_PATH];
   	WCHAR FullAplPath[MAX_PATH];
    LPTSTR lpFilePart;
    DWORD Length;
    static LPCWSTR UninstallKey = TEXT("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\");
    WCHAR RegPath[MAX_PATH];
    WCHAR UninstallPath[MAX_PATH * 2];
   	HKEY  hkey;
//
    if (argc < 7) {
        wprintf( TEXT("USAGE: AppDrv.exe <option> <full path to the inf file> <Application ID> <Product display name> <Product name> <Manufacturer name>\n") );
        wprintf( TEXT("<option> can be one of the following -\n") );
        wprintf( TEXT("/i - Install the driver\n") );
        wprintf( TEXT("/u - Uninstall the driver\n") );
        return 0;
    }

    //
    // Initialization
    //
    NeedReboot = FALSE;

    //
    // Get the path to the INF file
    //
	InfPath = argv[2];

    //
    // Fill in the fields of the INSTALLERINFO structure with information about
    // the application performing the install.
    //

    //
    // The application ID should be some random string unique to this 
    // application.
    //
    AppInfo.pApplicationId = argv[3];
    AppInfo.pDisplayName = argv[4];
    AppInfo.pProductName = argv[5];
    AppInfo.pMfgName = argv[6];
//
    Length = GetFullPathName(InfPath, MAX_PATH, FullInfPath, NULL);
    if(Length >= MAX_PATH) {
        wprintf(TEXT("ERROR: GetFullPathName, length = %wlu\n"), Length);
        return 0;
    }
    wprintf(TEXT("INFO: FullInfPath = %ws, length = %wlu\n"), FullInfPath, Length);

    Length = SearchPath(NULL, argv[0], TEXT(".EXE"), MAX_PATH, FullAplPath, &lpFilePart);
    if(Length >= MAX_PATH) {
        wprintf(TEXT("ERROR: GetFullPathName, length = %wlu\n"), Length);
        return 0;
    }
    wprintf(TEXT("INFO: FullAplPath = %ws (%ws), length = %wlu\n"), FullAplPath, argv[0], Length);

    lstrcpy(RegPath, UninstallKey);
	lstrcat(RegPath, AppInfo.pApplicationId);
//
    if (!lstrcmpi(argv[1], TEXT("/i"))) {
        //
        // Install the driver package, passing in the INSTALLERINFO structure with
        // information about the application performing the install. By passing in
        // this structure, the "MySoftware" application is expressing a dependency
        // on the driver being installed.
        //
        wprintf( TEXT("INFO: Attempting to install the driver package %ws.\n"), InfPath );
    
        Result = DriverPackageInstall(InfPath, 
                                      0,        // flags
                                      &AppInfo, // dependency information
                                      &NeedReboot);
    
        if (ERROR_SUCCESS == Result) {
            wprintf( TEXT("SUCCESS: installed driver package %ws.\n"), InfPath );
        } else if (ERROR_NO_MORE_ITEMS == Result) {
            wprintf( TEXT("INFO: the driver package %ws was successfully staged on the machine.\n"), InfPath );
            wprintf( TEXT("INFO: All devices found already have a better driver than what is contained in the specified inf file.\n"));
        } else if (ERROR_NO_SUCH_DEVINST == Result) {
            wprintf( TEXT("INFO: the driver package %ws was successfully staged on the machine.\n"), InfPath );
            wprintf( TEXT("INFO: There aren't any live devnodes with the device ID contained in the INF.\n"));
        } else if (ERROR_NO_DEVICE_ID == Result) {
            wprintf( TEXT("INFO: the driver package %ws was successfully staged on the machine.\n"), InfPath );
            wprintf( TEXT("INFO: No device IDs found in INF '%ws' for current platform.\n"), InfPath);
        } else {
            wprintf( TEXT("ERROR: An error occurred while installing driver package %ws. Error code - 0x%X.\n"), InfPath, Result);
            goto clean0;
        }
    
        if (NeedReboot){
            wprintf( TEXT("INFO: Machine will have to be rebooted to complete install.\n"));
        }
        //
        Result = ERROR_SUCCESS;
        if(RegCreateKeyEx(HKEY_LOCAL_MACHINE, RegPath, 0, 0, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL,
	    	&hkey, &Result) != ERROR_SUCCESS) {
                wprintf(TEXT("ERROR: RegCreateKeyEx Create Error.\n"));
                goto clean0;
	    }
        if (Result == REG_OPENED_EXISTING_KEY) {
                wprintf(TEXT("ERROR: Driver Already Installed.\n"));
	            RegCloseKey(hkey);
                goto clean0;
        }
	    Result = RegSetValueEx(hkey, TEXT("DisplayName"), 0, REG_SZ, (BYTE*)AppInfo.pDisplayName,
            2 *(lstrlen(AppInfo.pDisplayName) + 1));
        wprintf(TEXT("INFO: DisplayName = %ws., r = 0x%X\n"), AppInfo.pDisplayName, Result);
        
        Result = RegSetValueEx(hkey, TEXT("Publisher"), 0, REG_SZ, (BYTE*)AppInfo.pMfgName,
            2 *(lstrlen(AppInfo.pMfgName) + 1));
        wprintf(TEXT("INFO: Publisher = %ws., r = 0x%X\n"), AppInfo.pMfgName, Result);

        _snwprintf(UninstallPath, MAX_PATH * 2, TEXT("\"%ws\" /u \"%ws\" \"%ws\" \"%ws\" \"%ws\" \"%ws\""),
            FullAplPath, FullInfPath, argv[3], argv[4], argv[5], argv[6]);

        Result = RegSetValueEx(hkey, TEXT("UninstallString"), 0, REG_SZ, (BYTE*)UninstallPath,
            2 *(lstrlen(UninstallPath) + 1));
        wprintf(TEXT("INFO: UninstallPath = %ws., r = 0x%X\n"), UninstallPath, Result);

	    Result = RegCloseKey(hkey);
        if (Result != ERROR_SUCCESS) {
            wprintf(TEXT("ERROR: RegCloseKey.\n"));
            goto clean0;
        }
        //

    } else if (!lstrcmpi(argv[1], TEXT("/u"))) {
        //
        // Uninstall the driver package, passing in the INSTALLERINFO structure
        // with information about the application performing the uninstall. By
        // passing in this structure, the "MySoftware" application is removing
        // its dependency on the driver. The driver will actually be 
        // uninstalled only if there are no other applications dependent on it.
        // If there are other applications depending on the driver, then the call
        // to DriverPackageUninstall will simply remove the dependency of the 
        // "MySoftware" application on the driver, and return 
        // ERROR_DEPENDENT_APPLICATIONS_EXIST.
        //
        wprintf( TEXT("INFO: Attempting to uninstall the driver package %ws.\n"), InfPath );

        Result = DriverPackageUninstall(InfPath,
                                        0,        // flags
                                        &AppInfo, // dependency information
                                        &NeedReboot);
        if (ERROR_SUCCESS != Result) {
            if (ERROR_DEPENDENT_APPLICATIONS_EXIST == Result) {
                wprintf( TEXT("INFO: The driver package %ws was not uninstalled because there were some applications still depending on it.\n"), InfPath);
            } else {
                wprintf( TEXT("ERROR: An error occurred while uninstalling driver package %ws\n."), InfPath);
                goto clean0;
            }
        } else {
            wprintf( TEXT("SUCCESS: Uninstalled driver package %ws.\n"), InfPath );
        }
    } else {
        wprintf( TEXT("ERROR: An invalid command line option %ws was passed in!\n"), argv[1] );
    }

    RegDeleteKey(HKEY_LOCAL_MACHINE, RegPath);

    clean0:
        return 0;
}