OilfieldToolsNet/IFEM
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added: sanity freespace limit in the HDF5 writer to avoid corrupting files

Browse Source 
git-svn-id: http://svn.sintef.no/trondheim/IFEM/trunk@1212 e10b68d5-8a6e-419e-a041-bce267b0401d
This commit is contained in:
akva 2011-09-26 08:58:02 +00:00 committed by Knut Morten Okstad
parent f3954a730b
commit b730f43bdd
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/Utility/HDF5Writer.C
View File

@ -18,6 +18,8 @@
#include <sstream> #include <sstream>
#include <sys/stat.h> #include <sys/stat.h>
#include <numeric> #include <numeric>
#include <sys/statvfs.h>
#include <assert.h>
#ifdef HAS_HDF5 #ifdef HAS_HDF5
#include <hdf5.h> #include <hdf5.h>
@ -27,6 +29,8 @@
#include <mpi.h> #include <mpi.h>
#endif #endif
#define HDF5_SANITY_LIMIT 10*1024*1024LL // 10MB
HDF5Writer::HDF5Writer (const std::string& name, bool append, bool keepOpen) HDF5Writer::HDF5Writer (const std::string& name, bool append, bool keepOpen)
: DataWriter(name+".hdf5"), m_file(0), m_keepOpen(keepOpen) : DataWriter(name+".hdf5"), m_file(0), m_keepOpen(keepOpen)
@ -87,8 +91,21 @@ void HDF5Writer::openFile(int level)
if (m_flag == H5F_ACC_TRUNC) if (m_flag == H5F_ACC_TRUNC)
m_file = H5Fcreate(m_name.c_str(),m_flag,H5P_DEFAULT,acc_tpl); m_file = H5Fcreate(m_name.c_str(),m_flag,H5P_DEFAULT,acc_tpl);
else else {
// check free disk space - to protect against corrupting files
// due to out of space condition
if (m_flag == H5F_ACC_RDWR) {
char* cwd = get_current_dir_name();
struct statvfs vfs;
statvfs(cwd,&vfs);
if (((int64_t)vfs.f_bavail)*vfs.f_bsize < HDF5_SANITY_LIMIT) {
std::cerr << "HDF5Writer: Low disk space detected, bailing" << std::endl;
exit(1);
}
free(cwd);
}
m_file = H5Fopen(m_name.c_str(),m_flag,acc_tpl); m_file = H5Fopen(m_name.c_str(),m_flag,acc_tpl);
}
std::stringstream str; std::stringstream str;
str << '/' << level; str << '/' << level;