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 |
41 } // namespace | |
42 | |
43 void ScopedTempDir::Rename() { | |
36 for (int count = 0; count < 50; ++count) { | 44 for (int count = 0; count < 50; ++count) { |
Mark Mentovai
2015/02/11 17:25:50
50 should be a constant local to this file, in the
scottmg
2015/02/11 19:24:55
Done.
| |
37 // Try create a new temporary directory with random generated name. If the | 45 // Try to move to a new temporary directory with a randomly generated name. |
38 // one we generate exists, keep trying another path name until we reach some | 46 // If the one we try exists, retry with a new name until we reach some |
39 // limit. | 47 // limit. |
40 base::string16 new_dir_name = base::UTF8ToUTF16(base::StringPrintf( | 48 base::FilePath target_path = GenerateCandidateName(); |
41 "crashpad.test.%d.%I64x", GetCurrentProcessId(), base::RandUint64())); | 49 if (MoveFileEx(path_.value().c_str(), target_path.value().c_str(), 0)) { |
50 path_ = target_path; | |
51 return; | |
52 } | |
53 } | |
42 | 54 |
43 base::FilePath path_to_create = system_temp_dir.Append(new_dir_name); | 55 CHECK(false) << "Couldn't find temp dir name"; |
56 } | |
57 | |
58 // static | |
59 base::FilePath ScopedTempDir::CreateTemporaryDirectory() { | |
60 for (int count = 0; count < 50; ++count) { | |
Mark Mentovai
2015/02/11 17:25:50
because you use it here too.
scottmg
2015/02/11 19:24:55
Done.
| |
61 // Try to create a new temporary directory with random generated name. If | |
62 // the one we generate exists, keep trying another path name until we reach | |
63 // some limit. | |
64 base::FilePath path_to_create = GenerateCandidateName(); | |
44 if (CreateDirectory(path_to_create.value().c_str(), NULL)) | 65 if (CreateDirectory(path_to_create.value().c_str(), NULL)) |
45 return path_to_create; | 66 return path_to_create; |
46 } | 67 } |
47 | 68 |
48 CHECK(false) << "Couldn't find temp dir name"; | 69 CHECK(false) << "Couldn't find temp dir name"; |
49 return base::FilePath(); | 70 return base::FilePath(); |
50 } | 71 } |
51 | 72 |
52 // static | 73 // static |
53 void ScopedTempDir::RecursivelyDeleteTemporaryDirectory( | 74 void ScopedTempDir::RecursivelyDeleteTemporaryDirectory( |
(...skipping 25 matching lines...) Expand all Loading... | |
79 break; | 100 break; |
80 } | 101 } |
81 } | 102 } |
82 | 103 |
83 EXPECT_TRUE(FindClose(search_handle)); | 104 EXPECT_TRUE(FindClose(search_handle)); |
84 EXPECT_TRUE(RemoveDirectory(path.value().c_str())); | 105 EXPECT_TRUE(RemoveDirectory(path.value().c_str())); |
85 } | 106 } |
86 | 107 |
87 } // namespace test | 108 } // namespace test |
88 } // namespace crashpad | 109 } // namespace crashpad |
OLD | NEW |