Support/FileSystem: Add file_size implementation.

llvm-svn: 120867
This commit is contained in:
Michael J. Spencer 2010-12-04 00:31:48 +00:00
parent 1048e75fb9
commit 818ab4aad4
2 changed files with 35 additions and 0 deletions

View File

@ -263,6 +263,20 @@ error_code equivalent(const Twine &A, const Twine &B, bool &result) {
return make_error_code(errc::success);
}
error_code file_size(const Twine &path, uint64_t &result) {
SmallString<128> path_storage;
StringRef p = path.toNullTerminatedStringRef(path_storage);
struct stat status;
if (::stat(p.begin(), &status) == -1)
return error_code(errno, system_category());
if (!S_ISREG(status.st_mode))
return error_code(EPERM, system_category());
result = status.st_size;
return make_error_code(errc::success);
}
error_code unique_file(const Twine &model, int &result_fd,
SmallVectorImpl<char> &result_path) {
SmallString<128> Model;

View File

@ -406,6 +406,27 @@ error_code equivalent(const Twine &A, const Twine &B, bool &result) {
return make_error_code(errc::success);
}
error_code file_size(const Twine &path, uint64_t &result) {
SmallString<128> path_storage;
SmallVector<wchar_t, 128> path_utf16;
if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
path_utf16))
return ec;
WIN32_FILE_ATTRIBUTE_DATA FileData;
if (!::GetFileAttributesExW(path_utf16.begin(),
::GetFileExInfoStandard,
&FileData))
return make_error_code(windows_error(::GetLastError()));
result =
(uint64_t(FileData.nFileSizeHigh) << (sizeof(FileData.nFileSizeLow) * 8))
+ FileData.nFileSizeLow;
return make_error_code(errc::success);
}
error_code unique_file(const Twine &model, int &result_fd,
SmallVectorImpl<char> &result_path) {
// Use result_path as temp storage.