/[pcsx2_0.9.7]/branch/r3113_0.9.7_beta/3rdparty/wxWidgets/src/msw/volume.cpp
ViewVC logotype

Contents of /branch/r3113_0.9.7_beta/3rdparty/wxWidgets/src/msw/volume.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 32 - (show annotations) (download)
Tue Sep 7 03:29:01 2010 UTC (9 years, 11 months ago) by william
File size: 20682 byte(s)
branching from upstream revision (http://pcsx2.googlecode.com/svn/trunk
): r3113 to
https://svn.netsolutions.dnsalias.com/websvn/ps2/pcsx2/pcsx2_0.9.7/branch/r3113_0.9.7_beta
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: src/msw/volume.cpp
3 // Purpose: wxFSVolume - encapsulates system volume information
4 // Author: George Policello
5 // Modified by:
6 // Created: 28 Jan 02
7 // RCS-ID: $Id: volume.cpp 41903 2006-10-10 17:37:02Z PC $
8 // Copyright: (c) 2002 George Policello
9 // Licence: wxWindows licence
10 ///////////////////////////////////////////////////////////////////////////////
11
12 // ============================================================================
13 // declarations
14 // ============================================================================
15
16 // ----------------------------------------------------------------------------
17 // headers
18 // ----------------------------------------------------------------------------
19
20 #include "wx/wxprec.h"
21
22 #ifdef __BORLANDC__
23 #pragma hdrstop
24 #endif
25
26 #if wxUSE_FSVOLUME
27
28 #include "wx/volume.h"
29
30 #ifndef WX_PRECOMP
31 #if wxUSE_GUI
32 #include "wx/icon.h"
33 #endif
34 #include "wx/intl.h"
35 #include "wx/hashmap.h"
36 #endif // WX_PRECOMP
37
38 #include "wx/dir.h"
39 #include "wx/dynlib.h"
40 #include "wx/arrimpl.cpp"
41
42 #include <shellapi.h>
43 #include <shlobj.h>
44 #include "wx/msw/missing.h"
45
46 #if wxUSE_BASE
47
48 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
49 // Dynamic library function defs.
50 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
51
52 #if wxUSE_DYNLIB_CLASS
53 static wxDynamicLibrary s_mprLib;
54 #endif
55
56 typedef DWORD (WINAPI* WNetOpenEnumPtr)(DWORD, DWORD, DWORD, LPNETRESOURCE, LPHANDLE);
57 typedef DWORD (WINAPI* WNetEnumResourcePtr)(HANDLE, LPDWORD, LPVOID, LPDWORD);
58 typedef DWORD (WINAPI* WNetCloseEnumPtr)(HANDLE);
59
60 static WNetOpenEnumPtr s_pWNetOpenEnum;
61 static WNetEnumResourcePtr s_pWNetEnumResource;
62 static WNetCloseEnumPtr s_pWNetCloseEnum;
63
64 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
65 // Globals/Statics
66 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
67 static long s_cancelSearch = FALSE;
68
69 struct FileInfo
70 {
71 FileInfo(unsigned flag=0, wxFSVolumeKind type=wxFS_VOL_OTHER) :
72 m_flags(flag), m_type(type) {}
73
74 FileInfo(const FileInfo& other) { *this = other; }
75 FileInfo& operator=(const FileInfo& other)
76 {
77 m_flags = other.m_flags;
78 m_type = other.m_type;
79 return *this;
80 }
81
82 unsigned m_flags;
83 wxFSVolumeKind m_type;
84 };
85 WX_DECLARE_STRING_HASH_MAP(FileInfo, FileInfoMap);
86 // Cygwin bug (?) destructor for global s_fileInfo is called twice...
87 static FileInfoMap& GetFileInfoMap()
88 {
89 static FileInfoMap s_fileInfo(25);
90
91 return s_fileInfo;
92 }
93 #define s_fileInfo (GetFileInfoMap())
94
95 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
96 // Local helper functions.
97 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
98
99 //=============================================================================
100 // Function: GetBasicFlags
101 // Purpose: Set basic flags, primarily wxFS_VOL_REMOTE and wxFS_VOL_REMOVABLE.
102 // Notes: - Local and mapped drives are mounted by definition. We have no
103 // way to determine mounted status of network drives, so assume that
104 // all drives are mounted, and let the caller decide otherwise.
105 // - Other flags are 'best guess' from type of drive. The system will
106 // not report the file attributes with any degree of accuracy.
107 //=============================================================================
108 static unsigned GetBasicFlags(const wxChar* filename)
109 {
110 unsigned flags = wxFS_VOL_MOUNTED;
111
112 //----------------------------------
113 // 'Best Guess' based on drive type.
114 //----------------------------------
115 wxFSVolumeKind type;
116 switch(GetDriveType(filename))
117 {
118 case DRIVE_FIXED:
119 type = wxFS_VOL_DISK;
120 break;
121
122 case DRIVE_REMOVABLE:
123 flags |= wxFS_VOL_REMOVABLE;
124 type = wxFS_VOL_FLOPPY;
125 break;
126
127 case DRIVE_CDROM:
128 flags |= wxFS_VOL_REMOVABLE | wxFS_VOL_READONLY;
129 type = wxFS_VOL_CDROM;
130 break;
131
132 case DRIVE_REMOTE:
133 flags |= wxFS_VOL_REMOTE;
134 type = wxFS_VOL_NETWORK;
135 break;
136
137 case DRIVE_NO_ROOT_DIR:
138 flags &= ~wxFS_VOL_MOUNTED;
139 type = wxFS_VOL_OTHER;
140 break;
141
142 default:
143 type = wxFS_VOL_OTHER;
144 break;
145 }
146
147 //-----------------------------------------------------------------------
148 // The following most likely will not modify anything not set above,
149 // and will not work at all for network shares or empty CD ROM drives.
150 // But it is a good check if the Win API ever gets better about reporting
151 // this information.
152 //-----------------------------------------------------------------------
153 SHFILEINFO fi;
154 long rc;
155 rc = SHGetFileInfo(filename, 0, &fi, sizeof(fi), SHGFI_ATTRIBUTES );
156 if (!rc)
157 {
158 wxLogError(_("Cannot read typename from '%s'!"), filename);
159 }
160 else
161 {
162 if (fi.dwAttributes & SFGAO_READONLY)
163 flags |= wxFS_VOL_READONLY;
164 if (fi.dwAttributes & SFGAO_REMOVABLE)
165 flags |= wxFS_VOL_REMOVABLE;
166 }
167
168 //------------------
169 // Flags are cached.
170 //------------------
171 s_fileInfo[filename] = FileInfo(flags, type);
172
173 return flags;
174 } // GetBasicFlags
175
176 //=============================================================================
177 // Function: FilteredAdd
178 // Purpose: Add a file to the list if it meets the filter requirement.
179 // Notes: - See GetBasicFlags for remarks about the Mounted flag.
180 //=============================================================================
181 static bool FilteredAdd(wxArrayString& list, const wxChar* filename,
182 unsigned flagsSet, unsigned flagsUnset)
183 {
184 bool accept = true;
185 unsigned flags = GetBasicFlags(filename);
186
187 if (flagsSet & wxFS_VOL_MOUNTED && !(flags & wxFS_VOL_MOUNTED))
188 accept = false;
189 else if (flagsUnset & wxFS_VOL_MOUNTED && (flags & wxFS_VOL_MOUNTED))
190 accept = false;
191 else if (flagsSet & wxFS_VOL_REMOVABLE && !(flags & wxFS_VOL_REMOVABLE))
192 accept = false;
193 else if (flagsUnset & wxFS_VOL_REMOVABLE && (flags & wxFS_VOL_REMOVABLE))
194 accept = false;
195 else if (flagsSet & wxFS_VOL_READONLY && !(flags & wxFS_VOL_READONLY))
196 accept = false;
197 else if (flagsUnset & wxFS_VOL_READONLY && (flags & wxFS_VOL_READONLY))
198 accept = false;
199 else if (flagsSet & wxFS_VOL_REMOTE && !(flags & wxFS_VOL_REMOTE))
200 accept = false;
201 else if (flagsUnset & wxFS_VOL_REMOTE && (flags & wxFS_VOL_REMOTE))
202 accept = false;
203
204 // Add to the list if passed the filter.
205 if (accept)
206 list.Add(filename);
207
208 return accept;
209 } // FilteredAdd
210
211 //=============================================================================
212 // Function: BuildListFromNN
213 // Purpose: Append or remove items from the list
214 // Notes: - There is no way to find all disconnected NN items, or even to find
215 // all items while determining which are connected and not. So this
216 // function will find either all items or connected items.
217 //=============================================================================
218 static void BuildListFromNN(wxArrayString& list, NETRESOURCE* pResSrc,
219 unsigned flagsSet, unsigned flagsUnset)
220 {
221 HANDLE hEnum;
222 int rc;
223
224 //-----------------------------------------------
225 // Scope may be all drives or all mounted drives.
226 //-----------------------------------------------
227 unsigned scope = RESOURCE_GLOBALNET;
228 if (flagsSet & wxFS_VOL_MOUNTED)
229 scope = RESOURCE_CONNECTED;
230
231 //----------------------------------------------------------------------
232 // Enumerate all items, adding only non-containers (ie. network shares).
233 // Containers cause a recursive call to this function for their own
234 // enumeration.
235 //----------------------------------------------------------------------
236 if (rc = s_pWNetOpenEnum(scope, RESOURCETYPE_DISK, 0, pResSrc, &hEnum), rc == NO_ERROR)
237 {
238 DWORD count = 1;
239 DWORD size = 256;
240 NETRESOURCE* pRes = (NETRESOURCE*)malloc(size);
241 memset(pRes, 0, sizeof(NETRESOURCE));
242 while (rc = s_pWNetEnumResource(hEnum, &count, pRes, &size), rc == NO_ERROR || rc == ERROR_MORE_DATA)
243 {
244 if (s_cancelSearch)
245 break;
246
247 if (rc == ERROR_MORE_DATA)
248 {
249 pRes = (NETRESOURCE*)realloc(pRes, size);
250 count = 1;
251 }
252 else if (count == 1)
253 {
254 // Enumerate the container.
255 if (pRes->dwUsage & RESOURCEUSAGE_CONTAINER)
256 {
257 BuildListFromNN(list, pRes, flagsSet, flagsUnset);
258 }
259
260 // Add the network share.
261 else
262 {
263 wxString filename(pRes->lpRemoteName);
264
265 if (!filename.empty())
266 {
267 if (filename.Last() != '\\')
268 filename.Append('\\');
269
270 // The filter function will not know mounted from unmounted, and neither do we unless
271 // we are iterating using RESOURCE_CONNECTED, in which case they all are mounted.
272 // Volumes on disconnected servers, however, will correctly show as unmounted.
273 FilteredAdd(list, filename, flagsSet, flagsUnset&~wxFS_VOL_MOUNTED);
274 if (scope == RESOURCE_GLOBALNET)
275 s_fileInfo[filename].m_flags &= ~wxFS_VOL_MOUNTED;
276 }
277 }
278 }
279 else if (count == 0)
280 break;
281 }
282 free(pRes);
283 s_pWNetCloseEnum(hEnum);
284 }
285 } // BuildListFromNN
286
287 //=============================================================================
288 // Function: CompareFcn
289 // Purpose: Used to sort the NN list alphabetically, case insensitive.
290 //=============================================================================
291 static int CompareFcn(const wxString& first, const wxString& second)
292 {
293 return wxStricmp(first.c_str(), second.c_str());
294 } // CompareFcn
295
296 //=============================================================================
297 // Function: BuildRemoteList
298 // Purpose: Append Network Neighborhood items to the list.
299 // Notes: - Mounted gets transalated into Connected. FilteredAdd is told
300 // to ignore the Mounted flag since we need to handle it in a weird
301 // way manually.
302 // - The resulting list is sorted alphabetically.
303 //=============================================================================
304 static bool BuildRemoteList(wxArrayString& list, NETRESOURCE* pResSrc,
305 unsigned flagsSet, unsigned flagsUnset)
306 {
307 // NN query depends on dynamically loaded library.
308 if (!s_pWNetOpenEnum || !s_pWNetEnumResource || !s_pWNetCloseEnum)
309 {
310 wxLogError(_("Failed to load mpr.dll."));
311 return false;
312 }
313
314 // Don't waste time doing the work if the flags conflict.
315 if (flagsSet & wxFS_VOL_MOUNTED && flagsUnset & wxFS_VOL_MOUNTED)
316 return false;
317
318 //----------------------------------------------
319 // Generate the list according to the flags set.
320 //----------------------------------------------
321 BuildListFromNN(list, pResSrc, flagsSet, flagsUnset);
322 list.Sort(CompareFcn);
323
324 //-------------------------------------------------------------------------
325 // If mounted only is requested, then we only need one simple pass.
326 // Otherwise, we need to build a list of all NN volumes and then apply the
327 // list of mounted drives to it.
328 //-------------------------------------------------------------------------
329 if (!(flagsSet & wxFS_VOL_MOUNTED))
330 {
331 // generate.
332 wxArrayString mounted;
333 BuildListFromNN(mounted, pResSrc, flagsSet | wxFS_VOL_MOUNTED, flagsUnset & ~wxFS_VOL_MOUNTED);
334 mounted.Sort(CompareFcn);
335
336 // apply list from bottom to top to preserve indexes if removing items.
337 ssize_t iList = list.GetCount()-1;
338 for (ssize_t iMounted = mounted.GetCount()-1; iMounted >= 0 && iList >= 0; iMounted--)
339 {
340 int compare;
341 wxString all(list[iList]);
342 wxString mount(mounted[iMounted]);
343
344 while (compare =
345 wxStricmp(list[iList].c_str(), mounted[iMounted].c_str()),
346 compare > 0 && iList >= 0)
347 {
348 iList--;
349 all = list[iList];
350 }
351
352
353 if (compare == 0)
354 {
355 // Found the element. Remove it or mark it mounted.
356 if (flagsUnset & wxFS_VOL_MOUNTED)
357 list.RemoveAt(iList);
358 else
359 s_fileInfo[list[iList]].m_flags |= wxFS_VOL_MOUNTED;
360
361 }
362
363 iList--;
364 }
365 }
366
367 return true;
368 } // BuildRemoteList
369
370 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
371 // wxFSVolume
372 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
373
374 //=============================================================================
375 // Function: GetVolumes
376 // Purpose: Generate and return a list of all volumes (drives) available.
377 // Notes:
378 //=============================================================================
379 wxArrayString wxFSVolumeBase::GetVolumes(int flagsSet, int flagsUnset)
380 {
381 ::InterlockedExchange(&s_cancelSearch, FALSE); // reset
382
383 #if wxUSE_DYNLIB_CLASS
384 if (!s_mprLib.IsLoaded() && s_mprLib.Load(_T("mpr.dll")))
385 {
386 #ifdef UNICODE
387 s_pWNetOpenEnum = (WNetOpenEnumPtr)s_mprLib.GetSymbol(_T("WNetOpenEnumW"));
388 s_pWNetEnumResource = (WNetEnumResourcePtr)s_mprLib.GetSymbol(_T("WNetEnumResourceW"));
389 #else
390 s_pWNetOpenEnum = (WNetOpenEnumPtr)s_mprLib.GetSymbol(_T("WNetOpenEnumA"));
391 s_pWNetEnumResource = (WNetEnumResourcePtr)s_mprLib.GetSymbol(_T("WNetEnumResourceA"));
392 #endif
393 s_pWNetCloseEnum = (WNetCloseEnumPtr)s_mprLib.GetSymbol(_T("WNetCloseEnum"));
394 }
395 #endif
396
397 wxArrayString list;
398
399 //-------------------------------
400 // Local and mapped drives first.
401 //-------------------------------
402 // Allocate the required space for the API call.
403 const DWORD chars = GetLogicalDriveStrings(0, NULL);
404 TCHAR* buf = new TCHAR[chars+1];
405
406 // Get the list of drives.
407 GetLogicalDriveStrings(chars, buf);
408
409 // Parse the list into an array, applying appropriate filters.
410 TCHAR *pVol;
411 pVol = buf;
412 while (*pVol)
413 {
414 FilteredAdd(list, pVol, flagsSet, flagsUnset);
415 pVol = pVol + wxStrlen(pVol) + 1;
416 }
417
418 // Cleanup.
419 delete[] buf;
420
421 //---------------------------
422 // Network Neighborhood next.
423 //---------------------------
424
425 // not exclude remote and not removable
426 if (!(flagsUnset & wxFS_VOL_REMOTE) &&
427 !(flagsSet & wxFS_VOL_REMOVABLE)
428 )
429 {
430 // The returned list will be sorted alphabetically. We don't pass
431 // our in since we don't want to change to order of the local drives.
432 wxArrayString nn;
433 if (BuildRemoteList(nn, 0, flagsSet, flagsUnset))
434 {
435 for (size_t idx = 0; idx < nn.GetCount(); idx++)
436 list.Add(nn[idx]);
437 }
438 }
439
440 return list;
441 } // GetVolumes
442
443 //=============================================================================
444 // Function: CancelSearch
445 // Purpose: Instruct an active search to stop.
446 // Notes: - This will only sensibly be called by a thread other than the one
447 // performing the search. This is the only thread-safe function
448 // provided by the class.
449 //=============================================================================
450 void wxFSVolumeBase::CancelSearch()
451 {
452 ::InterlockedExchange(&s_cancelSearch, TRUE);
453 } // CancelSearch
454
455 //=============================================================================
456 // Function: constructor
457 // Purpose: default constructor
458 //=============================================================================
459 wxFSVolumeBase::wxFSVolumeBase()
460 {
461 m_isOk = false;
462 } // wxVolume
463
464 //=============================================================================
465 // Function: constructor
466 // Purpose: constructor that calls Create
467 //=============================================================================
468 wxFSVolumeBase::wxFSVolumeBase(const wxString& name)
469 {
470 Create(name);
471 } // wxVolume
472
473 //=============================================================================
474 // Function: Create
475 // Purpose: Finds, logs in, etc. to the request volume.
476 //=============================================================================
477 bool wxFSVolumeBase::Create(const wxString& name)
478 {
479 // assume fail.
480 m_isOk = false;
481
482 // supplied.
483 m_volName = name;
484
485 // Display name.
486 SHFILEINFO fi;
487 long rc = SHGetFileInfo(m_volName, 0, &fi, sizeof(fi), SHGFI_DISPLAYNAME);
488 if (!rc)
489 {
490 wxLogError(_("Cannot read typename from '%s'!"), m_volName.c_str());
491 return m_isOk;
492 }
493 m_dispName = fi.szDisplayName;
494
495 // all tests passed.
496 return m_isOk = true;
497 } // Create
498
499 //=============================================================================
500 // Function: IsOk
501 // Purpose: returns true if the volume is legal.
502 // Notes: For fixed disks, it must exist. For removable disks, it must also
503 // be present. For Network Shares, it must also be logged in, etc.
504 //=============================================================================
505 bool wxFSVolumeBase::IsOk() const
506 {
507 return m_isOk;
508 } // IsOk
509
510 //=============================================================================
511 // Function: GetKind
512 // Purpose: Return the type of the volume.
513 //=============================================================================
514 wxFSVolumeKind wxFSVolumeBase::GetKind() const
515 {
516 if (!m_isOk)
517 return wxFS_VOL_OTHER;
518
519 FileInfoMap::iterator itr = s_fileInfo.find(m_volName);
520 if (itr == s_fileInfo.end())
521 return wxFS_VOL_OTHER;
522
523 return itr->second.m_type;
524 }
525
526 //=============================================================================
527 // Function: GetFlags
528 // Purpose: Return the caches flags for this volume.
529 // Notes: - Returns -1 if no flags were cached.
530 //=============================================================================
531 int wxFSVolumeBase::GetFlags() const
532 {
533 if (!m_isOk)
534 return -1;
535
536 FileInfoMap::iterator itr = s_fileInfo.find(m_volName);
537 if (itr == s_fileInfo.end())
538 return -1;
539
540 return itr->second.m_flags;
541 } // GetFlags
542
543 #endif // wxUSE_BASE
544
545 // ============================================================================
546 // wxFSVolume
547 // ============================================================================
548
549 #if wxUSE_GUI
550
551 void wxFSVolume::InitIcons()
552 {
553 m_icons.Alloc(wxFS_VOL_ICO_MAX);
554 wxIcon null;
555 for (int idx = 0; idx < wxFS_VOL_ICO_MAX; idx++)
556 m_icons.Add(null);
557 }
558
559 //=============================================================================
560 // Function: GetIcon
561 // Purpose: return the requested icon.
562 //=============================================================================
563
564 wxIcon wxFSVolume::GetIcon(wxFSIconType type) const
565 {
566 wxCHECK_MSG( type >= 0 && (size_t)type < m_icons.GetCount(), wxNullIcon,
567 _T("wxFSIconType::GetIcon(): invalid icon index") );
568
569 // Load on demand.
570 if (m_icons[type].IsNull())
571 {
572 UINT flags = SHGFI_ICON;
573 switch (type)
574 {
575 case wxFS_VOL_ICO_SMALL:
576 flags |= SHGFI_SMALLICON;
577 break;
578
579 case wxFS_VOL_ICO_LARGE:
580 flags |= SHGFI_SHELLICONSIZE;
581 break;
582
583 case wxFS_VOL_ICO_SEL_SMALL:
584 flags |= SHGFI_SMALLICON | SHGFI_OPENICON;
585 break;
586
587 case wxFS_VOL_ICO_SEL_LARGE:
588 flags |= SHGFI_SHELLICONSIZE | SHGFI_OPENICON;
589 break;
590
591 case wxFS_VOL_ICO_MAX:
592 wxFAIL_MSG(_T("wxFS_VOL_ICO_MAX is not valid icon type"));
593 break;
594 }
595
596 SHFILEINFO fi;
597 long rc = SHGetFileInfo(m_volName, 0, &fi, sizeof(fi), flags);
598 m_icons[type].SetHICON((WXHICON)fi.hIcon);
599 if (!rc || !fi.hIcon)
600 wxLogError(_("Cannot load icon from '%s'."), m_volName.c_str());
601 }
602
603 return m_icons[type];
604 } // GetIcon
605
606 #endif // wxUSE_GUI
607
608 #endif // wxUSE_FSVOLUME

  ViewVC Help
Powered by ViewVC 1.1.22