Chromium Code Reviews| Index: util/test/scoped_temp_dir_win.cc |
| diff --git a/util/test/scoped_temp_dir_win.cc b/util/test/scoped_temp_dir_win.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..d1ef6346fb6ba886a2836d118e78ecdfd108b6b8 |
| --- /dev/null |
| +++ b/util/test/scoped_temp_dir_win.cc |
| @@ -0,0 +1,82 @@ |
| +// Copyright 2015 The Crashpad Authors. All rights reserved. |
| +// |
| +// Licensed under the Apache License, Version 2.0 (the "License"); |
| +// you may not use this file except in compliance with the License. |
| +// You may obtain a copy of the License at |
| +// |
| +// http://www.apache.org/licenses/LICENSE-2.0 |
| +// |
| +// Unless required by applicable law or agreed to in writing, software |
| +// distributed under the License is distributed on an "AS IS" BASIS, |
| +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| +// See the License for the specific language governing permissions and |
| +// limitations under the License. |
| + |
| +#include "util/test/scoped_temp_dir.h" |
| + |
| +#include <windows.h> |
| + |
| +#include "base/logging.h" |
| +#include "base/rand_util.h" |
| +#include "base/strings/string16.h" |
| +#include "base/strings/stringprintf.h" |
| +#include "base/strings/utf_string_conversions.h" |
| +#include "gtest/gtest.h" |
| + |
| +namespace crashpad { |
| +namespace test { |
| + |
| +// static |
| +base::FilePath ScopedTempDir::CreateTemporaryDirectory() { |
| + wchar_t temp_path[MAX_PATH + 1]; |
| + DWORD path_len = GetTempPath(MAX_PATH, temp_path); |
| + PCHECK(path_len != 0) << "GetTempPath"; |
| + base::FilePath system_temp_dir(temp_path); |
| + |
| + for (int count = 0; count < 50; ++count) { |
| + // Try create a new temporary directory with random generated name. If the |
| + // one we generate exists, keep trying another path name until we reach some |
| + // limit. |
| + base::string16 new_dir_name = base::UTF8ToUTF16(base::StringPrintf( |
| + "%d_%d", GetCurrentProcessId(), base::RandInt(0, 65536))); |
|
Mark Mentovai
2015/01/09 14:13:47
Can we put something about crashpad in the name so
scottmg
2015/01/13 21:20:56
Done.
|
| + |
| + base::FilePath path_to_create = system_temp_dir.Append(new_dir_name); |
| + if (CreateDirectory(path_to_create.value().c_str(), NULL)) |
| + return path_to_create; |
| + } |
| + |
| + CHECK(false) << "Couldn't find temp dir name"; |
| + return base::FilePath(); |
| +} |
| + |
| +// static |
| +void ScopedTempDir::RecursivelyDeleteTemporaryDirectory( |
| + const base::FilePath& path) { |
| + // SHFILEOPSTRUCT wants the path to be terminated with two NULLs, |
| + // so we have to use wcscpy because wcscpy_s writes non-NULLs |
| + // into the rest of the buffer. |
| + wchar_t double_terminated_path[MAX_PATH + 1] = {0}; |
|
Mark Mentovai
2015/01/09 14:13:47
Because this uses clobber-prone wcscpy, it should
|
| + wcscpy(double_terminated_path, path.value().c_str()); |
| + |
| + SHFILEOPSTRUCT file_operation = {0}; |
| + file_operation.wFunc = FO_DELETE; |
| + file_operation.pFrom = double_terminated_path; |
| + file_operation.fFlags = FOF_NOERRORUI | FOF_SILENT | FOF_NOCONFIRMATION; |
| + int err = SHFileOperation(&file_operation); |
|
Mark Mentovai
2015/01/09 14:13:47
I’m kinda unimpressed by this function expanding w
scottmg
2015/01/13 21:20:56
OK. Mostly was based on ::RemoveDirectory suggesti
|
| + |
| + // Since we're passing flags to the operation telling it to be silent, |
| + // it's possible for the operation to be aborted/cancelled without err |
| + // being set (although MSDN doesn't give any scenarios for how this can |
| + // happen). See MSDN for SHFileOperation and SHFILEOPTSTRUCT. |
| + EXPECT_FALSE(file_operation.fAnyOperationsAborted); |
| + |
| + // Some versions of Windows return ERROR_FILE_NOT_FOUND (0x2) when deleting |
| + // an empty directory and some return 0x402 when they should be returning |
| + // ERROR_FILE_NOT_FOUND. MSDN says Vista and up won't return 0x402. Windows 7 |
| + // can return DE_INVALIDFILES (0x7C) for nonexistent directories. |
| + EXPECT_TRUE(err == 0 || err == ERROR_FILE_NOT_FOUND || err == 0x402 || |
| + err == 0x7C); |
| +} |
| + |
| +} // namespace test |
| +} // namespace crashpad |