Windowsに挿入されたUSBを検出する

Question

私は現在、ホストコンピュータ上でデーモンとして動作するセキュリティツールpythonで書いています。 usbストレージデバイスが検出されるたびに、usbからすべてのファイルがホストコンピュータ上のあるディレクトリにコピーされます。この種のUSB検出/インターフェースを行う簡単な方法はありますか？前もって感謝します！

Answer 1

はい、RegisterDeviceNotification Windows API呼び出しを使用する必要があります。私が知る限り、この機能をラップするPythonモジュールはありませんので、ctypesを使用してこの関数を呼び出す必要があります。

幸いにも、あなたはこれをやりたかった最初の人ではないので、Web上にいくつかのコードサンプルが浮かんでいます。 WxPythonはcode sampleを提供していますが、あなたがデーモンを作成しているので、これはあなたに興味がないかもしれません。あなたはTim Goldenから恥知らず引き上げ、​​とpywin32の両方に依存している次のコードサンプルを、しようとする場合があります：

import win32serviceutil 
import win32service 
import win32event 
import servicemanager 

import win32gui 
import win32gui_struct 
struct = win32gui_struct.struct 
pywintypes = win32gui_struct.pywintypes 
import win32con 

GUID_DEVINTERFACE_USB_DEVICE = "{A5DCBF10-6530-11D2-901F-00C04FB951ED}" 
DBT_DEVICEARRIVAL = 0x8000 
DBT_DEVICEREMOVECOMPLETE = 0x8004 

import ctypes 

# 
# Cut-down clone of UnpackDEV_BROADCAST from win32gui_struct, to be 
# used for monkey-patching said module with correct handling 
# of the "name" param of DBT_DEVTYPE_DEVICEINTERFACE 
# 
def _UnpackDEV_BROADCAST (lparam): 
    if lparam == 0: return None 
    hdr_format = "iii" 
    hdr_size = struct.calcsize (hdr_format) 
    hdr_buf = win32gui.PyGetMemory (lparam, hdr_size) 
    size, devtype, reserved = struct.unpack ("iii", hdr_buf) 
    # Due to x64 alignment issues, we need to use the full format string over 
    # the entire buffer. ie, on x64: 
    # calcsize('iiiP') != calcsize('iii')+calcsize('P') 
    buf = win32gui.PyGetMemory (lparam, size) 

    extra = {} 
    if devtype == win32con.DBT_DEVTYP_DEVICEINTERFACE: 
    fmt = hdr_format + "16s" 
    _, _, _, guid_bytes = struct.unpack (fmt, buf[:struct.calcsize(fmt)]) 
    extra['classguid'] = pywintypes.IID (guid_bytes, True) 
    extra['name'] = ctypes.wstring_at (lparam + struct.calcsize(fmt)) 
    else: 
    raise NotImplementedError("unknown device type %d" % (devtype,)) 
    return win32gui_struct.DEV_BROADCAST_INFO(devtype, **extra) 
win32gui_struct.UnpackDEV_BROADCAST = _UnpackDEV_BROADCAST 

class DeviceEventService (win32serviceutil.ServiceFramework): 

    _svc_name_ = "DevEventHandler" 
    _svc_display_name_ = "Device Event Handler" 
    _svc_description_ = "Handle device notification events" 

    def __init__(self, args): 
    win32serviceutil.ServiceFramework.__init__ (self, args) 
    self.hWaitStop = win32event.CreateEvent (None, 0, 0, None) 
    # 
    # Specify that we're interested in device interface 
    # events for USB devices 
    # 
    filter = win32gui_struct.PackDEV_BROADCAST_DEVICEINTERFACE (
     GUID_DEVINTERFACE_USB_DEVICE 
    ) 
    self.hDevNotify = win32gui.RegisterDeviceNotification (
     self.ssh, # copy of the service status handle 
     filter, 
     win32con.DEVICE_NOTIFY_SERVICE_HANDLE 
    ) 

    # 
    # Add to the list of controls already handled by the underlying 
    # ServiceFramework class. We're only interested in device events 
    # 
    def GetAcceptedControls(self): 
    rc = win32serviceutil.ServiceFramework.GetAcceptedControls (self) 
    rc |= win32service.SERVICE_CONTROL_DEVICEEVENT 
    return rc 

    # 
    # Handle non-standard service events (including our device broadcasts) 
    # by logging to the Application event log 
    # 
    def SvcOtherEx(self, control, event_type, data): 
    if control == win32service.SERVICE_CONTROL_DEVICEEVENT: 
     info = win32gui_struct.UnpackDEV_BROADCAST(data) 
     # 
     # This is the key bit here where you'll presumably 
     # do something other than log the event. Perhaps pulse 
     # a named event or write to a secure pipe etc. etc. 
     # 
     if event_type == DBT_DEVICEARRIVAL: 
     servicemanager.LogMsg (
      servicemanager.EVENTLOG_INFORMATION_TYPE, 
      0xF000, 
      ("Device %s arrived" % info.name, '') 
     ) 
     elif event_type == DBT_DEVICEREMOVECOMPLETE: 
     servicemanager.LogMsg (
      servicemanager.EVENTLOG_INFORMATION_TYPE, 
      0xF000, 
      ("Device %s removed" % info.name, '') 
     ) 

    # 
    # Standard stuff for stopping and running service; nothing 
    # specific to device notifications 
    # 
    def SvcStop(self): 
    self.ReportServiceStatus (win32service.SERVICE_STOP_PENDING) 
    win32event.SetEvent (self.hWaitStop) 

    def SvcDoRun(self): 
    win32event.WaitForSingleObject (self.hWaitStop, win32event.INFINITE) 
    servicemanager.LogMsg (
     servicemanager.EVENTLOG_INFORMATION_TYPE, 
     servicemanager.PYS_SERVICE_STOPPED, 
     (self._svc_name_, '') 
    ) 

if __name__=='__main__': 
    win32serviceutil.HandleCommandLine (DeviceEventService) 

Answer 2

OKコード次のWindowsマシンの使用上のUSBデバイスを見つけるはるかに簡単な方法があります：

def locate_usb(): 
import win32file 
drive_list = [] 
drivebits=win32file.GetLogicalDrives() 
for d in range(1,26): 
    mask=1 << d 
    if drivebits & mask: 
     # here if the drive is at least there 
     drname='%c:\\' % chr(ord('A')+d) 
     t=win32file.GetDriveType(drname) 
     if t == win32file.DRIVE_REMOVABLE: 
      drive_list.append(drname) 
return drive_list 

コードが実際に使用しているOSは何https://mail.python.org/pipermail/python-win32/2006-December/005406.html