dir_access_unix.cpp 9.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418
  1. /*************************************************************************/
  2. /* dir_access_unix.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2017 Godot Engine contributors (cf. AUTHORS.md) */
  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. #include "dir_access_unix.h"
  31. #if defined(UNIX_ENABLED) || defined(LIBC_FILEIO_ENABLED)
  32. #ifndef ANDROID_ENABLED
  33. #include <sys/statvfs.h>
  34. #endif
  35. #include "core/list.h"
  36. #include "os/memory.h"
  37. #include "print_string.h"
  38. #include <errno.h>
  39. #include <stdio.h>
  40. #include <stdlib.h>
  41. #include <string.h>
  42. #ifdef HAVE_MNTENT
  43. #include <mntent.h>
  44. #endif
  45. DirAccess *DirAccessUnix::create_fs() {
  46. return memnew(DirAccessUnix);
  47. }
  48. Error DirAccessUnix::list_dir_begin() {
  49. list_dir_end(); //close any previous dir opening!
  50. //char real_current_dir_name[2048]; //is this enough?!
  51. //getcwd(real_current_dir_name,2048);
  52. //chdir(curent_path.utf8().get_data());
  53. dir_stream = opendir(current_dir.utf8().get_data());
  54. //chdir(real_current_dir_name);
  55. if (!dir_stream)
  56. return ERR_CANT_OPEN; //error!
  57. return OK;
  58. }
  59. bool DirAccessUnix::file_exists(String p_file) {
  60. GLOBAL_LOCK_FUNCTION
  61. if (p_file.is_rel_path())
  62. p_file = current_dir.plus_file(p_file);
  63. p_file = fix_path(p_file);
  64. struct stat flags;
  65. bool success = (stat(p_file.utf8().get_data(), &flags) == 0);
  66. if (success && S_ISDIR(flags.st_mode)) {
  67. success = false;
  68. }
  69. return success;
  70. }
  71. bool DirAccessUnix::dir_exists(String p_dir) {
  72. GLOBAL_LOCK_FUNCTION
  73. if (p_dir.is_rel_path())
  74. p_dir = get_current_dir().plus_file(p_dir);
  75. p_dir = fix_path(p_dir);
  76. struct stat flags;
  77. bool success = (stat(p_dir.utf8().get_data(), &flags) == 0);
  78. if (success && S_ISDIR(flags.st_mode))
  79. return true;
  80. return false;
  81. }
  82. uint64_t DirAccessUnix::get_modified_time(String p_file) {
  83. if (p_file.is_rel_path())
  84. p_file = current_dir.plus_file(p_file);
  85. p_file = fix_path(p_file);
  86. struct stat flags;
  87. bool success = (stat(p_file.utf8().get_data(), &flags) == 0);
  88. if (success) {
  89. return flags.st_mtime;
  90. } else {
  91. ERR_FAIL_V(0);
  92. };
  93. return 0;
  94. };
  95. String DirAccessUnix::get_next() {
  96. if (!dir_stream)
  97. return "";
  98. dirent *entry;
  99. entry = readdir(dir_stream);
  100. if (entry == NULL) {
  101. list_dir_end();
  102. return "";
  103. }
  104. //typedef struct stat Stat;
  105. struct stat flags;
  106. String fname = fix_unicode_name(entry->d_name);
  107. String f = current_dir.plus_file(fname);
  108. if (stat(f.utf8().get_data(), &flags) == 0) {
  109. if (S_ISDIR(flags.st_mode)) {
  110. _cisdir = true;
  111. } else {
  112. _cisdir = false;
  113. }
  114. } else {
  115. _cisdir = false;
  116. }
  117. _cishidden = (fname != "." && fname != ".." && fname.begins_with("."));
  118. return fname;
  119. }
  120. bool DirAccessUnix::current_is_dir() const {
  121. return _cisdir;
  122. }
  123. bool DirAccessUnix::current_is_hidden() const {
  124. return _cishidden;
  125. }
  126. void DirAccessUnix::list_dir_end() {
  127. if (dir_stream)
  128. closedir(dir_stream);
  129. dir_stream = 0;
  130. _cisdir = false;
  131. }
  132. #if defined(HAVE_MNTENT) && defined(X11_ENABLED)
  133. static bool _filter_drive(struct mntent *mnt) {
  134. // Ignore devices that don't point to /dev
  135. if (strncmp(mnt->mnt_fsname, "/dev", 4) != 0) {
  136. return false;
  137. }
  138. // Accept devices mounted at common locations
  139. if (strncmp(mnt->mnt_dir, "/media", 6) == 0 ||
  140. strncmp(mnt->mnt_dir, "/mnt", 4) == 0 ||
  141. strncmp(mnt->mnt_dir, "/home", 5) == 0 ||
  142. strncmp(mnt->mnt_dir, "/run/media", 10) == 0) {
  143. return true;
  144. }
  145. // Ignore everything else
  146. return false;
  147. }
  148. #endif
  149. static void _get_drives(List<String> *list) {
  150. #if defined(HAVE_MNTENT) && defined(X11_ENABLED)
  151. // Check /etc/mtab for the list of mounted partitions
  152. FILE *mtab = setmntent("/etc/mtab", "r");
  153. if (mtab) {
  154. struct mntent mnt;
  155. char strings[4096];
  156. while (getmntent_r(mtab, &mnt, strings, sizeof(strings))) {
  157. if (mnt.mnt_dir != NULL && _filter_drive(&mnt)) {
  158. // Avoid duplicates
  159. if (!list->find(mnt.mnt_dir)) {
  160. list->push_back(mnt.mnt_dir);
  161. }
  162. }
  163. }
  164. endmntent(mtab);
  165. }
  166. #endif
  167. // Add $HOME
  168. const char *home = getenv("HOME");
  169. if (home) {
  170. // Only add if it's not a duplicate
  171. if (!list->find(home)) {
  172. list->push_back(home);
  173. }
  174. // Check $HOME/.config/gtk-3.0/bookmarks
  175. char path[1024];
  176. snprintf(path, 1024, "%s/.config/gtk-3.0/bookmarks", home);
  177. FILE *fd = fopen(path, "r");
  178. if (fd) {
  179. char string[1024];
  180. while (fgets(string, 1024, fd)) {
  181. // Parse only file:// links
  182. if (strncmp(string, "file://", 7) == 0) {
  183. // Strip any unwanted edges on the strings and push_back if it's not a duplicate
  184. String fpath = String(string + 7).strip_edges();
  185. if (!list->find(fpath)) {
  186. list->push_back(fpath);
  187. }
  188. }
  189. }
  190. fclose(fd);
  191. }
  192. }
  193. list->sort();
  194. }
  195. int DirAccessUnix::get_drive_count() {
  196. List<String> list;
  197. _get_drives(&list);
  198. return list.size();
  199. }
  200. String DirAccessUnix::get_drive(int p_drive) {
  201. List<String> list;
  202. _get_drives(&list);
  203. ERR_FAIL_INDEX_V(p_drive, list.size(), "");
  204. return list[p_drive];
  205. }
  206. Error DirAccessUnix::make_dir(String p_dir) {
  207. GLOBAL_LOCK_FUNCTION
  208. if (p_dir.is_rel_path())
  209. p_dir = get_current_dir().plus_file(p_dir);
  210. p_dir = fix_path(p_dir);
  211. bool success = (mkdir(p_dir.utf8().get_data(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH) == 0);
  212. int err = errno;
  213. if (success) {
  214. return OK;
  215. };
  216. if (err == EEXIST) {
  217. return ERR_ALREADY_EXISTS;
  218. };
  219. return ERR_CANT_CREATE;
  220. }
  221. Error DirAccessUnix::change_dir(String p_dir) {
  222. GLOBAL_LOCK_FUNCTION
  223. p_dir = fix_path(p_dir);
  224. // prev_dir is the directory we are changing out of
  225. String prev_dir;
  226. char real_current_dir_name[2048];
  227. getcwd(real_current_dir_name, 2048);
  228. if (prev_dir.parse_utf8(real_current_dir_name))
  229. prev_dir = real_current_dir_name; //no utf8, maybe latin?
  230. // try_dir is the directory we are trying to change into
  231. String try_dir = "";
  232. if (p_dir.is_rel_path()) {
  233. String next_dir = current_dir + "/" + p_dir;
  234. next_dir = next_dir.simplify_path();
  235. try_dir = next_dir;
  236. } else {
  237. try_dir = p_dir;
  238. }
  239. bool worked = (chdir(try_dir.utf8().get_data()) == 0); // we can only give this utf8
  240. if (!worked) {
  241. return ERR_INVALID_PARAMETER;
  242. }
  243. // the directory exists, so set current_dir to try_dir
  244. current_dir = try_dir;
  245. chdir(prev_dir.utf8().get_data());
  246. return OK;
  247. }
  248. String DirAccessUnix::get_current_dir() {
  249. String base = _get_root_path();
  250. if (base != "") {
  251. String bd = current_dir.replace_first(base, "");
  252. if (bd.begins_with("/"))
  253. return _get_root_string() + bd.substr(1, bd.length());
  254. else
  255. return _get_root_string() + bd;
  256. }
  257. return current_dir;
  258. }
  259. Error DirAccessUnix::rename(String p_path, String p_new_path) {
  260. if (p_path.is_rel_path())
  261. p_path = get_current_dir().plus_file(p_path);
  262. p_path = fix_path(p_path);
  263. if (p_new_path.is_rel_path())
  264. p_new_path = get_current_dir().plus_file(p_new_path);
  265. p_new_path = fix_path(p_new_path);
  266. return ::rename(p_path.utf8().get_data(), p_new_path.utf8().get_data()) == 0 ? OK : FAILED;
  267. }
  268. Error DirAccessUnix::remove(String p_path) {
  269. if (p_path.is_rel_path())
  270. p_path = get_current_dir().plus_file(p_path);
  271. p_path = fix_path(p_path);
  272. struct stat flags;
  273. if ((stat(p_path.utf8().get_data(), &flags) != 0))
  274. return FAILED;
  275. if (S_ISDIR(flags.st_mode))
  276. return ::rmdir(p_path.utf8().get_data()) == 0 ? OK : FAILED;
  277. else
  278. return ::unlink(p_path.utf8().get_data()) == 0 ? OK : FAILED;
  279. }
  280. size_t DirAccessUnix::get_space_left() {
  281. #ifndef NO_STATVFS
  282. struct statvfs vfs;
  283. if (statvfs(current_dir.utf8().get_data(), &vfs) != 0) {
  284. return 0;
  285. };
  286. return vfs.f_bfree * vfs.f_bsize;
  287. #else
  288. #warning THIS IS BROKEN
  289. return 0;
  290. #endif
  291. };
  292. DirAccessUnix::DirAccessUnix() {
  293. dir_stream = 0;
  294. _cisdir = false;
  295. /* determine drive count */
  296. // set current directory to an absolute path of the current directory
  297. char real_current_dir_name[2048];
  298. getcwd(real_current_dir_name, 2048);
  299. if (current_dir.parse_utf8(real_current_dir_name))
  300. current_dir = real_current_dir_name;
  301. change_dir(current_dir);
  302. }
  303. DirAccessUnix::~DirAccessUnix() {
  304. list_dir_end();
  305. }
  306. #endif //posix_enabled