实时获得USB磁盘(再次更新)

by czy

10分感谢sinister老大指点,程序流层:

先通过设备消息得到新加入的盘符并且排除映射盘(net use/subst).

然后通过DeviceIoControl函数发送IOCTL_STORAGE_BASE equ FILE_DEVICE_MASS_STORAGE得到盘符的总线类别,而不是通过一般的GetDirverTyte,或是网上有些人说的IOCTL_STORAGE_GET_MEDIA_TYPES.

最后通过GetDirverTyte排除USB光驱.

要注意的是调用CreateFile的时候如果文件名是//./x:形式的盘符第二三四个参数一定要为NULL
网上很多代码这儿都有问题.另外由于MASM的INC文件中没有winioctl所以有些常量得自定义:)

.386
.model flat, stdcall
option casemap :none
;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
; Include 文件定义
;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
include ../include/user32.inc
includelib ../lib/user32.lib
include ../include/kernel32.inc
includelib ../lib/kernel32.lib
include ../include/windows.inc

CheckUsbDisk proto :dword

CTL_CODE MACRO DeviceType:=<0>, Function:=<0>, Method:=<0>, Access:=<0>
EXITM %(((DeviceType) SHL 16) OR ((Access) SHL 14) OR ((Function) SHL 2) OR (Method))
ENDM

;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
; Equ 等值定义
;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
ICO_MAIN equ 1000
DLG_MAIN equ 1000
IDC_TEXT equ 1001

DBT_DEVICEARRIVAL equ 8000h
DBT_DEVTYP_VOLUME equ 02h

_DEV_BROADCAST_HDR struct
dbch_size dd ?
dbch_devicetype dd ?
dbch_reserved dd ?
_DEV_BROADCAST_HDR ends

_DEV_BROADCAST_VOLUME struct
dbcv_size dd ?
dbcv_devicetype dd ?
dbcv_reserved dd ?
dbcv_unitmask dd ?
dbcv_flags dw ?
_DEV_BROADCAST_VOLUME ends

DBTF_MEDIA equ 01h
DBTF_NET equ 02h ;网络映射盘符或subst命令映射的盘符

FILE_ANY_ACCESS equ 0
METHOD_BUFFERED equ 0
FILE_DEVICE_MASS_STORAGE equ 2dh ;from ntddk.inc
IOCTL_STORAGE_BASE equ FILE_DEVICE_MASS_STORAGE ;from ntddstor.inc
IOCTL_STORAGE_QUERY_PROPERTY equ CTL_CODE(IOCTL_STORAGE_BASE, 500h, METHOD_BUFFERED, FILE_ANY_ACCESS)

BusTypeUsb equ 7

StorageDeviceProperty equ 0
PropertyStandardQuery equ 0

STORAGE_PROPERTY_QUERY struct
PropertyId dd ? ; 存储设备/适配器
QueryType dd ? ; 查询方式
AdditionalParameters dd ? ; 额外的数据(仅定义了象征性的1个字节)
STORAGE_PROPERTY_QUERY ends

STORAGE_DEVICE_DESCRIPTOR struct
Version dd ? ; 版本
theSize dd ? ; 结构大小
DeviceType db ? ; 设备类型
DeviceTypeModifier db ? ; SCSI-2额外的设备类型
RemovableMedia db ? ; 是否可移动
CommandQueueing db ? ; 是否支持命令队列
VendorIdOffset dd ? ; 厂家设定值的偏移
ProductIdOffset dd ? ; 产品ID的偏移
ProductRevisionOffset dd ? ; 产品版本的偏移
SerialNumberOffset dd ? ; 序列号的偏移
BusType dd ? ; 总线类型
RawPropertiesLength dd ? ; 额外的属性数据长度
RawDeviceProperties dd ? ; 额外的属性数据(仅定义了象征性的1个字节)
STORAGE_DEVICE_DESCRIPTOR ends

;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
; 数据段
;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
.data
disk3 db '//./',0
mao db ':',0
;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
; 代码段
;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>

.code

;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
_ProcDlgMain proc uses ebx edi esi hWnd,wMsg,wParam,lParam

mov eax,wMsg
;********************************************************************
.if eax == WM_CLOSE
invoke EndDialog,hWnd,NULL
;********************************************************************
.elseif eax == WM_DEVICECHANGE ;设备变化消息
mov eax,wParam
.if eax == DBT_DEVICEARRIVAL ;加入了可用的新设备
mov eax,lParam
assume eax:ptr _DEV_BROADCAST_HDR
mov eax,[eax].dbch_devicetype
assume eax:nothing
.if eax == DBT_DEVTYP_VOLUME ;新设备添加了一个盘符
mov eax,lParam
assume eax:ptr _DEV_BROADCAST_VOLUME
mov edx,[eax].dbcv_unitmask
;mov ecx,[eax].dbcv_flags
xor ebx,ebx
.while ebx<26
mov ecx,edx
and edx,01h
.if edx==1
mov cx,word ptr [eax].dbcv_flags
.if cx == 00h ;排除映射盘(subst或net use)
add ebx,65d ;盘符
invoke CheckUsbDisk,ebx
.endif
.break
.else
mov edx,ecx
shr edx,1
.endif
inc ebx
.endw
assume eax:nothing
.endif
.endif
;********************************************************************
.else
mov eax,FALSE
ret
.endif
mov eax,TRUE
ret

_ProcDlgMain endp

CheckUsbDisk proc uses ebx esi edi disk:dword
LOCAL buffer[128]:byte
LOCAL hDisk:dword
LOCAL notuse:dword
LOCAL Query:STORAGE_PROPERTY_QUERY
LOCAL DevDesc:STORAGE_DEVICE_DESCRIPTOR


invoke lstrcpy,addr buffer,offset disk3 ;buffer=//./
invoke lstrcat,addr buffer,addr disk ;buffer=//./X
invoke lstrcat,addr buffer,offset mao ;buffer=//./X:
invoke CreateFile,addr buffer, NULL, NULL,NULL, OPEN_EXISTING, NULL,NULL
mov hDisk,eax
.if eax !=INVALID_HANDLE_VALUE
mov Query.PropertyId,StorageDeviceProperty
mov Query.QueryType,PropertyStandardQuery
invoke DeviceIoControl,hDisk,IOCTL_STORAGE_QUERY_PROPERTY,addr Query, sizeof STORAGE_PROPERTY_QUERY,addr DevDesc,sizeof STORAGE_DEVICE_DESCRIPTOR,addr notuse, NULL
.if eax != NULL
mov eax,[ DevDesc.BusType]
.if eax == BusTypeUsb
invoke lstrcpy,addr buffer,addr disk ;buffer=X
invoke lstrcat,addr buffer,offset mao ;buffer=X:
invoke GetDriveType,addr buffer
.if eax != DRIVE_CDROM ;排除USB光驱
invoke MessageBox,0,addr buffer,addr buffer,1
.endif
.endif
.endif
.endif
invoke CloseHandle,hDisk
ret
CheckUsbDisk endp
;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
start:
invoke GetModuleHandle,NULL
invoke DialogBoxParam,eax,DLG_MAIN,NULL,offset _ProcDlgMain,NULL
invoke ExitProcess,NULL
;>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
end start