OberonCore

MODULE MyRegDev ["user32"];
   PROCEDURE RegisterDeviceNotificationW* (hRecipient, NotificationFilter, Flags: INTEGER): INTEGER;
   PROCEDURE UnregisterDeviceNotification* (handle: INTEGER): INTEGER;
END MyRegDev.

MODULE UsbNotifier;

IMPORT SYSTEM, WinApi, HostWindows, Windows, StdLog, MyRegDev;

CONST
   WM_DEVICECHANGE = 219H; (* Сообщение об изменении устройств *)
   DBT_DEVICEARRIVAL = 8000H; (* Устройство подключено *)
   DBT_DEVICEREMOVECOMPLETE = 8004H; (* Устройство отключено *)
   DBT_DEVTYP_VOLUME = 2H; (* Тип устройства - том (флешка) *)

TYPE

   DEV_BROADCAST_HDR1* = RECORD
      dbch_size*: INTEGER;
      dbch_devicetype*: INTEGER;
      dbch_reserved*: INTEGER;
   END;

   DEV_BROADCAST_HDR* = POINTER TO DEV_BROADCAST_HDR1;

   DEV_BROADCAST_VOLUME* = POINTER TO RECORD
      dbch_size*: INTEGER;
      dbch_devicetype*: INTEGER;
      dbch_reserved*: INTEGER;
      dbcv_unitmask*: SET;
      dbcv_flags*: SHORTINT;
   END;

VAR
   hwndMain: INTEGER;
   registered: WinApi.BOOL;
   devNotify: INTEGER;

(* Callback-процедура для обработки оконных сообщений *)
PROCEDURE [callback] MainWinProc(hwnd: INTEGER; msg: INTEGER; wParam: INTEGER; lParam: INTEGER): INTEGER;
VAR
   devHdr: DEV_BROADCAST_HDR;
   devVolume: DEV_BROADCAST_VOLUME;
   driveLetter: CHAR;
   driveMask: SET;
   i: INTEGER;
