Обнаружение вставленного USB в Windows

В настоящее время я пишу инструмент безопасности на Python, который работает как демон на главном компьютере. Всякий раз, когда обнаруживается запоминающее устройство USB, оно копирует все файлы с USB в какой-либо каталог на главном компьютере. Есть ли простой способ сделать такое обнаружение / интерфейс USB? Заранее спасибо!


person kjakeb    schedule 25.11.2010    source источник
comment
Какую ОС вы используете? Для Linux можно использовать dbus: redclay.altervista.org/ wiki / doku.php? id = projects: hal-automount   -  person unutbu    schedule 25.11.2010
comment
Для Linux см. Также stackoverflow.com/questions/469243/   -  person unutbu    schedule 25.11.2010
comment
Спасибо за ответы! В настоящее время я использую окна. Придется ли мне для этого использовать WinAPI? Еще раз спасибо!   -  person kjakeb    schedule 25.11.2010


Ответы (2)


Да, вам необходимо использовать RegisterDeviceNotification Windows Вызов API. Насколько мне известно, не существует модуля Python, который бы обертывал эту функциональность, поэтому вы должны использовать _2 _ для вызова этой функции.

К счастью, вы не первый, кто захотел это сделать, поэтому в сети есть несколько примеров кода. WxPython предоставляет образец кода, но, поскольку вы пишете демон, это может вас не интересовать. Вы можете попробовать следующий пример кода, который опирается как на ctypes, так и на 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)
person fmark    schedule 25.11.2010
comment
Когда скрипт выполняется в Windows 10 с параметром start, в консоли появляются следующие сообщения: Запуск службы DevEventHandler, Ошибка запуска службы: Доступ запрещен. - person Atalanttore; 21.10.2019

Хорошо, есть гораздо более простой способ найти USB-устройства на компьютере с Windows, используя следующий код:

import win32file

def locate_usb():
    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

код был фактически взят из https://mail.python.org/pipermail/python-win32/2006-DeDecember/005406.html

person Arjuna    schedule 23.10.2015
comment
Не знаю, насколько это надежно, но я думаю, для многих случаев использования этого будет достаточно. Спасибо за простой пример. - person Hakaishin; 30.01.2018
comment
См. Также основные функции Windows fileapi GetLogicalDrives и GetDriveTypeAA против W) - person djvg; 08.10.2019
comment
Хорошо работает. Эта реализация пропускает диск 'A:\' (позиция бита 0), но это не проблема, если предположить, что диск 'A:\' никогда не снимается. Можно использовать while mask < drivebits: ... вместо перебора всех 26 символов. - person djvg; 08.10.2019