dir_access_windows.cpp 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412
  1. /**************************************************************************/
  2. /* dir_access_windows.cpp */
  3. /**************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /**************************************************************************/
  8. /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
  9. /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /**************************************************************************/
  30. #if defined(WINDOWS_ENABLED)
  31. #include "dir_access_windows.h"
  32. #include "core/os/memory.h"
  33. #include "core/print_string.h"
  34. #include <stdio.h>
  35. #include <wchar.h>
  36. #define WIN32_LEAN_AND_MEAN
  37. #include <windows.h>
  38. /*
  39. [03:57] <reduz> yessopie, so i don't havemak to rely on unicows
  40. [03:58] <yessopie> reduz- yeah, all of the functions fail, and then you can call GetLastError () which will return 120
  41. [03:58] <drumstick> CategoryApl, hehe, what? :)
  42. [03:59] <CategoryApl> didn't Verona lead to some trouble
  43. [03:59] <yessopie> 120 = ERROR_CALL_NOT_IMPLEMENTED
  44. [03:59] <yessopie> (you can use that constant if you include winerr.h)
  45. [03:59] <CategoryApl> well answer with winning a compo
  46. [04:02] <yessopie> if ( SetCurrentDirectoryW ( L"." ) == FALSE && GetLastError () == ERROR_CALL_NOT_IMPLEMENTED ) { use ANSI }
  47. */
  48. struct DirAccessWindowsPrivate {
  49. HANDLE h; //handle for findfirstfile
  50. WIN32_FIND_DATA f;
  51. WIN32_FIND_DATAW fu; //unicode version
  52. };
  53. // CreateFolderAsync
  54. Error DirAccessWindows::list_dir_begin() {
  55. _cisdir = false;
  56. _cishidden = false;
  57. list_dir_end();
  58. p->h = FindFirstFileExW((current_dir + "\\*").c_str(), FindExInfoStandard, &p->fu, FindExSearchNameMatch, NULL, 0);
  59. return (p->h == INVALID_HANDLE_VALUE) ? ERR_CANT_OPEN : OK;
  60. }
  61. String DirAccessWindows::get_next() {
  62. if (p->h == INVALID_HANDLE_VALUE)
  63. return "";
  64. _cisdir = (p->fu.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY);
  65. _cishidden = (p->fu.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN);
  66. String name = p->fu.cFileName;
  67. if (FindNextFileW(p->h, &p->fu) == 0) {
  68. FindClose(p->h);
  69. p->h = INVALID_HANDLE_VALUE;
  70. }
  71. return name;
  72. }
  73. bool DirAccessWindows::current_is_dir() const {
  74. return _cisdir;
  75. }
  76. bool DirAccessWindows::current_is_hidden() const {
  77. return _cishidden;
  78. }
  79. void DirAccessWindows::list_dir_end() {
  80. if (p->h != INVALID_HANDLE_VALUE) {
  81. FindClose(p->h);
  82. p->h = INVALID_HANDLE_VALUE;
  83. }
  84. }
  85. int DirAccessWindows::get_drive_count() {
  86. return drive_count;
  87. }
  88. String DirAccessWindows::get_drive(int p_drive) {
  89. if (p_drive < 0 || p_drive >= drive_count)
  90. return "";
  91. return String::chr(drives[p_drive]) + ":";
  92. }
  93. Error DirAccessWindows::change_dir(String p_dir) {
  94. GLOBAL_LOCK_FUNCTION
  95. p_dir = fix_path(p_dir);
  96. wchar_t real_current_dir_name[2048];
  97. GetCurrentDirectoryW(2048, real_current_dir_name);
  98. String prev_dir = real_current_dir_name;
  99. SetCurrentDirectoryW(current_dir.c_str());
  100. bool worked = (SetCurrentDirectoryW(p_dir.c_str()) != 0);
  101. String base = _get_root_path();
  102. if (base != "") {
  103. GetCurrentDirectoryW(2048, real_current_dir_name);
  104. String new_dir;
  105. new_dir = String(real_current_dir_name).replace("\\", "/");
  106. if (!new_dir.begins_with(base)) {
  107. worked = false;
  108. }
  109. }
  110. if (worked) {
  111. GetCurrentDirectoryW(2048, real_current_dir_name);
  112. current_dir = real_current_dir_name; // TODO, utf8 parser
  113. current_dir = current_dir.replace("\\", "/");
  114. } //else {
  115. SetCurrentDirectoryW(prev_dir.c_str());
  116. //}
  117. return worked ? OK : ERR_INVALID_PARAMETER;
  118. }
  119. Error DirAccessWindows::make_dir(String p_dir) {
  120. GLOBAL_LOCK_FUNCTION
  121. p_dir = fix_path(p_dir);
  122. if (p_dir.is_rel_path())
  123. p_dir = current_dir.plus_file(p_dir);
  124. p_dir = p_dir.simplify_path().replace("/", "\\");
  125. bool success;
  126. int err;
  127. if (!p_dir.is_network_share_path()) {
  128. p_dir = "\\\\?\\" + p_dir;
  129. // Add "\\?\" to the path to extend max. path length past 248, if it's not a network share UNC path.
  130. // See https://msdn.microsoft.com/en-us/library/windows/desktop/aa363855(v=vs.85).aspx
  131. }
  132. success = CreateDirectoryW(p_dir.c_str(), NULL);
  133. err = GetLastError();
  134. if (success) {
  135. return OK;
  136. };
  137. if (err == ERROR_ALREADY_EXISTS || err == ERROR_ACCESS_DENIED) {
  138. return ERR_ALREADY_EXISTS;
  139. };
  140. return ERR_CANT_CREATE;
  141. }
  142. String DirAccessWindows::get_current_dir() {
  143. String base = _get_root_path();
  144. if (base != "") {
  145. String bd = current_dir.replace("\\", "/").replace_first(base, "");
  146. if (bd.begins_with("/"))
  147. return _get_root_string() + bd.substr(1, bd.length());
  148. else
  149. return _get_root_string() + bd;
  150. } else {
  151. }
  152. return current_dir;
  153. }
  154. String DirAccessWindows::get_current_dir_without_drive() {
  155. String dir = get_current_dir();
  156. if (_get_root_string() == "") {
  157. int p = current_dir.find(":");
  158. if (p != -1) {
  159. dir = dir.right(p + 1);
  160. }
  161. }
  162. return dir;
  163. }
  164. bool DirAccessWindows::file_exists(String p_file) {
  165. GLOBAL_LOCK_FUNCTION
  166. if (!p_file.is_abs_path())
  167. p_file = get_current_dir().plus_file(p_file);
  168. p_file = fix_path(p_file);
  169. //p_file.replace("/","\\");
  170. //WIN32_FILE_ATTRIBUTE_DATA fileInfo;
  171. DWORD fileAttr;
  172. fileAttr = GetFileAttributesW(p_file.c_str());
  173. if (INVALID_FILE_ATTRIBUTES == fileAttr)
  174. return false;
  175. return !(fileAttr & FILE_ATTRIBUTE_DIRECTORY);
  176. }
  177. bool DirAccessWindows::dir_exists(String p_dir) {
  178. GLOBAL_LOCK_FUNCTION
  179. if (p_dir.is_rel_path())
  180. p_dir = get_current_dir().plus_file(p_dir);
  181. p_dir = fix_path(p_dir);
  182. //p_dir.replace("/","\\");
  183. //WIN32_FILE_ATTRIBUTE_DATA fileInfo;
  184. DWORD fileAttr;
  185. fileAttr = GetFileAttributesW(p_dir.c_str());
  186. if (INVALID_FILE_ATTRIBUTES == fileAttr)
  187. return false;
  188. return (fileAttr & FILE_ATTRIBUTE_DIRECTORY);
  189. }
  190. Error DirAccessWindows::rename(String p_path, String p_new_path) {
  191. if (p_path.is_rel_path())
  192. p_path = get_current_dir().plus_file(p_path);
  193. p_path = fix_path(p_path);
  194. if (p_new_path.is_rel_path())
  195. p_new_path = get_current_dir().plus_file(p_new_path);
  196. p_new_path = fix_path(p_new_path);
  197. // If we're only changing file name case we need to do a little juggling
  198. if (p_path.to_lower() == p_new_path.to_lower()) {
  199. if (dir_exists(p_path)) {
  200. // The path is a dir; just rename
  201. return ::_wrename(p_path.c_str(), p_new_path.c_str()) == 0 ? OK : FAILED;
  202. }
  203. // The path is a file; juggle
  204. WCHAR tmpfile[MAX_PATH];
  205. if (!GetTempFileNameW(fix_path(get_current_dir()).c_str(), NULL, 0, tmpfile)) {
  206. return FAILED;
  207. }
  208. if (!::ReplaceFileW(tmpfile, p_path.c_str(), NULL, 0, NULL, NULL)) {
  209. DeleteFileW(tmpfile);
  210. return FAILED;
  211. }
  212. return ::_wrename(tmpfile, p_new_path.c_str()) == 0 ? OK : FAILED;
  213. } else {
  214. if (file_exists(p_new_path)) {
  215. if (remove(p_new_path) != OK) {
  216. return FAILED;
  217. }
  218. }
  219. return ::_wrename(p_path.c_str(), p_new_path.c_str()) == 0 ? OK : FAILED;
  220. }
  221. }
  222. Error DirAccessWindows::remove(String p_path) {
  223. if (p_path.is_rel_path())
  224. p_path = get_current_dir().plus_file(p_path);
  225. p_path = fix_path(p_path);
  226. DWORD fileAttr;
  227. fileAttr = GetFileAttributesW(p_path.c_str());
  228. if (INVALID_FILE_ATTRIBUTES == fileAttr)
  229. return FAILED;
  230. if ((fileAttr & FILE_ATTRIBUTE_DIRECTORY))
  231. return ::_wrmdir(p_path.c_str()) == 0 ? OK : FAILED;
  232. else
  233. return ::_wunlink(p_path.c_str()) == 0 ? OK : FAILED;
  234. }
  235. /*
  236. FileType DirAccessWindows::get_file_type(const String& p_file) const {
  237. wchar_t real_current_dir_name[2048];
  238. GetCurrentDirectoryW(2048,real_current_dir_name);
  239. String prev_dir=real_current_dir_name;
  240. bool worked SetCurrentDirectoryW(current_dir.c_str());
  241. DWORD attr;
  242. if (worked) {
  243. WIN32_FILE_ATTRIBUTE_DATA fileInfo;
  244. attr = GetFileAttributesExW(p_file.c_str(), GetFileExInfoStandard, &fileInfo);
  245. }
  246. SetCurrentDirectoryW(prev_dir.c_str());
  247. if (!worked)
  248. return FILE_TYPE_NONE;
  249. return (attr&FILE_ATTRIBUTE_DIRECTORY)?FILE_TYPE_
  250. }
  251. */
  252. uint64_t DirAccessWindows::get_space_left() {
  253. uint64_t bytes = 0;
  254. if (!GetDiskFreeSpaceEx(NULL, (PULARGE_INTEGER)&bytes, NULL, NULL))
  255. return 0;
  256. //this is either 0 or a value in bytes.
  257. return bytes;
  258. }
  259. String DirAccessWindows::get_filesystem_type() const {
  260. String path = fix_path(const_cast<DirAccessWindows *>(this)->get_current_dir());
  261. if (path.is_network_share_path()) {
  262. return "Network Share";
  263. }
  264. int unit_end = path.find(":");
  265. ERR_FAIL_COND_V(unit_end == -1, String());
  266. String unit = path.substr(0, unit_end + 1) + "\\";
  267. WCHAR szVolumeName[100];
  268. WCHAR szFileSystemName[10];
  269. DWORD dwSerialNumber = 0;
  270. DWORD dwMaxFileNameLength = 0;
  271. DWORD dwFileSystemFlags = 0;
  272. if (::GetVolumeInformationW(unit.c_str(),
  273. szVolumeName,
  274. sizeof(szVolumeName),
  275. &dwSerialNumber,
  276. &dwMaxFileNameLength,
  277. &dwFileSystemFlags,
  278. szFileSystemName,
  279. sizeof(szFileSystemName)) == TRUE) {
  280. return String(szFileSystemName);
  281. }
  282. ERR_FAIL_V("");
  283. }
  284. DirAccessWindows::DirAccessWindows() {
  285. p = memnew(DirAccessWindowsPrivate);
  286. p->h = INVALID_HANDLE_VALUE;
  287. current_dir = ".";
  288. drive_count = 0;
  289. #ifdef UWP_ENABLED
  290. Windows::Storage::StorageFolder ^ install_folder = Windows::ApplicationModel::Package::Current->InstalledLocation;
  291. change_dir(install_folder->Path->Data());
  292. #else
  293. DWORD mask = GetLogicalDrives();
  294. for (int i = 0; i < MAX_DRIVES; i++) {
  295. if (mask & (1 << i)) { //DRIVE EXISTS
  296. drives[drive_count] = 'A' + i;
  297. drive_count++;
  298. }
  299. }
  300. change_dir(".");
  301. #endif
  302. }
  303. DirAccessWindows::~DirAccessWindows() {
  304. list_dir_end();
  305. memdelete(p);
  306. }
  307. #endif //windows DirAccess support