3 Copyright (C) 2010 celeron55, Perttu Ahola <celeron55@gmail.com>
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License along
16 with this program; if not, write to the Free Software Foundation, Inc.,
17 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
29 #ifdef _WIN32 // WINDOWS
31 #define _WIN32_WINNT 0x0501
39 #define BUFSIZE MAX_PATH
41 std::vector<DirListNode> GetDirListing(std::string pathstring)
43 std::vector<DirListNode> listing;
45 WIN32_FIND_DATA FindFileData;
46 HANDLE hFind = INVALID_HANDLE_VALUE;
51 DirSpec = (LPTSTR) malloc (BUFSIZE);
55 errorstream<<"GetDirListing: Insufficient memory available"<<std::endl;
60 // Check that the input is not larger than allowed.
61 if (pathstring.size() > (BUFSIZE - 2))
63 errorstream<<"GetDirListing: Input directory is too large."<<std::endl;
68 //_tprintf (TEXT("Target directory is %s.\n"), pathstring.c_str());
70 sprintf(DirSpec, "%s", (pathstring + "\\*").c_str());
72 // Find the first file in the directory.
73 hFind = FindFirstFile(DirSpec, &FindFileData);
75 if (hFind == INVALID_HANDLE_VALUE)
77 errorstream<<"GetDirListing: Invalid file handle. Error is "
78 <<GetLastError()<<std::endl;
84 // Be very sure to not include '..' in the results, it will
85 // result in an epic failure when deleting stuff.
88 node.name = FindFileData.cFileName;
89 node.dir = FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
90 if(node.name != "." && node.name != "..")
91 listing.push_back(node);
93 // List all the other files in the directory.
94 while (FindNextFile(hFind, &FindFileData) != 0)
97 node.name = FindFileData.cFileName;
98 node.dir = FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
99 if(node.name != "." && node.name != "..")
100 listing.push_back(node);
103 dwError = GetLastError();
105 if (dwError != ERROR_NO_MORE_FILES)
107 errorstream<<"GetDirListing: FindNextFile error. Error is "
108 <<dwError<<std::endl;
118 if(retval != 0) listing.clear();
120 //for(unsigned int i=0; i<listing.size(); i++){
121 // infostream<<listing[i].name<<(listing[i].dir?" (dir)":" (file)")<<std::endl;
127 bool CreateDir(std::string path)
129 bool r = CreateDirectory(path.c_str(), NULL);
132 if(GetLastError() == ERROR_ALREADY_EXISTS)
137 bool PathExists(std::string path)
139 return (GetFileAttributes(path.c_str()) != INVALID_FILE_ATTRIBUTES);
142 bool RecursiveDelete(std::string path)
144 infostream<<"Removing \""<<path<<"\""<<std::endl;
148 // This silly function needs a double-null terminated string...
149 // Well, we'll just make sure it has at least two, then.
154 sfo.wFunc = FO_DELETE;
155 sfo.pFrom = path.c_str();
157 sfo.fFlags = FOF_SILENT | FOF_NOCONFIRMATION | FOF_NOCONFIRMMKDIR;
159 int r = SHFileOperation(&sfo);
162 errorstream<<"SHFileOperation returned "<<r<<std::endl;
170 #include <sys/types.h>
173 #include <sys/stat.h>
174 #include <sys/wait.h>
176 std::vector<DirListNode> GetDirListing(std::string pathstring)
178 std::vector<DirListNode> listing;
182 if((dp = opendir(pathstring.c_str())) == NULL) {
183 //infostream<<"Error("<<errno<<") opening "<<pathstring<<std::endl;
187 while ((dirp = readdir(dp)) != NULL) {
189 // Be very sure to not include '..' in the results, it will
190 // result in an epic failure when deleting stuff.
191 if(dirp->d_name[0]!='.'){
193 node.name = dirp->d_name;
194 if(node.name == "." || node.name == "..")
197 int isdir = -1; // -1 means unknown
200 POSIX doesn't define d_type member of struct dirent and
201 certain filesystems on glibc/Linux will only return
202 DT_UNKNOWN for the d_type member.
204 Also we don't know whether symlinks are directories or not.
206 #ifdef _DIRENT_HAVE_D_TYPE
207 if(dirp->d_type != DT_UNKNOWN && dirp->d_type != DT_LNK)
208 isdir = (dirp->d_type == DT_DIR);
209 #endif /* _DIRENT_HAVE_D_TYPE */
212 Was d_type DT_UNKNOWN, DT_LNK or nonexistent?
218 if (stat((pathstring + "/" + node.name).c_str(), &statbuf))
220 isdir = ((statbuf.st_mode & S_IFDIR) == S_IFDIR);
223 listing.push_back(node);
231 bool CreateDir(std::string path)
233 int r = mkdir(path.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
240 // If already exists, return true
247 bool PathExists(std::string path)
250 return (stat(path.c_str(),&st) == 0);
253 bool RecursiveDelete(std::string path)
256 Execute the 'rm' command directly, by fork() and execve()
259 infostream<<"Removing \""<<path<<"\""<<std::endl;
263 pid_t child_pid = fork();
268 char argv_data[3][10000];
269 strcpy(argv_data[0], "/bin/rm");
270 strcpy(argv_data[1], "-rf");
271 strncpy(argv_data[2], path.c_str(), 10000);
273 argv[0] = argv_data[0];
274 argv[1] = argv_data[1];
275 argv[2] = argv_data[2];
278 verbosestream<<"Executing '"<<argv[0]<<"' '"<<argv[1]<<"' '"
279 <<argv[2]<<"'"<<std::endl;
281 execv(argv[0], argv);
283 // Execv shouldn't return. Failed.
292 tpid = wait(&child_status);
293 //if(tpid != child_pid) process_terminated(tpid);
294 }while(tpid != child_pid);
295 return (child_status == 0);
301 bool RecursiveDeleteContent(std::string path)
303 infostream<<"Removing content of \""<<path<<"\""<<std::endl;
304 std::vector<DirListNode> list = GetDirListing(path);
305 for(unsigned int i=0; i<list.size(); i++)
307 if(trim(list[i].name) == "." || trim(list[i].name) == "..")
309 std::string childpath = path + DIR_DELIM + list[i].name;
310 bool r = RecursiveDelete(childpath);
313 errorstream<<"Removing \""<<childpath<<"\" failed"<<std::endl;
320 bool CreateAllDirs(std::string path)
324 std::vector<std::string> tocreate;
325 std::string basepath = path;
326 while(!PathExists(basepath))
328 tocreate.push_back(basepath);
329 pos = basepath.rfind(DIR_DELIM_C);
330 if(pos == std::string::npos)
332 basepath = basepath.substr(0,pos);
334 for(int i=tocreate.size()-1;i>=0;i--)
335 if(!CreateDir(tocreate[i]))