aboutsummaryrefslogtreecommitdiff
path: root/src/lombok/installer/WindowsDriveInfo.java
blob: a0cfcbc104dd5b66b9419f5da043fe17322dab36 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
/*
 * Copyright © 2009 Reinier Zwitserloot and Roel Spilker.
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package lombok.installer;

import java.util.ArrayList;
import java.util.List;

/**
 * This class uses native calls on windows to figure out all drives,
 * and, for each drive, if its a harddisk or something else.
 * 
 * The output is essentially equivalent to running windows executable:
 * <pre>fsutil fsinfo drives</pre>
 * and
 * <pre>fsutil fsinfo drivetype C:</pre>
 * 
 * except that (A) fsutil requires privileges, (B) someone might have moved
 * it out of the path or some such, and (C) its output is internationalized,
 * so unless you want to include a table of how to say "Fixed Disk" in 300
 * languages, this really is a superior solution.
 * <p>
 * To compile it, you'll need windows, as well as MinGW:
 * http://sourceforge.net/projects/mingw/files/
 * <p>
 * Fetch gcc 4.0.4+, you don't need anything extra. Toss /c/mingw/bin in
 * your git bash prompt's path (/etc/profile) and then run:
 * 
 * $ gcc -c \
 *     -I "/c/Program Files/Java/jdk1.6.0_14/include" \
 *     -I "/c/Program Files/Java/jdk1.6.0_14/include/win32" \
 *     -D__int64="long long" lombok_installer_WindowsDriveInfo.c
 * 
 * $ dllwrap.exe --add-stdcall-alias \
 *     -o WindowsDriveInfo.dll \
 *     lombok_installer_WindowsDriveInfo.o
 * 
 * You may get a warning along the lines of "Creating an export definition".
 * This is expected behaviour.
 * 
 * The DLL produced has been checked into the git repository so you won't
 * need to build this file again unless you make some changes to it.
 */
public class WindowsDriveInfo {
	/**
	 * Return a list of all available drive letters, such as ["A", "C", "D"].
	 */
	public List<String> getLogicalDrives() {
		int flags = getLogicalDrives0();
		
		List<String> letters = new ArrayList<String>();
		for ( int i = 0 ; i < 26 ; i++ ) {
			if ( (flags & (1 << i)) != 0 ) letters.add(Character.toString((char)('A' + i)));
		}
		
		return letters;
	}
	
	/**
	 * Calls kernel32's GetLogicalDrives, which returns an int containing
	 * flags; bit 0 corresponds to drive A, bit 25 to drive Z. on = disk exists.
	 */
	private native int getLogicalDrives0();
	
	/**
	 * Feed it a drive letter (such as 'A') to see if it is a fixed disk.
	 */
	public boolean isFixedDisk(String letter) {
		if ( letter.length() != 1 ) throw new IllegalArgumentException("Supply 1 letter, not: " + letter);
		char drive = Character.toUpperCase(letter.charAt(0));
		if ( drive < 'A' || drive > 'Z' ) throw new IllegalArgumentException(
				"A drive is indicated by a letter, so A-Z inclusive. Not " + drive);
		return getDriveType(drive + ":\\") == 3L;
	}
	
	/**
	 * Mirror of kernel32's GetDriveTypeA. You must pass in 'A:\\' - 
	 * so including both a colon and a backslash!
	 * 
	 * 0 = error
	 * 1 = doesn't exist
	 * 2 = removable drive
	 * 3 = fixed disk
	 * 4 = remote (network) disk
	 * 5 = cd-rom
	 * 6 = ram disk
	 */
	private native int getDriveType(String name);
	
	public static void main(String[] args) {
		System.loadLibrary("WindowsDriveInfo");
		WindowsDriveInfo info = new WindowsDriveInfo();
		
		for ( String letter : info.getLogicalDrives() ) {
			System.out.printf("Drive %s: - %s\n", letter,
					info.isFixedDisk(letter) ? "Fixed Disk" : "Not Fixed Disk");
		}
	}
}