| /*============================================================================ |
| CMake - Cross Platform Makefile Generator |
| Copyright 2000-2009 Kitware, Inc., Insight Software Consortium |
| |
| Distributed under the OSI-approved BSD License (the "License"); |
| see accompanying file Copyright.txt for details. |
| |
| This software is distributed WITHOUT ANY WARRANTY; without even the |
| implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. |
| See the License for more information. |
| ============================================================================*/ |
| #include "cmGeneratedFileStream.h" |
| |
| #include "cmSystemTools.h" |
| |
| #if defined(CMAKE_BUILD_WITH_CMAKE) |
| # include <cm_zlib.h> |
| #endif |
| |
| //---------------------------------------------------------------------------- |
| cmGeneratedFileStream::cmGeneratedFileStream(): |
| cmGeneratedFileStreamBase(), Stream() |
| { |
| } |
| |
| //---------------------------------------------------------------------------- |
| cmGeneratedFileStream::cmGeneratedFileStream(const char* name, bool quiet): |
| cmGeneratedFileStreamBase(name), |
| Stream(TempName.c_str()) |
| { |
| // Check if the file opened. |
| if(!*this && !quiet) |
| { |
| cmSystemTools::Error("Cannot open file for write: ", |
| this->TempName.c_str()); |
| cmSystemTools::ReportLastSystemError(""); |
| } |
| } |
| |
| //---------------------------------------------------------------------------- |
| cmGeneratedFileStream::~cmGeneratedFileStream() |
| { |
| // This is the first destructor called. Check the status of the |
| // stream and give the information to the private base. Next the |
| // stream will be destroyed which will close the temporary file. |
| // Finally the base destructor will be called to replace the |
| // destination file. |
| this->Okay = (*this)?true:false; |
| } |
| |
| //---------------------------------------------------------------------------- |
| cmGeneratedFileStream& |
| cmGeneratedFileStream::Open(const char* name, bool quiet, bool binaryFlag) |
| { |
| // Store the file name and construct the temporary file name. |
| this->cmGeneratedFileStreamBase::Open(name); |
| |
| // Open the temporary output file. |
| if ( binaryFlag ) |
| { |
| this->Stream::open(this->TempName.c_str(), |
| std::ios::out | std::ios::binary); |
| } |
| else |
| { |
| this->Stream::open(this->TempName.c_str(), std::ios::out); |
| } |
| |
| // Check if the file opened. |
| if(!*this && !quiet) |
| { |
| cmSystemTools::Error("Cannot open file for write: ", |
| this->TempName.c_str()); |
| cmSystemTools::ReportLastSystemError(""); |
| } |
| return *this; |
| } |
| |
| //---------------------------------------------------------------------------- |
| bool |
| cmGeneratedFileStream::Close() |
| { |
| // Save whether the temporary output file is valid before closing. |
| this->Okay = (*this)?true:false; |
| |
| // Close the temporary output file. |
| this->Stream::close(); |
| |
| // Remove the temporary file (possibly by renaming to the real file). |
| return this->cmGeneratedFileStreamBase::Close(); |
| } |
| |
| //---------------------------------------------------------------------------- |
| void cmGeneratedFileStream::SetCopyIfDifferent(bool copy_if_different) |
| { |
| this->CopyIfDifferent = copy_if_different; |
| } |
| |
| //---------------------------------------------------------------------------- |
| void cmGeneratedFileStream::SetCompression(bool compression) |
| { |
| this->Compress = compression; |
| } |
| |
| //---------------------------------------------------------------------------- |
| void cmGeneratedFileStream::SetCompressionExtraExtension(bool ext) |
| { |
| this->CompressExtraExtension = ext; |
| } |
| |
| //---------------------------------------------------------------------------- |
| cmGeneratedFileStreamBase::cmGeneratedFileStreamBase(): |
| Name(), |
| TempName(), |
| CopyIfDifferent(false), |
| Okay(false), |
| Compress(false), |
| CompressExtraExtension(true) |
| { |
| } |
| |
| //---------------------------------------------------------------------------- |
| cmGeneratedFileStreamBase::cmGeneratedFileStreamBase(const char* name): |
| Name(), |
| TempName(), |
| CopyIfDifferent(false), |
| Okay(false), |
| Compress(false), |
| CompressExtraExtension(true) |
| { |
| this->Open(name); |
| } |
| |
| //---------------------------------------------------------------------------- |
| cmGeneratedFileStreamBase::~cmGeneratedFileStreamBase() |
| { |
| this->Close(); |
| } |
| |
| //---------------------------------------------------------------------------- |
| void cmGeneratedFileStreamBase::Open(const char* name) |
| { |
| // Save the original name of the file. |
| this->Name = name; |
| |
| // Create the name of the temporary file. |
| this->TempName = name; |
| #if defined(__VMS) |
| this->TempName += "_tmp"; |
| #else |
| this->TempName += ".tmp"; |
| #endif |
| |
| // Make sure the temporary file that will be used is not present. |
| cmSystemTools::RemoveFile(this->TempName); |
| |
| std::string dir = cmSystemTools::GetFilenamePath(this->TempName); |
| cmSystemTools::MakeDirectory(dir.c_str()); |
| } |
| |
| //---------------------------------------------------------------------------- |
| bool cmGeneratedFileStreamBase::Close() |
| { |
| bool replaced = false; |
| |
| std::string resname = this->Name; |
| if ( this->Compress && this->CompressExtraExtension ) |
| { |
| resname += ".gz"; |
| } |
| |
| // Only consider replacing the destination file if no error |
| // occurred. |
| if(!this->Name.empty() && |
| this->Okay && |
| (!this->CopyIfDifferent || |
| cmSystemTools::FilesDiffer(this->TempName, resname))) |
| { |
| // The destination is to be replaced. Rename the temporary to the |
| // destination atomically. |
| if ( this->Compress ) |
| { |
| std::string gzname = this->TempName + ".temp.gz"; |
| if ( this->CompressFile(this->TempName.c_str(), gzname.c_str()) ) |
| { |
| this->RenameFile(gzname.c_str(), resname.c_str()); |
| } |
| cmSystemTools::RemoveFile(gzname); |
| } |
| else |
| { |
| this->RenameFile(this->TempName.c_str(), resname.c_str()); |
| } |
| |
| replaced = true; |
| } |
| |
| // Else, the destination was not replaced. |
| // |
| // Always delete the temporary file. We never want it to stay around. |
| cmSystemTools::RemoveFile(this->TempName); |
| |
| return replaced; |
| } |
| |
| //---------------------------------------------------------------------------- |
| #ifdef CMAKE_BUILD_WITH_CMAKE |
| int cmGeneratedFileStreamBase::CompressFile(const char* oldname, |
| const char* newname) |
| { |
| gzFile gf = gzopen(newname, "w"); |
| if ( !gf ) |
| { |
| return 0; |
| } |
| FILE* ifs = cmsys::SystemTools::Fopen(oldname, "r"); |
| if ( !ifs ) |
| { |
| return 0; |
| } |
| size_t res; |
| const size_t BUFFER_SIZE = 1024; |
| char buffer[BUFFER_SIZE]; |
| while ( (res = fread(buffer, 1, BUFFER_SIZE, ifs)) > 0 ) |
| { |
| if ( !gzwrite(gf, buffer, static_cast<int>(res)) ) |
| { |
| fclose(ifs); |
| gzclose(gf); |
| return 0; |
| } |
| } |
| fclose(ifs); |
| gzclose(gf); |
| return 1; |
| } |
| #else |
| int cmGeneratedFileStreamBase::CompressFile(const char*, const char*) |
| { |
| return 0; |
| } |
| #endif |
| |
| //---------------------------------------------------------------------------- |
| int cmGeneratedFileStreamBase::RenameFile(const char* oldname, |
| const char* newname) |
| { |
| return cmSystemTools::RenameFile(oldname, newname); |
| } |
| |
| //---------------------------------------------------------------------------- |
| void cmGeneratedFileStream::SetName(const std::string& fname) |
| { |
| this->Name = fname; |
| } |