WinPE 驱动程序最终解决方案 - 无忧启动论坛

// Header
#include <SetupAPI.h>
#include <DevGuid.h>
#include <RegStr.h>
#include <CfgMgr32.h>
#pragma comment(lib, "SetupAPI.lib")
#ifdef _UNICODE
#define STR_UpdateDriverForPlugAndPlayDevices "UpdateDriverForPlugAndPlayDevicesW"
#else
#define STR_UpdateDriverForPlugAndPlayDevices "UpdateDriverForPlugAndPlayDevicesA"
#endif
typedef BOOL (WINAPI* PUPNP)(HWND hWnd, PCTSTR ptzID, PCTSTR ptzPath, DWORD dwFlags, PBOOL bReboot);
// Compare device
PSTR DevCmp(PSTR pszStr1, PSTR pszStr2)
{
PSTR p = pszStr2;
for (UINT i = 0; *p; p++)
{
if ((*p == '&') && (++i == 2))
{
break;
}
}
UINT uCmp = (UINT) (p - pszStr2);
p = pszStr1;
while (*p)
{
if (UMemCmp(p, pszStr2, uCmp) == 0)
{
return p;
}
p++;
}
return NULL;
}
// CAB callback
UINT CALLBACK DevProc(PVOID pvContext, UINT uMsg, UINT_PTR upParam1, UINT_PTR upParam2)
{
static BOOL s_bExtract = FALSE;
static TCHAR s_tzDevID[MAX_NAME] = {0};
static TCHAR s_tzInfPath[MAX_PATH] = {0};
if (uMsg == SPFILENOTIFY_FILEINCABINET)
{
// Is INF file or followed by INF file
FILE_IN_CABINET_INFO *p = (FILE_IN_CABINET_INFO*) upParam1;
PCTSTR q = UStrRChr(p->NameInCabinet, '\\');
q = q ? (q + 1) : p->NameInCabinet;
if ((*q == '_') || s_bExtract)
{
// Extract INF or driver file
if (*q == '_')
{
q++;
}
UStrCopy(p->FullTargetName + GetTempPath(MAX_PATH, p->FullTargetName), q);
return FILEOP_DOIT;
}
else
{
// Skip
return FILEOP_SKIP;
}
}
else if (uMsg == SPFILENOTIFY_FILEEXTRACTED)
{
PCTSTR ptzTarget = ((FILEPATHS*) upParam1)->Target;
if (UStrStr(ptzTarget, TEXT(".INF")) == NULL)
{
// Not INF
return NO_ERROR;
}
// Open INF
PSTR pszFile = (PSTR) UFileLoad(ptzTarget, NULL);
if (pszFile == NULL)
{
return NO_ERROR;
}
for (PSTR p = (PSTR) pvContext; *p; p += UAStrLen(p) + 1)
{
if ((*p != '#') && DevCmp(pszFile, p))
{
// Found Device
UAStrToStr(s_tzDevID, p, MAX_NAME);
UStrCopy(s_tzInfPath, ptzTarget);
//UTrack(TEXT("\r\nFound %s: %s\r\n\r\n"), s_tzDevID, s_tzInfPath);
s_bExtract = TRUE;
*p = '#';
return NO_ERROR;
}
}
// This is next INF
if (s_bExtract)
{
// Previous is uses INF
s_bExtract = FALSE;
HMODULE hLib = LoadLibrary(TEXT("NewDev"));
if (hLib)
{
// Install INF
PUPNP p = (PUPNP) GetProcAddress(hLib, STR_UpdateDriverForPlugAndPlayDevices);
if (p)
{
BOOL bReboot = FALSE;
p(NULL, s_tzDevID, s_tzInfPath, 3, &bReboot);
UTrack(TEXT("\r\nInstall %s: %s\r\n\r\n"), s_tzDevID, s_tzInfPath);
}
FreeLibrary(hLib);
}
}
// Delete INF
UFileDelete(ptzTarget);
}
return NO_ERROR;
}
#define MAX_DevID (MAX_STR * 4)
UINT DevEnum(PSTR pszDev, PCTSTR ptzClass = REGSTR_KEY_PCIENUM)
{
// Lookup device
HDEVINFO hDev = SetupDiGetClassDevs(NULL, ptzClass, 0, DIGCF_PRESENT | DIGCF_ALLCLASSES );
if (hDev == INVALID_HANDLE_VALUE)
{
return E_FAIL;
}
// Lookup Device ID
UINT uSize = 0;
SP_DEVINFO_DATA sdDev = {sizeof(SP_DEVINFO_DATA)};
for (UINT i = 0; (uSize < MAX_DevID) && SetupDiEnumDeviceInfo(hDev, i, &sdDev); i++)
{
DWORD dwData;
DWORD dwSize = MAX_DevID- uSize;
if (SetupDiGetDeviceRegistryPropertyA(hDev, &sdDev, SPDRP_HARDWAREID, &dwData, (PBYTE) pszDev + uSize, dwSize, &dwSize))
{
#ifndef _DEBUG
// List problem device only
ULONG uProblem = 0;
ULONG uStatus = DN_HAS_PROBLEM;
CM_Get_DevNode_Status(&uStatus, &uProblem, sdDev.DevInst, 0);
if (uProblem)
#endif
{
uSize += UAStrLen(pszDev + uSize) + 1;
}
}
}
SetupDiDestroyDeviceInfoList(hDev);
pszDev[uSize] = pszDev[uSize + 1] = 0;
return uSize;
}
// Install device
HRESULT Devi(PTSTR ptzCmd)
{
// Skip if Control Key pressed.
if (GetAsyncKeyState(VK_CONTROL) & 0x8000)
{
return S_FALSE;
}
// Enum device ID
CHAR szDev[MAX_DevID];
UINT uSize = DevEnum(szDev, TEXT("PCI"));
uSize += DevEnum(szDev + uSize, TEXT("USB"));
if (uSize == 0)
{
// No device
return ERROR_DEVICE_NOT_CONNECTED;
}
#ifdef _DEBUG
for (PSTR p = szDev; *p; p += UAStrLen(p) + 1)
{
UTrack(TEXT("%hs\r\n"), p);
}
#endif
// Lookup CAB file
TCHAR tzPath[MAX_PATH];
ExpandEnvironmentStrings(ptzCmd, tzPath, MAX_PATH);
SetupIterateCabinet(tzPath, 0, DevProc, szDev);
return S_OK;
}

