| OLD | NEW |
| 1 // Copyright 2015 The Crashpad Authors. All rights reserved. | 1 // Copyright 2015 The Crashpad Authors. All rights reserved. |
| 2 // | 2 // |
| 3 // Licensed under the Apache License, Version 2.0 (the "License"); | 3 // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 // you may not use this file except in compliance with the License. | 4 // you may not use this file except in compliance with the License. |
| 5 // You may obtain a copy of the License at | 5 // You may obtain a copy of the License at |
| 6 // | 6 // |
| 7 // http://www.apache.org/licenses/LICENSE-2.0 | 7 // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 // | 8 // |
| 9 // Unless required by applicable law or agreed to in writing, software | 9 // Unless required by applicable law or agreed to in writing, software |
| 10 // distributed under the License is distributed on an "AS IS" BASIS, | 10 // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 // See the License for the specific language governing permissions and | 12 // See the License for the specific language governing permissions and |
| 13 // limitations under the License. | 13 // limitations under the License. |
| 14 | 14 |
| 15 #include "util/test/scoped_temp_dir.h" | 15 #include "util/test/scoped_temp_dir.h" |
| 16 | 16 |
| 17 #include <windows.h> | 17 #include <windows.h> |
| 18 | 18 |
| 19 #include "base/logging.h" | 19 #include "base/logging.h" |
| 20 #include "base/rand_util.h" | 20 #include "base/rand_util.h" |
| 21 #include "base/strings/string16.h" | 21 #include "base/strings/string16.h" |
| 22 #include "base/strings/stringprintf.h" | 22 #include "base/strings/stringprintf.h" |
| 23 #include "base/strings/utf_string_conversions.h" | 23 #include "base/strings/utf_string_conversions.h" |
| 24 #include "gtest/gtest.h" | 24 #include "gtest/gtest.h" |
| 25 | 25 |
| 26 namespace crashpad { | 26 namespace crashpad { |
| 27 namespace test { | 27 namespace test { |
| 28 | 28 |
| 29 // static | 29 namespace { |
| 30 base::FilePath ScopedTempDir::CreateTemporaryDirectory() { | 30 |
| 31 base::FilePath GenerateCandidateName() { |
| 31 wchar_t temp_path[MAX_PATH + 1]; | 32 wchar_t temp_path[MAX_PATH + 1]; |
| 32 DWORD path_len = GetTempPath(MAX_PATH, temp_path); | 33 DWORD path_len = GetTempPath(MAX_PATH, temp_path); |
| 33 PCHECK(path_len != 0) << "GetTempPath"; | 34 PCHECK(path_len != 0) << "GetTempPath"; |
| 34 base::FilePath system_temp_dir(temp_path); | 35 base::FilePath system_temp_dir(temp_path); |
| 36 base::string16 new_dir_name = base::UTF8ToUTF16(base::StringPrintf( |
| 37 "crashpad.test.%d.%I64x", GetCurrentProcessId(), base::RandUint64())); |
| 38 return system_temp_dir.Append(new_dir_name); |
| 39 } |
| 35 | 40 |
| 36 for (int count = 0; count < 50; ++count) { | 41 const int kRetries = 50; |
| 37 // Try create a new temporary directory with random generated name. If the | 42 |
| 38 // one we generate exists, keep trying another path name until we reach some | 43 } // namespace |
| 44 |
| 45 void ScopedTempDir::Rename() { |
| 46 for (int count = 0; count < kRetries; ++count) { |
| 47 // Try to move to a new temporary directory with a randomly generated name. |
| 48 // If the one we try exists, retry with a new name until we reach some |
| 39 // limit. | 49 // limit. |
| 40 base::string16 new_dir_name = base::UTF8ToUTF16(base::StringPrintf( | 50 base::FilePath target_path = GenerateCandidateName(); |
| 41 "crashpad.test.%d.%I64x", GetCurrentProcessId(), base::RandUint64())); | 51 if (MoveFileEx(path_.value().c_str(), target_path.value().c_str(), 0)) { |
| 52 path_ = target_path; |
| 53 return; |
| 54 } |
| 55 } |
| 42 | 56 |
| 43 base::FilePath path_to_create = system_temp_dir.Append(new_dir_name); | 57 CHECK(false) << "Couldn't find temp dir name"; |
| 58 } |
| 59 |
| 60 // static |
| 61 base::FilePath ScopedTempDir::CreateTemporaryDirectory() { |
| 62 for (int count = 0; count < kRetries; ++count) { |
| 63 // Try to create a new temporary directory with random generated name. If |
| 64 // the one we generate exists, keep trying another path name until we reach |
| 65 // some limit. |
| 66 base::FilePath path_to_create = GenerateCandidateName(); |
| 44 if (CreateDirectory(path_to_create.value().c_str(), NULL)) | 67 if (CreateDirectory(path_to_create.value().c_str(), NULL)) |
| 45 return path_to_create; | 68 return path_to_create; |
| 46 } | 69 } |
| 47 | 70 |
| 48 CHECK(false) << "Couldn't find temp dir name"; | 71 CHECK(false) << "Couldn't find temp dir name"; |
| 49 return base::FilePath(); | 72 return base::FilePath(); |
| 50 } | 73 } |
| 51 | 74 |
| 52 // static | 75 // static |
| 53 void ScopedTempDir::RecursivelyDeleteTemporaryDirectory( | 76 void ScopedTempDir::RecursivelyDeleteTemporaryDirectory( |
| (...skipping 25 matching lines...) Expand all Loading... |
| 79 break; | 102 break; |
| 80 } | 103 } |
| 81 } | 104 } |
| 82 | 105 |
| 83 EXPECT_TRUE(FindClose(search_handle)); | 106 EXPECT_TRUE(FindClose(search_handle)); |
| 84 EXPECT_TRUE(RemoveDirectory(path.value().c_str())); | 107 EXPECT_TRUE(RemoveDirectory(path.value().c_str())); |
| 85 } | 108 } |
| 86 | 109 |
| 87 } // namespace test | 110 } // namespace test |
| 88 } // namespace crashpad | 111 } // namespace crashpad |
| OLD | NEW |