1 // Copyright (c) 2016 The WebM project authors. All Rights Reserved. 2 // 3 // Use of this source code is governed by a BSD-style license 4 // that can be found in the LICENSE file in the root of the source 5 // tree. An additional intellectual property rights grant can be found 6 // in the file PATENTS. All contributing project authors may 7 // be found in the AUTHORS file in the root of the source tree. 8 #include "common/file_util.h" 9 10 #include <sys/stat.h> 11 #ifndef _MSC_VER 12 #include <unistd.h> // close() 13 #endif 14 15 #include <cstdio> 16 #include <cstdlib> 17 #include <cstring> 18 #include <fstream> 19 #include <ios> 20 21 namespace libwebm { 22 23 std::string GetTempFileName() { 24 #if !defined _MSC_VER && !defined __MINGW32__ 25 std::string temp_file_name_template_str = 26 std::string(std::getenv("TEST_TMPDIR") ? std::getenv("TEST_TMPDIR") : 27 ".") + 28 "/libwebm_temp.XXXXXX"; 29 char* temp_file_name_template = 30 new char[temp_file_name_template_str.length() + 1]; 31 memset(temp_file_name_template, 0, temp_file_name_template_str.length() + 1); 32 temp_file_name_template_str.copy(temp_file_name_template, 33 temp_file_name_template_str.length(), 0); 34 int fd = mkstemp(temp_file_name_template); 35 std::string temp_file_name = 36 (fd != -1) ? std::string(temp_file_name_template) : std::string(); 37 delete[] temp_file_name_template; 38 if (fd != -1) { 39 close(fd); 40 } 41 return temp_file_name; 42 #else 43 char tmp_file_name[_MAX_PATH]; 44 errno_t err = tmpnam_s(tmp_file_name); 45 if (err == 0) { 46 return std::string(tmp_file_name); 47 } 48 return std::string(); 49 #endif 50 } 51 52 uint64_t GetFileSize(const std::string& file_name) { 53 uint64_t file_size = 0; 54 #ifndef _MSC_VER 55 struct stat st; 56 st.st_size = 0; 57 if (stat(file_name.c_str(), &st) == 0) { 58 #else 59 struct _stat st; 60 st.st_size = 0; 61 if (_stat(file_name.c_str(), &st) == 0) { 62 #endif 63 file_size = st.st_size; 64 } 65 return file_size; 66 } 67 68 TempFileDeleter::TempFileDeleter() { file_name_ = GetTempFileName(); } 69 70 TempFileDeleter::~TempFileDeleter() { 71 std::ifstream file(file_name_.c_str()); 72 if (file.good()) { 73 file.close(); 74 std::remove(file_name_.c_str()); 75 } 76 } 77 78 } // namespace libwebm 79
