java获取u盘_java实现监听u盘示例分享

Java可以通过JDK中的Java Native Interface（JNI）调用操作系统的API来获取U盘的插拔事件，从而实现对U盘的监听。下面是一个简单的示例代码，演示了如何在Java中监听U盘的插拔事件：

import java.io.File;
import java.util.Arrays;

public class USBListener {
    static {
        System.loadLibrary("USBListener");
    }

    public static native void start();

    public static void main(String[] args) {
        USBListener.start();
    }

    public static void onUSBInserted(String drivePath) {
        System.out.println("U盘已插入：" + drivePath);
    }

    public static void onUSBRemoved(String drivePath) {
        System.out.println("U盘已拔出：" + drivePath);
    }
}

上述代码中，我们定义了一个`USBListener`类，其中包含了一个`start()`方法，用于启动U盘监听功能。在`start()`方法中，我们通过JNI调用了一个名为`startUSBListener()`的C++函数，该函数启动了一个线程，循环遍历系统中的所有驱动器，并检查它们是否是可移动设备。如果是可移动设备，则将其保存到一个列表中，并且在后续的循环中持续检查它们是否被插拔。

当U盘被插入或拔出时，`startUSBListener()`函数会调用`onUSBInserted()`或`onUSBRemoved()`方法，这两个方法都是Java中的静态方法，可以在任何地方被调用。在本例中，我们只是简单地打印了一条消息，表示U盘已经插入或拔出了。

我们还需要在C++中实现`startUSBListener()`函数的代码，这里我们使用了Windows API来获取系统中的驱动器列表，并检查它们是否是可移动设备。以下是`startUSBListener()`函数的C++实现代码：

#include <Windows.h>
#include <stdio.h>
#include <stdlib.h>
#include "USBListener.h"

#define MAX_DRIVES 26

void startUSBListener(JNIEnv *env, jclass cls) {
    DWORD drives = GetLogicalDrives();

    char driveLetter[] = "A:\\";
    for (int i = 0; i < MAX_DRIVES; i++) {
        if ((drives & (1 << i)) != 0) {
            driveLetter[0] = 'A' + i;
            UINT type = GetDriveType(driveLetter);
            if (type == DRIVE_REMOVABLE) {
                HANDLE hDevice = CreateFile(driveLetter, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL);
                if (hDevice != INVALID_HANDLE_VALUE) {
                    char buf[MAX_PATH];
                    DWORD len;
                    if (DeviceIoControl(hDevice, IOCTL_STORAGE_GET_DEVICE_NUMBER, NULL, 0, &buf, sizeof(buf), &len, NULL)) {
                        STORAGE_DEVICE_NUMBER *number = (STORAGE_DEVICE_NUMBER *) &buf;
                        char drivePath[MAX_PATH];
                        sprintf(drivePath, "\\\\.\\PhysicalDrive%d", number->DeviceNumber);
                        HANDLE hDrive = CreateFile(drivePath, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL);
                        if (hDrive != INVALID_HANDLE_VALUE) {
                            OVERLAPPED overlapped = {};
                            char buffer[512];
                            DWORD bytesReturned;
                            if (DeviceIoControl(hDrive, IOCTL_STORAGE_GET_DEVICE_NUMBER, NULL, 0, &buf, sizeof(buf), &len, &overlapped)) {
                                while (1) {
                                    Sleep(1000);
                                    if (DeviceIoControl(hDrive, IOCTL_STORAGE_CHECK_VERIFY2, NULL, 0, NULL, 0, &bytesReturned, &overlapped)) {
                                        USBListener::onUSBRemoved(driveLetter);
                                        break;
                                    } else {
                                        if (GetLastError() == ERROR_NOT_READY) {
                                            USBListener::onUSBInserted(driveLetter);
                                            break;
                                        }
                                    }
                                }
                            }
                            CloseHandle(hDrive);
                        }
                    }
                    CloseHandle(hDevice);
                }
            }
        }
    }
}

在上述代码中，我们首先调用`GetLogicalDrives()`函数获取系统中所有的驱动器列表，然后遍历这个列表，并检查每个驱动器是否是可移动设备。如果是可移动设备，则使用`CreateFile()`函数打开该设备，并获取其设备号。然后，我们将设备号作为参数，使用`CreateFile()`函数打开该设备对应的物理驱动器，并调用`DeviceIoControl()`函数发送`IOCTL_STORAGE_CHECK_VERIFY2`控制码，以检查该U盘是否已经被插入或拔出。

如果`DeviceIoControl()`函数返回了`ERROR_NOT_READY`错误码，则表示该U盘已经被拔出，我们就调用`USBListener::onUSBRemoved()`方法通知Java代码；如果`DeviceIoControl()`函数成功返回，则表示该U盘已经被插入，我们就调用`USBListener::onUSBInserted()`方法通知Java代码。在检测到U盘被插入或拔出后，我们需要使用`Sleep()`函数等待一段时间，以避免频繁地检查设备状态导致CPU占用过高。

最后，我们需要使用JNI将C++代码编译成动态链接库，以供Java代码调用。以下是`USBListener.h`文件的内容：

#include <jni.h>

#ifdef __cplusplus
extern "C" {
#endif

JNIEXPORT void JNICALL Java_USBListener_startUSBListener(JNIEnv *, jclass);

#ifdef __cplusplus
}
#endif

然后，我们需要使用C++编译器将上述代码编译成动态链接库，以供Java代码调用。在Windows下，可以使用MinGW或Visual C++等编译器来编译，例如：

g++ -shared -o USBListener.dll -I"C:\Program Files\Java\jdk1.8.0_231\include" -I"C:\Program Files\Java\jdk1.8.0_231\include\win32" USBListener.cpp

编译完成后，我们就可以在Java代码中使用`System.loadLibrary()`方法加载该动态链接库，并调用`USBListener.start()`方法启动U盘监听功能了。