xbmc/platform/win32/filesystem/Win32SMBFile.cpp

   1 /*
   2  *  Copyright (C) 2014-2018 Team Kodi
   3  *  This file is part of Kodi - https://kodi.tv
   4  *
   5  *  SPDX-License-Identifier: GPL-2.0-or-later
   6  *  See LICENSES/README.md for more information.
   7  */
   8
   9 #include "Win32SMBFile.h"
  10
  11 #include "URL.h"
  12 #include "Win32SMBDirectory.h"
  13
  14 #include "platform/win32/WIN32Util.h"
  15
  16 #include <cassert>
  17
  18 #include <Windows.h>
  19
  20 using namespace XFILE;
  21
  22 // local helper
  23 static bool worthTryToConnect(const DWORD lastErr)
  24 {
  25   return lastErr != ERROR_INVALID_DATA && // used to indicate internal errors
  26          lastErr != ERROR_FILE_NOT_FOUND      && lastErr != ERROR_BAD_NET_NAME  &&
  27          lastErr != ERROR_NO_NET_OR_BAD_PATH  && lastErr != ERROR_NO_NETWORK    &&
  28          lastErr != ERROR_BAD_NETPATH;
  29 }
  30
  31
  32 bool CWin32SMBFile::ConnectAndAuthenticate(const CURL& url)
  33 {
  34   CURL authUrl(url);
  35   CWin32SMBDirectory smbDir;
  36   return smbDir.ConnectAndAuthenticate(authUrl, false);
  37 }
  38
  39 CWin32SMBFile::CWin32SMBFile() : CWin32File(true)
  40 { }
  41
  42 CWin32SMBFile::~CWin32SMBFile()
  43 { /* cleanup by CWin32File destructor */ }
  44
  45 bool CWin32SMBFile::Open(const CURL& url)
  46 {
  47   assert(url.IsProtocol("smb")); // function suitable only for SMB files
  48   if (CWin32File::Open(url))
  49     return true;
  50
  51   return worthTryToConnect(m_lastSMBFileErr) && ConnectAndAuthenticate(url) && CWin32File::Open(url);
  52 }
  53
  54 bool CWin32SMBFile::OpenForWrite(const CURL& url, bool bOverWrite /*= false*/)
  55 {
  56   assert(url.IsProtocol("smb")); // function suitable only for SMB files
  57   if (CWin32File::OpenForWrite(url, bOverWrite))
  58     return true;
  59
  60   return worthTryToConnect(m_lastSMBFileErr) && ConnectAndAuthenticate(url) && CWin32File::OpenForWrite(url, bOverWrite);
  61 }
  62
  63 bool CWin32SMBFile::Delete(const CURL& url)
  64 {
  65   assert(url.IsProtocol("smb")); // function suitable only for SMB files
  66
  67   if (CWin32File::Delete(url))
  68     return true;
  69
  70   return worthTryToConnect(m_lastSMBFileErr) && ConnectAndAuthenticate(url) && CWin32File::Delete(url);
  71 }
  72
  73 bool CWin32SMBFile::Rename(const CURL& urlCurrentName, const CURL& urlNewName)
  74 {
  75   assert(urlCurrentName.IsProtocol("smb")); // function suitable only for SMB files
  76   assert(urlNewName.IsProtocol("smb")); // function suitable only for SMB files
  77
  78   if (CWin32File::Rename(urlCurrentName, urlNewName))
  79     return true;
  80
  81   return worthTryToConnect(m_lastSMBFileErr) && ConnectAndAuthenticate(urlCurrentName) && ConnectAndAuthenticate(urlNewName) &&
  82     CWin32File::Rename(urlCurrentName, urlNewName);
  83 }
  84
  85 bool CWin32SMBFile::SetHidden(const CURL& url, bool hidden)
  86 {
  87   assert(url.IsProtocol("smb")); // function suitable only for SMB files
  88
  89   std::wstring pathnameW(CWIN32Util::ConvertPathToWin32Form(url));
  90   if (pathnameW.empty())
  91     return false;
  92
  93   DWORD attrs = GetFileAttributesW(pathnameW.c_str());
  94   if (attrs == INVALID_FILE_ATTRIBUTES)
  95   {
  96     if (!worthTryToConnect(GetLastError()) || !ConnectAndAuthenticate(url))
  97       return false;
  98     attrs = GetFileAttributesW(pathnameW.c_str());
  99     if (attrs == INVALID_FILE_ATTRIBUTES)
 100       return false;
 101   }
 102
 103   if ((attrs & FILE_ATTRIBUTE_DIRECTORY) != 0)
 104     return false;
 105
 106   // check whether attribute is already set/cleared
 107   if (((attrs & FILE_ATTRIBUTE_HIDDEN) != 0) == hidden)
 108     return true;
 109
 110   if (hidden)
 111     return SetFileAttributesW(pathnameW.c_str(), attrs | FILE_ATTRIBUTE_HIDDEN) != 0;
 112   return SetFileAttributesW(pathnameW.c_str(), attrs & ~FILE_ATTRIBUTE_HIDDEN) != 0;
 113 }
 114
 115 bool CWin32SMBFile::Exists(const CURL& url)
 116 {
 117   assert(url.IsProtocol("smb")); // function suitable only for SMB files
 118   std::wstring pathnameW(CWIN32Util::ConvertPathToWin32Form(url));
 119   if (pathnameW.empty())
 120     return false;
 121
 122   DWORD attrs = GetFileAttributesW(pathnameW.c_str());
 123   if (attrs == INVALID_FILE_ATTRIBUTES)
 124   {
 125     if (!worthTryToConnect(GetLastError()) || !ConnectAndAuthenticate(url))
 126       return false;
 127     attrs = GetFileAttributesW(pathnameW.c_str());
 128     if (attrs == INVALID_FILE_ATTRIBUTES)
 129       return false;
 130   }
 131   return (attrs & FILE_ATTRIBUTE_DIRECTORY) == 0;
 132 }
 133
 134 int CWin32SMBFile::Stat(const CURL& url, struct __stat64* statData)
 135 {
 136   assert(url.IsProtocol("smb")); // function suitable only for SMB files
 137
 138   if (CWin32File::Stat(url, statData) == 0)
 139     return 0;
 140
 141   if (!worthTryToConnect(m_lastSMBFileErr) || !ConnectAndAuthenticate(url))
 142     return -1;
 143
 144   return CWin32File::Stat(url, statData);
 145 }