一些 PECMD 封装函数和调用 WinApi 示例代码

Bluebells

一些实用的(?) PECMD 示例代码分享给大家参考参考
代码质量可能不太好, 一些术语描述可能有误, 因为本人没啥编程基础, 希望大家谅解

EFI 变量设置封装函数
简单的 Ini 文本文件读写封装函数
获取目标驱动器的磁盘号和分区号
判断当前安全启动状态
获取系统版本号
获取计算机名和当前登录用户名
脱机注册表封装函数

Bluebells

1.     ;IniRead 有五个参数, 第一个参数为目标 .ini 完整文件名, 第二个参数为节名, 第三个参数为键名, 第四个参数为默认值, 第五个参数为存放读取的值的变量
   
2.     ;当无法读取目标键的值时, 其返回值(第五个参数变量值)将为默认值(第四个参数定义的值)
   
3.     _SUB IniRead
   
4.     TEAM ENVI &FileName=%~1| ENVI &SectionName=%~2| ENVI &KeyName=%~3| ENVI &DefValue=%~4
   
5.     FIND %&FileName%=,EXIT _SUB
   
6.     FIND %&SectionName%=,EXIT _SUB
   
7.     FIND %&KeyName%=,EXIT _SUB
   
8.     FIND %&DefValue%=,EXIT _SUB
   
9.     CALL $--qd --ret:&ret Kernel32.dll,GetPrivateProfileStringW,$%&SectionName%,$%&KeyName%,$%&DefValue%,0,0,$%&FileName%
   
10.     TEAM CALC #&CC = %&ret% + 1| CALC #&nSize = %&CC% * 2
    
11.     SET$ &ValueName=*%&nSize% 0
    
12.     CALL $--qd --ret:&ret Kernel32.dll,GetPrivateProfileStringW,$%&SectionName%,$%&KeyName%,$%&DefValue%,*&ValueName,#%&CC%,$%&FileName%
    
13.     ENVI-ret %~5=%&ValueName%
    
14.     _END
    
15. 
    
16.     ;IniWrite 有四个参数, 第一个参数为目标 .ini 完整文件名, 第二个参数为节名, 第三个参数为键名, 第四个参数为值名, 第五个参数为返回值(若返回零则写入失败)
    
17.     ;其中第四个和第五个参数为可选, 当第四个参数留空时, 将移除目标键的值
    
18.     _SUB IniWrite
    
19.     TEAM ENVI &FileName=%~1| ENVI &SectionName=%~2| ENVI &KeyName=%~3| ENVI &ValueName=%~4
    
20.     FIND %&FileName%=,EXIT _SUB
    
21.     FIND %&SectionName%=,EXIT _SUB
    
22.     FIND %&KeyName%=,EXIT _SUB
    
23.     CALL $--qd --ret:&ret Kernel32.dll,WritePrivateProfileStringW,$%&SectionName%,$%&KeyName%,$%&ValueName%,$%&FileName%
    
24.     ENVI-ret %~5=%&ret%
    
25.     _END

复制代码

PS: mdyblog 老大的 PECMD 的示例代码(readini.wcs)中有读取 Ini 文本文件的封装代码段, 不过挺复杂的

Bluebells

1. ;只有一个输入参数和两个输出参数
   
2. ;输入参数为路径, 支持驱动器号(盘符), 路径命名空间(\\?\HarddiskVolumeX, \\?\HarddiskX\PartitionX 等)
   
3. ;第一个输出参数为设备号, 第二个输出参数为分区号
   
4. _SUB GetDevicePartNum
   
5.     SET &drv=%~1
   
6.     ENVI-ret %~2=
   
7.     ENVI-ret %~3=
   
8.     SET &prefix=\\?\
   
9.     SET &gr=
   
10.     LSTR aStr1=4,%drv%
    
11.     LSTR aStr2=7,%drv%
    
12.     FIND $%aStr1%=\\?\,SET &prefix=
    
13.     FIND $%aStr2%=\Device,SET &gr=GLOBALROOT
    
14.    
    
15.     SET &FILE_ANY_ACCESS=0x00000000
    
16.     SET &OPEN_EXISTING=3
    
17.     SET &IOCTL_STORAGE_GET_DEVICE_NUMBER=0x2D1080
    
18.     SET$# &udtQuery=*4 0 *4 0 *4 0
    
19.     ENVI$# &&cbBytesReturned=*8 0
    
20.     CALL $--qd --ret:&&h Kernel32.dll,CreateFileW,$%prefix%%gr%%drv%,#0,#%FILE_ANY_ACCESS%,#0,#%OPEN_EXISTING%,#0,#0
    
21.     IFEX #%h%=-1, EXIT
    
22.     CALL $--qd --ret:&&ret Kernel32.dll,DeviceIoControl,#%h%,#%IOCTL_STORAGE_GET_DEVICE_NUMBER%,#0,#0,*&udtQuery,#12,*&cbBytesReturned,#0 //待处理
    
