1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/toolkit/crashreporter/google-breakpad/src/common/tests/auto_tempdir.h Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,100 @@
1.4 +// Copyright (c) 2011, Google Inc.
1.5 +// All rights reserved.
1.6 +//
1.7 +// Redistribution and use in source and binary forms, with or without
1.8 +// modification, are permitted provided that the following conditions are
1.9 +// met:
1.10 +//
1.11 +// * Redistributions of source code must retain the above copyright
1.12 +// notice, this list of conditions and the following disclaimer.
1.13 +// * Redistributions in binary form must reproduce the above
1.14 +// copyright notice, this list of conditions and the following disclaimer
1.15 +// in the documentation and/or other materials provided with the
1.16 +// distribution.
1.17 +// * Neither the name of Google Inc. nor the names of its
1.18 +// contributors may be used to endorse or promote products derived from
1.19 +// this software without specific prior written permission.
1.20 +//
1.21 +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
1.22 +// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
1.23 +// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
1.24 +// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
1.25 +// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
1.26 +// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
1.27 +// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
1.28 +// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
1.29 +// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
1.30 +// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
1.31 +// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
1.32 +
1.33 +// Utility class for creating a temporary directory for unit tests
1.34 +// that is deleted in the destructor.
1.35 +#ifndef GOOGLE_BREAKPAD_COMMON_TESTS_AUTO_TEMPDIR
1.36 +#define GOOGLE_BREAKPAD_COMMON_TESTS_AUTO_TEMPDIR
1.37 +
1.38 +#include <dirent.h>
1.39 +#include <sys/types.h>
1.40 +
1.41 +#include <string>
1.42 +
1.43 +#include "breakpad_googletest_includes.h"
1.44 +#include "common/using_std_string.h"
1.45 +
1.46 +#if !defined(__ANDROID__)
1.47 +#define TEMPDIR "/tmp"
1.48 +#else
1.49 +#define TEMPDIR "/data/local/tmp"
1.50 +#include "common/android/testing/mkdtemp.h"
1.51 +#endif
1.52 +
1.53 +namespace google_breakpad {
1.54 +
1.55 +class AutoTempDir {
1.56 + public:
1.57 + AutoTempDir() {
1.58 + char temp_dir[] = TEMPDIR "/breakpad.XXXXXX";
1.59 + EXPECT_TRUE(mkdtemp(temp_dir) != NULL);
1.60 + path_.assign(temp_dir);
1.61 + }
1.62 +
1.63 + ~AutoTempDir() {
1.64 + DeleteRecursively(path_);
1.65 + }
1.66 +
1.67 + const string& path() const {
1.68 + return path_;
1.69 + }
1.70 +
1.71 + private:
1.72 + void DeleteRecursively(const string& path) {
1.73 + // First remove any files in the dir
1.74 + DIR* dir = opendir(path.c_str());
1.75 + if (!dir)
1.76 + return;
1.77 +
1.78 + dirent* entry;
1.79 + while ((entry = readdir(dir)) != NULL) {
1.80 + if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0)
1.81 + continue;
1.82 + string entry_path = path + "/" + entry->d_name;
1.83 + struct stat stats;
1.84 + EXPECT_TRUE(lstat(entry_path.c_str(), &stats) == 0);
1.85 + if (S_ISDIR(stats.st_mode))
1.86 + DeleteRecursively(entry_path);
1.87 + else
1.88 + EXPECT_TRUE(unlink(entry_path.c_str()) == 0);
1.89 + }
1.90 + EXPECT_TRUE(closedir(dir) == 0);
1.91 + EXPECT_TRUE(rmdir(path.c_str()) == 0);
1.92 + }
1.93 +
1.94 + // prevent copy construction and assignment
1.95 + AutoTempDir(const AutoTempDir&);
1.96 + AutoTempDir& operator=(const AutoTempDir&);
1.97 +
1.98 + string path_;
1.99 +};
1.100 +
1.101 +} // namespace google_breakpad
1.102 +
1.103 +#endif // GOOGLE_BREAKPAD_COMMON_TESTS_AUTO_TEMPDIR
