windows设备驱动安装接口（自己仿写）
2024-08-28 15:54:24
/*****************************************
Author:foo_hack This is File named:Setup.h The Funtion Implement in Setup.cpp ******************************************/ #include "stdafx.h" void UDBG(TCHAR *msg, ...); void FreeAllInfFiles();
int FindAllInfFiles(WCHAR* csDir);
void InstallDriver();
int UpdateDriver();
void FindAllDevices(const WCHAR* asHID);
void FreeDevices();
void RemoveDevices();
BOOL Reboot();
BOOL ConfigRight(WCHAR* szPath); VOID SearchRegistryHardID(PWCHAR pHardIDBuffer); //////////////////////////////////////////////////////////////////// BOOL IsDeviceInstallInProgress(VOID);
int RemoveDriver(_TCHAR *HardwareID);
VOID UninstallWDMDriver(LPCTSTR theHardware);
BOOL UnicodeToAnsi(LPCWSTR Source, const WORD wLen, LPSTR Destination, const WORD sLen);
BOOL AnsiToUnicode(LPCSTR Source, const WORD sLen, LPWSTR Destination, const WORD wLen);
//BOOL GetINFData(FILE *pFile);
VOID FindComma(LPSTR szData);
//BOOL GetSectionData(FILE* pFile, const char* szKey, const char bIsVender);
//BOOL IsInstalled();
BOOL InstallClassDriver(LPCTSTR theINFName);
BOOL StartInstallWDMDriver(LPCTSTR theInfName) ; BOOL FindExistingDevice(IN LPTSTR HardwareId); VOID InitialGlobalVar(); //////////////////////////////////////////////////////////////////// //extern WORD g_wVender;
//extern WORD g_wHardware;
//extern TCHAR g_strVender[20][64];
//extern TCHAR g_strHardware[20][64];
//extern TCHAR g_strHID[MAX_PATH+1]; /////////////////////////////////////////////////////////////////// typedef struct
{
WCHAR asHID[MAX_PATH];
int nInfNum; // g_asInfList下标号 int nNumCone;
int nNumNotCone;
bool bNeedUpdate; bool bWillBeUpdate;
bool bSuccess;
//int nFail;
} HID_INFO, *PHID_INFO; typedef struct
{
WCHAR asInfName[MAX_PATH]; // 文件名
WCHAR asFullPath[MAX_PATH]; // 全路径
WCHAR asDesPath[MAX_PATH];
WCHAR *asDesName;
WCHAR asVersion[MAX_PATH];
WCHAR asClassName[MAX_PATH];
WCHAR asProvider[MAX_PATH];
WCHAR asDSign[MAX_PATH];
bool bWillBeInstall;
int nResultInstall;
}INF_INFO, *PINF_INFO; #define MAX_SYMBOL_LINK 250 typedef struct Hid_Ddvices{
WCHAR asHID[MAX_PATH]; WCHAR* asDeviceInstanceID[MAX_SYMBOL_LINK];
bool bLink[MAX_SYMBOL_LINK];
bool bSuccess[MAX_SYMBOL_LINK];
bool bWillBeDelete[MAX_SYMBOL_LINK]; int nMaxCount;
int nCount; bool bNeedReboot; Hid_Ddvices()
{
nCount = nMaxCount = ;
bNeedReboot = false; ZeroMemory(asHID, sizeof(asHID)+ sizeof(asDeviceInstanceID) +
sizeof(bLink) + sizeof(bSuccess) + sizeof(bWillBeDelete));
}
}HID_DEVICES, *PHID_DEVICES; #define MAX_INF_COUNT 40
#define MAX_HID_COUNT 250 extern PINF_INFO g_InfList[MAX_INF_COUNT];
extern int g_nInfCount; // INF文 extern PHID_INFO g_HIDInfo[MAX_HID_COUNT];
extern int g_nHIDCount ; // HID个数 extern HID_DEVICES g_HIDDevices;
Setup.h
#include "stdafx.h"
#include <newdev.h>
#include <setupapi.h>
#include <cfgmgr32.h>
#include <regstr.h>
#include <strsafe.h>
#include <string.h> #include <Aclapi.h>
#include "setup.h" #pragma comment(lib,"Advapi32") #pragma comment(lib, "newdev.lib")
#pragma comment(lib, "setupapi.lib") void GetHIDList(int nInfIndex); DWORD DeleteTree(HKEY RootKey, const WCHAR *pSubKey); // 不带变参
void ULOG(TCHAR *msg)
{
OutputDebugString(_T("UsbKitApp:"));
OutputDebugString(msg);
OutputDebugString(_T("\r\n"));
} #ifdef DEBUG
// 带变参的Log函数
void UDBG(TCHAR *msg, ...)
{
TCHAR strMsg[]; va_list argp;
va_start(argp, msg);
_vstprintf_s(strMsg, , msg, argp);
va_end(argp);
OutputDebugString(_T("UsbKitApp:"));
OutputDebugString(strMsg);
OutputDebugString(_T("\r\n"));
}
#else
void UDBG(TCHAR* msg, ...)
{ }
#endif PINF_INFO g_InfList[MAX_INF_COUNT];
int g_nInfMaxCount = ;
int g_nInfCount = ; // INF文件数 PHID_INFO g_HIDInfo[MAX_HID_COUNT];
int g_nHIDMaxCount = ;
int g_nHIDCount = ; // HID个数 HID_DEVICES g_HIDDevices; void FreeDevices()
{
for(int i = ; i < g_HIDDevices.nMaxCount; i++)
{
delete g_HIDDevices.asDeviceInstanceID[i];
} ZeroMemory(&g_HIDDevices, sizeof(g_HIDDevices));
} void FreeAllInfFiles()
{
for(int i = ; i < g_nHIDMaxCount; i++)
delete g_HIDInfo[i]; g_nHIDMaxCount = g_nHIDCount = ; for(int i = ; i < g_nInfMaxCount; i++)
delete g_InfList[i]; g_nInfMaxCount = g_nInfCount = ;
} //在特定路径下寻找所有的inf文件
int FindAllInfFiles(WCHAR* csDir = NULL)
{
WIN32_FIND_DATA FindFileData;
WCHAR dir[MAX_PATH];
WCHAR path[MAX_PATH];
HANDLE hFind = INVALID_HANDLE_VALUE; if(csDir == NULL)
{
// 在当前路径中搜索所有inf文件
GetCurrentDirectory(MAX_PATH, dir);
//StringCchCopy(path, MAX_PATH, dir);//此函数不支持win XP SP1
wcscpy(path,dir);
}
else
{
if(!GetFullPathName(csDir, MAX_PATH, dir, NULL))
GetCurrentDirectory(MAX_PATH, dir); //StringCchCopy(path, MAX_PATH, dir); wcscpy(path,dir);
} //StringCchCat(path, MAX_PATH, L"\\*.inf");//此函数不支持win XP SP1
wcscat(path,L"\\*.inf"); wprintf(path);
printf("\n"); __try{ g_nInfCount = ;
g_nHIDCount = ; // 遍历
hFind = FindFirstFile(path, &FindFileData); if (hFind == INVALID_HANDLE_VALUE)
{
printf("没有发现inf文件\n");
__leave;
} do{
if(g_nInfCount >= g_nInfMaxCount)
{
g_InfList[g_nInfCount] = new INF_INFO;
g_nInfMaxCount = g_nInfCount + ;
} ZeroMemory(g_InfList[g_nInfCount], sizeof(INF_INFO)); //StringCchCopy(path, MAX_PATH, dir);
wcscpy(path,dir);
//StringCchCat(path, MAX_PATH, L"\\");
wcscat(path,L"\\");
//StringCchCat(path, MAX_PATH, FindFileData.cFileName);
wcscat(path,FindFileData.cFileName);
//StringCchCopy(g_InfList[g_nInfCount]->asInfName, MAX_PATH, FindFileData.cFileName);
wcscpy(g_InfList[g_nInfCount]->asInfName,FindFileData.cFileName);
//StringCchCopy(g_InfList[g_nInfCount]->asFullPath, MAX_PATH, path);
wcscpy(g_InfList[g_nInfCount]->asFullPath,path);
// 查找每个inf文件中的所有HID
wprintf(L"找到文件：%s\n", FindFileData.cFileName); GetHIDList(g_nInfCount);
g_nInfCount ++; }while (FindNextFile(hFind, &FindFileData) != && g_nInfMaxCount < MAX_INF_COUNT) ; }
__finally{
if(hFind != INVALID_HANDLE_VALUE)
{
FindClose(hFind);
}
} return g_nHIDCount;
} //查找特定inf文件中的所有HID（硬件ID）
void GetHIDList(int nInfIndex)
{
INFCONTEXT infcont;
INFCONTEXT infcont_dev; HINF hInf = INVALID_HANDLE_VALUE;
DWORD config_flags, problem, status;
BOOL reboot;
WCHAR inf_path[MAX_PATH]; WCHAR direct[MAX_PATH];
WCHAR manu[MAX_PATH];
WCHAR manu_os[MAX_PATH];
WCHAR hid[MAX_PATH];
WCHAR tmp_id[MAX_PATH]; WCHAR os[][]; __try
{
// 取INF路径
wprintf(g_InfList[nInfIndex]->asFullPath);
if(!GetFullPathName(g_InfList[nInfIndex]->asFullPath, MAX_PATH, inf_path, NULL))
{
printf(".inf文件 %s 未找到\n", g_InfList[nInfIndex]->asInfName);
__leave;
} wprintf(L"INF 文件路径：%s\n", inf_path); // 打开INF文件
hInf = SetupOpenInfFile(inf_path, NULL, INF_STYLE_WIN4, NULL); if(hInf == INVALID_HANDLE_VALUE)
{
wprintf(L"打开文件失败：%s", inf_path);
__leave;
} if(!SetupFindFirstLine(hInf, L"version", NULL, &infcont))
{
printf("找不到[Version]域，不是合法的Inf文件\n");
__leave;
} //
// 获取Version信息
// do
{
if(!SetupGetStringField(&infcont, , direct, sizeof(direct), NULL))
continue; _wcslwr(direct);//把字符串转为小写
if( == wcscmp(direct, L"class"))
{
if(SetupGetStringField(&infcont, , direct, sizeof(direct), NULL))
{
//StringCchCopy(g_InfList[nInfIndex]->asClassName, MAX_PATH, direct);
wcscpy(g_InfList[nInfIndex]->asClassName,direct);
}
}
else if( == wcscmp(direct, L"driverver"))
{
if(SetupGetStringField(&infcont, , direct, sizeof(direct), NULL))
{
//StringCchCopy(g_InfList[nInfIndex]->asVersion, MAX_PATH, direct);
wcscpy(g_InfList[nInfIndex]->asVersion,direct); if(SetupGetStringField(&infcont, , direct, sizeof(direct), NULL))
{
//StringCchCat(g_InfList[nInfIndex]->asVersion, MAX_PATH, L", ");
wcscat(g_InfList[nInfIndex]->asVersion,L", ");
//StringCchCat(g_InfList[nInfIndex]->asVersion, MAX_PATH, direct);
wcscat(g_InfList[nInfIndex]->asVersion,direct);
}
}
}
else if( == wcscmp(direct, L"provider"))
{
if(SetupGetStringField(&infcont, , direct, sizeof(direct), NULL))
{
//StringCchCopy(g_InfList[nInfIndex]->asProvider, MAX_PATH, direct);
wcscpy(g_InfList[nInfIndex]->asProvider,direct);
}
}
else if(wcsstr(direct, L"catalogfile"))//返回第一个子字符创的指针
{
if(SetupGetStringField(&infcont, , direct, sizeof(direct), NULL))
{
//StringCchCopy(g_InfList[nInfIndex]->asDSign, MAX_PATH, direct);
wcscpy(g_InfList[nInfIndex]->asDSign,direct);
}
}
} while (SetupFindNextLine(&infcont, &infcont)); printf("INF 信息：\n");
wprintf(L"%s\n", g_InfList[nInfIndex]->asClassName);
wprintf(L"%s\n", g_InfList[nInfIndex]->asProvider);
wprintf(L"%s\n", g_InfList[nInfIndex]->asDSign);
wprintf(L"%s\n", g_InfList[nInfIndex]->asVersion); if(g_nHIDMaxCount >= MAX_HID_COUNT)
__leave; //
// 搜索设备描述符
// if(!SetupFindFirstLine(hInf, L"Manufacturer", NULL, &infcont))
{
printf("没有可用设备描述符\n");
__leave;
} SYSTEM_INFO siSysInfo;
GetSystemInfo(&siSysInfo); do{
int i = ;
int j = ; // 一行行搜索
if(!SetupGetStringField(&infcont, , manu, sizeof(manu), NULL))
{
continue;
} wprintf(manu); printf("\n"); for(; i < ; i++)
{
if(!SetupGetStringField(&infcont, i+, os[i], sizeof(os[]), NULL))
break;
else
printf("OS: %s\n", os[i]); _wcslwr(os[i]);
} // 寻找最佳的平台匹配
if(siSysInfo.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64) // X64
{
for(; j < i; j++)
{
if(wcscmp(os[j], L"ntamd64"))
{
//StringCchCopy(manu_os, sizeof(manu_os), manu);
wcscpy(manu_os,manu);
//StringCchCat(manu_os, sizeof(manu_os), L".");
wcscat(manu_os,L".");
//StringCchCat(manu_os, sizeof(manu_os), os[j]);
wcscat(manu_os,os[i]); // 如果找到[XXX.ntamd64],就用它；否则去掉os后缀
if(SetupFindFirstLine(hInf, manu_os, NULL, &infcont_dev))
//StringCchCopy(manu, sizeof(manu), manu_os);
wcscpy(manu,manu_os); break;
}
}
}
else if(siSysInfo.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_IA64)
{
for(; j < i; j++)
{
if( == wcsstr(os[j], L"ntia64"))
{
//StringCchCopy(manu_os, sizeof(manu_os), manu);
wcscpy(manu_os,manu);
//StringCchCat(manu_os, sizeof(manu_os), L".");
wcscat(manu_os,L".");
//StringCchCat(manu_os, sizeof(manu_os), os[j]);
wcscat(manu_os,os[j]); // 如果找到[XXX.ntamd64],就用它；否则去掉os后缀
if(SetupFindFirstLine(hInf, manu_os, NULL, &infcont_dev))
//StringCchCopy(manu, sizeof(manu), manu_os);
wcscpy(manu,manu_os); break;
}
}
}
else if(siSysInfo.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_INTEL)
{
for(; j < i; j++)
{
if( == wcsstr(os[j], L"ntx86"))
{
//StringCchCopy(manu_os, sizeof(manu_os), manu);
wcscpy(manu_os,manu);
//StringCchCat(manu_os, sizeof(manu_os), L".");
wcscat(manu_os,L".");
//StringCchCat(manu_os, sizeof(manu_os), os[j]);
wcscat(manu_os,os[j]);
wprintf(manu_os);
printf("\n");
// 如果找到[XXX.ntamd64],就用它；否则去掉os后缀
if(SetupFindFirstLine(hInf, manu_os, NULL, &infcont_dev))
//StringCchCopy(manu, sizeof(manu), manu_os);
wcscpy(manu,manu_os); break;
}
}
} // 最后，如果最符合的版本没有找到，就看看.nt有没有
if(NULL == wcsstr(manu, L".ntx86") &&
NULL == wcsstr(manu, L".ntia64") &&
NULL == wcsstr(manu, L".ntamd64"))
{
for(j = ; j < i; j++)
{
if( == wcsstr(os[j], L".nt"))
{
//StringCchCopy(manu_os, sizeof(manu_os), manu);
wcscpy(manu_os,manu);
//StringCchCat(manu_os, sizeof(manu_os), L".");
wcscat(manu_os,L".");
//StringCchCat(manu_os, sizeof(manu_os), os[j]);
wcscat(manu_os,os[j]); // 如果找到[XXX.ntamd64],就用它；否则去掉os后缀
if(SetupFindFirstLine(hInf, manu_os, NULL, &infcont_dev))
//StringCchCopy(manu, sizeof(manu), manu_os);
wcscpy(manu,manu_os); break;
}
}
} wprintf(L"设备域名：%s\n", manu);
if(!SetupFindFirstLine(hInf, manu, NULL, &infcont_dev))
{
printf("找不到设备模块\n");
continue;
} do{
if(!SetupGetStringField(&infcont_dev, , hid, sizeof(hid), NULL))
continue; // 保存HID记录
// _wcslwr(hid);
wprintf(L"HID: %s\n", hid); if(g_nHIDMaxCount <= g_nHIDCount)
{
g_HIDInfo[g_nHIDCount] = new HID_INFO;
g_nHIDMaxCount = g_nHIDCount + ;
} ZeroMemory(g_HIDInfo[g_nHIDCount], sizeof(HID_INFO));
//StringCchCopy(g_HIDInfo[g_nHIDCount]->asHID, MAX_PATH, hid);
wcscpy(g_HIDInfo[g_nHIDCount]->asHID,hid);
g_HIDInfo[g_nHIDCount]->nInfNum = nInfIndex;
g_nHIDCount++; }while(SetupFindNextLine(&infcont_dev, &infcont_dev) && g_nHIDMaxCount < MAX_HID_COUNT); // 获取下一个设备ID
}while(SetupFindNextLine(&infcont, &infcont) && g_nHIDMaxCount < MAX_HID_COUNT);
}
__finally
{
if(hInf != INVALID_HANDLE_VALUE)
SetupCloseInfFile(hInf);
}
} BOOL Reboot()
{
HANDLE Token = NULL;
TOKEN_PRIVILEGES NewPrivileges;
LUID Luid; __try
{
// 使能ShutDown特权
if(!OpenProcessToken(GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES,&Token))
__leave; if(!LookupPrivilegeValue(NULL,SE_SHUTDOWN_NAME,&Luid))
__leave; NewPrivileges.PrivilegeCount = ;
NewPrivileges.Privileges[].Luid = Luid;
NewPrivileges.Privileges[].Attributes = SE_PRIVILEGE_ENABLED; AdjustTokenPrivileges(Token, FALSE, &NewPrivileges, , NULL, NULL);
}
__finally
{
if(Token) CloseHandle(Token);
} //尝试重启
return InitiateSystemShutdownEx(NULL, NULL, , FALSE, TRUE, REASON_PLANNED_FLAG | REASON_HWINSTALL);
} // 删除设备对应的驱动程序，调用SetupDiCallClassInstaller函数，并执行DIF_REMOVE动作
// 参数Devs和DevInfo用来唯一标识目标设备
bool RemoveDevice(__in HDEVINFO Devs, __in PSP_DEVINFO_DATA DevInfo)
{
UDBG(_T("[RemoveDriverForSpecifiedDevice]")); SP_REMOVEDEVICE_PARAMS rmdParams;
SP_DEVINSTALL_PARAMS devParams; rmdParams.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER);
rmdParams.ClassInstallHeader.InstallFunction = DIF_REMOVE;
rmdParams.Scope = DI_REMOVEDEVICE_GLOBAL;
rmdParams.HwProfile = ; if(SetupDiSetClassInstallParams(Devs, DevInfo, &rmdParams.ClassInstallHeader, sizeof(rmdParams)) &&
SetupDiCallClassInstaller(DIF_REMOVE, Devs, DevInfo))
{
wprintf(_T("卸载驱动成功\n")); // 检查是否需要重启机器
devParams.cbSize = sizeof(devParams);
if(SetupDiGetDeviceInstallParams(Devs, DevInfo, &devParams) &&
(devParams.Flags & (DI_NEEDRESTART|DI_NEEDREBOOT)))
{
g_HIDDevices.bNeedReboot = true;
} return true;
} return false;
} void RemoveDevices()
{
HDEVINFO dev_info;
SP_DEVINFO_DATA dev_info_data;
int nDevIndex;
WCHAR id[MAX_PATH]; g_HIDDevices.bNeedReboot = false; // 寻找所有设备，设置Flag值为DIGCF_ALLCLASSES。
dev_info = SetupDiGetClassDevs(NULL, NULL, NULL, DIGCF_ALLCLASSES); // class名无效
if(dev_info == INVALID_HANDLE_VALUE)
{
printf("无效的HDEVINFO句柄\n");
return;
} nDevIndex = ; // 枚举设备
dev_info_data.cbSize = sizeof(dev_info_data);
while(SetupDiEnumDeviceInfo(dev_info, nDevIndex, &dev_info_data))
{
if(CR_SUCCESS != CM_Get_Device_ID(dev_info_data.DevInst, id, MAX_PATH, ))
continue; _wcslwr(id); for(int i = ; i < g_HIDDevices.nCount; i++)
{
if(g_HIDDevices.bWillBeDelete[i] &&
== wcscmp(id, g_HIDDevices.asDeviceInstanceID[i]))
{
wprintf(id);
printf("\n");
g_HIDDevices.bSuccess[i] = RemoveDevice(dev_info, &dev_info_data);
}
} nDevIndex++;
} SetupDiDestroyDeviceInfoList(dev_info);
} //该函数主要就是根据硬件ID查找在本地机器上有没有这样一个设备，\
把查找到的ID放到全局变量中，还检测设备有没有连接系统
void FindAllDevices(const WCHAR* asHID)
{
HDEVINFO dev_info;
SP_DEVINFO_DATA dev_info_data; int nDevIndex;
WCHAR id[MAX_PATH];
WCHAR *p;
ULONG problem;
ULONG status; WCHAR* ashid = _wcsdup(asHID);
_wcslwr(ashid);//转换ashid中的大写字母 printf("FindAllDevices\n"); g_HIDDevices.nCount = ; // 寻找所有设备，设置Flag值为DIGCF_ALLCLASSES。
dev_info = SetupDiGetClassDevs(NULL, NULL, NULL, DIGCF_ALLCLASSES); // class名无效
if(dev_info == INVALID_HANDLE_VALUE)
{
printf("无效的HDEVINFO句柄\n");
free(ashid);
return;
} nDevIndex = ; // 枚举设备，从0开始枚举设备
dev_info_data.cbSize = sizeof(dev_info_data);
while(SetupDiEnumDeviceInfo(dev_info, nDevIndex, &dev_info_data))
{
if(CR_SUCCESS != CM_Get_Device_ID(dev_info_data.DevInst, id, MAX_PATH, ))
continue; (id);
if(wcsstr(id, ashid))
{
if(g_HIDDevices.nCount >= g_HIDDevices.nMaxCount)
{
g_HIDDevices.asDeviceInstanceID[g_HIDDevices.nMaxCount] = new WCHAR[MAX_PATH];
g_HIDDevices.nMaxCount++;
} // Device实例ID。把id拷贝到parameter 1中
//StringCchCopy(g_HIDDevices.asDeviceInstanceID[g_HIDDevices.nCount], MAX_PATH, id);
wcscpy(g_HIDDevices.asDeviceInstanceID[g_HIDDevices.nCount],id);
// 连接状态
g_HIDDevices.bLink[g_HIDDevices.nCount] = false;
if(CM_Get_DevNode_Status(&status,
&problem,
dev_info_data.DevInst,
) != CR_NO_SUCH_DEVINST)
{
g_HIDDevices.bLink[g_HIDDevices.nCount] = true;
} g_HIDDevices.nCount++;
} nDevIndex++;
} SetupDiDestroyDeviceInfoList(dev_info);
free(ashid);
} void InstallDriver()
{
HDEVINFO dev_info;
SP_DEVINFO_DATA dev_info_data;
int nDevIndex;
WCHAR id[MAX_PATH];
WCHAR tmp_id[MAX_PATH];
WCHAR *p;
ULONG problem;
ULONG status;
int nInfIndex; for(nInfIndex = ; nInfIndex < g_nInfCount; nInfIndex++)
{
if(false == g_InfList[nInfIndex]->bWillBeInstall)
continue; // 拷贝INF文件（预安装到Driver Store中）
wprintf(L"Parameter 1 is %s\n",g_InfList[nInfIndex] ->asFullPath);
//getch();
//wprintf(L"Parameter 1 is %s\n",g_InfList[nInfIndex] ->asDesPath);
if(FALSE == SetupCopyOEMInf(g_InfList[nInfIndex]->asFullPath, NULL, SPOST_PATH,,
, ,
NULL, NULL))
{
printf("SetCopyOEMInf1 Error Code is %d\n",GetLastError()); if(GetLastError() == ERROR_FILE_EXISTS)
{
g_InfList[nInfIndex]->nResultInstall = ;
wprintf(L"located in inf dir's name:%s\n",g_InfList[nInfIndex]->asDesPath); PWCHAR pSubStr1,pSubWork1;
WCHAR OemFileName3[]; if(NULL != (pSubStr1 = wcsstr((wchar_t*)g_InfList[nInfIndex]->asDesPath,L"oem")))
{
pSubWork1 = pSubStr1; while((*pSubWork1) != 'f')
{
pSubWork1++;
} pSubWork1++; *pSubWork1 = '\0'; wcscpy(OemFileName3,pSubStr1); wprintf(L"OemFileName3 is %s\n",OemFileName3); //getch();
} if(SetupUninstallOEMInf(OemFileName3, SUOI_FORCEDELETE, NULL))
{
printf("new Driver Package Has Deleted succced! on windows XP\n"); if(FALSE == SetupCopyOEMInf(g_InfList[nInfIndex]->asFullPath, NULL, SPOST_PATH, SP_COPY_NOOVERWRITE,
g_InfList[nInfIndex]->asDesPath, MAX_PATH,
NULL, &g_InfList[nInfIndex]->asDesName))
{
printf("SetCopyOEMInf2 Error Code is %d\n",GetLastError());
}
else
printf("SetupCopyOEMInf2 succeed!\n"); //getch(); } }
else
g_InfList[nInfIndex]->nResultInstall = -;
}else
{ printf("Install Succeed!\n"); //getch();
g_InfList[nInfIndex]->nResultInstall = ;
} BOOL bResult = FALSE;
HDEVINFO DeviceInfoSet = INVALID_HANDLE_VALUE;
SP_DEVINFO_DATA DeviceInfoData;
GUID ClassGUID;
TCHAR ClassName[MAX_CLASS_NAME_LEN];
TCHAR hwIdList[LINE_LEN+];
DWORD Status; if (!SetupDiGetINFClass (g_InfList[nInfIndex]->asFullPath, &ClassGUID, ClassName, sizeof(ClassName)/sizeof(TCHAR), ) )
{
printf("SetupDiGetINFClass fail,Error Code is %d\n",GetLastError());
} DeviceInfoSet = SetupDiGetClassDevs(&ClassGUID,ClassName,NULL,DIGCF_PRESENT); if(DeviceInfoSet == INVALID_HANDLE_VALUE)
{
printf("SetupDiGetClassDevs fail,Error Code is %d\n",GetLastError());
} /*DWORD wIdx = 0; while (TRUE)
{
DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
//找到所有的硬件设备,并且可以得到所有的硬件设备的详细信息
if (SetupDiEnumDeviceInfo(DeviceInfoSet, wIdx, &DeviceInfoData))
{ printf("SetupDiEnumDeviceInfo Success!\n"); char Buffer[2048] = {0}; //可以在前面得到的指向某一个具体设备信息集合的指针中取出某一项信息
if (SetupDiGetDeviceRegistryProperty(DeviceInfoSet, &DeviceInfoData, SPDRP_HARDWAREID,
0L, (PBYTE)Buffer, 2048, 0L))
{
printf("SetupDiGetDeviceRegistryProperty Success!\n"); wprintf(L"Buffer is %s\n",Buffer); if (!lstrcmpi(L"USB\\VID_0451&PID_AF32&REV_0000", (LPTSTR)Buffer))
{
printf("theHardware is matched!\n"); DWORD ConfigFlag; if(SetupDiGetDeviceRegistryProperty(DeviceInfoSet,&DeviceInfoData,SPDRP_CONFIGFLAGS,
0L,(PBYTE)&ConfigFlag,sizeof(ConfigFlag),0L))
{ ConfigFlag |= CONFIGFLAG_REINSTALL; SetupDiSetDeviceRegistryProperty(DeviceInfoSet,&DeviceInfoData,SPDRP_CONFIGFLAGS,
(PBYTE)&ConfigFlag,sizeof(ConfigFlag)); } if(!UpdateDriverForPlugAndPlayDevices(NULL,L"USB\\VID_0451&PID_AF32&REV_0000",
g_InfList[nInfIndex] ->asFullPath,INSTALLFLAG_FORCE,NULL))
{
printf("SetupDiGetINFClass fail,Error Code is %x\n",GetLastError());
} break;
} }
}
else
{ break;
} wIdx++;
} */ /*wprintf(L"ClassName is %s\n",ClassName); DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA); if (!SetupDiCreateDeviceInfo(DeviceInfoSet, L"USB\\VID_0451&PID_AF32\\7LUCK", &ClassGUID, NULL, NULL, DICD_INHERIT_CLASSDRVS, &DeviceInfoData) )
{
printf("SetupDiCreateDeviceInfo fail,Error Code is %x\n",GetLastError());
} ZeroMemory (hwIdList, sizeof(hwIdList) ); lstrcpyn (hwIdList, L"USB\\VID_0451&PID_AF32", LINE_LEN); if (!SetupDiSetDeviceRegistryProperty (DeviceInfoSet, &DeviceInfoData, SPDRP_HARDWAREID, (LPBYTE)hwIdList,
( lstrlen (hwIdList) + 1 + 1) * sizeof (TCHAR) ) )
{
printf("SetupDiSetDeviceRegistryProperty fail,Error Code is %d\n",GetLastError()); } if (!SetupDiCallClassInstaller(DIF_REGISTERDEVICE, DeviceInfoSet, &DeviceInfoData) )
{
printf("SetupDiCallClassInstaller fail,Error Code is %d\n",GetLastError());
}*/ /*if (!UpdateDriverForPlugAndPlayDevices(
NULL,
L"USB\\VID_0451&PID_AF32",
g_InfList[nInfIndex]->asFullPath,
INSTALLFLAG_FORCE,
NULL) )
{
printf("UpdateDriverForPlugAndPlayDevices,Error Code is %x\n",GetLastError());
}*/ //Return Value：0xe0000203指定设备不在当前系统，0xe000020b没为设备指定设备集
/*if(!UpdateDriverForPlugAndPlayDevices(NULL,L"USB\\VID_0451&PID_AF32",
g_InfList[nInfIndex]->asFullPath,INSTALLFLAG_FORCE,NULL))
{
。 printf("UpdateDriverForPlugAndPlayDevices Failed! Error Code is %x\n",GetLastError());
}*/ getch(); /*if(!UpdateDriverForPlugAndPlayDevices(NULL,L"USB\\VID_0451&PID_AF32",
g_InfList[nInfIndex]->asFullPath,INSTALLFLAG_FORCE,NULL))
{
printf("UpdateDriverForPlugAndPlayDevices Failed! Error Code is %x\n",GetLastError());
}*/ //getch(); for(int i = ; i < g_nHIDCount; i++)
{
if(g_HIDInfo[i]->nInfNum == nInfIndex)
{
g_HIDInfo[i]->nNumCone = ;
g_HIDInfo[i]->nNumNotCone = ;
g_HIDInfo[i]->bNeedUpdate = false;
}
} // 获取设备实例数
if(g_InfList[nInfIndex]->nResultInstall == && g_InfList[nInfIndex]->asClassName[])
{ wprintf(g_InfList[nInfIndex]->asClassName);
printf("\n"); // 寻找所有设备，设置Flag值为DIGCF_ALLCLASSES。
dev_info = SetupDiGetClassDevs(NULL, NULL, NULL, DIGCF_ALLCLASSES); // class名无效
if(dev_info == INVALID_HANDLE_VALUE){
printf("class 名无效\n");
continue;
} nDevIndex = ; // 枚举设备
dev_info_data.cbSize = sizeof(dev_info_data);
while(SetupDiEnumDeviceInfo(dev_info, nDevIndex, &dev_info_data))
{
// 取设备ID
if(SetupDiGetDeviceRegistryProperty(dev_info, &dev_info_data,
SPDRP_HARDWAREID,
NULL,
(BYTE *)tmp_id,
sizeof(tmp_id),
NULL))
{
//获得的是一个字符串列表，故而需要遍历
for(p = tmp_id; *p; p += (wcslen(p) + ))
{
_wcslwr(p);
for(int x = ; x < g_nHIDCount; x++)
{
if(g_HIDInfo[x]->nInfNum != nInfIndex)
continue; if(wcsstr(p, g_HIDInfo[x]->asHID)){
g_HIDInfo[x]->bNeedUpdate = true;// 此设备已存在，需要更新。
if(CM_Get_DevNode_Status(&status,
&problem,
dev_info_data.DevInst,
) == CR_NO_SUCH_DEVINST)
{
g_HIDInfo[x]->nNumNotCone++;
}
else
g_HIDInfo[x]->nNumCone++;
}
}
}
} nDevIndex++;
} SetupDiDestroyDeviceInfoList(dev_info);
} SetupDiDestroyDeviceInfoList(dev_info); }
} // 更新驱动程序
int UpdateDriver()
{
HDEVINFO dev_info;
SP_DEVINFO_DATA dev_info_data;
INFCONTEXT infcont;
HINF hInf = NULL;
DWORD config_flags, problem, status;
BOOL reboot;
char inf_path[MAX_PATH];
char id[MAX_PATH];
char tmp_id[MAX_PATH];
char *p;
int dev_index; dev_info_data.cbSize = sizeof(SP_DEVINFO_DATA); __try
{
for(int i = ; i < g_nHIDCount; i++)
{
if(g_HIDInfo[i]->bWillBeUpdate)
{
if(g_HIDInfo[i]->nNumCone)
{
// 为当前连接到系统的所有设备更新驱动
g_HIDInfo[i]->bSuccess = UpdateDriverForPlugAndPlayDevices(NULL, g_HIDInfo[i]->asHID,
g_InfList[g_HIDInfo[i]->nInfNum]->asDesPath,
INSTALLFLAG_FORCE,
&reboot);
} if(g_HIDInfo[i]->nNumNotCone)
{
dev_index = ; // 寻找所有设备，设置Flag值为DIGCF_ALLCLASSES。
dev_info = SetupDiGetClassDevs(NULL, NULL, NULL, DIGCF_ALLCLASSES); if(dev_info == INVALID_HANDLE_VALUE)
continue; // 枚举设备
while(SetupDiEnumDeviceInfo(dev_info, dev_index, &dev_info_data))
{
// 先取设备ID，判断是否是当前更新设备
if(SetupDiGetDeviceRegistryProperty(dev_info, &dev_info_data,
SPDRP_HARDWAREID, NULL,
(BYTE *)tmp_id,
sizeof(tmp_id),
NULL))
{
//获得的是一个字符串列表，故而需要遍历
for(p = tmp_id; *p; p += (strlen(p) + ))
{
strlwr(p); if(strstr(p, id))
{
// 判断此设备是不是当前未连接在系统
if(CM_Get_DevNode_Status(&status,
&problem,
dev_info_data.DevInst,
) == CR_NO_SUCH_DEVINST)
{
// 取当前配置值
if(SetupDiGetDeviceRegistryProperty(dev_info,
&dev_info_data,
SPDRP_CONFIGFLAGS,
NULL,
(BYTE *)&config_flags,
sizeof(config_flags),
NULL))
{
// 与CONFIGFLAG_REINSTALL或
config_flags |= CONFIGFLAG_REINSTALL; // 将新的配置值写入
SetupDiSetDeviceRegistryProperty(dev_info,
&dev_info_data,
SPDRP_CONFIGFLAGS,
(BYTE *)&config_flags,
sizeof(config_flags));
}
} // 找到一个即可，跳出。
break;
}
}
} dev_index++;
} SetupDiDestroyDeviceInfoList(dev_info);
}
}
} }
__finally
{
} return ;
} /////////////////////以下函数是移入的////////////////////////////////////// BOOL IsDeviceInstallInProgress (VOID)
{
return !(CM_WaitNoPendingInstallEvents() == WAIT_OBJECT_0);
} int RemoveDriver(_TCHAR *HardwareID)
{
HDEVINFO DeviceInfoSet;
SP_DEVINFO_DATA DeviceInfoData;
DWORD i,err;
//GUID devGUID ={36fc9e60-c465-11cf-8056-444553540000}; DeviceInfoSet = SetupDiGetClassDevs(NULL, // All Classes
,
,
DIGCF_ALLCLASSES | DIGCF_PRESENT ); // All devices present on system if (DeviceInfoSet == INVALID_HANDLE_VALUE)
{
printf("GetClassDevs(All Present Devices)\n");
return ;
} //
// Enumerate through all Devices.
//
DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
for (i=;SetupDiEnumDeviceInfo(DeviceInfoSet,i,&DeviceInfoData);i++)
{
DWORD DataT;
LPTSTR p,buffer = NULL;
DWORD buffersize = ; //
// We won't know the size of the HardwareID buffer until we call
// this function. So call it with a null to begin with, and then
// use the required buffer size to Alloc the nessicary space.
// Keep calling we have success or an unknown failure.
//
while (!SetupDiGetDeviceRegistryProperty(
DeviceInfoSet,
&DeviceInfoData,
SPDRP_HARDWAREID,
&DataT,
(PBYTE)buffer,
buffersize,
&buffersize))
{
if (GetLastError() == ERROR_INVALID_DATA)
{
//
// May be a Legacy Device with no HardwareID. Continue.
//
break;
}
else if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
{
//
// We need to change the buffer size.
//
if (buffer)
LocalFree(buffer);
buffer = (LPTSTR)LocalAlloc(LPTR,buffersize);
}
else
{
//
// Unknown Failure.
//
printf("GetDeviceRegistryProperty");
goto cleanup_DeviceInfo;
}
} if (GetLastError() == ERROR_INVALID_DATA)
continue; //
// Compare each entry in the buffer multi-sz list with our HardwareID. //
for (p = buffer; *p && (p < &buffer[buffersize]);p+=lstrlen(p)*sizeof(TCHAR) + ) {
//_tprintf(TEXT("Compare device ID: [%s]/n"),p); if (!_tcscmp(HardwareID,p))
{
//_tprintf(TEXT("Found! [%s]/n"),p); //
// Worker function to remove device.
//
//if (SetupDiCallClassInstaller(DIF_REMOVE,
// DeviceInfoSet,
// &DeviceInfoData)) if (SetupDiRemoveDevice(DeviceInfoSet, &DeviceInfoData)) {
printf("CallClassInstaller(REMOVE)\n");
}
else
printf("Remove Driver Fail\n");
break;
} //printf("TTTTTTTTTTTTTTTTT is %s\n",p);
//getch();
} if (buffer) LocalFree(buffer);
} if ((GetLastError()!=NO_ERROR)&&(GetLastError()!=ERROR_NO_MORE_ITEMS))
{
printf("EnumDeviceInfo\n");
} //
// Cleanup.
//
cleanup_DeviceInfo:
err = GetLastError();
SetupDiDestroyDeviceInfoList(DeviceInfoSet); return err;
} VOID UninstallWDMDriver(LPCTSTR theHardware)
{ /* HKEY hKey;
//删除安装包inf，不然刷新设备会重新安装
if(INVALID_HANDLE_VALUE==(hKey = SetupDiOpenDevRegKey(hDevInfo,&spDevInfoData,DICS_FLAG_GLOBAL,0,
DIREG_DEV,KEY_ALL_ACCESS)))
{
printf("Open Key Has Failed.\n");
}*/ //RegGetValue(hKey,L"InfPath");
HKEY hKey[] = {};
DWORD type_1=REG_SZ;
BYTE buffer[] = {};
BYTE buffer2[] = {};
DWORD cbData_1=;
PWCHAR pSubStr = NULL;
PWCHAR pSubWork = NULL;
WCHAR SubKeyName[] = {};
DWORD SizeSubKeyName = sizeof(WCHAR)*;
DWORD Index = ;
WCHAR KeyPath[] = L"SYSTEM\\CurrentControlSet\\Enum\\USB\\VID_0451&PID_AF32";
//WCHAR constKeyPath[512] = L"SYSTEM\\CurrentControlSet\\Enum\\";
//wcscat(KeyPath,theHardware);
//wcscat(constKeyPath,theHardware);
WCHAR Path[] = L"MACHINE\\SYSTEM\\CurrentControlSet\\Enum\\USB\\VID_0451&PID_AF32";
//wcscat(Path,theHardware);
WCHAR DeleteKeyPath[] = L"SYSTEM\\CurrentControlSet\\Enum\\USB\\VID_0451&PID_AF32" ;
//wcscat(DeleteKeyPath,theHardware);
//WCHAR EndChar = 'f'; //if(ConfigRight(Path))
// printf("Set Right Success!\n"); //这里如果是KEY_ALL_ACCESS就会返回失败，以前尝试注册表提权ConfigRight失败。原因是无SYSTEM权限
//
if(ERROR_SUCCESS !=RegOpenKeyEx(HKEY_LOCAL_MACHINE,KeyPath,,KEY_ALL_ACCESS,&hKey[]))
printf("UninstallWDMDriver OpenKey Fail! error code is %d\n",GetLastError()); printf("OpenKey Success!\n"); //getch(); ZeroMemory(KeyPath,sizeof(WCHAR)*); //枚举子键 while(ERROR_SUCCESS == RegEnumKey(hKey[],Index,SubKeyName,SizeSubKeyName/*,NULL,NULL,NULL,NULL*/))
{ wcscpy(KeyPath,L"SYSTEM\\CurrentControlSet\\Enum\\USB\\VID_0451&PID_AF32");
wcscat(KeyPath,L"\\");
wcscat(KeyPath,SubKeyName); Index++; // ZeroMemory(buffer,sizeof(BYTE)*512);
//ZeroMemory(buffer2,sizeof(BYTE)*512);
// ZeroMemory(SubKeyName,SizeSubKeyName);
//ZeroMemory(KeyPath,sizeof(WCHAR)*512); if(ERROR_SUCCESS==RegOpenKeyEx(HKEY_LOCAL_MACHINE,KeyPath,,KEY_READ,&hKey[Index]))
printf("OpenKey Success!\n"); if( ERROR_SUCCESS==RegQueryValueEx(hKey[Index],L"DeviceDesc",NULL,&type_1,buffer,&cbData_1)) printf("Get Buffer1 Success!\n"); printf("%d\n",GetLastError()); if( ERROR_SUCCESS==RegQueryValueEx(hKey[Index],L"Mfg",NULL,&type_1,buffer2,&cbData_1))
printf("Get Buffer2 Success!\n"); printf("%d\n",GetLastError()); RegCloseKey(hKey[Index]); //搜索OEM文件 if(NULL == (pSubStr = wcsstr((wchar_t*)buffer,L"oem")))
{
printf("Search SubString_buffer Fail!\n"); if(NULL == (pSubStr = wcsstr((wchar_t*)buffer2,L"oem")))
{
printf("Search SubString_buffer2 Fail!\n"); ZeroMemory(buffer,sizeof(BYTE)*);
ZeroMemory(buffer2,sizeof(BYTE)*);
ZeroMemory(SubKeyName,SizeSubKeyName);
ZeroMemory(KeyPath,sizeof(WCHAR)*);
//KeyPath = NULL;
continue;
}
else
{
// ZeroMemory(buffer,sizeof(BYTE)*512);
//ZeroMemory(buffer2,sizeof(BYTE)*512);
//ZeroMemory(SubKeyName,SizeSubKeyName);
//ZeroMemory(KeyPath,sizeof(WCHAR)*512);
//KeyPath = NULL;
//continue; printf("Search Success!\n"); pSubWork = pSubStr; while((*pSubWork) != 'f')
{
pSubWork++;
} pSubWork++; *pSubWork = '\0'; WCHAR OemFileName1[]; wcscpy(OemFileName1,pSubStr); SP_DEVINFO_DATA spDevInfoData = {};
HDEVINFO hDevInfo = 0L;
WORD wIdx, wCount = ; //得到设备信息结构的句柄
hDevInfo = SetupDiGetClassDevs(0L, 0L, 0L, DIGCF_ALLCLASSES | DIGCF_PRESENT);
if (hDevInfo == INVALID_HANDLE_VALUE)
{
printf("Fail!\n");
return;
} wIdx = ;
while (TRUE)
{
spDevInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
//找到所有的硬件设备,并且可以得到所有的硬件设备的详细信息
if (SetupDiEnumDeviceInfo(hDevInfo, wIdx, &spDevInfoData))
{
char Buffer[] = {}; //可以在前面得到的指向某一个具体设备信息集合的指针中取出某一项信息
if (SetupDiGetDeviceRegistryProperty(hDevInfo, &spDevInfoData, SPDRP_HARDWAREID,
0L, (PBYTE)Buffer, , 0L))
{
if (!lstrcmpi(theHardware, (LPTSTR)Buffer))
{
//从系统中删除一个注册的设备接口
if (!SetupDiRemoveDevice(hDevInfo, &spDevInfoData))
printf("Remove Fail is %d!\n",GetLastError());
//if(!SetupDiCallClassInstaller(DIF_REMOVE,hDevInfo,&spDevInfoData))
// printf("Remove Fail is %d!\n",GetLastError());
wCount++;
}
}
}
else
break;
wIdx++;
} if (wCount != )
_tprintf(_T("UnInstall Successed...\n")); //getch(); //wprintf("%s\n",(wchar_t*)pSubStr);
if(SetupUninstallOEMInf(OemFileName1, SUOI_FORCEDELETE, NULL))
printf("Driver Package Has Deleted succced!\n"); //销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo); /* for(int i = 0; i <= Index; i++)
{
RegCloseKey(hKey[i]);
}*/ /*if(ERROR_SUCCESS==RegDeleteTree(hKey[0],NULL))
{
printf("Registry Has Delete Success!\n");
return ;
}*/ RegCloseKey(hKey[]); DeleteTree(HKEY_LOCAL_MACHINE,DeleteKeyPath); //printf("Registry Has Delete Su",GetLastError()); return ;
}
}
else
{
// ZeroMemory(buffer,sizeof(BYTE)*512);
//ZeroMemory(buffer2,sizeof(BYTE)*512);
//ZeroMemory(SubKeyName,SizeSubKeyName);
//ZeroMemory(KeyPath,sizeof(WCHAR)*512);
//KeyPath = NULL;
//continue; printf("Search Success!\n"); pSubWork = pSubStr; while((*pSubWork) != 'f')
{
pSubWork++;
} pSubWork++; *pSubWork = '\0'; WCHAR OemFileName2[]; wcscpy(OemFileName2,pSubStr); SP_DEVINFO_DATA spDevInfoData = {};
HDEVINFO hDevInfo = 0L;
WORD wIdx, wCount = ; //得到设备信息结构的句柄
hDevInfo = SetupDiGetClassDevs(0L, 0L, 0L, DIGCF_ALLCLASSES | DIGCF_PRESENT);
if (hDevInfo == INVALID_HANDLE_VALUE)
{
printf("Fail!\n");
return;
} wIdx = ;
while (TRUE)
{
spDevInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
//找到所有的硬件设备,并且可以得到所有的硬件设备的详细信息
if (SetupDiEnumDeviceInfo(hDevInfo, wIdx, &spDevInfoData))
{
char Buffer[] = {}; //可以在前面得到的指向某一个具体设备信息集合的指针中取出某一项信息
if (SetupDiGetDeviceRegistryProperty(hDevInfo, &spDevInfoData, SPDRP_HARDWAREID,
0L, (PBYTE)Buffer, , 0L))
{ printf("SetupDiGetDeviceRegistryProperty Success!\n"); if (!lstrcmpi(theHardware, (LPTSTR)Buffer))
{ //从系统中删除一个注册的设备接口
if (!SetupDiRemoveDevice(hDevInfo, &spDevInfoData))
printf("Remove Fail is %d!\n",GetLastError());
//if(!SetupDiCallClassInstaller(DIF_REMOVE,hDevInfo,&spDevInfoData))
// printf("Remove Fail is %d!\n",GetLastError()); wCount++;
} }
}
else
break;
wIdx++;
} if (wCount != )
_tprintf(_T("UnInstall Successed...\n")); // getch();
//wprintf("%s\n",(wchar_t*)pSubStr);
if(SetupUninstallOEMInf(OemFileName2, SUOI_FORCEDELETE, NULL))
printf("Driver Package Has Deleted succced!\n"); //销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo); /*for(int i = 0; i <= Index; i++)
{
RegCloseKey(hKey[i]);
}*/
//
/*if(ERROR_SUCCESS==RegDeleteTree(hKey[0],NULL))
{
printf("Registry Has Delete Success!\n");
return;
}*/ RegCloseKey(hKey[]); DeleteTree(HKEY_LOCAL_MACHINE,DeleteKeyPath); // printf("Registry Has Delete Fail! Error Code is %d\n",GetLastError()); return;
} } printf("Here is windows XP!\n"); //getch(); SP_DEVINFO_DATA spDevInfoData = {};
HDEVINFO hDevInfo = 0L;
WORD wIdx, wCount = ;
WCHAR oem_file_name[] = {};
//GUID usb_guid = {36fc9e60,c465,11cf,8056,444553540000};
//得到设备信息结构的句柄
hDevInfo = SetupDiGetClassDevs(NULL, NULL, NULL, DIGCF_ALLCLASSES | DIGCF_PRESENT); if (hDevInfo == INVALID_HANDLE_VALUE)
{
printf("Fail!\n");
printf("SetupDiGetClassDevs Error Code is %d\n",GetLastError());
return;
} wIdx = ; printf("hDevInfo is %x\n",hDevInfo); while (TRUE)
{
spDevInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
//找到所有的硬件设备,并且可以得到所有的硬件设备的详细信息
if (SetupDiEnumDeviceInfo(hDevInfo, wIdx, &spDevInfoData))
{ printf("SetupDiEnumDeviceInfo Success!\n"); char Buffer[] = {}; //可以在前面得到的指向某一个具体设备信息集合的指针中取出某一项信息
if (SetupDiGetDeviceRegistryProperty(hDevInfo, &spDevInfoData, SPDRP_HARDWAREID,
0L, (PBYTE)Buffer, , 0L))
{
printf("SetupDiGetDeviceRegistryProperty Success!\n"); wprintf(L"Buffer is %s\n",Buffer); if (!lstrcmpi(theHardware, (LPTSTR)Buffer))
{
printf("theHardware is matched!\n"); HKEY hDevKey;
BYTE buffer3[] = {};
DWORD buffer3_size = ;
//打开有关设备注册表
hDevKey = SetupDiOpenDevRegKey(hDevInfo,&spDevInfoData, DICS_FLAG_GLOBAL,,DIREG_DRV,KEY_ALL_ACCESS); if(hDevKey == INVALID_HANDLE_VALUE)
{
printf("SetupDiOpenRegKey Error Code is %d\n",GetLastError());
} if(ERROR_SUCCESS == RegQueryValueEx(hDevKey,L"InfPath",NULL,NULL,buffer3,&buffer3_size))
{
wprintf(L"Get Buffer3 Success! Buffer3 is %s\n",buffer3);
} wcscpy(oem_file_name,(wchar_t*)buffer3);
wprintf(oem_file_name);
printf("\n");
//getch();
RegCloseKey(hDevKey);
//从系统中删除一个注册的设备接口
/*if (!SetupDiRemoveDevice(hDevInfo, &spDevInfoData))
{
printf("Remove Fail is %d!\n",GetLastError());
}*/ /*if(!SetupDiCallClassInstaller(DIF_REMOVE,hDevInfo,&spDevInfoData))
printf("Remove Fail is %d!\n",GetLastError());*/
SP_REMOVEDEVICE_PARAMS rmdParams;
SP_DEVINSTALL_PARAMS devParams; rmdParams.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER);
rmdParams.ClassInstallHeader.InstallFunction = DIF_REMOVE;
rmdParams.Scope = DI_REMOVEDEVICE_GLOBAL;
rmdParams.HwProfile = ; if(SetupDiSetClassInstallParams(hDevInfo, &spDevInfoData, &rmdParams.ClassInstallHeader, sizeof(rmdParams)) &&
SetupDiCallClassInstaller(DIF_REMOVE, hDevInfo, &spDevInfoData))
{ }
wCount++;
} }
}
else
{ break;
} wIdx++;
} if (wCount != )
_tprintf(_T(" UnInstall Successed on windows XP...\n")); if(SetupUninstallOEMInf(oem_file_name, SUOI_FORCEDELETE, NULL))
printf("Driver Package Has Deleted succced! on windows XP\n"); //getch(); //销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo); //递归删除DeleteKeyPath下子键
/*DeleteTree(HKEY_LOCAL_MACHINE,DeleteKeyPath); HKEY hKey2;//删除DeleteKeyPath键本身 if(ERROR_SUCCESS != RegOpenKeyEx(HKEY_LOCAL_MACHINE,DeleteKeyPath,0,KEY_ALL_ACCESS,&hKey2))
{
printf("Can not Open Key!(Delete part!) Error Code is %d\n",GetLastError());
return ;
} RegDeleteKey(hKey2,DeleteKeyPath); RegCloseKey(hKey2);*/ //InitialGlobalVar();
return;
} BOOL UnicodeToAnsi(LPCWSTR Source, const WORD wLen, LPSTR Destination, const WORD sLen)
{
return WideCharToMultiByte(CP_ACP, WC_COMPOSITECHECK, Source, wLen, Destination,
sLen, 0L, 0L);
} BOOL AnsiToUnicode(LPCSTR Source, const WORD sLen, LPWSTR Destination, const WORD wLen)
{
return MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, Source, sLen, Destination, wLen);
} /*BOOL GetINFData(FILE *pFile)
{
WORD wLoop; if (!g_wVender || !g_wHardware)
InitialGlobalVar();
if (GetSectionData(pFile, "[Manufacturer]", TRUE) == FALSE)
return FALSE; //rewind(pFile); for (wLoop = 0; wLoop < g_wVender; wLoop++)
{
CHAR szVender[64] = {0};
UnicodeToAnsi(g_strVender[wLoop], _tcslen(g_strVender[wLoop]), szVender, 64);
GetSectionData(pFile, "Device.NTamd64", FALSE);
}
if (g_wHardware != 0)
{
if (IsInstalled() == TRUE)//如果已经安装
return FALSE;
else
return TRUE;
}
return FALSE;
} */ /*VOID InitialGlobalVar()
{
WORD wLoop; g_wVender = g_wHardware = 0;
for (wLoop = 0; wLoop < 20; wLoop++)
{
RtlZeroMemory(g_strVender[wLoop], sizeof(TCHAR)*64);
RtlZeroMemory(g_strHardware[wLoop], sizeof(TCHAR)*64);
}
} */ VOID FindComma(LPSTR szData)
{
WORD wLen = (WORD)strlen(szData);
WORD wIdx;
WORD wLoop;
CHAR szTmp[] = {}; for (wIdx = , wLoop = ; wLoop < wLen; wLoop++)
{
if (szData[wLoop] == ',')
szData[wLoop] = '.';
else if (szData[wLoop] == ' ')
continue;
szTmp[wIdx++] = szData[wLoop];
}
memcpy(szData, szTmp, wIdx*sizeof(char));
szData[wIdx] = ;
} VOID StrLTrim(LPSTR szData)
{
LPSTR ptr = szData;
//判断是否为空格
while (isspace(*ptr))
ptr++; if (strcmp(ptr, szData))
{
WORD wLen = (WORD)(strlen(szData) - (ptr - szData));
memmove(szData, ptr, (wLen+)*sizeof(char));
}
} VOID StrRTrim(LPSTR szData)
{
LPSTR ptr = szData;
LPSTR pTmp = NULL; //debug模式下 使用isspace判断中文 需要设置编码
#if defined(WIN32) && defined(_DEBUG)
char* locale = setlocale( LC_ALL, ".OCP" );
#endif while (*ptr != )
{
//判断是否为空格
if (isspace(*ptr))
{
if (!pTmp)
pTmp = ptr;
}
else
pTmp = NULL;
ptr++;
} if (pTmp)
{
*pTmp = ;
memmove(szData, szData, strlen(szData) - strlen(pTmp));
}
} //从字符串右边开始截取字符串
VOID StrRight(LPSTR szData, WORD wCount)
{
WORD wLen = (WORD)strlen(szData) - wCount; if (wCount > 0x7FFF)//负数
wCount = ;
if (wCount >= (WORD)strlen(szData))
return; memmove(szData, szData + wLen, wCount * sizeof(char));
szData[wCount] = ;
} VOID ConvertGUIDToString(const GUID guid, LPSTR pData)
{
CHAR szData[] = {};
CHAR szTmp[] = {};
WORD wLoop; sprintf_s(pData, _countof(szData), "%04X-%02X-%02X-", guid.Data1, guid.Data2, guid.Data3);
for (wLoop = ; wLoop < ; wLoop++)
{
if (wLoop == )
strcat_s(szData, "-");
sprintf_s(szTmp, _countof(szTmp), "%02X", guid.Data4[wLoop]);
strcat_s(szData, szTmp);
} memcpy(pData + strlen(pData), szData, strlen(szData));
} /*BOOL IsInstalled()
{
HDEVINFO hDevInfo = 0L;
SP_DEVINFO_DATA spDevInfoData = {0L};
WORD wIdx;
BOOL bIsFound; //得到设备信息结构的句柄
hDevInfo = SetupDiGetClassDevs(0L, 0, 0, DIGCF_ALLCLASSES | DIGCF_PRESENT);
if (hDevInfo == INVALID_HANDLE_VALUE)
{
printf("SetupDiGetClassDevs is %d",GetLastError());
return FALSE;
} spDevInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
wIdx = 0;
bIsFound = 0;
while (++wIdx)
{
//找到所有的硬件设备,并且可以得到所有的硬件设备的详细信息
if (SetupDiEnumDeviceInfo(hDevInfo, wIdx, &spDevInfoData))
{
LPTSTR ptr;
LPBYTE pBuffer = NULL;
DWORD dwData = 0L;
DWORD dwRetVal;
DWORD dwBufSize = 0L; while (TRUE)
{
//可以在前面得到的指向某一个具体设备信息集合的指针中取出某一项信息
dwRetVal = SetupDiGetDeviceRegistryProperty(hDevInfo, &spDevInfoData, SPDRP_HARDWAREID,
&dwData, (PBYTE)pBuffer, dwBufSize, &dwBufSize);
if (!dwRetVal)
dwRetVal = GetLastError();
else
break;
if (dwRetVal == ERROR_INVALID_DATA)
break;
else if (dwRetVal == ERROR_INSUFFICIENT_BUFFER)
{
if (pBuffer)
LocalFree(pBuffer);
pBuffer = (LPBYTE)LocalAlloc(LPTR, dwBufSize);
}
else
{
printf("SetupDiGetDeviceRegistryProperty is %d",dwRetVal);
//销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo);
return FALSE;
}
} if (dwRetVal == ERROR_INVALID_DATA)
continue; for (ptr = (LPTSTR)pBuffer; *ptr && (ptr < (LPTSTR)&pBuffer[dwBufSize]); ptr += _tcslen(ptr) + sizeof(TCHAR))
{
WORD wLoop; for (wLoop = 0; wLoop < g_wHardware; wLoop++)
{
if (!_tcscmp(g_strHardware[wLoop], ptr))
{
bIsFound = TRUE;
break;
}
}
}
if (pBuffer)
LocalFree(pBuffer);
if (bIsFound)
break;
}
}
//销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo);
return bIsFound;
} */ //寻找指定的节名 如果找到返回TRUE 反之返回FALSE
BOOL FindSectionName(FILE *pFile, const char *szKey)
{
char szData[] = {}; if (!pFile)
return FALSE; //将文件内部的位置指针重新指向一个流（数据流/文件）的开头
rewind(pFile);
//循环读取文件内容
while (!feof(pFile))
{
//读取一行
fgets(szData, , pFile);
//去除前后空格
StrLTrim(szData);
StrRTrim(szData); if (strcmp(szKey, szData) == )
return TRUE;
}
return FALSE;
} //得到INF文件中节的数量
/*BOOL GetSectionData(FILE* pFile, const char* szKey, const char bIsVender)
{
char szData[128] = {0}; if (bIsVender)
strcpy_s(szData, szKey);
else
sprintf_s(szData, _countof(szData), "[%s]", szKey); if (FindSectionName(pFile, szData) == FALSE)
return FALSE; RtlZeroMemory(szData, sizeof(char)*128);
while (!feof(pFile))
{
char *str = NULL;
fgets(szData, 127, pFile);
szData[strlen(szData)-1] = 0;
StrLTrim(szData);
StrRTrim(szData);
if (!*szData)
continue;
if (szData[0] == ';')
continue; if (strchr(szData, '['))
{
StrLTrim(szData);
if (szData[0] != ';')
return 1;
else
continue;
} if (bIsVender)
str = strchr(szData, '=');
else
str = strchr(szData, ','); if (*str)
{
char szTmp[128] = {0};
WORD pos = (WORD)(str - szData + 1); StrRight(szData, (short)(strlen(szData)-pos));
StrLTrim(szData);
StrRTrim(szData);
FindComma(szData);
if (bIsVender)
{
AnsiToUnicode(szData, strlen(szData), g_strVender[g_wVender++], 64);
}
else
{
AnsiToUnicode(szData, strlen(szData), g_strHardware[g_wHardware++], 64);
}
} //end if
}
return TRUE;
} */ //实质性的安装驱动
BOOL InstallClassDriver(LPCTSTR theINFName)
{
GUID guid = {};
SP_DEVINFO_DATA spDevData = {};
HDEVINFO hDevInfo = 0L;
TCHAR className[MAX_CLASS_NAME_LEN] = {};
LPTSTR pHID = NULL;
WORD wLoop;
BOOL bRebootRequired; //取得此驱动的GUID值 ,className也是输出参数
if (!SetupDiGetINFClass(theINFName, &guid, className, MAX_CLASS_NAME_LEN, ))
{
printf( "SetupDiGetINFClass is %d\n",GetLastError());
return FALSE;
} //创建设备信息块列表
hDevInfo = SetupDiCreateDeviceInfoList(&guid, );
if (hDevInfo == INVALID_HANDLE_VALUE)
{
printf("SetupDiCreateDeviceInfoList is %d\n",GetLastError());
return FALSE;
} spDevData.cbSize = sizeof(SP_DEVINFO_DATA);
//创建设备信息块
if (!SetupDiCreateDeviceInfo(hDevInfo, className, &guid, 0L, 0L, DICD_GENERATE_ID, &spDevData))
{
printf("SetupDiCreateDeviceInfo is %d",GetLastError());
//销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo);
return FALSE;
} // for (wLoop = 0; wLoop < g_wHardware; wLoop++)
//{
if (pHID)
LocalFree(pHID); pHID = (LPTSTR)LocalAlloc(LPTR, _tcslen(L"USB\\VID_0451&PID_AF32")**sizeof(TCHAR));
if (!pHID)
{
printf("LocalAlloc is %d",GetLastError());
//销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo);
return FALSE;
} _tcscpy_s(pHID, _tcslen(L"USB\\VID_0451&PID_AF32")*,
L"USB\\VID_0451&PID_AF32");
//设定硬件ID
if (!SetupDiSetDeviceRegistryProperty(hDevInfo, &spDevData, SPDRP_HARDWAREID, (PBYTE)pHID,
(DWORD)(_tcslen(L"USB\\VID_0451&PID_AF32")**sizeof(TCHAR))))
{
printf("SetupDiSetDeviceRegistryProperty is %d",GetLastError());
//销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo);
LocalFree(pHID);
return FALSE;
}
//调用相应的类程序来注册设备
if (!SetupDiCallClassInstaller(DIF_REGISTERDEVICE, hDevInfo, &spDevData))
{
printf("SetupDiCallClassInstaller is %d", GetLastError());
//销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo);
LocalFree(pHID);
return FALSE;
} bRebootRequired = FALSE;
//安装更新和硬件ID相匹配的驱动程序
if (!UpdateDriverForPlugAndPlayDevices(0L, L"USB\\VID_0451&PID_AF32"
, theINFName,
INSTALLFLAG_FORCE, &bRebootRequired))
{
DWORD dwErrorCode = GetLastError();
//调用相应的类程序来移除设备
if (!SetupDiCallClassInstaller(DIF_REMOVE, hDevInfo, &spDevData))
printf("SetupDiCallClassInstaller(Remove) is %d",GetLastError());
printf("UpdateDriverForPlugAndPlayDevices is %d",(WORD)dwErrorCode);
//销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo);
LocalFree(pHID); //getch();
return FALSE;
}
LocalFree(pHID);
pHID = NULL; // }
//销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo);
_tprintf(_T("Install Successed\n"));
return TRUE;
} // 安装WDM驱动的测试工作
BOOL StartInstallWDMDriver(LPCTSTR theInfName)
{
HDEVINFO hDevInfo = 0L;
GUID guid = {0L};
SP_DEVINSTALL_PARAMS spDevInst = {0L};
TCHAR strClass[MAX_CLASS_NAME_LEN] = {0L}; //取得此驱动的GUID值
if (!SetupDiGetINFClass(theInfName, &guid, strClass, MAX_CLASS_NAME_LEN, ))
{
printf("SetupDiGetINFClass is %d",GetLastError());
return FALSE;
} //得到设备信息结构的句柄
hDevInfo = SetupDiGetClassDevs(&guid, 0L, 0L, DIGCF_PRESENT | DIGCF_ALLCLASSES | DIGCF_PROFILE);
if (!hDevInfo)
{
printf("SetupDiGetClassDevs is %d",GetLastError());
return FALSE;
} spDevInst.cbSize = sizeof(SP_DEVINSTALL_PARAMS);
//获得指定设备的安装信息
if (!SetupDiGetDeviceInstallParams(hDevInfo, 0L, &spDevInst))
{
printf("SetupDiGetDeviceInstallParams is %d",GetLastError());
return FALSE;
} spDevInst.Flags = DI_ENUMSINGLEINF;
spDevInst.FlagsEx = DI_FLAGSEX_ALLOWEXCLUDEDDRVS;
_tcscpy_s(spDevInst.DriverPath, _countof(spDevInst.DriverPath), theInfName); //为设备信息集或者是一个实际的设备信息单元设置或清除类安装参数
if (!SetupDiSetDeviceInstallParams(hDevInfo, , &spDevInst))
{ printf("SetupDiSetDeviceInstallParams is %d",GetLastError());
return FALSE;
} //获取这个设备的驱动程序信息列表
if (!SetupDiBuildDriverInfoList(hDevInfo, , SPDIT_CLASSDRIVER))
{ printf("SetupDiDeviceInstallParams is %d",GetLastError());
return FALSE;
} //销毁一个设备信息集合
SetupDiDestroyDeviceInfoList(hDevInfo); //进入安装设备驱动函数
return InstallClassDriver(theInfName);
} BOOL FindExistingDevice(IN LPTSTR HardwareId)
{
HDEVINFO DeviceInfoSet;
SP_DEVINFO_DATA DeviceInfoData;
DWORD i,err;
BOOL Found; //
// Create a Device Information Set with all present devices.
//
DeviceInfoSet = SetupDiGetClassDevs(NULL, // All Classes
,
,
DIGCF_ALLCLASSES | DIGCF_PRESENT ); // All devices present on system if (DeviceInfoSet == INVALID_HANDLE_VALUE)
{
return printf("GetClassDevs(All Present Devices)"); } //_tprintf(TEXT("Search for Device ID: [%s]/n"),HardwareId); //
// Enumerate through all Devices.
//
Found = FALSE;
DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
for (i=;SetupDiEnumDeviceInfo(DeviceInfoSet,i,&DeviceInfoData);i++)
{
DWORD DataT;
LPTSTR p,buffer = NULL;
DWORD buffersize = ; //
// We won't know the size of the HardwareID buffer until we call
// this function. So call it with a null to begin with, and then
// use the required buffer size to Alloc the nessicary space.
// Keep calling we have success or an unknown failure.
//
while (!SetupDiGetDeviceRegistryProperty(
DeviceInfoSet,
&DeviceInfoData,
SPDRP_HARDWAREID,
&DataT,
(PBYTE)buffer,
buffersize,
&buffersize))
{
if (GetLastError() == ERROR_INVALID_DATA)
{
//
// May be a Legacy Device with no HardwareID. Continue.
//
break;
}
else if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
{
//
// We need to change the buffer size.
//
if (buffer)
LocalFree(buffer);
buffer = (LPTSTR)LocalAlloc(LPTR,buffersize);
}
else
{
//
// Unknown Failure.
//
printf("GetDeviceRegistryProperty");
goto cleanup_DeviceInfo;
}
} if (GetLastError() == ERROR_INVALID_DATA)
continue; //
// Compare each entry in the buffer multi-sz list with our HardwareID. //
for (p=buffer;*p&&(p<&buffer[buffersize]);p+=lstrlen(p)+sizeof(TCHAR)) {
//_tprintf(TEXT("Compare device ID: [%s]/n"),p); if (!_tcscmp(HardwareId,p))
{
//_tprintf(TEXT("Found! [%s]/n"),p);
Found = TRUE;
break;
}
} if (buffer) LocalFree(buffer);
if (Found) break;
} if (GetLastError() != NO_ERROR)
{
printf("EnumDeviceInfo");
} //
// Cleanup.
//
cleanup_DeviceInfo:
err = GetLastError();
SetupDiDestroyDeviceInfoList(DeviceInfoSet);
SetLastError(err); return err == NO_ERROR; //???
} BOOL ConfigRight(WCHAR* szPath)
{
if (szPath == NULL)
{
return FALSE;
}
/*BOOL bRet = TRUE; SE_OBJECT_TYPE objectType;
PACL newDACL;
objectType = SE_REGISTRY_KEY; if (SetEntriesInAclW(0, NULL, NULL, &newDACL) != ERROR_SUCCESS)
{
return FALSE;
} DWORD dwRet = 0; EXPLICIT_ACCESSW ea;
memset(&ea, 0, sizeof(EXPLICIT_ACCESS)); BuildExplicitAccessWithNameW(&ea,L"Everyone", KEY_ALL_ACCESS, SET_ACCESS, SUB_CONTAINERS_AND_OBJECTS_INHERIT); if (SetEntriesInAclW(1, &ea, NULL, &newDACL) != ERROR_SUCCESS)
{
bRet = FALSE;
goto Clean_Up;
}
DWORD ret; //调用失败，访问被拒绝
if (ERROR_SUCCESS!= (ret = SetNamedSecurityInfoW(szPath, objectType, DACL_SECURITY_INFORMATION, NULL, NULL, newDACL, NULL)))
{
bRet = FALSE;
printf("ret value is %d\n",ret);
goto Clean_Up;
} Clean_Up:
if (newDACL != NULL)
{
LocalFree((HLOCAL)newDACL);
}*/
PACL pOldDacl=NULL;
PACL pNewDacl=NULL;
DWORD dRet;
EXPLICIT_ACCESS eia;
PSECURITY_DESCRIPTOR pSID=NULL; dRet = GetNamedSecurityInfo(szPath,SE_REGISTRY_KEY,DACL_SECURITY_INFORMATION,NULL,NULL,&pOldDacl,NULL,&pSID);// 获取SAM主键的DACL if(dRet!=ERROR_SUCCESS)
{
return ;
}
//创建一个ACE,允许Everyone组成员完全控制对象,并允许子对象继承此权限
ZeroMemory(&eia,sizeof(EXPLICIT_ACCESS));
BuildExplicitAccessWithName(&eia,L"SYSTEM",KEY_ALL_ACCESS,SET_ACCESS,SUB_CONTAINERS_AND_OBJECTS_INHERIT);
// 将新的ACE加入DACL
dRet = SetEntriesInAcl(,&eia,pOldDacl,&pNewDacl);
if(dRet!=ERROR_SUCCESS)
{
return ;
}
// 更新SAM主键的DACL
dRet = SetNamedSecurityInfo(szPath,SE_REGISTRY_KEY,DACL_SECURITY_INFORMATION,NULL,NULL,pNewDacl,NULL); if(dRet!=ERROR_SUCCESS)
{
printf("%d\n",dRet);
return ;
}
//return bRet;
} DWORD DeleteTree(HKEY RootKey, const WCHAR *pSubKey)
{ HKEY hKey;
DWORD nRet;
DWORD NameCnt,NameMaxLen;
DWORD KeyCnt,KeyMaxLen,MaxDateLen;
//static WCHAR sFormat[256] = L"";
//wcscat(sFormat, L"----"); static int dwDeep = -;
dwDeep++; nRet=RegOpenKeyEx(RootKey,pSubKey,,KEY_ALL_ACCESS,&hKey);
if(nRet!=ERROR_SUCCESS)
{
printf("Can not Open Key!(Delete part!) Error Code is %d\n",GetLastError());
return ;
} nRet = RegQueryInfoKey(hKey,NULL,NULL,NULL,&KeyCnt,&KeyMaxLen,NULL,&NameCnt,
&NameMaxLen,&MaxDateLen,NULL,NULL); if(nRet == ERROR_SUCCESS)
{
printf("RegQueryInfoKey Success!\n"); for(int dwIndex = KeyCnt - ; dwIndex >= ; dwIndex--) //枚举键值
{
WCHAR sKeyName[] = {};
RegEnumKey(hKey, dwIndex, sKeyName, sizeof(sKeyName)); HKEY hKeySub;
DWORD KeyCntSub;
WCHAR pSubKeyTemp[] = {};
wcscpy(pSubKeyTemp, pSubKey);
wcscat(pSubKeyTemp, L"\\");
wcscat(pSubKeyTemp, sKeyName); nRet = RegOpenKeyEx(RootKey,pSubKeyTemp,,KEY_ALL_ACCESS,&hKeySub); if(nRet == ERROR_SUCCESS)
{
printf("RegOpenKeyEx Success!\n"); nRet = RegQueryInfoKey(hKeySub,NULL,NULL,NULL,&KeyCntSub,&KeyMaxLen,NULL,&NameCnt,
&NameMaxLen,&MaxDateLen,NULL,NULL); if(nRet == ERROR_SUCCESS)
{
if (KeyCntSub != )
{
DeleteTree(RootKey, pSubKeyTemp);
} RegCloseKey(hKeySub);
}
} // cout << sFormat << sKeyName << endl; RegDeleteKey(RootKey ,pSubKeyTemp);
} RegCloseKey(hKey);
} // sFormat[strlen(sFormat) - 4] = 0; if (dwDeep == )
{
RegDeleteKey(RootKey ,pSubKey);
} return ;
} VOID SearchRegistryHardID(PWCHAR pHardIDBuffer)
{
const PWCHAR pReg = L"SYSTEM\\CurrentControlSet\\Enum\\USB";
HKEY hKey; if(ERROR_SUCCESS !=RegOpenKeyEx(HKEY_LOCAL_MACHINE,pReg,,KEY_ALL_ACCESS,&hKey))
printf("in SearchOpenKey Fail! error code is %d\n",GetLastError()); DWORD Index = ;
WCHAR SubKeyName[] = {}; while(ERROR_SUCCESS == RegEnumKey(hKey,Index,SubKeyName,sizeof(WCHAR)*))
{
if(!lstrcmpi(SubKeyName,L"VID_0451&PID_AF32"))
{
wcscpy(pHardIDBuffer,L"VID_0451&PID_AF32");
RegCloseKey(hKey);
return;
} else if(!lstrcmpi(SubKeyName,L"VID_04B41&PID_8613"))
{
wcscpy(pHardIDBuffer,L"VID_04B41&PID_8613");
RegCloseKey(hKey);
return;
} else if(!lstrcmpi(SubKeyName,L"VID_0547&PID_1002"))
{
wcscpy(pHardIDBuffer,L"VID_0547&PID_1002");
RegCloseKey(hKey);
return;
} ZeroMemory(SubKeyName,sizeof(WCHAR)*); Index++;
} }
Setup.cpp
// InstallWDFDriver.cpp : Defines the entry point for the console application.
// #include "stdafx.h"
#include "setup.h"
#include "Shlwapi.h" #pragma comment(lib,"Shlwapi.lib") BOOL SystemShutdown(); int _tmain(int argc, _TCHAR* argv[])
{
WCHAR Wlp_USB_PATH[] = L"C:\\Windows\\System32\\drivers\\WLP_USB_Driver.sys";
//WCHAR Wdf_USB_PATH[] = L"C:\\Windows\\System32\\drivers\\CYUSB3.sys";
WCHAR New_Inf_Path[] = L"D:\\wlp driver\\WDF Driver _new\\win7\\x64\\cyusb3.inf";
WCHAR InfFileName[] = L"\\cyusb3.inf";
CHAR szInfPath[] = {};
FILE *fp = NULL; WCHAR CurrentDirBuffer[] = {};
WCHAR OutCurrentDirBuffer[] = {}; GetModuleFileName(NULL,CurrentDirBuffer,_countof(CurrentDirBuffer));
//GetCurrentDirectory(215,CurrentDirBuffer);
//lstrcat(CurrentDirBuffer, InfFileName);
_wsplitpath_s(CurrentDirBuffer,NULL,,OutCurrentDirBuffer,_countof(OutCurrentDirBuffer),
NULL,,NULL,);
printf("This Process is Driver Installtion Program...\n"); //getch();
//WCHAR HIDBuffer[128] = {0}; if(PathFileExists(Wlp_USB_PATH))
{
printf("Exists WLP_USB_Driver.sys!\n"); //SearchRegistryHardID(HIDBuffer); //WCHAR HardID[215] = L"USB\\"; //wcscat(HardID,HIDBuffer);
//getch(); UninstallWDMDriver(L"USB\\VID_0451&PID_AF32&REV_0000"); if(DeleteFile(Wlp_USB_PATH))
{
printf("WLP_USB_Driver.sys File has Deleted!\n");
} // getch(); /* if(FindExistingDevice(L"USB\\VID_0451&PID_AF32&REV_0000"))
{
printf("找打了！\n"); getch();
}
else
{
printf("没找到！\n");
getch();
}*/ printf("Get Started Install the New DLF Driver...\n"); // RemoveDriver(L"USB\\VID_0451&PID_AF32&REV_0000");
while(IsDeviceInstallInProgress())
{
printf("Has Driver Installtion Program is Processing!\n");
} printf("Get Started analyse inf File!\n"); UnicodeToAnsi(New_Inf_Path, _tcslen(New_Inf_Path), szInfPath, ); FindAllInfFiles(OutCurrentDirBuffer/* L"D:\\wlp driver\\WDF Driver _new\\win7\\x64"*/); for(int i = ; i < g_nHIDCount; i++)
g_HIDInfo[i]->bNeedUpdate = false; for(int i = ; i < g_nInfCount; i++)
g_InfList[i]->bWillBeInstall = false; int SearchInfIndex = ; while(SearchInfIndex < g_nInfCount)
{
if(g_InfList[SearchInfIndex] != NULL)
{
g_InfList[SearchInfIndex]->bWillBeInstall = TRUE;
} SearchInfIndex++;
} InstallDriver(); //UpdateDriver();
printf("Install Successed!\n");
//if ((fopen_s(&fp, szInfPath, "r"))!=0)
//{
// _tprintf(_T("can not open file %s\n"), CurrentDirBuffer);
//return 0;
//} // GetINFData(fp);
// fclose(fp); // 安装WDM驱动 /*if (StartInstallWDMDriver(New_Inf_Path) == FALSE)
{
_tprintf(_T("Start Install WDF Driver failed\n"));
getch();
return 0;
} */ /*if(IDOK == MessageBox(NULL,L"Need to Restart the System to Complete Installation!",
L"Tip",MB_OK))
{
if(SystemShutdown())
{
printf("ShutDown the System And Restart the System!\n");
} }*/ //getch(); printf("Press any key to continue!\n"); }
else
{
printf("No Exists WLP_USB_Driver.sys!\n"); //getch(); UninstallWDMDriver(L"USB\\VID_0451&PID_AF32&REV_0000"); printf("Get Started Install the New DLF Driver...\n"); printf("Get Started analyse inf File!\n"); UnicodeToAnsi(New_Inf_Path, _tcslen(New_Inf_Path), szInfPath, ); /*if ((fopen_s(&fp, szInfPath, "r"))!=0)
{
_tprintf(_T("can not open file %s\n"), CurrentDirBuffer);
return 0;
} */ // GetINFData(fp);
//fclose(fp); // 安装WDM驱动 /*if (StartInstallWDMDriver(New_Inf_Path) == FALSE)
{
_tprintf(_T("Start Install WDF Driver failed\n"));
getch();
return 0;
} */ FindAllInfFiles( OutCurrentDirBuffer/*L"D:\\wlp driver\\WDF Driver _new\\win7\\x64"*/); for(int i = ; i < g_nHIDCount; i++)
g_HIDInfo[i]->bNeedUpdate = false; for(int i = ; i < g_nInfCount; i++)
g_InfList[i]->bWillBeInstall = false; int SearchInfIndex = ; while(SearchInfIndex < g_nInfCount)
{
if(g_InfList[SearchInfIndex] != NULL)
{
g_InfList[SearchInfIndex]->bWillBeInstall = TRUE;
} SearchInfIndex++;
} InstallDriver(); printf("Install Successed!\n"); /*if(IDOK == MessageBox(NULL,L"Need to Restart the System to Complete Installation!",
L"Tip",MB_OK))
{
if(SystemShutdown())
{
printf("ShutDown the System And Restart the System!\n");
} }*/ //getch(); printf("Press any key to continue!\n"); } return ;
} BOOL SystemShutdown()
{
HANDLE hToken; TOKEN_PRIVILEGES tkp;
/* OpenProcessToken() 这个函数的作用是打开一个进程的访问令牌
GetCurrentProcess() 函数的作用是得到本进程的句柄*/
if (!OpenProcessToken(GetCurrentProcess(),
TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,&hToken))
return FALSE;
// LookupPrivilegeValue() 的作用是修改进程的权限
LookupPrivilegeValue(NULL,SE_SHUTDOWN_NAME,
&tkp.Privileges[].Luid);
tkp.PrivilegeCount = ; // one privilege to set 赋给本进程特权
tkp.Privileges[].Attributes = SE_PRIVILEGE_ENABLED;
// AdjustTokenPrivileges（）的作用是通知Windows NT修改本进程的权利
AdjustTokenPrivileges(hToken, FALSE, &tkp, ,
(PTOKEN_PRIVILEGES)NULL, ); if (GetLastError() != ERROR_SUCCESS) //失败
return FALSE; if (!ExitWindowsEx(EWX_REBOOT, )) //参数在这里设置。强行退出WINDOWS（EWX_FORCE）。
return FALSE;
return TRUE;
}
main.cpp
附加产物：
发现MSDN的代码样例中由实现设备管理器功能的命令行工具
https://code.msdn.microsoft.com/DevCon-Sample-4e95d71c#content
最新文章
- 锐捷linux客户端常用命令（主要用来连接校园网或公司局域网）
- 当 IDENTITY_INSERT 设置为 OFF 时，不能向表 'OrderList' 中的标识列插入显式值
- 【XLL API 函数】xlGetBinaryName
- FreeCodeCamp 中级算法（个人向）
- c++继承中的内存布局
- 从前有个聊天室（socket&threading）
- VRTK实现瞬移需要添加的脚本
- Mysql介绍和实践总结
- Mac Os系统设置
- python中type dtype astype 的用法
- python持久化
- gis 相关资料
- Linux基础命令---yes
- android开发 gradle 总结
- spring session实现集群中session共享
- Django-DRF-视图的演变
- Vue实例对象的数据选项（火柴）
- 文件复制（shutil）
- 在Eclipse中开发WEB项目
- RGB555转RGB565
热门文章
- HDOJ 1061 Rightmost Digit（循环问题）
- Android中使用shape制作一个旋转的progressbar
- FZU 11月月赛D题：双向搜索+二分
- 大型分布式C++框架《二：大包处理过程》
- zookeeper[1] （转）ZooKeeper Programmer's Guide（zookeeper编程向导）---中文
- Gson 基础教程 —— 自定义类型适配器(TypeAdapter)
- struts2.1.3之后使用自定义Filter
- 【BUG】Kewastunpackstats(): Bad Magic 1 (0x。。。。, 0)
- android requestDisallowInterceptTouchEvent用途
- 关于在xp（sp3 专业版）下安装sql2005开发版图解