23.     CALL $--qd --ret:&&ret Kernel32.dll,CloseHandle,#%h%
    
24.     ENVI?int &udtQuery=&&DeviceNum:4
    
25.     ENVI?int &udtQuery=&&PartNum:8
    
26.     ENVI-ret %~2=%&DeviceNum%
    
27.     ENVI-ret %~3=%&PartNum%
    
28. _END
    
29. 
    
30. GetDevicePartNum C: &1 &2
    
31. GetDevicePartNum \\?\HarddiskVolume2 &3 &4
    
32. MESS 设备(磁盘)号: %1%\n分区号: %2%\n\n设备(磁盘)号: %3%\n分区号: %4%
    

复制代码

a66

了解

smile_z

学习一下

红毛樱木

强贴留名

it323

了解一下。

Bluebells

1. ;此封装函数有三个参数, 第一个参数用于存放返回的操作系统主要版本号, 第二个参数用于存放返回的操作系统次要版本号, 第三个参数用于存放返回的操作系统内部版本号
   
2. _SUB RtlGetVersion
   
3.     SET$# &VersionInformation=*4 0 *4 0 *4 0 *4 0 *4 0 *256 0
   
4.     SET-long &VersionInformation=276
   
5.     CALL $--qd --ret:&ret ntdll.dll,RtlGetVersion,*&VersionInformation
   
6.     IFEX #%&ret%<>0,EXIT
   
7.     TEAM ENVI?int &VersionInformation=&MajorVersion:4| ENVI?int &VersionInformation=&MinorVersion:8| ENVI?int &VersionInformation=&BuildNumber:12
   
8.     ENVI-ret %~1=%MajorVersion%
   
9.     ENVI-ret %~2=%MinorVersion%
   
10.     ENVI-ret %~3=%BuildNumber%
    
11. _END
    
12. 
    
13. ;参数同上
    
14. _SUB GetNTVersion
    
15.     TEAM SET-long &pMajorVersion=| SET-long &pMinorVersion=| SET-long &pBuildNumber=
    
16.     CALL $--qd --ret:&ret ntdll.dll,RtlGetNtVersionNumbers,*&pMajorVersion,*&pMinorVersion,*&pBuildNumber
    
17.     IFEX #%&ret%=0,EXIT
    
18.     TEAM ENVI?int &pMajorVersion=&MajorVersion| ENVI?int &pMinorVersion=&MinorVersion| ENVI?int &pBuildNumber=&BuildNumber
    
19.     CALC #&BuildNumber = %&BuildNumber% & 0xFFFF
    
20.     ENVI-ret %~1=%MajorVersion%
    
21.     ENVI-ret %~2=%MinorVersion%
    
22.     ENVI-ret %~3=%BuildNumber%
    
23. _END
    

复制代码

PS: RtlGetNtVersionNumbers 为非公开函数, 且支持系统版本范围为 NT5.1+; 感谢 wintoflash 老大帮忙处理获取正确的 BuildNumber 值

yyz2191958

技术贴 支持

Bluebells

1. ;获取计算机名称, 仅一个参数, 用于存放返回的计算机名称
   
2. _SUB GetComputerName
   
3.     SET$# &nSize=*4 0
   
4.     CALL $--qd --ret:&ret Kernel32.dll,GetComputerNameW,*,*&nSize
   
5.     ENVI?int &nSize=&BufferSize
   
6.     IFEX $%&BufferSize%=0,EXIT _SUB
   
7.     SET$ &lpBuffer=*%&BufferSize% 0
   
8.     CALL $--qd --ret:&&ret Kernel32.dll,GetComputerNameW,*&lpBuffer,*&nSize
   
9.     IFEX $%&ret%=0,EXIT _SUB
   
10.     ENVI-ret %~1=%&lpBuffer%
    
11. _END
    
12. 
    
13. ;获取系统当前登录用户名, 仅一个参数, 用于存放返回的当前登录用户名
    
14. _SUB GetUserName
    
15.     ENVI-ret %~1=
    
16.     SET$# &pcbBuffer=*4 0
    
17.     CALL $--qd --ret:&ret Advapi32.dll,GetUserNameW,*,*&pcbBuffer
    
18.     ENVI?int &pcbBuffer=&BufferSize
    
19.     IFEX $%&BufferSize%=0,EXIT _SUB
    
20.     SET$ &lpBuffer=*%&BufferSize% 0
    
21.     CALL $--qd --ret:&ret Advapi32.dll,GetUserNameW,*&lpBuffer,*&pcbBuffer
    
22.     IFEX $%&ret%=0,EXIT _SUB
    
23.     ENVI-ret %~1=%&lpBuffer%
    
24. _END
    
25. 
    
26. ;调用示例
    
27. CALL GetComputerName &ComputerName
    
28. CALL GetUserName &UserName
    
29. MESS 计算机名称: %&ComputerName%\n\n用户名: %&UserName%
    

复制代码

handsome_xiang

感谢分享！

没想到

谢谢分享

wn168cn@163.com

支持原创

张小仙

学习一下。