briar/components/net/sf/briar/plugins/file/WindowsRemovableDriveFinder.java

package net.sf.briar.plugins.file;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;

import com.sun.jna.platform.win32.Kernel32;

class WindowsRemovableDriveFinder implements RemovableDriveFinder {

	// http://msdn.microsoft.com/en-us/library/windows/desktop/aa364939.aspx
	private static final int DRIVE_REMOVABLE = 2;

	public Collection<File> findRemovableDrives() throws IOException {
		File[] roots = File.listRoots();
		if(roots == null) throw new IOException();
		List<File> drives = new ArrayList<File>();
		for(File root : roots) {
			try {
				int type = Kernel32.INSTANCE.GetDriveType(root.getPath());
				if(type == DRIVE_REMOVABLE) drives.add(root);
			} catch(RuntimeException e) {
				throw new IOException(e.toString());
			}
		}
		return Collections.unmodifiableList(drives);
	}
}