Changeset 497 for library/bdm/osutils.cpp

Timestamp:

08/11/09 09:56:48 (15 years ago)

Author:

vbarta

Message:

moved cross-platform directory scanning support from tests to core (osutils)

Files:

• library/bdm/osutils.cpp (modified) (2 diffs)

library/bdm/osutils.cpp

@@ -13 +13 @@
 #include "osutils.h"
 #include <iostream>
+#include <stdexcept>
+#include <string>
 
 #ifdef WIN32
+#include "dirent.h"
 #include <direct.h>
+#define rmdir _rmdir
+#define unlink _unlink
 #else
+#include <dirent.h>
+#include <unistd.h>
 #include <sys/types.h>
 #include <sys/stat.h>
@@ -44 +51 @@
         }
 }
+
+bool remove_all ( const char *path ) {
+        DIR *dir;
+        dirent *de;
+
+        bool rv = true;
+        if ( ( dir = opendir ( path ) ) != 0 ) {
+                try {
+                        std::string top ( path );
+                        top += "/";
+
+                        while ( ( de = readdir ( dir ) ) != 0 ) {
+                                if ( strcmp ( de->d_name, "." ) && strcmp ( de->d_name, ".." ) ) {
+                                        std::string subpath ( top );
+                                        subpath += de->d_name;
+                                        remove_all ( subpath.c_str() );
+                                }
+                        }
+                } catch ( ... ) {
+                        closedir ( dir );
+                        throw;
+                }
+
+                closedir ( dir );
+
+                if ( rmdir ( path ) ) {
+                        std::string msg = "can't remove dir ";
+                        msg += path;
+                        throw std::runtime_error ( msg );
+                }
+        } else {
+                if ( ( errno == ENOTDIR ) || // Linux 
+                     ( errno == EINVAL ) ) { // Windows
+                        if ( unlink ( path ) ) {
+                                std::string msg = "can't remove file ";
+                                msg += path;
+                                throw std::runtime_error ( msg );
+                        }
+                } else {
+                        if ( errno != ENOENT ) {
+                                std::string msg = "can't remove ";
+                                msg += path;
+                                throw std::runtime_error ( msg );
+                        } else {
+                                // it wasn't there in the first place
+                                rv = false;
+                        }
+                }
+        }
+
+        return rv;
+}