michael@0: // Copyright (c) 2011, Google Inc.
michael@0: // All rights reserved.
michael@0: //
michael@0: // Redistribution and use in source and binary forms, with or without
michael@0: // modification, are permitted provided that the following conditions are
michael@0: // met:
michael@0: //
michael@0: //     * Redistributions of source code must retain the above copyright
michael@0: // notice, this list of conditions and the following disclaimer.
michael@0: //     * Redistributions in binary form must reproduce the above
michael@0: // copyright notice, this list of conditions and the following disclaimer
michael@0: // in the documentation and/or other materials provided with the
michael@0: // distribution.
michael@0: //     * Neither the name of Google Inc. nor the names of its
michael@0: // contributors may be used to endorse or promote products derived from
michael@0: // this software without specific prior written permission.
michael@0: //
michael@0: // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
michael@0: // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
michael@0: // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
michael@0: // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
michael@0: // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
michael@0: // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
michael@0: // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
michael@0: // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
michael@0: // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
michael@0: // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
michael@0: // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
michael@0: 
michael@0: // Utility class for creating a temporary directory for unit tests
michael@0: // that is deleted in the destructor.
michael@0: #ifndef GOOGLE_BREAKPAD_COMMON_TESTS_AUTO_TEMPDIR
michael@0: #define GOOGLE_BREAKPAD_COMMON_TESTS_AUTO_TEMPDIR
michael@0: 
michael@0: #include <dirent.h>
michael@0: #include <sys/types.h>
michael@0: 
michael@0: #include <string>
michael@0: 
michael@0: #include "breakpad_googletest_includes.h"
michael@0: #include "common/using_std_string.h"
michael@0: 
michael@0: #if !defined(__ANDROID__)
michael@0: #define TEMPDIR "/tmp"
michael@0: #else
michael@0: #define TEMPDIR "/data/local/tmp"
michael@0: #include "common/android/testing/mkdtemp.h"
michael@0: #endif
michael@0: 
michael@0: namespace google_breakpad {
michael@0: 
michael@0: class AutoTempDir {
michael@0:  public:
michael@0:   AutoTempDir() {
michael@0:     char temp_dir[] = TEMPDIR "/breakpad.XXXXXX";
michael@0:     EXPECT_TRUE(mkdtemp(temp_dir) != NULL);
michael@0:     path_.assign(temp_dir);
michael@0:   }
michael@0: 
michael@0:   ~AutoTempDir() {
michael@0:     DeleteRecursively(path_);
michael@0:   }
michael@0: 
michael@0:   const string& path() const {
michael@0:     return path_;
michael@0:   }
michael@0: 
michael@0:  private:
michael@0:   void DeleteRecursively(const string& path) {
michael@0:     // First remove any files in the dir
michael@0:     DIR* dir = opendir(path.c_str());
michael@0:     if (!dir)
michael@0:       return;
michael@0: 
michael@0:     dirent* entry;
michael@0:     while ((entry = readdir(dir)) != NULL) {
michael@0:       if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0)
michael@0:         continue;
michael@0:       string entry_path = path + "/" + entry->d_name;
michael@0:       struct stat stats;
michael@0:       EXPECT_TRUE(lstat(entry_path.c_str(), &stats) == 0);
michael@0:       if (S_ISDIR(stats.st_mode))
michael@0:         DeleteRecursively(entry_path);
michael@0:       else
michael@0:         EXPECT_TRUE(unlink(entry_path.c_str()) == 0);
michael@0:     }
michael@0:     EXPECT_TRUE(closedir(dir) == 0);
michael@0:     EXPECT_TRUE(rmdir(path.c_str()) == 0);
michael@0:   }
michael@0: 
michael@0:   // prevent copy construction and assignment
michael@0:   AutoTempDir(const AutoTempDir&);
michael@0:   AutoTempDir& operator=(const AutoTempDir&);
michael@0: 
michael@0:   string path_;
michael@0: };
michael@0: 
michael@0: }  // namespace google_breakpad
michael@0: 
michael@0: #endif  // GOOGLE_BREAKPAD_COMMON_TESTS_AUTO_TEMPDIR