xbmc/filesystem/UDFDirectory.cpp

   1 /*
   2  *  Copyright (C) 2010 Team Boxee
   3  *      http://www.boxee.tv
   4  *
   5  *  Copyright (C) 2010-2018 Team Kodi
   6  *  This file is part of Kodi - https://kodi.tv
   7  *
   8  *  SPDX-License-Identifier: GPL-2.0-or-later
   9  *  See LICENSES/README.md for more information.
  10  */
  11
  12 #include "UDFDirectory.h"
  13
  14 #include "FileItem.h"
  15 #include "URL.h"
  16 #include "Util.h"
  17 #include "filesystem/UDFBlockInput.h"
  18 #include "utils/URIUtils.h"
  19
  20 #include <udfread/udfread.h>
  21
  22 using namespace XFILE;
  23
  24 bool CUDFDirectory::GetDirectory(const CURL& url, CFileItemList& items)
  25 {
  26   CURL url2(url);
  27   if (!url2.IsProtocol("udf"))
  28   {
  29     url2.Reset();
  30     url2.SetProtocol("udf");
  31     url2.SetHostName(url.Get());
  32   }
  33
  34   std::string strRoot(url2.Get());
  35   std::string strSub(url2.GetFileName());
  36
  37   URIUtils::AddSlashAtEnd(strRoot);
  38   URIUtils::AddSlashAtEnd(strSub);
  39
  40   auto udf = udfread_init();
  41
  42   if (!udf)
  43     return false;
  44
  45   CUDFBlockInput udfbi;
  46
  47   auto bi = udfbi.GetBlockInput(url2.GetHostName());
  48
  49   if (udfread_open_input(udf, bi) < 0)
  50   {
  51     udfread_close(udf);
  52     return false;
  53   }
  54
  55   auto path = udfread_opendir(udf, strSub.c_str());
  56   if (!path)
  57   {
  58     udfread_close(udf);
  59     return false;
  60   }
  61
  62   struct udfread_dirent dirent;
  63
  64   while (udfread_readdir(path, &dirent))
  65   {
  66     if (dirent.d_type == UDF_DT_DIR)
  67     {
  68       std::string filename = dirent.d_name;
  69       if (filename != "." && filename != "..")
  70       {
  71         CFileItemPtr pItem(new CFileItem(filename));
  72         std::string strDir(strRoot + filename);
  73         URIUtils::AddSlashAtEnd(strDir);
  74         pItem->SetPath(strDir);
  75         pItem->m_bIsFolder = true;
  76
  77         items.Add(pItem);
  78       }
  79     }
  80     else
  81     {
  82       std::string filename = dirent.d_name;
  83       std::string filenameWithPath{strSub + filename};
  84       auto file = udfread_file_open(udf, filenameWithPath.c_str());
  85       if (!file)
  86         continue;
  87
  88       CFileItemPtr pItem(new CFileItem(filename));
  89       pItem->SetPath(strRoot + filename);
  90       pItem->m_bIsFolder = false;
  91       pItem->m_dwSize = udfread_file_size(file);
  92       items.Add(pItem);
  93
  94       udfread_file_close(file);
  95     }
  96   }
  97
  98   udfread_closedir(path);
  99   udfread_close(udf);
 100
 101   return true;
 102 }
 103
 104 bool CUDFDirectory::Exists(const CURL& url)
 105 {
 106   CFileItemList items;
 107   return GetDirectory(url, items);
 108 }