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.
28 #ifdef _WIN32 // WINDOWS
30 #define _WIN32_WINNT 0x0501
38 #define BUFSIZE MAX_PATH
40 std::vector<DirListNode> GetDirListing(std::string pathstring)
42 std::vector<DirListNode> listing;
44 WIN32_FIND_DATA FindFileData;
45 HANDLE hFind = INVALID_HANDLE_VALUE;
50 DirSpec = (LPTSTR) malloc (BUFSIZE);
54 printf( "Insufficient memory available\n" );
59 // Check that the input is not larger than allowed.
60 if (pathstring.size() > (BUFSIZE - 2))
62 _tprintf(TEXT("Input directory is too large.\n"));
67 //_tprintf (TEXT("Target directory is %s.\n"), pathstring.c_str());
69 sprintf(DirSpec, "%s", (pathstring + "\\*").c_str());
71 // Find the first file in the directory.
72 hFind = FindFirstFile(DirSpec, &FindFileData);
74 if (hFind == INVALID_HANDLE_VALUE)
76 _tprintf (TEXT("Invalid file handle. Error is %u.\n"),
83 // Be very sure to not include '..' in the results, it will
84 // result in an epic failure when deleting stuff.
87 node.name = FindFileData.cFileName;
88 node.dir = FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
89 if(node.name != "." && node.name != "..")
90 listing.push_back(node);
92 // List all the other files in the directory.
93 while (FindNextFile(hFind, &FindFileData) != 0)
96 node.name = FindFileData.cFileName;
97 node.dir = FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
98 if(node.name != "." && node.name != "..")
99 listing.push_back(node);
102 dwError = GetLastError();
104 if (dwError != ERROR_NO_MORE_FILES)
106 _tprintf (TEXT("FindNextFile error. Error is %u.\n"),
117 if(retval != 0) listing.clear();
119 //for(unsigned int i=0; i<listing.size(); i++){
120 // std::cout<<listing[i].name<<(listing[i].dir?" (dir)":" (file)")<<std::endl;
126 bool CreateDir(std::string path)
128 bool r = CreateDirectory(path.c_str(), NULL);
131 if(GetLastError() == ERROR_ALREADY_EXISTS)
136 bool PathExists(std::string path)
138 return (GetFileAttributes(path.c_str()) != INVALID_FILE_ATTRIBUTES);
141 bool RecursiveDelete(std::string path)
143 std::cerr<<"Removing \""<<path<<"\""<<std::endl;
147 // This silly function needs a double-null terminated string...
148 // Well, we'll just make sure it has at least two, then.
153 sfo.wFunc = FO_DELETE;
154 sfo.pFrom = path.c_str();
156 sfo.fFlags = FOF_SILENT | FOF_NOCONFIRMATION | FOF_NOCONFIRMMKDIR;
158 int r = SHFileOperation(&sfo);
161 std::cerr<<"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 //std::cout<<"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 std::cerr<<"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 std::cerr<<"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 std::cerr<<"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 std::cerr<<"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 CreateDir(tocreate[i]);