复制代码

欢迎使用 RXPE - Rescue Windows XP Environment 扩展设置工具！

一、简介

RXPE 是为扩展设置 Windows PE 而制作的小工具。

◎ 支持常规的扩展设置命令。
◎ 支持外部命令增加程序的功能。
◎ 支持命令行、命令列表文件和图形界面，所有命令均使用统一格式，一看就会使用。
◎ 细微周到的功能支持。
◎ 短小精干，运行迅速。
◎ 源代码公开，方便更改定制。在源代码中增加扩展命令非常容易。

二、命令

■ Regi <HKLM|HKCU|HKCR|HKU|HKCC><\子项\>[键值][=[#]数据]

  ◎ 功能：设置或删除注册表数据。

  ◎ 参数：
※ 子项             所选 ROOTKEY 下注册表项的完整名。

※ 键值             要操作的键值。如果省略，则操作默认键值；如果为“!”且没有“=”，则用于删除整个子项。

※ 数据             要设置的数据。如果没有“=”则删除；如果有“#”则表示 REG_DWORD 整数类型的数据，否则为字符串。

  ◎ 示例：Regi HKCU\SOFTWRE\RXPE\Version=#1200

  ◎ 备注：无。

■ Envi [名称][=值]

  ◎ 功能：如果大写的 ENVI 将设置系统的环境变量；否则仅设置用于 RXPE.EXE 的环境变量。

  ◎ 参数：分别指定环境变量名称和值。如果不指定值，则删除环境变量；如果不指定名称和值，则设置以下环境变量：
      Favorites    收藏夹目录
      Desktop       桌面目录
      StartMenu    开始菜单目录
      Startup       启动菜单目录
      Programs    程序菜单目录
      Recent       新近文件目录
      SendTo       发送到目录
      System       系统目录
      Windows       Windows 目录
      Personal    我的文档目录
      QuickLaunch 快速启动目录

  ◎ 示例：ENVI

  ◎ 备注：如果名称的最后一个字符为大写，才会通知系系使环境变量更改生效。

■ Link <快捷方式路径><,目标路径>[,运行参数][,图标路径][,图标索引]

  ◎ 功能：添加快捷方式。

  ◎ 参数：

※ 快捷方式路径    指定要生成的快捷方式的路径（.lnk 可以省略）。

※ 目标路径       指定快捷方式的目标路径。如果目标路径的结尾处有字符“?”，则快捷方式的起始位置（或者说当前目录）为临时文件夹“%TEMP%”；否则，为目标文件所在的目录。如果目标路径的结尾处有字符“<”，则快捷方式的启动方式为最小化方式启动；如果是“>”，则以隐藏方式启动；否则，为正常方式启动。

※ 运行参数       目标程序运行参数。

※ 图标路径       快捷方式图标的路径。

※ 图标索引       快捷方式图标的在文件资源中的序号，0 为第一个图标（不填写则默认）。

  ◎ 示例：Link $Desktop\宽带连接,PPPOE.CMD<,,RASDIAL.DLL,19

  ◎ 备注：无。

■ Dele <文件路径>

  ◎ 功能：删除文件或目录。

  ◎ 参数：指定要删除的文件路径，支持环境变量扩展和通配符。

  ◎ 示例：Dele %SystemRoot%\INF\*.PNF

  ◎ 备注：如果以 RAMDISK 方式启动系统，可删除一些启动后无用的文件（如 2 MB 的 NTOSKRNL.EXE）来增加 RAMDISK 的可写空间。

■ Devi <CAB 路径>

  ◎ 功能：从 CAB 文件中安装驱动程序。

  ◎ 参数：CAB 完整路径。把每个驱动单独放在一个目录中，然后把所有的驱动程序按备注中的说明，打包成 CAB 即可。

  ◎ 示例：DEVI %SystemRoot%\DRV.CAB

  ◎ 备注：CAB 文件中的。其中的 INF 文件必须以“_”开头（使用改名工具，添加“_”前缀）；并且，请最后放一个目录和空的INF（如“ZZZ\ZZ.INF”），否则最后一个驱动不会自动安装（但会解压）。可以用此功能安装驱动程序，然后删除掉 CAB 文件（如果是 RAMDISK 方式启动），以腾出空间。

■ Send <按键代码1>[,按键代码2][,按键代码3]...

  ◎ 功能：模拟按键。

  ◎ 参数：虚拟按键代码，如 VK_RETURN，请参相关考编程文档。

  ◎ 示例：Send 0x12<,0x09<,0x09>,0x12>

  ◎ 备注：上面的示例模拟 Alt+Tab。如果按键代码以“<”结束，则仅模拟按键按下；“>”则仅模拟按键弹起；否则模拟按下并弹起。按键代码支持 16 进制，如“SendKey 0x90”表示按下 NumLock 键。

■ Wait <毫秒>

  ◎ 功能：等待指定时间后再继续执行命令。

  ◎ 参数：指定时间，单位为毫秒。

  ◎ 示例：Wait 5000

  ◎ 备注：无。

■ Shut [0|1|2]...

  ◎ 功能：注销|关机|重起。

  ◎ 参数：关机方式。

  ◎ 示例：Shut

  ◎ 备注：使用经修改的 USER32.DLL，配合此命令使用，将可以拦截所有的关闭系统调用，达到正确关闭的目的（注意，此功能是快速关机，可能不会保存所有数据）。

■ Exec <程序路径>[ 参数]

  ◎ 功能：执行外部命令。

  ◎ 格式：执行外部命令。

  ◎ 参数：指定程序路径。

  ◎ 示例：ExeCmd CMD.EXE /C "DEL /Q /F %TEMP%"

  ◎ 备注：如果路径结尾处有字符“?”，则等待外部命令执行完后才继续执行下一条命令；否则立即返回并执行下一条。如果目标路径的结尾处有字符“<”，则快捷方式的启动方式为最小化方式启动；如果是“>”，则以隐藏方式启动；否则，为正常方式启动。另外，此命令可以支持使用特定参数调用 DLL 函数，功能比 RegSvr32 强大的多，命令格式为“<DLL 路径,函数名称>[,[#]参数1][,[#]参数2][,[#]参数3][,[#]参数4]”。

■ Kill <进程名称>

  ◎ 功能：强制终止指定的进程。

  ◎ 参数：进程名称，即 EXE 文件名（不包含路径）。

  ◎ 示例：Kill WinLogon

  ◎ 备注：此命令将终止与 <进程名称> 前部分匹配的所有进程。

■ Load [文件路径]

  ◎ 功能：逐条运行命令文件中的命令。

  ◎ 参数：指定文件的路径。如果省略，假设 RXPE 本身的 EXE 文件名为 RXPE.EXE，则自动搜索每个分区上的 RXPE.INI 和 RXPE\RXPE.INI 文件。把 RXPE.EXE 改名即可实现自定义命令文件的名称。

  ◎ 示例：Load

  ◎ 备注：RXPE 支持把所有的命令写在一个文件种逐条执行。RXPE 命令文件最后一行必须是空行，否则最后一行无法被执行。注意，如果在命令文件中使用此命令，要仔细分析，防止嵌套调用。

■ Init [文件路径]

  ◎ 功能：初始化桌面，然后执行 Load 功能。

  ◎ 参数：请参看 Load。

  ◎ 示例：Init

  ◎ 备注：此功能用于代替 WinLogon 登录系统，RXPE 可以完全控制登录进度。

■ Logo [[!]文件路径]

  ◎ 功能：显示启动画面。

  ◎ 参数：图片文件路径，支持 BMP/JPG/PNG/GIF 等各式（需要 GDI+ 支持）。如果参数为空，则关闭启动画面（渐隐淡出）。

  ◎ 示例：Logo !%SystemRoot%\RXPE.JPG

  ◎ 备注：上例中，将使用内部图片显示启动画面。“!”表示后台执行，不阻塞。

■ Text [文字][#颜色][,Left][,Top][,Right][,Bottom]

  ◎ 功能：在启动画面中显示进度文字。

  ◎ 参数：文字为空则不显示文字。

  ◎ 示例：TEXT 正在注册组件……#0xFFDDDD,4,745,128,768

  ◎ 备注：首次显示文字的时候必须指定位置，否则不会显示。

■ Help

  ◎ 功能：显示帮助信息。

  ◎ 参数：无。

  ◎ 示例：Help

  ◎ 备注：无。

三、备注

◎ RXPE 命令文件必须使用 UNICODE 格式。

◎ RXPE 命令不区分大小写（ENVI 除外）。

◎ RXPE 命令之后可以跟数量不限的空格和制表符。

◎ 上面的命令中 <> 内的参数表示必须输入，[] 内的表示可以省略。

◎ RXPE 中使用的路径支持环境变量替换（如“%TEMP%\X.TXT”）。

◎ 通常情况下，您只需要编辑一个命令文件 RXPE.INI，里面列出需要执行的命令。然后在 PE 启动的时候执行“RXPE.EXE Init”即可。

四、致谢

◎ 感谢老九的热心帮助和意见。
◎ 感谢无忧论坛提供的讨论场所。

要获取更多信息，请访问 WWW.YONSM.NET

Yonsm
Yonsm@163.com
WWW.Yonsm.NET
2007.1.4，杭州