experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

PosixFileSystem::CopyFile: Truncate file before overwrite

Browse Source 
Without O_TRUNC, the original content of the target file will be left
over if the src file is shorter.
This commit is contained in:
Takeo Sawada 2019-05-08 15:19:12 +09:00 committed by Takeo Sawada
parent 8a8a109e56
commit 83c4b2792b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/platform/posix/posix_file_system.cc
View File

@ -333,11 +333,13 @@ Status PosixFileSystem::CopyFile(const string& src, const string& target) {
return IOError(src, errno);
}
string translated_target = TranslateName(target);
// O_WRONLY | O_CREAT:
// O_WRONLY | O_CREAT | O_TRUNC:
// Open file for write and if file does not exist, create the file.
// If file exists, truncate its size to 0.
// When creating file, use the same permissions as original
mode_t mode = sbuf.st_mode & (S_IRWXU | S_IRWXG | S_IRWXO);
int target_fd = open(translated_target.c_str(), O_WRONLY | O_CREAT, mode);
int target_fd =
open(translated_target.c_str(), O_WRONLY | O_CREAT | O_TRUNC, mode);
if (target_fd < 0) {
close(src_fd);
return IOError(target, errno);