+++ /dev/null
-/* Copyright (c) Xen Project.
- * Copyright (c) Cloud Software Group, Inc.
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms,
- * with or without modification, are permitted provided
- * that the following conditions are met:
- *
- * * Redistributions of source code must retain the above
- * copyright notice, this list of conditions and the
- * following disclaimer.
- * * Redistributions in binary form must reproduce the above
- * copyright notice, this list of conditions and the
- * following disclaimer in the documentation and/or other
- * materials provided with the distribution.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
- * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
- * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
- * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
- * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
- * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
- * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
- * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
- * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
- * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
- */
-
-#include <windows.h>
-#include <setupapi.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <strsafe.h>
-#include <malloc.h>
-#include <assert.h>
-#include <iphlpapi.h>
-
-#include <version.h>
-#include <revision.h>
-
-#define stringify_literal(_text) #_text
-#define stringify(_text) stringify_literal(_text)
-#define __MODULE__ stringify(PROJECT)
-
-__user_code;
-
-#define MAXIMUM_BUFFER_SIZE 1024
-
-#define SERVICES_KEY "SYSTEM\\CurrentControlSet\\Services"
-
-#define SERVICE_KEY(_Driver) \
- SERVICES_KEY ## "\\" ## #_Driver
-
-#define UNPLUG_KEY \
- SERVICE_KEY(XEN) ## "\\Unplug"
-
-#define SETTINGS_KEY \
- SERVICE_KEY(XENVIF) ## "\\Settings"
-
-#define CONTROL_KEY "SYSTEM\\CurrentControlSet\\Control"
-
-#define CLASS_KEY \
- CONTROL_KEY ## "\\Class"
-
-#define ENUM_KEY "SYSTEM\\CurrentControlSet\\Enum"
-
-#define NETWORK_CLASS_GUID \
- "{4d36e972-e325-11ce-bfc1-08002be10318}"
-
-static VOID
-#pragma prefast(suppress:6262) // Function uses '1036' bytes of stack: exceeds /analyze:stacksize'1024'
-__Log(
- IN const CHAR *Format,
- IN ...
- )
-{
- TCHAR Buffer[MAXIMUM_BUFFER_SIZE];
- va_list Arguments;
- size_t Length;
- SP_LOG_TOKEN LogToken;
- DWORD Category;
- DWORD Flags;
- HRESULT Result;
-
- va_start(Arguments, Format);
- Result = StringCchVPrintf(Buffer, MAXIMUM_BUFFER_SIZE, Format, Arguments);
- va_end(Arguments);
-
- if (Result != S_OK && Result != STRSAFE_E_INSUFFICIENT_BUFFER)
- return;
-
- Result = StringCchLength(Buffer, MAXIMUM_BUFFER_SIZE, &Length);
- if (Result != S_OK)
- return;
-
- LogToken = SetupGetThreadLogToken();
- Category = TXTLOG_VENDOR;
- Flags = TXTLOG_WARNING;
-
- SetupWriteTextLog(LogToken, Category, Flags, Buffer);
- Length = __min(MAXIMUM_BUFFER_SIZE - 1, Length + 2);
-
- __analysis_assume(Length < MAXIMUM_BUFFER_SIZE);
- __analysis_assume(Length >= 2);
- Buffer[Length] = '\0';
- Buffer[Length - 1] = '\n';
- Buffer[Length - 2] = '\r';
-
- OutputDebugString(Buffer);
-}
-
-#define Log(_Format, ...) \
- __Log(__MODULE__ "|" __FUNCTION__ ": " _Format, __VA_ARGS__)
-
-static PTCHAR
-GetErrorMessage(
- IN DWORD Error
- )
-{
- PTCHAR Message;
- ULONG Index;
-
- if (!FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
- FORMAT_MESSAGE_FROM_SYSTEM |
- FORMAT_MESSAGE_IGNORE_INSERTS,
- NULL,
- Error,
- MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
- (LPTSTR)&Message,
- 0,
- NULL))
- return NULL;
-
- for (Index = 0; Message[Index] != '\0'; Index++) {
- if (Message[Index] == '\r' || Message[Index] == '\n') {
- Message[Index] = '\0';
- break;
- }
- }
-
- return Message;
-}
-
-static FORCEINLINE const CHAR *
-__FunctionName(
- IN DI_FUNCTION Function
- )
-{
-#define _NAME(_Function) \
- case DIF_ ## _Function: \
- return #_Function;
-
- switch (Function) {
- _NAME(INSTALLDEVICE);
- _NAME(REMOVE);
- _NAME(SELECTDEVICE);
- _NAME(ASSIGNRESOURCES);
- _NAME(PROPERTIES);
- _NAME(FIRSTTIMESETUP);
- _NAME(FOUNDDEVICE);
- _NAME(SELECTCLASSDRIVERS);
- _NAME(VALIDATECLASSDRIVERS);
- _NAME(INSTALLCLASSDRIVERS);
- _NAME(CALCDISKSPACE);
- _NAME(DESTROYPRIVATEDATA);
- _NAME(VALIDATEDRIVER);
- _NAME(MOVEDEVICE);
- _NAME(DETECT);
- _NAME(INSTALLWIZARD);
- _NAME(DESTROYWIZARDDATA);
- _NAME(PROPERTYCHANGE);
- _NAME(ENABLECLASS);
- _NAME(DETECTVERIFY);
- _NAME(INSTALLDEVICEFILES);
- _NAME(ALLOW_INSTALL);
- _NAME(SELECTBESTCOMPATDRV);
- _NAME(REGISTERDEVICE);
- _NAME(NEWDEVICEWIZARD_PRESELECT);
- _NAME(NEWDEVICEWIZARD_SELECT);
- _NAME(NEWDEVICEWIZARD_PREANALYZE);
- _NAME(NEWDEVICEWIZARD_POSTANALYZE);
- _NAME(NEWDEVICEWIZARD_FINISHINSTALL);
- _NAME(INSTALLINTERFACES);
- _NAME(DETECTCANCEL);
- _NAME(REGISTER_COINSTALLERS);
- _NAME(ADDPROPERTYPAGE_ADVANCED);
- _NAME(ADDPROPERTYPAGE_BASIC);
- _NAME(TROUBLESHOOTER);
- _NAME(POWERMESSAGEWAKE);
- default:
- break;
- }
-
- return "UNKNOWN";
-
-#undef _NAME
-}
-
-static BOOLEAN
-ClearUnplugRequest(
- IN PTCHAR ClassName
- )
-{
- HKEY UnplugKey;
- HRESULT Error;
-
- Error = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
- UNPLUG_KEY,
- 0,
- KEY_ALL_ACCESS,
- &UnplugKey);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail1;
- }
-
- Error = RegDeleteValue(UnplugKey, ClassName);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail2;
- }
-
- RegCloseKey(UnplugKey);
-
- return TRUE;
-
-fail2:
- Log("fail2");
-
- RegCloseKey(UnplugKey);
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-AllowUpdate(
- IN PTCHAR DriverName,
- OUT PBOOLEAN Allow
- )
-{
- TCHAR ServiceKeyName[MAX_PATH];
- HKEY ServiceKey;
- HRESULT Result;
- HRESULT Error;
- DWORD ValueLength;
- DWORD Value;
- DWORD Type;
-
- Log("====> (%s)", DriverName);
-
- Result = StringCbPrintf(ServiceKeyName,
- MAX_PATH,
- SERVICES_KEY "\\%s",
- DriverName);
- assert(SUCCEEDED(Result));
-
- Error = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
- ServiceKeyName,
- 0,
- KEY_READ,
- &ServiceKey);
- if (Error != ERROR_SUCCESS) {
- if (Error == ERROR_FILE_NOT_FOUND) {
- Value = 1;
- goto done;
- }
-
- SetLastError(Error);
- goto fail1;
- }
-
- ValueLength = sizeof (Value);
-
- Error = RegQueryValueEx(ServiceKey,
- "AllowUpdate",
- NULL,
- &Type,
- (LPBYTE)&Value,
- &ValueLength);
- if (Error != ERROR_SUCCESS) {
- if (Error == ERROR_FILE_NOT_FOUND) {
- Type = REG_DWORD;
- Value = 1;
- } else {
- SetLastError(Error);
- goto fail2;
- }
- }
-
- if (Type != REG_DWORD) {
- SetLastError(ERROR_BAD_FORMAT);
- goto fail3;
- }
-
- RegCloseKey(ServiceKey);
-
-done:
- if (Value == 0) {
- Log("DISALLOWED");
- *Allow = FALSE;
- }
-
- Log("<====");
-
- return TRUE;
-
-fail3:
- Log("fail3");
-
-fail2:
- Log("fail2");
-
- RegCloseKey(ServiceKey);
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-AllowInstall(
- OUT PBOOLEAN Allow
- )
-{
- BOOLEAN Success;
- HRESULT Error;
-
- Log("====>");
-
- *Allow = TRUE;
-
- Success = AllowUpdate("XENVIF", Allow);
- if (!Success)
- goto fail1;
-
- Log("<====");
-
- return TRUE;
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-OpenEnumKey(
- OUT PHKEY EnumKey
- )
-{
- HRESULT Error;
-
- Error = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
- ENUM_KEY,
- 0,
- KEY_READ,
- EnumKey);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail1;
- }
-
- return TRUE;
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-OpenBusKey(
- IN PTCHAR BusKeyName,
- OUT PHKEY BusKey
- )
-{
- BOOLEAN Success;
- HKEY EnumKey;
- HRESULT Error;
-
- Success = OpenEnumKey(&EnumKey);
- if (!Success)
- goto fail1;
-
- Error = RegOpenKeyEx(EnumKey,
- BusKeyName,
- 0,
- KEY_READ,
- BusKey);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail2;
- }
-
- RegCloseKey(EnumKey);
-
- return TRUE;
-
-fail2:
- Log("fail2");
-
- RegCloseKey(EnumKey);
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-OpenDeviceKey(
- IN PTCHAR BusKeyName,
- IN PTCHAR DeviceKeyName,
- OUT PHKEY DeviceKey
- )
-{
- BOOLEAN Success;
- HKEY BusKey;
- HRESULT Error;
-
- Success = OpenBusKey(BusKeyName, &BusKey);
- if (!Success)
- goto fail1;
-
- Error = RegOpenKeyEx(BusKey,
- DeviceKeyName,
- 0,
- KEY_READ,
- DeviceKey);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail2;
- }
-
- RegCloseKey(BusKey);
-
- return TRUE;
-
-fail2:
- Log("fail2");
-
- RegCloseKey(BusKey);
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-GetDriverKeyName(
- IN HKEY DeviceKey,
- OUT PTCHAR *Name,
- OUT DWORD *ConfigFlags
- )
-{
- HRESULT Error;
- DWORD SubKeys;
- DWORD MaxSubKeyLength;
- DWORD SubKeyLength;
- PTCHAR SubKeyName;
- DWORD Index;
- HKEY SubKey;
- PTCHAR DriverKeyName;
- DWORD Flags;
-
- Error = RegQueryInfoKey(DeviceKey,
- NULL,
- NULL,
- NULL,
- &SubKeys,
- &MaxSubKeyLength,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail1;
- }
-
- SubKeyLength = MaxSubKeyLength + sizeof (TCHAR);
-
- SubKeyName = malloc(SubKeyLength);
- if (SubKeyName == NULL)
- goto fail2;
-
- SubKey = NULL;
- DriverKeyName = NULL;
- Flags = 0;
-
- for (Index = 0; Index < SubKeys; Index++) {
- DWORD MaxValueLength;
- DWORD ConfigFlagsLength;
- DWORD DriverKeyNameLength;
- DWORD Type;
-
- SubKeyLength = MaxSubKeyLength + sizeof (TCHAR);
- memset(SubKeyName, 0, SubKeyLength);
-
- Error = RegEnumKeyEx(DeviceKey,
- Index,
- (LPTSTR)SubKeyName,
- &SubKeyLength,
- NULL,
- NULL,
- NULL,
- NULL);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail3;
- }
-
- Error = RegOpenKeyEx(DeviceKey,
- SubKeyName,
- 0,
- KEY_READ,
- &SubKey);
- if (Error != ERROR_SUCCESS) {
- SubKey = NULL;
- continue;
- }
-
- Error = RegQueryInfoKey(SubKey,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- &MaxValueLength,
- NULL,
- NULL);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail4;
- }
-
- ConfigFlagsLength = (DWORD)sizeof(DWORD);
-
- Error = RegQueryValueEx(SubKey,
- "ConfigFlags",
- NULL,
- &Type,
- (LPBYTE)&Flags,
- &ConfigFlagsLength);
- if (Error != ERROR_SUCCESS ||
- Type != REG_DWORD)
- Flags = 0;
-
- DriverKeyNameLength = MaxValueLength + sizeof (TCHAR);
-
- DriverKeyName = calloc(1, DriverKeyNameLength);
- if (DriverKeyName == NULL)
- goto fail5;
-
- Error = RegQueryValueEx(SubKey,
- "Driver",
- NULL,
- &Type,
- (LPBYTE)DriverKeyName,
- &DriverKeyNameLength);
- if (Error == ERROR_SUCCESS &&
- Type == REG_SZ)
- break;
-
- free(DriverKeyName);
- DriverKeyName = NULL;
- Flags = 0;
-
- RegCloseKey(SubKey);
- SubKey = NULL;
- }
-
- Log("%s", (DriverKeyName != NULL) ? DriverKeyName : "none found");
-
- if (SubKey != NULL)
- RegCloseKey(SubKey);
-
- free(SubKeyName);
-
- *Name = DriverKeyName;
- *ConfigFlags = Flags;
-
- return TRUE;
-
-fail5:
- Log("fail5");
-
-fail4:
- Log("fail4");
-
- if (SubKey != NULL)
- RegCloseKey(SubKey);
-
-fail3:
- Log("fail3");
-
- free(SubKeyName);
-
-fail2:
- Log("fail2");
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-OpenClassKey(
- OUT PHKEY ClassKey
- )
-{
- HRESULT Error;
-
- Error = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
- CLASS_KEY,
- 0,
- KEY_READ,
- ClassKey);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail1;
- }
-
- return TRUE;
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-OpenDriverKey(
- IN PTCHAR DriverKeyName,
- OUT PHKEY DriverKey
- )
-{
- BOOLEAN Success;
- HKEY ClassKey;
- HRESULT Error;
-
- Success = OpenClassKey(&ClassKey);
- if (!Success)
- goto fail1;
-
- Error = RegOpenKeyEx(ClassKey,
- DriverKeyName,
- 0,
- KEY_READ,
- DriverKey);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail2;
- }
-
- RegCloseKey(ClassKey);
-
- return TRUE;
-
-fail2:
- Log("fail2");
-
- RegCloseKey(ClassKey);
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-#define DEFINE_REVISION(_N, _C, _V, _ST, _SU) \
- (_N)
-
-static DWORD DeviceRevision[] = {
- DEFINE_REVISION_TABLE
-};
-
-#undef DEFINE_REVISION
-
-static BOOLEAN
-SupportDeviceID(
- IN PTCHAR DeviceID
- )
-{
- unsigned int Revision;
- int Count;
- DWORD Index;
- HRESULT Error;
-
- DeviceID = strrchr(DeviceID, '&');
- assert(DeviceID != NULL);
- DeviceID++;
-
- Count = sscanf_s(DeviceID,
- "REV_%8x",
- &Revision);
- if (Count != 1) {
- SetLastError(ERROR_BAD_FORMAT);
- goto fail1;
- }
-
- for (Index = 0; Index < ARRAYSIZE(DeviceRevision); Index++) {
- if (Revision == DeviceRevision[Index])
- goto found;
- }
-
- SetLastError(ERROR_FILE_NOT_FOUND);
- goto fail2;
-
-found:
- Log("%x", Revision);
-
- return TRUE;
-
-fail2:
- Log("fail2");
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-GetMatchingDeviceID(
- IN HKEY DriverKey,
- OUT PTCHAR *MatchingDeviceID
- )
-{
- HRESULT Error;
- DWORD MaxValueLength;
- DWORD MatchingDeviceIDLength;
- DWORD Type;
- DWORD Index;
-
- Error = RegQueryInfoKey(DriverKey,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- &MaxValueLength,
- NULL,
- NULL);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail1;
- }
-
- MatchingDeviceIDLength = MaxValueLength + sizeof (TCHAR);
-
- *MatchingDeviceID = calloc(1, MatchingDeviceIDLength);
- if (*MatchingDeviceID == NULL)
- goto fail2;
-
- Error = RegQueryValueEx(DriverKey,
- "MatchingDeviceId",
- NULL,
- &Type,
- (LPBYTE)*MatchingDeviceID,
- &MatchingDeviceIDLength);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail3;
- }
-
- if (Type != REG_SZ) {
- SetLastError(ERROR_BAD_FORMAT);
- goto fail4;
- }
-
- for (Index = 0; Index < strlen(*MatchingDeviceID); Index++)
- (*MatchingDeviceID)[Index] = (CHAR)toupper((*MatchingDeviceID)[Index]);
-
- Log("%s", *MatchingDeviceID);
-
- return TRUE;
-
-fail4:
- Log("fail4");
-
-fail3:
- Log("fail3");
-
- free(*MatchingDeviceID);
-
-fail2:
- Log("fail2");
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-SupportChildDrivers(
- VOID
- )
-{
- BOOLEAN Success;
- HKEY XenbusKey;
- HRESULT Error;
- DWORD SubKeys;
- DWORD MaxSubKeyLength;
- DWORD SubKeyLength;
- PTCHAR SubKeyName;
- HKEY DeviceKey;
- PTCHAR DriverKeyName;
- DWORD ConfigFlags;
- HKEY DriverKey;
- PTCHAR MatchingDeviceID;
- DWORD Index;
-
- Log("====>");
-
- Success = OpenBusKey("XENVIF", &XenbusKey);
- if (!Success) {
- // If there is no key then this must be a fresh installation
- if (GetLastError() == ERROR_FILE_NOT_FOUND)
- goto done;
-
- goto fail1;
- }
-
- Error = RegQueryInfoKey(XenbusKey,
- NULL,
- NULL,
- NULL,
- &SubKeys,
- &MaxSubKeyLength,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail2;
- }
-
- SubKeyLength = MaxSubKeyLength + sizeof (TCHAR);
-
- SubKeyName = malloc(SubKeyLength);
- if (SubKeyName == NULL)
- goto fail3;
-
- for (Index = 0; Index < SubKeys; Index++) {
- SubKeyLength = MaxSubKeyLength + sizeof (TCHAR);
- memset(SubKeyName, 0, SubKeyLength);
-
- Error = RegEnumKeyEx(XenbusKey,
- Index,
- (LPTSTR)SubKeyName,
- &SubKeyLength,
- NULL,
- NULL,
- NULL,
- NULL);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail4;
- }
-
- Success = OpenDeviceKey("XENVIF", SubKeyName, &DeviceKey);
- if (!Success)
- goto fail5;
-
- Success = GetDriverKeyName(DeviceKey, &DriverKeyName, &ConfigFlags);
- if (!Success)
- goto fail6;
-
- if (DriverKeyName == NULL)
- goto loop1;
-
- if (ConfigFlags & 0x20)
- goto loop2;
-
- Success = OpenDriverKey(DriverKeyName, &DriverKey);
- if (!Success)
- goto loop3;
-
- Success = GetMatchingDeviceID(DriverKey, &MatchingDeviceID);
- if (!Success)
- goto loop4;
-
- Success = SupportDeviceID(MatchingDeviceID);
- if (!Success)
- goto fail7;
-
- free(MatchingDeviceID);
-
- loop4:
- RegCloseKey(DriverKey);
-
- loop3:
-
- loop2:
- free(DriverKeyName);
-
- loop1:
- RegCloseKey(DeviceKey);
- }
-
- free(SubKeyName);
-
- RegCloseKey(XenbusKey);
-
-done:
- Log("<====");
-
- return TRUE;
-
-fail7:
- Log("fail7");
-
- free(MatchingDeviceID);
-
- RegCloseKey(DriverKey);
-
- free(DriverKeyName);
-
-fail6:
- Log("fail6");
-
- RegCloseKey(DeviceKey);
-
-fail5:
- Log("fail5");
-
-fail4:
- Log("fail4");
-
- free(SubKeyName);
-
-fail3:
- Log("fail3");
-
-fail2:
- Log("fail2");
-
- RegCloseKey(XenbusKey);
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-CreateSettingsKey(
- IN PTCHAR KeyName,
- OUT PHKEY Key
- )
-{
- HKEY SettingsKey;
- HRESULT Error;
-
- Error = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
- SETTINGS_KEY,
- 0,
- KEY_ALL_ACCESS,
- &SettingsKey);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail1;
- }
-
- Error = RegCreateKeyEx(SettingsKey,
- KeyName,
- 0,
- NULL,
- REG_OPTION_NON_VOLATILE,
- KEY_ALL_ACCESS,
- NULL,
- Key,
- NULL);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail2;
- }
-
- RegCloseKey(SettingsKey);
-
- return TRUE;
-
-fail2:
- Log("fail2");
-
- RegCloseKey(SettingsKey);
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-CopyNetworkID(
- IN PTCHAR VifIndex,
- IN HKEY DriverKey
- )
-{
- BOOLEAN Success;
- HRESULT Error;
- TCHAR NetCfgInstanceId[MAX_PATH];
- DWORD NetCfgInstanceIdLength;
- DWORD Value;
- DWORD ValueLength;
- NET_LUID NetLuid;
- HKEY SettingsKey;
-
- NetCfgInstanceIdLength = sizeof(NetCfgInstanceId);
- Error = RegQueryValueEx(DriverKey,
- "NetCfgInstanceId",
- NULL,
- NULL,
- (LPBYTE)NetCfgInstanceId,
- &NetCfgInstanceIdLength);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail1;
- }
-
- NetLuid.Value = 0ull;
-
- ValueLength = sizeof(DWORD);
- Error = RegQueryValueEx(DriverKey,
- "NetLuidIndex",
- NULL,
- NULL,
- (LPBYTE)&Value,
- &ValueLength);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail2;
- }
-
- NetLuid.Info.NetLuidIndex = Value;
-
- ValueLength = sizeof(DWORD);
- Error = RegQueryValueEx(DriverKey,
- "*IfType",
- NULL,
- NULL,
- (LPBYTE)&Value,
- &ValueLength);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail3;
- }
-
- NetLuid.Info.IfType = Value;
-
- Log("VIF/%s was %s", VifIndex, NetCfgInstanceId);
-
- Success = CreateSettingsKey(VifIndex, &SettingsKey);
- if (!Success)
- goto fail4;
-
- Error = RegSetValueEx(SettingsKey,
- "NetCfgInstanceId",
- 0,
- REG_SZ,
- (const BYTE*)NetCfgInstanceId,
- NetCfgInstanceIdLength);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail5;
- }
-
- Error = RegSetValueEx(SettingsKey,
- "NetLuid",
- 0,
- REG_BINARY,
- (const BYTE*)&NetLuid,
- sizeof(NET_LUID));
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail6;
- }
-
- Value = 1;
- Error = RegSetValueEx(SettingsKey,
- "HasSettings",
- 0,
- REG_DWORD,
- (const BYTE*)&Value,
- sizeof(DWORD));
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail7;
- }
-
- RegCloseKey(SettingsKey);
-
- return TRUE;
-
-fail7:
- Log("fail7");
-
-fail6:
- Log("fail6");
-
-fail5:
- Log("fail5");
-
- RegCloseKey(SettingsKey);
-
-fail4:
- Log("fail4");
-
-fail3:
- Log("fail3");
-
-fail2:
- Log("fail2");
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static BOOLEAN
-CopyNetworkIDs(
- VOID
- )
-{
- BOOLEAN Success;
- HRESULT Error;
- HKEY ClassKey;
- DWORD SubKeys;
- DWORD MaxSubKeyLength;
- DWORD Index;
- DWORD SubKeyLength;
- PTCHAR SubKeyName;
-
- Log("====>");
-
- Success = OpenDriverKey(NETWORK_CLASS_GUID, &ClassKey);
- if (!Success)
- goto fail1;
-
- Error = RegQueryInfoKey(ClassKey,
- NULL,
- NULL,
- NULL,
- &SubKeys,
- &MaxSubKeyLength,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail2;
- }
-
- SubKeyLength = MaxSubKeyLength + sizeof (TCHAR);
-
- SubKeyName = malloc(SubKeyLength);
- if (SubKeyName == NULL)
- goto fail3;
-
- for (Index = 0; Index < SubKeys; ++Index) {
- HKEY DriverKey;
- TCHAR DeviceInstanceID[MAX_PATH];
- PTCHAR VifIndex;
- DWORD Length;
-
- SubKeyLength = MaxSubKeyLength + sizeof (TCHAR);
- memset(SubKeyName, 0, SubKeyLength);
-
- Error = RegEnumKeyEx(ClassKey,
- Index,
- (LPTSTR)SubKeyName,
- &SubKeyLength,
- NULL,
- NULL,
- NULL,
- NULL);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto fail4;
- }
-
- Error = RegOpenKeyEx(ClassKey,
- SubKeyName,
- 0,
- KEY_READ,
- &DriverKey);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- continue;
- }
-
- Length = sizeof(DeviceInstanceID);
- Error = RegQueryValueEx(DriverKey,
- "DeviceInstanceID",
- NULL,
- NULL,
- (LPBYTE)DeviceInstanceID,
- &Length);
- if (Error != ERROR_SUCCESS) {
- SetLastError(Error);
- goto loop1;
- }
-
- if (_strnicmp("XENVIF", DeviceInstanceID, strlen("XENVIF")) != 0)
- goto loop2;
-
- VifIndex = strrchr(DeviceInstanceID, '\\');
- if (VifIndex == NULL)
- goto loop3;
-
- ++VifIndex;
-
- CopyNetworkID(VifIndex, DriverKey);
-
- loop3:
- loop2:
- loop1:
- RegCloseKey(DriverKey);
- }
-
- free(SubKeyName);
- RegCloseKey(ClassKey);
-
- Log("<====");
-
- return TRUE;
-
-fail4:
- Log("fail4");
-
- free(SubKeyName);
-
-fail3:
- Log("fail3");
-
-fail2:
- Log("fail2");
-
- RegCloseKey(ClassKey);
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return FALSE;
-}
-
-static HRESULT
-DifInstallPreProcess(
- IN HDEVINFO DeviceInfoSet,
- IN PSP_DEVINFO_DATA DeviceInfoData,
- IN PCOINSTALLER_CONTEXT_DATA Context
- )
-{
- HRESULT Error;
- BOOLEAN Success;
- BOOLEAN Allow;
-
- UNREFERENCED_PARAMETER(DeviceInfoSet);
- UNREFERENCED_PARAMETER(DeviceInfoData);
- UNREFERENCED_PARAMETER(Context);
-
- Log("====>");
-
- Success = AllowInstall(&Allow);
- if (!Success)
- goto fail1;
-
- if (!Allow) {
- SetLastError(ERROR_ACCESS_DENIED);
- goto fail2;
- }
-
- Success = SupportChildDrivers();
- if (!Success)
- goto fail3;
-
- Success = CopyNetworkIDs();
- if (!Success)
- goto fail4;
-
- Log("<====");
-
- return NO_ERROR;
-
-fail4:
- Log("fail4");
-
-fail3:
- Log("fail3");
-
-fail2:
- Log("fail2");
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return Error;
-}
-
-static HRESULT
-DifInstallPostProcess(
- IN HDEVINFO DeviceInfoSet,
- IN PSP_DEVINFO_DATA DeviceInfoData,
- IN PCOINSTALLER_CONTEXT_DATA Context
- )
-{
- UNREFERENCED_PARAMETER(DeviceInfoSet);
- UNREFERENCED_PARAMETER(DeviceInfoData);
- UNREFERENCED_PARAMETER(Context);
-
- Log("<===>");
-
- return NO_ERROR;
-}
-
-static DECLSPEC_NOINLINE HRESULT
-DifInstall(
- IN HDEVINFO DeviceInfoSet,
- IN PSP_DEVINFO_DATA DeviceInfoData,
- IN PCOINSTALLER_CONTEXT_DATA Context
- )
-{
- SP_DEVINSTALL_PARAMS DeviceInstallParams;
- HRESULT Error;
-
- DeviceInstallParams.cbSize = sizeof (DeviceInstallParams);
-
- if (!SetupDiGetDeviceInstallParams(DeviceInfoSet,
- DeviceInfoData,
- &DeviceInstallParams))
- goto fail1;
-
- Log("Flags = %08x", DeviceInstallParams.Flags);
-
- if (!Context->PostProcessing) {
- Error = DifInstallPreProcess(DeviceInfoSet, DeviceInfoData, Context);
-
- if (Error == NO_ERROR)
- Error = ERROR_DI_POSTPROCESSING_REQUIRED;
- } else {
- Error = Context->InstallResult;
-
- if (Error == NO_ERROR) {
- (VOID) DifInstallPostProcess(DeviceInfoSet, DeviceInfoData, Context);
- } else {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("NOT RUNNING (DifInstallPreProcess Error: %s)", Message);
- LocalFree(Message);
- }
-
- Error = NO_ERROR;
- }
-
- return Error;
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return Error;
-}
-
-static HRESULT
-DifRemovePreProcess(
- IN HDEVINFO DeviceInfoSet,
- IN PSP_DEVINFO_DATA DeviceInfoData,
- IN PCOINSTALLER_CONTEXT_DATA Context
- )
-{
- UNREFERENCED_PARAMETER(DeviceInfoSet);
- UNREFERENCED_PARAMETER(DeviceInfoData);
- UNREFERENCED_PARAMETER(Context);
-
- Log("<===>");
-
- return NO_ERROR;
-}
-
-static HRESULT
-DifRemovePostProcess(
- IN HDEVINFO DeviceInfoSet,
- IN PSP_DEVINFO_DATA DeviceInfoData,
- IN PCOINSTALLER_CONTEXT_DATA Context
- )
-{
- UNREFERENCED_PARAMETER(DeviceInfoSet);
- UNREFERENCED_PARAMETER(DeviceInfoData);
- UNREFERENCED_PARAMETER(Context);
-
- Log("<===>");
-
- return NO_ERROR;
-}
-
-static DECLSPEC_NOINLINE HRESULT
-DifRemove(
- IN HDEVINFO DeviceInfoSet,
- IN PSP_DEVINFO_DATA DeviceInfoData,
- IN PCOINSTALLER_CONTEXT_DATA Context
- )
-{
- SP_DEVINSTALL_PARAMS DeviceInstallParams;
- HRESULT Error;
-
- DeviceInstallParams.cbSize = sizeof (DeviceInstallParams);
-
- if (!SetupDiGetDeviceInstallParams(DeviceInfoSet,
- DeviceInfoData,
- &DeviceInstallParams))
- goto fail1;
-
- Log("Flags = %08x", DeviceInstallParams.Flags);
-
- if (!Context->PostProcessing) {
- Error = DifRemovePreProcess(DeviceInfoSet, DeviceInfoData, Context);
-
- if (Error == NO_ERROR)
- Error = ERROR_DI_POSTPROCESSING_REQUIRED;
- } else {
- Error = Context->InstallResult;
-
- if (Error == NO_ERROR) {
- (VOID) DifRemovePostProcess(DeviceInfoSet, DeviceInfoData, Context);
- } else {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("NOT RUNNING (DifRemovePreProcess Error: %s)", Message);
- LocalFree(Message);
- }
-
- Error = NO_ERROR;
- }
-
- return Error;
-
-fail1:
- Error = GetLastError();
-
- {
- PTCHAR Message;
-
- Message = GetErrorMessage(Error);
- Log("fail1 (%s)", Message);
- LocalFree(Message);
- }
-
- return Error;
-}
-
-DWORD CALLBACK
-Entry(
- IN DI_FUNCTION Function,
- IN HDEVINFO DeviceInfoSet,
- IN PSP_DEVINFO_DATA DeviceInfoData,
- IN PCOINSTALLER_CONTEXT_DATA Context
- )
-{
- HRESULT Error;
-
- Log("%s (%s) ===>",
- MAJOR_VERSION_STR "." MINOR_VERSION_STR "." MICRO_VERSION_STR "." BUILD_NUMBER_STR,
- DAY_STR "/" MONTH_STR "/" YEAR_STR);
-
- if (!Context->PostProcessing) {
- Log("%s PreProcessing",
- __FunctionName(Function));
- } else {
- Log("%s PostProcessing (%08x)",
- __FunctionName(Function),
- Context->InstallResult);
- }
-
- switch (Function) {
- case DIF_INSTALLDEVICE: {
- SP_DRVINFO_DATA DriverInfoData;
- BOOLEAN DriverInfoAvailable;
-
- DriverInfoData.cbSize = sizeof (DriverInfoData);
- DriverInfoAvailable = SetupDiGetSelectedDriver(DeviceInfoSet,
- DeviceInfoData,
- &DriverInfoData) ?
- TRUE :
- FALSE;
-
- // If there is no driver information then the NULL driver is being
- // installed. Treat this as we would a DIF_REMOVE.
- Error = (DriverInfoAvailable) ?
- DifInstall(DeviceInfoSet, DeviceInfoData, Context) :
- DifRemove(DeviceInfoSet, DeviceInfoData, Context);
- break;
- }
- case DIF_REMOVE:
- Error = DifRemove(DeviceInfoSet, DeviceInfoData, Context);
- break;
- default:
- if (!Context->PostProcessing) {
- Error = NO_ERROR;
- } else {
- Error = Context->InstallResult;
- }
-
- break;
- }
-
- Log("%s (%s) <===",
- MAJOR_VERSION_STR "." MINOR_VERSION_STR "." MICRO_VERSION_STR "." BUILD_NUMBER_STR,
- DAY_STR "/" MONTH_STR "/" YEAR_STR);
-
- return (DWORD)Error;
-}
-
-DWORD CALLBACK
-Version(
- IN HWND Window,
- IN HINSTANCE Module,
- IN PTCHAR Buffer,
- IN INT Reserved
- )
-{
- UNREFERENCED_PARAMETER(Window);
- UNREFERENCED_PARAMETER(Module);
- UNREFERENCED_PARAMETER(Buffer);
- UNREFERENCED_PARAMETER(Reserved);
-
- Log("%s (%s)",
- MAJOR_VERSION_STR "." MINOR_VERSION_STR "." MICRO_VERSION_STR "." BUILD_NUMBER_STR,
- DAY_STR "/" MONTH_STR "/" YEAR_STR);
-
- return NO_ERROR;
-}
-
-static FORCEINLINE const CHAR *
-__ReasonName(
- IN DWORD Reason
- )
-{
-#define _NAME(_Reason) \
- case DLL_ ## _Reason: \
- return #_Reason;
-
- switch (Reason) {
- _NAME(PROCESS_ATTACH);
- _NAME(PROCESS_DETACH);
- _NAME(THREAD_ATTACH);
- _NAME(THREAD_DETACH);
- default:
- break;
- }
-
- return "UNKNOWN";
-
-#undef _NAME
-}
-
-BOOL WINAPI
-DllMain(
- IN HINSTANCE Module,
- IN DWORD Reason,
- IN PVOID Reserved
- )
-{
- UNREFERENCED_PARAMETER(Module);
- UNREFERENCED_PARAMETER(Reserved);
-
- Log("%s (%s): %s",
- MAJOR_VERSION_STR "." MINOR_VERSION_STR "." MICRO_VERSION_STR "." BUILD_NUMBER_STR,
- DAY_STR "/" MONTH_STR "/" YEAR_STR,
- __ReasonName(Reason));
-
- return TRUE;
-}
projects / pvdrivers / win / xenvif.git / commitdiff