src/OS/FlashCardEnumerator.hpp

   1 /*
   2 Copyright_License {
   3
   4   XCSoar Glide Computer - http://www.xcsoar.org/
   5   Copyright (C) 2000-2013 The XCSoar Project
   6   A detailed list of copyright holders can be found in the file "AUTHORS".
   7
   8   This program is free software; you can redistribute it and/or
   9   modify it under the terms of the GNU General Public License
  10   as published by the Free Software Foundation; either version 2
  11   of the License, or (at your option) any later version.
  12
  13   This program is distributed in the hope that it will be useful,
  14   but WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16   GNU General Public License for more details.
  17
  18   You should have received a copy of the GNU General Public License
  19   along with this program; if not, write to the Free Software
  20   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  21 }
  22 */
  23
  24 #ifndef XCSOAR_FLASH_CARD_ENUMERATOR_HPP
  25 #define XCSOAR_FLASH_CARD_ENUMERATOR_HPP
  26
  27 #include "Util/StringUtil.hpp"
  28 #include "Compatibility/path.h"
  29
  30 #ifdef HAVE_NOTE_PRJ_DLL
  31 #include "NotePrjDLL.hpp"
  32 #endif
  33
  34 #include <windows.h>
  35
  36 class FlashCardEnumerator {
  37 #ifdef HAVE_NOTE_PRJ_DLL
  38   const NotePrjDLL note_prj;
  39 #endif
  40
  41   HANDLE handle;
  42   WIN32_FIND_DATA data;
  43   bool first;
  44
  45 public:
  46 #ifdef HAVE_NOTE_PRJ_DLL
  47   FlashCardEnumerator()
  48     :handle(note_prj.IsDefined()
  49             ? note_prj.FindFirstFlashCard(&data)
  50             : ::FindFirstFile(_T(DIR_SEPARATOR_S "*"), &data)),
  51      first(true) {}
  52 #else
  53   FlashCardEnumerator()
  54     :handle(::FindFirstFile(_T(DIR_SEPARATOR_S "*"), &data)), first(true) {}
  55 #endif
  56
  57   ~FlashCardEnumerator() {
  58     if (handle != INVALID_HANDLE_VALUE)
  59       ::FindClose(handle);
  60   }
  61
  62   const TCHAR *Next() {
  63     if (handle == INVALID_HANDLE_VALUE)
  64       return NULL;
  65
  66 #ifdef HAVE_NOTE_PRJ_DLL
  67     if (note_prj.IsDefined()) {
  68       if (first)
  69         first = false;
  70       else if (!note_prj.FindNextFlashCard(handle, &data) ||
  71                /* On my Dell Axim x51v, the last entry returned by
  72                   FindNextFlashCard() is empty; workaround: */
  73                StringIsEmpty(data.cFileName))
  74         return NULL;
  75       return data.cFileName;
  76     }
  77 #endif
  78
  79     enum {
  80       /* Directories with the "TEMPORARY" flag set seem to be
  81          mountable flash drives - use this trick on platforms where
  82          note_prj.dll is unavailable */
  83       FLASH = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_TEMPORARY,
  84     };
  85
  86     do {
  87       if (first)
  88         first = false;
  89       else if (!::FindNextFile(handle, &data))
  90         return NULL;
  91     } while ((data.dwFileAttributes & FLASH) != FLASH);
  92
  93     return data.cFileName;
  94   }
  95 };
  96
  97 #endif