039df94e87
Honor TEST_TMPDIR environment variable for creating temporary test files in non-windows platforms. This helps in environments where we cannot create a temporary file in the current directory. Change-Id: I39e0a19580635d752171e0573dd21ec838cb81a4
79 lines
2.2 KiB
C++
79 lines
2.2 KiB
C++
// Copyright (c) 2016 The WebM project authors. All Rights Reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style license
|
|
// that can be found in the LICENSE file in the root of the source
|
|
// tree. An additional intellectual property rights grant can be found
|
|
// in the file PATENTS. All contributing project authors may
|
|
// be found in the AUTHORS file in the root of the source tree.
|
|
#include "common/file_util.h"
|
|
|
|
#include <sys/stat.h>
|
|
#ifndef _MSC_VER
|
|
#include <unistd.h> // close()
|
|
#endif
|
|
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <fstream>
|
|
#include <ios>
|
|
|
|
namespace libwebm {
|
|
|
|
std::string GetTempFileName() {
|
|
#if !defined _MSC_VER && !defined __MINGW32__
|
|
std::string temp_file_name_template_str =
|
|
std::string(std::getenv("TEST_TMPDIR") ? std::getenv("TEST_TMPDIR") :
|
|
".") +
|
|
"/libwebm_temp.XXXXXX";
|
|
char* temp_file_name_template =
|
|
new char[temp_file_name_template_str.length() + 1];
|
|
memset(temp_file_name_template, 0, temp_file_name_template_str.length() + 1);
|
|
temp_file_name_template_str.copy(temp_file_name_template,
|
|
temp_file_name_template_str.length(), 0);
|
|
int fd = mkstemp(temp_file_name_template);
|
|
std::string temp_file_name =
|
|
(fd != -1) ? std::string(temp_file_name_template) : std::string();
|
|
delete[] temp_file_name_template;
|
|
if (fd != -1) {
|
|
close(fd);
|
|
}
|
|
return temp_file_name;
|
|
#else
|
|
char tmp_file_name[_MAX_PATH];
|
|
errno_t err = tmpnam_s(tmp_file_name);
|
|
if (err == 0) {
|
|
return std::string(tmp_file_name);
|
|
}
|
|
return std::string();
|
|
#endif
|
|
}
|
|
|
|
uint64_t GetFileSize(const std::string& file_name) {
|
|
uint64_t file_size = 0;
|
|
#ifndef _MSC_VER
|
|
struct stat st;
|
|
st.st_size = 0;
|
|
if (stat(file_name.c_str(), &st) == 0) {
|
|
#else
|
|
struct _stat st;
|
|
st.st_size = 0;
|
|
if (_stat(file_name.c_str(), &st) == 0) {
|
|
#endif
|
|
file_size = st.st_size;
|
|
}
|
|
return file_size;
|
|
}
|
|
|
|
TempFileDeleter::TempFileDeleter() { file_name_ = GetTempFileName(); }
|
|
|
|
TempFileDeleter::~TempFileDeleter() {
|
|
std::ifstream file(file_name_.c_str());
|
|
if (file.good()) {
|
|
file.close();
|
|
std::remove(file_name_.c_str());
|
|
}
|
|
}
|
|
|
|
} // namespace libwebm
|