src/common/filesys.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/common/filesys.cpp
   3 // Purpose:     wxFileSystem class - interface for opening files
   4 // Author:      Vaclav Slavik
   5 // Copyright:   (c) 1999 Vaclav Slavik
   6 // CVS-ID:      $Id$
   7 // Licence:     wxWindows licence
   8 /////////////////////////////////////////////////////////////////////////////
   9
  10 #include "wx/wxprec.h"
  11
  12 #ifdef __BORLANDC__
  13     #pragma hdrstop
  14 #endif
  15
  16
  17 #if wxUSE_FILESYSTEM
  18
  19 #include "wx/filesys.h"
  20
  21 #ifndef WX_PRECOMP
  22     #include "wx/log.h"
  23     #include "wx/module.h"
  24 #endif
  25
  26 #include "wx/wfstream.h"
  27 #include "wx/mimetype.h"
  28 #include "wx/filename.h"
  29 #include "wx/tokenzr.h"
  30 #include "wx/private/fileback.h"
  31
  32
  33 //--------------------------------------------------------------------------------
  34 // wxFileSystemHandler
  35 //--------------------------------------------------------------------------------
  36
  37 IMPLEMENT_ABSTRACT_CLASS(wxFileSystemHandler, wxObject)
  38
  39
  40 wxString wxFileSystemHandler::GetMimeTypeFromExt(const wxString& location)
  41 {
  42     wxString ext, mime;
  43     wxString loc = GetRightLocation(location);
  44     wxChar c;
  45     int l = loc.length(), l2;
  46
  47     l2 = l;
  48     for (int i = l-1; i >= 0; i--)
  49     {
  50         c = loc[(unsigned int) i];
  51         if ( c == wxT('#') )
  52             l2 = i + 1;
  53         if ( c == wxT('.') )
  54         {
  55             ext = loc.Right(l2-i-1);
  56             break;
  57         }
  58         if ( (c == wxT('/')) || (c == wxT('\\')) || (c == wxT(':')) )
  59             return wxEmptyString;
  60     }
  61
  62 #if wxUSE_MIMETYPE
  63     static bool s_MinimalMimeEnsured = false;
  64     if (!s_MinimalMimeEnsured)
  65     {
  66         static const wxFileTypeInfo fallbacks[] =
  67         {
  68             wxFileTypeInfo(_T("image/jpeg"),
  69                            wxEmptyString,
  70                            wxEmptyString,
  71                            _T("JPEG image (from fallback)"),
  72                            _T("jpg"), _T("jpeg"), _T("JPG"), _T("JPEG"), NULL),
  73             wxFileTypeInfo(_T("image/gif"),
  74                            wxEmptyString,
  75                            wxEmptyString,
  76                            _T("GIF image (from fallback)"),
  77                            _T("gif"), _T("GIF"), NULL),
  78             wxFileTypeInfo(_T("image/png"),
  79                            wxEmptyString,
  80                            wxEmptyString,
  81                            _T("PNG image (from fallback)"),
  82                            _T("png"), _T("PNG"), NULL),
  83             wxFileTypeInfo(_T("image/bmp"),
  84                            wxEmptyString,
  85                            wxEmptyString,
  86                            _T("windows bitmap image (from fallback)"),
  87                            _T("bmp"), _T("BMP"), NULL),
  88             wxFileTypeInfo(_T("text/html"),
  89                            wxEmptyString,
  90                            wxEmptyString,
  91                            _T("HTML document (from fallback)"),
  92                            _T("htm"), _T("html"), _T("HTM"), _T("HTML"), NULL),
  93             // must terminate the table with this!
  94             wxFileTypeInfo()
  95         };
  96         wxTheMimeTypesManager->AddFallbacks(fallbacks);
  97         s_MinimalMimeEnsured = true;
  98     }
  99
 100     wxFileType *ft = wxTheMimeTypesManager->GetFileTypeFromExtension(ext);
 101     if ( !ft || !ft -> GetMimeType(&mime) )
 102     {
 103         mime = wxEmptyString;
 104     }
 105
 106     delete ft;
 107
 108     return mime;
 109 #else
 110     if ( ext.IsSameAs(wxT("htm"), false) || ext.IsSameAs(_T("html"), false) )
 111         return wxT("text/html");
 112     if ( ext.IsSameAs(wxT("jpg"), false) || ext.IsSameAs(_T("jpeg"), false) )
 113         return wxT("image/jpeg");
 114     if ( ext.IsSameAs(wxT("gif"), false) )
 115         return wxT("image/gif");
 116     if ( ext.IsSameAs(wxT("png"), false) )
 117         return wxT("image/png");
 118     if ( ext.IsSameAs(wxT("bmp"), false) )
 119         return wxT("image/bmp");
 120     return wxEmptyString;
 121 #endif
 122 }
 123
 124
 125
 126 wxString wxFileSystemHandler::GetProtocol(const wxString& location) const
 127 {
 128     wxString s = wxEmptyString;
 129     int i, l = location.length();
 130     bool fnd = false;
 131
 132     for (i = l-1; (i >= 0) && ((location[i] != wxT('#')) || (!fnd)); i--) {
 133         if ((location[i] == wxT(':')) && (i != 1 /*win: C:\path*/)) fnd = true;
 134     }
 135     if (!fnd) return wxT("file");
 136     for (++i; (i < l) && (location[i] != wxT(':')); i++) s << location[i];
 137     return s;
 138 }
 139
 140
 141 wxString wxFileSystemHandler::GetLeftLocation(const wxString& location) const
 142 {
 143     int i;
 144     bool fnd = false;
 145
 146     for (i = location.length()-1; i >= 0; i--) {
 147         if ((location[i] == wxT(':')) && (i != 1 /*win: C:\path*/)) fnd = true;
 148         else if (fnd && (location[i] == wxT('#'))) return location.Left(i);
 149     }
 150     return wxEmptyString;
 151 }
 152
 153 wxString wxFileSystemHandler::GetRightLocation(const wxString& location) const
 154 {
 155     int i, l = location.length();
 156     int l2 = l + 1;
 157
 158     for (i = l-1;
 159          (i >= 0) &&
 160          ((location[i] != wxT(':')) || (i == 1) || (location[i-2] == wxT(':')));
 161          i--)
 162     {
 163         if (location[i] == wxT('#')) l2 = i + 1;
 164     }
 165     if (i == 0) return wxEmptyString;
 166     else return location.Mid(i + 1, l2 - i - 2);
 167 }
 168
 169 wxString wxFileSystemHandler::GetAnchor(const wxString& location) const
 170 {
 171     wxChar c;
 172     int l = location.length();
 173
 174     for (int i = l-1; i >= 0; i--) {
 175         c = location[i];
 176         if (c == wxT('#')) return location.Right(l-i-1);
 177         else if ((c == wxT('.')) || (c == wxT('/')) || (c == wxT('\\')) || (c == wxT(':'))) return wxEmptyString;
 178     }
 179     return wxEmptyString;
 180 }
 181
 182
 183 wxString wxFileSystemHandler::FindFirst(const wxString& WXUNUSED(spec),
 184                                         int WXUNUSED(flags))
 185 {
 186     return wxEmptyString;
 187 }
 188
 189 wxString wxFileSystemHandler::FindNext()
 190 {
 191     return wxEmptyString;
 192 }
 193
 194 //--------------------------------------------------------------------------------
 195 // wxLocalFSHandler
 196 //--------------------------------------------------------------------------------
 197
 198
 199 wxString wxLocalFSHandler::ms_root;
 200
 201 bool wxLocalFSHandler::CanOpen(const wxString& location)
 202 {
 203     return GetProtocol(location) == wxT("file");
 204 }
 205
 206 wxFSFile* wxLocalFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location)
 207 {
 208     // location has Unix path separators
 209     wxString right = GetRightLocation(location);
 210     wxFileName fn = wxFileSystem::URLToFileName(right);
 211     wxString fullpath = ms_root + fn.GetFullPath();
 212
 213     if (!wxFileExists(fullpath))
 214         return (wxFSFile*) NULL;
 215
 216     // we need to check whether we can really read from this file, otherwise
 217     // wxFSFile is not going to work
 218 #if wxUSE_FILE
 219     wxFileInputStream *is = new wxFileInputStream(fullpath);
 220 #elif wxUSE_FFILE
 221     wxFFileInputStream *is = new wxFFileInputStream(fullpath);
 222 #else
 223 #error One of wxUSE_FILE or wxUSE_FFILE must be set to 1 for wxFSHandler to work
 224 #endif
 225     if ( !is->Ok() )
 226     {
 227         delete is;
 228         return (wxFSFile*) NULL;
 229     }
 230
 231     return new wxFSFile(is,
 232                         right,
 233                         GetMimeTypeFromExt(location),
 234                         GetAnchor(location)
 235 #if wxUSE_DATETIME
 236                         ,wxDateTime(wxFileModificationTime(fullpath))
 237 #endif // wxUSE_DATETIME
 238                         );
 239 }
 240
 241 wxString wxLocalFSHandler::FindFirst(const wxString& spec, int flags)
 242 {
 243     wxFileName fn = wxFileSystem::URLToFileName(GetRightLocation(spec));
 244     return wxFindFirstFile(ms_root + fn.GetFullPath(), flags);
 245 }
 246
 247 wxString wxLocalFSHandler::FindNext()
 248 {
 249     return wxFindNextFile();
 250 }
 251
 252
 253
 254 //-----------------------------------------------------------------------------
 255 // wxFileSystem
 256 //-----------------------------------------------------------------------------
 257
 258 IMPLEMENT_DYNAMIC_CLASS(wxFileSystem, wxObject)
 259 IMPLEMENT_ABSTRACT_CLASS(wxFSFile, wxObject)
 260
 261
 262 wxList wxFileSystem::m_Handlers;
 263
 264
 265 wxFileSystem::~wxFileSystem()
 266 {
 267     WX_CLEAR_HASH_MAP(wxFSHandlerHash, m_LocalHandlers)
 268 }
 269
 270
 271 static wxString MakeCorrectPath(const wxString& path)
 272 {
 273     wxString p(path);
 274     wxString r;
 275     int i, j, cnt;
 276
 277     cnt = p.length();
 278     for (i = 0; i < cnt; i++)
 279       if (p.GetChar(i) == wxT('\\')) p.GetWritableChar(i) = wxT('/'); // Want to be windows-safe
 280
 281     if (p.Left(2) == wxT("./")) { p = p.Mid(2); cnt -= 2; }
 282
 283     if (cnt < 3) return p;
 284
 285     r << p.GetChar(0) << p.GetChar(1);
 286
 287     // skip trailing ../.., if any
 288     for (i = 2; i < cnt && (p.GetChar(i) == wxT('/') || p.GetChar(i) == wxT('.')); i++) r << p.GetChar(i);
 289
 290     // remove back references: translate dir1/../dir2 to dir2
 291     for (; i < cnt; i++)
 292     {
 293         r << p.GetChar(i);
 294         if (p.GetChar(i) == wxT('/') && p.GetChar(i-1) == wxT('.') && p.GetChar(i-2) == wxT('.'))
 295         {
 296             for (j = r.length() - 2; j >= 0 && r.GetChar(j) != wxT('/') && r.GetChar(j) != wxT(':'); j--) {}
 297             if (j >= 0 && r.GetChar(j) != wxT(':'))
 298             {
 299                 for (j = j - 1; j >= 0 && r.GetChar(j) != wxT('/') && r.GetChar(j) != wxT(':'); j--) {}
 300                 r.Remove(j + 1);
 301             }
 302         }
 303     }
 304
 305     for (; i < cnt; i++) r << p.GetChar(i);
 306
 307     return r;
 308 }
 309
 310
 311 void wxFileSystem::ChangePathTo(const wxString& location, bool is_dir)
 312 {
 313     int i, pathpos = -1;
 314
 315     m_Path = MakeCorrectPath(location);
 316
 317     if (is_dir)
 318     {
 319         if (m_Path.length() > 0 && m_Path.Last() != wxT('/') && m_Path.Last() != wxT(':'))
 320             m_Path << wxT('/');
 321     }
 322
 323     else
 324     {
 325         for (i = m_Path.length()-1; i >= 0; i--)
 326         {
 327             if (m_Path[(unsigned int) i] == wxT('/'))
 328             {
 329                 if ((i > 1) && (m_Path[(unsigned int) (i-1)] == wxT('/')) && (m_Path[(unsigned int) (i-2)] == wxT(':')))
 330                 {
 331                     i -= 2;
 332                     continue;
 333                 }
 334                 else
 335                 {
 336                     pathpos = i;
 337                     break;
 338                 }
 339             }
 340             else if (m_Path[(unsigned int) i] == wxT(':')) {
 341                 pathpos = i;
 342                 break;
 343             }
 344         }
 345         if (pathpos == -1)
 346         {
 347             for (i = 0; i < (int) m_Path.length(); i++)
 348             {
 349                 if (m_Path[(unsigned int) i] == wxT(':'))
 350                 {
 351                     m_Path.Remove(i+1);
 352                     break;
 353                 }
 354             }
 355             if (i == (int) m_Path.length())
 356                 m_Path = wxEmptyString;
 357         }
 358         else
 359         {
 360             m_Path.Remove(pathpos+1);
 361         }
 362     }
 363 }
 364
 365
 366
 367 wxFileSystemHandler *wxFileSystem::MakeLocal(wxFileSystemHandler *h)
 368 {
 369     wxClassInfo *classinfo = h->GetClassInfo();
 370
 371     if (classinfo->IsDynamic())
 372     {
 373         wxFileSystemHandler*& local = m_LocalHandlers[classinfo];
 374         if (!local)
 375             local = (wxFileSystemHandler*)classinfo->CreateObject();
 376         return local;
 377     }
 378     else
 379     {
 380         return h;
 381     }
 382 }
 383
 384
 385
 386 wxFSFile* wxFileSystem::OpenFile(const wxString& location, int flags)
 387 {
 388     if ((flags & wxFS_READ) == 0)
 389         return NULL;
 390
 391     wxString loc = MakeCorrectPath(location);
 392     unsigned i, ln;
 393     wxChar meta;
 394     wxFSFile *s = NULL;
 395     wxList::compatibility_iterator node;
 396
 397     ln = loc.length();
 398     meta = 0;
 399     for (i = 0; i < ln; i++)
 400     {
 401         switch (loc[i])
 402         {
 403             case wxT('/') : case wxT(':') : case wxT('#') :
 404                 meta = loc[i];
 405                 break;
 406         }
 407         if (meta != 0) break;
 408     }
 409     m_LastName = wxEmptyString;
 410
 411     // try relative paths first :
 412     if (meta != wxT(':'))
 413     {
 414         node = m_Handlers.GetFirst();
 415         while (node)
 416         {
 417             wxFileSystemHandler *h = (wxFileSystemHandler*) node -> GetData();
 418             if (h->CanOpen(m_Path + loc))
 419             {
 420                 s = MakeLocal(h)->OpenFile(*this, m_Path + loc);
 421                 if (s) { m_LastName = m_Path + loc; break; }
 422             }
 423             node = node->GetNext();
 424         }
 425     }
 426
 427     // if failed, try absolute paths :
 428     if (s == NULL)
 429     {
 430         node = m_Handlers.GetFirst();
 431         while (node)
 432         {
 433             wxFileSystemHandler *h = (wxFileSystemHandler*) node->GetData();
 434             if (h->CanOpen(loc))
 435             {
 436                 s = MakeLocal(h)->OpenFile(*this, loc);
 437                 if (s) { m_LastName = loc; break; }
 438             }
 439             node = node->GetNext();
 440         }
 441     }
 442
 443     if (s && (flags & wxFS_SEEKABLE) != 0 && !s->GetStream()->IsSeekable())
 444     {
 445         wxBackedInputStream *stream;
 446         stream = new wxBackedInputStream(s->DetachStream());
 447         stream->FindLength();
 448         s->SetStream(stream);
 449     }
 450
 451     return (s);
 452 }
 453
 454
 455
 456 wxString wxFileSystem::FindFirst(const wxString& spec, int flags)
 457 {
 458     wxList::compatibility_iterator node;
 459     wxString spec2(spec);
 460
 461     m_FindFileHandler = NULL;
 462
 463     for (int i = spec2.length()-1; i >= 0; i--)
 464         if (spec2[(unsigned int) i] == wxT('\\')) spec2.GetWritableChar(i) = wxT('/'); // Want to be windows-safe
 465
 466     node = m_Handlers.GetFirst();
 467     while (node)
 468     {
 469         wxFileSystemHandler *h = (wxFileSystemHandler*) node -> GetData();
 470         if (h -> CanOpen(m_Path + spec2))
 471         {
 472             m_FindFileHandler = MakeLocal(h);
 473             return m_FindFileHandler -> FindFirst(m_Path + spec2, flags);
 474         }
 475         node = node->GetNext();
 476     }
 477
 478     node = m_Handlers.GetFirst();
 479     while (node)
 480     {
 481         wxFileSystemHandler *h = (wxFileSystemHandler*) node -> GetData();
 482         if (h -> CanOpen(spec2))
 483         {
 484             m_FindFileHandler = MakeLocal(h);
 485             return m_FindFileHandler -> FindFirst(spec2, flags);
 486         }
 487         node = node->GetNext();
 488     }
 489
 490     return wxEmptyString;
 491 }
 492
 493
 494
 495 wxString wxFileSystem::FindNext()
 496 {
 497     if (m_FindFileHandler == NULL) return wxEmptyString;
 498     else return m_FindFileHandler -> FindNext();
 499 }
 500
 501 bool wxFileSystem::FindFileInPath(wxString *pStr,
 502                                   const wxChar *path,
 503                                   const wxChar *basename)
 504 {
 505     // we assume that it's not empty
 506     wxCHECK_MSG( !wxIsEmpty(basename), false,
 507                 _T("empty file name in wxFileSystem::FindFileInPath"));
 508
 509     // skip path separator in the beginning of the file name if present
 510     if ( wxIsPathSeparator(*basename) )
 511        basename++;
 512
 513     wxStringTokenizer tokenizer(path, wxPATH_SEP);
 514     while ( tokenizer.HasMoreTokens() )
 515     {
 516         wxString strFile = tokenizer.GetNextToken();
 517         if ( !wxEndsWithPathSeparator(strFile) )
 518             strFile += wxFILE_SEP_PATH;
 519         strFile += basename;
 520
 521         wxFSFile *file = OpenFile(strFile);
 522         if ( file )
 523         {
 524             delete file;
 525             *pStr = strFile;
 526             return true;
 527         }
 528     }
 529
 530     return false;
 531 }
 532
 533 void wxFileSystem::AddHandler(wxFileSystemHandler *handler)
 534 {
 535     // prepend the handler to the beginning of the list because handlers added
 536     // last should have the highest priority to allow overriding them
 537     m_Handlers.Insert((size_t)0, handler);
 538 }
 539
 540 wxFileSystemHandler* wxFileSystem::RemoveHandler(wxFileSystemHandler *handler)
 541 {
 542     // if handler has already been removed (or deleted)
 543     // we return NULL. This is by design in case
 544     // CleanUpHandlers() is called before RemoveHandler
 545     // is called, as we cannot control the order
 546     // which modules are unloaded
 547     if (!m_Handlers.DeleteObject(handler))
 548         return NULL;
 549
 550     return handler;
 551 }
 552
 553
 554 bool wxFileSystem::HasHandlerForPath(const wxString &location)
 555 {
 556     for ( wxList::compatibility_iterator node = m_Handlers.GetFirst();
 557            node; node = node->GetNext() )
 558     {
 559         wxFileSystemHandler *h = (wxFileSystemHandler*) node->GetData();
 560         if (h->CanOpen(location))
 561             return true;
 562     }
 563
 564     return false;
 565 }
 566
 567 void wxFileSystem::CleanUpHandlers()
 568 {
 569     WX_CLEAR_LIST(wxList, m_Handlers);
 570 }
 571
 572 static const wxString g_unixPathString(wxT("/"));
 573 static const wxString g_nativePathString(wxFILE_SEP_PATH);
 574
 575 // Returns the native path for a file URL
 576 wxFileName wxFileSystem::URLToFileName(const wxString& url)
 577 {
 578     wxString path = url;
 579
 580     if ( path.Find(wxT("file://")) == 0 )
 581     {
 582         path = path.Mid(7);
 583     }
 584     else if ( path.Find(wxT("file:")) == 0 )
 585     {
 586         path = path.Mid(5);
 587     }
 588     // Remove preceding double slash on Mac Classic
 589 #if defined(__WXMAC__) && !defined(__UNIX__)
 590     else if ( path.Find(wxT("//")) == 0 )
 591         path = path.Mid(2);
 592 #endif
 593
 594     path.Replace(wxT("%25"), wxT("%"));
 595     path.Replace(wxT("%3A"), wxT(":"));
 596
 597 #ifdef __WXMSW__
 598     // file urls either start with a forward slash (local harddisk),
 599     // otherwise they have a servername/sharename notation,
 600     // which only exists on msw and corresponds to a unc
 601     if ( path[0u] == wxT('/') && path [1u] != wxT('/'))
 602     {
 603         path = path.Mid(1);
 604     }
 605     else if ( (url.Find(wxT("file://")) == 0) &&
 606               (path.Find(wxT('/')) != wxNOT_FOUND) &&
 607               (path.length() > 1) && (path[1u] != wxT(':')) )
 608     {
 609         path = wxT("//") + path;
 610     }
 611 #endif
 612
 613     path.Replace(g_unixPathString, g_nativePathString);
 614
 615     return wxFileName(path, wxPATH_NATIVE);
 616 }
 617
 618 // Returns the file URL for a native path
 619 wxString wxFileSystem::FileNameToURL(const wxFileName& filename)
 620 {
 621     wxFileName fn = filename;
 622     fn.Normalize(wxPATH_NORM_DOTS | wxPATH_NORM_TILDE | wxPATH_NORM_ABSOLUTE);
 623     wxString url = fn.GetFullPath(wxPATH_NATIVE);
 624
 625 #ifndef __UNIX__
 626     // unc notation, wxMSW
 627     if ( url.Find(wxT("\\\\")) == 0 )
 628     {
 629         url = wxT("//") + url.Mid(2);
 630     }
 631     else
 632     {
 633         url = wxT("/") + url;
 634 #ifdef __WXMAC__
 635         url = wxT("/") + url;
 636 #endif
 637
 638     }
 639 #endif
 640
 641     url.Replace(g_nativePathString, g_unixPathString);
 642     url.Replace(wxT("%"), wxT("%25"));
 643     url.Replace(wxT(":"), wxT("%3A"));
 644     url = wxT("file:") + url;
 645     return url;
 646 }
 647
 648
 649 ///// Module:
 650
 651 class wxFileSystemModule : public wxModule
 652 {
 653     DECLARE_DYNAMIC_CLASS(wxFileSystemModule)
 654
 655     public:
 656         wxFileSystemModule() :
 657             wxModule(),
 658             m_handler(NULL)
 659         {
 660         }
 661
 662         virtual bool OnInit()
 663         {
 664             m_handler = new wxLocalFSHandler;
 665             wxFileSystem::AddHandler(m_handler);
 666             return true;
 667         }
 668         virtual void OnExit()
 669         {
 670             delete wxFileSystem::RemoveHandler(m_handler);
 671
 672             wxFileSystem::CleanUpHandlers();
 673         }
 674
 675     private:
 676         wxFileSystemHandler* m_handler;
 677
 678 };
 679
 680 IMPLEMENT_DYNAMIC_CLASS(wxFileSystemModule, wxModule)
 681
 682 #endif
 683   // wxUSE_FILESYSTEM