1 /*
2 * Copyright (c) 2003, 2015, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25
26 #define OEMRESOURCE
27
28 #ifdef DEBUG
29 // Warning : do not depend on anything in <awt.h>. Including this file
30 // is a fix for 4507525 to use the same operator new and delete as AWT.
31 // This file should stand independent of AWT and should ultimately be
32 // put into its own DLL.
33 #include <awt.h>
34 #else
35 // Include jni_util.h first, so JNU_* macros can be redefined
36 #include "jni_util.h"
37 // Borrow some macros from awt.h
38 #define JNU_NewStringPlatform(env, x) env->NewString(reinterpret_cast<jchar*>(x), static_cast<jsize>(_tcslen(x)))
39 #define JNU_GetStringPlatformChars(env, x, y) reinterpret_cast<LPCWSTR>(env->GetStringChars(x, y))
40 #define JNU_ReleaseStringPlatformChars(env, x, y) env->ReleaseStringChars(x, reinterpret_cast<const jchar*>(y))
41 #endif // DEBUG
42
43 #include <windows.h>
44 #include <shlobj.h>
45 #include <shellapi.h>
46 #include "jlong.h"
47 #include "alloc.h"
48
49 #include "stdhdrs.h"
50
51 // Copy from shlguid.h which is no longer in PlatformSDK
52 #ifndef DEFINE_SHLGUID
53 #define DEFINE_SHLGUID(name, l, w1, w2) DEFINE_GUID(name, l, w1, w2, 0xC0, 0, 0, 0, 0, 0, 0, 0x46)
54 #endif
55
56 // {93F2F68C-1D1B-11d3-A30E-00C04F79ABD1}
57 DEFINE_GUID(IID_IShellFolder2, 0x93f2f68c, 0x1d1b, 0x11d3, 0xa3, 0xe, 0x0, 0xc0, 0x4f, 0x79, 0xab, 0xd1);
58
59 #undef IID_IShellLinkW
60 #undef IID_IExtractIconW
61 // copied from shlguid.h
62 DEFINE_SHLGUID(IID_IShellLinkW, 0x000214F9L, 0, 0);
63 DEFINE_SHLGUID(IID_IExtractIconW, 0x000214FAL, 0, 0);
64
65 //#include <sun_awt_shell_Win32ShellFolder2.h>
66
67 #ifndef DASSERT
68 #define DASSERT(x)
69 #endif
70 #define DEFINE_FIELD_ID(var, cls, field, type) \
71 jfieldID var = env->GetFieldID(cls, field, type); \
72 DASSERT(var != NULL); \
73 CHECK_NULL_RETURN(var, NULL);
74
75 #define EXCEPTION_CHECK \
76 if(env->ExceptionCheck()) { \
77 throw std::bad_alloc(); \
78 }
79
80 // Shell Functions
81 typedef BOOL (WINAPI *DestroyIconType)(HICON);
82 typedef HINSTANCE (WINAPI *FindExecutableType)(LPCTSTR,LPCTSTR,LPTSTR);
83 typedef HICON (WINAPI *ImageList_GetIconType)(HIMAGELIST,int,UINT);
84 typedef BOOL (WINAPI *GetIconInfoType)(HICON,PICONINFO);
85 typedef HRESULT (WINAPI *SHGetDesktopFolderType)(IShellFolder**);
86 typedef DWORD* (WINAPI *SHGetFileInfoType)(LPCTSTR,DWORD,SHFILEINFO*,UINT,UINT);
87 typedef HRESULT (WINAPI *SHGetMallocType)(IMalloc**);
88 typedef BOOL (WINAPI *SHGetPathFromIDListType)(LPCITEMIDLIST,LPTSTR);
89 typedef HRESULT (WINAPI *SHGetSpecialFolderLocationType)(HWND,int,LPITEMIDLIST*);
90
91 static DestroyIconType fn_DestroyIcon;
92 static FindExecutableType fn_FindExecutable;
93 static GetIconInfoType fn_GetIconInfo;
94 static ImageList_GetIconType fn_ImageList_GetIcon;
95 static SHGetDesktopFolderType fn_SHGetDesktopFolder;
96 static SHGetFileInfoType fn_SHGetFileInfo;
97 static SHGetMallocType fn_SHGetMalloc;
98 static SHGetPathFromIDListType fn_SHGetPathFromIDList;
99 static SHGetSpecialFolderLocationType fn_SHGetSpecialFolderLocation;
100
101 // Field IDs
102 static jmethodID MID_pIShellFolder;
103 static jfieldID FID_pIShellIcon;
104 static jmethodID MID_relativePIDL;
105 static jfieldID FID_displayName;
106 static jfieldID FID_folderType;
107
108 // Other statics
109 static IMalloc* pMalloc;
110 static IShellFolder* pDesktop;
111
112 // Some macros from awt.h, because it is not included in release
113 #ifndef IS_WIN2000
114 #define IS_WIN2000 (LOBYTE(LOWORD(::GetVersion())) >= 5)
115 #endif
116 #ifndef IS_WINXP
117 #define IS_WINXP ((IS_WIN2000 && HIBYTE(LOWORD(::GetVersion())) >= 1) || LOBYTE(LOWORD(::GetVersion())) > 5)
118 #endif
119 #ifndef IS_WINVISTA
120 #define IS_WINVISTA (!(::GetVersion() & 0x80000000) && LOBYTE(LOWORD(::GetVersion())) >= 6)
121 #endif
122
123
124 extern "C" {
125
126 static BOOL initShellProcs()
127 {
128 static HMODULE libShell32 = NULL;
129 static HMODULE libUser32 = NULL;
130 static HMODULE libComCtl32 = NULL;
131 // If already initialized, return TRUE
132 if (libShell32 != NULL && libUser32 != NULL) {
133 return TRUE;
134 }
135 // Load libraries
136 libShell32 = JDK_LoadSystemLibrary("shell32.dll");
137 if (libShell32 == NULL) {
138 return FALSE;
139 }
140 libUser32 = JDK_LoadSystemLibrary("user32.dll");
141 if (libUser32 == NULL) {
142 return FALSE;
143 }
144 libComCtl32 = JDK_LoadSystemLibrary("comctl32.dll");
145 if (libComCtl32 == NULL) {
146 return FALSE;
147 }
148
149 // Set up procs - libComCtl32
150 fn_ImageList_GetIcon = (ImageList_GetIconType)GetProcAddress(libComCtl32, "ImageList_GetIcon");
151 if (fn_ImageList_GetIcon == NULL) {
152 return FALSE;
153 }
154
155 // Set up procs - libShell32
156 fn_FindExecutable = (FindExecutableType)GetProcAddress(
157 libShell32, "FindExecutableW");
158 if (fn_FindExecutable == NULL) {
159 return FALSE;
160 }
161 fn_SHGetDesktopFolder = (SHGetDesktopFolderType)GetProcAddress(libShell32,
162 "SHGetDesktopFolder");
163 if (fn_SHGetDesktopFolder == NULL) {
164 return FALSE;
165 }
166 fn_SHGetFileInfo = (SHGetFileInfoType)GetProcAddress(
167 libShell32, "SHGetFileInfoW");
168 if (fn_SHGetFileInfo == NULL) {
169 return FALSE;
170 }
171 fn_SHGetMalloc = (SHGetMallocType)GetProcAddress(libShell32,
172 "SHGetMalloc");
173 if (fn_SHGetMalloc == NULL) {
174 return FALSE;
175 }
176 // Set up IMalloc
177 if (fn_SHGetMalloc(&pMalloc) != S_OK) {
178 return FALSE;
179 }
180 fn_SHGetPathFromIDList = (SHGetPathFromIDListType)GetProcAddress(
181 libShell32, "SHGetPathFromIDListW");
182 if (fn_SHGetPathFromIDList == NULL) {
183 return FALSE;
184 }
185 fn_SHGetSpecialFolderLocation = (SHGetSpecialFolderLocationType)
186 GetProcAddress(libShell32, "SHGetSpecialFolderLocation");
187 if (fn_SHGetSpecialFolderLocation == NULL) {
188 return FALSE;
189 }
190
191 // Set up procs - libUser32
192 fn_GetIconInfo = (GetIconInfoType)GetProcAddress(libUser32, "GetIconInfo");
193 if (fn_GetIconInfo == NULL) {
194 return FALSE;
195 }
196 fn_DestroyIcon = (DestroyIconType)GetProcAddress(libUser32, "DestroyIcon");
197 if (fn_DestroyIcon == NULL) {
198 return FALSE;
199 }
200 return TRUE;
201 }
202
203 // To call real JNU_NewStringPlatform
204 #undef JNU_NewStringPlatform
205 static jstring jstringFromSTRRET(JNIEnv* env, LPITEMIDLIST pidl, STRRET* pStrret) {
206 switch (pStrret->uType) {
207 case STRRET_CSTR :
208 return JNU_NewStringPlatform(env, reinterpret_cast<const char*>(pStrret->cStr));
209 case STRRET_OFFSET :
210 // Note : this may need to be WCHAR instead
211 return JNU_NewStringPlatform(env,
212 (CHAR*)pidl + pStrret->uOffset);
213 case STRRET_WSTR :
214 return env->NewString(reinterpret_cast<const jchar*>(pStrret->pOleStr),
215 static_cast<jsize>(wcslen(pStrret->pOleStr)));
216 }
217 return NULL;
218 }
219 // restoring the original definition
220 #define JNU_NewStringPlatform(env, x) env->NewString(reinterpret_cast<jchar*>(x), static_cast<jsize>(_tcslen(x)))
221
222 /*
223 * Class: sun_awt_shell_Win32ShellFolder2
224 * Method: initIDs
225 * Signature: ()V
226 */
227 JNIEXPORT void JNICALL Java_sun_awt_shell_Win32ShellFolder2_initIDs
228 (JNIEnv* env, jclass cls)
229 {
230 if (!initShellProcs()) {
231 JNU_ThrowInternalError(env, "Could not initialize shell library");
232 return;
233 }
234 MID_pIShellFolder = env->GetMethodID(cls, "setIShellFolder", "(J)V");
235 CHECK_NULL(MID_pIShellFolder);
236 FID_pIShellIcon = env->GetFieldID(cls, "pIShellIcon", "J");
237 CHECK_NULL(FID_pIShellIcon);
238 MID_relativePIDL = env->GetMethodID(cls, "setRelativePIDL", "(J)V");
239 CHECK_NULL(MID_relativePIDL);
240 FID_displayName = env->GetFieldID(cls, "displayName", "Ljava/lang/String;");
241 CHECK_NULL(FID_displayName);
242 FID_folderType = env->GetFieldID(cls, "folderType", "Ljava/lang/String;");
243 CHECK_NULL(FID_folderType);
244 }
245
246
247 /*
248 * Class: sun_awt_shell_Win32ShellFolderManager2
249 * Method: initializeCom
250 * Signature: ()V
251 */
252 JNIEXPORT void JNICALL Java_sun_awt_shell_Win32ShellFolderManager2_initializeCom
253 (JNIEnv* env, jclass cls)
254 {
255 HRESULT hr = ::CoInitialize(NULL);
256 if (FAILED(hr)) {
257 char c[64];
258 sprintf(c, "Could not initialize COM: HRESULT=0x%08X", hr);
259 JNU_ThrowInternalError(env, c);
260 }
261 }
262
263 /*
264 * Class: sun_awt_shell_Win32ShellFolderManager2
265 * Method: uninitializeCom
266 * Signature: ()V
267 */
268 JNIEXPORT void JNICALL Java_sun_awt_shell_Win32ShellFolderManager2_uninitializeCom
269 (JNIEnv* env, jclass cls)
270 {
271 ::CoUninitialize();
272 }
273
274 static IShellIcon* getIShellIcon(IShellFolder* pIShellFolder) {
275 // http://msdn.microsoft.com/library/en-us/shellcc/platform/Shell/programmersguide/shell_int/shell_int_programming/std_ifaces.asp
276 HRESULT hres;
277 IShellIcon* pIShellIcon;
278 if (pIShellFolder != NULL) {
279 hres = pIShellFolder->QueryInterface(IID_IShellIcon, (void**)&pIShellIcon);
280 if (SUCCEEDED(hres)) {
281 return pIShellIcon;
282 }
283 }
284 return (IShellIcon*)NULL;
285 }
286
287
288 /*
289 * Class: sun_awt_shell_Win32ShellFolder2
290 * Method: getIShellIcon
291 * Signature: (J)J
292 */
293 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_getIShellIcon
294 (JNIEnv* env, jclass cls, jlong parentIShellFolder)
295 {
296 return (jlong)getIShellIcon((IShellFolder*)parentIShellFolder);
297 }
298
299
300 /*
301 * Class: sun_awt_shell_Win32ShellFolder2
302 * Method: initDesktop
303 * Signature: ()V
304 */
305 JNIEXPORT void JNICALL Java_sun_awt_shell_Win32ShellFolder2_initDesktop
306 (JNIEnv* env, jobject desktop)
307 {
308 // Get desktop IShellFolder
309 HRESULT res = fn_SHGetDesktopFolder(&pDesktop);
310 if (res != S_OK) {
311 JNU_ThrowInternalError(env, "Could not get desktop shell folder");
312 return;
313 }
314 // Set field ID for pIShellFolder
315 env->CallVoidMethod(desktop, MID_pIShellFolder, (jlong)pDesktop);
316 // Get desktop relative PIDL
317 LPITEMIDLIST relPIDL;
318 res = fn_SHGetSpecialFolderLocation(NULL, CSIDL_DESKTOP, &relPIDL);
319 if (res != S_OK) {
320 JNU_ThrowInternalError(env,
321 "Could not get desktop shell folder ID list");
322 return;
323 }
324 // Set field ID for relative PIDL
325 env->CallVoidMethod(desktop, MID_relativePIDL, (jlong)relPIDL);
326 }
327
328 /*
329 * Class: sun_awt_shell_Win32ShellFolder2
330 * Method: initSpecial
331 * Signature: (JI)V
332 */
333 JNIEXPORT void JNICALL Java_sun_awt_shell_Win32ShellFolder2_initSpecial
334 (JNIEnv* env, jobject folder, jlong desktopIShellFolder, jint folderType)
335 {
336 // Get desktop IShellFolder interface
337 IShellFolder* pDesktop = (IShellFolder*)desktopIShellFolder;
338 if (pDesktop == NULL) {
339 JNU_ThrowInternalError(env, "Desktop shell folder missing");
340 return;
341 }
342 // Get special folder relative PIDL
343 LPITEMIDLIST relPIDL;
344 HRESULT res = fn_SHGetSpecialFolderLocation(NULL, folderType,
345 &relPIDL);
346 if (res != S_OK) {
347 JNU_ThrowIOException(env, "Could not get shell folder ID list");
348 return;
349 }
350 // Set field ID for relative PIDL
351 env->CallVoidMethod(folder, MID_relativePIDL, (jlong)relPIDL);
352 // Get special folder IShellFolder interface
353 IShellFolder* pFolder;
354 res = pDesktop->BindToObject(relPIDL, NULL, IID_IShellFolder,
355 (void**)&pFolder);
356 if (res != S_OK) {
357 JNU_ThrowInternalError(env,
358 "Could not bind shell folder to interface");
359 return;
360 }
361 // Set field ID for pIShellFolder
362 env->CallVoidMethod(folder, MID_pIShellFolder, (jlong)pFolder);
363 }
364
365
366 /*
367 * Class: sun_awt_shell_Win32ShellFolder2
368 * Method: getNextPIDLEntry
369 * Signature: (J)J
370 */
371 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_getNextPIDLEntry
372 (JNIEnv* env, jclass cls, jlong jpIDL)
373 {
374 LPITEMIDLIST pIDL = (LPITEMIDLIST)jpIDL;
375
376 // Check for valid pIDL.
377 if(pIDL == NULL)
378 return NULL;
379
380 // Get the size of the specified item identifier.
381 int cb = pIDL->mkid.cb;
382
383 // If the size is zero, it is the end of the list.
384 if (cb == 0)
385 return NULL;
386
387 // Add cb to pidl (casting to increment by bytes).
388 pIDL = (LPITEMIDLIST)(((LPBYTE)pIDL) + cb);
389
390 // Return NULL if it is null-terminating, or a pidl otherwise.
391 return (pIDL->mkid.cb == 0) ? 0 : (jlong)pIDL;
392 }
393
394
395 /*
396 * Class: sun_awt_shell_Win32ShellFolder2
397 * Method: copyFirstPIDLEntry
398 * Signature: (J)J
399 */
400 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_copyFirstPIDLEntry
401 (JNIEnv* env, jclass cls, jlong jpIDL)
402 {
403 LPITEMIDLIST pIDL = (LPITEMIDLIST)jpIDL;
404 if (pIDL == NULL) {
405 return 0;
406 }
407 // Get the size of the specified item identifier.
408 int cb = pIDL->mkid.cb;
409
410 // If the size is zero, it is the end of the list.
411 if (cb == 0)
412 return 0;
413
414 if (!IS_SAFE_SIZE_ADD(cb, sizeof(SHITEMID))) {
415 return 0;
416 }
417 // Allocate space for this as well as null-terminating entry.
418 LPITEMIDLIST newPIDL = (LPITEMIDLIST)pMalloc->Alloc(cb + sizeof(SHITEMID));
419
420 // Copy data.
421 memcpy(newPIDL, pIDL, cb);
422
423 // Set null terminator for next entry.
424 LPITEMIDLIST nextPIDL = (LPITEMIDLIST)(((LPBYTE)newPIDL) + cb);
425 nextPIDL->mkid.cb = 0;
426
427 return (jlong)newPIDL;
428 }
429
430 static int pidlLength(LPITEMIDLIST pIDL) {
431 int len = 0;
432 while (pIDL->mkid.cb != 0) {
433 int cb = pIDL->mkid.cb;
434 len += cb;
435 pIDL = (LPITEMIDLIST)(((LPBYTE)pIDL) + cb);
436 }
437 return len;
438 }
439
440 /*
441 * Class: sun_awt_shell_Win32ShellFolder2
442 * Method: combinePIDLs
443 * Signature: (J)J
444 */
445 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_combinePIDLs
446 (JNIEnv* env, jclass cls, jlong jppIDL, jlong jpIDL)
447 {
448 // Combine an absolute (fully qualified) pidl in a parent with the relative
449 // pidl of a child object to create a new absolute pidl for the child.
450
451 LPITEMIDLIST parentPIDL = (LPITEMIDLIST)jppIDL;
452 LPITEMIDLIST relativePIDL = (LPITEMIDLIST)jpIDL;
453
454 int len1 = pidlLength(parentPIDL);
455 int len2 = pidlLength(relativePIDL);
456
457 if (!IS_SAFE_SIZE_ADD(len1, len2) || !IS_SAFE_SIZE_ADD(len1 + len2, sizeof(SHITEMID))) {
458 return 0;
459 }
460 LPITEMIDLIST newPIDL = (LPITEMIDLIST)pMalloc->Alloc(len1 + len2 + sizeof(SHITEMID));
461 memcpy(newPIDL, parentPIDL, len1);
462 memcpy(((LPBYTE) newPIDL) + len1, relativePIDL, len2);
463 LPITEMIDLIST nullTerminator = (LPITEMIDLIST)(((LPBYTE) newPIDL) + len1 + len2);
464 nullTerminator->mkid.cb = 0;
465
466 return (jlong) newPIDL;
467 }
468
469
470 /*
471 * Class: sun_awt_shell_Win32ShellFolder2
472 * Method: releasePIDL
473 * Signature: (J)V
474 */
475 JNIEXPORT void JNICALL Java_sun_awt_shell_Win32ShellFolder2_releasePIDL
476 (JNIEnv* env, jclass cls, jlong pIDL)
477 {
478 if (pIDL != 0L) {
479 pMalloc->Free((LPITEMIDLIST)pIDL);
480 }
481 }
482
483
484 /*
485 * Class: sun_awt_shell_Win32ShellFolder2
486 * Method: releaseIShellFolder
487 * Signature: (J)V
488 */
489 JNIEXPORT void JNICALL Java_sun_awt_shell_Win32ShellFolder2_releaseIShellFolder
490 (JNIEnv* env, jclass cls, jlong pIShellFolder)
491 {
492 if (pIShellFolder != 0L) {
493 ((IShellFolder*)pIShellFolder)->Release();
494 }
495 }
496
497
498 /*
499 * Class: sun_awt_shell_Win32ShellFolder2
500 * Method: compareIDs
501 * Signature: (JJJ)I
502 */
503 JNIEXPORT jint JNICALL Java_sun_awt_shell_Win32ShellFolder2_compareIDs
504 (JNIEnv* env, jclass cls, jlong jpParentIShellFolder, jlong pIDL1, jlong pIDL2)
505 {
506 IShellFolder* pParentIShellFolder = (IShellFolder*)jpParentIShellFolder;
507 if (pParentIShellFolder == NULL) {
508 return 0;
509 }
510 return pParentIShellFolder->CompareIDs(0, (LPCITEMIDLIST) pIDL1, (LPCITEMIDLIST) pIDL2);
511 }
512
513
514 /*
515 * Class: sun_awt_shell_Win32ShellFolder2
516 * Method: getAttributes0
517 * Signature: (JJI)J
518 */
519 JNIEXPORT jint JNICALL Java_sun_awt_shell_Win32ShellFolder2_getAttributes0
520 (JNIEnv* env, jclass cls, jlong jpParentIShellFolder, jlong jpIDL, jint attrsMask)
521 {
522 IShellFolder* pParentIShellFolder = (IShellFolder*)jpParentIShellFolder;
523 if (pParentIShellFolder == NULL) {
524 return 0;
525 }
526 LPCITEMIDLIST pIDL = (LPCITEMIDLIST)jpIDL;
527 if (pIDL == NULL) {
528 return 0;
529 }
530 ULONG attrs = attrsMask;
531 HRESULT res = pParentIShellFolder->GetAttributesOf(1, &pIDL, &attrs);
532 return attrs;
533 }
534
535
536 /*
537 * Class: sun_awt_shell_Win32ShellFolder2
538 * Method: getFileSystemPath0
539 * Signature: (I)Ljava/lang/String;
540 */
541 JNIEXPORT jstring JNICALL Java_sun_awt_shell_Win32ShellFolder2_getFileSystemPath0
542 (JNIEnv* env, jclass cls, jint csidl)
543 {
544 LPITEMIDLIST relPIDL;
545 TCHAR szBuf[MAX_PATH];
546 HRESULT res = fn_SHGetSpecialFolderLocation(NULL, csidl, &relPIDL);
547 if (res != S_OK) {
548 JNU_ThrowIOException(env, "Could not get shell folder ID list");
549 return NULL;
550 }
551 if (fn_SHGetPathFromIDList(relPIDL, szBuf)) {
552 return JNU_NewStringPlatform(env, szBuf);
553 } else {
554 return NULL;
555 }
556 }
557
558 /*
559 * Class: sun_awt_shell_Win32ShellFolder2
560 * Method: getEnumObjects
561 * Signature: (JZ)J
562 */
563 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_getEnumObjects
564 (JNIEnv* env, jobject folder, jlong pIShellFolder,
565 jboolean isDesktop, jboolean includeHiddenFiles)
566 {
567 IShellFolder* pFolder = (IShellFolder*)pIShellFolder;
568 if (pFolder == NULL) {
569 return 0;
570 }
571 DWORD dwFlags = SHCONTF_FOLDERS | SHCONTF_NONFOLDERS;
572 if (includeHiddenFiles) {
573 dwFlags |= SHCONTF_INCLUDEHIDDEN;
574 }
575 /*
576 if (!isDesktop) {
577 dwFlags = dwFlags | SHCONTF_NONFOLDERS;
578 }
579 */
580 IEnumIDList* pEnum;
581 if (pFolder->EnumObjects(NULL, dwFlags, &pEnum) != S_OK) {
582 return 0;
583 }
584 return (jlong)pEnum;
585 }
586
587 /*
588 * Class: sun_awt_shell_Win32ShellFolder2
589 * Method: getNextChild
590 * Signature: (J)J
591 */
592 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_getNextChild
593 (JNIEnv* env, jobject folder, jlong pEnumObjects)
594 {
595 IEnumIDList* pEnum = (IEnumIDList*)pEnumObjects;
596 if (pEnum == NULL) {
597 return 0;
598 }
599 LPITEMIDLIST pidl;
600 if (pEnum->Next(1, &pidl, NULL) != S_OK) {
601 return 0;
602 }
603 return (jlong)pidl;
604 }
605
606 /*
607 * Class: sun_awt_shell_Win32ShellFolder2
608 * Method: releaseEnumObjects
609 * Signature: (J)V
610 */
611 JNIEXPORT void JNICALL Java_sun_awt_shell_Win32ShellFolder2_releaseEnumObjects
612 (JNIEnv* env, jobject folder, jlong pEnumObjects)
613 {
614 IEnumIDList* pEnum = (IEnumIDList*)pEnumObjects;
615 if (pEnum == NULL) {
616 return;
617 }
618 pEnum->Release();
619 }
620
621 /*
622 * Class: sun_awt_shell_Win32ShellFolder2
623 * Method: bindToObject
624 * Signature: (JJ)J
625 */
626 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_bindToObject
627 (JNIEnv* env, jclass cls, jlong parentIShellFolder, jlong relativePIDL)
628 {
629 IShellFolder* pParent = (IShellFolder*)parentIShellFolder;
630 if (pParent == NULL) {
631 return 0;
632 }
633 LPITEMIDLIST pidl = (LPITEMIDLIST)relativePIDL;
634 if (pidl == NULL) {
635 return 0;
636 }
637 IShellFolder* pFolder;
638 HRESULT hr = pParent->BindToObject(pidl, NULL, IID_IShellFolder, (void**)&pFolder);
639 if (SUCCEEDED (hr)) {
640 return (jlong)pFolder;
641 }
642 return 0;
643 }
644
645
646 /*
647 * Class: sun_awt_shell_Win32ShellFolder2
648 * Method: getLinkLocation
649 * Signature: (JJZ)J;
650 */
651 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_getLinkLocation
652 (JNIEnv* env, jclass cls, jlong parentIShellFolder, jlong relativePIDL, jboolean resolve)
653 {
654 HRESULT hres;
655 STRRET strret;
656 OLECHAR olePath[MAX_PATH]; // wide-char version of path name
657 LPWSTR wstr;
658
659 IShellFolder* pParent = (IShellFolder*)parentIShellFolder;
660 if (pParent == NULL) {
661 return NULL;
662 }
663
664 LPITEMIDLIST pidl = (LPITEMIDLIST)relativePIDL;
665 if (pidl == NULL) {
666 return NULL;
667 }
668
669 hres = pParent->GetDisplayNameOf(pidl, SHGDN_NORMAL | SHGDN_FORPARSING, &strret);
670 if (FAILED(hres)) {
671 return NULL;
672 }
673
674 switch (strret.uType) {
675 case STRRET_CSTR :
676 // IShellFolder::ParseDisplayName requires the path name in Unicode.
677 MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, strret.cStr, -1, olePath, MAX_PATH);
678 wstr = olePath;
679 break;
680
681 case STRRET_OFFSET :
682 MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, (CHAR *)pidl + strret.uOffset, -1, olePath, MAX_PATH);
683 wstr = olePath;
684 break;
685
686 case STRRET_WSTR :
687 wstr = strret.pOleStr;
688 break;
689 }
690
691 IShellLinkW* psl;
692 hres = ::CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLinkW, (LPVOID *)&psl);
693 if (SUCCEEDED(hres)) {
694 IPersistFile* ppf;
695 hres = psl->QueryInterface(IID_IPersistFile, (void**)&ppf);
696 if (SUCCEEDED(hres)) {
697 hres = ppf->Load(wstr, STGM_READ);
698 if (SUCCEEDED(hres)) {
699 if (resolve) {
700 hres = psl->Resolve(NULL, 0);
701 // Ignore failure
702 }
703 pidl = (LPITEMIDLIST)NULL;
704 hres = psl->GetIDList(&pidl);
705 }
706 ppf->Release();
707 }
708 psl->Release();
709 }
710
711 if (strret.uType == STRRET_WSTR) {
712 CoTaskMemFree(strret.pOleStr);
713 }
714 if (SUCCEEDED(hres)) {
715 return (jlong)pidl;
716 } else {
717 return 0;
718 }
719 }
720
721
722 /*
723 * Class: sun_awt_shell_Win32ShellFolder2
724 * Method: parseDisplayName0
725 * Signature: (JLjava/lang/String;)J
726 */
727 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_parseDisplayName0
728 (JNIEnv* env, jclass cls, jlong jpIShellFolder, jstring jname)
729 {
730
731 // Get desktop IShellFolder interface
732 IShellFolder* pIShellFolder = (IShellFolder*)jpIShellFolder;
733 if (pIShellFolder == NULL) {
734 JNU_ThrowInternalError(env, "Desktop shell folder missing");
735 return 0;
736 }
737 // Get relative PIDL for name
738 LPITEMIDLIST pIDL;
739 int nLength = env->GetStringLength(jname);
740 const jchar* strPath = env->GetStringChars(jname, NULL);
741 JNU_CHECK_EXCEPTION_RETURN(env, 0);
742 jchar* wszPath = new jchar[nLength + 1];
743 wcsncpy(reinterpret_cast<LPWSTR>(wszPath), reinterpret_cast<LPCWSTR>(strPath), nLength);
744 wszPath[nLength] = 0;
745 HRESULT res = pIShellFolder->ParseDisplayName(NULL, NULL,
746 reinterpret_cast<LPWSTR>(wszPath), NULL, &pIDL, NULL);
747 if (res != S_OK) {
748 JNU_ThrowIOException(env, "Could not parse name");
749 pIDL = 0;
750 }
751 delete[] wszPath;
752 env->ReleaseStringChars(jname, strPath);
753 return (jlong)pIDL;
754 }
755
756
757 /*
758 * Class: sun_awt_shell_Win32ShellFolder2
759 * Method: getDisplayNameOf
760 * Signature: (JJI)Ljava/lang/String;
761 */
762 JNIEXPORT jstring JNICALL Java_sun_awt_shell_Win32ShellFolder2_getDisplayNameOf
763 (JNIEnv* env, jclass cls, jlong parentIShellFolder, jlong relativePIDL, jint attrs)
764 {
765 IShellFolder* pParent = (IShellFolder*)parentIShellFolder;
766 if (pParent == NULL) {
767 return NULL;
768 }
769 LPITEMIDLIST pidl = (LPITEMIDLIST)relativePIDL;
770 if (pidl == NULL) {
771 return NULL;
772 }
773 STRRET strret;
774 if (pParent->GetDisplayNameOf(pidl, attrs, &strret) != S_OK) {
775 return NULL;
776 }
777 jstring result = jstringFromSTRRET(env, pidl, &strret);
778 if (strret.uType == STRRET_WSTR) {
779 CoTaskMemFree(strret.pOleStr);
780 }
781 return result;
782 }
783
784 /*
785 * Class: sun_awt_shell_Win32ShellFolder2
786 * Method: getFolderType
787 * Signature: (J)Ljava/lang/String;
788 */
789 JNIEXPORT jstring JNICALL Java_sun_awt_shell_Win32ShellFolder2_getFolderType
790 (JNIEnv* env, jclass cls, jlong pIDL)
791 {
792 SHFILEINFO fileInfo;
793 if (fn_SHGetFileInfo((LPCTSTR)pIDL, 0L, &fileInfo, sizeof(fileInfo),
794 SHGFI_TYPENAME | SHGFI_PIDL) == 0) {
795 return NULL;
796 }
797 return JNU_NewStringPlatform(env, fileInfo.szTypeName);
798 }
799
800 /*
801 * Class: sun_awt_shell_Win32ShellFolder2
802 * Method: getExecutableType
803 * Signature: (Ljava/lang/String;)Ljava/lang/String;
804 */
805 JNIEXPORT jstring JNICALL Java_sun_awt_shell_Win32ShellFolder2_getExecutableType
806 (JNIEnv* env, jobject folder, jstring path)
807 {
808 TCHAR szBuf[MAX_PATH];
809 LPCTSTR szPath = JNU_GetStringPlatformChars(env, path, NULL);
810 if (szPath == NULL) {
811 return NULL;
812 }
813 HINSTANCE res = fn_FindExecutable(szPath, szPath, szBuf);
814 JNU_ReleaseStringPlatformChars(env, path, szPath);
815 if ((UINT_PTR)res < 32) {
816 return NULL;
817 }
818 return JNU_NewStringPlatform(env, szBuf);
819 }
820
821
822 /*
823 * Class: sun_awt_shell_Win32ShellFolder2
824 * Method: getIcon
825 * Signature: (Ljava/lang/String;Z)J
826 */
827 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_getIcon
828 (JNIEnv* env, jclass cls, jstring absolutePath, jboolean getLargeIcon)
829 {
830 HICON hIcon = NULL;
831 SHFILEINFO fileInfo;
832 LPCTSTR pathStr = JNU_GetStringPlatformChars(env, absolutePath, NULL);
833 JNU_CHECK_EXCEPTION_RETURN(env, 0);
834 if (fn_SHGetFileInfo(pathStr, 0L, &fileInfo, sizeof(fileInfo),
835 SHGFI_ICON | (getLargeIcon ? 0 : SHGFI_SMALLICON)) != 0) {
836 hIcon = fileInfo.hIcon;
837 }
838 JNU_ReleaseStringPlatformChars(env, absolutePath, pathStr);
839 return (jlong)hIcon;
840 }
841
842 /*
843 * Class: sun_awt_shell_Win32ShellFolder2
844 * Method: getIconIndex
845 * Signature: (JJ)I
846 */
847 JNIEXPORT jint JNICALL Java_sun_awt_shell_Win32ShellFolder2_getIconIndex
848 (JNIEnv* env, jclass cls, jlong pIShellIconL, jlong relativePIDL)
849 {
850 IShellIcon* pIShellIcon = (IShellIcon*)pIShellIconL;
851 LPITEMIDLIST pidl = (LPITEMIDLIST)relativePIDL;
852 if (pIShellIcon == NULL && pidl == NULL) {
853 return 0;
854 }
855
856 INT index = -1;
857
858 HRESULT hres;
859 // http://msdn.microsoft.com/library/en-us/shellcc/platform/Shell/programmersguide/shell_int/shell_int_programming/std_ifaces.asp
860 if (pIShellIcon != NULL) {
861 hres = pIShellIcon->GetIconOf(pidl, GIL_FORSHELL, &index);
862 }
863
864 return (jint)index;
865 }
866
867
868 /*
869 * Class: sun_awt_shell_Win32ShellFolder2
870 * Method: extractIcon
871 * Signature: (JJZZ)J
872 */
873 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_extractIcon
874 (JNIEnv* env, jclass cls, jlong pIShellFolderL, jlong relativePIDL,
875 jboolean getLargeIcon, jboolean getDefaultIcon)
876 {
877 IShellFolder* pIShellFolder = (IShellFolder*)pIShellFolderL;
878 LPITEMIDLIST pidl = (LPITEMIDLIST)relativePIDL;
879 if (pIShellFolder == NULL || pidl == NULL) {
880 return 0;
881 }
882
883 HICON hIcon = NULL;
884
885 HRESULT hres;
886 IExtractIconW* pIcon;
887 hres = pIShellFolder->GetUIObjectOf(NULL, 1, const_cast<LPCITEMIDLIST*>(&pidl),
888 IID_IExtractIconW, NULL, (void**)&pIcon);
889 if (SUCCEEDED(hres)) {
890 WCHAR szBuf[MAX_PATH];
891 INT index;
892 UINT flags;
893 UINT uFlags = getDefaultIcon ? GIL_DEFAULTICON : GIL_FORSHELL | GIL_ASYNC;
894 hres = pIcon->GetIconLocation(uFlags, szBuf, MAX_PATH, &index, &flags);
895 if (SUCCEEDED(hres)) {
896 HICON hIconLarge;
897 hres = pIcon->Extract(szBuf, index, &hIconLarge, &hIcon, (16 << 16) + 32);
898 if (SUCCEEDED(hres)) {
899 if (getLargeIcon) {
900 fn_DestroyIcon((HICON)hIcon);
901 hIcon = hIconLarge;
902 } else {
903 fn_DestroyIcon((HICON)hIconLarge);
904 }
905 }
906 } else if (hres == E_PENDING) {
907 pIcon->Release();
908 return E_PENDING;
909 }
910 pIcon->Release();
911 }
912 return (jlong)hIcon;
913 }
914
915
916 /*
917 * Class: sun_awt_shell_Win32ShellFolder2
918 * Method: disposeIcon
919 * Signature: (J)V
920 */
921 JNIEXPORT void JNICALL Java_sun_awt_shell_Win32ShellFolder2_disposeIcon
922 (JNIEnv* env, jclass cls, jlong hicon)
923 {
924 fn_DestroyIcon((HICON)hicon);
925 }
926
927 /*
928 * Class: sun_awt_shell_Win32ShellFolder2
929 * Method: getIconBits
930 * Signature: (JI)[I
931 */
932 JNIEXPORT jintArray JNICALL Java_sun_awt_shell_Win32ShellFolder2_getIconBits
933 (JNIEnv* env, jclass cls, jlong hicon, jint iconSize)
934 {
935 jintArray iconBits = NULL;
936
937 // Get the icon info
938 ICONINFO iconInfo;
939 if (fn_GetIconInfo((HICON)hicon, &iconInfo)) {
940 // Get the screen DC
941 HDC dc = GetDC(NULL);
942 if (dc != NULL) {
943 // Set up BITMAPINFO
944 BITMAPINFO bmi;
945 memset(&bmi, 0, sizeof(BITMAPINFO));
946 bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
947 bmi.bmiHeader.biWidth = iconSize;
948 bmi.bmiHeader.biHeight = -iconSize;
949 bmi.bmiHeader.biPlanes = 1;
950 bmi.bmiHeader.biBitCount = 32;
951 bmi.bmiHeader.biCompression = BI_RGB;
952 // Extract the color bitmap
953 int nBits = iconSize * iconSize;
954 long colorBits[1024];
955 GetDIBits(dc, iconInfo.hbmColor, 0, iconSize, colorBits, &bmi, DIB_RGB_COLORS);
956 // XP supports alpha in some icons, and depending on device.
957 // This should take precedence over the icon mask bits.
958 BOOL hasAlpha = FALSE;
959 if (IS_WINXP) {
960 for (int i = 0; i < nBits; i++) {
961 if ((colorBits[i] & 0xff000000) != 0) {
962 hasAlpha = TRUE;
963 break;
964 }
965 }
966 }
967 if (!hasAlpha) {
968 // Extract the mask bitmap
969 long maskBits[1024];
970 GetDIBits(dc, iconInfo.hbmMask, 0, iconSize, maskBits, &bmi, DIB_RGB_COLORS);
971 // Copy the mask alphas into the color bits
972 for (int i = 0; i < nBits; i++) {
973 if (maskBits[i] == 0) {
974 colorBits[i] |= 0xff000000;
975 }
976 }
977 }
978 // Release DC
979 ReleaseDC(NULL, dc);
980 // Create java array
981 iconBits = env->NewIntArray(nBits);
982 if (!(env->ExceptionCheck())) {
983 // Copy values to java array
984 env->SetIntArrayRegion(iconBits, 0, nBits, colorBits);
985 }
986 }
987 // Fix 4745575 GDI Resource Leak
988 // MSDN
989 // GetIconInfo creates bitmaps for the hbmMask and hbmColor members of ICONINFO.
990 // The calling application must manage these bitmaps and delete them when they
991 // are no longer necessary.
992 ::DeleteObject(iconInfo.hbmColor);
993 ::DeleteObject(iconInfo.hbmMask);
994 }
995 return iconBits;
996 }
997
998 /*
999 * Class: sun_awt_shell_Win32ShellFolder2
1000 * Method: getStandardViewButton0
1001 * Signature: (I)[I
1002 */
1003 JNIEXPORT jintArray JNICALL Java_sun_awt_shell_Win32ShellFolder2_getStandardViewButton0
1004 (JNIEnv* env, jclass cls, jint iconIndex)
1005 {
1006 jintArray result = NULL;
1007
1008 // Create a toolbar
1009 HWND hWndToolbar = ::CreateWindowEx(0, TOOLBARCLASSNAME, NULL,
1010 0, 0, 0, 0, 0,
1011 NULL, NULL, NULL, NULL);
1012
1013 if (hWndToolbar != NULL) {
1014 SendMessage(hWndToolbar, TB_LOADIMAGES, (WPARAM)IDB_VIEW_SMALL_COLOR, (LPARAM)HINST_COMMCTRL);
1015
1016 HIMAGELIST hImageList = (HIMAGELIST) SendMessage(hWndToolbar, TB_GETIMAGELIST, 0, 0);
1017
1018 if (hImageList != NULL) {
1019 HICON hIcon = ImageList_GetIcon(hImageList, iconIndex, ILD_TRANSPARENT);
1020
1021 if (hIcon != NULL) {
1022 result = Java_sun_awt_shell_Win32ShellFolder2_getIconBits(env, cls, ptr_to_jlong(hIcon), 16);
1023
1024 DestroyIcon(hIcon);
1025 }
1026
1027 ImageList_Destroy(hImageList);
1028 }
1029
1030 DestroyWindow(hWndToolbar);
1031 }
1032
1033 return result;
1034 }
1035
1036 /*
1037 * Class: sun_awt_shell_Win32ShellFolder2
1038 * Method: getSystemIcon
1039 * Signature: (I)J
1040 */
1041 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_getSystemIcon
1042 (JNIEnv* env, jclass cls, jint iconID)
1043 {
1044 return (jlong)LoadIcon(NULL, MAKEINTRESOURCE(iconID));
1045 }
1046
1047
1048 /*
1049 * Class: sun_awt_shell_Win32ShellFolder2
1050 * Method: getIconResource
1051 * Signature: (Ljava/lang/String;IIIZ)J
1052 */
1053 JNIEXPORT jlong JNICALL Java_sun_awt_shell_Win32ShellFolder2_getIconResource
1054 (JNIEnv* env, jclass cls, jstring libName, jint iconID,
1055 jint cxDesired, jint cyDesired, jboolean useVGAColors)
1056 {
1057 const char *pLibName = env->GetStringUTFChars(libName, NULL);
1058 JNU_CHECK_EXCEPTION_RETURN(env, 0);
1059 HINSTANCE libHandle = (HINSTANCE)JDK_LoadSystemLibrary(pLibName);
1060 if (libHandle != NULL) {
1061 UINT fuLoad = (useVGAColors && !IS_WINXP) ? LR_VGACOLOR : 0;
1062 return ptr_to_jlong(LoadImage(libHandle, MAKEINTRESOURCE(iconID),
1063 IMAGE_ICON, cxDesired, cyDesired,
1064 fuLoad));
1065 }
1066 return 0;
1067 }
1068
1069
1070 /*
1071 * Helper function for creating Java column info object
1072 */
1073 static jobject CreateColumnInfo(JNIEnv *pEnv,
1074 jclass *pClass, jmethodID *pConstructor,
1075 SHELLDETAILS *psd, ULONG visible)
1076 {
1077 jstring str = jstringFromSTRRET(pEnv, NULL, &(psd->str));
1078 JNU_CHECK_EXCEPTION_RETURN(pEnv, NULL);
1079
1080 return pEnv->NewObject(*pClass, *pConstructor,
1081 str,
1082 (jint)(psd->cxChar * 6), // TODO: is 6 OK for converting chars to pixels?
1083 (jint)psd->fmt, (jboolean) visible);
1084 }
1085
1086
1087 /*
1088 * Class: sun_awt_shell_Win32ShellFolder2
1089 * Method: doGetColumnInfo
1090 * Signature: (J)[Lsun/awt/shell/ShellFolderColumnInfo;
1091 */
1092 JNIEXPORT jobjectArray JNICALL
1093 Java_sun_awt_shell_Win32ShellFolder2_doGetColumnInfo
1094 (JNIEnv *env, jobject obj, jlong iShellFolder)
1095 {
1096
1097 HRESULT hr;
1098 IShellFolder *pIShellFolder = (IShellFolder*) iShellFolder;
1099 IUnknown *pIUnknown = NULL;
1100
1101 jclass columnClass = env->FindClass("sun/awt/shell/ShellFolderColumnInfo");
1102 if(NULL == columnClass) {
1103 return NULL;
1104 }
1105
1106 jmethodID columnConstructor =
1107 env->GetMethodID(columnClass, "<init>", "(Ljava/lang/String;IIZ)V");
1108 if(NULL == columnConstructor) {
1109 return NULL;
1110 }
1111
1112 // We'are asking the object the list of available columns
1113 SHELLDETAILS sd;
1114
1115 hr = pIShellFolder->QueryInterface(IID_IShellFolder2, (void**)&pIUnknown);
1116 if(SUCCEEDED (hr)) {
1117
1118 // The folder exposes IShellFolder2 interface
1119 IShellFolder2 *pIShellFolder2 = (IShellFolder2*) pIUnknown;
1120
1121 // Count columns
1122 int colNum = -1;
1123 hr = S_OK;
1124 do{
1125 hr = pIShellFolder2->GetDetailsOf(NULL, ++colNum, &sd);
1126 } while (SUCCEEDED (hr));
1127
1128 jobjectArray columns =
1129 env->NewObjectArray((jsize) colNum, columnClass, NULL);
1130 if(NULL == columns) {
1131 pIShellFolder2->Release();
1132 return NULL;
1133 }
1134
1135 // Fill column details list
1136 SHCOLSTATEF csFlags;
1137 colNum = 0;
1138 hr = S_OK;
1139 while (SUCCEEDED (hr)) {
1140 hr = pIShellFolder2->GetDetailsOf(NULL, colNum, &sd);
1141
1142 if (SUCCEEDED (hr)) {
1143 hr = pIShellFolder2->GetDefaultColumnState(colNum, &csFlags);
1144 if (SUCCEEDED (hr)) {
1145 if(!(csFlags & SHCOLSTATE_HIDDEN)) {
1146 jobject column = CreateColumnInfo(env,
1147 &columnClass, &columnConstructor,
1148 &sd, csFlags & SHCOLSTATE_ONBYDEFAULT);
1149 if(!column){
1150 pIShellFolder2->Release();
1151 return NULL;
1152 }
1153 env->SetObjectArrayElement(columns, (jsize) colNum, column);
1154 }
1155 }
1156 colNum++;
1157 }
1158 }
1159
1160 pIShellFolder2->Release();
1161
1162 return columns;
1163 }
1164
1165 hr = pIShellFolder->CreateViewObject(NULL, IID_IShellDetails, (void**)&pIUnknown);
1166 if(SUCCEEDED (hr)) {
1167 // The folder exposes IShellDetails interface
1168 IShellDetails *pIShellDetails = (IShellDetails*) pIUnknown;
1169
1170 // Count columns
1171 int colNum = -1;
1172 hr = S_OK;
1173 do{
1174 hr = pIShellDetails->GetDetailsOf(NULL, ++colNum, &sd);
1175 } while (SUCCEEDED (hr));
1176
1177 jobjectArray columns =
1178 env->NewObjectArray((jsize) colNum, columnClass, NULL);
1179 if(NULL == columns) {
1180 pIShellDetails->Release();
1181 return NULL;
1182 }
1183
1184 // Fill column details list
1185 colNum = 0;
1186 hr = S_OK;
1187 while (SUCCEEDED (hr)) {
1188 hr = pIShellDetails->GetDetailsOf(NULL, colNum, &sd);
1189 if (SUCCEEDED (hr)) {
1190 jobject column = CreateColumnInfo(env,
1191 &columnClass, &columnConstructor,
1192 &sd, 1);
1193 if(!column){
1194 pIShellDetails->Release();
1195 return NULL;
1196 }
1197 env->SetObjectArrayElement(columns, (jsize) colNum++, column);
1198 }
1199 }
1200
1201 pIShellDetails->Release();
1202
1203 return columns;
1204 }
1205
1206 // The folder exposes neither IShellFolder2 nor IShelDetails
1207 return NULL;
1208
1209 }
1210
1211 /*
1212 * Class: sun_awt_shell_Win32ShellFolder2
1213 * Method: doGetColumnValue
1214 * Signature: (JJI)Ljava/lang/Object;
1215 */
1216 JNIEXPORT jobject JNICALL
1217 Java_sun_awt_shell_Win32ShellFolder2_doGetColumnValue
1218 (JNIEnv *env, jobject obj, jlong iShellFolder,
1219 jlong jpidl, jint columnIdx)
1220 {
1221
1222 HRESULT hr;
1223 IShellFolder *pIShellFolder = (IShellFolder*) iShellFolder;
1224 IUnknown *pIUnknown = NULL;
1225
1226
1227 LPITEMIDLIST pidl = (LPITEMIDLIST) jpidl;
1228 SHELLDETAILS sd;
1229
1230 hr = pIShellFolder->QueryInterface(IID_IShellFolder2, (void**)&pIUnknown);
1231 if(SUCCEEDED (hr)) {
1232 // The folder exposes IShellFolder2 interface
1233 IShellFolder2 *pIShellFolder2 = (IShellFolder2*) pIUnknown;
1234 hr = pIShellFolder2->GetDetailsOf(pidl, (UINT)columnIdx, &sd);
1235 pIShellFolder2->Release();
1236 if (SUCCEEDED (hr)) {
1237 STRRET strRet = sd.str;
1238 return jstringFromSTRRET(env, pidl, &strRet);
1239 }
1240 }
1241
1242 hr = pIShellFolder->CreateViewObject(NULL, IID_IShellDetails, (void**)&pIUnknown);
1243 if(SUCCEEDED (hr)) {
1244 // The folder exposes IShellDetails interface
1245 IShellDetails *pIShellDetails = (IShellDetails*) pIUnknown;
1246 hr = pIShellDetails->GetDetailsOf(pidl, (UINT)columnIdx, &sd);
1247 pIShellDetails->Release();
1248 if (SUCCEEDED (hr)) {
1249 STRRET strRet = sd.str;
1250 return jstringFromSTRRET(env, pidl, &strRet);
1251 }
1252 }
1253
1254 // The folder exposes neither IShellFolder2 nor IShelDetails
1255 return NULL;
1256 }
1257
1258 /*
1259 * Class: sun_awt_shell_Win32ShellFolder2
1260 * Method: compareIDsByColumn
1261 * Signature: (JJJI)I
1262 */
1263 JNIEXPORT jint JNICALL
1264 Java_sun_awt_shell_Win32ShellFolder2_compareIDsByColumn
1265 (JNIEnv* env, jclass cls, jlong jpParentIShellFolder,
1266 jlong pIDL1, jlong pIDL2, jint columnIdx)
1267 {
1268 IShellFolder* pParentIShellFolder = (IShellFolder*)jpParentIShellFolder;
1269 if (pParentIShellFolder == NULL) {
1270 return 0;
1271 }
1272
1273 HRESULT hr = pParentIShellFolder->CompareIDs(
1274 (UINT) columnIdx,
1275 (LPCITEMIDLIST) pIDL1,
1276 (LPCITEMIDLIST) pIDL2);
1277 if (SUCCEEDED (hr)) {
1278 return (jint) (short) HRESULT_CODE(hr);
1279 }
1280
1281 return 0;
1282 }
1283
1284 /*
1285 * Class: sun_awt_shell_Win32ShellFolder2
1286 * Method: loadKnownFolders
1287 * Signature: (V)[BLsun/awt/shell/Win32ShellFolder2$KnownfolderDefenition;
1288 */
1289 JNIEXPORT jobjectArray JNICALL Java_sun_awt_shell_Win32ShellFolder2_loadKnownFolders
1290 (JNIEnv* env, jclass cls )
1291 {
1292 IKnownFolderManager* pkfm = NULL;
1293 HRESULT hr = CoCreateInstance(CLSID_KnownFolderManager, NULL,
1294 CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&pkfm));
1295 if (!SUCCEEDED(hr)) return NULL;
1296
1297 TRY;
1298
1299 jclass cl = env->FindClass("sun/awt/shell/Win32ShellFolder2$KnownFolderDefinition");
1300 CHECK_NULL_RETURN(cl, NULL);
1301 DEFINE_FIELD_ID(field_guid, cl, "guid", "Ljava/lang/String;")
1302 DEFINE_FIELD_ID(field_name, cl, "name", "Ljava/lang/String;");
1303 DEFINE_FIELD_ID(field_description, cl, "description", "Ljava/lang/String;");
1304 DEFINE_FIELD_ID(field_parent, cl, "parent", "Ljava/lang/String;");
1305 DEFINE_FIELD_ID(field_relativePath, cl, "relativePath", "Ljava/lang/String;");
1306 DEFINE_FIELD_ID(field_parsingName, cl, "parsingName", "Ljava/lang/String;");
1307 DEFINE_FIELD_ID(field_tooltip, cl, "tooltip", "Ljava/lang/String;");
1308 DEFINE_FIELD_ID(field_localizedName, cl, "localizedName", "Ljava/lang/String;");
1309 DEFINE_FIELD_ID(field_icon, cl, "icon", "Ljava/lang/String;");
1310 DEFINE_FIELD_ID(field_security, cl, "security", "Ljava/lang/String;");
1311 DEFINE_FIELD_ID(field_path, cl, "path", "Ljava/lang/String;");
1312 DEFINE_FIELD_ID(field_saveLocation, cl, "saveLocation", "Ljava/lang/String;");
1313 DEFINE_FIELD_ID(field_category, cl, "category", "I");
1314 DEFINE_FIELD_ID(field_attributes, cl, "attributes", "J");
1315 DEFINE_FIELD_ID(field_defenitionFlags, cl, "defenitionFlags", "I");
1316 DEFINE_FIELD_ID(field_ftidType, cl, "ftidType", "Ljava/lang/String;");
1317
1318 jobjectArray result;
1319 KNOWNFOLDERID* pFoldersIds = NULL;
1320 UINT count = 0;
1321 if (SUCCEEDED(pkfm->GetFolderIds(&pFoldersIds, &count))) {
1322 jmethodID initMethod;
1323 try {
1324 result = env->NewObjectArray(count, cl, NULL);
1325 initMethod = env->GetMethodID(cl, "<init>", "()V");
1326 EXCEPTION_CHECK
1327 } catch (std::bad_alloc&) {
1328 CoTaskMemFree(pFoldersIds);
1329 pkfm->Release();
1330 throw;
1331 }
1332 for(UINT i = 0; i < count; ++i)
1333 {
1334 jobject fld;
1335 const KNOWNFOLDERID& folderId = pFoldersIds[i];
1336 LPOLESTR guid = NULL;
1337 try {
1338 fld = env->NewObject(cl, initMethod);
1339 if (fld) {
1340 env->SetObjectArrayElement(result, i, fld);
1341 }
1342 EXCEPTION_CHECK
1343
1344 if (SUCCEEDED(StringFromCLSID(folderId, &guid))) {
1345 jstring jstr = JNU_NewStringPlatform(env, guid);
1346 if (jstr) {
1347 env->SetObjectField(fld, field_guid, jstr);
1348 }
1349 CoTaskMemFree(guid);
1350 EXCEPTION_CHECK
1351 }
1352 } catch (std::bad_alloc&) {
1353 CoTaskMemFree(pFoldersIds);
1354 pkfm->Release();
1355 throw;
1356 }
1357
1358 IKnownFolder* pFolder = NULL;
1359 if (SUCCEEDED(pkfm->GetFolder(folderId, &pFolder))) {
1360 KNOWNFOLDER_DEFINITION kfDef;
1361 if (SUCCEEDED(pFolder->GetFolderDefinition(&kfDef)))
1362 {
1363 try {
1364 jstring jstr = JNU_NewStringPlatform(env, kfDef.pszName);
1365 if(jstr) {
1366 env->SetObjectField(fld, field_name, jstr);
1367 }
1368 EXCEPTION_CHECK
1369 if (kfDef.pszDescription) {
1370 jstr = JNU_NewStringPlatform(env, kfDef.pszDescription);
1371 if (jstr) {
1372 env->SetObjectField(fld, field_description, jstr);
1373 }
1374 EXCEPTION_CHECK
1375 }
1376 EXCEPTION_CHECK
1377 if (SUCCEEDED(StringFromCLSID(kfDef.fidParent, &guid))) {
1378 jstr = JNU_NewStringPlatform(env, guid);
1379 if (jstr) {
1380 env->SetObjectField(fld, field_parent, jstr);
1381 }
1382 CoTaskMemFree(guid);
1383 EXCEPTION_CHECK
1384 }
1385 if (kfDef.pszRelativePath) {
1386 jstr = JNU_NewStringPlatform(env, kfDef.pszRelativePath);
1387 if (jstr) {
1388 env->SetObjectField(fld, field_relativePath, jstr);
1389 }
1390 EXCEPTION_CHECK
1391 }
1392 if (kfDef.pszParsingName) {
1393 jstr = JNU_NewStringPlatform(env, kfDef.pszParsingName);
1394 if (jstr) {
1395 env->SetObjectField(fld, field_parsingName, jstr);
1396 }
1397 EXCEPTION_CHECK
1398 }
1399 if (kfDef.pszTooltip) {
1400 jstr = JNU_NewStringPlatform(env, kfDef.pszTooltip);
1401 if (jstr) {
1402 env->SetObjectField(fld, field_tooltip, jstr);
1403 }
1404 EXCEPTION_CHECK
1405 }
1406 if (kfDef.pszLocalizedName) {
1407 jstr = JNU_NewStringPlatform(env, kfDef.pszLocalizedName);
1408 if (jstr) {
1409 env->SetObjectField(fld, field_localizedName, jstr);
1410 }
1411 EXCEPTION_CHECK
1412 }
1413 if (kfDef.pszIcon) {
1414 jstr = JNU_NewStringPlatform(env, kfDef.pszIcon);
1415 if (jstr) {
1416 env->SetObjectField(fld, field_icon, jstr);
1417 }
1418 EXCEPTION_CHECK
1419 }
1420 if (kfDef.pszSecurity) {
1421 jstr = JNU_NewStringPlatform(env, kfDef.pszSecurity);
1422 if (jstr) {
1423 env->SetObjectField(fld, field_security, jstr);
1424 }
1425 EXCEPTION_CHECK
1426 }
1427 if (SUCCEEDED(StringFromCLSID(kfDef.ftidType, &guid))) {
1428 jstr = JNU_NewStringPlatform(env, guid);
1429 if (jstr) {
1430 env->SetObjectField(fld, field_ftidType, jstr);
1431 }
1432 CoTaskMemFree(guid);
1433 EXCEPTION_CHECK
1434 }
1435 env->SetIntField(fld, field_category, kfDef.category);
1436 env->SetIntField(fld, field_defenitionFlags, kfDef.kfdFlags);
1437 env->SetLongField(fld, field_attributes, kfDef.dwAttributes);
1438
1439 LPWSTR folderPath = NULL;
1440 if (SUCCEEDED(pFolder->GetPath(KF_FLAG_NO_ALIAS, &folderPath))
1441 && folderPath) {
1442 jstr = JNU_NewStringPlatform(env, folderPath);
1443 if (jstr) {
1444 env->SetObjectField(fld, field_path, jstr);
1445 }
1446 CoTaskMemFree(folderPath);
1447 EXCEPTION_CHECK
1448 }
1449
1450 IShellLibrary *plib = NULL;
1451 hr = CoCreateInstance(CLSID_ShellLibrary, NULL,
1452 CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&plib));
1453 if (SUCCEEDED(hr)) {
1454 hr = plib->LoadLibraryFromKnownFolder(folderId, STGM_READWRITE);
1455 if (SUCCEEDED(hr)) {
1456 IShellItem *item = NULL;
1457 hr = plib->GetDefaultSaveFolder(DSFT_DETECT,
1458 IID_PPV_ARGS(&item));
1459 if (SUCCEEDED(hr) && item) {
1460 LPWSTR loc = NULL;
1461 hr = item->GetDisplayName(SIGDN_FILESYSPATH, &loc);
1462 if (SUCCEEDED(hr) && loc)
1463 {
1464 jstr = JNU_NewStringPlatform(env, loc);
1465 if (jstr) {
1466 env->SetObjectField(fld, field_saveLocation, jstr);
1467 }
1468 CoTaskMemFree(loc);
1469 }
1470 item->Release();
1471 }
1472 }
1473 plib->Release();
1474 EXCEPTION_CHECK
1475 }
1476 FreeKnownFolderDefinitionFields(&kfDef);
1477 } catch (std::bad_alloc&) {
1478 FreeKnownFolderDefinitionFields(&kfDef);
1479 pFolder->Release();
1480 CoTaskMemFree(pFoldersIds);
1481 pkfm->Release();
1482 throw;
1483 }
1484 }
1485 }
1486 pFolder->Release();
1487 }
1488 CoTaskMemFree(pFoldersIds);
1489 }
1490 pkfm->Release();
1491 return result;
1492 CATCH_BAD_ALLOC_RET(NULL);
1493 }
1494
1495 } // extern "C"