BEGIN
   IF msg = WM_DEVICECHANGE THEN
      (* Проверяем, что устройство подключено *)
      IF wParam = DBT_DEVICEARRIVAL THEN
         devHdr := SYSTEM.VAL(DEV_BROADCAST_HDR, lParam);
         IF devHdr # NIL THEN
            (* Проверяем, что это том (флешка) *)
            IF devHdr.dbch_devicetype = DBT_DEVTYP_VOLUME THEN
               devVolume := SYSTEM.VAL(DEV_BROADCAST_VOLUME, lParam);;
               driveMask := devVolume.dbcv_unitmask;
               
               (* Определяем букву подключенного диска *)
               i := 0;
               WHILE (i < 26) & ~(i IN driveMask) DO
                  INC(i)
               END;
               IF i < 26 THEN
                  (* Выводим сообщение о подключении флешки *)
                  StdLog.String ("Флешка подключена на диск: "); StdLog.Char(CHR(ORD("A") + i)); StdLog.Ln
               END
            END
         END
      (* Проверяем, что устройство отключено *)
      ELSIF wParam = DBT_DEVICEREMOVECOMPLETE THEN
         devHdr := SYSTEM.VAL(DEV_BROADCAST_HDR, lParam);
         IF (devHdr # NIL) & (devHdr.dbch_devicetype = DBT_DEVTYP_VOLUME) THEN
            StdLog.String ("Флешка отключена"); StdLog.Ln
         END
      END
   END;
   
   (* Передаем сообщение на стандартную обработку *)
   RETURN WinApi.DefWindowProc(hwnd, msg, wParam, lParam)
END MainWinProc;

(* Процедура создания окна *)
PROCEDURE CreateMainWindow*;
VAR
   wc: WinApi.WNDCLASS;
   className: WinApi.PtrSTR;
   style: SET;
BEGIN
   IF hwndMain = 0 THEN
      
      (* Заполняем структуру класса окна *)
      className := "USBNotifierWindow";
      wc.style := {};
      wc.lpfnWndProc := MainWinProc;
      wc.cbClsExtra := 0;
      wc.cbWndExtra := 0;
      wc.hInstance := WinApi.GetModuleHandle(NIL);
      wc.hIcon := 0;
      wc.hCursor := 0;
      wc.hbrBackground := 0;
      wc.lpszMenuName := NIL;
      wc.lpszClassName := className;
      
      (* Регистрируем класс окна *)
      IF WinApi.RegisterClass(wc) = WinApi.FALSE THEN
         StdLog.String("Ошибка регистрации класса окна"); StdLog.Int(WinApi.GetLastError()); StdLog.Ln
      ELSE
         (*Создаем невидимое окно *)
         style := WinApi.WS_OVERLAPPEDWINDOW;
         hwndMain := WinApi.CreateWindowEx({}, className, "USB Notifier", style, WinApi.CW_USEDEFAULT, WinApi.CW_USEDEFAULT, 300, 200, 0, 0, wc.hInstance, 0);
         
         IF hwndMain = WinApi.FALSE THEN
            StdLog.String("Ошибка создания окна"); StdLog.Ln
         ELSE
            (* Делаем окно невидимым *)
            registered := WinApi.ShowWindow(hwndMain, WinApi.SW_HIDE);
         END
      END
   END
END CreateMainWindow;

(* Процедура уничтожения окна *)
PROCEDURE DestroyMainWindow*;
VAR result: WinApi.BOOL;
BEGIN
   IF hwndMain # 0 THEN
      IF WinApi.DestroyWindow(hwndMain) = WinApi.FALSE THEN
         StdLog.String ("Не удалось удалить окно"); StdLog.Ln
      ELSE
         IF WinApi.UnregisterClassW ("USBNotifierWindow", WinApi.GetModuleHandle(NIL)) = WinApi.FALSE THEN
            StdLog.String ("Не удалось удалить класс"); StdLog.Ln
         ELSE
            IF (devNotify # 0) & (MyRegDev.UnregisterDeviceNotification (devNotify) = WinApi.FALSE) THEN StdLog.String ("Не удалось отменить регистрацию."); StdLog.Ln END
         END;
         hwndMain := 0;
         registered := WinApi.FALSE;
      END
   END
END DestroyMainWindow;

(* Процедура регистрации для получения уведомлений об устройствах *)
PROCEDURE RegisterForDeviceNotifications*;
CONST
   DEVICE_NOTIFY_WINDOW_HANDLE = 0;
VAR
   filter: DEV_BROADCAST_HDR;
   
BEGIN
   IF hwndMain = WinApi.FALSE THEN
      CreateMainWindow()
   ELSE
      (*DestroyMainWindow*)
   END;
   
   IF hwndMain # 0 THEN
      (* Заполняем структуру фильтра *)
      NEW(filter);
      filter.dbch_size := SIZE(DEV_BROADCAST_HDR1);
      StdLog.Int(filter.dbch_size);StdLog.Ln;
      filter.dbch_devicetype := DBT_DEVTYP_VOLUME;
      filter.dbch_reserved := 0;
      
      (* Регистрируем для получения уведомлений *)
      devNotify := MyRegDev.RegisterDeviceNotificationW(hwndMain, SYSTEM.ADR(filter), DEVICE_NOTIFY_WINDOW_HANDLE);
      IF devNotify = WinApi.FALSE THEN
         StdLog.String("Ошибка регистрации уведомлений:"); StdLog.Int(WinApi.GetLastError()); StdLog.Ln;
         (*DestroyMainWindow*)
      ELSE
         StdLog.String("Регистрация для уведомлений USB выполнена успешно");
      END
   END
END RegisterForDeviceNotifications;

(* Инициализация модуля *)
BEGIN
   hwndMain := 0;
   registered := WinApi.FALSE;
END UsbNotifier.

UsbNotifier.RegisterForDeviceNotifications

UsbNotifier.DestroyMainWindow

VAR filter: DEV_BROADCAST_VOLUME;

NEW(filter);
filter.dbch_size := SIZE(DEV_BROADCAST_VOLUME1);  (* Или SIZE(filter^), в зависимости от системы *)
filter.dbch_devicetype := DBT_DEVTYP_VOLUME;
filter.dbch_reserved := 0;
filter.dbcv_unitmask := {0..25};  (* Маска для всех дисков A-Z (0x03FFFFFF) *)
filter.dbcv_flags := 0;  (* Или 1 (DBTF_MEDIA) для фокуса на съёмных носителях *)

oberondevNotify := MyRegDev.RegisterDeviceNotificationW(hwndMain, SYSTEM.VAL(INTEGER, filter), DEVICE_NOTIFY_WINDOW_HANDLE);