1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
//===- unittests/BuildSystem/TempDir.cpp --------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#include "TempDir.h"
#include "llbuild/Basic/FileSystem.h"
#include "llbuild/Basic/PlatformUtility.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/Path.h"
#include "llvm/Support/SourceMgr.h"
#if defined(_WIN32)
#include "llvm/Support/ConvertUTF.h"
#include <windows.h>
#endif
llbuild::TmpDir::TmpDir(llvm::StringRef namePrefix) {
llvm::SmallString<256> tempDirPrefix;
llvm::sys::path::system_temp_directory(true, tempDirPrefix);
llvm::sys::path::append(tempDirPrefix, namePrefix);
std::error_code ec =
llvm::sys::fs::createUniqueDirectory(tempDirPrefix.str(), tempDir);
assert(!ec);
(void)ec;
}
llbuild::TmpDir::~TmpDir() {
#if defined(_WIN32)
// On windows you can't delete a directory is it's your current working
// directory. So if we're in the directory we're trying to delete, move out
// of it first.
wchar_t wCurrPath[MAX_PATH];
GetCurrentDirectoryW(MAX_PATH, wCurrPath);
llvm::SmallVector<llvm::UTF16, 20> wTmpPath;
llvm::convertUTF8ToUTF16String(str(), wTmpPath);
if (lstrcmpW(wCurrPath, (LPCWSTR)wTmpPath.data()) == 0) {
std::string currPath = str();
StringRef parent = llvm::sys::path::parent_path(currPath);
llbuild::basic::sys::chdir(parent.str().c_str());
}
#endif
auto fs = basic::createLocalFileSystem();
bool result = fs->remove(tempDir.c_str());
assert(result);
(void)result;
}
const char *llbuild::TmpDir::c_str() { return tempDir.c_str(); }
std::string llbuild::TmpDir::str() const { return tempDir.str(); }
|
